paper.js/src/item/Raster.js

Raster = Item.extend({
	beans: true,

	// TODO: implement url / type, width, height
	// TODO: have PlacedSymbol & Raster inherit from a shared class?
	initialize: function(image) {
		this.base();
		if (image) {
			this.image = image;
			// TODO: cross browser compatible?
			var width = image.naturalWidth;
			var height = image.naturalHeight;
			this._size = new Size(width, height);
			this._bounds = new Rectangle(-width / 2, -height / 2, width, height);
			this.matrix = new Matrix();
		}
	},

	/**
	* The size of the raster in pixels.
	*/
	getSize: function() {
		return this._size;
	},
	
	setSize: function() {
		var size = Size.read(arguments);
		var canvas = CanvasProvider.getCanvas(size.width, size.height);
		var context = canvas.getContext('2d');
		context.drawImage(this._canvas ? this._canvas : this.image,
			0, 0, size.width, size.height);
		// If we already had a canvas, return it to be reused.
		if (this._canvas)
			CanvasProvider.returnCanvas(this._canvas);
		this._size = size;
		this._context = null;
		this._canvas = canvas;
	},
	
	/**
	 * The width of the raster in pixels.
	 */
	getWidth: function() {
		return this._size.width;
	},
	
	/**
	 * The height of the raster in pixels.
	 */
	getHeight: function() {
		return this._size.height;
	},
	
	// TODO: getPpi
	// TODO: getSubImage
	// TODO: getImage
	// TODO: drawImage
	
	/**
	 * {@grouptitle Pixels}
	 * 
	 * Gets the color of a pixel in the raster.
	 * @param x
	 * @param y
	 */
	getPixel: function() {
		var point = Point.read(arguments);
		var ctx = this.context;
		var pixels = ctx.getImageData(point.x + 0.5, point.y + 0.5, 1, 1).data;
		var channels = [];
		for (var i = 0; i < 4; i++)
			channels.push(pixels[i] / 255);
		return Color.read(channels);
	},
	
	// TODO: setPixel(point, color)
	setPixel: function(x, y, color) {
		color = Color.read(arguments, 2);
		var ctx = this.context;
		var imageData = ctx.getImageData(x, y, 1, 1);
		imageData.data[0] = color.red * 255;
		imageData.data[1] = color.green * 255;
		imageData.data[2] = color.blue * 255;
		imageData.data[3] = color.alpha != -1 ? color.alpha * 255 : 255;
		ctx.putImageData(imageData, x, y);
	},
	
	getContext: function() {
		if (!this._context)
			this._context = this.canvas.getContext('2d');
		return this._context;
	},
	
	setContext: function(context) {
		this._context = context;
	},
	
	getCanvas: function() {
		if (!this._canvas) {
			this._canvas = CanvasProvider.getCanvas(this.size.width, this.size.height);
			this.ctx = this._canvas.getContext('2d');
			this.ctx.drawImage(this.image, 0, 0);
		}
		return this._canvas;
	},
	
	setCanvas: function(canvas) {
		CanvasProvider.returnCanvas(this._canvas);
		this._ctx = null;
		this._canvas = canvas;
	},
	
	transformContent: function(matrix, flags) {
		var bounds = this.bounds;
		var coords = [bounds.x, bounds.y,
			bounds.x + bounds.width, bounds.y + bounds.height];
		matrix.transform(coords, 0, coords, 0, 2);
		this.matrix.preConcatenate(matrix);
		bounds.x = coords[0];
		bounds.y = coords[1];
		bounds.width = coords[2] - coords[0];
		bounds.height = coords[3] - coords[1];
	},
	
	getBounds: function() {
		return this._bounds;
	},
	
	draw: function(ctx) {
		ctx.save();
		this.matrix.applyToContext(ctx);
		ctx.drawImage(this._canvas || this.image,
				-this.size.width / 2, -this.size.height / 2);
		ctx.restore();
	}
}, new function() {
	function getAverageColor(pixels) {
		var channels = [0, 0, 0];
		var total = 0;
		for (var i = 0, l = pixels.length / 4; i < l; i++) {
			var offset = i * 4;
			var alpha = pixels[offset + 3] / 255;
			total += alpha;
			channels[0] += pixels[offset] * alpha;
			channels[1] += pixels[offset + 1] * alpha;
			channels[2] += pixels[offset + 2] * alpha;
		}
		for (var i = 0; i < 3; i++)
			channels[i] /= total * 255;
		return total ? Color.read(channels) : null;
	}
	
	return {
		/**
		 * {@grouptitle Average Color}
		 * Calculates the average color of the image within the given path,
		 * rectangle or point. This can be used for creating raster image
		 * effects.
		 * 
		 * @param object
		 * @return the average color contained in the area covered by the
		 * specified path, rectangle or point.
		 */
		getAverageColor: function(object) {
			var image;
			if (object) {
				var bounds, path;
				if (object instanceof Path) {
					// TODO: what if the path is smaller than 1 px?
					// TODO: how about rounding of bounds.size?
					// TODO: test with compound paths.
					path = object;
					bounds = object.bounds;
				} else if (object.width) {
					bounds = new Rectangle(object);
				} else if (object.x) {
					bounds = new Rectangle(object.x - 0.5, object.y - 0.5, 1, 1);
				}
				
				var canvas = CanvasProvider.getCanvas(bounds.size);
				var ctx = canvas.getContext('2d');
				var delta = bounds.topLeft.multiply(-1);
				ctx.translate(delta.x, delta.y);
				if (path) {
					var style = object.style;
					path.draw(ctx);
					ctx.clip();
					path.style = style;
				}
				var matrix = this.matrix.clone();
				var transMatrix = Matrix.getTranslateInstance(delta);
				matrix.preConcatenate(transMatrix);
				matrix.applyToContext(ctx);
				ctx.drawImage(this._canvas || this.image,
						-this.size.width / 2, -this.size.height / 2);
				image = canvas;
			} else {
				image = this.image;
			}
			var size = 32;
			var sampleCanvas = CanvasProvider.getCanvas(size);
			var ctx = sampleCanvas.getContext('2d');
			ctx.drawImage(image, 0, 0, size, size);
			var pixels = context.getImageData(0.5, 0.5, size, size).data;
			var color = getAverageColor(pixels);
			CanvasProvider.returnCanvas(tempCanvas);
			if(image instanceof HTMLCanvasElement)
				CanvasProvider.returnCanvas(image);
			return color;
		}
	}
});