ai-chatbot/services/pascal-editor/packages/plugin-trees/src/geometry.ts

// ez-tree loads its inlined textures at module scope (needs `document`), so
// this module must only be imported from lazy client modules (renderers,
// systems, tools, previews) — never from `index.ts`, a definition, or
// `floorplan.ts`, or SSR/prerender crashes. Pure helpers shared with the
// flower/grass builders live in `variant-utils.ts` for that reason.
import { Tree } from '@dgreenheck/ez-tree'
import type { BufferGeometry, Material, Mesh, Object3D } from 'three'
import { ezPresetOf } from './presets'
import type { TreeNode } from './schema'
import { naturalHeight } from './variant-utils'
import { toWindMaterial } from './wind-node'

/** The geometry-affecting fields of a tree. Two trees with the same spec share
 * one generated variant (and thus one InstancedMesh set). Per-instance fields
 * (position/rotation/height) are deliberately NOT here — they're cheap matrix
 * work, not geometry. */
export type TreeSpec = Pick<
  TreeNode,
  | 'preset'
  | 'size'
  | 'treeType'
  | 'seed'
  | 'foliageDensity'
  | 'trunkThickness'
  | 'leafless'
  | 'leafColor'
  | 'branchColor'
>

export function treeSpecOf(node: TreeNode): TreeSpec {
  // Default the non-override fields (nodes persisted before a field existed load
  // without it). The four overrides are left as-is — `undefined` means "inherit
  // the ez-tree preset" (its own seed/type/tints), resolved in `generateTree`.
  return {
    preset: node.preset ?? 'oak',
    size: node.size ?? 'medium',
    treeType: node.treeType,
    seed: node.seed,
    foliageDensity: node.foliageDensity ?? 1,
    trunkThickness: node.trunkThickness ?? 1,
    leafless: node.leafless ?? false,
    leafColor: node.leafColor,
    branchColor: node.branchColor,
  }
}

/** Stable variant id. Trees with the same key share one set of InstancedMeshes. */
export function treeVariantKey(spec: TreeSpec): string {
  return [
    spec.preset,
    spec.size,
    spec.treeType,
    spec.seed,
    spec.foliageDensity,
    spec.trunkThickness,
    spec.leafless,
    spec.leafColor,
    spec.branchColor,
  ].join(':')
}

/** `#rrggbb` → 0xrrggbb, defaulting to white on anything missing/unparseable. */
function hexToInt(hex: string | undefined): number {
  const n = Number.parseInt((hex ?? '').replace('#', ''), 16)
  return Number.isFinite(n) ? n : 0xffffff
}

/**
 * Generate an ez-tree for a spec. ez-tree's `Tree` is a `THREE.Group`; textures
 * are inlined in the library (no asset hosting). `loadPreset` owns the full look
 * (seed, growth model, tints, branch/leaf structure); the curated params then
 * apply *on top* — but only where the node actually set them, so an unset field
 * keeps the preset's value (its canonical silhouette/colours). `trunkThickness`
 * and `foliageDensity` are multipliers (1 = preset default). Pure given its
 * inputs — same spec ⇒ same tree — which lets the renderer cache one generation
 * per variant.
 */
export function generateTree(spec: TreeSpec): Tree {
  const tree = new Tree()
  tree.loadPreset(ezPresetOf(spec.preset, spec.size))
  if (spec.seed != null) tree.options.seed = spec.seed
  if (spec.treeType != null) (tree.options as { type: string }).type = spec.treeType

  const radius = tree.options.branch.radius as unknown as Record<string, number>
  for (const level of Object.keys(radius)) {
    const value = radius[level]
    if (value !== undefined) radius[level] = value * spec.trunkThickness
  }

  const leaves = tree.options.leaves as { count: number; tint: number }
  leaves.count = spec.leafless ? 0 : Math.round(leaves.count * spec.foliageDensity)
  if (spec.leafColor != null) leaves.tint = hexToInt(spec.leafColor)
  if (spec.branchColor != null)
    (tree.options.bark as { tint: number }).tint = hexToInt(spec.branchColor)

  tree.generate()
  return tree
}

/** A renderable sub-mesh of a tree: geometry (baked into tree-local space) +
 * its material. The instanced renderer builds one InstancedMesh per sub-mesh
 * per variant. */
export type TreeSubMesh = { geometry: BufferGeometry; material: Material | Material[] }

/** Geometry + height for one tree variant, generated once and shared across
 * every instance of that spec. */
export type TreeVariantData = { subMeshes: TreeSubMesh[]; naturalHeight: number }

const variantCache = new Map<string, TreeVariantData>()

/**
 * Cached geometry for a spec. ez-tree's `generate()` is heavy, so it runs once
 * per variant; the resulting geometries/materials are retained here and shared
 * by every instance. The renderer must NOT dispose them (it sets `dispose={null}`
 * on the InstancedMesh).
 */
export function getVariantData(spec: TreeSpec): TreeVariantData {
  const key = treeVariantKey(spec)
  const cached = variantCache.get(key)
  if (cached) return cached
  const tree = generateTree(spec)
  const data: TreeVariantData = {
    subMeshes: extractSubMeshes(tree),
    naturalHeight: naturalHeight(tree),
  }
  variantCache.set(key, data)
  return data
}

/** Extract the leaf/bark sub-meshes, baking each mesh's local transform into a
 * cloned geometry so instance matrices only carry the node's own transform. */
export function extractSubMeshes(tree: Object3D): TreeSubMesh[] {
  const out: TreeSubMesh[] = []
  tree.traverse((child) => {
    const mesh = child as Partial<Mesh>
    if (mesh.isMesh && mesh.geometry && mesh.material) {
      const geometry = mesh.geometry.clone()
      ;(child as Mesh).updateMatrix()
      geometry.applyMatrix4((child as Mesh).matrix)
      // Swap ez-tree's plain materials for swaying node materials (WebGPU/TSL).
      const material = Array.isArray(mesh.material)
        ? mesh.material.map(toWindMaterial)
        : toWindMaterial(mesh.material)
      out.push({ geometry, material })
    }
  })
  return out
}