import * as THREE from 'three';
import {RoomEnvironment} from 'three/addons/environments/RoomEnvironment.js';
import {bodyMatrices,controllableJoints,homeProgress} from './vendor/kinematics.js';

export const STUDIO_DIRECTION=[1,0.65,1.4] as const;

export type StudioLighting={exposure?:number;environment?:number;hemisphere?:number;key?:number};
/** Shared studio, with optional batch lighting tuned for thumbnails on a pale canvas. */
export function createStudio(renderer:THREE.WebGLRenderer,lighting:StudioLighting={}){
  renderer.setClearColor(0,0);
  renderer.outputColorSpace=THREE.SRGBColorSpace;
  renderer.toneMapping=THREE.ACESFilmicToneMapping;
  renderer.toneMappingExposure=lighting.exposure??1.35;
  const scene=new THREE.Scene();
  const pmrem=new THREE.PMREMGenerator(renderer),room=new RoomEnvironment();
  const environment=pmrem.fromScene(room,0.04);
  scene.environment=environment.texture;
  scene.environmentIntensity=lighting.environment??1;
  room.dispose();pmrem.dispose();
  scene.add(new THREE.HemisphereLight(0xffffff,0xdce0ef,lighting.hemisphere??2));
  const light=new THREE.DirectionalLight(0xffffff,lighting.key??3);
  light.position.set(3,5,4);scene.add(light);
  return {scene,dispose:()=>environment.dispose()};
}

export function setHomePose(model:any):Record<string,number>{
  const values:Record<string,number>={};
  for(const joint of controllableJoints(model.articulation))values[joint.id]=homeProgress(joint);
  if(model.articulation)for(const [id,matrix] of Object.entries(bodyMatrices(model.articulation,values))){
    const body=model.bodies.get(id);
    if(body){body.matrixAutoUpdate=false;body.matrix.set(...matrix as number[]);body.matrixWorldNeedsUpdate=true;}
  }
  model.scene.updateMatrixWorld(true);
  return values;
}

export function objectBounds(object:THREE.Object3D){
  object.updateMatrixWorld(true);
  const box=new THREE.Box3().setFromObject(object);
  const size=box.getSize(new THREE.Vector3()),center=box.getCenter(new THREE.Vector3());
  if(box.isEmpty()||![...size,...center].every(Number.isFinite))throw new Error('Empty or non-finite model bounds.');
  return {box,center,radius:Math.max(size.length()/2,0.01)};
}

/** Fits all eight world bounds corners to an orthographic square, including thin objects. */
export function thumbnailCamera(object:THREE.Object3D,padding=1.12){
  const {box,center,radius}=objectBounds(object);
  const camera=new THREE.OrthographicCamera(-1,1,1,-1,radius/1000,radius*20);
  camera.position.copy(center).add(new THREE.Vector3(...STUDIO_DIRECTION).normalize().multiplyScalar(radius*4));
  camera.lookAt(center);camera.updateMatrixWorld(true);
  const projected=new THREE.Box3();
  for(const x of [box.min.x,box.max.x])for(const y of [box.min.y,box.max.y])for(const z of [box.min.z,box.max.z]){
    projected.expandByPoint(new THREE.Vector3(x,y,z).applyMatrix4(camera.matrixWorldInverse));
  }
  const middle=projected.getCenter(new THREE.Vector3());
  const extent=Math.max(projected.max.x-projected.min.x,projected.max.y-projected.min.y,radius/100)*padding/2;
  camera.left=middle.x-extent;camera.right=middle.x+extent;
  camera.bottom=middle.y-extent;camera.top=middle.y+extent;
  camera.updateProjectionMatrix();
  return camera;
}