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

diff --git a/Scripts/main.js b/Scripts/main.js
index ececd6a..1a30f7a 100644
--- a/Scripts/main.js
+++ b/Scripts/main.js
@@ -1,34 +1,21 @@
-import { Face, Spring, Cloth } from './cloth.js';
 
 
-class Point {
-  constructor(x, y) {
-    this.x = x;
-    this.y = y;
-  }
+function addLights(scene){
+  scene.add(new THREE.AmbientLight(0x222222));
 
-  add(that) {
-    return new Point(
-      this.x + that.x,
-      this.y + that.y
-    );
-  }
+  const light1 = new THREE.PointLight(0xffffff, 1, 50);
+  light1.position.set(15, 1, 40);
+  scene.add(light1);
 
-  sub(that) {
-    return new Point(
-      this.x - that.x,
-      this.y - that.y
-    );
-  }
+  const light2 = new THREE.PointLight(0xffffff, 1, 50);
+  light2.position.set(-15, 0, 40);
+  scene.add(light2);
 
-  dist(that) {
-    let a = this.x - that.x;
-    let b = this.y - that.y;
-    return Math.sqrt(a * a + b * b)
-  }
+  const light3 = new THREE.PointLight(0xffffff, 1, 50);
+  light3.position.set(0, -1, 40);
+  scene.add(light3);
 }
 
-
 /**
  * setup THREE JS Scene, Camera and Renderer
  */
@@ -38,17 +25,39 @@ function setup_scene(canvasSpace) {
   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);
 
   /** position camera */
-  camera.position.y = 5;
-  camera.position.z = 10;
-
+  camera.position.z = 2;
+  addLights(scene);
   return [scene, camera, renderer];
 }
 
@@ -56,8 +65,7 @@ function setup_scene(canvasSpace) {
 document.body.onload = init;
 
 function init() {
-  let mousePos = new Point();
-  let previousClothSimulation;
+  let mousePos = new THREE.Vector2();
   
   /**
    * Space left empty under canvas
@@ -66,43 +74,74 @@ function init() {
   const canvasSpace = 200;
 
   /** Constant Frame Time */
-  const frameTime = 1000.0 / 60.0;
+  const frameTime = 1000.0 / 200.0;
 
   /** Setup scene */
   let [scene, camera, renderer] = setup_scene(canvasSpace);
   
-  /** setup cloth and generate debug mesh */
-  let cloth = new Cloth();
-  cloth.createBasic(10, 10, 10, 10);
-  //cloth.createDebugMesh(scene);
-
-  const material = new THREE.MeshBasicMaterial({ color: 0x0000ff });
-  const mesh = new THREE.Mesh(cloth.geometry, material);
-  //const mesh = new THREE.WireframeGeometry(cloth.geometry);
-  //const line = new THREE.LineSegments(mesh);
-  //line.material.depthTest = false;
-  //line.material.opacity = 0.25;
-  //line.material.transparent = true;
-  scene.add(mesh);
+  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
+   * @param {number} dt - time passed since last frame in ms
    */
   function animate(dt) {
-    cloth.simulate(dt);
-
+    /** 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 () {
-    let w = window.innerWidth;
-    let h = window.innerHeight - 200;
+    w = window.innerWidth;
+    h = window.innerHeight - canvasSpace;
     canvas.width = w;
     canvas.height = h;
   }
@@ -114,12 +153,58 @@ function init() {
    * start rendering
    */
   if (canvas.getContext) {
-    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