refactor drawing code

1 files changed, 5 insertions, 1 deletions

diff --git a/src/drone_controller.hh b/src/drone_controller.hh
index 08dac42..bcf1a1a 100644
--- a/src/drone_controller.hh
+++ b/src/drone_controller.hh
@@ -3,11 +3,14 @@
 
 #include "drone.hh"
 
+#include "opengl_widget.hh"
+
 #include <QJsonObject>
 #include <QTimer>
+#include <QOpenGLExtraFunctions>
 
 
-class DroneController : public QObject {
+class DroneController : public QObject, public Painter {
 	Q_OBJECT
 
 	int framerate;
@@ -22,6 +25,7 @@ class DroneController : public QObject {
 public:
 	DroneController(const QJsonObject &json);
 	int getDuration() const;
+	void draw(QOpenGLExtraFunctions *f) const;
 
 signals:
 	void frameChanged(int frame);