index.js 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608
  1. import TextIconOverlay from 'bmaplib.texticonoverlay';
  2. /**
  3. * 获取一个扩展的视图范围,把上下左右都扩大一样的像素值。
  4. * @param {Map} map BMap.Map的实例化对象
  5. * @param {BMap.Bounds} bounds BMap.Bounds的实例化对象
  6. * @param {Number} gridSize 要扩大的像素值
  7. *
  8. * @return {BMap.Bounds} 返回扩大后的视图范围。
  9. */
  10. var getExtendedBounds = function(map, bounds, gridSize){
  11. bounds = cutBoundsInRange(bounds);
  12. var pixelNE = map.pointToPixel(bounds.getNorthEast());
  13. var pixelSW = map.pointToPixel(bounds.getSouthWest());
  14. pixelNE.x += gridSize;
  15. pixelNE.y -= gridSize;
  16. pixelSW.x -= gridSize;
  17. pixelSW.y += gridSize;
  18. var newNE = map.pixelToPoint(pixelNE);
  19. var newSW = map.pixelToPoint(pixelSW);
  20. return new BMap.Bounds(newSW, newNE);
  21. };
  22. /**
  23. * 按照百度地图支持的世界范围对bounds进行边界处理
  24. * @param {BMap.Bounds} bounds BMap.Bounds的实例化对象
  25. *
  26. * @return {BMap.Bounds} 返回不越界的视图范围
  27. */
  28. var cutBoundsInRange = function (bounds) {
  29. var maxX = getRange(bounds.getNorthEast().lng, -180, 180);
  30. var minX = getRange(bounds.getSouthWest().lng, -180, 180);
  31. var maxY = getRange(bounds.getNorthEast().lat, -74, 74);
  32. var minY = getRange(bounds.getSouthWest().lat, -74, 74);
  33. return new BMap.Bounds(new BMap.Point(minX, minY), new BMap.Point(maxX, maxY));
  34. };
  35. /**
  36. * 对单个值进行边界处理。
  37. * @param {Number} i 要处理的数值
  38. * @param {Number} min 下边界值
  39. * @param {Number} max 上边界值
  40. *
  41. * @return {Number} 返回不越界的数值
  42. */
  43. var getRange = function (i, mix, max) {
  44. mix && (i = Math.max(i, mix));
  45. max && (i = Math.min(i, max));
  46. return i;
  47. };
  48. /**
  49. * 判断给定的对象是否为数组
  50. * @param {Object} source 要测试的对象
  51. *
  52. * @return {Boolean} 如果是数组返回true,否则返回false
  53. */
  54. var isArray = function (source) {
  55. return '[object Array]' === Object.prototype.toString.call(source);
  56. };
  57. /**
  58. * 返回item在source中的索引位置
  59. * @param {Object} item 要测试的对象
  60. * @param {Array} source 数组
  61. *
  62. * @return {Number} 如果在数组内,返回索引,否则返回-1
  63. */
  64. var indexOf = function(item, source){
  65. var index = -1;
  66. if(isArray(source)){
  67. if (source.indexOf) {
  68. index = source.indexOf(item);
  69. } else {
  70. for (var i = 0, m; m = source[i]; i++) {
  71. if (m === item) {
  72. index = i;
  73. break;
  74. }
  75. }
  76. }
  77. }
  78. return index;
  79. };
  80. /**
  81. *@exports MarkerClusterer as BMapLib.MarkerClusterer
  82. */
  83. /**
  84. * MarkerClusterer
  85. * @class 用来解决加载大量点要素到地图上产生覆盖现象的问题,并提高性能
  86. * @constructor
  87. * @param {Map} map 地图的一个实例。
  88. * @param {Json Object} options 可选参数,可选项包括:<br />
  89. * markers {Array<Marker>} 要聚合的标记数组<br />
  90. * girdSize {Number} 聚合计算时网格的像素大小,默认60<br />
  91. * maxZoom {Number} 最大的聚合级别,大于该级别就不进行相应的聚合<br />
  92. * minClusterSize {Number} 最小的聚合数量,小于该数量的不能成为一个聚合,默认为2<br />
  93. * isAverangeCenter {Boolean} 聚合点的落脚位置是否是所有聚合在内点的平均值,默认为否,落脚在聚合内的第一个点<br />
  94. * styles {Array<IconStyle>} 自定义聚合后的图标风格,请参考TextIconOverlay类<br />
  95. */
  96. var MarkerClusterer = function(map, options){
  97. try {
  98. BMap;
  99. } catch (e) {
  100. throw Error('Baidu Map JS API is not ready yet!');
  101. }
  102. if (!map){
  103. return;
  104. }
  105. this._map = map;
  106. this._markers = [];
  107. this._clusters = [];
  108. var opts = options || {};
  109. this._gridSize = opts["gridSize"] || 60;
  110. this._maxZoom = opts["maxZoom"] || 18;
  111. this._minClusterSize = opts["minClusterSize"] || 2;
  112. this._isAverageCenter = false;
  113. if (opts['isAverageCenter'] != undefined) {
  114. this._isAverageCenter = opts['isAverageCenter'];
  115. }
  116. this._styles = opts["styles"] || [];
  117. var that = this;
  118. this._map.addEventListener("zoomend",function(){
  119. that._redraw();
  120. });
  121. this._map.addEventListener("moveend",function(){
  122. that._redraw();
  123. });
  124. var mkrs = opts["markers"];
  125. isArray(mkrs) && this.addMarkers(mkrs);
  126. };
  127. /**
  128. * 添加要聚合的标记数组。
  129. * @param {Array<Marker>} markers 要聚合的标记数组
  130. *
  131. * @return 无返回值。
  132. */
  133. MarkerClusterer.prototype.addMarkers = function(markers){
  134. if (!markers.length) {
  135. return
  136. }
  137. for(var i = 0, len = markers.length; i <len ; i++){
  138. this._pushMarkerTo(markers[i]);
  139. }
  140. this._createClusters();
  141. };
  142. /**
  143. * 把一个标记添加到要聚合的标记数组中
  144. * @param {BMap.Marker} marker 要添加的标记
  145. *
  146. * @return 无返回值。
  147. */
  148. MarkerClusterer.prototype._pushMarkerTo = function(marker){
  149. var index = indexOf(marker, this._markers);
  150. if(index === -1){
  151. marker.isInCluster = false;
  152. this._markers.push(marker);//Marker拖放后enableDragging不做变化,忽略
  153. }
  154. };
  155. /**
  156. * 添加一个聚合的标记。
  157. * @param {BMap.Marker} marker 要聚合的单个标记。
  158. * @return 无返回值。
  159. */
  160. MarkerClusterer.prototype.addMarker = function(marker) {
  161. this._pushMarkerTo(marker);
  162. this._createClusters();
  163. };
  164. /**
  165. * 根据所给定的标记,创建聚合点
  166. * @return 无返回值
  167. */
  168. MarkerClusterer.prototype._createClusters = function(){
  169. var mapBounds = this._map.getBounds();
  170. if (!mapBounds.getCenter()) {
  171. return
  172. }
  173. var extendedBounds = getExtendedBounds(this._map, mapBounds, this._gridSize);
  174. for(var i = 0, marker; marker = this._markers[i]; i++){
  175. if(!marker.isInCluster && extendedBounds.containsPoint(marker.getPosition()) ){
  176. this._addToClosestCluster(marker);
  177. }
  178. }
  179. };
  180. /**
  181. * 根据标记的位置,把它添加到最近的聚合中
  182. * @param {BMap.Marker} marker 要进行聚合的单个标记
  183. *
  184. * @return 无返回值。
  185. */
  186. MarkerClusterer.prototype._addToClosestCluster = function (marker){
  187. var distance = 4000000;
  188. var clusterToAddTo = null;
  189. var position = marker.getPosition();
  190. for(var i = 0, cluster; cluster = this._clusters[i]; i++){
  191. var center = cluster.getCenter();
  192. if(center){
  193. var d = this._map.getDistance(center, marker.getPosition());
  194. if(d < distance){
  195. distance = d;
  196. clusterToAddTo = cluster;
  197. }
  198. }
  199. }
  200. if (clusterToAddTo && clusterToAddTo.isMarkerInClusterBounds(marker)){
  201. clusterToAddTo.addMarker(marker);
  202. } else {
  203. var cluster = new Cluster(this);
  204. cluster.addMarker(marker);
  205. this._clusters.push(cluster);
  206. }
  207. };
  208. /**
  209. * 清除上一次的聚合的结果
  210. * @return 无返回值。
  211. */
  212. MarkerClusterer.prototype._clearLastClusters = function(){
  213. for(var i = 0, cluster; cluster = this._clusters[i]; i++){
  214. cluster.remove();
  215. }
  216. this._clusters = [];//置空Cluster数组
  217. this._removeMarkersFromCluster();//把Marker的cluster标记设为false
  218. };
  219. /**
  220. * 清除某个聚合中的所有标记
  221. * @return 无返回值
  222. */
  223. MarkerClusterer.prototype._removeMarkersFromCluster = function(){
  224. for(var i = 0, marker; marker = this._markers[i]; i++){
  225. marker.isInCluster = false;
  226. }
  227. };
  228. /**
  229. * 把所有的标记从地图上清除
  230. * @return 无返回值
  231. */
  232. MarkerClusterer.prototype._removeMarkersFromMap = function(){
  233. for(var i = 0, marker; marker = this._markers[i]; i++){
  234. marker.isInCluster = false;
  235. var label = marker.getLabel();
  236. this._map.removeOverlay(marker);
  237. marker.setLabel(label);
  238. }
  239. };
  240. /**
  241. * 删除单个标记
  242. * @param {BMap.Marker} marker 需要被删除的marker
  243. *
  244. * @return {Boolean} 删除成功返回true,否则返回false
  245. */
  246. MarkerClusterer.prototype._removeMarker = function(marker) {
  247. var index = indexOf(marker, this._markers);
  248. if (index === -1) {
  249. return false;
  250. }
  251. this._map.removeOverlay(marker);
  252. this._markers.splice(index, 1);
  253. return true;
  254. };
  255. /**
  256. * 删除单个标记
  257. * @param {BMap.Marker} marker 需要被删除的marker
  258. *
  259. * @return {Boolean} 删除成功返回true,否则返回false
  260. */
  261. MarkerClusterer.prototype.removeMarker = function(marker) {
  262. var success = this._removeMarker(marker);
  263. if (success) {
  264. this._clearLastClusters();
  265. this._createClusters();
  266. }
  267. return success;
  268. };
  269. /**
  270. * 删除一组标记
  271. * @param {Array<BMap.Marker>} markers 需要被删除的marker数组
  272. *
  273. * @return {Boolean} 删除成功返回true,否则返回false
  274. */
  275. MarkerClusterer.prototype.removeMarkers = function(markers) {
  276. var success = false;
  277. for (var i = 0; i < markers.length; i++) {
  278. var r = this._removeMarker(markers[i]);
  279. success = success || r;
  280. }
  281. if (success) {
  282. this._clearLastClusters();
  283. this._createClusters();
  284. }
  285. return success;
  286. };
  287. /**
  288. * 从地图上彻底清除所有的标记
  289. * @return 无返回值
  290. */
  291. MarkerClusterer.prototype.clearMarkers = function() {
  292. this._clearLastClusters();
  293. this._removeMarkersFromMap();
  294. this._markers = [];
  295. };
  296. /**
  297. * 重新生成,比如改变了属性等
  298. * @return 无返回值
  299. */
  300. MarkerClusterer.prototype._redraw = function () {
  301. this._clearLastClusters();
  302. this._createClusters();
  303. };
  304. /**
  305. * 获取网格大小
  306. * @return {Number} 网格大小
  307. */
  308. MarkerClusterer.prototype.getGridSize = function() {
  309. return this._gridSize;
  310. };
  311. /**
  312. * 设置网格大小
  313. * @param {Number} size 网格大小
  314. * @return 无返回值
  315. */
  316. MarkerClusterer.prototype.setGridSize = function(size) {
  317. this._gridSize = size;
  318. this._redraw();
  319. };
  320. /**
  321. * 获取聚合的最大缩放级别。
  322. * @return {Number} 聚合的最大缩放级别。
  323. */
  324. MarkerClusterer.prototype.getMaxZoom = function() {
  325. return this._maxZoom;
  326. };
  327. /**
  328. * 设置聚合的最大缩放级别
  329. * @param {Number} maxZoom 聚合的最大缩放级别
  330. * @return 无返回值
  331. */
  332. MarkerClusterer.prototype.setMaxZoom = function(maxZoom) {
  333. this._maxZoom = maxZoom;
  334. this._redraw();
  335. };
  336. /**
  337. * 获取聚合的样式风格集合
  338. * @return {Array<IconStyle>} 聚合的样式风格集合
  339. */
  340. MarkerClusterer.prototype.getStyles = function() {
  341. return this._styles;
  342. };
  343. /**
  344. * 设置聚合的样式风格集合
  345. * @param {Array<IconStyle>} styles 样式风格数组
  346. * @return 无返回值
  347. */
  348. MarkerClusterer.prototype.setStyles = function(styles) {
  349. this._styles = styles;
  350. this._redraw();
  351. };
  352. /**
  353. * 获取单个聚合的最小数量。
  354. * @return {Number} 单个聚合的最小数量。
  355. */
  356. MarkerClusterer.prototype.getMinClusterSize = function() {
  357. return this._minClusterSize;
  358. };
  359. /**
  360. * 设置单个聚合的最小数量。
  361. * @param {Number} size 单个聚合的最小数量。
  362. * @return 无返回值。
  363. */
  364. MarkerClusterer.prototype.setMinClusterSize = function(size) {
  365. this._minClusterSize = size;
  366. this._redraw();
  367. };
  368. /**
  369. * 获取单个聚合的落脚点是否是聚合内所有标记的平均中心。
  370. * @return {Boolean} true或false。
  371. */
  372. MarkerClusterer.prototype.isAverageCenter = function() {
  373. return this._isAverageCenter;
  374. };
  375. /**
  376. * 获取聚合的Map实例。
  377. * @return {Map} Map的示例。
  378. */
  379. MarkerClusterer.prototype.getMap = function() {
  380. return this._map;
  381. };
  382. /**
  383. * 获取所有的标记数组。
  384. * @return {Array<Marker>} 标记数组。
  385. */
  386. MarkerClusterer.prototype.getMarkers = function() {
  387. return this._markers;
  388. };
  389. /**
  390. * 获取聚合的总数量。
  391. * @return {Number} 聚合的总数量。
  392. */
  393. MarkerClusterer.prototype.getClustersCount = function() {
  394. var count = 0;
  395. for(var i = 0, cluster; cluster = this._clusters[i]; i++){
  396. cluster.isReal() && count++;
  397. }
  398. return count;
  399. };
  400. /**
  401. * @ignore
  402. * Cluster
  403. * @class 表示一个聚合对象,该聚合,包含有N个标记,这N个标记组成的范围,并有予以显示在Map上的TextIconOverlay等。
  404. * @constructor
  405. * @param {MarkerClusterer} markerClusterer 一个标记聚合器示例。
  406. */
  407. function Cluster(markerClusterer){
  408. this._markerClusterer = markerClusterer;
  409. this._map = markerClusterer.getMap();
  410. this._minClusterSize = markerClusterer.getMinClusterSize();
  411. this._isAverageCenter = markerClusterer.isAverageCenter();
  412. this._center = null;//落脚位置
  413. this._markers = [];//这个Cluster中所包含的markers
  414. this._gridBounds = null;//以中心点为准,向四边扩大gridSize个像素的范围,也即网格范围
  415. this._isReal = false; //真的是个聚合
  416. this._clusterMarker = new TextIconOverlay(this._center, this._markers.length, {"styles":this._markerClusterer.getStyles()});
  417. //this._map.addOverlay(this._clusterMarker);
  418. }
  419. /**
  420. * 向该聚合添加一个标记。
  421. * @param {Marker} marker 要添加的标记。
  422. * @return 无返回值。
  423. */
  424. Cluster.prototype.addMarker = function(marker){
  425. if(this.isMarkerInCluster(marker)){
  426. return false;
  427. }//也可用marker.isInCluster判断,外面判断OK,这里基本不会命中
  428. if (!this._center){
  429. this._center = marker.getPosition();
  430. this.updateGridBounds();//
  431. } else {
  432. if(this._isAverageCenter){
  433. var l = this._markers.length + 1;
  434. var lat = (this._center.lat * (l - 1) + marker.getPosition().lat) / l;
  435. var lng = (this._center.lng * (l - 1) + marker.getPosition().lng) / l;
  436. this._center = new BMap.Point(lng, lat);
  437. this.updateGridBounds();
  438. }//计算新的Center
  439. }
  440. marker.isInCluster = true;
  441. this._markers.push(marker);
  442. var len = this._markers.length;
  443. if(len < this._minClusterSize ){
  444. this._map.addOverlay(marker);
  445. //this.updateClusterMarker();
  446. return true;
  447. } else if (len === this._minClusterSize) {
  448. for (var i = 0; i < len; i++) {
  449. var label = this._markers[i].getLabel();
  450. this._markers[i].getMap() && this._map.removeOverlay(this._markers[i]);
  451. this._markers[i].setLabel(label);
  452. }
  453. }
  454. this._map.addOverlay(this._clusterMarker);
  455. this._isReal = true;
  456. this.updateClusterMarker();
  457. return true;
  458. };
  459. /**
  460. * 判断一个标记是否在该聚合中。
  461. * @param {Marker} marker 要判断的标记。
  462. * @return {Boolean} true或false。
  463. */
  464. Cluster.prototype.isMarkerInCluster= function(marker){
  465. if (this._markers.indexOf) {
  466. return this._markers.indexOf(marker) != -1;
  467. } else {
  468. for (var i = 0, m; m = this._markers[i]; i++) {
  469. if (m === marker) {
  470. return true;
  471. }
  472. }
  473. }
  474. return false;
  475. };
  476. /**
  477. * 判断一个标记是否在该聚合网格范围中。
  478. * @param {Marker} marker 要判断的标记。
  479. * @return {Boolean} true或false。
  480. */
  481. Cluster.prototype.isMarkerInClusterBounds = function(marker) {
  482. return this._gridBounds.containsPoint(marker.getPosition());
  483. };
  484. Cluster.prototype.isReal = function(marker) {
  485. return this._isReal;
  486. };
  487. /**
  488. * 更新该聚合的网格范围。
  489. * @return 无返回值。
  490. */
  491. Cluster.prototype.updateGridBounds = function() {
  492. var bounds = new BMap.Bounds(this._center, this._center);
  493. this._gridBounds = getExtendedBounds(this._map, bounds, this._markerClusterer.getGridSize());
  494. };
  495. /**
  496. * 更新该聚合的显示样式,也即TextIconOverlay。
  497. * @return 无返回值。
  498. */
  499. Cluster.prototype.updateClusterMarker = function () {
  500. if (this._map.getZoom() > this._markerClusterer.getMaxZoom()) {
  501. this._clusterMarker && this._map.removeOverlay(this._clusterMarker);
  502. for (var i = 0, marker; marker = this._markers[i]; i++) {
  503. this._map.addOverlay(marker);
  504. }
  505. return;
  506. }
  507. if (this._markers.length < this._minClusterSize) {
  508. this._clusterMarker.hide();
  509. return;
  510. }
  511. this._clusterMarker.setPosition(this._center);
  512. this._clusterMarker.setText(this._markers.length);
  513. this._clusterMarker.addEventListener && !this._clusterMarker._hasClickEvent && this._clusterMarker.addEventListener("click", function(event){
  514. this._clusterMarker._hasClickEvent = true;
  515. this._markers && this._map.setViewport(this.getBounds());
  516. }.bind(this));
  517. };
  518. /**
  519. * 删除该聚合。
  520. * @return 无返回值。
  521. */
  522. Cluster.prototype.remove = function(){
  523. for (var i = 0, m; m = this._markers[i]; i++) {
  524. var label = this._markers[i].getLabel();
  525. this._markers[i].getMap() && this._map.removeOverlay(this._markers[i]);
  526. this._markers[i].setLabel(label);
  527. }//清除散的标记点
  528. this._map.removeOverlay(this._clusterMarker);
  529. this._markers.length = 0;
  530. delete this._markers;
  531. }
  532. /**
  533. * 获取该聚合所包含的所有标记的最小外接矩形的范围。
  534. * @return {BMap.Bounds} 计算出的范围。
  535. */
  536. Cluster.prototype.getBounds = function() {
  537. var bounds = new BMap.Bounds(this._center,this._center);
  538. for (var i = 0, marker; marker = this._markers[i]; i++) {
  539. bounds.extend(marker.getPosition());
  540. }
  541. return bounds;
  542. };
  543. /**
  544. * 获取该聚合的落脚点。
  545. * @return {BMap.Point} 该聚合的落脚点。
  546. */
  547. Cluster.prototype.getCenter = function() {
  548. return this._center;
  549. };
  550. export default MarkerClusterer;