PRE2017 3 11 Python Code

From Control Systems Technology Group

(Difference between revisions)
Jump to: navigation, search
Line 1: Line 1:
 +
__NOTOC__
 +
==Import required packages==
<pre>
<pre>
#!/usr/bin/python
#!/usr/bin/python
Line 8: Line 10:
import signal # Cleanup when ending script
import signal # Cleanup when ending script
import sys # Needed to stop script if no camera
import sys # Needed to stop script if no camera
 +
</pre>
 +
==Set up constants==
 +
 +
<pre>
pin_s1  = 29 # Continuous servo 1
pin_s1  = 29 # Continuous servo 1
pin_s2  = 33            # Continuous servo 2
pin_s2  = 33            # Continuous servo 2
Line 27: Line 33:
imgname = "image2.jpg"  # File frame is temporarily saved in
imgname = "image2.jpg"  # File frame is temporarily saved in
-
camslot = 0 # Default camera slot
 
# Imports allowed QRcodes
# Imports allowed QRcodes
Line 33: Line 38:
with open("authcodes") as file:
with open("authcodes") as file:
passcodelist = [line.strip() for line in file]
passcodelist = [line.strip() for line in file]
 +
</pre>
 +
==Stops the script safely if CTRL+C is pressed==
 +
 +
<pre>
# Releases pins when script is interrupted
# Releases pins when script is interrupted
def end_run(signal,frame):
def end_run(signal,frame):
Line 45: Line 54:
# Runs end_run if ctrl+C is clicked
# Runs end_run if ctrl+C is clicked
signal.signal(signal.SIGINT, end_run)
signal.signal(signal.SIGINT, end_run)
 +
</pre>
 +
==Takes a picture with the camera==
 +
 +
<pre>
# Takes picture from camera, reads and evaluates QRcode
# Takes picture from camera, reads and evaluates QRcode
def QRscanner(passcodelist, camslot):
def QRscanner(passcodelist, camslot):
camera = cv2.VideoCapture(camslot) # Initializes camera viewer
camera = cv2.VideoCapture(camslot) # Initializes camera viewer
grabbed, im = camera.read()    # Reads figure
grabbed, im = camera.read()    # Reads figure
 +
</pre>
 +
==Checks if a picture is taken, and if it is not, it looks for other cameras or stops the script==
 +
 +
<pre>
if not grabbed: # Searches for working camera
if not grabbed: # Searches for working camera
if camslot < 5:
if camslot < 5:
Line 63: Line 80:
GPIO.cleanup()
GPIO.cleanup()
sys.exit()
sys.exit()
 +
</pre>
 +
==Reads QR-codes in the image and checks if a correct code is presented==
 +
 +
<pre>
cv2.imwrite(imgname,im) # Converts image to jpg
cv2.imwrite(imgname,im) # Converts image to jpg
img = Image.open(imgname) # Reads jpg
img = Image.open(imgname) # Reads jpg
Line 76: Line 97:
print("Wrong code")
print("Wrong code")
return False, camslot
return False, camslot
 +
</pre>
 +
==Sets up the output pins that control the servos==
 +
 +
<pre>
GPIO.setmode(GPIO.BOARD)    # Board numbering sceme pins
GPIO.setmode(GPIO.BOARD)    # Board numbering sceme pins
GPIO.setup(pin_s1, GPIO.OUT) # Sets pins as output
GPIO.setup(pin_s1, GPIO.OUT) # Sets pins as output
Line 84: Line 109:
servo1.start(s1center) # Starts servos in neutral position
servo1.start(s1center) # Starts servos in neutral position
servo2.start(s2center)
servo2.start(s2center)
 +
</pre>
 +
==Sets up variables, starts the script, and runs the QR-code reader until correct code is presented or CTRL+C is pressed==
 +
 +
<pre>
 +
camslot = 0 # Default camera slot
running = True
running = True
print("Program Running, provide QR-code")
print("Program Running, provide QR-code")
Line 90: Line 120:
while running:
while running:
rightcode, camslot = QRscanner(passcodelist, camslot)  # Runs QR-reader
rightcode, camslot = QRscanner(passcodelist, camslot)  # Runs QR-reader
 +
</pre>
 +
 +
==Makes Cylinders and platform move if the correct code is presented==
 +
 +
<pre>
if rightcode:
if rightcode:
servo1.ChangeDutyCycle(s1left) # Cylinders open
servo1.ChangeDutyCycle(s1left) # Cylinders open

Revision as of 17:48, 24 March 2018

Import required packages

#!/usr/bin/python
from pyzbar.pyzbar import decode	# QR reader
from PIL import Image               	# Image reader
import cv2                     		# Image reader
import RPi.GPIO as GPIO    	     	# GPIO pin control
import time               	 	# Posibility to delay
import signal				# Cleanup when ending script
import sys				# Needed to stop script if no camera

Set up constants

pin_s1  = 29		# Continuous servo 1
pin_s2  = 33            # Continuous servo 2
freq 	= 50        	# Continuous servo frequency [Hz]

s1left   = 5.00         # 100% velocity to left
s1center = 6.70         # no movement
s1right  = 8.00         # 100% velocity to right

s2left   = 5.00         # 100% velocity to left
s2center = 6.60         # no movement
s2right  = 8.00         # 100% velocity to right

cyldelay      = 2 	# Time it takes for cylinder to rotate
dropdelay     = 5    	# Time it takes to drop package
pauzedelay    = 2 	# Time between cylinder rotation and platform movement
platformdelay = 2     	# Time it takes for platform to move down

imgname = "image2.jpg"  # File frame is temporarily saved in

# Imports allowed QRcodes
passcodelist = []
with open("authcodes") as file:
	passcodelist = [line.strip() for line in file]

Stops the script safely if CTRL+C is pressed

# Releases pins when script is interrupted
def end_run(signal,frame):
	global running
	print("Ctrl+C captured, ending script")
	running = False
	servo1.stop()
	servo2.stop()
	GPIO.cleanup()

# Runs end_run if ctrl+C is clicked
signal.signal(signal.SIGINT, end_run)

Takes a picture with the camera

# Takes picture from camera, reads and evaluates QRcode
def QRscanner(passcodelist, camslot):
	camera = cv2.VideoCapture(camslot)	# Initializes camera viewer
	grabbed, im = camera.read()     	# Reads figure

Checks if a picture is taken, and if it is not, it looks for other cameras or stops the script

	if not grabbed:				# Searches for working camera
		if camslot < 5:
			camstr = "No camera connected in slot: " + str(camslot)
			print(camstr)
			camslot = camslot + 1
			return False, camslot
		else:
			print("No camera connected")
			servo1.stop()
			servo2.stop()
			GPIO.cleanup()
			sys.exit()

Reads QR-codes in the image and checks if a correct code is presented

	cv2.imwrite(imgname,im)		# Converts image to jpg
	img = Image.open(imgname)	# Reads jpg
	decodedObjects = decode(img)    # Decodes image
	for obj in decodedObjects:     	# Makes sure every code is read
	   	qrdata = obj.data       # Reads the alphanumerical code
		for passcode in passcodelist:
    			if passcode in qrdata:      # Verifies if code is legit
				print("Correct code, cylinders start opening")
        			return True, camslot
			else:
				print("Wrong code")
	return False, camslot

Sets up the output pins that control the servos

GPIO.setmode(GPIO.BOARD)    	# Board numbering sceme pins
GPIO.setup(pin_s1, GPIO.OUT)	# Sets pins as output
GPIO.setup(pin_s2, GPIO.OUT)
servo1 = GPIO.PWM(pin_s1, freq)	# Assigns frequency to pins
servo2 = GPIO.PWM(pin_s2, freq)
servo1.start(s1center)		# Starts servos in neutral position
servo2.start(s2center)

Sets up variables, starts the script, and runs the QR-code reader until correct code is presented or CTRL+C is pressed

camslot = 0 		# Default camera slot
running = True
print("Program Running, provide QR-code")

while running:
	rightcode, camslot = QRscanner(passcodelist, camslot)  # Runs QR-reader

Makes Cylinders and platform move if the correct code is presented

	if rightcode:
		servo1.ChangeDutyCycle(s1left)	 # Cylinders open
    		time.sleep(cyldelay)             # Duration of cylinder movement
    		servo1.ChangeDutyCycle(s1center) # Cylinders stop moving
		print("Cylinders are open, package can be dropped")
    		time.sleep(dropdelay)            # Duration of package dropping
    		print("Package dropped, cylinders start closing")
		servo1.ChangeDutyCycle(s1right)  # Cylinders close
   		time.sleep(cyldelay)             # Duration of cylinder movement
    		servo1.ChangeDutyCycle(s1center) # Cylinders stop moving
		print("Cylinders are closed")
		time.sleep(pauzedelay)           # Time buffer
		print("Platform starts moving down")
		servo2.ChangeDutyCycle(s2left)   # Platform moves down
    		time.sleep(platformdelay)        # Duration of platform movement
    		servo2.ChangeDutyCycle(s2center) # Platform stops moving
		print("Platform stopped moving down, new code can be scanned")
Personal tools