271 lines
10 KiB
JavaScript
271 lines
10 KiB
JavaScript
|
|
module.exports = function(RED) {
|
|
"use strict";
|
|
var fs = require('fs');
|
|
var PNG = require('pngjs').PNG;
|
|
var spawn = require('child_process').spawn;
|
|
var execSync = require('child_process').execSync;
|
|
|
|
var hatCommand = __dirname+'/unihat';
|
|
var allOK = true;
|
|
|
|
try {
|
|
var cpuinfo = fs.readFileSync("/proc/cpuinfo").toString();
|
|
if (cpuinfo.indexOf(": BCM") === -1) {
|
|
RED.log.warn("rpi-unicorn : "+RED._("node-red:rpi-gpio.errors.ignorenode"));
|
|
allOK = false;
|
|
}
|
|
else if (!fs.existsSync("/usr/share/doc/python-rpi.gpio")) {
|
|
RED.log.warn("rpi-unicorn : "+RED._("node-red:rpi-gpio.errors.libnotfound"));
|
|
allOK = false;
|
|
}
|
|
else if (!(1 & parseInt ((fs.statSync(hatCommand).mode & parseInt ("777", 8)).toString (8)[0]))) {
|
|
RED.log.warn("rpi-unicorn : "+RED._("node-red:rpi-gpio.errors.needtobeexecutable",{command:hatCommand}));
|
|
allOK = false;
|
|
}
|
|
}
|
|
catch(err) {
|
|
RED.log.warn("rpi-unicorn : "+RED._("node-red:rpi-gpio.errors.ignorenode"));
|
|
allOK = false;
|
|
}
|
|
|
|
// the magic to make python print stuff immediately
|
|
process.env.PYTHONUNBUFFERED = 1;
|
|
|
|
function UnicornHatNode(n) {
|
|
RED.nodes.createNode(this,n);
|
|
this.png = n.png;
|
|
this.bright = n.bright || 20;
|
|
this.items = {};
|
|
var node = this;
|
|
|
|
var pic = new Buffer(192);
|
|
pic.fill(0);
|
|
var ready = false;
|
|
if (node.png) {
|
|
if (node.png.split(",").length === 3) {
|
|
var c = node.png.split(",");
|
|
for (var i=0; i<192; i++) {
|
|
pic[i] = c[0];
|
|
pic[i+1] = c[1];
|
|
pic[i+2] = c[2];
|
|
i += 2;
|
|
}
|
|
ready = true;
|
|
}
|
|
else {
|
|
try {
|
|
var file = fs.readFileSync(node.png);
|
|
new PNG().parse( file, function(error, data) {
|
|
if (error) {
|
|
node.warn("error reading : "+node.png);
|
|
ready = true;
|
|
}
|
|
else {
|
|
var pix = data.data;
|
|
var j=0;
|
|
for (var i=0; i<192; i++) {
|
|
pic[i] = pix[j];
|
|
pic[i+1] = pix[j+1];
|
|
pic[i+2] = pix[j+2];
|
|
i += 2;
|
|
j += 4;
|
|
}
|
|
ready = true;
|
|
}
|
|
});
|
|
}
|
|
catch(e) {
|
|
node.warn("error loading : "+node.png);
|
|
ready = true;
|
|
}
|
|
}
|
|
}
|
|
else { ready = true; }
|
|
|
|
function inputlistener(msg) {
|
|
if (typeof msg.payload === "string") {
|
|
var a,b,i,j,x,y;
|
|
msg.payload = msg.payload.replace('"','');
|
|
var s = msg.payload.toUpperCase().split(",");
|
|
var doDraw = true;
|
|
if (s.length === 1) {
|
|
if ((s[0] == "CLS") || (s[0] == "CLR") || (s[0] == "CLEAR")) {
|
|
//console.log("CLEAR")
|
|
pic.fill(0);
|
|
node.items = {};
|
|
}
|
|
if ((s[0] == "DEL") || (s[0] == "DELETE")) {
|
|
//console.log("DELETE")
|
|
node.items = {};
|
|
}
|
|
}
|
|
else if (s.length === 2) {
|
|
if (s[0] === "BRIGHTNESS") {
|
|
//console.log("BRIGHTNESS",s[1])
|
|
node.child.stdin.write("B"+s[1]+"\n");
|
|
}
|
|
if (s[0] === "ROTATE") {
|
|
//console.log("ROTATE",s[1])
|
|
node.child.stdin.write("R"+s[1]+"\n");
|
|
}
|
|
}
|
|
else if (s.length === 3) {
|
|
//console.log("BACKGROUND",s)
|
|
for (i=0; i<192; i++) {
|
|
pic[i] = s[0];
|
|
pic[i+1] = s[1];
|
|
pic[i+2] = s[2];
|
|
i += 2;
|
|
}
|
|
}
|
|
else if (s.length % 5 === 0) { // handles pixel updates
|
|
if (msg.topic) {
|
|
node.items[msg.topic] = msg.payload;
|
|
}
|
|
else {
|
|
node.child.stdin.write('P'+msg.payload+'\n');
|
|
doDraw = false;
|
|
x = [];
|
|
y = [];
|
|
for (a = 0; a < s.length; a++) {
|
|
//console.log("PIXELS",a);
|
|
if (s[a] === "*") {
|
|
x[0] = 0;
|
|
x[1] = 7;
|
|
}
|
|
else if (s[a].indexOf("-") !== -1) {
|
|
x = s[a].split("-").sort();
|
|
}
|
|
else { x[0] = x[1] = s[a]; }
|
|
if (s[a+1] === "*") {
|
|
y[0] = 0;
|
|
y[1] = 7;
|
|
}
|
|
else if (s[a+1].indexOf("-") !== -1) {
|
|
y = s[a+1].split("-").sort();
|
|
}
|
|
else { y[0] = y[1] = s[a+1]; }
|
|
|
|
for (j = y[0]; j <= y[1]; j++) {
|
|
for (i = x[0]; i <= x[1]; i++) {
|
|
pic[i*3+j*24] = s[a+2];
|
|
pic[i*3+j*24+1] = s[a+3];
|
|
pic[i*3+j*24+2] = s[a+4];
|
|
}
|
|
}
|
|
|
|
a += 4;
|
|
}
|
|
}
|
|
}
|
|
else if (s.length === 192) { // handle complete buffer refresh.
|
|
for (var h=0; h<192; h++) {
|
|
pic[h] = s[h];
|
|
pic[h+1] = s[h+1];
|
|
pic[h+2] = s[h+2];
|
|
h += 2;
|
|
}
|
|
}
|
|
else {
|
|
if (node.items.hasOwnProperty(msg.topic)) { delete node.items[msg.topic]; }
|
|
}
|
|
|
|
if (doDraw) {
|
|
var pixels = new Buffer(192);
|
|
pic.copy(pixels);
|
|
for (var p in node.items) {
|
|
if (node.items.hasOwnProperty(p)) {
|
|
b = node.items[p].split(",");
|
|
x = [];
|
|
y = [];
|
|
for (a = 0; a < b.length; a++) {
|
|
if (b[a] === "*") {
|
|
x[0] = 0;
|
|
x[1] = 7;
|
|
}
|
|
else if (b[a].indexOf("-") !== -1) {
|
|
x = b[a].split("-").sort();
|
|
}
|
|
else { x[0] = x[1] = b[a]; }
|
|
if (b[a+1] === "*") {
|
|
y[0] = 0;
|
|
y[1] = 7;
|
|
}
|
|
else if (b[a+1].indexOf("-") !== -1) {
|
|
y = b[a+1].split("-").sort();
|
|
}
|
|
else { y[0] = y[1] = b[a+1]; }
|
|
for (j = y[0]; j <= y[1]; j++) {
|
|
for (i = x[0]; i <= x[1]; i++) {
|
|
pixels[i*3+j*24] = b[a+2];
|
|
pixels[i*3+j*24+1] = b[a+3];
|
|
pixels[i*3+j*24+2] = b[a+4];
|
|
}
|
|
}
|
|
a += 4;
|
|
}
|
|
}
|
|
}
|
|
node.child.stdin.write(pixels);
|
|
node.child.stdin.write("\n");
|
|
}
|
|
}
|
|
else { node.warn("Input not a string"); }
|
|
}
|
|
|
|
if (allOK === true) {
|
|
node.child = spawn(hatCommand, [node.bright]);
|
|
node.status({fill:"green",shape:"dot",text:"ok"});
|
|
|
|
node.on("input", inputlistener);
|
|
|
|
node.child.stdout.on('data', function (data) {
|
|
if (RED.settings.verbose) { node.log("out: "+data+" :"); }
|
|
});
|
|
|
|
node.child.stderr.on('data', function (data) {
|
|
if (RED.settings.verbose) { node.log("err: "+data+" :"); }
|
|
});
|
|
|
|
node.child.on('close', function () {
|
|
node.child = null;
|
|
if (RED.settings.verbose) { node.log(RED._("rpi-gpio.status.closed")); }
|
|
if (node.finished) {
|
|
node.status({fill:"grey",shape:"ring",text:"closed"});
|
|
node.finished();
|
|
}
|
|
else { node.status({fill:"red",shape:"ring",text:"stopped"}); }
|
|
});
|
|
|
|
node.child.on('error', function (err) {
|
|
if (err.errno === "ENOENT") { node.error(RED._("rpi-gpio.errors.commandnotfound")); }
|
|
else if (err.errno === "EACCES") { node.error(RED._("rpi-gpio.errors.commandnotexecutable")); }
|
|
else { node.error(RED._("rpi-gpio.errors.error")+': ' + err.errno); }
|
|
});
|
|
|
|
node.on("close", function(done) {
|
|
node.status({fill:"grey",shape:"ring",text:"closed"});
|
|
if (node.tout) { clearTimeout(node.tout); }
|
|
if (node.child != null) {
|
|
node.finished = done;
|
|
node.child.kill('SIGKILL');
|
|
}
|
|
else { done(); }
|
|
});
|
|
|
|
var nowready = function() {
|
|
node.tout = setTimeout( function() {
|
|
if (ready) { inputlistener({payload:"0"}); }
|
|
else { nowready(); }
|
|
}, 100);
|
|
}
|
|
nowready();
|
|
}
|
|
else {
|
|
node.status({fill:"grey",shape:"dot",text:"node-red:rpi-gpio.status.not-available"});
|
|
}
|
|
}
|
|
RED.nodes.registerType("rpi-unicorn",UnicornHatNode);
|
|
}
|