add VibrationMotor

Author: thijstriemstra

VibrationMotor/VibrationMotor.cpp

@@ -0,0 +1,38 @@
+/*  Copyright (c) 2022, Collab
+ *  All rights reserved
+*/
+
+#include "VibrationMotor.h"
+
+VibrationMotor::VibrationMotor(
+  int motor_pin,
+  unsigned int period
+) {
+  _motorPin = motor_pin;
+  _period = period;
+}
+
+void VibrationMotor::begin() {
+  pinMode(_motorPin, OUTPUT);
+}
+
+void VibrationMotor::loop() {
+  if (_enabled) {
+    if ((millis() - _enablePoint) >= _period) {
+      disable();
+    }
+  }
+}
+
+void VibrationMotor::enable() {
+  _enabled = true;
+  _enablePoint = millis();
+
+  digitalWrite(_motorPin, HIGH);
+}
+
+void VibrationMotor::disable() {
+  _enabled = false;
+
+  digitalWrite(_motorPin, LOW);
+}

VibrationMotor/VibrationMotor.h

@@ -0,0 +1,29 @@
+/*  Copyright (c) 2022, Collab
+ *  All rights reserved
+*/
+#ifndef VibrationMotor_h
+#define VibrationMotor_h
+
+#include <Arduino.h>
+
+class VibrationMotor
+{
+  public:
+    VibrationMotor(
+      int motor_pin,
+      unsigned int period = 250
+    );
+    void begin();
+    void loop();
+    void enable();
+    void disable();
+
+  private:
+    int _motorPin;
+    unsigned int _period;
+    unsigned int _enablePoint;
+
+    bool _enabled = false;
+};
+
+#endif