class AirPuff():
def __init__(self, portName):
self.ser = None
if portName is None or portName == "None":
return
self.ser = SerialPort(serialPortName=portName, baudRate=57600)
self.ser.startReadThread()
def puff(self):
if self.ser is None:
print "No air puff connected."
return
self.ser.write("x")
def open_valve(self):
if self.ser is None:
print "No air puff connected."
return
self.ser.write("1")
def close_valve(self):
if self.ser is None:
print "No air puff connected."
return
self.ser.write("0")
class AirPuff():
def __init__(self, portName):
self.ser = None
if portName is None or portName == "None":
return
self.ser = SerialPort(serialPortName=portName, baudRate=57600)
self.ser.startReadThread()
def puff(self):
if self.ser is None:
print "No air puff connected."
return
self.ser.write("x")
def open_valve(self):
if self.ser is None:
print "No air puff connected."
return
self.ser.write("1")
def close_valve(self):
if self.ser is None:
print "No air puff connected."
return
self.ser.write("0")
class ArduinoSensor():
def __init__(self, serialPortName):
self.serialPort = SerialPort(serialPortName)
self.serialPort.startReadThread()
#thresholds (reasonable defaults given)
self.lickThreshold = 1000
self.tapThreshold = 1000
self.irThreshold = 100
#sensor state variables
self._lick = False
self._tap = False
#sensor history entries are tuples of (booleanState, timestamp).
self.historyLick = []
self.historyTap = []
#full sensor data, not really used except for debugging
self.lickRaw = []
self.tapRaw = []
#reference to objects we need to talk to
self.task = None
self.tracer = None
def checkSensors(self):
# Read from serial
updates = self.serialPort.getUpdates()
# Add sensor data to log
for update in updates:
(data, t) = update
toks = data.rstrip().split()
if len(toks) < 2 or not toks[1].isdigit:
continue
try:
level = int(toks[1])
if toks[0] == "L":
self.lickRaw.append((level,t))
self.reportChange("lick " + str(level), t)
self.set_lick_state(level,t)
if toks[0] == "T":
self.tapRaw.append((level,t))
self.reportChange("tap " + str(level), t)
self.set_tap_state(level,t)
except:
continue
# parse / update state as needed
for update in updates:
(evt, t) = update
if evt == "Lx":
self.lick = True
self.historyLick.append(update)
elif evt == "Lo":
self.lick = False
self.historyLick.append(update)
elif evt == "Tx":
self.tap = True
self.historyTap.append(update)
elif evt == "To":
self.tap = False
self.historyTap.append(update)
return updates
def set_lick_state(self, level, t):
"""
Sets the state of the lick sensor.
If the state has changed (crossed threshold), change the state and note it in the log.
"""
if not self.lick and level > self.lickThreshold:
self.lick = True
self.historyLick.append(("Lx", t))
elif self.lick and level < self.lickThreshold:
self.lick = False
self.historyLick.append(("Lo", t))
def set_tap_state(self, level, t):
"""
Sets the state of the tap sensor.
If the state has changed (crossed threshold), change the state and note it in the log.
"""
if not self.tap and level > self.tapThreshold:
self.tap = True
self.historyTap.append(("Tx", t))
elif self.tap and level < self.tapThreshold:
self.tap = False
self.historyTap.append(("To", t))
def reportChange(self, evt, timestamp):
"""
Reports lick and tap events to the task.
Sends sensor readings to tracer.
"""
if self.task is not None:
self.task.notify(evt, timestamp)
if self.tracer is not None:
self.tracer.notify(evt, timestamp)
'''
Properties. All setters call reportChange().
'''
@property
def lick(self):
return _lick
@lick.setter
def lick(self, value):
self._lick = value
if self._lick:
self.reportChange("Lx", time.time())
else:
self.reportChange("Lo", time.time())
@property
def tap(self):
return _tap
@tap.setter
def tap(self, value):
self._tap = value
if self._tap:
self.reportChange("Tx", time.time())
else:
self.reportChange("To", time.time())
'''
OK, now we have some boolean functions.
These are used by CauseEffect objects, which poll sensor states to determine task progression.
'''
# lick sensor
def isLicking(self):
return self.lick
def isNotLicking(self):
return not self.lick
def isTapping(self):
return self.tap
def isNotTapping(self):
return not self.tap
def hasLickedSince(self, secondsAgo):
#checks whether there was a recent lick (no more than secondsAgo)
if self.isLicking:
#is licking right now
return True
if len(self.historyLick) == 0:
#never licked
return False
#last event must have been an offset; check when it happened
(state, lickOffsetTime) = self.historyLick[-1]
return (time.time() - lickOffsetTime) <= secondsAgo
def hasNotLickedSinceAtLeast(self, secondsAgo):
return not hasLickedSince(self, secondsAgo)
class ArduinoSensor():
def __init__(self, serialPortName):
self.serialPort = SerialPort(serialPortName)
self.serialPort.startReadThread()
#thresholds (reasonable defaults given)
self.lickThreshold = 1000
self.tapThreshold = 1000
self.irThreshold = 100
#sensor state variables
self._lick = False
self._tap = False
#sensor history entries are tuples of (booleanState, timestamp).
self.historyLick = []
self.historyTap = []
#full sensor data, not really used except for debugging
self.lickRaw = []
self.tapRaw = []
#reference to objects we need to talk to
self.task = None
self.tracer = None
def checkSensors(self):
# Read from serial
updates = self.serialPort.getUpdates()
# Add sensor data to log
for update in updates:
(data, t) = update
toks = data.rstrip().split()
if len(toks) < 2 or not toks[1].isdigit:
continue
try:
level = int(toks[1])
if toks[0] == "L":
self.lickRaw.append((level, t))
self.reportChange("lick " + str(level), t)
self.set_lick_state(level, t)
if toks[0] == "T":
self.tapRaw.append((level, t))
self.reportChange("tap " + str(level), t)
self.set_tap_state(level, t)
except:
continue
# parse / update state as needed
for update in updates:
(evt, t) = update
if evt == "Lx":
self.lick = True
self.historyLick.append(update)
elif evt == "Lo":
self.lick = False
self.historyLick.append(update)
elif evt == "Tx":
self.tap = True
self.historyTap.append(update)
elif evt == "To":
self.tap = False
self.historyTap.append(update)
return updates
def set_lick_state(self, level, t):
"""
Sets the state of the lick sensor.
If the state has changed (crossed threshold), change the state and note it in the log.
"""
if not self.lick and level > self.lickThreshold:
self.lick = True
self.historyLick.append(("Lx", t))
elif self.lick and level < self.lickThreshold:
self.lick = False
self.historyLick.append(("Lo", t))
def set_tap_state(self, level, t):
"""
Sets the state of the tap sensor.
If the state has changed (crossed threshold), change the state and note it in the log.
"""
if not self.tap and level > self.tapThreshold:
self.tap = True
self.historyTap.append(("Tx", t))
elif self.tap and level < self.tapThreshold:
self.tap = False
self.historyTap.append(("To", t))
def reportChange(self, evt, timestamp):
"""
Reports lick and tap events to the task.
Sends sensor readings to tracer.
"""
if self.task is not None:
self.task.notify(evt, timestamp)
if self.tracer is not None:
self.tracer.notify(evt, timestamp)
'''
Properties. All setters call reportChange().
'''
@property
def lick(self):
return _lick
@lick.setter
def lick(self, value):
self._lick = value
if self._lick:
self.reportChange("Lx", time.time())
else:
self.reportChange("Lo", time.time())
@property
def tap(self):
return _tap
@tap.setter
def tap(self, value):
self._tap = value
if self._tap:
self.reportChange("Tx", time.time())
else:
self.reportChange("To", time.time())
'''
OK, now we have some boolean functions.
These are used by CauseEffect objects, which poll sensor states to determine task progression.
'''
# lick sensor
def isLicking(self):
return self.lick
def isNotLicking(self):
return not self.lick
def isTapping(self):
return self.tap
def isNotTapping(self):
return not self.tap
def hasLickedSince(self, secondsAgo):
#checks whether there was a recent lick (no more than secondsAgo)
if self.isLicking:
#is licking right now
return True
if len(self.historyLick) == 0:
#never licked
return False
#last event must have been an offset; check when it happened
(state, lickOffsetTime) = self.historyLick[-1]
return (time.time() - lickOffsetTime) <= secondsAgo
def hasNotLickedSinceAtLeast(self, secondsAgo):
return not hasLickedSince(self, secondsAgo)