X-Git-Url: https://gitweb.ps.run/cloth_sim/blobdiff_plain/fc396e0a9cd744fa4d2c37f0052f395320580cc5..f97f4ee25759ffaa6a4d4709f45fc8b7b5b24973:/Scripts/main.js

diff --git a/Scripts/main.js b/Scripts/main.js
index 843c19a..1a30f7a 100644
--- a/Scripts/main.js
+++ b/Scripts/main.js
@@ -1,135 +1,210 @@
-function init() {
-  class Point {
-    constructor(x, y) {
-      this.x = x;
-      this.y = y;
-    }
-
-    add(that) {
-      return new Point(
-        this.x + that.x,
-        this.y + that.y
-      );
-    }
-
-    sub(that) {
-      return new Point(
-        this.x - that.x,
-        this.y - that.y
-      );
-    }
-
-    dist(that) {
-      let a = this.x - that.x;
-      let b = this.y - that.y;
-      return Math.sqrt(a * a + b * b)
-    }
-  }
-
-  class Cloth {
-    geometry = new THREE.Geometry();
-
-    
-
-    static CreateBasic(width, height, numPointsWidth, numPointsHeight) {
-      let vertices = [];
-      let faces = [];
-
-      let stepWidth = width / (numPointsWidth - 1);
-      let stepHeight = height / (numPointsHeight - 1);
-
-      for (let y = 0; y < numPointsHeight; y++) {
-        for (let x = 0; x < numPointsWidth; x++) {
-          vertices.push(
-            new THREE.Vector3(x * stepWidth, height - y * stepHeight, 0)
-          );
-        }
-      }
-
-      function getVertexIndex(x, y) {
-        return y * numPointsWidth + x;
-      }
-      
-      for (let y = 0; y < numPointsHeight - 1; y++) {
-        for (let x = 0; x < numPointsWidth - 1; x++) {
-          faces.push(
-            new THREE.Face3(
-              getVertexIndex(x, y),
-              getVertexIndex(x, y + 1),
-              getVertexIndex(x + 1, y),
-            )
-          );
-          faces.push(
-            new THREE.Face3(
-              getVertexIndex(x + 1, y),
-              getVertexIndex(x, y + 1),
-              getVertexIndex(x + 1, y + 1),
-            )
-          );
-        }
-      }
-
-      return this.CreateExplicit(vertices, faces);
-    }
 
-    static CreateExplicit(vertices, faces) {
-      let result = new Cloth();
 
-      for (let i in vertices) {
-        result.geometry.vertices.push(vertices[i]);
-      }
-      for (let i in faces) {
-        result.geometry.faces.push(faces[i]);
-      }
+function addLights(scene){
+  scene.add(new THREE.AmbientLight(0x222222));
 
-      result.geometry.computeBoundingSphere();
+  const light1 = new THREE.PointLight(0xffffff, 1, 50);
+  light1.position.set(15, 1, 40);
+  scene.add(light1);
 
-      return result;
-    }
-  }
+  const light2 = new THREE.PointLight(0xffffff, 1, 50);
+  light2.position.set(-15, 0, 40);
+  scene.add(light2);
 
-  let mousePos = new Point();
-
-  const canvasSpace = 200;
+  const light3 = new THREE.PointLight(0xffffff, 1, 50);
+  light3.position.set(0, -1, 40);
+  scene.add(light3);
+}
 
+/**
+ * setup THREE JS Scene, Camera and Renderer
+ */
+function setup_scene(canvasSpace) {
   const scene = new THREE.Scene();
   const camera = new THREE.PerspectiveCamera(75, window.innerWidth / (window.innerHeight - canvasSpace), 0.1, 1000);
-
   const renderer = new THREE.WebGLRenderer();
+  /** size canvas to leave some space for UI */
   renderer.setSize(window.innerWidth, window.innerHeight - canvasSpace);
+  renderer.antialias = true;
+  /** embed canvas in HTML */
   document.getElementById("threejscontainer").appendChild(renderer.domElement);
 
+  /** add orbit controls */
+  const controls = new THREE.OrbitControls(camera, renderer.domElement);
+  controls.target.set(0, 0, 0);
+  controls.update();
+
+  /** add scene background */
+  const loader = new THREE.TextureLoader();
+  const texture = loader.load(
+    'Textures/tears_of_steel_bridge_2k.jpg',
+    () => {
+      const rt = new THREE.WebGLCubeRenderTarget(texture.image.height);
+      rt.fromEquirectangularTexture(renderer, texture);
+      scene.background = rt;
+    });
+
+  /** add flag pole */
+  const geometry = new THREE.CylinderGeometry(0.02, 0.02, 5, 32);
+  const material = new THREE.MeshStandardMaterial({color: 0xffffff});
+  const cylinder = new THREE.Mesh(geometry, material);
+  cylinder.position.set(-0.5, -2.25, 0);
+  scene.add(cylinder);
+
+  /** add global light */
   const directionalLight = new THREE.DirectionalLight(0xffffff, 1);
   scene.add(directionalLight);
 
-  let cloth = Cloth.CreateBasic(10, 10, 4, 4);
-  const material = new THREE.MeshBasicMaterial({ color: 0x00ff00 });
-  const mesh = new THREE.Mesh(cloth.geometry, material);
-  scene.add(mesh);
+  /** position camera */
+  camera.position.z = 2;
+  addLights(scene);
+  return [scene, camera, renderer];
+}
+
+/** call "init" when document is fully loaded */
+document.body.onload = init;
+
+function init() {
+  let mousePos = new THREE.Vector2();
+  
+  /**
+   * Space left empty under canvas
+   * for UI elements
+   */
+  const canvasSpace = 200;
 
-  camera.position.y = 5;
-  camera.position.z = 10;
+  /** Constant Frame Time */
+  const frameTime = 1000.0 / 200.0;
+
+  /** Setup scene */
+  let [scene, camera, renderer] = setup_scene(canvasSpace);
+  
+  const loader = new THREE.TextureLoader();
+  //Red color: 0xC70039
+
+  /** Create cloth and generate mesh */
+  const cloth = new Cloth(1, 0.5, 20, 20);
+  const clothGeometry = cloth.generateGeometry();
+  const clothMaterial = new THREE.MeshStandardMaterial({ map: loader.load('Textures/hsrm2.png'), color: 0xffffff, side: THREE.DoubleSide, flatShading: false });
+  //const clothMaterial = new THREE.MeshStandardMaterial({ color: 0xC70039, side: THREE.DoubleSide, flatShading: false });
+  const clothMesh = new THREE.Mesh(clothGeometry, clothMaterial);
+  scene.add(clothMesh);
+  
+  /** Register wind checkbox */
+  document.getElementById("windToggle").checked = options.wind;
+  document.getElementById("windToggle").onchange = (e) => {
+    options.wind = e.target.checked;
+  };
 
+  let raycaster = new THREE.Raycaster();
+  let intersects;
+  let windKeyDown = false;
+  let dragKeyDown = false;
+  let draggedIndex = -1;
+  /**
+   * function called every frame
+   * @param {number} dt - time passed since last frame in ms
+   */
   function animate(dt) {
-    requestAnimationFrame(animate);
+    /** run simulation step */
+    cloth.simulate(dt / 1000);
+
+    cloth.updateGeometry(clothGeometry);
+
+    /** raycast from camera to cursor */
+    raycaster.setFromCamera(new THREE.Vector2((mousePos.x / w) * 2 - 1, ((h - mousePos.y) / h) * 2 - 1), camera);
+    intersects = raycaster.intersectObject(clothMesh);
+
+    /** provide user interaction */
+    if (intersects.length > 0) {
+      if (windKeyDown)
+        cloth.blow(camera.position, intersects);
+      /** "grab" vertex index */
+      if (dragKeyDown && draggedIndex == -1)
+        draggedIndex = intersects[0].face.a;
+    }
+    if (dragKeyDown && draggedIndex != -1)
+      cloth.drag(calculateMousePosToWorld(mousePos), draggedIndex);
+
+    /** queue next frame */
+    setTimeout(() => {
+      animate(frameTime);
+    }, frameTime);
+
     renderer.render(scene, camera);
   }
 
+
+  /** add callback for window resize */
   let canvas = document.getElementsByTagName("canvas")[0];
+  let w = window.innerWidth;
+  let h = window.innerHeight - canvasSpace;
   let resize = function () {
     w = window.innerWidth;
-    h = window.innerHeight - 200;
+    h = window.innerHeight - canvasSpace;
     canvas.width = w;
     canvas.height = h;
   }
   window.onresize = resize;
   resize();
+
+  /**
+   * if canvas has been successfully initialized
+   * start rendering
+   */
   if (canvas.getContext) {
-    ctx = canvas.getContext('2d');
-    animate(performance.now());
+    animate(frameTime);
   }
+
+  
+
+  /** add mouse move callback */
   canvas.onmousemove = (evt) => {
     mousePos.x = evt.clientX;
     mousePos.y = evt.clientY;
   };
+
+  /**
+   * Prevent context menu while blowing wind
+   */
+  canvas.addEventListener('contextmenu', function(evt) { 
+    evt.preventDefault();
+  }, false);
+
+  /** register key events */
+  document.onkeydown = (evt) => {
+    if (evt.code === "KeyW")
+      windKeyDown = true;
+    if (evt.code === "KeyD")
+      dragKeyDown = true;
+  };
+
+  document.onkeyup = (evt) => {
+    if (evt.code === "KeyW")
+      windKeyDown = false;
+    if (evt.code === "KeyD") {
+      dragKeyDown = false;
+      draggedIndex = -1;
+    }
+  };
+
+  /** helper function to turn mouse position into 3D coordinates */
+  function calculateMousePosToWorld(mousePos){
+    var vec = new THREE.Vector3(); // create once and reuse
+    var pos = new THREE.Vector3(); // create once and reuse
+
+    vec.set(
+      (mousePos.x / window.innerWidth) * 2 - 1,
+    - (mousePos.y / window.innerHeight) * 2 + 1,
+      0.5);
+
+    vec.unproject(camera);
+
+    vec.sub(camera.position).normalize();
+
+    var distance = - camera.position.z / vec.z;
+
+    pos.copy(camera.position).add(vec.multiplyScalar(distance));
+    return pos;
+  }
 }
\ No newline at end of file