server.js

//// Node.js server ////

//require('look').start();
// var memwatch = require('memwatch');
// var agent = require('webkit-devtools-agent');
// agent.start()

var controller = require('./controller.js');
var drawers1D = require('./drawers1D.js');
var alienblob = require('./alienblob.js');
var bzr = require('./bzr.js');
var video = require('./video.js');
var off = require('./off.js');
var gradient = require('./gradient.js');
var leds = require('./leds.js');
var palette = require('./palette.js');
var camera = require('./camera.js');
var allBaseColors = require('./kuler.js').allBaseColors;

var fs = require('fs');
var express = require('express');
var app = express();
var http = require('http').Server(app);
var socketIO = require('socket.io');
var io = socketIO(http);
var path = require('path');

//// Server config variables ////

// *** These must match settings on the Teensy ***
var WIDTH = 64;
var HEIGHT = 32;
// *** End settings match ***

var DEPTH = 24; // bit depth: 24 or 48
var NUM_COLORS = 1<<12; // colors in the gradient of each palette
var FPS = 60;
var CAMERA_FPS = 10;
var START_DRAWER = 'Bzr';
var DRAWER_CHANGE_INTERVAL = 10000;
var CAM_SIZE = [1280, 960];//[640, 480];
var layoutLeftToRight = false; // only used for serial port connections
var ENABLE_AUDIO = false;
var ENABLE_CAMERA = false;
var UPDATE_IMAGE_INTERVAL = 0;

//// End Server config variables ///


//// Global variables ////
var device;
if (process.argv.length > 2) {
    device = process.argv[2];
} else {
    device = "/dev/ttyACM0"; // for direct serial access to Teensy
    // device = "ws://localhost:7890"; // for fadecandy
}

// start the camera used by face detection and the VideoDrawer
if (ENABLE_CAMERA) {
    var cam = new camera.Camera(CAM_SIZE, CAMERA_FPS);
}

var allDrawers = 
    [new drawers1D.GradientDrawer(), 
     new drawers1D.WipeDrawer(),
     new drawers1D.WaveDrawer(),
     new drawers1D.SparkleDrawer(),
     new drawers1D.PulseDrawer(),
     new alienblob.AlienBlobDrawer(WIDTH, HEIGHT, NUM_COLORS),
     new bzr.BzrDrawer(WIDTH, HEIGHT, NUM_COLORS),
     new video.VideoDrawer(WIDTH, HEIGHT, NUM_COLORS, cam),
     // new gradient.GradientDrawer(WIDTH, HEIGHT, NUM_COLORS),
     new off.OffDrawer(WIDTH, HEIGHT, NUM_COLORS)
    ];

var availableDrawers = {};  // drawers that can be selected from the UI
for (var i = 0; i < allDrawers.length; i++) {
    if ((HEIGHT == 1 && allDrawers[i].type().indexOf("1D") > -1) ||
	      (HEIGHT > 1 && allDrawers[i].type().indexOf("2D") > -1)) {
	       availableDrawers[allDrawers[i].name] = allDrawers[i];
    }
}

//// Start the patterns ////
console.log('start');
var paletteMgr = new palette.PaletteManager(allBaseColors, NUM_COLORS);
var leds = new leds.LEDs(WIDTH, HEIGHT, DEPTH, device, layoutLeftToRight, UPDATE_IMAGE_INTERVAL);
var control = new controller.Controller(leds, paletteMgr, availableDrawers, 
					START_DRAWER, DRAWER_CHANGE_INTERVAL, cam, ENABLE_AUDIO);

function func() { 
    control.loop(); 
};
setInterval(func, 1000 / FPS);

//// Start the http server ///
app.get('/', function(req, res) {
    res.sendFile(path.join(__dirname, 'public', 'index.html'));
});

app.get('/image', function(req, res) {
    var image = "data:image/png;base64," + leds.pngData.toString('base64');
    //leds.pngData = '';
    res.send(image);
});

app.use(express.static(path.join(__dirname, 'public')));

http.listen(8080, function(){
    console.log('listening on *:8080');
});



//// Handle the websockets connection to the client ////
io.sockets.on('connection', function (socket) {
    // get the allowed programs
    socket.on('get drawers', function (data, fn) {
      	console.log('get drawers: ' + JSON.stringify(data));
      	var settings = control.getSettings();
      	fn({active: {name: control.currDrawer.name, ranges: settings.ranges, values: settings.values},
                  allNames: Object.keys(availableDrawers)});
    });
    
    // set the running program, return it's settings
    // plus the palette
    socket.on('set drawer', function (drawerName, fn) {
      	console.log('set drawer: ' + drawerName);
      	control.changeDrawer(availableDrawers[drawerName]);
      	var settings = control.getSettings();
      	fn({drawer: drawerName, ranges: settings.ranges, values: settings.values});
    });
    
    // update the settings on the server
    socket.on('set settings', function (settingVals, fn) {
      	console.log('set settings: ' + JSON.stringify(settingVals));
      	control.setSettings(settingVals);
      	//fn();
    });
    
    // randomize the settings on the server
    socket.on('randomize settings', function (data, fn) {
      	console.log('randomize settings: ' + data);
      	control.randomizeSettings();
      	fn(control.getSettings().values);
    });
});

function cleanup() {
    if (ENABLE_CAMERA)
      	cam.stop();
    leds.stop(function(err) { 
	     if (err) console.log("Error when stopping leds: " + err);
	         process.exit();
    });
}

process.on('SIGINT', function() {
    console.log("Caught sigint");
    cleanup();
});
process.on('SIGTERM', function() {
    console.log("Caught sigterm");
    cleanup();
});