Make extrude's arguments more flexible and powerful

[nt3d] / nt3d.js

/* nt3d - Javascript library for doing some 3D stuff.
 * Copyright (C) 2012  Scott Worley <ScottWorley@ScottWorley.com>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Affero General Public License as
 *  published by the Free Software Foundation, either version 3 of the
 *  License, or (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Affero General Public License for more details.
 *
 *  You should have received a copy of the GNU Affero General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

nt3d = {
        triangle: function(a, b, c) {
                return [a, b, c];
        },
        quad: function(a, b, c, d) {
                return this.triangle(a, b, c).concat(
                       this.triangle(c, d, a));
        },
        trianglefan: function(fan) {
                var result = [];
                for (var i = 2; i < fan.length; i++) {
                        result.push(fan[0], fan[i-1], fan[i]);
                }
                return result;
        },
        closed_trianglefan: function(fan) {
                return this.trianglefan(fan.concat([fan[1]]));
        },
        quadstrip: function(strip) {
                if (strip.length % 2 != 0) {
                        alert("quadstrip length not divisble by 2!");
                }
                var result = [];
                for (var i = 2; i < strip.length; i += 2) {
                        result = result.concat(this.quad(strip[i-2], strip[i-1], strip[i+1], strip[i]));
                }
                return result;
        },
        closed_quadstrip: function(strip) {
                return this.quadstrip(strip.concat([strip[0], strip[1]]));
        },
        circle: function(r, n) {
                var points = [];
                for (var i = 0; i < n; i++) {
                        points.push([r*Math.cos(2*Math.PI*i/n),
                                     r*Math.sin(2*Math.PI*i/n),
                                     0]);
                }
                return points;
        },
        cone: function(base_center, apex, radius, steps) {
                var base = this.circle(radius, steps);
                base = this.rotate_onto(base, [0,0,1], this.sub(apex, base_center));
                base = this.translate(base, base_center);
                return this.closed_trianglefan([apex].concat(base)).concat(
                       this.trianglefan(base.reverse()));
        },
        to_function: function(thing, make_indexer) {
                // If thing is a point, just yield thing every time.
                // If thing is a list of points && make_indexer, index into thing.
                // If thing is already a function, just return it.
                if (({}).toString.call(thing) === "[object Function]") {
                        return thing;  // Already a function
                }
                if (make_indexer && Array.isArray(thing[0])) {
                        // Looks like a list of points.
                        return function(i) { return thing[i]; }
                }
                return function() { return thing; }
        },
        extrude: function(path, shape, shapenormals, pathnormals) {

                var guts_result = nt3d._extrude_guts(path, shape, shapenormals, pathnormals);
                // Add the end-caps
                // XXX: This doesn't work if shape is not convex
                return guts_result.points.concat(
                        nt3d.trianglefan(guts_result.first_loop.reverse()),
                        nt3d.trianglefan(guts_result.last_loop));

        },
        closed_extrude: function(path, shape, shapenormals, pathnormals) {
                var guts_result = nt3d._extrude_guts(path, shape, shapenormals, pathnormals);
                // Stitch the ends together
                return guts_result.points.concat(
                        nt3d.closed_quadstrip(nt3d.zip(guts_result.first_loop, guts_result.last_loop)));
        },
        _extrude_guts: function(path, shape, shapenormals, pathnormals) {
                var shape_fun = this.to_function(shape, false);
                var shapenormal_fun = this.to_function(shapenormals, true);
                var pathnormal_fun = this.to_function(pathnormals, true);
                var result = { points: [] };
                var prev_loop;
                for (var i = 0; i < path.length; i++) {
                        var shapenormali = shapenormal_fun(i, path[i]);
                        var pathnormali = pathnormal_fun(i, path[i], shapenormali);

                        // Fix pathnormali to be perfectly perpendicular to
                        // shapenormali.  pathnormali must be perpendicular to
                        // shapenormali or the second rotation will take loop
                        // back out of the shapenormali plane that the first
                        // rotation so carefully placed it in.  But, letting
                        // callers be sloppy with the pathnormals can greatly
                        // simplify generating them -- so much so that you can
                        // often just pass a constant to use the same value
                        // along the whole path.
                        pathnormali = this.project_to_orthogonal(shapenormali, pathnormali);

                        var shapei = shape_fun(i, path[i], shapenormali, pathnormali);

                        // loop is shapei in 3d with (0,0) at path[i], shape's
                        // z axis in the direction of shapenormali, and shape's
                        // x axis in the direction of pathnormali.  We tack
                        // [1,0,0] onto the end as a hack to see where it ends
                        // up after the first rotation.  This is removed later.
                        var loop = shapei.concat([[1,0,0]]);

                        // This is done in three steps:
                        // 1. Rotate shape out of the xy plane so that [0,0,1]
                        //    becomes shapenormali.  This puts the shape in
                        //    the correct plane, but does not constrain its
                        //    rotation about shapenormali.
                        loop = this.rotate_onto(loop, [0,0,1], shapenormali);
                        var shapex = loop.pop();

                        // 2. Rotate around shapenormali so that [1,0,0]
                        //    becomes pathnormali.
                        loop = this.rotate_onto(loop, shapex, pathnormali);

                        // (This would probably be faster and more numerically stable
                        // if the two rotations were applied as one combined operation
                        // rather than separate steps.)

                        // 3. Translate to path[i].
                        loop = this.translate(loop, path[i]);

                        if (i == 0) {
                                result.first_loop = loop;
                        } else {
                                result.points = result.points.concat(nt3d.closed_quadstrip(nt3d.zip(loop, prev_loop)));
                        }
                        prev_loop = loop;
                }
                result.last_loop = prev_loop;
                return result;
        },
        zip: function(a, b) {
                var result = [];
                if (a.length != b.length) {
                        alert("Zip over different-sized inputs");
                }
                for (var i = 0; i < a.length; i++) {
                        result.push(a[i], b[i]);
                }
                return result;
        },
        magnitude: function(a) {
                return Math.sqrt(a[0]*a[0] + a[1]*a[1] + a[2]*a[2]);
        },
        unit: function(a) {
                return this.scale(a, 1 / this.magnitude(a));
        },
        sub: function(a, b) {
                return [a[0] - b[0],
                        a[1] - b[1],
                        a[2] - b[2]];
        },
        neg: function(a) {
                return [-a[0], -a[1], -a[2]];
        },
        dot: function(a, b) {
                return a[0]*b[0] + a[1]*b[1] + a[2]*b[2];
        },
        scale: function(v, s) { // Scale vector v by scalar s
                return [s*v[0], s*v[1], s*v[2]];
        },
        cross: function(a, b) {
                return [a[1]*b[2] - a[2]*b[1],
                        a[2]*b[0] - a[0]*b[2],
                        a[0]*b[1] - a[1]*b[0]];
        },
        normal: function(a, b, c) {
                return this.cross(this.sub(a, b), this.sub(b, c));
        },
        project: function(a, b) { // Project b onto a
                var a_magnitude = this.magnitude(a);
                return this.scale(a, this.dot(a, b) / a_magnitude * a_magnitude);
        },
        project_to_orthogonal: function(a, b) {
                // The nearest thing to b that is orthogonal to a
                return this.sub(b, this.project(a, b));
        },
        translate: function(points, offset) {
                var translated = [];
                for (var i = 0; i < points.length; i++) {
                        translated[i] = [points[i][0] + offset[0],
                                         points[i][1] + offset[1],
                                         points[i][2] + offset[2]];
                }
                return translated;
        },
        angle_between: function(a, b) { // a and b must be unit vectors
                return Math.acos(this.dot(a, b));
        },
        rotate_about_origin: function(points, axis, angle) { // axis must be a unit vector
                // From http://inside.mines.edu/~gmurray/ArbitraryAxisRotation/
                var cosangle = Math.cos(angle);
                var sinangle = Math.sin(angle);
                var rotated = [];
                for (var i = 0; i < points.length; i++) {
                        var p = points[i];
                        var tmp = this.dot(p, axis) * (1 - cosangle);
                        rotated[i] = [
                                axis[0]*tmp + p[0]*cosangle + (-axis[2]*p[1] + axis[1]*p[2])*sinangle,
                                axis[1]*tmp + p[1]*cosangle + ( axis[2]*p[0] - axis[0]*p[2])*sinangle,
                                axis[2]*tmp + p[2]*cosangle + (-axis[1]*p[0] + axis[0]*p[1])*sinangle];
                }
                return rotated;
        },
        rotate_onto: function(points, a, b) {
                // Rotate points such that a (in points-space) maps onto b
                // by crossing a and b to get a rotation axis and using
                // angle_between to get a rotation angle.
                var angle = this.angle_between(this.unit(a), this.unit(b));
                if (Math.abs(angle) < 1e-15) {
                        // No siginificant rotation to perform.  Bail to avoid
                        // NaNs and numerical error
                        return points;
                }
                var axis = this.unit(this.cross(a, b));
                return this.rotate_about_origin(points, axis, angle);
        },
        rotate: function(points, center, axis, angle) { // axis must be a unit vector
                return this.translate(
                        this.rotate_about_origin(
                                this.translate(points, this.neg(center)),
                                axis,
                                angle),
                        center);
        },
        go: function() {
                // Remove any previous download links
                var old_download_link = document.getElementById("nt3d_download");
                if (old_download_link) {
                        old_download_link.parentNode.removeChild(old_download_link);
                }

                // Continue in a callback, so that there's not a stale download
                // link hanging around while we process.
                setTimeout(function(the_this) { (function() {

                // Get params from form
                var params = [];
                for (var i = 0; i < this.user_params.length; i++) {
                        var as_string = this.form.elements["param"+i].value;
                        var as_num = +as_string;
                        params[i] = isNaN(as_num) ? as_string : as_num;
                }

                // Run user_function
                this.points = this.user_function.apply(null, params);
                if (this.points.length % 3 != 0) {
                        alert("Points list length not divisble by 3!");
                }
                var n = this.points.length / 3;

                // Make STL
                this.stl = "solid " + this.user_function.name + "\n";
                for (var i = 0; i < n; i++) {
                        var a = this.points[i*3+0];
                        var b = this.points[i*3+1];
                        var c = this.points[i*3+2];
                        var normal = this.normal(a, b, c);
                        this.stl += "facet normal " + normal[0] + " " + normal[1] + " " + normal[2] + "\n" +
                                "outer loop\n" +
                                "vertex " + a[0] + " " + a[1] + " " + a[2] + "\n"+
                                "vertex " + b[0] + " " + b[1] + " " + b[2] + "\n"+
                                "vertex " + c[0] + " " + c[1] + " " + c[2] + "\n"+
                                "endloop\n" +
                                "endfacet\n";
                }
                this.stl += "endsolid " + this.user_function.name + "\n";


                // Offer result as download
                var download_link = document.createElement("a");
                download_link.appendChild(document.createTextNode("Download!"));
                download_link.setAttribute("id", "nt3d_download");
                download_link.setAttribute("style", "background-color: blue");
                download_link.setAttribute("download", this.user_function.name + ".stl");
                download_link.setAttribute("href", "data:application/sla," + encodeURIComponent(this.stl));
                this.ui.appendChild(download_link);
                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);

                }).call(the_this); }, 0, this); // (We were in a callback this whole time, remember?)
        },
        framework: function (f, params) {
                this.user_function = f;
                this.user_params = params;

                // Make the UI
                this.ui = document.getElementById("nt3dui");
                if (!this.ui) {
                        this.ui = document.createElement("div");
                        this.ui.setAttribute("id", "nt3dui");
                        document.body.appendChild(this.ui);
                }
                this.form = document.createElement("form");
                this.form.setAttribute("onsubmit", "nt3d.go(); return false");
                this.ui.appendChild(this.form);
                var table = document.createElement("table");
                this.form.appendChild(table);
                var tr = document.createElement("tr");
                table.appendChild(tr);
                var th = document.createElement("th");
                th.appendChild(document.createTextNode("Variable"));
                tr.appendChild(th);
                th = document.createElement("th");
                th.appendChild(document.createTextNode("Value"));
                tr.appendChild(th);
                for (var i = 0; i < params.length; i++) {
                        tr = document.createElement("tr");
                        table.appendChild(tr);
                        var td = document.createElement("td");
                        td.appendChild(document.createTextNode(params[i][0]));
                        tr.appendChild(td);
                        td = document.createElement("td");
                        var input = document.createElement("input");
                        input.setAttribute("name", "param" + i);
                        input.setAttribute("value", params[i][1]);
                        td.appendChild(input);
                        tr.appendChild(td);
                }
                var go = document.createElement("input");
                go.setAttribute("type", "button");
                go.setAttribute("value", "Go!");
                go.setAttribute("onclick", "nt3d.go()");
                this.form.appendChild(go);
        }
};