import {DiagramEntity, DiagramEntityType} from "../types/diagramEntity"; import {AbstractMesh, InstancedMesh, Mesh, Quaternion, Scene, Vector3} from "@babylonjs/core"; import {DiagramConnection} from "../diagramConnection"; import {updateTextNode} from "../../util/functions/updateTextNode"; import log from "loglevel"; import {v4 as uuidv4} from 'uuid'; import {buildStandardMaterial} from "../../materials/functions/buildStandardMaterial"; export function buildMeshFromDiagramEntity(entity: DiagramEntity, scene: Scene): AbstractMesh { const logger = log.getLogger('buildMeshFromDiagramEntity'); if (!entity) { logger.error("buildMeshFromDiagramEntity: entity is null"); return null; } switch (entity.type) { case DiagramEntityType.USER: logger.debug("buildMeshFromDiagramEntity: entity is user"); break; default: } generateId(entity); const newMesh: AbstractMesh = createNewInstanceIfNecessary(entity, scene); return mapMetadata(entity, newMesh, scene); } function createNewInstanceIfNecessary(entity: DiagramEntity, scene: Scene): AbstractMesh { const logger = log.getLogger('createNewInstanceIfNecessary'); const oldMesh: AbstractMesh = scene.getMeshById(entity.id); let newMesh: AbstractMesh; if (oldMesh) { logger.debug(`mesh ${oldMesh.id} already exists`); newMesh = oldMesh; } else { if (entity.template == "#connection-template") { const connection: DiagramConnection = new DiagramConnection(entity.from, entity.to, entity.id, scene); logger.debug(`connection.mesh = ${connection.mesh.id}`); newMesh = connection.mesh; } else { const toolMesh = scene.getMeshById("tool-" + entity.template + "-" + entity.color); if (toolMesh && !oldMesh) { newMesh = new InstancedMesh(entity.id, (toolMesh as Mesh)); newMesh.metadata = {template: entity.template, exportable: true, tool: false}; } else { logger.warn('no tool mesh found for ' + entity.template + "-" + entity.color); } } } return newMesh; } function generateId(entity: DiagramEntity) { if (!entity.id) { entity.id = "id" + uuidv4(); } } function mapMetadata(entity: DiagramEntity, newMesh: AbstractMesh, scene: Scene): AbstractMesh { const logger = log.getLogger('mapMetaqdata'); if (newMesh) { if (!newMesh.metadata) { newMesh.metadata = {}; } if (entity.position) { newMesh.position = xyztovec(entity.position); } if (entity.rotation) { if (newMesh.rotationQuaternion) { newMesh.rotationQuaternion = Quaternion.FromEulerAngles(entity.rotation.x, entity.rotation.y, entity.rotation.z); } else { newMesh.rotation = xyztovec(entity.rotation); } } if (entity.parent) { const parent_node = scene.getNodeById(entity.parent); if (parent_node) { newMesh.parent = parent_node; newMesh.metadata.parent = entity.parent; } } if (entity.scale) { newMesh.scaling = xyztovec(entity.scale); } if (!newMesh.material && newMesh?.metadata?.template != "#object-template") { logger.warn("new material created, this shouldn't happen"); newMesh.material = buildStandardMaterial("material-" + entity.id, scene, entity.color); } if (entity.text) { newMesh.metadata.text = entity.text; updateTextNode(newMesh, entity.text); } if (entity.from) { newMesh.metadata.from = entity.from; } if (entity.to) { newMesh.metadata.to = entity.to; } } else { logger.error("buildMeshFromDiagramEntity: mesh is null after it should have been created"); } return newMesh; } function xyztovec(xyz: { x, y, z }): Vector3 { return new Vector3(xyz.x, xyz.y, xyz.z); }