streaming-kamera/src/schrittmotor.py
2021-03-05 15:26:18 +01:00

199 lines
6.6 KiB
Python

from src.einstellungen import *
#import RPi.GPIO as GPIO
import time
import pigpio
import threading
import math
pi = pigpio.pi()
#GPIO.setmode(GPIO.BCM)
#GPIO.setwarnings(False)
class motor():
def __init__(self, name, sp1, sp2, sp3, sp4, t1, t2, te, es, halten, schritte):
self.name = name
self.sp1 = sp1
self.sp2 = sp2
self.sp3 = sp3
self.sp4 = sp4
self.t1 = t1
self.t2 = t2
self.te = te
self.endschalter = es
self.halten = halten
self.schritte_max = schritte
self.pos = 0
self.freq = freq
self.freq_halten = freq_halten
self.cyc_halten = cyc_halten
self.pwm1 = 0
self.pwm2 = 0
self.pwm3 = 0
self.pwm4 = 0
def start(self):
pi.set_mode(self.sp1, pigpio.OUTPUT)
pi.set_mode(self.sp2, pigpio.OUTPUT)
pi.set_mode(self.sp3, pigpio.OUTPUT)
pi.set_mode(self.sp4, pigpio.OUTPUT)
#GPIO.setup(self.sp1, GPIO.OUT)
#GPIO.setup(self.sp2, GPIO.OUT)
#GPIO.setup(self.sp3, GPIO.OUT)
#GPIO.setup(self.sp4, GPIO.OUT)
pi.set_PWM_range(self.sp1, 100)
pi.set_PWM_range(self.sp2, 100)
pi.set_PWM_range(self.sp3, 100)
pi.set_PWM_range(self.sp4, 100)
pi.set_PWM_frequency(self.sp1, self.freq)
pi.set_PWM_dutycycle(self.sp1, 0)
#self.pwm1 = GPIO.PWM(self.sp1, self.freq)
#self.pwm1.start(0)
pi.set_PWM_frequency(self.sp2, self.freq)
pi.set_PWM_dutycycle(self.sp2, 0)
#self.pwm2 = GPIO.PWM(self.sp2, self.freq)
#self.pwm2.start(0)
pi.set_PWM_frequency(self.sp3, self.freq)
pi.set_PWM_dutycycle(self.sp3, 0)
#self.pwm3 = GPIO.PWM(self.sp3, self.freq)
#self.pwm3.start(0)
if self.halten == 1:
pi.set_PWM_frequency(self.sp4, self.freq_halten)
pi.set_PWM_dutycycle(self.sp4, cyc_halten)
#self.pwm4 = GPIO.PWM(self.sp4, self.freq_halten)
#self.pwm4.start(cyc_halten)
else:
pi.set_PWM_frequency(self.sp4, self.freq)
pi.set_PWM_dutycycle(self.sp4, 0)
#self.pwm4 = GPIO.PWM(self.sp4, self.freq)
#self.pwm4.start(0)
def vorwaerts(self, schritte, eichen=0):
ges_schritte = schritte * 8
if eichen == 1:
temp_t1 = self.te
temp_t2 = self.te
else:
temp_t1 = self.t1
temp_t2 = self.t2
if self.halten == 1:
pi.set_PWM_frequency(self.sp4, self.freq)
#self.pwm4.ChangeFrequency(self.freq)
for i in range(int(schritte)):
if self.pos >= self.schritte_max:
break
else:
self.pos += 1
temp_pos = i * 8
einzelschritt(self.sp1, 1, ges_schritte, temp_pos, temp_t1, temp_t2)
einzelschritt(self.sp4, 0, ges_schritte, temp_pos + 1, temp_t1, temp_t2)
einzelschritt(self.sp2, 1, ges_schritte, temp_pos + 2, temp_t1, temp_t2)
einzelschritt(self.sp1, 0, ges_schritte, temp_pos + 3, temp_t1, temp_t2)
einzelschritt(self.sp3, 1, ges_schritte, temp_pos + 4, temp_t1, temp_t2)
einzelschritt(self.sp2, 0, ges_schritte, temp_pos + 5, temp_t1, temp_t2)
einzelschritt(self.sp4, 1, ges_schritte, temp_pos + 6, temp_t1, temp_t2)
einzelschritt(self.sp3, 0, ges_schritte, temp_pos + 7, temp_t1, temp_t2)
#print(str(self.pos))
if self.halten == 1:
pi.set_PWM_frequency(self.sp4, self.freq_halten)
pi.set_PWM_dutycycle(self.sp4, cyc_halten)
#self.pwm4.ChangeFrequency(self.freq_halten)
#self.pwm4.ChangeDutyCycle(self.cyc_halten)
else:
pi.set_PWM_dutycycle(self.sp4, 0)
#self.pwm4.ChangeDutyCycle(0)
def rueckwaerts(self, schritte, eichen=0):
ges_schritte = schritte * 8
if eichen == 1:
temp_t1 = self.te
temp_t2 = self.te
else:
temp_t1 = self.t1
temp_t2 = self.t2
if self.halten == 1:
pi.set_PWM_frequency(self.sp4, self.freq)
#self.pwm4.ChangeFrequency(self.freq)
for i in range(int(schritte)):
#if GPIO.input(self.endschalter) == GPIO.LOW:
if pi.read(self.endschalter) == 0:
print('Endschalter ' + self.name)
self.pos = 0
break
else:
self.pos -= 1
temp_pos = i * 8
einzelschritt(self.sp3, 1, ges_schritte, temp_pos, temp_t1, temp_t2)
einzelschritt(self.sp4, 0, ges_schritte, temp_pos + 1, temp_t1, temp_t2)
einzelschritt(self.sp2, 1, ges_schritte, temp_pos + 2, temp_t1, temp_t2)
einzelschritt(self.sp3, 0, ges_schritte, temp_pos + 3, temp_t1, temp_t2)
einzelschritt(self.sp1, 1, ges_schritte, temp_pos + 4, temp_t1, temp_t2)
einzelschritt(self.sp2, 0, ges_schritte, temp_pos + 5, temp_t1, temp_t2)
einzelschritt(self.sp4, 1, ges_schritte, temp_pos + 6, temp_t1, temp_t2)
einzelschritt(self.sp1, 0, ges_schritte, temp_pos + 7, temp_t1, temp_t2)
#print(str(self.pos))
if self.halten == 1:
pi.set_PWM_frequency(self.sp4, self.freq_halten)
pi.set_PWM_dutycycle(self.sp4, cyc_halten)
#self.pwm4.ChangeFrequency(self.freq_halten)
#self.pwm4.ChangeDutyCycle(self.cyc_halten)
else:
pi.set_PWM_dutycycle(self.sp4, 0)
#self.pwm4.ChangeDutyCycle(0)
def eichen(self):
#if GPIO.input(self.endschalter) == GPIO.LOW:
if pi.read(self.endschalter) == 0:
self.vorwaerts(10, 1)
self.rueckwaerts(self.schritte_max, 1)
self.pos = 0
def gehe_zu(self, position):
temp_schritte = int(position) - self.pos
print(str(temp_schritte))
if temp_schritte > 0:
self.vorwaerts(temp_schritte)
else:
self.rueckwaerts(abs(temp_schritte))
def einzelschritt(spule, status, ges_schritte, schritt, t1, t2):
w = (schritt * 100 / (ges_schritte - 1)) * 1.8
s = round(math.sin(math.radians(w)), 2)
t = t2 - (s * (t2 - t1))
if status == 1:
for i in range(0, 91, 5):
dc = round(math.sin(math.radians(i)) * 100, 0)
pi.set_PWM_dutycycle(spule, dc)
#spule.ChangeDutyCycle(dc)
time.sleep(t)
else:
for i in range(0, 91, 5):
dc = round(math.cos(math.radians(i)) * 100, 0)
pi.set_PWM_dutycycle(spule, dc)
#spule.ChangeDutyCycle(dc)
time.sleep(t)