From 3804ae53768678031e16e54c9d6d94857adca6b3 Mon Sep 17 00:00:00 2001
From: Yeicor <4929005+Yeicor@users.noreply.github.com>
Date: Sat, 10 Feb 2024 19:56:47 +0100
Subject: [PATCH] add support for orthographic camera!

---
 src/App.vue              |  4 +---
 src/OrientationGizmo.vue |  4 ++++
 src/Tools.vue            | 36 ++++++++++++++++++++++++++++++++++--
 3 files changed, 39 insertions(+), 5 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index 57783ec..0763022 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -63,10 +63,8 @@ let modelViewerInfo: Ref<typeof ModelViewerInfo | null> = ref(null);
 
 <!--suppress CssUnusedSymbol -->
 <style>
-html, body, #main {
+html, body {
   height: 100%;
-  margin: 0;
-  padding: 0;
   overflow: hidden;
 }
 </style>
diff --git a/src/OrientationGizmo.vue b/src/OrientationGizmo.vue
index 58f68a7..526d489 100644
--- a/src/OrientationGizmo.vue
+++ b/src/OrientationGizmo.vue
@@ -40,6 +40,10 @@ function createGizmo(expectedParent: HTMLElement, scene: ModelScene): HTMLElemen
     //console.log("New camera position", newLookFrom)
     scene.getCamera().position.copy(newLookFrom);
     scene.getCamera().lookAt(lookAt);
+    if ((scene as any).__perspectiveCamera) { // HACK: Make the hacky ortho also work
+      (scene as any).__perspectiveCamera.position.copy(newLookFrom);
+      (scene as any).__perspectiveCamera.lookAt(lookAt);
+    }
     scene.queueRender();
   }
   return gizmo;
diff --git a/src/Tools.vue b/src/Tools.vue
index d3b54ac..5e15a54 100644
--- a/src/Tools.vue
+++ b/src/Tools.vue
@@ -1,21 +1,53 @@
 <script setup lang="ts">
 import {VBtn, VIcon} from "vuetify/lib/components";
+import {ref} from "vue";
 import OrientationGizmo from "./OrientationGizmo.vue";
+import type {ModelScene} from "@google/model-viewer/lib/three-components/ModelScene";
+import {OrthographicCamera} from "three";
 
 const props = defineProps({
   modelViewerInfo: Object
 });
 
+
+function syncOrthoCamera(force: boolean) {
+  if (!props.modelViewerInfo) return;
+  let scene: ModelScene = props.modelViewerInfo.scene
+  let perspectiveCam = (scene as any).__perspectiveCamera;
+  if (force || perspectiveCam && scene.camera != perspectiveCam) {
+    // Get zoom level from perspective camera
+    let dist = scene.getTarget().distanceToSquared(perspectiveCam.position);
+    let w = scene.aspect * dist ** 1.1 / 4000;
+    let h = dist ** 1.1 / 4000;
+    (scene as any).camera = new OrthographicCamera(-w, w, h, -h, perspectiveCam.near, perspectiveCam.far);
+    scene.camera.position.copy(perspectiveCam.position);
+    scene.camera.lookAt(scene.getTarget().clone().add(scene.target.position));
+    requestAnimationFrame(() => syncOrthoCamera(false));
+  }
+}
+
+let toggleProjectionText = ref('PERSP'); // Default to perspective camera
 function toggleProjection() {
   if (!props.modelViewerInfo) return;
-  console.log('Toggling projection', props.modelViewerInfo);
+  let scene: ModelScene = props.modelViewerInfo.scene
+  let prevCam = scene.camera;
+  let wasPerspectiveCamera = prevCam.isPerspectiveCamera;
+  if (wasPerspectiveCamera) {
+    (scene as any).__perspectiveCamera = prevCam; // Save the default perspective camera
+    // This hack also needs to sync the camera position and target
+    requestAnimationFrame(() => syncOrthoCamera(true));
+  } else {
+    // Restore the default perspective camera
+    scene.camera = (scene as any).__perspectiveCamera;
+  }
+  toggleProjectionText.value = wasPerspectiveCamera ? 'ORTHO' : 'PERSP';
 }
 
 </script>
 
 <template>
   <orientation-gizmo v-if="props.modelViewerInfo" :scene="props.modelViewerInfo.scene"/>
-  <v-btn icon="mdi-projector" @click="toggleProjection"><span class="icon-detail">PERSP</span>
+  <v-btn icon="mdi-projector" @click="toggleProjection"><span class="icon-detail">{{ toggleProjectionText }}</span>
     <v-icon icon="mdi-projector"></v-icon>
   </v-btn>
 </template>