From 4107b962ad151e43fc4081711e6d712c24720197 Mon Sep 17 00:00:00 2001
From: Stewart Allen <stewart.allen@gmail.com>
Date: Fri, 19 Jul 2024 20:12:15 -0400
Subject: [PATCH] fix twist for chamfered gears

---
 src/mesh/work.js | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/src/mesh/work.js b/src/mesh/work.js
index 002ea928..96a3ced3 100644
--- a/src/mesh/work.js
+++ b/src/mesh/work.js
@@ -410,19 +410,18 @@ let model = {
             const srad = shaft / 2;
             poly.addInner( newPolygon().centerCircle({ x:0, y:0, z:0 }, srad, nump) );
         }
-        const verts = poly.extrude(height || 15, { chamfer });
+        const zh = height || 15;
+        const verts = poly.extrude(zh, { chamfer });
         if (twist) {
             const rad = base.util.toRadians(twist);
             for (let i=0; i<verts.length; i += 3) {
-                if (Math.abs(verts[i+2]) < 0.001) {
-                    let [x, y] = base.util.rotate(
-                        verts[i],
-                        verts[i+1],
-                        rad
-                    );
-                    verts[i] = x;
-                    verts[i+1] = y;
-                }
+                let [x, y] = base.util.rotate(
+                    verts[i],
+                    verts[i+1],
+                    rad * (verts[i+2] / zh)
+                );
+                verts[i] = x;
+                verts[i+1] = y;
             }
         }
         send.done(verts);