leds/pattern.py

import sys
import json
import random

this = sys.modules[__name__]

this.encoded = None
this.pattern = None
this.values = {
    "stack": 0,
    "r": 0,
    "g": 0,
    "b": 0,
    "tick": 0,
    "index": 0
}


def constant(target, arg):
    return arg


def add(target, arg):
    res = target + arg
    if res > 255:
        res = 255
    elif res < 0:
        res = 0
    return res


def sub(target, arg):
    res = target - arg
    if res > 255:
        res = 255
    elif res < 0:
        res = 0
    return res


def rand(target, arg):
    return random.randrange(0, 255)


def _apply(target, arg, func):
    if type(arg is int):
        return func(target, arg)
    elif type(arg is str) and arg in this.values:
        return func(target, this.values[arg])


def apply(targets, args, func):
    for target in range(len(targets)):
        if this.values[targets[target]['channel']] != None:
            if target in args:
                this.values[targets[target]['channel']] = _apply(
                    this.values[targets[target]['channel']], args[target], func)
            else:
                this.values[targets[target]['channel']] = _apply(
                    this.values[targets[target]['channel']], 0, func)


this.instructions = {
    "CONSTANT": constant,
    "ADD": add,
    "SUB": sub,
    "RAND": rand
}


def default(index, tick):
    return (((index + tick) * 5) % 255, (tick * 42) % 255, (tick * 50) % 255)


def pat(index, tick, previous_values):
    if this.pattern != None:
        for i in range(len(this.pattern)):
            name = this.pattern[i]['instruction']['name']
            targets = this.pattern[i]['instruction']['targets']
            args = []
            if 'args' in this.pattern[i]['instruction']:
                args = this.pattern[i]['instruction']['args']
            if this.instructions[name] != None:
                apply(targets, args, this.instructions[name])
        r = this.values["r"]
        g = this.values["g"]
        b = this.values["b"]
        this.values["r"] = 0
        this.values["g"] = 0
        this.values["b"] = 0
        this.values["stack"] = 0
        return (r, g, b)
    else:
        return default(index, tick)


def parse(str):
    this.encoded = str
    print(this.encoded)
    this.pattern = json.loads(this.encoded)