From 267cf2c11f71b274a347378932b7ca57309397d5 Mon Sep 17 00:00:00 2001
From: btzr-io <btzr.io@gmail.com>
Date: Wed, 15 Aug 2018 00:15:15 -0600
Subject: [PATCH] update gui controls

---
 .../component/viewers/threeViewer/index.jsx   | 32 +++++++++++++++----
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/src/renderer/component/viewers/threeViewer/index.jsx b/src/renderer/component/viewers/threeViewer/index.jsx
index 02a1405e9..ab9903785 100644
--- a/src/renderer/component/viewers/threeViewer/index.jsx
+++ b/src/renderer/component/viewers/threeViewer/index.jsx
@@ -136,6 +136,8 @@ class ThreeViewer extends React.PureComponent<Props, State> {
           if (child.material) child.material.dispose();
         }
       });
+      // Clean up controls
+      if (this.controls) this.controls.dispose();
       // It's unclear if we need this:
       if (this.renderer) {
         this.renderer.renderLists.dispose();
@@ -157,25 +159,36 @@ class ThreeViewer extends React.PureComponent<Props, State> {
 
   transformGroup(group) {
     ThreeViewer.fitMeshToCamera(group);
-    this.updateControlsTarget(group.position);
+    this.updateControlsTarget(this.mesh.position);
   }
 
   createInterfaceControls() {
     if (this.guiContainer && this.mesh) {
+      this.gui = new dat.GUI({ autoPlace: false, name: 'controls' });
+
       const config = {
         color: this.materialColor,
-        wireframe: false,
       };
 
-      this.gui = new dat.GUI({ autoPlace: false });
+      config.reset = () => {
+        // Reset material color
+        config.color = this.materialColor;
+        this.material.color.set(config.color);
+        // Reset wireframe
+        this.material.wireframe = false;
+        // Reset camera
+        this.restoreCamera();
+      };
 
-      const colorPicker = this.gui.addColor(config, 'color');
+      // Color picker
+      const colorPicker = this.gui.addColor(config, 'color').listen();
 
       colorPicker.onChange(color => {
         this.material.color.set(color);
       });
 
       this.gui.add(this.material, 'wireframe').listen();
+      this.gui.add(config, 'reset');
       this.guiContainer.current.appendChild(this.gui.domElement);
     }
   }
@@ -187,10 +200,11 @@ class ThreeViewer extends React.PureComponent<Props, State> {
     controls.enableDamping = true;
     controls.dampingFactor = 0.75;
     controls.enableZoom = true;
-    controls.minDistance = 1;
-    controls.maxDistance = 50;
+    controls.minDistance = 5;
+    controls.maxDistance = 14;
     controls.autoRotate = autoRotate;
     controls.enablePan = false;
+    controls.saveState();
     return controls;
   }
 
@@ -245,6 +259,12 @@ class ThreeViewer extends React.PureComponent<Props, State> {
     this.controls.update();
   }
 
+  restoreCamera() {
+    this.controls.reset();
+    this.camera.position.set(-9.5, 14, 11);
+    this.updateControlsTarget(this.mesh.position);
+  }
+
   renderStl(data) {
     this.createGeometry(data);
     this.mesh = new THREE.Mesh(this.geometry, this.material);