@antv/g2
Version:
the Grammar of Graphics in Javascript
46 lines (40 loc) • 1.21 kB
text/typescript
import { path as d3path } from 'd3-path';
import { applyStyle } from '../utils';
import { select } from '../../utils/selection';
import { ShapeComponent as SC } from '../../runtime';
export type SmoothOptions = Record<string, any>;
/**
* Connect 2 points with a smooth line, used in tree.
*/
export const Smooth: SC<SmoothOptions> = (options, context) => {
const { ...style } = options;
const { document } = context;
return (points, value, defaults) => {
const { color: defaultColor, ...rest } = defaults;
const { color = defaultColor, transform } = value;
const [from, to] = points;
const path = d3path();
path.moveTo(from[0], from[1]);
path.bezierCurveTo(
from[0] / 2 + to[0] / 2,
from[1],
from[0] / 2 + to[0] / 2,
to[1],
to[0],
to[1],
);
return select(document.createElement('path', {}))
.call(applyStyle, rest)
.style('d', path.toString())
.style('stroke', color)
.style('transform', transform)
.call(applyStyle, style)
.node();
};
};
Smooth.props = {
defaultMarker: 'smooth',
defaultEnterAnimation: 'fadeIn',
defaultUpdateAnimation: 'morphing',
defaultExitAnimation: 'fadeOut',
};