thingzi-logic-twinkly/lib/frame.js

Twinkly lights control via node red

class Frame {
    constructor(ledCount) {
        this.leds = new Array(ledCount);
        this.ledCount = ledCount;
    }

    static repeatedColors(ledCount, colors) {
        let frame = new Frame(ledCount);
        for (let i = 0; i < ledCount; i++) {
            frame.put(i, colors[i % colors.length]);
        }
        return frame;
    }

    static singleColor(ledCount, color) {
        return this.repeatedColors(ledCount, [color]);
    }

    addTwinkle(count, minIntensity) {
        const probabilityFlicker = 0.01;
        for (let i = 0; i < this.ledCount; i++) {
            let isFlicker = Math.random() < probabilityFlicker;
            let intensity = isFlicker ? 0.8 + gaussRandom() / 20 : 0.9 + gaussRandom() / 25;
            this.scaleBrightness(i, Math.min(1.0, Math.max(0, intensity)));
        }
    }

    scaleBrightness(index, scale) {
        scale = Math.max(scale, 0);
        let color = this.get(index);
        if (color instanceof Array) {
            color.forEach(c => scale = limitBrightnessScale(c, scale));
            this.put(index, color.map(c => c * scale));
        } else {
            console.error(`scaleBrightness: unknown color format ${color} len=${color.length}`);
        }
    }

    get(index) {
        return this.leds[index];
    }

    put(index, color) {
        this.leds[index] = color;
    }

    getArray(ledProfile) {
        let colorCount = ledProfile.length;
        let array = new Uint8Array(this.ledCount * colorCount);
        for (let i = 0; i < this.ledCount; i++) {
            let color = this.leds[i];
            if (color instanceof Array && (color.length === 3 || color.length === 4)) {
                let [cR, cG, cB, cW] = color;
                if (colorCount === 4) {
                    array[colorCount * i] = color.length === 4 ? color[3] : 0; // W
                    array[colorCount * i + 1] = color[0]; // R
                    array[colorCount * i + 2] = color[1]; // G
                    array[colorCount * i + 3] = color[2]; // B
                } else {
                    array[colorCount * i] = color[0];     // R
                    array[colorCount * i + 1] = color[1]; // G
                    array[colorCount * i + 2] = color[2]; // B
                }
            } else {
                array[colorCount * i] = 0;     // R
                array[colorCount * i + 1] = 0; // G
                array[colorCount * i + 2] = 0; // B
                console.error(`getArray: unknown color format ${color} len=${color.length}`);
            }
        }
        return array;
    }
}

function gaussRandom() {
    // https://stackoverflow.com/a/36481059/168939
    let u = 0, v = 0;
    while (u === 0) u = Math.random();
    while (v === 0) v = Math.random();
    return Math.sqrt(-2.0 * Math.log(u)) * Math.cos(2.0 * Math.PI * v);
}

/** Limit scale so that it won't cause an overflow */
function limitBrightnessScale(value, scale) {
    if (value * scale > 255.0) {
        return 255.0 / value;
    }
    return scale;
}

exports.Frame = Frame;