#import utime, badge, binascii
import badge

counter = 0
direction = True
serviceMode = 0

# This function gets called by the home application at boot
def setup():
    global serviceMode, brightness, color_pick
    serviceMode = badge.nvs_get_u8('stijlrijder', 'serviceMode', 0)
    brightMode  = badge.nvs_get_u8('stijlrijder', 'brightness', 2)

    # Translate brightness setting to percentage
    if(brightMode<1):
        brightness = 1
    elif(brightMode == 1):
        brightness = 10
    elif(brightMode == 2):
        brightness = 50
    else:
        brightness = 100

    # Set the appropriate color pick
    if(serviceMode<1):
        # print "STIJLRIJDER Disabled! Please enable in app!"
        return True
    elif(serviceMode==1):
        # GREEN
        color_pick = [255, 0, 0, 0]
    elif(serviceMode==2):
        # RED
        color_pick = [0, 255, 0, 0]
    elif(serviceMode==3):
        # BLUE
        color_pick = [0, 0, 255, 0]
    elif(serviceMode==4):
        # PINK
        color_pick = [0, 255, 155, 0]
    elif(serviceMode==5):
        # WHITE
        color_pick = [0, 0, 0, 255]
    elif(serviceMode==6):
        # PURPLE
        color_pick = [0, 255, 255, 0]

    badge.leds_enable()

# Set led(pos) to GRBW values in the ledbuffer
def set_led(ledbuffer, pos, grbw):
    pos = pos*4
    ledbuffer[pos]   = grbw[0]
    ledbuffer[pos+1] = grbw[1]
    ledbuffer[pos+2] = grbw[2]
    ledbuffer[pos+3] = grbw[3]
    return ledbuffer

# grbw = arr[4], phase = 0-10, brightness = 0-100
def dim_pixel(grbw, phase, brightness):
    ret = [0,0,0,0]
    multiply = (phase / 10) * (brightness / 100)
    ret[0] = round(grbw[0] * multiply)
    ret[1] = round(grbw[1] * multiply)
    ret[2] = round(grbw[2] * multiply)
    ret[3] = round(grbw[3] * multiply)
    return ret

def leds(counter, direction):
    global serviceMode, brightness, color_pick
    led = 0

    led = round(counter/10)
               
    if direction:
        prevLed = max(0, led-1)
        nextLed = min(5, led+1)
    else:
        prevLed = min(5, led+1)
        nextLed = max(0, led-1)
    
    # pre-clearing output array
    output = []
    for i in range(0,24):
        output.append(0)

    # Draw current, prev and next led
    brightNext = (counter+5) % 10
    brightPrev = 11 - brightNext

    set_led(output, nextLed, dim_pixel(color_pick, brightNext, brightness))
    set_led(output, prevLed, dim_pixel(color_pick, brightPrev, brightness))
    set_led(output, led,     dim_pixel(color_pick, 10, brightness))
    
    return output


def loop():
    global serviceMode
    global direction
    global counter
        
    if (serviceMode>0): 
        if direction:
            counter = min(50, counter+1)
        else:
            counter = max(0, counter-1)
        if counter==50 or counter==0:
            direction = not direction
        values = leds(counter, direction)
        badge.leds_send_data(bytes(values),24)
        return 5
    return 0