Fine-tune fire mode

[kaka/cakelight.git] / src / kaka / cakelight / mode / FireMode.java

package kaka.cakelight.mode;

import kaka.cakelight.Color;
import kaka.cakelight.LedFrame;
import kaka.cakelight.util.SimplexNoise3D;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

public class FireMode extends AmbientMode {
    private final Color[] colors;
    private SimplexNoise3D noise = new SimplexNoise3D(0);

    public FireMode(Color... colors) {
	assert colors.length > 1;
	List<Color> list = Arrays.asList(colors);
	Collections.reverse(list);
	this.colors = list.toArray(new Color[colors.length]);
    }

    @Override
    protected void updateFrame(LedFrame frame, long time, int count) {
	for (int i = 0; i < config.leds.getCount(); i++) {
	    double x = frame.xOf(i);
	    double y = frame.yOf(i);
	    double v = Math.pow(Math.min(1, Math.max(0, noise.getr(0.0, 1.0, 1, x * 10.0, y + (time / 2500.0), time / 5000.0))), 1.5);
	    frame.setLedColor(i, getColorAt(Math.min(1, Math.max(0, (v + y) * 0.5))));
	}
    }

    private Color getColorAt(double value) { // 0.0 to 1.0
	double localRange = 1.0 / (colors.length - 1);
	int index = (int)(value / localRange);
	double localValue = (value / localRange) - index;
	if (index == colors.length - 1) {
	    return colors[colors.length - 1];
	} else {
	    return colors[index].interpolate(colors[index + 1], localValue);
	}
    }
}