def __init__(self, server, sock, address):
super(RobotControl, self).__init__(server, sock, address)
# setup GPIO pins for proximity sensors
for PIN in self.PROXIMITY_GPIO:
GPIO.setup(PIN, GPIO.IN)
# try to add event detection for proximity GPIO pins
for PIN, val in self.PROXIMITY_GPIO.items():
# wait until the GPIO is configured as an input
while GPIO.gpio_function(PIN) != GPIO.IN:
GPIO.setup(PIN, GPIO.IN)
GPIO.add_event_detect(PIN, GPIO.FALLING, self.__proximityDetect,
10)
for LED in self.LEDS_GPIO.itervalues():
GPIO.setup(LED, GPIO.OUT)
GPIO.output(LED, GPIO.LOW)
# Connected to Sabertooth S2 as "emergency stop" button
GPIO.setup(self.STOP_GPIO, GPIO.OUT)
GPIO.output(self.STOP_GPIO, GPIO.HIGH)
# Triggers when button pressed
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
while GPIO.gpio_function(self.STOP_BUTTON) != GPIO.IN:
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
GPIO.add_event_detect(self.STOP_BUTTON, GPIO.RISING, self.__stopButton,
10)
# HMC5883L Magnetometer
self.mag = MAG(declination=(11, 35))
# HC-SR04 pin setup
# ECHO_TRIGGER initiates ultrasonic pulse
GPIO.setup(self.ECHO_TRIGGER, GPIO.OUT)
# ECHO_RETURN - needs to be level shifted from 5.0V to 3.3V
# time of +ve pulse is the distance
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
while GPIO.gpio_function(self.ECHO_RETURN) != GPIO.IN:
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
GPIO.add_event_detect(self.ECHO_RETURN, GPIO.BOTH, self.__measureEcho,
1)
GPIO.output(self.ECHO_TRIGGER, GPIO.LOW)
# Start servo scanning movement thread
self.SCAN = True
threading.Thread(target=self.__servoScan).start()
# Start HC-SR04 timing/measurement thread
threading.Thread(target=self.__HCSR04).start()
self.do_beep(0.25)
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.HIGH)
self.saber = Sabertooth(self.UART, self.TTY)
self.saber.setRamp(15)
def __init__(self, speed_max=1.654, speed_limit=(1, 90), track=0.656, rcrit=1, ramp=15):
"""
speed_max: Maximum physical speed of drive motors (m/s).
speed_limit: Tuple: speed limit for forward/reverse drive
operations (m/s), speed limit for tight turns (deg/s).
track: Track width for the drive wheels (m).
rcrit: Critical radius inside which speeds measured angularly.
ramp: Sabertooth ramp identifier.
"""
# Drive system parameters.
self.speed_max = speed_max
self.speed_limit = speed_limit
self.track = track
self.rcrit = rcrit
self.ramp = ramp
if (self.speed_limit[0] > self.speed_max):
return None
# Instantiate motor driver object (com. over UART4).
TTY = "ttyO4"
self.saber = Sabertooth(TTY)
if (self.saber == None):
return None
self.saber.setRamp(self.ramp)
return None
def __init__(self, server, sock, address):
super(RobotControl, self).__init__(server, sock, address)
# setup GPIO pins for proximity sensors
for PIN in self.PROXIMITY_GPIO:
GPIO.setup(PIN, GPIO.IN)
# try to add event detection for proximity GPIO pins
for PIN, val in self.PROXIMITY_GPIO.items():
# wait until the GPIO is configured as an input
while GPIO.gpio_function(PIN) != GPIO.IN:
GPIO.setup(PIN, GPIO.IN)
GPIO.add_event_detect(PIN, GPIO.FALLING, self.__proximityDetect, 10)
for LED in self.LEDS_GPIO.itervalues():
GPIO.setup(LED, GPIO.OUT)
GPIO.output(LED, GPIO.LOW)
# Connected to Sabertooth S2 as "emergency stop" button
GPIO.setup(self.STOP_GPIO, GPIO.OUT)
GPIO.output(self.STOP_GPIO, GPIO.HIGH)
# Triggers when button pressed
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
while GPIO.gpio_function(self.STOP_BUTTON) != GPIO.IN:
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
GPIO.add_event_detect(self.STOP_BUTTON, GPIO.RISING, self.__stopButton, 10)
# HMC5883L Magnetometer
self.mag = MAG(declination=(11,35))
# HC-SR04 pin setup
# ECHO_TRIGGER initiates ultrasonic pulse
GPIO.setup(self.ECHO_TRIGGER, GPIO.OUT)
# ECHO_RETURN - needs to be level shifted from 5.0V to 3.3V
# time of +ve pulse is the distance
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
while GPIO.gpio_function(self.ECHO_RETURN) != GPIO.IN:
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
GPIO.add_event_detect(self.ECHO_RETURN, GPIO.BOTH, self.__measureEcho, 1)
GPIO.output(self.ECHO_TRIGGER, GPIO.LOW)
# Start servo scanning movement thread
self.SCAN = True
threading.Thread(target=self.__servoScan).start()
# Start HC-SR04 timing/measurement thread
threading.Thread(target=self.__HCSR04).start()
self.do_beep(0.25)
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.HIGH)
self.saber = Sabertooth(self.UART, self.TTY)
self.saber.setRamp(15)
class Drive():
"""
Drive: A class that abstracts Sabertooth to useful differential drive
commands. This class is not meant to be sufficient for
controlled operation of a vehicle. An additional abstraction
layer to take vehicle dynamics into account is recommended.
"""
def __init__(self, speed_max=1.654, speed_limit=(1, 90), track=0.656, rcrit=1, ramp=15):
"""
speed_max: Maximum physical speed of drive motors (m/s).
speed_limit: Tuple: speed limit for forward/reverse drive
operations (m/s), speed limit for tight turns (deg/s).
track: Track width for the drive wheels (m).
rcrit: Critical radius inside which speeds measured angularly.
ramp: Sabertooth ramp identifier.
"""
# Drive system parameters.
self.speed_max = speed_max
self.speed_limit = speed_limit
self.track = track
self.rcrit = rcrit
self.ramp = ramp
if (self.speed_limit[0] > self.speed_max):
return None
# Instantiate motor driver object (com. over UART4).
TTY = "ttyO4"
self.saber = Sabertooth(TTY)
if (self.saber == None):
return None
self.saber.setRamp(self.ramp)
return None
def __del__(self):
self.saber.stop()
return
def drive(self, direction="fwd", speed=0, turn="no", radius=0.25):
"""
drive: Drive in specified direction at specified speed with
specified radius of turning. Note that it is recommended
that the control layer bring the vehicle to a stop before
allowing the radius of turning to cross the critical value.
direction: fwd or rev. Corresponds to CCW or CW for a left
turn, CW or CCW for right turn
speed: tangential speed (m/s)
turn: left, no, or right
radius: radius of turning with respect to axle center
"""
# Stupidity checks.
validcmds = ["fwd", "rev"]
if (direction not in validcmds):
return -1
validcmds = ["left", "no", "right"]
if (turn not in validcmds):
return -1
if radius < 0:
radius = 0
# Drive algorithm.
if turn == "no":
# Check speed limit. Ignore if malformed.
if speed < 0:
return -1
elif speed > self.speed_limit[0]:
return -1
# Calculate speed percentage.
speed = int(speed/self.speed_max*100)
# Debug logging.
logging.debug("drive (straight): %s %d" %(direction, speed))
# Command motor driver.
self.saber.independentDrive(direction, speed, direction, speed)
else:
if radius >= self.rcrit:
# Check speed limit.
if speed < 0:
return -1
elif speed > self.speed_limit[0]:
return -1
# Wheel speeds assuming left turn.
speed_l = speed*(2*radius - self.track)/(2*radius)
speed_r = speed*(2*radius + self.track)/(2*radius)
# Check for saturation.
if speed_r > self.speed_max:
# Truncate speed to max attainable at turning radius.
logging.debug("drive (turn): speed truncated due to saturation")
speed_l = self.speed_max/speed_r*speed_l
speed_r = self.speed_max
# Swap if turning right.
if turn == "right":
speed_l, speed_r = speed_r, speed_l
# Calculate speed percentages.
speed_l = abs(int(speed_l/self.speed_max*100))
speed_r = abs(int(speed_r/self.speed_max*100))
# Debug logging.
logging.debug("drive (turn): %s %d %s %d" %(direction, speed_l, direction, speed_r))
# Command motor driver.
self.saber.independentDrive(direction, speed_l, direction, speed_r)
else:
# Check speed limit.
if speed < 0:
return -1
elif speed > self.speed_limit[1]:
return -1
# Convert speed to rad/s.
speed = math.pi*speed/180
# Wheel speeds assuming left turn.
speed_l = speed*(2*radius - self.track)/2
speed_r = speed*(2*radius + self.track)/2
# Check for saturation.
if speed_r > self.speed_max:
# Truncate speed to max attainable at turning radius.
logging.debug("drive (turn): speed truncated due to saturation")
speed_l = self.speed_max/speed_r*speed_l
speed_r = self.speed_max
# Swap if turning right.
if turn == "right":
speed_l, speed_r = speed_r, speed_l
# Determine directions assuming driving forward.
if speed_l < 0:
dir_l = "rev"
else:
dir_l = "fwd"
if speed_r < 0:
dir_r = "rev"
else:
dir_r = "fwd"
# Swap if driving reverse.
if direction == "rev":
dir_l, dir_r = dir_r, dir_l
# Calculate speed percentages.
speed_l = abs(int(speed_l/self.speed_max*100))
speed_r = abs(int(speed_r/self.speed_max*100))
# Debug logging.
logging.debug("drive (turn): %s %d %s %d" %(dir_l, speed_l, dir_r, speed_r))
# Command motor driver.
self.saber.independentDrive(dir_l, speed_l, dir_r, speed_r)
def stop(self):
"""
stop: Stops all drive using drive.
"""
self.drive("fwd", 0)
class RobotControl(SimpleWebSocketServer.WebSocket):
saber = None
UART = "UART1"
TTY = "ttyO1"
STOP_GPIO = "P9_13"
STOP_BUTTON = "P9_15"
PROXIMITY_GPIO = {"P9_11": {"name": "right"}, "P9_12": {"name": "left"}}
LEDS_GPIO = {"RED_pin": "P8_10", "GREEN_pin": "P8_11"}
SPEAKER_PWM = "P8_13"
SERVO_PWM = "P8_34"
ECHO_RETURN = "P8_15"
ECHO_TRIGGER = "P9_14"
SPEED = 10
CMD = None
OBSTACLE = False
SPEAKER = False
SCAN = False
angle = 0.0
distance = (0.0, 0.0)
COUNTING = False
startTime = datetime.now()
endTime = datetime.now()
delta = endTime - startTime
# cmd values that can be sent for a JSON "drive" event
# maps command to corresponding function to control motors
commands = {
"fwd": "do_forward",
"rev": "do_reverse",
"left": "turn_left",
"right": "turn_right",
"stop": "do_stop",
"speed": "set_speed"
}
def __init__(self, server, sock, address):
super(RobotControl, self).__init__(server, sock, address)
# setup GPIO pins for proximity sensors
for PIN in self.PROXIMITY_GPIO:
GPIO.setup(PIN, GPIO.IN)
# try to add event detection for proximity GPIO pins
for PIN, val in self.PROXIMITY_GPIO.items():
# wait until the GPIO is configured as an input
while GPIO.gpio_function(PIN) != GPIO.IN:
GPIO.setup(PIN, GPIO.IN)
GPIO.add_event_detect(PIN, GPIO.FALLING, self.__proximityDetect,
10)
for LED in self.LEDS_GPIO.itervalues():
GPIO.setup(LED, GPIO.OUT)
GPIO.output(LED, GPIO.LOW)
# Connected to Sabertooth S2 as "emergency stop" button
GPIO.setup(self.STOP_GPIO, GPIO.OUT)
GPIO.output(self.STOP_GPIO, GPIO.HIGH)
# Triggers when button pressed
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
while GPIO.gpio_function(self.STOP_BUTTON) != GPIO.IN:
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
GPIO.add_event_detect(self.STOP_BUTTON, GPIO.RISING, self.__stopButton,
10)
# HMC5883L Magnetometer
self.mag = MAG(declination=(11, 35))
# HC-SR04 pin setup
# ECHO_TRIGGER initiates ultrasonic pulse
GPIO.setup(self.ECHO_TRIGGER, GPIO.OUT)
# ECHO_RETURN - needs to be level shifted from 5.0V to 3.3V
# time of +ve pulse is the distance
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
while GPIO.gpio_function(self.ECHO_RETURN) != GPIO.IN:
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
GPIO.add_event_detect(self.ECHO_RETURN, GPIO.BOTH, self.__measureEcho,
1)
GPIO.output(self.ECHO_TRIGGER, GPIO.LOW)
# Start servo scanning movement thread
self.SCAN = True
threading.Thread(target=self.__servoScan).start()
# Start HC-SR04 timing/measurement thread
threading.Thread(target=self.__HCSR04).start()
self.do_beep(0.25)
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.HIGH)
self.saber = Sabertooth(self.UART, self.TTY)
self.saber.setRamp(15)
def __speaker(self):
if self.SPEAKER:
return
self.SPEAKER = True
PWM.start(self.SPEAKER_PWM, 50, 3000)
for dir in [-1, 2]:
for x in range(3, 20):
PWM.set_frequency(self.SPEAKER_PWM, 3000 + (dir * x * 100))
sleep(0.05)
PWM.stop(self.SPEAKER_PWM)
self.SPEAKER = False
return
def __proximityDetect(self, channel):
if self.OBSTACLE or (self.CMD in ["stop", None]):
return # already handling a problem or not moving
threading.Thread(target=self.__speaker).start()
self.OBSTACLE = True
self.sendJSON(
"obstacle", {
"sensor": "%s" % (channel),
"name": "%s" % (self.PROXIMITY_GPIO[channel]["name"])
})
self.saber.driveMotor("both", "rev", int(float(self.SPEED) * 1.5))
delay = 1 - (float(self.SPEED) / 200)
sleep(delay)
self.do_stop(self.SPEED)
self.OBSTACLE = False
return
def __stopButton(self, channel):
self.SCAN = False
GPIO.output(self.STOP_GPIO, GPIO.HIGH)
self.do_stop(self.SPEED)
return
def __servoScan(self):
PWM.start(self.SERVO_PWM, 3, 50)
while self.SCAN:
for a in xrange(0, 180, 1):
self.angle = a
duty = 3.7 + (a / 180.0) * 9.5
if self.SCAN:
PWM.set_duty_cycle(self.SERVO_PWM, duty)
else:
break
sleep(0.01)
for a in xrange(181, 1, -1):
self.angle = a
duty = 3.7 + (a / 180.0) * 9.5
if self.SCAN:
PWM.set_duty_cycle(self.SERVO_PWM, duty)
else:
break
sleep(0.01)
return
def __measureEcho(self, channel):
# Get time when Echo line goes high (ie: RISING edge)
if GPIO.input(self.ECHO_RETURN) == GPIO.HIGH and not self.COUNTING:
self.startTime = datetime.now()
self.COUNTING = True
# Get time when Echo line goes low (ie: FALLING edge)
elif GPIO.input(self.ECHO_RETURN) == GPIO.LOW and self.COUNTING:
self.endTime = datetime.now()
# delta is the period of the echo (roughly)
self.delta = self.endTime - self.startTime
self.COUNTING = False
return
def __HCSR04(self):
measures = []
while self.SCAN:
# Set Trigger to HIGH
GPIO.output("P9_14", GPIO.HIGH)
# Wait a tiny amount (should be 10us). This is not deterministic
# when running stock Linux. To get better use a RTOS or the PRUSS
sleep(0.001)
# Set Trigger to LOW
# This will cause the HC-SR04 to output an ultrasonic pulse and start
# listening for the return
GPIO.output("P9_14", GPIO.LOW)
# Sleep for a bit so that we don't send out another pulse before the
# previous one has finished
sleep(0.05)
# 58.77 is the magic number for cm per microsecond
d = min(200.0, max(self.delta.microseconds / 58.77, 17.0))
# only include value if it is reasonable/valid
if d > 17 and d < 200:
measures.insert(0, d)
if len(measures) > 3:
measures.pop()
# take an average of the last few measures as the distance
if len(measures) > 0:
self.distance = sum(measures) / len(measures)
else:
self.distance = 20.0
if self.distance < 20 and not self.OBSTACLE:
self.OBSTACLE = True
if self.angle < 90:
self.sendJSON("obstacle", {
"sensor": "P9_12",
"name": "left"
})
else:
self.sendJSON("obstacle", {
"sensor": "P9_11",
"name": "right"
})
self.OBSTACLE = False
return
def do_beep(self, duration):
PWM.start(self.SPEAKER_PWM, 50, 3000)
sleep(duration)
PWM.stop(self.SPEAKER_PWM)
def do_forward(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("both", "fwd", self.SPEED)
def do_reverse(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("both", "rev", self.SPEED)
def do_stop(self, set_speed):
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.HIGH)
GPIO.output(self.LEDS_GPIO["GREEN_pin"], GPIO.LOW)
self.saber.stop()
def turn_left(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("left", "fwd", self.SPEED)
self.saber.driveMotor("right", "rev", self.SPEED)
delay = 1 - (float(self.SPEED) / 100)
sleep(delay)
if self.CMD in ["fwd", "rev"]:
getattr(self, self.commands[self.CMD])(int(self.SPEED))
else:
self.do_stop(self.SPEED)
def turn_right(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("right", "fwd", self.SPEED)
self.saber.driveMotor("left", "rev", self.SPEED)
delay = 1 - (float(self.SPEED) / 100)
sleep(delay)
if self.CMD in ["fwd", "rev"]:
getattr(self, self.commands[self.CMD])(int(self.SPEED))
else:
self.do_stop(self.SPEED)
def set_speed(self, new_speed):
self.SPEED = new_speed
if (self.CMD != "speed") and (self.CMD != None):
getattr(self, self.commands[self.CMD])(int(self.SPEED))
def sendJSON(self, event, data):
try:
self.sendMessage(json.dumps({"event": event, "data": data}))
except Exception as n:
print "sendJSON: ", n
def handleMessage(self):
if self.data is None:
self.data = ''
msg = json.loads(str(self.data))
if msg['event'] == 'drive' and msg['data']['cmd'] in self.commands:
if msg['data']['speed'] == None:
msg['data']['speed'] = self.SPEED
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.LOW)
GPIO.output(self.LEDS_GPIO["GREEN_pin"], GPIO.HIGH)
getattr(self, self.commands[msg['data']['cmd']])(int(
msg['data']['speed']))
if msg['data']['cmd'] in ["fwd", "rev", "stop"]:
self.CMD = msg['data']['cmd']
self.SPEED = int(msg['data']['speed'])
self.sendJSON("ack", {
"cmd":
"%s %d" % (self.commands[msg['data']['cmd']], self.SPEED)
})
elif (msg["event"] == 'fetch'):
if msg['data']['cmd'] == 'angle':
self.sendJSON("angle", {"angle": "%2.2f" % (self.angle)})
elif msg['data']['cmd'] == 'heading':
self.sendJSON("heading",
{"angle": "%2.2f" % (self.mag.getHeading())})
def handleConnected(self):
print self.address, 'connected'
def handleClose(self):
self.do_beep(0.25)
for PIN in self.PROXIMITY_GPIO:
GPIO.remove_event_detect(PIN)
GPIO.remove_event_detect(self.STOP_BUTTON)
GPIO.remove_event_detect(self.ECHO_RETURN)
self.SCAN = False
sleep(1)
PWM.stop(self.SERVO_PWM)
PWM.stop(self.SPEAKER_PWM)
self.saber.stop()
print self.address, 'closed'
class RobotControl(SimpleWebSocketServer.WebSocket):
saber = None
UART = "UART1"
TTY ="ttyO1"
STOP_GPIO = "P9_13"
STOP_BUTTON = "P9_15"
PROXIMITY_GPIO = {"P9_11": {"name": "right"},
"P9_12": {"name": "left"}
}
LEDS_GPIO = { "RED_pin": "P8_10",
"GREEN_pin": "P8_11" }
SPEAKER_PWM = "P8_13"
SERVO_PWM = "P8_34"
ECHO_RETURN = "P8_15"
ECHO_TRIGGER = "P9_14"
SPEED = 10
CMD = None
OBSTACLE = False
SPEAKER = False
SCAN = False
angle = 0.0
distance = (0.0, 0.0)
COUNTING = False
startTime = datetime.now()
endTime = datetime.now()
delta = endTime-startTime
# cmd values that can be sent for a JSON "drive" event
# maps command to corresponding function to control motors
commands = {"fwd": "do_forward",
"rev": "do_reverse",
"left": "turn_left",
"right": "turn_right",
"stop": "do_stop",
"speed": "set_speed"
}
def __init__(self, server, sock, address):
super(RobotControl, self).__init__(server, sock, address)
# setup GPIO pins for proximity sensors
for PIN in self.PROXIMITY_GPIO:
GPIO.setup(PIN, GPIO.IN)
# try to add event detection for proximity GPIO pins
for PIN, val in self.PROXIMITY_GPIO.items():
# wait until the GPIO is configured as an input
while GPIO.gpio_function(PIN) != GPIO.IN:
GPIO.setup(PIN, GPIO.IN)
GPIO.add_event_detect(PIN, GPIO.FALLING, self.__proximityDetect, 10)
for LED in self.LEDS_GPIO.itervalues():
GPIO.setup(LED, GPIO.OUT)
GPIO.output(LED, GPIO.LOW)
# Connected to Sabertooth S2 as "emergency stop" button
GPIO.setup(self.STOP_GPIO, GPIO.OUT)
GPIO.output(self.STOP_GPIO, GPIO.HIGH)
# Triggers when button pressed
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
while GPIO.gpio_function(self.STOP_BUTTON) != GPIO.IN:
GPIO.setup(self.STOP_BUTTON, GPIO.IN)
GPIO.add_event_detect(self.STOP_BUTTON, GPIO.RISING, self.__stopButton, 10)
# HMC5883L Magnetometer
self.mag = MAG(declination=(11,35))
# HC-SR04 pin setup
# ECHO_TRIGGER initiates ultrasonic pulse
GPIO.setup(self.ECHO_TRIGGER, GPIO.OUT)
# ECHO_RETURN - needs to be level shifted from 5.0V to 3.3V
# time of +ve pulse is the distance
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
while GPIO.gpio_function(self.ECHO_RETURN) != GPIO.IN:
GPIO.setup(self.ECHO_RETURN, GPIO.IN)
GPIO.add_event_detect(self.ECHO_RETURN, GPIO.BOTH, self.__measureEcho, 1)
GPIO.output(self.ECHO_TRIGGER, GPIO.LOW)
# Start servo scanning movement thread
self.SCAN = True
threading.Thread(target=self.__servoScan).start()
# Start HC-SR04 timing/measurement thread
threading.Thread(target=self.__HCSR04).start()
self.do_beep(0.25)
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.HIGH)
self.saber = Sabertooth(self.UART, self.TTY)
self.saber.setRamp(15)
def __speaker(self):
if self.SPEAKER:
return
self.SPEAKER = True
PWM.start(self.SPEAKER_PWM, 50, 3000)
for dir in [-1,2]:
for x in range(3,20):
PWM.set_frequency(self.SPEAKER_PWM, 3000 + (dir * x * 100))
sleep(0.05)
PWM.stop(self.SPEAKER_PWM)
self.SPEAKER = False
return
def __proximityDetect(self, channel):
if self.OBSTACLE or (self.CMD in ["stop", None]):
return # already handling a problem or not moving
threading.Thread(target=self.__speaker).start()
self.OBSTACLE = True
self.sendJSON("obstacle", {"sensor": "%s" % (channel),
"name": "%s" % (self.PROXIMITY_GPIO[channel]["name"])})
self.saber.driveMotor("both", "rev", int(float(self.SPEED)*1.5))
delay = 1 - (float(self.SPEED)/200)
sleep(delay)
self.do_stop(self.SPEED)
self.OBSTACLE = False
return
def __stopButton(self, channel):
self.SCAN = False
GPIO.output(self.STOP_GPIO, GPIO.HIGH)
self.do_stop(self.SPEED)
return
def __servoScan(self):
PWM.start(self.SERVO_PWM, 3, 50)
while self.SCAN:
for a in xrange(0,180,1):
self.angle = a
duty = 3.7 + (a/180.0)*9.5
if self.SCAN:
PWM.set_duty_cycle(self.SERVO_PWM, duty)
else:
break
sleep(0.01)
for a in xrange(181,1,-1):
self.angle = a
duty = 3.7 + (a/180.0)*9.5
if self.SCAN:
PWM.set_duty_cycle(self.SERVO_PWM, duty)
else:
break
sleep(0.01)
return
def __measureEcho(self, channel):
# Get time when Echo line goes high (ie: RISING edge)
if GPIO.input(self.ECHO_RETURN) == GPIO.HIGH and not self.COUNTING:
self.startTime = datetime.now()
self.COUNTING = True
# Get time when Echo line goes low (ie: FALLING edge)
elif GPIO.input(self.ECHO_RETURN) == GPIO.LOW and self.COUNTING:
self.endTime = datetime.now()
# delta is the period of the echo (roughly)
self.delta = self.endTime - self.startTime
self.COUNTING = False
return
def __HCSR04(self):
measures = []
while self.SCAN:
# Set Trigger to HIGH
GPIO.output("P9_14", GPIO.HIGH)
# Wait a tiny amount (should be 10us). This is not deterministic
# when running stock Linux. To get better use a RTOS or the PRUSS
sleep(0.001)
# Set Trigger to LOW
# This will cause the HC-SR04 to output an ultrasonic pulse and start
# listening for the return
GPIO.output("P9_14", GPIO.LOW)
# Sleep for a bit so that we don't send out another pulse before the
# previous one has finished
sleep(0.05)
# 58.77 is the magic number for cm per microsecond
d = min(200.0, max(self.delta.microseconds/58.77,17.0))
# only include value if it is reasonable/valid
if d > 17 and d < 200:
measures.insert(0, d)
if len(measures) > 3:
measures.pop()
# take an average of the last few measures as the distance
if len(measures) > 0:
self.distance = sum(measures)/len(measures)
else:
self.distance = 20.0
if self.distance < 20 and not self.OBSTACLE:
self.OBSTACLE = True
if self.angle < 90:
self.sendJSON("obstacle", {"sensor": "P9_12", "name": "left"})
else:
self.sendJSON("obstacle", {"sensor": "P9_11", "name": "right"})
self.OBSTACLE = False
return
def do_beep(self, duration):
PWM.start(self.SPEAKER_PWM, 50, 3000)
sleep(duration)
PWM.stop(self.SPEAKER_PWM)
def do_forward(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("both", "fwd", self.SPEED)
def do_reverse(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("both", "rev", self.SPEED)
def do_stop(self, set_speed):
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.HIGH)
GPIO.output(self.LEDS_GPIO["GREEN_pin"], GPIO.LOW)
self.saber.stop()
def turn_left(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("left", "fwd", self.SPEED)
self.saber.driveMotor("right", "rev", self.SPEED)
delay = 1 - (float(self.SPEED)/100)
sleep(delay)
if self.CMD in ["fwd", "rev"]:
getattr(self, self.commands[self.CMD])(int(self.SPEED))
else:
self.do_stop(self.SPEED)
def turn_right(self, set_speed):
if set_speed != None:
self.SPEED = set_speed
self.saber.driveMotor("right", "fwd", self.SPEED)
self.saber.driveMotor("left", "rev", self.SPEED)
delay = 1 - (float(self.SPEED)/100)
sleep(delay)
if self.CMD in ["fwd", "rev"]:
getattr(self, self.commands[self.CMD])(int(self.SPEED))
else:
self.do_stop(self.SPEED)
def set_speed(self, new_speed):
self.SPEED = new_speed
if (self.CMD != "speed") and (self.CMD != None):
getattr(self, self.commands[self.CMD])(int(self.SPEED))
def sendJSON(self, event, data):
try:
self.sendMessage(json.dumps({"event": event, "data": data}))
except Exception as n:
print "sendJSON: ", n
def handleMessage(self):
if self.data is None:
self.data = ''
msg = json.loads(str(self.data))
if msg['event'] == 'drive' and msg['data']['cmd'] in self.commands:
if msg['data']['speed'] == None:
msg['data']['speed'] = self.SPEED
GPIO.output(self.LEDS_GPIO["RED_pin"], GPIO.LOW)
GPIO.output(self.LEDS_GPIO["GREEN_pin"], GPIO.HIGH)
getattr(self, self.commands[msg['data']['cmd']])(int(msg['data']['speed']))
if msg['data']['cmd'] in ["fwd", "rev", "stop"]:
self.CMD = msg['data']['cmd']
self.SPEED = int(msg['data']['speed'])
self.sendJSON("ack", {"cmd": "%s %d" % (self.commands[msg['data']['cmd']], self.SPEED)})
elif (msg["event"] == 'fetch'):
if msg['data']['cmd'] == 'angle':
self.sendJSON("angle", {"angle": "%2.2f" % (self.angle)})
elif msg['data']['cmd'] == 'heading':
self.sendJSON("heading", {"angle": "%2.2f" % (self.mag.getHeading())})
def handleConnected(self):
print self.address, 'connected'
def handleClose(self):
self.do_beep(0.25)
for PIN in self.PROXIMITY_GPIO:
GPIO.remove_event_detect(PIN)
GPIO.remove_event_detect(self.STOP_BUTTON)
GPIO.remove_event_detect(self.ECHO_RETURN)
self.SCAN = False
sleep(1)
PWM.stop(self.SERVO_PWM)
PWM.stop(self.SPEAKER_PWM)
self.saber.stop()
print self.address, 'closed'