Skip to content

Commit

Permalink
testing lightshow
Browse files Browse the repository at this point in the history
  • Loading branch information
Unreal-Dan committed Aug 15, 2024
1 parent d097c54 commit 6480d1c
Show file tree
Hide file tree
Showing 6 changed files with 6,128 additions and 0 deletions.
1 change: 1 addition & 0 deletions docs/_config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ heading_anchors: true
markdown: kramdown
include:
- assets/images/
- assets/js/
- core/
- desktop/
- gloves/
Expand Down
248 changes: 248 additions & 0 deletions docs/assets/js/Lightshow.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,248 @@
export default class Lightshow {
static count = 0;

constructor(vortexLib, canvasId, options = {}) {
this.id = Lightshow.count++;
this.type = options.type || 'scrolling'; // Default to 'scrolling' if no type is provided
this.setupCanvas(canvasId);
this.setupOffScreenCanvas(canvasId);
this.initializeVortex(vortexLib);
this.configureDisplay(options.sectionCount || 20);
this.loadPatData();
this.clearCanvas();
this.boundScrollingDraw = this.scrollingDraw.bind(this); // Bind the scrolling draw function
this.boundFlashDraw = this.flashDraw.bind(this); // Bind the flash draw function
}

setupCanvas(canvasId) {
this.canvas = document.getElementById(canvasId);
if (!this.canvas) throw new Error(`Canvas with ID ${canvasId} not found`);

const container = this.canvas.parentElement; // Get the container of the canvas
// Set the canvas dimensions based on its container size
this.canvas.width = container.offsetWidth;
this.canvas.height = container.offsetHeight;

this.ctx = this.canvas.getContext('2d');
this.patId = canvasId.split('_')[1];
}

setupOffScreenCanvas(canvasId) {
this.offScreenCanvas = document.createElement('canvas');
this.offScreenCanvas.width = this.canvas.width;
this.offScreenCanvas.height = this.canvas.height;
this.offCtx = this.offScreenCanvas.getContext('2d');
}

initializeVortex(vortexLib) {
this.vortexLib = vortexLib;
this.vortex = new vortexLib.Vortex();
this.vortex.init();
this.vortex.setLedCount(1);
this.modes = this.vortex.engine().modes();
this.vortexLib.RunTick(this.vortex); // Initial tick
}

configureDisplay(sectionCount) {
this.sectionCount = sectionCount;
this.sectionWidth = this.canvas.width / this.sectionCount;
this.history = [];
this.animationFrameId = null;
}

loadPatData() {
const patDataAttr = this.canvas.getAttribute('data-pat');
if (patDataAttr) {
this.applyPatData(JSON.parse(patDataAttr));
} else {
console.log("No pat data for lightshow " + this.canvas.id);
}
}

clearCanvas() {
this.offCtx.fillStyle = 'rgba(0, 0, 0)';
this.offCtx.fillRect(0, 0, this.offScreenCanvas.width, this.offScreenCanvas.height);
}

applyPatData(patData) {
this.patData = patData;
let colorSet = new this.vortexLib.Colorset();

patData.colorset.forEach(hex => {
let color = hex.replace('0x', '#');
let rgb = /^#?([a-f\d]{2})([a-f\d]{2})([a-f\d]{2})$/i.exec(color);
if (rgb) {
colorSet.addColor(new this.vortexLib.RGBColor(
parseInt(rgb[1], 16),
parseInt(rgb[2], 16),
parseInt(rgb[3], 16)
));
}
});

let previewMode = this.getCurrentMode();
if (!previewMode) return;

previewMode.setColorset(colorSet, this.ledCount());
this.setPatternAndArgs(previewMode);
}

getCurrentMode() {
return this.modes.curMode();
}

setPatternAndArgs(mode) {
let patternId = this.vortexLib.intToPatternID(this.patData.pattern_id);
mode.setPattern(patternId, this.ledCount(), null, null);

let patternArgs = new this.vortexLib.PatternArgs();
this.patData.args.forEach(arg => patternArgs.addArgs(arg));
this.vortex.setPatternArgs(this.ledCount(), patternArgs, false);

mode.init(); // Re-initialize to apply new args
}

ledCount() {
return this.vortex.engine().leds().ledCount();
}

scrollingDraw() {
if (this._pause) return;

// Perform drawing operations on the off-screen canvas
this.offCtx.fillStyle = 'rgba(0, 0, 0, 0.4)';
this.offCtx.fillRect(0, 0, this.offScreenCanvas.width, this.offScreenCanvas.height);

const newColor = this.vortexLib.RunTick(this.vortex);
if (newColor) {
this.history.push({ color: newColor[0], x: 0, opacity: 1 });
}

this.history.forEach(segment => this.drawSegment(segment));

this.history = this.history.filter(segment => segment.x < this.offScreenCanvas.width);

// Copy the off-screen canvas to the on-screen canvas in one operation
this.ctx.drawImage(this.offScreenCanvas, 0, 0);

this.animationFrameId = requestAnimationFrame(this.boundScrollingDraw);
}

flashDraw() {
if (this._pause) return;

const newColor = this.vortexLib.RunTick(this.vortex);
if (newColor) {
this.ctx.fillStyle = `rgba(${newColor[0].red}, ${newColor[0].green}, ${newColor[0].blue}, 1)`;
this.ctx.fillRect(0, 0, this.canvas.width, this.canvas.height);
}

this.animationFrameId = requestAnimationFrame(this.boundFlashDraw);
}

drawSegment(segment) {
if (segment.color.red !== 0 || segment.color.green !== 0 || segment.color.blue !== 0) {
this.offCtx.fillStyle = `rgba(${segment.color.red}, ${segment.color.green}, ${segment.color.blue}, ${segment.opacity})`;
this.offCtx.fillRect(segment.x, 0, this.sectionWidth, this.offScreenCanvas.height);
}
segment.x += this.sectionWidth;
}

start() {
this._pause = false;
if (!this.animationFrameId) {
if (this.type === 'flashing') {
this.animationFrameId = requestAnimationFrame(this.boundFlashDraw);
} else {
this.animationFrameId = requestAnimationFrame(this.boundScrollingDraw);
}
}
}

stop() {
this._pause = true;
if (this.animationFrameId) {
cancelAnimationFrame(this.animationFrameId);
this.animationFrameId = null;
}
}

// get the pattern
getPattern() {
const demoMode = this.modes.curMode();
return demoMode.getPattern(this.vortexLib.LedPos.LED_0);
}

// set the pattern
setPattern(patternIDValue) {
// the selected dropdown pattern
const selectedPattern = this.vortexLib.PatternID.values[patternIDValue];
// grab the 'preview' mode for the current mode (randomizer)
let demoMode = this.modes.curMode();
// set the pattern of the demo mode to the selected dropdown pattern on all LED positions
// with null args and null colorset (so they are defaulted and won't change)
demoMode.setPattern(selectedPattern, this.ledCount(), null, null);
// re-initialize the demo mode so it takes the new args into consideration
demoMode.init();
}

// get colorset
getColorset() {
const demoMode = this.modes.curMode();
if (!demoMode) {
return new this.vortexLib.Colorset();
}
return demoMode.getColorset(this.vortexLib.LedPos.LED_0);
}

// update colorset
setColorset(colorset) {
// grab the 'preview' mode for the current mode (randomizer)
let demoMode = this.modes.curMode();
if (!demoMode) {
return;
}
// set the colorset of the demo mode
demoMode.setColorset(colorset, this.ledCount());
// re-initialize the demo mode because num colors may have changed
demoMode.init();
}

// add a color to the colorset
addColor(r, g, b) {
let set = this.getColorset();
set.addColor(new this.vortexLib.RGBColor(r, g, b));
this.setColorset(set);
}

// delete a color from the colorset
delColor(index) {
let set = this.getColorset();
if (set.numColors() <= 1) {
return;
}
set.removeColor(index);
this.setColorset(set);
}

// update a color in the colorset
updateColor(index, r, g, b) {
let set = this.getColorset();
set.set(index, new this.vortexLib.RGBColor(r, g, b));
this.setColorset(set);
}

// randomize the pattern
randomize() {
this.vortex.openRandomizer();
this.vortex.longClick(0);
this.vortex.shortClick(0);
this.vortex.longClick(0);
// whatever reason we need 3 ticks to clear through the longClick
// randomize idk it really shouldn't take that long
for (let i = 0; i < 3; ++i) {
this.vortexLib.RunTick(this.vortex);
}
}
}

Loading

0 comments on commit 6480d1c

Please sign in to comment.