poly.js 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225
  1. var Path = require("zrender/lib/graphic/Path");
  2. var vec2 = require("zrender/lib/core/vector");
  3. var fixClipWithShadow = require("zrender/lib/graphic/helper/fixClipWithShadow");
  4. // Poly path support NaN point
  5. var vec2Min = vec2.min;
  6. var vec2Max = vec2.max;
  7. var scaleAndAdd = vec2.scaleAndAdd;
  8. var v2Copy = vec2.copy; // Temporary variable
  9. var v = [];
  10. var cp0 = [];
  11. var cp1 = [];
  12. function isPointNull(p) {
  13. return isNaN(p[0]) || isNaN(p[1]);
  14. }
  15. function drawSegment(ctx, points, start, segLen, allLen, dir, smoothMin, smoothMax, smooth, smoothMonotone, connectNulls) {
  16. var prevIdx = 0;
  17. var idx = start;
  18. for (var k = 0; k < segLen; k++) {
  19. var p = points[idx];
  20. if (idx >= allLen || idx < 0) {
  21. break;
  22. }
  23. if (isPointNull(p)) {
  24. if (connectNulls) {
  25. idx += dir;
  26. continue;
  27. }
  28. break;
  29. }
  30. if (idx === start) {
  31. ctx[dir > 0 ? 'moveTo' : 'lineTo'](p[0], p[1]);
  32. v2Copy(cp0, p);
  33. } else {
  34. if (smooth > 0) {
  35. var nextIdx = idx + dir;
  36. var nextP = points[nextIdx];
  37. if (connectNulls) {
  38. // Find next point not null
  39. while (nextP && isPointNull(points[nextIdx])) {
  40. nextIdx += dir;
  41. nextP = points[nextIdx];
  42. }
  43. }
  44. var ratioNextSeg = 0.5;
  45. var prevP = points[prevIdx];
  46. var nextP = points[nextIdx]; // Last point
  47. if (!nextP || isPointNull(nextP)) {
  48. v2Copy(cp1, p);
  49. } else {
  50. // If next data is null in not connect case
  51. if (isPointNull(nextP) && !connectNulls) {
  52. nextP = p;
  53. }
  54. vec2.sub(v, nextP, prevP);
  55. var lenPrevSeg;
  56. var lenNextSeg;
  57. if (smoothMonotone === 'x' || smoothMonotone === 'y') {
  58. var dim = smoothMonotone === 'x' ? 0 : 1;
  59. lenPrevSeg = Math.abs(p[dim] - prevP[dim]);
  60. lenNextSeg = Math.abs(p[dim] - nextP[dim]);
  61. } else {
  62. lenPrevSeg = vec2.dist(p, prevP);
  63. lenNextSeg = vec2.dist(p, nextP);
  64. } // Use ratio of seg length
  65. ratioNextSeg = lenNextSeg / (lenNextSeg + lenPrevSeg);
  66. scaleAndAdd(cp1, p, v, -smooth * (1 - ratioNextSeg));
  67. } // Smooth constraint
  68. vec2Min(cp0, cp0, smoothMax);
  69. vec2Max(cp0, cp0, smoothMin);
  70. vec2Min(cp1, cp1, smoothMax);
  71. vec2Max(cp1, cp1, smoothMin);
  72. ctx.bezierCurveTo(cp0[0], cp0[1], cp1[0], cp1[1], p[0], p[1]); // cp0 of next segment
  73. scaleAndAdd(cp0, p, v, smooth * ratioNextSeg);
  74. } else {
  75. ctx.lineTo(p[0], p[1]);
  76. }
  77. }
  78. prevIdx = idx;
  79. idx += dir;
  80. }
  81. return k;
  82. }
  83. function getBoundingBox(points, smoothConstraint) {
  84. var ptMin = [Infinity, Infinity];
  85. var ptMax = [-Infinity, -Infinity];
  86. if (smoothConstraint) {
  87. for (var i = 0; i < points.length; i++) {
  88. var pt = points[i];
  89. if (pt[0] < ptMin[0]) {
  90. ptMin[0] = pt[0];
  91. }
  92. if (pt[1] < ptMin[1]) {
  93. ptMin[1] = pt[1];
  94. }
  95. if (pt[0] > ptMax[0]) {
  96. ptMax[0] = pt[0];
  97. }
  98. if (pt[1] > ptMax[1]) {
  99. ptMax[1] = pt[1];
  100. }
  101. }
  102. }
  103. return {
  104. min: smoothConstraint ? ptMin : ptMax,
  105. max: smoothConstraint ? ptMax : ptMin
  106. };
  107. }
  108. var Polyline = Path.extend({
  109. type: 'ec-polyline',
  110. shape: {
  111. points: [],
  112. smooth: 0,
  113. smoothConstraint: true,
  114. smoothMonotone: null,
  115. connectNulls: false
  116. },
  117. style: {
  118. fill: null,
  119. stroke: '#000'
  120. },
  121. brush: fixClipWithShadow(Path.prototype.brush),
  122. buildPath: function (ctx, shape) {
  123. var points = shape.points;
  124. var i = 0;
  125. var len = points.length;
  126. var result = getBoundingBox(points, shape.smoothConstraint);
  127. if (shape.connectNulls) {
  128. // Must remove first and last null values avoid draw error in polygon
  129. for (; len > 0; len--) {
  130. if (!isPointNull(points[len - 1])) {
  131. break;
  132. }
  133. }
  134. for (; i < len; i++) {
  135. if (!isPointNull(points[i])) {
  136. break;
  137. }
  138. }
  139. }
  140. while (i < len) {
  141. i += drawSegment(ctx, points, i, len, len, 1, result.min, result.max, shape.smooth, shape.smoothMonotone, shape.connectNulls) + 1;
  142. }
  143. }
  144. });
  145. var Polygon = Path.extend({
  146. type: 'ec-polygon',
  147. shape: {
  148. points: [],
  149. // Offset between stacked base points and points
  150. stackedOnPoints: [],
  151. smooth: 0,
  152. stackedOnSmooth: 0,
  153. smoothConstraint: true,
  154. smoothMonotone: null,
  155. connectNulls: false
  156. },
  157. brush: fixClipWithShadow(Path.prototype.brush),
  158. buildPath: function (ctx, shape) {
  159. var points = shape.points;
  160. var stackedOnPoints = shape.stackedOnPoints;
  161. var i = 0;
  162. var len = points.length;
  163. var smoothMonotone = shape.smoothMonotone;
  164. var bbox = getBoundingBox(points, shape.smoothConstraint);
  165. var stackedOnBBox = getBoundingBox(stackedOnPoints, shape.smoothConstraint);
  166. if (shape.connectNulls) {
  167. // Must remove first and last null values avoid draw error in polygon
  168. for (; len > 0; len--) {
  169. if (!isPointNull(points[len - 1])) {
  170. break;
  171. }
  172. }
  173. for (; i < len; i++) {
  174. if (!isPointNull(points[i])) {
  175. break;
  176. }
  177. }
  178. }
  179. while (i < len) {
  180. var k = drawSegment(ctx, points, i, len, len, 1, bbox.min, bbox.max, shape.smooth, smoothMonotone, shape.connectNulls);
  181. drawSegment(ctx, stackedOnPoints, i + k - 1, k, len, -1, stackedOnBBox.min, stackedOnBBox.max, shape.stackedOnSmooth, smoothMonotone, shape.connectNulls);
  182. i += k + 1;
  183. ctx.closePath();
  184. }
  185. }
  186. });
  187. exports.Polyline = Polyline;
  188. exports.Polygon = Polygon;