@antv/g-base
Version:
A common util collection for antv projects
133 lines • 5.97 kB
JavaScript
import { Quad as QuadUtil, Cubic as CubicUtil, Arc as EllipseArcUtil } from '@antv/g-math';
import { path2Segments } from '@antv/path-util';
import { isNumberEqual, max, min } from '@antv/util';
import { mergeArrowBBox } from './util';
function getPathBox(segments, lineWidth) {
var xArr = [];
var yArr = [];
var segmentsWithAngle = [];
for (var i = 0; i < segments.length; i++) {
var segment = segments[i];
var currentPoint = segment.currentPoint, params = segment.params, prePoint = segment.prePoint;
var box = void 0;
switch (segment.command) {
case 'Q':
box = QuadUtil.box(prePoint[0], prePoint[1], params[1], params[2], params[3], params[4]);
break;
case 'C':
box = CubicUtil.box(prePoint[0], prePoint[1], params[1], params[2], params[3], params[4], params[5], params[6]);
break;
case 'A':
var arcParams = segment.arcParams;
box = EllipseArcUtil.box(arcParams.cx, arcParams.cy, arcParams.rx, arcParams.ry, arcParams.xRotation, arcParams.startAngle, arcParams.endAngle);
break;
default:
xArr.push(currentPoint[0]);
yArr.push(currentPoint[1]);
break;
}
if (box) {
segment.box = box;
xArr.push(box.x, box.x + box.width);
yArr.push(box.y, box.y + box.height);
}
if (lineWidth && (segment.command === 'L' || segment.command === 'M') && segment.prePoint && segment.nextPoint) {
segmentsWithAngle.push(segment);
}
}
// bbox calculation should ignore NaN for path attribute
// ref: https://github.com/antvis/g/issues/210
// ref: https://github.com/antvis/G2/issues/3109
xArr = xArr.filter(function (item) { return !Number.isNaN(item) && item !== Infinity && item !== -Infinity; });
yArr = yArr.filter(function (item) { return !Number.isNaN(item) && item !== Infinity && item !== -Infinity; });
var minX = min(xArr);
var minY = min(yArr);
var maxX = max(xArr);
var maxY = max(yArr);
if (segmentsWithAngle.length === 0) {
return {
x: minX,
y: minY,
width: maxX - minX,
height: maxY - minY,
};
}
for (var i = 0; i < segmentsWithAngle.length; i++) {
var segment = segmentsWithAngle[i];
var currentPoint = segment.currentPoint;
var extra = void 0;
if (currentPoint[0] === minX) {
extra = getExtraFromSegmentWithAngle(segment, lineWidth);
minX = minX - extra.xExtra;
}
else if (currentPoint[0] === maxX) {
extra = getExtraFromSegmentWithAngle(segment, lineWidth);
maxX = maxX + extra.xExtra;
}
if (currentPoint[1] === minY) {
extra = getExtraFromSegmentWithAngle(segment, lineWidth);
minY = minY - extra.yExtra;
}
else if (currentPoint[1] === maxY) {
extra = getExtraFromSegmentWithAngle(segment, lineWidth);
maxY = maxY + extra.yExtra;
}
}
return {
x: minX,
y: minY,
width: maxX - minX,
height: maxY - minY,
};
}
function getExtraFromSegmentWithAngle(segment, lineWidth) {
var prePoint = segment.prePoint, currentPoint = segment.currentPoint, nextPoint = segment.nextPoint;
var currentAndPre = Math.pow(currentPoint[0] - prePoint[0], 2) + Math.pow(currentPoint[1] - prePoint[1], 2);
var currentAndNext = Math.pow(currentPoint[0] - nextPoint[0], 2) + Math.pow(currentPoint[1] - nextPoint[1], 2);
var preAndNext = Math.pow(prePoint[0] - nextPoint[0], 2) + Math.pow(prePoint[1] - nextPoint[1], 2);
// 以 currentPoint 为顶点的夹角
var currentAngle = Math.acos((currentAndPre + currentAndNext - preAndNext) / (2 * Math.sqrt(currentAndPre) * Math.sqrt(currentAndNext)));
// 夹角为空、 0 或 PI 时,不需要计算夹角处的额外宽度
// 注意: 由于计算精度问题,夹角为 0 的情况计算出来的角度可能是一个很小的值,还需要判断其与 0 是否近似相等
if (!currentAngle || Math.sin(currentAngle) === 0 || isNumberEqual(currentAngle, 0)) {
return {
xExtra: 0,
yExtra: 0,
};
}
var xAngle = Math.abs(Math.atan2(nextPoint[1] - currentPoint[1], nextPoint[0] - currentPoint[0]));
var yAngle = Math.abs(Math.atan2(nextPoint[0] - currentPoint[0], nextPoint[1] - currentPoint[1]));
// 将夹角转为锐角
xAngle = xAngle > Math.PI / 2 ? Math.PI - xAngle : xAngle;
yAngle = yAngle > Math.PI / 2 ? Math.PI - yAngle : yAngle;
// 这里不考虑在水平和垂直方向的投影,直接使用最大差值
// 由于上层统一加减了二分之一线宽,这里需要进行弥补
var extra = {
// 水平方向投影
xExtra: Math.cos(currentAngle / 2 - xAngle) * ((lineWidth / 2) * (1 / Math.sin(currentAngle / 2))) - lineWidth / 2 || 0,
// 垂直方向投影
yExtra: Math.cos(yAngle - currentAngle / 2) * ((lineWidth / 2) * (1 / Math.sin(currentAngle / 2))) - lineWidth / 2 || 0,
};
return extra;
}
export default function (shape) {
var attrs = shape.attr();
var path = attrs.path, stroke = attrs.stroke;
var lineWidth = stroke ? attrs.lineWidth : 0; // 只有有 stroke 时,lineWidth 才生效
var segments = shape.get('segments') || path2Segments(path);
var _a = getPathBox(segments, lineWidth), x = _a.x, y = _a.y, width = _a.width, height = _a.height;
var bbox = {
minX: x,
minY: y,
maxX: x + width,
maxY: y + height,
};
bbox = mergeArrowBBox(shape, bbox);
return {
x: bbox.minX,
y: bbox.minY,
width: bbox.maxX - bbox.minX,
height: bbox.maxY - bbox.minY,
};
}
//# sourceMappingURL=path.js.map