Create getElement function

Author: sarcadass

lib/Granim.js

@@ -1,15 +1,7 @@
 'use strict';
 
 function Granim(options) {
-	if(options.element instanceof HTMLCanvasElement)
-		this.canvas = options.element
-	else if (typeof options.element === "string")
-		this.canvas = document.querySelector(options.element)
-	else
-		throw new Error('The element you used is neither a String, nor a HTMLCanvasElement');
-	if(!this.canvas){
-		throw new Error('`' + options.element + '` could not be found in the DOM');
-	}
+	this.getElement(options.element);
 	this.x1 = 0;
 	this.y1 = 0;
 	this.name = options.name || false;
@@ -22,6 +14,7 @@ function Granim(options) {
 	this.previousTimeStamp = null;
 	this.progress = 0;
 	this.isPaused = false;
+	this.isCleared = false;
 	this.isPausedBecauseNotInView = false;
 	this.iscurrentColorsSet = false;
 	this.context = this.canvas.getContext('2d');
@@ -89,6 +82,8 @@ Granim.prototype.makeGradient = require('./makeGradient.js');
 
 Granim.prototype.getDimensions = require('./getDimensions.js');
 
+Granim.prototype.getElement = require('./getElement.js');
+
 Granim.prototype.animateColors = require('./animateColors.js');
 
 Granim.prototype.getLightness = require('./getLightness.js');

lib/getElement.js

@@ -0,0 +1,17 @@
+'use strict';
+
+module.exports = function(element) {
+	if (element instanceof HTMLCanvasElement) {
+		this.canvas = element;
+
+	} else if (typeof element === "string") {
+		this.canvas = document.querySelector(element);
+
+	} else {
+		throw new Error('The element you used is neither a String, nor a HTMLCanvasElement');
+	}
+
+	if (!this.canvas) {
+		throw new Error('`' + element + '` could not be found in the DOM');
+	}
+};