heatmap.js 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731
  1. /*
  2. * heatmap.js v2.0.5 | JavaScript Heatmap Library
  3. *
  4. * Copyright 2008-2016 Patrick Wied <heatmapjs@patrick-wied.at> - All rights reserved.
  5. * Dual licensed under MIT and Beerware license
  6. *
  7. * :: 2016-09-05 01:16
  8. */
  9. ;
  10. (function (name, context, factory) {
  11. // Supports UMD. AMD, CommonJS/Node.js and browser context
  12. if (typeof module !== "undefined" && module.exports) {
  13. module.exports = factory();
  14. } else if (typeof define === "function" && define.amd) {
  15. define(factory);
  16. } else {
  17. context[name] = factory();
  18. }
  19. })("h337", this, function () {
  20. // Heatmap Config stores default values and will be merged with instance config
  21. var HeatmapConfig = {
  22. defaultRadius: 40,
  23. defaultRenderer: 'canvas2d',
  24. defaultGradient: {
  25. "1": "rgba(255,0,0,1)",
  26. "0.85": "rgba(255,255,0,1)",
  27. "0.55": "rgba(0,255,0,1)",
  28. "0.25": "rgba(0,255,255,1)",
  29. },
  30. defaultMaxOpacity: 1,
  31. defaultMinOpacity: 0,
  32. defaultBlur: .85,
  33. defaultXField: 'x',
  34. defaultYField: 'y',
  35. defaultValueField: 'value',
  36. plugins: {}
  37. };
  38. var Store = (function StoreClosure() {
  39. var Store = function Store(config) {
  40. this._coordinator = {};
  41. this._data = [];
  42. this._radi = [];
  43. this._min = 10;
  44. this._max = 1;
  45. this._xField = config['xField'] || config.defaultXField;
  46. this._yField = config['yField'] || config.defaultYField;
  47. this._valueField = config['valueField'] || config.defaultValueField;
  48. if (config["radius"]) {
  49. this._cfgRadius = config["radius"];
  50. }
  51. };
  52. var defaultRadius = HeatmapConfig.defaultRadius;
  53. Store.prototype = {
  54. // when forceRender = false -> called from setData, omits renderall event
  55. _organiseData: function (dataPoint, forceRender) {
  56. var x = dataPoint[this._xField];
  57. var y = dataPoint[this._yField];
  58. var radi = this._radi;
  59. var store = this._data;
  60. var max = this._max;
  61. var min = this._min;
  62. var value = dataPoint[this._valueField] || 1;
  63. var radius = dataPoint.radius || this._cfgRadius || defaultRadius;
  64. if (!store[x]) {
  65. store[x] = [];
  66. radi[x] = [];
  67. }
  68. if (!store[x][y]) {
  69. store[x][y] = value;
  70. radi[x][y] = radius;
  71. } else {
  72. store[x][y] += value;
  73. }
  74. var storedVal = store[x][y];
  75. if (storedVal > max) {
  76. if (!forceRender) {
  77. this._max = storedVal;
  78. } else {
  79. this.setDataMax(storedVal);
  80. }
  81. return false;
  82. } else if (storedVal < min) {
  83. if (!forceRender) {
  84. this._min = storedVal;
  85. } else {
  86. this.setDataMin(storedVal);
  87. }
  88. return false;
  89. } else {
  90. return {
  91. x: x,
  92. y: y,
  93. value: value,
  94. radius: radius,
  95. min: min,
  96. max: max
  97. };
  98. }
  99. },
  100. _unOrganizeData: function () {
  101. var unorganizedData = [];
  102. var data = this._data;
  103. var radi = this._radi;
  104. for (var x in data) {
  105. for (var y in data[x]) {
  106. unorganizedData.push({
  107. x: x,
  108. y: y,
  109. radius: radi[x][y],
  110. value: data[x][y]
  111. });
  112. }
  113. }
  114. return {
  115. min: this._min,
  116. max: this._max,
  117. data: unorganizedData
  118. };
  119. },
  120. _onExtremaChange: function () {
  121. this._coordinator.emit('extremachange', {
  122. min: this._min,
  123. max: this._max
  124. });
  125. },
  126. addData: function () {
  127. if (arguments[0].length > 0) {
  128. var dataArr = arguments[0];
  129. var dataLen = dataArr.length;
  130. while (dataLen--) {
  131. this.addData.call(this, dataArr[dataLen]);
  132. }
  133. } else {
  134. // add to store
  135. var organisedEntry = this._organiseData(arguments[0], true);
  136. if (organisedEntry) {
  137. // if it's the first datapoint initialize the extremas with it
  138. if (this._data.length === 0) {
  139. this._min = this._max = organisedEntry.value;
  140. }
  141. this._coordinator.emit('renderpartial', {
  142. min: this._min,
  143. max: this._max,
  144. data: [organisedEntry]
  145. });
  146. }
  147. }
  148. return this;
  149. },
  150. setData: function (data) {
  151. var dataPoints = data.data;
  152. var pointsLen = dataPoints.length;
  153. // reset data arrays
  154. this._data = [];
  155. this._radi = [];
  156. for (var i = 0; i < pointsLen; i++) {
  157. this._organiseData(dataPoints[i], false);
  158. }
  159. this._max = data.max;
  160. this._min = data.min || 0;
  161. this._onExtremaChange();
  162. this._coordinator.emit('renderall', this._getInternalData());
  163. return this;
  164. },
  165. removeData: function () {
  166. // TODO: implement
  167. },
  168. setDataMax: function (max) {
  169. this._max = max;
  170. this._onExtremaChange();
  171. this._coordinator.emit('renderall', this._getInternalData());
  172. return this;
  173. },
  174. setDataMin: function (min) {
  175. this._min = min;
  176. this._onExtremaChange();
  177. this._coordinator.emit('renderall', this._getInternalData());
  178. return this;
  179. },
  180. setCoordinator: function (coordinator) {
  181. this._coordinator = coordinator;
  182. },
  183. _getInternalData: function () {
  184. return {
  185. max: this._max,
  186. min: this._min,
  187. data: this._data,
  188. radi: this._radi
  189. };
  190. },
  191. getData: function () {
  192. return this._unOrganizeData();
  193. }
  194. /*,
  195. TODO: rethink.
  196. getValueAt: function(point) {
  197. var value;
  198. var radius = 100;
  199. var x = point.x;
  200. var y = point.y;
  201. var data = this._data;
  202. if (data[x] && data[x][y]) {
  203. return data[x][y];
  204. } else {
  205. var values = [];
  206. // radial search for datapoints based on default radius
  207. for(var distance = 1; distance < radius; distance++) {
  208. var neighbors = distance * 2 +1;
  209. var startX = x - distance;
  210. var startY = y - distance;
  211. for(var i = 0; i < neighbors; i++) {
  212. for (var o = 0; o < neighbors; o++) {
  213. if ((i == 0 || i == neighbors-1) || (o == 0 || o == neighbors-1)) {
  214. if (data[startY+i] && data[startY+i][startX+o]) {
  215. values.push(data[startY+i][startX+o]);
  216. }
  217. } else {
  218. continue;
  219. }
  220. }
  221. }
  222. }
  223. if (values.length > 0) {
  224. return Math.max.apply(Math, values);
  225. }
  226. }
  227. return false;
  228. }*/
  229. };
  230. return Store;
  231. })();
  232. var Canvas2dRenderer = (function Canvas2dRendererClosure() {
  233. var _getColorPalette = function (config) {
  234. var gradientConfig = config.gradient || config.defaultGradient;
  235. var paletteCanvas = document.createElement('canvas');
  236. var paletteCtx = paletteCanvas.getContext('2d');
  237. paletteCanvas.width = 256;
  238. paletteCanvas.height = 1;
  239. var gradient = paletteCtx.createLinearGradient(0, 0, 256, 1);
  240. for (var key in gradientConfig) {
  241. gradient.addColorStop(key, gradientConfig[key]);
  242. }
  243. paletteCtx.fillStyle = gradient;
  244. paletteCtx.fillRect(0, 0, 256, 1);
  245. return paletteCtx.getImageData(0, 0, 256, 1).data;
  246. };
  247. var _getPointTemplate = function (radius, blurFactor) {
  248. var tplCanvas = document.createElement('canvas');
  249. var tplCtx = tplCanvas.getContext('2d');
  250. var x = radius;
  251. var y = radius;
  252. tplCanvas.width = tplCanvas.height = radius * 2;
  253. if (blurFactor == 1) {
  254. tplCtx.beginPath();
  255. tplCtx.arc(x, y, radius, 0, 2 * Math.PI, false);
  256. tplCtx.fillStyle = 'rgba(0,0,0,1)';
  257. tplCtx.fill();
  258. } else {
  259. var gradient = tplCtx.createRadialGradient(x, y, radius * blurFactor, x, y, radius);
  260. gradient.addColorStop(0, 'rgba(0,0,0,1)');
  261. gradient.addColorStop(1, 'rgba(0,0,0,0)');
  262. tplCtx.fillStyle = gradient;
  263. tplCtx.fillRect(0, 0, 2 * radius, 2 * radius);
  264. }
  265. return tplCanvas;
  266. };
  267. var _prepareData = function (data) {
  268. var renderData = [];
  269. var min = data.min;
  270. var max = data.max;
  271. var radi = data.radi;
  272. var data = data.data;
  273. var xValues = Object.keys(data);
  274. var xValuesLen = xValues.length;
  275. while (xValuesLen--) {
  276. var xValue = xValues[xValuesLen];
  277. var yValues = Object.keys(data[xValue]);
  278. var yValuesLen = yValues.length;
  279. while (yValuesLen--) {
  280. var yValue = yValues[yValuesLen];
  281. var value = data[xValue][yValue];
  282. var radius = radi[xValue][yValue];
  283. renderData.push({
  284. x: xValue,
  285. y: yValue,
  286. value: value,
  287. radius: radius
  288. });
  289. }
  290. }
  291. return {
  292. min: min,
  293. max: max,
  294. data: renderData
  295. };
  296. };
  297. function Canvas2dRenderer(config) {
  298. var container = config.container;
  299. var shadowCanvas = this.shadowCanvas = document.createElement('canvas');
  300. var canvas = this.canvas = config.canvas || document.createElement('canvas');
  301. var renderBoundaries = this._renderBoundaries = [10000, 10000, 0, 0];
  302. var computed = getComputedStyle(config.container) || {};
  303. canvas.className = 'heatmap-canvas';
  304. this._width = canvas.width = shadowCanvas.width = config.width || +(computed.width.replace(/px/, ''));
  305. this._height = canvas.height = shadowCanvas.height = config.height || +(computed.height.replace(/px/, ''));
  306. this.shadowCtx = shadowCanvas.getContext('2d');
  307. this.ctx = canvas.getContext('2d');
  308. // @TODO:
  309. // conditional wrapper
  310. canvas.style.cssText = shadowCanvas.style.cssText = 'position:absolute;left:0;top:0;';
  311. container.style.position = 'relative';
  312. container.appendChild(canvas);
  313. this._palette = _getColorPalette(config);
  314. this._templates = {};
  315. this._setStyles(config);
  316. };
  317. Canvas2dRenderer.prototype = {
  318. renderPartial: function (data) {
  319. if (data.data.length > 0) {
  320. this._drawAlpha(data);
  321. this._colorize();
  322. }
  323. },
  324. renderAll: function (data) {
  325. // reset render boundaries
  326. this._clear();
  327. if (data.data.length > 0) {
  328. this._drawAlpha(_prepareData(data));
  329. this._colorize();
  330. }
  331. },
  332. _updateGradient: function (config) {
  333. this._palette = _getColorPalette(config);
  334. },
  335. updateConfig: function (config) {
  336. if (config['gradient']) {
  337. this._updateGradient(config);
  338. }
  339. this._setStyles(config);
  340. },
  341. setDimensions: function (width, height) {
  342. this._width = width;
  343. this._height = height;
  344. this.canvas.width = this.shadowCanvas.width = width;
  345. this.canvas.height = this.shadowCanvas.height = height;
  346. },
  347. _clear: function () {
  348. this.shadowCtx.clearRect(0, 0, this._width, this._height);
  349. this.ctx.clearRect(0, 0, this._width, this._height);
  350. },
  351. _setStyles: function (config) {
  352. this._blur = (config.blur == 0) ? 0 : (config.blur || config.defaultBlur);
  353. if (config.backgroundColor) {
  354. this.canvas.style.backgroundColor = config.backgroundColor;
  355. }
  356. this._width = this.canvas.width = this.shadowCanvas.width = config.width || this._width;
  357. this._height = this.canvas.height = this.shadowCanvas.height = config.height || this._height;
  358. this._opacity = (config.opacity || 0) * 255;
  359. this._maxOpacity = (config.maxOpacity || config.defaultMaxOpacity) * 255;
  360. this._minOpacity = (config.minOpacity || config.defaultMinOpacity) * 255;
  361. this._useGradientOpacity = !!config.useGradientOpacity;
  362. },
  363. _drawAlpha: function (data) {
  364. var min = this._min = data.min;
  365. var max = this._max = data.max;
  366. var data = data.data || [];
  367. var dataLen = data.length;
  368. // on a point basis?
  369. var blur = 1 - this._blur;
  370. while (dataLen--) {
  371. var point = data[dataLen];
  372. var x = point.x;
  373. var y = point.y;
  374. var radius = point.radius;
  375. // if value is bigger than max
  376. // use max as value
  377. var value = Math.min(point.value, max);
  378. var rectX = x - radius;
  379. var rectY = y - radius;
  380. var shadowCtx = this.shadowCtx;
  381. var tpl;
  382. if (!this._templates[radius]) {
  383. this._templates[radius] = tpl = _getPointTemplate(radius, blur);
  384. } else {
  385. tpl = this._templates[radius];
  386. }
  387. // value from minimum / value range
  388. // => [0, 1]
  389. var templateAlpha = (value - min) / (max - min);
  390. // this fixes #176: small values are not visible because globalAlpha < .01 cannot be read from imageData
  391. shadowCtx.globalAlpha = templateAlpha < .01 ? .01 : templateAlpha;
  392. shadowCtx.drawImage(tpl, rectX, rectY);
  393. // update renderBoundaries
  394. if (rectX < this._renderBoundaries[0]) {
  395. this._renderBoundaries[0] = rectX;
  396. }
  397. if (rectY < this._renderBoundaries[1]) {
  398. this._renderBoundaries[1] = rectY;
  399. }
  400. if (rectX + 2 * radius > this._renderBoundaries[2]) {
  401. this._renderBoundaries[2] = rectX + 2 * radius;
  402. }
  403. if (rectY + 2 * radius > this._renderBoundaries[3]) {
  404. this._renderBoundaries[3] = rectY + 2 * radius;
  405. }
  406. }
  407. },
  408. _colorize: function () {
  409. var x = this._renderBoundaries[0];
  410. var y = this._renderBoundaries[1];
  411. var width = this._renderBoundaries[2] - x;
  412. var height = this._renderBoundaries[3] - y;
  413. var maxWidth = this._width;
  414. var maxHeight = this._height;
  415. var opacity = this._opacity;
  416. var maxOpacity = this._maxOpacity;
  417. var minOpacity = this._minOpacity;
  418. var useGradientOpacity = this._useGradientOpacity;
  419. if (x < 0) {
  420. x = 0;
  421. }
  422. if (y < 0) {
  423. y = 0;
  424. }
  425. if (x + width > maxWidth) {
  426. width = maxWidth - x;
  427. }
  428. if (y + height > maxHeight) {
  429. height = maxHeight - y;
  430. }
  431. var img = this.shadowCtx.getImageData(x, y, width, height);
  432. var imgData = img.data;
  433. var len = imgData.length;
  434. var palette = this._palette;
  435. for (var i = 3; i < len; i += 4) {
  436. var alpha = imgData[i];
  437. var offset = alpha * 4;
  438. if (!offset) {
  439. continue;
  440. }
  441. var finalAlpha;
  442. if (opacity > 0) {
  443. finalAlpha = opacity;
  444. } else {
  445. if (alpha < maxOpacity) {
  446. if (alpha < minOpacity) {
  447. finalAlpha = minOpacity;
  448. } else {
  449. finalAlpha = alpha;
  450. }
  451. } else {
  452. finalAlpha = maxOpacity;
  453. }
  454. }
  455. imgData[i - 3] = palette[offset];
  456. imgData[i - 2] = palette[offset + 1];
  457. imgData[i - 1] = palette[offset + 2];
  458. imgData[i] = useGradientOpacity ? palette[offset + 3] : finalAlpha;
  459. }
  460. img.data = imgData;
  461. this.ctx.putImageData(img, x, y);
  462. this._renderBoundaries = [1000, 1000, 0, 0];
  463. },
  464. getValueAt: function (point) {
  465. var value;
  466. var shadowCtx = this.shadowCtx;
  467. var img = shadowCtx.getImageData(point.x, point.y, 1, 1);
  468. var data = img.data[3];
  469. var max = this._max;
  470. var min = this._min;
  471. value = (Math.abs(max - min) * (data / 255)) >> 0;
  472. return value;
  473. },
  474. getDataURL: function () {
  475. return this.canvas.toDataURL();
  476. }
  477. };
  478. return Canvas2dRenderer;
  479. })();
  480. var Renderer = (function RendererClosure() {
  481. var rendererFn = false;
  482. if (HeatmapConfig['defaultRenderer'] === 'canvas2d') {
  483. rendererFn = Canvas2dRenderer;
  484. }
  485. return rendererFn;
  486. })();
  487. var Util = {
  488. merge: function () {
  489. var merged = {};
  490. var argsLen = arguments.length;
  491. for (var i = 0; i < argsLen; i++) {
  492. var obj = arguments[i]
  493. for (var key in obj) {
  494. merged[key] = obj[key];
  495. }
  496. }
  497. return merged;
  498. }
  499. };
  500. // Heatmap Constructor
  501. var Heatmap = (function HeatmapClosure() {
  502. var Coordinator = (function CoordinatorClosure() {
  503. function Coordinator() {
  504. this.cStore = {};
  505. };
  506. Coordinator.prototype = {
  507. on: function (evtName, callback, scope) {
  508. var cStore = this.cStore;
  509. if (!cStore[evtName]) {
  510. cStore[evtName] = [];
  511. }
  512. cStore[evtName].push((function (data) {
  513. return callback.call(scope, data);
  514. }));
  515. },
  516. emit: function (evtName, data) {
  517. var cStore = this.cStore;
  518. if (cStore[evtName]) {
  519. var len = cStore[evtName].length;
  520. for (var i = 0; i < len; i++) {
  521. var callback = cStore[evtName][i];
  522. callback(data);
  523. }
  524. }
  525. }
  526. };
  527. return Coordinator;
  528. })();
  529. var _connect = function (scope) {
  530. var renderer = scope._renderer;
  531. var coordinator = scope._coordinator;
  532. var store = scope._store;
  533. coordinator.on('renderpartial', renderer.renderPartial, renderer);
  534. coordinator.on('renderall', renderer.renderAll, renderer);
  535. coordinator.on('extremachange', function (data) {
  536. scope._config.onExtremaChange &&
  537. scope._config.onExtremaChange({
  538. min: data.min,
  539. max: data.max,
  540. gradient: scope._config['gradient'] || scope._config['defaultGradient']
  541. });
  542. });
  543. store.setCoordinator(coordinator);
  544. };
  545. function Heatmap() {
  546. var config = this._config = Util.merge(HeatmapConfig, arguments[0] || {});
  547. this._coordinator = new Coordinator();
  548. if (config['plugin']) {
  549. var pluginToLoad = config['plugin'];
  550. if (!HeatmapConfig.plugins[pluginToLoad]) {
  551. throw new Error('Plugin \'' + pluginToLoad + '\' not found. Maybe it was not registered.');
  552. } else {
  553. var plugin = HeatmapConfig.plugins[pluginToLoad];
  554. // set plugin renderer and store
  555. this._renderer = new plugin.renderer(config);
  556. this._store = new plugin.store(config);
  557. }
  558. } else {
  559. this._renderer = new Renderer(config);
  560. this._store = new Store(config);
  561. }
  562. _connect(this);
  563. };
  564. // @TODO:
  565. // add API documentation
  566. Heatmap.prototype = {
  567. addData: function () {
  568. this._store.addData.apply(this._store, arguments);
  569. return this;
  570. },
  571. removeData: function () {
  572. this._store.removeData && this._store.removeData.apply(this._store, arguments);
  573. return this;
  574. },
  575. setData: function () {
  576. this._store.setData.apply(this._store, arguments);
  577. return this;
  578. },
  579. setDataMax: function () {
  580. this._store.setDataMax.apply(this._store, arguments);
  581. return this;
  582. },
  583. setDataMin: function () {
  584. this._store.setDataMin.apply(this._store, arguments);
  585. return this;
  586. },
  587. configure: function (config) {
  588. this._config = Util.merge(this._config, config);
  589. this._renderer.updateConfig(this._config);
  590. this._coordinator.emit('renderall', this._store._getInternalData());
  591. return this;
  592. },
  593. repaint: function () {
  594. this._coordinator.emit('renderall', this._store._getInternalData());
  595. return this;
  596. },
  597. getData: function () {
  598. return this._store.getData();
  599. },
  600. getDataURL: function () {
  601. return this._renderer.getDataURL();
  602. },
  603. getValueAt: function (point) {
  604. if (this._store.getValueAt) {
  605. return this._store.getValueAt(point);
  606. } else if (this._renderer.getValueAt) {
  607. return this._renderer.getValueAt(point);
  608. } else {
  609. return null;
  610. }
  611. }
  612. };
  613. return Heatmap;
  614. })();
  615. // core
  616. var heatmapFactory = {
  617. create: function (config) {
  618. return new Heatmap(config);
  619. },
  620. register: function (pluginKey, plugin) {
  621. HeatmapConfig.plugins[pluginKey] = plugin;
  622. }
  623. };
  624. return heatmapFactory;
  625. });