thomas
/
CatToy


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279
							from wifi import Wifi
from config import Config
from toy import Toy
import random
from machine import Timer

max_laser_power = 0.1

limits = [
    # pan_min, pan_max, tilt_min, tilt_max, name
    (84, 120, 53, 76, 'office desk, front right')
]

timerRunning = False
timerData = None
outlineIndex = 0

def buildPage(header, footer):
    html = """<!DOCTYPE html>
    <html>
        <head>
            <title>Cat Toy</title>
        </head>
        <body>
            <h1>Cat Toy</h1>
            %s
            <h2>Laser Pointer</h2>
            <form action="/laser" method="get">
                Turn power:
                <input type="range" name="s" min="0" max="100" step="1" value="%d">
                <input type="submit" value="Set"><br>
            </form>
            <form action="/laser" method="get">
                Or just turn it:
                <input type="submit" name="s" value="Off">
            </form>
            <h2>Pan and Tilt Servos</h2>
            <form action="/servos" method="get">
                Pan: <input type="text" name="s1"><br>
                Tilt: <input type="text" name="s2"><br>
                <input type="submit" value="Move">
            </form>
            <form action="/random_move" method="get">
                <input type="submit" name="s" value="Go to random position">
            </form>
            <h2>Auto Play</h2>
            <form action="/repeat" method="get">
                Limits:
                <select name="limit">
                    %s
                </select><br>
                Steps: <input type="text" name="steps" value="100"><br>
                Duration: <input type="text" name="duration" value="1000">ms<br>
                <input type="submit" name="s" value="Random">
                <input type="submit" name="s" value="Outline"><br>
                Status: %s
            </form>
            %s
        </body>
    </html>
    """

    sl = ""
    for pan_min, pan_max, tilt_min, tilt_max, name in limits:
        val = name.replace(' ', '_').replace(',', '').lower()
        sl += '<option value="' + val + '">' + name + '</option>'
    sl += '<option value="">None</option>'

    status = "No program running"
    if timerRunning:
        status = "Program in progress"

    page = html % (header, int(max_laser_power * 100.0), sl, status, footer)
    return page

random.seed()
t = Toy()

def rootCallback(request):
    return buildPage(
        '<p>Welcome to the Cat Toy interface by <a href="https://www.xythobuz.de">xythobuz</a>.</p>',
        "<p><b>Limits:</b> tMin={} tMax={} pMin={} pMax={}</p>".format(t.tilt_min, t.tilt_max, t.pan_min, t.pan_max)
    )

def servoCallback(request):
    q = request.find("/servos?")
    p1 = request.find("s1=")
    p2 = request.find("s2=")
    if (q < 0) or (p1 < 0) or (p2 < 0):
        print("servo query error: q={} p1={} p2={}".format(q, p1, p2))
        return buildPage(
            '<p>Error: no servo arguments found in URL query string.</p>',
            '<p><a href="/">Back to main page</a></p>'
        )

    servos = [p1, p2]
    result = []
    for p in servos:
        pe = request.find("&s", p)
        if (pe < 0) or (p + 3 >= pe) or (pe - (p + 3) > 3):
            pe = request.find(" HTTP", p)
        if (pe < 0) or (p + 3 >= pe) or (pe - (p + 3) > 3):
            print("servo query error: p={} pe={}".format(p, pe))
            return buildPage(
                '<p>Error parsing query string.</p>',
                '<p><a href="/">Back to main page</a></p>'
            )
        r = request[p + 3 : pe]
        s = int(r)
        result.append(s)

    print("servos: pan={} tilt={}", result[0], result[1])
    t.angle(t.pan, result[0])
    t.angle(t.tilt, result[1])

    return buildPage(
        '<p>Servos move to s1=' + str(result[0]) + ' s2=' + str(result[1]) + '.</p>',
        '<p><a href="/">Back to main page</a></p>'
    )

def laserCallback(request):
    value = 0.0
    text = "off"

    if request.find("?s=") == 10:
        pe = request.find(" HTTP", 10)
        r = request[13 : pe]
        if r != "Off":
            value = int(r)
            text = "to " + str(r) + '%'

    print("laser: {}%".format(value))
    t.laser(value / 100.0)

    return buildPage(
        '<p>Laser turned ' + text + '!</p>',
        '<p><a href="/">Back to main page</a></p>'
    )

def randomMoveCallback(request):
    tilt = random.randint(t.tilt_min, t.tilt_max)
    pan = random.randint(t.pan_min, t.pan_max)
    print("random: tilt={} pan={}".format(tilt, pan))
    t.angle(t.tilt, tilt)
    t.angle(t.pan, pan)
    return buildPage(
        '<p>Random move to pan={} tilt={}</p>'.format(pan, tilt),
        '<p><a href="/">Back to main page</a></p>'
    )

def doMove(pan_min, pan_max, tilt_min, tilt_max, dur):
    tilt = random.randint(tilt_min, tilt_max)
    pan = random.randint(pan_min, pan_max)
    print("random move: tilt={} pan={} duration={}".format(tilt, pan, dur))
    t.angle(t.tilt, tilt)
    t.angle(t.pan, pan)

def doOutline(pan_min, pan_max, tilt_min, tilt_max, dur):
    global outlineIndex
    points = [
        (pan_min, tilt_min),
        (pan_min, tilt_max),
        (pan_max, tilt_max),
        (pan_max, tilt_min)
    ]
    outlineIndex = (outlineIndex + 1) % 4
    pan, tilt = points[outlineIndex]
    print("outline move: tilt={} pan={} duration={}".format(tilt, pan, dur))
    t.angle(t.tilt, tilt)
    t.angle(t.pan, pan)

def timerCallback(unused):
    global timerRunning, timerData

    if not timerRunning:
        return

    pan_min, pan_max, tilt_min, tilt_max, steps, duration, outline = timerData

    dur = duration
    if not outline:
        if dur < 200:
            dur = random.randint(200, 2000)
        else:
            dur = random.randint(200, duration)
    else:
        if dur < 200:
            dur = 500

    if steps > 0:
        steps -= 1
        if not outline:
            doMove(pan_min, pan_max, tilt_min, tilt_max, dur)
        else:
            doOutline(pan_min, pan_max, tilt_min, tilt_max, dur)
        tim = Timer(period = dur, mode=Timer.ONE_SHOT, callback = timerCallback)
    else:
        timerRunning = False
        t.laser(0.0)

    timerData = (pan_min, pan_max, tilt_min, tilt_max, steps, duration, outline)

def startRepeat(pan_min, pan_max, tilt_min, tilt_max, steps, duration, outline):
    global timerRunning, timerData
    timerData = (pan_min, pan_max, tilt_min, tilt_max, steps, duration, outline)

    if not timerRunning:
        timerRunning = True
        t.laser(max_laser_power)
        timerCallback(None)

def stopRepeat():
    global timerRunning, timerData
    timerRunning = False
    t.laser(0.0)

def repeatCallback(request):
    q = request.find("/repeat?")
    pl = request.find("limit=", q)
    ps = request.find("steps=", pl)
    pd = request.find("duration=", ps)
    pp = request.find("s=", pd)
    if (q < 0) or (pl < 0) or (ps < 0) or (pd < 0) or (pp < 0):
        print("repeat query error: q={} pl={} ps={} pd={} pp={}".format(q, pl, ps, pd, pp))
        return buildPage(
            '<p>Error: no repeat arguments found in URL query string.</p>',
            '<p><a href="/">Back to main page</a></p>'
        )

    data = [("limit=", pl), ("steps=", ps), ("duration=", pd), ("s=", pp)]
    result = []
    for s, p in data:
        #print(p)
        pe = request.find("&", p)
        #print(pe)
        if (pe < 0) or (p + len(s) > pe) or (pe - (p + 3) > 40):
            pe = request.find(" HTTP", p)
            #print(pe)
        if (pe < 0) or (p + len(s) > pe) or (pe - (p + 3) > 40):
            print("repeat query error: p={} pe={}".format(p, pe))
            return buildPage(
                '<p>Error parsing query string.</p>',
                '<p><a href="/">Back to main page</a></p>'
            )
        r = request[p + len(s) : pe]
        result.append(r)
        #print()

    print("repeat: limit={} steps={} duration={} s={}".format(result[0], result[1], result[2], result[3]))

    if len(result[0]) == 0:
        stopRepeat()
        return buildPage(
            '<p>Stopped repeated automatic moves!</p>',
            '<p><a href="/">Back to main page</a></p>'
        )

    outline = False
    if result[3].lower() == "outline":
        outline = True

    for pan_min, pan_max, tilt_min, tilt_max, name in limits:
        val = name.replace(' ', '_').replace(',', '').lower()
        if result[0] == val:
            startRepeat(pan_min, pan_max, tilt_min, tilt_max, int(result[1]), int(result[2]), outline)
            break

    return buildPage(
        '<p>Starting moves with limit={} steps={} duration={}</p>'.format(result[0], result[1], result[2]),
        '<p><a href="/">Back to main page</a></p>'
    )

w = Wifi(Config.ssid, Config.password)
w.add_handler("/", rootCallback)
w.add_handler("/servos", servoCallback)
w.add_handler("/laser", laserCallback)
w.add_handler("/random_move", randomMoveCallback)
w.add_handler("/repeat", repeatCallback)
w.listen()