(function (root, factory) {
if (typeof exports === 'object') {
module.exports = factory();
} else if (typeof define === 'function' && define.amd) {
define(factory);
} else {
root.BMapLib = root.BMapLib || {};
root.BMapLib.Heatmap = root.BMapLib.Heatmap || factory();
}
})(this, function() {
function inherits (subClass, superClass, className) {
var key, proto,
selfProps = subClass.prototype,
clazz = new Function();
clazz.prototype = superClass.prototype;
proto = subClass.prototype = new clazz();
for (key in selfProps) {
proto[key] = selfProps[key];
}
subClass.prototype.constructor = subClass;
subClass.superClass = superClass.prototype;
// 类名标识,兼容Class的toString,基本没用
if ("string" == typeof className) {
proto._className = className;
}
};
var heatmapFactory = (function(){
// store object constructor
// a heatmap contains a store
// the store has to know about the heatmap in order to trigger heatmap updates when datapoints get added
var store = function store(hmap){
var _ = {
// data is a two dimensional array
// a datapoint gets saved as data[point-x-value][point-y-value]
// the value at [point-x-value][point-y-value] is the occurrence of the datapoint
data: [],
// tight coupling of the heatmap object
heatmap: hmap
};
// the max occurrence - the heatmaps radial gradient alpha transition is based on it
this.max = 1;
this.get = function(key){
return _[key];
};
this.set = function(key, value){
_[key] = value;
};
}
store.prototype = {
// function for adding datapoints to the store
// datapoints are usually defined by x and y but could also contain a third parameter which represents the occurrence
addDataPoint: function(x, y){
if(x < 0 || y < 0)
return;
var me = this,
heatmap = me.get("heatmap"),
data = me.get("data");
if(!data[x])
data[x] = [];
if(!data[x][y])
data[x][y] = 0;
// if count parameter is set increment by count otherwise by 1
data[x][y]+=(arguments.length<3)?1:arguments[2];
me.set("data", data);
// do we have a new maximum?
if(me.max < data[x][y]){
// max changed, we need to redraw all existing(lower) datapoints
heatmap.get("actx").clearRect(0,0,heatmap.get("width"),heatmap.get("height"));
me.setDataSet({ max: data[x][y], data: data }, true);
return;
}
heatmap.drawAlpha(x, y, data[x][y], true);
},
setDataSet: function(obj, internal){
var me = this,
heatmap = me.get("heatmap"),
data = [],
d = obj.data,
dlen = d.length;
// clear the heatmap before the data set gets drawn
heatmap.clear();
this.max = obj.max;
// if a legend is set, update it
heatmap.get("legend") && heatmap.get("legend").update(obj.max);
if(internal != null && internal){
for(var one in d){
// jump over undefined indexes
if(one === undefined)
continue;
for(var two in d[one]){
if(two === undefined)
continue;
// if both indexes are defined, push the values into the array
heatmap.drawAlpha(one, two, d[one][two], false);
}
}
}else{
while(dlen--){
var point = d[dlen];
heatmap.drawAlpha(point.x, point.y, point.count, false);
if(!data[point.x])
data[point.x] = [];
if(!data[point.x][point.y])
data[point.x][point.y] = 0;
data[point.x][point.y] = point.count;
}
}
heatmap.colorize();
this.set("data", d);
},
exportDataSet: function(){
var me = this,
data = me.get("data"),
exportData = [];
for(var one in data){
// jump over undefined indexes
if(one === undefined)
continue;
for(var two in data[one]){
if(two === undefined)
continue;
// if both indexes are defined, push the values into the array
exportData.push({x: parseInt(one, 10), y: parseInt(two, 10), count: data[one][two]});
}
}
return { max: me.max, data: exportData };
},
generateRandomDataSet: function(points){
var heatmap = this.get("heatmap"),
w = heatmap.get("width"),
h = heatmap.get("height");
var randomset = {},
max = Math.floor(Math.random()*1000+1);
randomset.max = max;
var data = [];
while(points--){
data.push({x: Math.floor(Math.random()*w+1), y: Math.floor(Math.random()*h+1), count: Math.floor(Math.random()*max+1)});
}
randomset.data = data;
this.setDataSet(randomset);
}
};
var legend = function legend(config){
this.config = config;
var _ = {
element: null,
labelsEl: null,
gradientCfg: null,
ctx: null
};
this.get = function(key){
return _[key];
};
this.set = function(key, value){
_[key] = value;
};
this.init();
};
legend.prototype = {
init: function(){
var me = this,
config = me.config,
title = config.title || "Legend",
position = config.position,
offset = config.offset || 10,
gconfig = config.gradient,
labelsEl = document.createElement("ul"),
labelsHtml = "",
grad, element, gradient, positionCss = "";
me.processGradientObject();
// Positioning
// top or bottom
if(position.indexOf('t') > -1){
positionCss += 'top:'+offset+'px;';
}else{
positionCss += 'bottom:'+offset+'px;';
}
// left or right
if(position.indexOf('l') > -1){
positionCss += 'left:'+offset+'px;';
}else{
positionCss += 'right:'+offset+'px;';
}
element = document.createElement("div");
element.style.cssText = "border-radius:5px;position:absolute;"+positionCss+"font-family:Helvetica; width:256px;z-index:10000000000; background:rgba(255,255,255,1);padding:10px;border:1px solid black;margin:0;";
element.innerHTML = "
"+title+"
";
// create gradient in canvas
labelsEl.style.cssText = "position:relative;font-size:12px;display:block;list-style:none;list-style-type:none;margin:0;height:15px;";
// create gradient element
gradient = document.createElement("div");
gradient.style.cssText = ["position:relative;display:block;width:256px;height:15px;border-bottom:1px solid black; background-image:url(",me.createGradientImage(),");"].join("");
element.appendChild(labelsEl);
element.appendChild(gradient);
me.set("element", element);
me.set("labelsEl", labelsEl);
me.update(1);
},
processGradientObject: function(){
// create array and sort it
var me = this,
gradientConfig = this.config.gradient,
gradientArr = [];
for(var key in gradientConfig){
if(gradientConfig.hasOwnProperty(key)){
gradientArr.push({ stop: key, value: gradientConfig[key] });
}
}
gradientArr.sort(function(a, b){
return (a.stop - b.stop);
});
gradientArr.unshift({ stop: 0, value: 'rgba(0,0,0,0)' });
me.set("gradientArr", gradientArr);
},
createGradientImage: function(){
var me = this,
gradArr = me.get("gradientArr"),
length = gradArr.length,
canvas = document.createElement("canvas"),
ctx = canvas.getContext("2d"),
grad;
// the gradient in the legend including the ticks will be 256x15px
canvas.width = "256";
canvas.height = "15";
grad = ctx.createLinearGradient(0,5,256,10);
for(var i = 0; i < length; i++){
grad.addColorStop(1/(length-1) * i, gradArr[i].value);
}
ctx.fillStyle = grad;
ctx.fillRect(0,5,256,10);
ctx.strokeStyle = "black";
ctx.beginPath();
for(var i = 0; i < length; i++){
ctx.moveTo(((1/(length-1)*i*256) >> 0)+.5, 0);
ctx.lineTo(((1/(length-1)*i*256) >> 0)+.5, (i==0)?15:5);
}
ctx.moveTo(255.5, 0);
ctx.lineTo(255.5, 15);
ctx.moveTo(255.5, 4.5);
ctx.lineTo(0, 4.5);
ctx.stroke();
// we re-use the context for measuring the legends label widths
me.set("ctx", ctx);
return canvas.toDataURL();
},
getElement: function(){
return this.get("element");
},
update: function(max){
var me = this,
gradient = me.get("gradientArr"),
ctx = me.get("ctx"),
labels = me.get("labelsEl"),
labelText, labelsHtml = "", offset;
for(var i = 0; i < gradient.length; i++){
labelText = max*gradient[i].stop >> 0;
offset = (ctx.measureText(labelText).width/2) >> 0;
if(i == 0){
offset = 0;
}
if(i == gradient.length-1){
offset *= 2;
}
labelsHtml += ''+labelText+'';
}
labels.innerHTML = labelsHtml;
}
};
// heatmap object constructor
var heatmap = function heatmap(config){
// private variables
var _ = {
radius : 40,
element : {},
canvas : {},
acanvas: {},
ctx : {},
actx : {},
legend: null,
visible : true,
width : 0,
height : 0,
max : false,
gradient : false,
opacity: 180,
premultiplyAlpha: false,
bounds: {
l: 1000,
r: 0,
t: 1000,
b: 0
},
debug: false
};
// heatmap store containing the datapoints and information about the maximum
// accessible via instance.store
this.store = new store(this);
this.get = function(key){
return _[key];
};
this.set = function(key, value){
_[key] = value;
};
// configure the heatmap when an instance gets created
this.configure(config);
// and initialize it
this.init();
};
// public functions
heatmap.prototype = {
configure: function(config){
var me = this,
rout, rin;
me.set("radius", config["radius"] || 40);
me.set("element", (config.element instanceof Object)?config.element:document.getElementById(config.element));
me.set("visible", (config.visible != null)?config.visible:true);
me.set("max", config.max || false);
me.set("gradient", config.gradient || { 0.45: "rgb(0,0,255)", 0.55: "rgb(0,255,255)", 0.65: "rgb(0,255,0)", 0.95: "yellow", 1.0: "rgb(255,0,0)"}); // default is the common blue to red gradient
me.set("opacity", parseInt(255/(100/config.opacity), 10) || 180);
me.set("width", config.width || 0);
me.set("height", config.height || 0);
me.set("debug", config.debug);
if(config.legend){
var legendCfg = config.legend;
legendCfg.gradient = me.get("gradient");
me.set("legend", new legend(legendCfg));
}
},
resize: function () {
var me = this,
element = me.get("element"),
canvas = me.get("canvas"),
acanvas = me.get("acanvas");
canvas.width = acanvas.width = me.get("width") || element.style.width.replace(/px/, "") || me.getWidth(element);
this.set("width", canvas.width);
canvas.height = acanvas.height = me.get("height") || element.style.height.replace(/px/, "") || me.getHeight(element);
this.set("height", canvas.height);
},
init: function(){
var me = this,
canvas = document.createElement("canvas"),
acanvas = document.createElement("canvas"),
ctx = canvas.getContext("2d"),
actx = acanvas.getContext("2d"),
element = me.get("element");
me.initColorPalette();
me.set("canvas", canvas);
me.set("ctx", ctx);
me.set("acanvas", acanvas);
me.set("actx", actx);
me.resize();
canvas.style.cssText = acanvas.style.cssText = "position:absolute;top:0;left:0;z-index:10000000;";
if(!me.get("visible"))
canvas.style.display = "none";
element.appendChild(canvas);
if(me.get("legend")){
element.appendChild(me.get("legend").getElement());
}
// debugging purposes only
if(me.get("debug"))
document.body.appendChild(acanvas);
actx.shadowOffsetX = 15000;
actx.shadowOffsetY = 15000;
actx.shadowBlur = 15;
},
initColorPalette: function(){
var me = this,
canvas = document.createElement("canvas"),
gradient = me.get("gradient"),
ctx, grad, testData;
canvas.width = "1";
canvas.height = "256";
ctx = canvas.getContext("2d");
grad = ctx.createLinearGradient(0,0,1,256);
// Test how the browser renders alpha by setting a partially transparent pixel
// and reading the result. A good browser will return a value reasonably close
// to what was set. Some browsers (e.g. on Android) will return a ridiculously wrong value.
testData = ctx.getImageData(0,0,1,1);
testData.data[0] = testData.data[3] = 64; // 25% red & alpha
testData.data[1] = testData.data[2] = 0; // 0% blue & green
ctx.putImageData(testData, 0, 0);
testData = ctx.getImageData(0,0,1,1);
me.set("premultiplyAlpha", (testData.data[0] < 60 || testData.data[0] > 70));
for(var x in gradient){
grad.addColorStop(x, gradient[x]);
}
ctx.fillStyle = grad;
ctx.fillRect(0,0,1,256);
me.set("gradient", ctx.getImageData(0,0,1,256).data);
},
getWidth: function(element){
var width = element.offsetWidth;
if(element.style.paddingLeft){
width+=element.style.paddingLeft;
}
if(element.style.paddingRight){
width+=element.style.paddingRight;
}
return width;
},
getHeight: function(element){
var height = element.offsetHeight;
if(element.style.paddingTop){
height+=element.style.paddingTop;
}
if(element.style.paddingBottom){
height+=element.style.paddingBottom;
}
return height;
},
colorize: function(x, y){
// get the private variables
var me = this,
width = me.get("width"),
radius = me.get("radius"),
height = me.get("height"),
actx = me.get("actx"),
ctx = me.get("ctx"),
x2 = radius * 3,
premultiplyAlpha = me.get("premultiplyAlpha"),
palette = me.get("gradient"),
opacity = me.get("opacity"),
bounds = me.get("bounds"),
left, top, bottom, right,
image, length, alpha, offset, finalAlpha;
if(x != null && y != null){
if(x+x2>width){
x=width-x2;
}
if(x<0){
x=0;
}
if(y<0){
y=0;
}
if(y+x2>height){
y=height-x2;
}
left = x;
top = y;
right = x + x2;
bottom = y + x2;
}else{
if(bounds['l'] < 0){
left = 0;
}else{
left = bounds['l'];
}
if(bounds['r'] > width){
right = width;
}else{
right = bounds['r'];
}
if(bounds['t'] < 0){
top = 0;
}else{
top = bounds['t'];
}
if(bounds['b'] > height){
bottom = height;
}else{
bottom = bounds['b'];
}
}
image = actx.getImageData(left, top, right-left, bottom-top);
length = image.data.length;
// loop thru the area
for(var i=3; i < length; i+=4){
// [0] -> r, [1] -> g, [2] -> b, [3] -> alpha
alpha = image.data[i],
offset = alpha*4;
if(!offset)
continue;
// we ve started with i=3
// set the new r, g and b values
finalAlpha = (alpha < opacity)?alpha:opacity;
image.data[i-3]=palette[offset];
image.data[i-2]=palette[offset+1];
image.data[i-1]=palette[offset+2];
if (premultiplyAlpha) {
// To fix browsers that premultiply incorrectly, we'll pass in a value scaled
// appropriately so when the multiplication happens the correct value will result.
image.data[i-3] /= 255/finalAlpha;
image.data[i-2] /= 255/finalAlpha;
image.data[i-1] /= 255/finalAlpha;
}
// we want the heatmap to have a gradient from transparent to the colors
// as long as alpha is lower than the defined opacity (maximum), we'll use the alpha value
image.data[i] = finalAlpha;
}
// the rgb data manipulation didn't affect the ImageData object(defined on the top)
// after the manipulation process we have to set the manipulated data to the ImageData object
// image.data = imageData;
ctx.putImageData(image, left, top);
},
drawAlpha: function(x, y, count, colorize){
// storing the variables because they will be often used
var me = this,
radius = me.get("radius"),
ctx = me.get("actx"),
max = me.get("max"),
bounds = me.get("bounds"),
xb = x - (1.5 * radius) >> 0, yb = y - (1.5 * radius) >> 0,
xc = x + (1.5 * radius) >> 0, yc = y + (1.5 * radius) >> 0;
ctx.shadowColor = ('rgba(0,0,0,'+((count)?(count/me.store.max):'0.1')+')');
ctx.shadowOffsetX = 15000;
ctx.shadowOffsetY = 15000;
ctx.shadowBlur = 15;
ctx.beginPath();
ctx.arc(x - 15000, y - 15000, radius, 0, Math.PI * 2, true);
ctx.closePath();
ctx.fill();
if(colorize){
// finally colorize the area
me.colorize(xb,yb);
}else{
// or update the boundaries for the area that then should be colorized
if(xb < bounds["l"]){
bounds["l"] = xb;
}
if(yb < bounds["t"]){
bounds["t"] = yb;
}
if(xc > bounds['r']){
bounds['r'] = xc;
}
if(yc > bounds['b']){
bounds['b'] = yc;
}
}
},
toggleDisplay: function(){
var me = this,
visible = me.get("visible"),
canvas = me.get("canvas");
if(!visible)
canvas.style.display = "block";
else
canvas.style.display = "none";
me.set("visible", !visible);
},
// dataURL export
getImageData: function(){
return this.get("canvas").toDataURL();
},
clear: function(){
var me = this,
w = me.get("width"),
h = me.get("height");
me.store.set("data",[]);
// @TODO: reset stores max to 1
//me.store.max = 1;
me.get("ctx").clearRect(0,0,w,h);
me.get("actx").clearRect(0,0,w,h);
},
cleanup: function(){
var me = this;
me.get("element").removeChild(me.get("canvas"));
}
};
return {
create: function(config){
return new heatmap(config);
},
util: {
mousePosition: function(ev){
// this doesn't work right
// rather use
/*
// this = element to observe
var x = ev.pageX - this.offsetLeft;
var y = ev.pageY - this.offsetTop;
*/
var x, y;
if (ev.layerX) { // Firefox
x = ev.layerX;
y = ev.layerY;
} else if (ev.offsetX) { // Opera
x = ev.offsetX;
y = ev.offsetY;
}
if(typeof(x)=='undefined')
return;
return [x,y];
}
}
};
})();
var HeatmapOverlay = function(opts) {
try {
BMap;
} catch (e) {
throw Error('Baidu Map JS API is not ready yet!');
}
if (!HeatmapOverlay._isExtended) {
HeatmapOverlay._isExtended = true;
inherits(HeatmapOverlay, BMap.Overlay, "HeatmapOverlay");
var newHeatmap = new HeatmapOverlay(opts);
this.__proto__ = newHeatmap.__proto__;
}
// HeatmapOverlay.prototype = new BMap.Overlay();
this.conf = opts;
this.heatmap = null;
this.latlngs = [];
this.bounds = null;
this._moveendHandler = this._moveendHandler.bind(this);
}
HeatmapOverlay.prototype.initialize = function(map) {
this._map = map;
var el = document.createElement("div");
el.style.position = "absolute";
el.style.top = 0;
el.style.left = 0;
el.style.border = 0;
el.style.width = this._map.getSize().width + "px";
el.style.height = this._map.getSize().height + "px";
this.conf.element = el;
map.getPanes().mapPane.appendChild(el);
this.heatmap = heatmapFactory.create(this.conf);
this._div = el;
return el;
}
HeatmapOverlay.prototype.draw = function() {
var currentBounds = this._map.getBounds();
if (currentBounds.equals(this.bounds)) {
return;
}
this.bounds = currentBounds;
var ne = this._map.pointToOverlayPixel(currentBounds.getNorthEast()),
sw = this._map.pointToOverlayPixel(currentBounds.getSouthWest());
if (!ne || !sw) {
return
}
var topY = ne.y,
leftX = sw.x,
h = sw.y - ne.y,
w = ne.x - sw.x;
this.conf.element.style.left = leftX + 'px';
this.conf.element.style.top = topY + 'px';
this.conf.element.style.width = w + 'px';
this.conf.element.style.height = h + 'px';
this.heatmap.store.get("heatmap").resize();
if (this.latlngs.length > 0) {
this.heatmap.clear();
var len = this.latlngs.length;
var d = {
max: this.heatmap.store.max,
data: []
};
while (len--) {
var latlng = this.latlngs[len].latlng;
if (!currentBounds.containsPoint(latlng)) {
continue;
}
var divPixel = this._map.pointToOverlayPixel(latlng),
screenPixel = new BMap.Pixel(divPixel.x - leftX, divPixel.y - topY);
var roundedPoint = this.pixelTransform(screenPixel);
d.data.push({
x: roundedPoint.x,
y: roundedPoint.y,
count: this.latlngs[len].c
});
}
this.heatmap.store.setDataSet(d);
}
}
HeatmapOverlay.prototype.pixelTransform = function(p) {
var w = this.heatmap.get("width"),
h = this.heatmap.get("height");
while (p.x < 0) {
p.x += w;
}
while (p.x > w) {
p.x -= w;
}
while (p.y < 0) {
p.y += h;
}
while (p.y > h) {
p.y -= h;
}
p.x = (p.x >> 0);
p.y = (p.y >> 0);
return p;
}
HeatmapOverlay.prototype._moveendHandler = function (e) {
this.setDataSet(this._data);
delete this._data;
this._map.removeEventListener('moveend', this._moveendHandler);
}
HeatmapOverlay.prototype.setDataSet = function(data) {
if (!this._map) {
return;
}
var currentBounds = this._map.getBounds();
var ne = this._map.pointToOverlayPixel(currentBounds.getNorthEast()),
sw = this._map.pointToOverlayPixel(currentBounds.getSouthWest());
if (!ne || !sw) {
this._data = data
this._map.addEventListener('moveend', this._moveendHandler);
}
var mapdata = {
max: data.max,
data: []
};
var d = data.data,
dlen = d.length;
this.latlngs = [];
while (dlen--) {
var latlng = new BMap.Point(d[dlen].lng, d[dlen].lat);
this.latlngs.push({
latlng: latlng,
c: d[dlen].count
});
if (!currentBounds.containsPoint(latlng)) {
continue;
}
var divPixel = this._map.pointToOverlayPixel(latlng),
leftX = this._map.pointToOverlayPixel(currentBounds.getSouthWest()).x,
topY = this._map.pointToOverlayPixel(currentBounds.getNorthEast()).y,
screenPixel = new BMap.Pixel(divPixel.x - leftX, divPixel.y - topY);
var point = this.pixelTransform(screenPixel);
mapdata.data.push({
x: point.x,
y: point.y,
count: d[dlen].count
});
}
this.heatmap.clear();
this.heatmap.store.setDataSet(mapdata);
}
HeatmapOverlay.prototype.addDataPoint = function(lng, lat, count) {
var latlng = new BMap.Point(lng, lat),
point = this.pixelTransform(this._map.pointToOverlayPixel(latlng));
this.heatmap.store.addDataPoint(point.x, point.y, count);
this.latlngs.push({
latlng: latlng,
c: count
});
}
HeatmapOverlay.prototype.toggle = function() {
this.heatmap.toggleDisplay();
}
return HeatmapOverlay;
});