99 lines
2.4 KiB
TypeScript
99 lines
2.4 KiB
TypeScript
|
import type {IShorthandCompoundProps} from '@theatre/core'
|
||
|
|
import {types} from '@theatre/core'
|
||
|
|
import type {Object3D} from 'three'
|
||
|
|
import type {IconID} from './icons'
|
||
|
|
|
||
|
|
export type Helper = Object3D & {
|
||
|
|
update?: () => void
|
||
|
|
}
|
||
|
|
type PropConfig<T> = {
|
||
|
|
parse: (props: Record<string, any>) => T
|
||
|
|
apply: (value: T, object: any) => void
|
||
|
|
type: IShorthandCompoundProps
|
||
|
|
}
|
||
|
|
type Props = Record<string, PropConfig<any>>
|
||
|
|
type Meta<T> = {
|
||
|
|
useTransformControls: boolean
|
||
|
|
updateObject?: (object: T) => void
|
||
|
|
icon: IconID
|
||
|
|
dimensionless?: boolean
|
||
|
|
createHelper: (object: T) => Helper
|
||
|
|
}
|
||
|
|
export type ObjectConfig<T> = {props: Props} & Meta<T>
|
||
|
|
export type EditableFactoryConfig = Partial<
|
||
|
|
Record<keyof JSX.IntrinsicElements, ObjectConfig<any>>
|
||
|
|
>
|
||
|
|
|
||
|
|
type Vector3 = {
|
||
|
|
x: number
|
||
|
|
y: number
|
||
|
|
z: number
|
||
|
|
}
|
||
|
|
|
||
|
|
export const createVector = (components?: [number, number, number]) => {
|
||
|
|
return components
|
||
|
|
? {x: components[0], y: components[1], z: components[2]}
|
||
|
|
: {
|
||
|
|
x: 0,
|
||
|
|
y: 0,
|
||
|
|
z: 0,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export const createVectorPropConfig = (
|
||
|
|
key: string,
|
||
|
|
defaultValue = createVector(),
|
||
|
|
{nudgeMultiplier = 0.01} = {},
|
||
|
|
): PropConfig<Vector3> => ({
|
||
|
|
parse: (props) => {
|
||
|
|
const vector = props[key]
|
||
|
|
? Array.isArray(props[key])
|
||
|
|
? createVector(props[key] as any)
|
||
|
|
: {
|
||
|
|
x: props[key].x,
|
||
|
|
y: props[key].y,
|
||
|
|
z: props[key].z,
|
||
|
|
}
|
||
|
|
: defaultValue
|
||
|
|
;(['x', 'y', 'z'] as const).forEach((axis) => {
|
||
|
|
if (props[`${key}-${axis}` as any])
|
||
|
|
vector[axis] = props[`${key}-${axis}` as any]
|
||
|
|
})
|
||
|
|
return vector
|
||
|
|
},
|
||
|
|
apply: (value, object) => {
|
||
|
|
object[key].set(value.x, value.y, value.z)
|
||
|
|
},
|
||
|
|
type: {
|
||
|
|
[key]: {
|
||
|
|
x: types.number(defaultValue.x, {nudgeMultiplier}),
|
||
|
|
y: types.number(defaultValue.y, {nudgeMultiplier}),
|
||
|
|
z: types.number(defaultValue.z, {nudgeMultiplier}),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
})
|
||
|
|
|
||
|
|
export const createNumberPropConfig = (
|
||
|
|
key: string,
|
||
|
|
defaultValue: number = 0,
|
||
|
|
{nudgeMultiplier = 0.01} = {},
|
||
|
|
): PropConfig<number> => ({
|
||
|
|
parse: (props) => {
|
||
|
|
return props[key] ?? defaultValue ?? 0
|
||
|
|
},
|
||
|
|
apply: (value, object) => {
|
||
|
|
object[key] = value
|
||
|
|
},
|
||
|
|
type: {
|
||
|
|
[key]: types.number(defaultValue, {nudgeMultiplier}),
|
||
|
|
},
|
||
|
|
})
|
||
|
|
|
||
|
|
export const extendObjectProps = <T extends {props: {}}>(
|
||
|
|
objectConfig: T,
|
||
|
|
extension: Props,
|
||
|
|
) => ({
|
||
|
|
...objectConfig,
|
||
|
|
props: {...objectConfig.props, ...extension},
|
||
|
|
})
|