1 files changed, 12 insertions, 4 deletions
diff --git a/interactioncanvas.h b/interactioncanvas.h
index e380d47..397865d 100644
--- a/interactioncanvas.h
+++ b/interactioncanvas.h
@@ -9,14 +9,21 @@
 #include <QPainter>
 
 #include "line.h"
+#include "rectangle.h"
+#include "ellipse.h"
+#include "polygon.h"
 
 class InteractionCanvas : public QLabel
 {
 public:
-    InteractionCanvas();
+    InteractionCanvas(QWidget *parent);
     ~InteractionCanvas();
 
-    // QWidget interface
+    void addLine();
+    void addRectangle();
+    void addEllipse();
+    void addPolygon();
+
 protected:
     void mouseMoveEvent(QMouseEvent *mouseEvent);
     void mousePressEvent(QMouseEvent *mouseEvent);
@@ -26,14 +33,15 @@ protected:
 
 private:
     void changeSelectedLine(Line *newSelectedLine);
-    QList<Line> m_Lines;
+    void addNewLine(Line *newLine);
+
+    QList<Line *> m_Lines;
     Line *m_NewLine;
     Line *m_SelectedLine;
     QPoint m_MousePressPoint;
     QPoint m_MouseReleasePoint;
     QPoint m_LastMousePosition;
 
-    // QWidget interface
 protected:
     void paintEvent(QPaintEvent *paintEvent);
 };