-
Notifications
You must be signed in to change notification settings - Fork 0
/
pump.py
61 lines (45 loc) · 1.55 KB
/
pump.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
from stepper import Stepper
import machine
import _thread
# used for the cooling/OD pump
class VariablePump:
def __init__(self, step_pin):
self.stepper = Stepper(step_pin, None)
self.isRunning = False
self.maxSpeedDelay = 100
self.minSpeedDelay = 2000
self.setSpeed(0)
# speed is float between 0 - 1
def setSpeed(self, speed):
assert (speed <= 1) and (speed >= 0)
speedRange = self.minSpeedDelay - self.maxSpeedDelay
speedDelay = self.minSpeedDelay - (speedRange * speed)
self.stepper.set_step_time(speedDelay)
# do not use outside of the class
def __loop(self):
while self.isRunning:
self.stepper.steps(1000)
def startMotor(self):
self.isRunning = True
_thread.start_new_thread(self.__loop, ())
def stopMotor(self):
self.isRunning = False
class PWMPump:
def __init__(self, step_pin):
self.pwm = machine.PWM(step_pin)
self.pwm.freq(10000)
# used for the feeding pump
class PrecisionPump:
def __init__(self, step_pin, dir_pin):
self.stepper = Stepper(step_pin, dir_pin)
self.stepper.set_step_time(750)
self.dir = 1
self.stepsPerML = 5680 #TODO: determine proper amount
def feed(self, ml_amount):
stepsToTake = ml_amount * self.stepsPerML * self.dir
self.stepper.steps(stepsToTake)
# just for testing
def step(self, steps):
self.stepper.steps(steps)
def reverse(self):
self.dir = 1 if self.dir == 1 else -1