2021-05-28 14:37:23 +00:00
|
|
|
import { v4 as uuid } from 'uuid'
|
|
|
|
import * as vec from 'utils/vec'
|
|
|
|
import { DrawShape, ShapeType } from 'types'
|
|
|
|
import { registerShapeUtils } from './index'
|
|
|
|
import { intersectPolylineBounds } from 'utils/intersections'
|
|
|
|
import { boundsContainPolygon } from 'utils/bounds'
|
|
|
|
import getStroke from 'perfect-freehand'
|
2021-05-27 22:07:27 +00:00
|
|
|
import {
|
|
|
|
getBoundsFromPoints,
|
2021-05-28 16:25:43 +00:00
|
|
|
getRotatedCorners,
|
2021-05-27 22:07:27 +00:00
|
|
|
getSvgPathFromStroke,
|
|
|
|
translateBounds,
|
2021-05-28 14:37:23 +00:00
|
|
|
} from 'utils/utils'
|
2021-05-27 18:51:25 +00:00
|
|
|
|
2021-05-28 14:37:23 +00:00
|
|
|
const pathCache = new WeakMap<number[][], string>([])
|
2021-05-27 17:59:40 +00:00
|
|
|
|
|
|
|
const draw = registerShapeUtils<DrawShape>({
|
|
|
|
boundsCache: new WeakMap([]),
|
|
|
|
|
|
|
|
create(props) {
|
|
|
|
return {
|
|
|
|
id: uuid(),
|
|
|
|
type: ShapeType.Draw,
|
|
|
|
isGenerated: false,
|
2021-05-28 14:37:23 +00:00
|
|
|
name: 'Draw',
|
|
|
|
parentId: 'page0',
|
2021-05-27 17:59:40 +00:00
|
|
|
childIndex: 0,
|
|
|
|
point: [0, 0],
|
|
|
|
points: [[0, 0]],
|
|
|
|
rotation: 0,
|
|
|
|
...props,
|
|
|
|
style: {
|
|
|
|
strokeWidth: 2,
|
2021-05-28 14:37:23 +00:00
|
|
|
strokeLinecap: 'round',
|
|
|
|
strokeLinejoin: 'round',
|
2021-05-27 17:59:40 +00:00
|
|
|
...props.style,
|
2021-05-28 20:30:27 +00:00
|
|
|
fill: props.style.stroke,
|
2021-05-27 17:59:40 +00:00
|
|
|
},
|
|
|
|
}
|
|
|
|
},
|
|
|
|
|
2021-05-27 18:51:25 +00:00
|
|
|
render(shape) {
|
2021-05-28 20:30:27 +00:00
|
|
|
const { id, point, points, style } = shape
|
2021-05-27 18:51:25 +00:00
|
|
|
|
2021-05-28 14:37:23 +00:00
|
|
|
if (!pathCache.has(points)) {
|
2021-05-28 16:25:43 +00:00
|
|
|
if (points.length < 2) {
|
|
|
|
const left = vec.add(point, [6, 0])
|
|
|
|
let d: number[][] = []
|
|
|
|
for (let i = 0; i < 10; i++) {
|
|
|
|
d.push(vec.rotWith(left, point, i * ((Math.PI * 2) / 8)))
|
|
|
|
}
|
|
|
|
pathCache.set(points, getSvgPathFromStroke(d))
|
|
|
|
} else {
|
2021-05-28 20:30:27 +00:00
|
|
|
pathCache.set(
|
|
|
|
points,
|
|
|
|
getSvgPathFromStroke(
|
|
|
|
getStroke(points, { size: Number(style.strokeWidth) * 2 })
|
|
|
|
)
|
|
|
|
)
|
2021-05-28 16:25:43 +00:00
|
|
|
}
|
2021-05-27 18:51:25 +00:00
|
|
|
}
|
|
|
|
|
2021-05-28 14:37:23 +00:00
|
|
|
return <path id={id} d={pathCache.get(points)} />
|
2021-05-27 17:59:40 +00:00
|
|
|
},
|
|
|
|
|
|
|
|
applyStyles(shape, style) {
|
|
|
|
Object.assign(shape.style, style)
|
2021-05-28 20:30:27 +00:00
|
|
|
shape.style.fill = shape.style.stroke
|
2021-05-27 17:59:40 +00:00
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
getBounds(shape) {
|
|
|
|
if (!this.boundsCache.has(shape)) {
|
|
|
|
const bounds = getBoundsFromPoints(shape.points)
|
|
|
|
this.boundsCache.set(shape, bounds)
|
|
|
|
}
|
|
|
|
|
|
|
|
return translateBounds(this.boundsCache.get(shape), shape.point)
|
|
|
|
},
|
|
|
|
|
|
|
|
getRotatedBounds(shape) {
|
2021-05-28 16:25:43 +00:00
|
|
|
return getBoundsFromPoints(
|
|
|
|
getRotatedCorners(this.getBounds(shape), shape.rotation)
|
|
|
|
)
|
2021-05-27 17:59:40 +00:00
|
|
|
},
|
|
|
|
|
|
|
|
getCenter(shape) {
|
2021-05-28 16:25:43 +00:00
|
|
|
const bounds = this.getRotatedBounds(shape)
|
2021-05-27 17:59:40 +00:00
|
|
|
return [bounds.minX + bounds.width / 2, bounds.minY + bounds.height / 2]
|
|
|
|
},
|
|
|
|
|
|
|
|
hitTest(shape, point) {
|
|
|
|
let pt = vec.sub(point, shape.point)
|
|
|
|
let prev = shape.points[0]
|
|
|
|
|
|
|
|
for (let i = 1; i < shape.points.length; i++) {
|
|
|
|
let curr = shape.points[i]
|
|
|
|
if (vec.distanceToLineSegment(prev, curr, pt) < 4) {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
prev = curr
|
|
|
|
}
|
|
|
|
|
|
|
|
return false
|
|
|
|
},
|
|
|
|
|
|
|
|
hitTestBounds(this, shape, brushBounds) {
|
|
|
|
const b = this.getBounds(shape)
|
|
|
|
const center = [b.minX + b.width / 2, b.minY + b.height / 2]
|
|
|
|
|
|
|
|
const rotatedCorners = [
|
|
|
|
[b.minX, b.minY],
|
|
|
|
[b.maxX, b.minY],
|
|
|
|
[b.maxX, b.maxY],
|
|
|
|
[b.minX, b.maxY],
|
|
|
|
].map((point) => vec.rotWith(point, center, shape.rotation))
|
|
|
|
|
|
|
|
return (
|
|
|
|
boundsContainPolygon(brushBounds, rotatedCorners) ||
|
|
|
|
intersectPolylineBounds(
|
|
|
|
shape.points.map((point) => vec.add(point, shape.point)),
|
|
|
|
brushBounds
|
|
|
|
).length > 0
|
|
|
|
)
|
|
|
|
},
|
|
|
|
|
|
|
|
rotateTo(shape, rotation) {
|
|
|
|
shape.rotation = rotation
|
2021-05-28 16:25:43 +00:00
|
|
|
// console.log(shape.points.map(([x, y]) => [x, y]))
|
|
|
|
// const bounds = this.getBounds(shape)
|
|
|
|
// const center = [bounds.width / 2, bounds.height / 2]
|
|
|
|
// shape.points = shape.points.map((pt) => vec.rotWith(pt, center, rotation))
|
2021-05-27 17:59:40 +00:00
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
translateTo(shape, point) {
|
2021-05-28 20:30:27 +00:00
|
|
|
shape.point = vec.toPrecision(point)
|
2021-05-27 17:59:40 +00:00
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
transform(shape, bounds, { initialShape, scaleX, scaleY }) {
|
|
|
|
const initialShapeBounds = this.boundsCache.get(initialShape)
|
|
|
|
shape.points = initialShape.points.map(([x, y]) => {
|
|
|
|
return [
|
|
|
|
bounds.width *
|
|
|
|
(scaleX < 0
|
|
|
|
? 1 - x / initialShapeBounds.width
|
|
|
|
: x / initialShapeBounds.width),
|
|
|
|
bounds.height *
|
|
|
|
(scaleY < 0
|
|
|
|
? 1 - y / initialShapeBounds.height
|
|
|
|
: y / initialShapeBounds.height),
|
|
|
|
]
|
|
|
|
})
|
|
|
|
|
|
|
|
const newBounds = getBoundsFromPoints(shape.points)
|
|
|
|
|
|
|
|
shape.point = vec.sub(
|
|
|
|
[bounds.minX, bounds.minY],
|
|
|
|
[newBounds.minX, newBounds.minY]
|
|
|
|
)
|
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
transformSingle(shape, bounds, info) {
|
|
|
|
this.transform(shape, bounds, info)
|
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
setParent(shape, parentId) {
|
|
|
|
shape.parentId = parentId
|
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
setChildIndex(shape, childIndex) {
|
|
|
|
shape.childIndex = childIndex
|
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
setPoints(shape, points) {
|
|
|
|
// const bounds = getBoundsFromPoints(points)
|
|
|
|
// const corner = [bounds.minX, bounds.minY]
|
|
|
|
// const nudged = points.map((point) => vec.sub(point, corner))
|
|
|
|
// this.boundsCache.set(shape, translategetBoundsFromPoints(nudged))
|
|
|
|
// shape.point = vec.add(shape.point, corner)
|
|
|
|
|
|
|
|
shape.points = points
|
|
|
|
|
|
|
|
return this
|
|
|
|
},
|
|
|
|
|
|
|
|
canTransform: true,
|
|
|
|
canChangeAspectRatio: true,
|
|
|
|
})
|
|
|
|
|
|
|
|
export default draw
|