Notaði fyrrum p_controller og bætti við virtual pi_controller

include/controller.h

@@ -0,0 +1,18 @@
+#ifndef CONTROLLER_H
+#define CONTROLLER_H
+
+class Controller
+{
+public:
+    virtual ~Controller()
+    {
+    }
+
+    virtual double update(double ref, double actual) = 0;
+
+    virtual void reset()
+    {
+    }
+};
+
+#endif

include/initialization_state.h

@@ -7,7 +7,7 @@
 class InitializationState : public State
 {
 private:
-    static const unsigned long WAIT_MS = 2000;
+    static const unsigned long WAIT_MS = 2000UL;
 
     unsigned long enter_time_ms = 0;
 

include/motor_controller.h

@@ -4,7 +4,8 @@
 #include "rotary_encoder_driver.h"
 #include "motor_driver.h"
 //#include "p_controller.h"
-#include "pi_controller.h"
+//#include "pi_controller.h"
+#include "controller.h"
 #include "digital_out.h"
 
 class State;
@@ -26,7 +27,8 @@ private:
     RotaryEncoderDriver& encoder;
     MotorDriver& motor;
     //P_controller& controller;
-    PI_controller& controller;
+    //PI_controller& controller;
+    Controller& controller;
 
     Digital_out led;
 
@@ -37,7 +39,8 @@ public:
     MotorController(
         RotaryEncoderDriver& encoder,
         MotorDriver& motor,
-        P_controller& controller,
+        //P_controller& controller,
+        Controller& controller,
         State& initialization,
         State& operational,
         State& stopped
@@ -75,11 +78,15 @@ public:
 
     void reset_speed_measurement();
 
+    void reset_controller();
+
+
+
     void led_on();
 
     void led_off();
 
-    void reset_controller();
+    
 };
 
 

include/p_controller.h

@@ -1,7 +1,9 @@
 #ifndef P_CONTROLLER_H
 #define P_CONTROLLER_H
 
-class P_controller
+#include "controller.h"
+
+class P_controller : public Controller
 {
 private:
     double Kp;
@@ -9,7 +11,7 @@ private:
 public:
     P_controller(double Kp);
 
-    double update(double ref, double actual);
+    double update(double ref, double actual) override;
 
 
     

include/pi_controller.h

@@ -1,12 +1,15 @@
 #ifndef PI_CONTROLLER_H
 #define PI_CONTROLLER_H
 
-class PI_controller
+#include "controller.h"
+
+class PI_controller : public Controller
 {
 private:
     double Kp;
     double Ti;
     double T;
+
     double integral;
 
 public:
@@ -16,9 +19,9 @@ public:
         double T
     );
 
-    double update(double ref, double actual);
+    double update(double ref, double actual) override;
 
-    void reset();
+    void reset() override;
 };
 
 #endif