-- Scripts juste pour tester l'effet train
-- tout sur la couleur: https://www.w3schools.com/colors/default.asp
-- roue des couleurs: https://iro.js.org/?ref=oldsite

print("\n a_train3.lua zf181215.1434 \n")

nbleds=36
fade1=0.05   fade2=0.2   fade3=0.4   fade4=1
R1=255   G1=0   B1=0
R2=255   G2=0   B2=0
train_speed=50
ws2812.init()
myLedStrip = ws2812.newBuffer(nbleds, 3)
myLedStrip1 = ws2812.newBuffer(nbleds, 3) 
myLedStrip2 = ws2812.newBuffer(nbleds, 3) 

function RGB_clear()
    myLedStrip:fill(0, 0, 0)   ws2812.write(myLedStrip)
end

function RGB_reform()
    rR1=B1   rG1=R1   rB1=G1
    rR2=B2   rG2=R2   rB2=G2
end

function train1_fill()
    RGB_reform()
    myLedStrip1:fill(0,0,0)
    myLedStrip1:set(1, rR1*fade1, rG1*fade1, rB1*fade1)
    myLedStrip1:set(2, rR1*fade2, rG1*fade2, rB1*fade2)
    myLedStrip1:set(3, rR1*fade3, rG1*fade3, rB1*fade3)
    myLedStrip1:set(4, rR1*fade4, rG1*fade4, rB1*fade4)
end

function train2_fill()
    RGB_reform()
    myLedStrip2:fill(0,0,0)
    myLedStrip2:set(nbleds,   rR2*fade1, rG2*fade1, rB2*fade1)
    myLedStrip2:set(nbleds-1, rR2*fade2, rG2*fade2, rB2*fade2)
    myLedStrip2:set(nbleds-2, rR2*fade3, rG2*fade3, rB2*fade3)
    myLedStrip2:set(nbleds-3, rR2*fade4, rG2*fade4, rB2*fade4)
end

function train_mix()
    myLedStrip:mix(255, myLedStrip1, 255, myLedStrip2)
end

function train_shift()
    myLedStrip1:shift(1, ws2812.SHIFT_CIRCULAR)
    myLedStrip2:shift(-1, ws2812.SHIFT_CIRCULAR)
end

function train_write()
    ws2812.write(myLedStrip)
end

function train_start()
    train3timer1=tmr.create()
    tmr.alarm(train3timer1, train_speed,  tmr.ALARM_AUTO, function()
        train_shift()
        train_mix()
        train_write()
    end)
end

function train_stop()
    tmr.unregister(train3timer1)
    RGB_clear()
end

RGB_clear()
train1_fill()
train2_fill()
train_start()