pixdisp/drivers/driver.js

'use strict';

/**
 * Basic matrix driver which every implementation can derive from
 */
class Driver {
	constructor() {

		// Some sane-ish defaults
		this.setSize( 16, 16 );
		this.setBrightness( 1 );

		this.flipHorizontal = false;
		this.flipVertical = false;

	}

	/**
	 * Sets the display size
	 */
	setSize( w, h ) {
		this.width = w;
		this.height = h;

		this.clearMatrix();
	}

	/**
	 * Return an object containing the width and height
	 */
	getSize() {
		return {
			width: this.width,
			height: this.height
		};
	}

	/**
	 * Sets brightness between 0.0f and 1.0f
	 */
	setBrightness( v ) {
		this.brightness = v;
	}

	/**
	 * Gets the current brightness
	 */
	getBrightness() {
		return this.brightness;
	}

	/**
	 * CLear the matrix with the following color, from 0..255
	 */
	clear( r = 0, g = 0, b = 0, a = 1 ) {
		for ( let x = 0; x < this.width; x++ ) {
			for ( let y = 0; y < this.height; y++) {
				this.setPixel( x, y, r, g, b, a );
			}
		}
	}

	/**
	 * Set pixel color
	 */
	setPixel( x, y, r = 0, g = 0, b = 0, a = 1 ) {
		if ( x >= 0 && x < this.width &&
			y >= 0 && y < this.height ) {
			this.matrix[x][y].r = r;
			this.matrix[x][y].g = g;
			this.matrix[x][y].b = b;
			this.matrix[x][y].a = a;
		}
	}

	/**
	 * Return the pixel color
	 */
	getPixel( x, y ) {
		if ( x >= 0 && x <= this.width &&
			y >= 0 && y <= this.height ) {
			return {
				r: this.matrix[x][y].r,
				g: this.matrix[x][y].g,
				b: this.matrix[x][y].b,
				a: this.matrix[x][y].a
			};
		}

		return {
			r: -1,
			g: -1,
			b: -1,
			a: -1
		};
	}

	/**
	 * Draw a line from x1,y1 to x2,y2
	 */
	drawLine( x1, y1, x2, y2, r = 0, g = 0, b = 0, a = 1 ) {
		let dx = x2 - x1;
		let dy = y2 - y1;

		for ( let x = x1; x <= x2; x++ ) {
			let y = Math.floor( y1 + dy * ( x - x1 ) / dx );

			this.setPixel( x, y, r, g, b, a );
		}
	}

	/**
	 * Draw a rectangle from x,y with dimensions w,h
	 */
	drawRect( x1, y1, w, h, r = 0, g = 0, b = 0, a = 1 ) {
		for ( let x = x1; x < x1 + w; x++ ) {
			this.setPixel( x, y1, r, g, b, a );
			this.setPixel( x, y1 + h - 1, r, g, b, a );
		}

		for ( let y = y1; y < y1 + h - 1; y++ ) {
			this.setPixel( x1, y, r, g, b, a );
			this.setPixel( x1 + w - 1, y, r, g, b, a );
		}
	}

	/**
	 * Draw a rectangle from x,y with dimensions w,h
	 */
	drawRectFilled( x1, y1, w, h, r = 0, g = 0, b = 0, a = 1 ) {
		for ( let x = x1; x < x1 + w; x++ ) {
			for ( let y = y1; y < y1 + h; y++ ) {
				this.setPixel( x, y, r, g, b, a );
			}
		}
	}

	/**
	 * Draws a circle at x,y with radius r
	 */
	drawCircle( x1, y1, radius, r = 0, g = 0, b = 0, a = 1 ) {
		let x = radius - 1;
		let y = 0;
		let dx = 1;
		let dy = 1;
		let err = dx - ( radius << 1 );

		while ( x >= y ) {
			this.setPixel( x1 + x, y1 + y, r, g, b, a );
			this.setPixel( x1 + y, y1 + x, r, g, b, a );
			this.setPixel( x1 - y, y1 + x, r, g, b, a );
			this.setPixel( x1 - x, y1 + y, r, g, b, a );
			this.setPixel( x1 - x, y1 - y, r, g, b, a );
			this.setPixel( x1 - y, y1 - x, r, g, b, a );
			this.setPixel( x1 + y, y1 - x, r, g, b, a );
			this.setPixel( x1 + x, y1 - y, r, g, b, a );

			if ( err <= 0 ) {
				y++;
				err += dy;
				dy += 2;
			}

			if ( err > 0 ) {
				x--;
				dx += 2;
				err += dx - ( radius << 1 );
			}
		}
	}

	/**
	 * Returns the current matrix at any given state
	 */
	getMatrix() {
		return this.matrix;
	}

	/**
	 * Get the buffer for writing
	 * @return {Buffer}
	 */
	getBuffer() {
		this.flipMatrix( this.flipHorizontal, this.flipVertical );

		let buffer = new Buffer( this.width * this.height * 3 );

		let size = this.getSize();

		// @TODO: for now this only supports square displays!
		for ( let y = 0; y < size.height; y++ ) {
			for ( let x = 0; x < size.width; x++  ) {
				buffer[ y * size.height * 3 + x * 3 + 0 ] = this.matrix[ x ][ y ].r * this.matrix[ x ][ y ].a * this.brightness;
				buffer[ y * size.height * 3 + x * 3 + 1 ] = this.matrix[ x ][ y ].g * this.matrix[ x ][ y ].a * this.brightness;
				buffer[ y * size.height * 3 + x * 3 + 2 ] = this.matrix[ x ][ y ].b * this.matrix[ x ][ y ].a * this.brightness;
			}
		}

		return buffer;
	}

	/**
	 * Write output to the device. Implement at driver level.
	 */
	write( buffer ) { //eslint-disable-line no-unused-vars
		console.log( 'Driver should implement this' );
	}

	/**
	 * Flip the matrix along it's axis.
	 */
	flipMatrix( horizontal, vertical ) {
		if ( horizontal === true ) {
			for ( let i = 0; i < this.matrix.length; i++ ) {
				this.matrix[ i ].reverse();
			}
		}

		if ( vertical === true ) {
			this.matrix.reverse();
		}
	}

	/**
	 * Create the internal frame
	 */
	clearMatrix() {
		this.matrix = [];

		for ( let y = 0; y < this.height; y++ ) {
			this.matrix.push( [] );

			for ( let x = 0; x < this.width; x++  ) {
				this.matrix[ y ].push( {
					r: 0,
					g: 0,
					b: 0,
					a: 1
				} );
			}
		}
	}
}

exports.Driver = Driver;