From 608f13ea90cdad08c9230194bc076f1fdcc3992d Mon Sep 17 00:00:00 2001
From: limengxun <limengxun45@outlook.com>
Date: Fri, 7 Jun 2024 19:16:14 +0800
Subject: [PATCH] add keys to change ellipsoid size

---
 index.html |  1 +
 main.js    | 18 ++++++++++++++++--
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/index.html b/index.html
index c25c895..f960262 100644
--- a/index.html
+++ b/index.html
@@ -239,6 +239,7 @@ <h3 class="nohf">WebGL 3D Gaussian Splat Viewer</h3>
 - press 0-9 to switch to one of the pre-loaded camera views
 - press '-' or '+'key to cycle loaded cameras
 - press p to resume default animation
+- press m/n keys to increase/decrease ellipsoid size
 - drag and drop .ply file to convert to .splat
 - drag and drop cameras.json to load cameras
 </div>
diff --git a/main.js b/main.js
index 94d6b1a..593d0ce 100644
--- a/main.js
+++ b/main.js
@@ -661,6 +661,7 @@ uniform highp usampler2D u_texture;
 uniform mat4 projection, view;
 uniform vec2 focal;
 uniform vec2 viewport;
+uniform float splat_scale;
 
 in vec2 position;
 in int index;
@@ -707,8 +708,8 @@ void main () {
     vec2 vCenter = vec2(pos2d) / pos2d.w;
     gl_Position = vec4(
         vCenter 
-        + position.x * majorAxis / viewport 
-        + position.y * minorAxis / viewport, 0.0, 1.0);
+        + position.x * majorAxis * splat_scale / viewport 
+        + position.y * minorAxis * splat_scale / viewport, 0.0, 1.0);
 
 }
 `.trim();
@@ -820,6 +821,10 @@ async function main() {
     const u_viewport = gl.getUniformLocation(program, "viewport");
     const u_focal = gl.getUniformLocation(program, "focal");
     const u_view = gl.getUniformLocation(program, "view");
+    const u_splat_scale = gl.getUniformLocation(program, "splat_scale");
+
+    let splat_scale = 1.0;
+    gl.uniform1f(u_splat_scale, splat_scale);
 
     // positions
     const triangleVertices = new Float32Array([-2, -2, 2, -2, 2, 2, -2, 2]);
@@ -1301,6 +1306,15 @@ async function main() {
             inv = translate4(inv, 0, 0, -d);
         }
 
+        if (activeKeys.includes("KeyN")) {
+            splat_scale = Math.max(0.1, splat_scale - 0.01);
+            gl.uniform1f(u_splat_scale, splat_scale);
+        }
+        if (activeKeys.includes("KeyM")) {
+            splat_scale = Math.min(2, splat_scale + 0.01);
+            gl.uniform1f(u_splat_scale, splat_scale);
+        }
+
         viewMatrix = invert4(inv);
 
         if (carousel) {