]> git.scottworley.com Git - nt3d/blob - nt3d.js
d0d716125a3b9357c1820f6e23868bf5e1b57920
[nt3d] / nt3d.js
1 /* nt3d - Javascript library for doing some 3D stuff.
2 * Copyright (C) 2012 Scott Worley <ScottWorley@ScottWorley.com>
3 *
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU Affero General Public License as
6 * published by the Free Software Foundation, either version 3 of the
7 * License, or (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU Affero General Public License for more details.
13 *
14 * You should have received a copy of the GNU Affero General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
16 */
17
18 nt3d = {
19 triangle: function(a, b, c) {
20 return [a, b, c];
21 },
22 quad: function(a, b, c, d) {
23 return this.triangle(a, b, c).concat(
24 this.triangle(c, d, a));
25 },
26 trianglefan: function(fan) {
27 var result = [];
28 for (var i = 2; i < fan.length; i++) {
29 result.push(fan[0], fan[i-1], fan[i]);
30 }
31 return result;
32 },
33 closed_trianglefan: function(fan) {
34 return this.trianglefan(fan.concat([fan[1]]));
35 },
36 quadstrip: function(strip) {
37 if (strip.length % 2 != 0) {
38 alert("quadstrip length not divisble by 2!");
39 }
40 var result = [];
41 for (var i = 2; i < strip.length; i += 2) {
42 result = result.concat(this.quad(strip[i-2], strip[i-1], strip[i+1], strip[i]));
43 }
44 return result;
45 },
46 closed_quadstrip: function(strip) {
47 return this.quadstrip(strip.concat([strip[0], strip[1]]));
48 },
49 extrude: function(shape, path, shapenormals, pathnormals) {
50 var guts_result = nt3d._extrude_guts(shape, path, shapenormals, pathnormals);
51 // Add the end-caps
52 // XXX: This doesn't work if shape is not convex
53 return guts_result.points.concat(
54 nt3d.trianglefan(guts_result.first_loop.reverse()),
55 nt3d.trianglefan(guts_result.last_loop));
56
57 },
58 closed_extrude: function(shape, path, shapenormals, pathnormals) {
59 var guts_result = nt3d._extrude_guts(shape, path, shapenormals, pathnormals);
60 // Stitch the ends together
61 return guts_result.points.concat(
62 nt3d.closed_quadstrip(nt3d.zip(guts_result.first_loop, guts_result.last_loop)));
63 },
64 _fix_pathnormals: function(shapenormals, pathnormals) {
65 // Fix pathnormals[i] to be perfectly perpendicular to
66 // shapenormals[i]. This lets extrude callers be sloppy
67 // with pathnormals, which can greatly simplify things.
68 var fixedpathnormals = [];
69 for (var i = 0; i < pathnormals.length; i++) {
70 var proj = this.project(shapenormals[i], pathnormals[i]);
71 fixedpathnormals[i] = this.sub(pathnormals[i], proj);
72 }
73 return fixedpathnormals;
74 },
75 _extrude_guts: function(shape, path, shapenormals, pathnormals) {
76 var fixedpathnormals = this._fix_pathnormals(shapenormals, pathnormals);
77 var result = { points: [] };
78 var prev_loop;
79 for (var i = 0; i < path.length; i++) {
80 // loop is shape in 3d with (0,0) at path[i], shape's
81 // z axis in the direction of shapenormals[i], and
82 // shape's x axis in the direction of pathnormals[i].
83 var loop = [];
84 // This is done in three steps:
85 // 1. Rotate shape out of the xy plane so that [0,0,1]
86 // becomes shapenormals[i] by crossing [0,0,1] and
87 // shapenormals[i] to get a rotation axis and taking
88 // their dot product to get a rotation angle. This
89 // puts the shape in the correct plane, but does not
90 // constrain its rotation about shapenormals[i].
91 var rot1axis = this.unit(this.cross([0,0,1], shapenormals[i]));
92 var rot1angle = this.angle_between([0,0,1], this.unit(shapenormals[i]));
93 // 2. Rotate around shapenormals[i] so that [1,0,0]
94 // becomes fixedpathnormals[i].
95 var rot2axis = this.unit(shapenormals[i]);
96 var rot2angle = this.angle_between([1,0,0], this.unit(fixedpathnormals[i]));
97 // 3. Translate to path[i].
98 // This would probably be faster and more numerically stable
99 // if the two rotations were applied as one combined operation
100 // rather than separate steps.
101 for (var j = 0; j < shape.length; j++) {
102 var p = [shape[j][0], shape[j][1], 0];
103 if (rot1angle > 1e-7) {
104 p = this.rotate(p, rot1axis, rot1angle);
105 }
106 if (rot2angle > 1e-7) {
107 p = this.rotate(p, rot2axis, rot2angle);
108 }
109 p = this.translate(p, path[i]);
110 loop[j] = p;
111 }
112 if (i == 0) {
113 result.first_loop = loop;
114 } else {
115 result.points = result.points.concat(nt3d.closed_quadstrip(nt3d.zip(loop, prev_loop)));
116 }
117 prev_loop = loop;
118 }
119 result.last_loop = prev_loop;
120 return result;
121 },
122 zip: function(a, b) {
123 var result = [];
124 if (a.length != b.length) {
125 alert("Zip over different-sized inputs");
126 }
127 for (var i = 0; i < a.length; i++) {
128 result.push(a[i], b[i]);
129 }
130 return result;
131 },
132 magnitude: function(a) {
133 return Math.sqrt(a[0]*a[0] + a[1]*a[1] + a[2]*a[2]);
134 },
135 unit: function(a) {
136 return this.scale(a, 1 / this.magnitude(a));
137 },
138 sub: function(a, b) {
139 return [a[0] - b[0],
140 a[1] - b[1],
141 a[2] - b[2]];
142 },
143 dot: function(a, b) {
144 return a[0]*b[0] + a[1]*b[1] + a[2]*b[2];
145 },
146 scale: function(v, s) { // Scale vector v by scalar s
147 return [s*v[0], s*v[1], s*v[2]];
148 },
149 cross: function(a, b) {
150 return [a[1]*b[2] - a[2]*b[1],
151 a[2]*b[0] - a[0]*b[2],
152 a[0]*b[1] - a[1]*b[0]];
153 },
154 normal: function(a, b, c) {
155 return this.cross(this.sub(a, b), this.sub(b, c));
156 },
157 project: function(a, b) { // Project b onto a
158 var a_magnitude = this.magnitude(a);
159 return this.scale(a, this.dot(a, b) / a_magnitude * a_magnitude);
160 },
161 translate: function(a, b) {
162 return [a[0] + b[0], a[1] + b[1], a[2] + b[2]];
163 },
164 angle_between: function(a, b) { // a and b must be unit vectors
165 return Math.acos(this.dot(a, b));
166 },
167 rotate: function(point, axis, angle) { // axis must be a unit vector
168 // From http://inside.mines.edu/~gmurray/ArbitraryAxisRotation/
169 var cosangle = Math.cos(angle);
170 var sinangle = Math.sin(angle);
171 var tmp = this.dot(point, axis) * (1 - cosangle);
172 return [axis[0]*tmp + point[0]*cosangle + (-axis[2]*point[1] + axis[1]*point[2])*sinangle,
173 axis[1]*tmp + point[1]*cosangle + ( axis[2]*point[0] - axis[0]*point[2])*sinangle,
174 axis[2]*tmp + point[2]*cosangle + (-axis[1]*point[0] + axis[0]*point[1])*sinangle];
175 },
176 go: function() {
177 // Get params from form
178 var params = [];
179 for (var i = 0; i < this.user_params.length; i++) {
180 params[i] = this.form.elements["param"+i].value;
181 }
182
183 // Run user_function
184 this.points = this.user_function.apply(null, params);
185 if (this.points.length % 3 != 0) {
186 alert("Points list length not divisble by 3!");
187 }
188 var n = this.points.length / 3;
189
190 // Make STL
191 this.stl = "solid " + this.user_function.name + "\n";
192 for (var i = 0; i < n; i++) {
193 var a = this.points[i*3+0];
194 var b = this.points[i*3+1];
195 var c = this.points[i*3+2];
196 var normal = this.normal(a, b, c);
197 this.stl += "facet normal " + normal[0] + " " + normal[1] + " " + normal[2] + "\n" +
198 "outer loop\n" +
199 "vertex " + a[0] + " " + a[1] + " " + a[2] + "\n"+
200 "vertex " + b[0] + " " + b[1] + " " + b[2] + "\n"+
201 "vertex " + c[0] + " " + c[1] + " " + c[2] + "\n"+
202 "endloop\n" +
203 "endfacet\n";
204 }
205 this.stl += "endsolid " + this.user_function.name + "\n";
206
207 // Remove any previous download links
208 var old_download_link = document.getElementById("nt3d_download");
209 if (old_download_link) {
210 old_download_link.parentNode.removeChild(old_download_link);
211 }
212
213 // Offer result as download
214 var download_link = document.createElement("a");
215 download_link.appendChild(document.createTextNode("Download!"));
216 download_link.setAttribute("id", "nt3d_download");
217 download_link.setAttribute("style", "background-color: blue");
218 download_link.setAttribute("download", this.user_function.name + ".stl");
219 download_link.setAttribute("href", "data:application/sla," + encodeURIComponent(this.stl));
220 this.ui.appendChild(download_link);
221 setTimeout(function() { download_link.setAttribute("style", "-webkit-transition: background-color 0.4s; -moz-transition: background-color 0.4s; -o-transition: background-color 0.4s; -ms-transition: background-color 0.4s; transition: background-color 0.4s; background-color: inherit"); }, 0);
222 },
223 framework: function (f, params) {
224 this.user_function = f;
225 this.user_params = params;
226
227 // Make the UI
228 this.ui = document.getElementById("nt3dui");
229 if (!this.ui) {
230 this.ui = document.createElement("div");
231 this.ui.setAttribute("id", "nt3dui");
232 document.body.appendChild(this.ui);
233 }
234 this.form = document.createElement("form");
235 this.form.setAttribute("onsubmit", "nt3d.go(); return false");
236 this.ui.appendChild(this.form);
237 var table = document.createElement("table");
238 this.form.appendChild(table);
239 var tr = document.createElement("tr");
240 table.appendChild(tr);
241 var th = document.createElement("th");
242 th.appendChild(document.createTextNode("Variable"));
243 tr.appendChild(th);
244 th = document.createElement("th");
245 th.appendChild(document.createTextNode("Value"));
246 tr.appendChild(th);
247 for (var i = 0; i < params.length; i++) {
248 tr = document.createElement("tr");
249 table.appendChild(tr);
250 var td = document.createElement("td");
251 td.appendChild(document.createTextNode(params[i][0]));
252 tr.appendChild(td);
253 td = document.createElement("td");
254 var input = document.createElement("input");
255 input.setAttribute("name", "param" + i);
256 input.setAttribute("value", params[i][1]);
257 td.appendChild(input);
258 tr.appendChild(td);
259 }
260 var go = document.createElement("input");
261 go.setAttribute("type", "button");
262 go.setAttribute("value", "Go!");
263 go.setAttribute("onclick", "nt3d.go()");
264 this.form.appendChild(go);
265 }
266 };