tldraw/state/shape-utils/ray.tsx
2021-07-09 20:52:08 +01:00

94 lines
2.1 KiB
TypeScript

import { getFromCache, uniqueId } from 'utils/utils'
import vec from 'utils/vec'
import { RayShape, ShapeType } from 'types'
import { intersectCircleBounds } from 'utils/intersections'
import { ThinLine } from 'components/canvas/misc'
import { translateBounds, boundsContained } from 'utils'
import { defaultStyle, getShapeStyle } from 'state/shape-styles'
import { registerShapeUtils } from './register'
const ray = registerShapeUtils<RayShape>({
boundsCache: new WeakMap([]),
defaultProps: {
id: uniqueId(),
type: ShapeType.Ray,
name: 'Ray',
parentId: 'page1',
childIndex: 0,
point: [0, 0],
direction: [0, 1],
rotation: 0,
style: defaultStyle,
},
shouldRender(shape, prev) {
return shape.direction !== prev.direction || shape.style !== prev.style
},
render(shape) {
const { direction } = shape
const styles = getShapeStyle(shape.style)
const [x2, y2] = vec.add([0, 0], vec.mul(direction, 10000))
return (
<>
<ThinLine x1={0} y1={0} x2={x2} y2={y2} stroke={styles.stroke} />
<circle r={4} fill="transparent" />
<use href="#dot" />
</>
)
},
getRotatedBounds(shape) {
return this.getBounds(shape)
},
getBounds(shape) {
const bounds = getFromCache(this.boundsCache, shape, (cache) => {
cache.set(shape, {
minX: 0,
maxX: 1,
minY: 0,
maxY: 1,
width: 1,
height: 1,
})
})
return translateBounds(bounds, shape.point)
},
getCenter(shape) {
return shape.point
},
hitTest() {
return true
},
hitTestBounds(this, shape, brushBounds) {
const shapeBounds = this.getBounds(shape)
return (
boundsContained(shapeBounds, brushBounds) ||
intersectCircleBounds(shape.point, 4, brushBounds).length > 0
)
},
transform(shape, bounds) {
shape.point = [bounds.minX, bounds.minY]
return this
},
transformSingle(shape, bounds, info) {
return this.transform(shape, bounds, info)
},
canTransform: false,
canChangeAspectRatio: false,
canStyleFill: false,
})
export default ray