1 files changed, 7 insertions, 6 deletions
diff --git a/src/curveeditor/curveeditor.h b/src/curveeditor/curveeditor.h
index f5e1af8..a2c5873 100644
--- a/src/curveeditor/curveeditor.h
+++ b/src/curveeditor/curveeditor.h
@@ -27,12 +27,11 @@
 
 #include <QWidget>
 
-class QTreeView;
-
 namespace DesignTools {
 
 class CurveEditorModel;
-class CurveEditorView;
+class GraphicsView;
+class TreeView;
 
 class CurveEditor : public QWidget
 {
@@ -41,12 +40,14 @@ class CurveEditor : public QWidget
 public:
     CurveEditor(CurveEditorModel *model, QWidget *parent = nullptr);
 
-    CurveEditorView *view() const;
+    void zoomX(double zoom);
+
+    void zoomY(double zoom);
 
 private:
-    QTreeView *m_tree;
+    TreeView *m_tree;
 
-    CurveEditorView *m_view;
+    GraphicsView *m_view;
 };
 
 } // End namespace DesignTools.