path.js 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. import { Quad as QuadUtil, Cubic as CubicUtil, Arc as EllipseArcUtil } from '@antv/g-math';
  2. import { path2Segments } from '@antv/path-util';
  3. import { isNumberEqual, max, min } from '@antv/util';
  4. import { mergeArrowBBox } from './util';
  5. function getPathBox(segments, lineWidth) {
  6. var xArr = [];
  7. var yArr = [];
  8. var segmentsWithAngle = [];
  9. for (var i = 0; i < segments.length; i++) {
  10. var segment = segments[i];
  11. var currentPoint = segment.currentPoint, params = segment.params, prePoint = segment.prePoint;
  12. var box = void 0;
  13. switch (segment.command) {
  14. case 'Q':
  15. box = QuadUtil.box(prePoint[0], prePoint[1], params[1], params[2], params[3], params[4]);
  16. break;
  17. case 'C':
  18. box = CubicUtil.box(prePoint[0], prePoint[1], params[1], params[2], params[3], params[4], params[5], params[6]);
  19. break;
  20. case 'A':
  21. var arcParams = segment.arcParams;
  22. box = EllipseArcUtil.box(arcParams.cx, arcParams.cy, arcParams.rx, arcParams.ry, arcParams.xRotation, arcParams.startAngle, arcParams.endAngle);
  23. break;
  24. default:
  25. xArr.push(currentPoint[0]);
  26. yArr.push(currentPoint[1]);
  27. break;
  28. }
  29. if (box) {
  30. segment.box = box;
  31. xArr.push(box.x, box.x + box.width);
  32. yArr.push(box.y, box.y + box.height);
  33. }
  34. if (lineWidth && (segment.command === 'L' || segment.command === 'M') && segment.prePoint && segment.nextPoint) {
  35. segmentsWithAngle.push(segment);
  36. }
  37. }
  38. // bbox calculation should ignore NaN for path attribute
  39. // ref: https://github.com/antvis/g/issues/210
  40. // ref: https://github.com/antvis/G2/issues/3109
  41. xArr = xArr.filter(function (item) { return !Number.isNaN(item) && item !== Infinity && item !== -Infinity; });
  42. yArr = yArr.filter(function (item) { return !Number.isNaN(item) && item !== Infinity && item !== -Infinity; });
  43. var minX = min(xArr);
  44. var minY = min(yArr);
  45. var maxX = max(xArr);
  46. var maxY = max(yArr);
  47. if (segmentsWithAngle.length === 0) {
  48. return {
  49. x: minX,
  50. y: minY,
  51. width: maxX - minX,
  52. height: maxY - minY,
  53. };
  54. }
  55. for (var i = 0; i < segmentsWithAngle.length; i++) {
  56. var segment = segmentsWithAngle[i];
  57. var currentPoint = segment.currentPoint;
  58. var extra = void 0;
  59. if (currentPoint[0] === minX) {
  60. extra = getExtraFromSegmentWithAngle(segment, lineWidth);
  61. minX = minX - extra.xExtra;
  62. }
  63. else if (currentPoint[0] === maxX) {
  64. extra = getExtraFromSegmentWithAngle(segment, lineWidth);
  65. maxX = maxX + extra.xExtra;
  66. }
  67. if (currentPoint[1] === minY) {
  68. extra = getExtraFromSegmentWithAngle(segment, lineWidth);
  69. minY = minY - extra.yExtra;
  70. }
  71. else if (currentPoint[1] === maxY) {
  72. extra = getExtraFromSegmentWithAngle(segment, lineWidth);
  73. maxY = maxY + extra.yExtra;
  74. }
  75. }
  76. return {
  77. x: minX,
  78. y: minY,
  79. width: maxX - minX,
  80. height: maxY - minY,
  81. };
  82. }
  83. function getExtraFromSegmentWithAngle(segment, lineWidth) {
  84. var prePoint = segment.prePoint, currentPoint = segment.currentPoint, nextPoint = segment.nextPoint;
  85. var currentAndPre = Math.pow(currentPoint[0] - prePoint[0], 2) + Math.pow(currentPoint[1] - prePoint[1], 2);
  86. var currentAndNext = Math.pow(currentPoint[0] - nextPoint[0], 2) + Math.pow(currentPoint[1] - nextPoint[1], 2);
  87. var preAndNext = Math.pow(prePoint[0] - nextPoint[0], 2) + Math.pow(prePoint[1] - nextPoint[1], 2);
  88. // 以 currentPoint 为顶点的夹角
  89. var currentAngle = Math.acos((currentAndPre + currentAndNext - preAndNext) / (2 * Math.sqrt(currentAndPre) * Math.sqrt(currentAndNext)));
  90. // 夹角为空、 0 或 PI 时,不需要计算夹角处的额外宽度
  91. // 注意: 由于计算精度问题,夹角为 0 的情况计算出来的角度可能是一个很小的值,还需要判断其与 0 是否近似相等
  92. if (!currentAngle || Math.sin(currentAngle) === 0 || isNumberEqual(currentAngle, 0)) {
  93. return {
  94. xExtra: 0,
  95. yExtra: 0,
  96. };
  97. }
  98. var xAngle = Math.abs(Math.atan2(nextPoint[1] - currentPoint[1], nextPoint[0] - currentPoint[0]));
  99. var yAngle = Math.abs(Math.atan2(nextPoint[0] - currentPoint[0], nextPoint[1] - currentPoint[1]));
  100. // 将夹角转为锐角
  101. xAngle = xAngle > Math.PI / 2 ? Math.PI - xAngle : xAngle;
  102. yAngle = yAngle > Math.PI / 2 ? Math.PI - yAngle : yAngle;
  103. // 这里不考虑在水平和垂直方向的投影,直接使用最大差值
  104. // 由于上层统一加减了二分之一线宽,这里需要进行弥补
  105. var extra = {
  106. // 水平方向投影
  107. xExtra: Math.cos(currentAngle / 2 - xAngle) * ((lineWidth / 2) * (1 / Math.sin(currentAngle / 2))) - lineWidth / 2 || 0,
  108. // 垂直方向投影
  109. yExtra: Math.cos(yAngle - currentAngle / 2) * ((lineWidth / 2) * (1 / Math.sin(currentAngle / 2))) - lineWidth / 2 || 0,
  110. };
  111. return extra;
  112. }
  113. export default function (shape) {
  114. var attrs = shape.attr();
  115. var path = attrs.path, stroke = attrs.stroke;
  116. var lineWidth = stroke ? attrs.lineWidth : 0; // 只有有 stroke 时,lineWidth 才生效
  117. var segments = shape.get('segments') || path2Segments(path);
  118. var _a = getPathBox(segments, lineWidth), x = _a.x, y = _a.y, width = _a.width, height = _a.height;
  119. var bbox = {
  120. minX: x,
  121. minY: y,
  122. maxX: x + width,
  123. maxY: y + height,
  124. };
  125. bbox = mergeArrowBBox(shape, bbox);
  126. return {
  127. x: bbox.minX,
  128. y: bbox.minY,
  129. width: bbox.maxX - bbox.minX,
  130. height: bbox.maxY - bbox.minY,
  131. };
  132. }
  133. //# sourceMappingURL=path.js.map