kopia lustrzana https://github.com/GuyCarver/MicroPython
118 wiersze
3.1 KiB
Python
118 wiersze
3.1 KiB
Python
#cat trainer and treat dispenser.
|
|
|
|
from machine import Pin
|
|
from time import sleep_ms
|
|
from uos import urandom
|
|
|
|
class cat(object):
|
|
'''Control activation of triggers and LEDs, look for input and dispense a treat
|
|
as a result. The LED indicates which switch is active. If that switch is
|
|
triggered, a treat is dispensed using an archimedes screw device and a new switch is activated.'''
|
|
|
|
_LEDPINS = (2, 0, 4, 5)
|
|
_SWITCHPINS = (15, 13, 12, 10)
|
|
_SCREWPIN = 16
|
|
_RANDOMCOUNT = 256
|
|
_FRAMERATE = 100 #ms
|
|
_ONVALUE = 100
|
|
_OFFVALUE = 0
|
|
_SCREWTIME = 5500
|
|
_DELAYTIME = _FRAMERATE * 50 #Delay for 5 seconds between switch activations.
|
|
|
|
def __init__( self ):
|
|
super(cat, self).__init__()
|
|
print('Starting.')
|
|
|
|
rs = urandom(cat._RANDOMCOUNT)
|
|
div = 255 / len(cat._LEDPINS)
|
|
#Create array of random indexes for the LED/switches.
|
|
self._randoms = [0 for r in rs] #int(r / div) for r in rs]
|
|
self._randindex = 0
|
|
|
|
self._active = -1
|
|
self._delay = 0
|
|
|
|
self._leds = [Pin(p, Pin.OUT) for p in cat._LEDPINS]
|
|
self._switches = [Pin(p, Pin.IN) for p in cat._SWITCHPINS]
|
|
|
|
self._screwpin = Pin(cat._SCREWPIN, Pin.OUT)
|
|
self.screwon = False
|
|
|
|
self._prev = 0
|
|
|
|
self.next() #Pick 1st active LED/switch.
|
|
|
|
@property
|
|
def screwon( self ):
|
|
return self._screw > 0
|
|
|
|
@screwon.setter
|
|
def screwon( self, aTF ):
|
|
''' Turn screw drive on/off. '''
|
|
print('Screw On:', aTF)
|
|
if aTF:
|
|
self._screw = cat._SCREWTIME #Start screw on timer.
|
|
self._screwpin.value(cat._ONVALUE)
|
|
else:
|
|
self._screwpin.value(cat._OFFVALUE)
|
|
self._screw = 0 #reset the screw on timer to be done.
|
|
|
|
def noactive( self ):
|
|
''' Turn all LED/switches off. '''
|
|
self._active = -1
|
|
self.updateleds()
|
|
|
|
def updateleds( self ):
|
|
''' Update LED to turn the correct one on. '''
|
|
for i, l in enumerate(self._leds):
|
|
l.value(1 if i == self._active else 0)
|
|
|
|
def next( self ):
|
|
''' Pick a new switch/LED to make active. '''
|
|
self._active = self._randoms[self._randindex]
|
|
self._randindex += 1
|
|
if self._randindex >= len(self._randoms):
|
|
self._randindex = 0
|
|
|
|
self.updateleds()
|
|
|
|
def dispense( self ):
|
|
''' Dispense a treat and pick a new active switch. '''
|
|
self.screwon = True
|
|
self._delay = cat._DELAYTIME
|
|
self.noactive()
|
|
|
|
def check( self ):
|
|
''' Check to see if switch was hit and released. '''
|
|
v = self._switches[self._active].value()
|
|
if v != self._prev:
|
|
if v == 0: #If the switch isn't touched, but it was before.
|
|
self.dispense()
|
|
self._prev = v
|
|
|
|
def updatescrew( self, dt ):
|
|
''' update screw timer and shut off when up. '''
|
|
if self.screwon:
|
|
self._screw -= dt
|
|
if self._screw <= 0:
|
|
self.screwon = False
|
|
|
|
def updateinput( self, dt ):
|
|
''' update led lighting. '''
|
|
if self._delay:
|
|
self._delay -= dt
|
|
if self._delay <= 0:
|
|
self.next()
|
|
self._delay = 0
|
|
else:
|
|
self.check()
|
|
|
|
def run( self ):
|
|
'''Main loop for cat treat dispenser.'''
|
|
while 1:
|
|
self.updatescrew(cat._FRAMERATE)
|
|
self.updateinput(cat._FRAMERATE)
|
|
sleep_ms(cat._FRAMERATE)
|
|
|
|
|