-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest
More file actions
49 lines (40 loc) · 1.3 KB
/
test
File metadata and controls
49 lines (40 loc) · 1.3 KB
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
#import gpio and time
import RPi.GPIO as GPIO
import time
GPIO.setwarnings(False)
GPIO.cleanup()
GPIO.setmode(GPIO.BOARD)
# Set up GPIO pins for IR beams and solenoids
ir_beam1_pin = 16
led1 = 7
odor_solenoid_pin = 29
vacuum_solenoid_pin = 33
GPIO.setup(ir_beam1_pin, GPIO.IN)
GPIO.setup(led1,GPIO.OUT, initial=GPIO.LOW)
GPIO.setup(odor_solenoid_pin, GPIO.OUT, initial=GPIO.LOW)
GPIO.setup(vacuum_solenoid_pin, GPIO.OUT, initial=GPIO.LOW)
# Define odor delivery function
def deliver_odor(odor_solenoid_pin, duration):
GPIO.output(odor_solenoid_pin, GPIO.HIGH)
time.sleep(duration)
GPIO.output(odor_solenoid_pin, GPIO.LOW)
# Define vacuum cease function
def vacuum_cease(vacuum_solenoid_pin, duration):
GPIO.output(vacuum_solenoid_pin, GPIO.HIGH)
time.sleep(duration)
GPIO.output(vacuum_solenoid_pin, GPIO.LOW)
# Define trial function
def run_trial(trial_duration):
start_time = time.time()
while time.time() < start_time + trial_duration:
if GPIO.input(ir_beam1_pin):
deliver_odor(odor_solenoid_pin, 1)
cease_vacuum(vacuum_solenoid_pin, 1)
time.sleep(60)
# Run two trials
for trial in range(2):
print(f"Starting trial {trial + 1}")
run_trial(240) # four minutes
print(f"Trial {trial + 1} complete")
# Clean up GPIO
GPIO.cleanup()