1 files changed, 9 insertions, 2 deletions
diff --git a/qt-ui/diveplanner.h b/qt-ui/diveplanner.h
index 7dca3bb85..33f647fe6 100644
--- a/qt-ui/diveplanner.h
+++ b/qt-ui/diveplanner.h
@@ -4,6 +4,13 @@
 #include <QGraphicsView>
 #include <QGraphicsPathItem>
 
+class DiveHandler : public QGraphicsEllipseItem{
+public:
+    DiveHandler();
+
+	QGraphicsLineItem *from;
+	QGraphicsLineItem *to;
+};
 class DivePlanner : public QGraphicsView {
 	Q_OBJECT
 public:
@@ -16,11 +23,11 @@ protected:
     void clear_generated_deco();
 	void create_deco_stop();
 	bool isPointOutOfBoundaries(QPointF point);
-	
+
 private:
     DivePlanner(QWidget* parent = 0);
 	QList<QGraphicsLineItem*> lines;
-	QList<QGraphicsEllipseItem*> handles;
+	QList<DiveHandler *> handles;
 	QGraphicsLineItem *verticalLine;
 	QGraphicsLineItem *horizontalLine;
 };