index.js

module.exports = function(THREE) {
  var BACK, COPLANAR, EPSILON, FRONT, SPANNING, returning,
    __bind = function(fn, me){ return function(){ return fn.apply(me, arguments); }; },
    __slice = [].slice,
    __hasProp = {}.hasOwnProperty,
    __extends = function(child, parent) { for (var key in parent) { if (__hasProp.call(parent, key)) child[key] = parent[key]; } function ctor() { this.constructor = child; } ctor.prototype = parent.prototype; child.prototype = new ctor(); child.__super__ = parent.prototype; return child; };

  EPSILON = 1e-5;

  COPLANAR = 0;

  FRONT = 1;

  BACK = 2;

  SPANNING = 3;

  returning = function(value, fn) {
    fn();
    return value;
  };

  function ThreeBSP(treeIsh, matrix) {
    this.matrix = matrix;
    this.intersect = __bind(this.intersect, this);
    this.union = __bind(this.union, this);
    this.subtract = __bind(this.subtract, this);
    this.toGeometry = __bind(this.toGeometry, this);
    this.toMesh = __bind(this.toMesh, this);
    this.toTree = __bind(this.toTree, this);
    if (this.matrix == null) {
      this.matrix = new THREE.Matrix4();
    }
    this.tree = this.toTree(treeIsh);
  }

  ThreeBSP.prototype.toTree = function(treeIsh) {
    var face, geometry, i, polygons, _fn, _i, _len, _ref,
      _this = this;
    if (treeIsh instanceof ThreeBSP.Node) {
      return treeIsh;
    }
    polygons = [];
    geometry = treeIsh instanceof THREE.Geometry ? treeIsh : treeIsh instanceof THREE.Mesh ? (treeIsh.updateMatrix(), this.matrix = treeIsh.matrix.clone(), treeIsh.geometry) : void 0;
    _ref = geometry.faces;
    _fn = function(face, i) {
      var faceVertexUvs, idx, polygon, vIndex, vName, vertex, _j, _len1, _ref1, _ref2;
      faceVertexUvs = (_ref1 = geometry.faceVertexUvs) != null ? _ref1[0][i] : void 0;
      if (faceVertexUvs == null) {
        faceVertexUvs = [new THREE.Vector2(), new THREE.Vector2(), new THREE.Vector2(), new THREE.Vector2()];
      }
      polygon = new ThreeBSP.Polygon();
      _ref2 = ['a', 'b', 'c', 'd'];
      for (vIndex = _j = 0, _len1 = _ref2.length; _j < _len1; vIndex = ++_j) {
        vName = _ref2[vIndex];
        if ((idx = face[vName]) != null) {
          vertex = geometry.vertices[idx];
          vertex = new ThreeBSP.Vertex(vertex.x, vertex.y, vertex.z, face.vertexNormals[0], new THREE.Vector2(faceVertexUvs[vIndex].x, faceVertexUvs[vIndex].y));
          vertex.applyMatrix4(_this.matrix);
          polygon.vertices.push(vertex);
        }
      }
      return polygons.push(polygon.calculateProperties());
    };
    for (i = _i = 0, _len = _ref.length; _i < _len; i = ++_i) {
      face = _ref[i];
      _fn(face, i);
    }
    return new ThreeBSP.Node(polygons);
  };

  ThreeBSP.prototype.toMesh = function(material) {
    var geometry, mesh,
      _this = this;
    if (material == null) {
      material = new THREE.MeshNormalMaterial();
    }
    geometry = this.toGeometry();
    return returning((mesh = new THREE.Mesh(geometry, material)), function() {
      mesh.position.getPositionFromMatrix(_this.matrix);
      return mesh.rotation.setFromRotationMatrix(_this.matrix);
    });
  };

  ThreeBSP.prototype.toGeometry = function() {
    var geometry, matrix,
      _this = this;
    matrix = new THREE.Matrix4().getInverse(this.matrix);
    return returning((geometry = new THREE.Geometry()), function() {
      var face, idx, polyVerts, polygon, v, vertUvs, verts, _i, _len, _ref, _results;
      _ref = _this.tree.allPolygons();
      _results = [];
      for (_i = 0, _len = _ref.length; _i < _len; _i++) {
        polygon = _ref[_i];
        polyVerts = (function() {
          var _j, _len1, _ref1, _results1;
          _ref1 = polygon.vertices;
          _results1 = [];
          for (_j = 0, _len1 = _ref1.length; _j < _len1; _j++) {
            v = _ref1[_j];
            _results1.push(v.clone().applyMatrix4(matrix));
          }
          return _results1;
        })();
        _results.push((function() {
          var _j, _ref1, _results1;
          _results1 = [];
          for (idx = _j = 2, _ref1 = polyVerts.length; 2 <= _ref1 ? _j < _ref1 : _j > _ref1; idx = 2 <= _ref1 ? ++_j : --_j) {
            verts = [polyVerts[0], polyVerts[idx - 1], polyVerts[idx]];
            vertUvs = (function() {
              var _k, _len1, _ref2, _ref3, _results2;
              _results2 = [];
              for (_k = 0, _len1 = verts.length; _k < _len1; _k++) {
                v = verts[_k];
                _results2.push(new THREE.Vector2((_ref2 = v.uv) != null ? _ref2.x : void 0, (_ref3 = v.uv) != null ? _ref3.y : void 0));
              }
              return _results2;
            })();
            face = (function(func, args, ctor) {
              ctor.prototype = func.prototype;
              var child = new ctor, result = func.apply(child, args);
              return Object(result) === result ? result : child;
            })(THREE.Face3, __slice.call((function() {
              var _k, _len1, _results2;
              _results2 = [];
              for (_k = 0, _len1 = verts.length; _k < _len1; _k++) {
                v = verts[_k];
                _results2.push(geometry.vertices.push(v) - 1);
              }
              return _results2;
            })()).concat([polygon.normal.clone()]), function(){});
            geometry.faces.push(face);
            _results1.push(geometry.faceVertexUvs[0].push(vertUvs));
          }
          return _results1;
        })());
      }
      return _results;
    });
  };

  ThreeBSP.prototype.subtract = function(other) {
    var them, us, _ref;
    _ref = [this.tree.clone(), other.tree.clone()], us = _ref[0], them = _ref[1];
    us.invert().clipTo(them);
    them.clipTo(us).invert().clipTo(us).invert();
    return new ThreeBSP(us.build(them.allPolygons()).invert(), this.matrix);
  };

  ThreeBSP.prototype.union = function(other) {
    var them, us, _ref;
    _ref = [this.tree.clone(), other.tree.clone()], us = _ref[0], them = _ref[1];
    us.clipTo(them);
    them.clipTo(us).invert().clipTo(us).invert();
    return new ThreeBSP(us.build(them.allPolygons()), this.matrix);
  };

  ThreeBSP.prototype.intersect = function(other) {
    var them, us, _ref;
    _ref = [this.tree.clone(), other.tree.clone()], us = _ref[0], them = _ref[1];
    them.clipTo(us.invert()).invert().clipTo(us.clipTo(them));
    return new ThreeBSP(us.build(them.allPolygons()).invert(), this.matrix);
  };

  ThreeBSP.Vertex = (function(_super) {
    __extends(Vertex, _super);
    function Vertex(x, y, z, normal, uv) {
      this.normal = normal != null ? normal : new THREE.Vector3();
      this.uv = uv != null ? uv : new THREE.Vector2();
      this.interpolate = __bind(this.interpolate, this);
      this.lerp = __bind(this.lerp, this);
      Vertex.__super__.constructor.call(this, x, y, z);
    }

    Vertex.prototype.clone = function() {
      return new ThreeBSP.Vertex(this.x, this.y, this.z, this.normal.clone(), this.uv.clone());
    };

    Vertex.prototype.lerp = function(v, alpha) {
      var _this = this;
      return returning(Vertex.__super__.lerp.apply(this, arguments), function() {
        _this.uv.add(v.uv.clone().sub(_this.uv).multiplyScalar(alpha));
        return _this.normal.lerp(v, alpha);
      });
    };

    Vertex.prototype.interpolate = function() {
      var args, _ref;
      args = 1 <= arguments.length ? __slice.call(arguments, 0) : [];
      return (_ref = this.clone()).lerp.apply(_ref, args);
    };

    return Vertex;

  })(THREE.Vector3);

  ThreeBSP.Polygon = (function() {
    function Polygon(vertices, normal, w) {
      this.vertices = vertices != null ? vertices : [];
      this.normal = normal;
      this.w = w;
      this.subdivide = __bind(this.subdivide, this);
      this.tessellate = __bind(this.tessellate, this);
      this.classifySide = __bind(this.classifySide, this);
      this.classifyVertex = __bind(this.classifyVertex, this);
      this.invert = __bind(this.invert, this);
      this.clone = __bind(this.clone, this);
      this.calculateProperties = __bind(this.calculateProperties, this);
      if (this.vertices.length) {
        this.calculateProperties();
      }
    }

    Polygon.prototype.calculateProperties = function() {
      var _this = this;
      return returning(this, function() {
        var a, b, c, _ref;
        _ref = _this.vertices, a = _ref[0], b = _ref[1], c = _ref[2];
        _this.normal = b.clone().sub(a).cross(c.clone().sub(a)).normalize();
        return _this.w = _this.normal.clone().dot(a);
      });
    };

    Polygon.prototype.clone = function() {
      var v;
      return new ThreeBSP.Polygon((function() {
        var _i, _len, _ref, _results;
        _ref = this.vertices;
        _results = [];
        for (_i = 0, _len = _ref.length; _i < _len; _i++) {
          v = _ref[_i];
          _results.push(v.clone());
        }
        return _results;
      }).call(this), this.normal.clone(), this.w);
    };

    Polygon.prototype.invert = function() {
      var _this = this;
      return returning(this, function() {
        _this.normal.multiplyScalar(-1);
        _this.w *= -1;
        return _this.vertices.reverse();
      });
    };

    Polygon.prototype.classifyVertex = function(vertex) {
      var side;
      side = this.normal.dot(vertex) - this.w;
      switch (false) {
        case !(side < -EPSILON):
          return BACK;
        case !(side > EPSILON):
          return FRONT;
        default:
          return COPLANAR;
      }
    };

    Polygon.prototype.classifySide = function(polygon) {
      var back, front, tally, v, _i, _len, _ref, _ref1,
        _this = this;
      _ref = [0, 0], front = _ref[0], back = _ref[1];
      tally = function(v) {
        switch (_this.classifyVertex(v)) {
          case FRONT:
            return front += 1;
          case BACK:
            return back += 1;
        }
      };
      _ref1 = polygon.vertices;
      for (_i = 0, _len = _ref1.length; _i < _len; _i++) {
        v = _ref1[_i];
        tally(v);
      }
      if (front > 0 && back === 0) {
        return FRONT;
      }
      if (front === 0 && back > 0) {
        return BACK;
      }
      if ((front === back && back === 0)) {
        return COPLANAR;
      }
      return SPANNING;
    };

    Polygon.prototype.tessellate = function(poly) {
      var b, count, f, i, j, polys, t, ti, tj, v, vi, vj, _i, _len, _ref, _ref1, _ref2,
        _this = this;
      _ref = {
        f: [],
        b: [],
        count: poly.vertices.length
      }, f = _ref.f, b = _ref.b, count = _ref.count;
      if (this.classifySide(poly) !== SPANNING) {
        return [poly];
      }
      _ref1 = poly.vertices;
      for (i = _i = 0, _len = _ref1.length; _i < _len; i = ++_i) {
        vi = _ref1[i];
        vj = poly.vertices[(j = (i + 1) % count)];
        _ref2 = (function() {
          var _j, _len1, _ref2, _results;
          _ref2 = [vi, vj];
          _results = [];
          for (_j = 0, _len1 = _ref2.length; _j < _len1; _j++) {
            v = _ref2[_j];
            _results.push(this.classifyVertex(v));
          }
          return _results;
        }).call(this), ti = _ref2[0], tj = _ref2[1];
        if (ti !== BACK) {
          f.push(vi);
        }
        if (ti !== FRONT) {
          b.push(vi);
        }
        if ((ti | tj) === SPANNING) {
          t = (this.w - this.normal.dot(vi)) / this.normal.dot(vj.clone().sub(vi));
          v = vi.interpolate(vj, t);
          f.push(v);
          b.push(v);
        }
      }
      return returning((polys = []), function() {
        if (f.length >= 3) {
          polys.push(new ThreeBSP.Polygon(f));
        }
        if (b.length >= 3) {
          return polys.push(new ThreeBSP.Polygon(b));
        }
      });
    };

    Polygon.prototype.subdivide = function(polygon, coplanar_front, coplanar_back, front, back) {
      var poly, side, _i, _len, _ref, _results;
      _ref = this.tessellate(polygon);
      _results = [];
      for (_i = 0, _len = _ref.length; _i < _len; _i++) {
        poly = _ref[_i];
        side = this.classifySide(poly);
        switch (side) {
          case FRONT:
            _results.push(front.push(poly));
            break;
          case BACK:
            _results.push(back.push(poly));
            break;
          case COPLANAR:
            if (this.normal.dot(poly.normal) > 0) {
              _results.push(coplanar_front.push(poly));
            } else {
              _results.push(coplanar_back.push(poly));
            }
            break;
          default:
            throw new Error("BUG: Polygon of classification " + side + " in subdivision");
        }
      }
      return _results;
    };

    return Polygon;

  })();

  ThreeBSP.Node = (function() {
    Node.prototype.clone = function() {
      var node,
        _this = this;
      return returning((node = new ThreeBSP.Node()), function() {
        var p, _ref, _ref1, _ref2;
        node.divider = (_ref = _this.divider) != null ? _ref.clone() : void 0;
        node.polygons = (function() {
          var _i, _len, _ref1, _results;
          _ref1 = this.polygons;
          _results = [];
          for (_i = 0, _len = _ref1.length; _i < _len; _i++) {
            p = _ref1[_i];
            _results.push(p.clone());
          }
          return _results;
        }).call(_this);
        node.front = (_ref1 = _this.front) != null ? _ref1.clone() : void 0;
        return node.back = (_ref2 = _this.back) != null ? _ref2.clone() : void 0;
      });
    };

    function Node(polygons) {
      this.clipTo = __bind(this.clipTo, this);
      this.clipPolygons = __bind(this.clipPolygons, this);
      this.invert = __bind(this.invert, this);
      this.allPolygons = __bind(this.allPolygons, this);
      this.isConvex = __bind(this.isConvex, this);
      this.build = __bind(this.build, this);
      this.clone = __bind(this.clone, this);
      this.polygons = [];
      if ((polygons != null) && polygons.length) {
        this.build(polygons);
      }
    }

    Node.prototype.build = function(polygons) {
      var _this = this;
      return returning(this, function() {
        var poly, polys, side, sides, _i, _len, _results;
        sides = {
          front: [],
          back: []
        };
        if (_this.divider == null) {
          _this.divider = polygons[0].clone();
        }
        for (_i = 0, _len = polygons.length; _i < _len; _i++) {
          poly = polygons[_i];
          _this.divider.subdivide(poly, _this.polygons, _this.polygons, sides.front, sides.back);
        }
        _results = [];
        for (side in sides) {
          if (!__hasProp.call(sides, side)) continue;
          polys = sides[side];
          if (polys.length) {
            if (_this[side] == null) {
              _this[side] = new ThreeBSP.Node();
            }
            _results.push(_this[side].build(polys));
          } else {
            _results.push(void 0);
          }
        }
        return _results;
      });
    };

    Node.prototype.isConvex = function(polys) {
      var inner, outer, _i, _j, _len, _len1;
      for (_i = 0, _len = polys.length; _i < _len; _i++) {
        inner = polys[_i];
        for (_j = 0, _len1 = polys.length; _j < _len1; _j++) {
          outer = polys[_j];
          if (inner !== outer && outer.classifySide(inner) !== BACK) {
            return false;
          }
        }
      }
      return true;
    };

    Node.prototype.allPolygons = function() {
      var _ref, _ref1;
      return this.polygons.slice().concat(((_ref1 = this.front) != null ? _ref1.allPolygons() : void 0) || []).concat(((_ref = this.back) != null ? _ref.allPolygons() : void 0) || []);
    };

    Node.prototype.invert = function() {
      var _this = this;
      return returning(this, function() {
        var flipper, poly, _i, _j, _len, _len1, _ref, _ref1, _ref2;
        _ref = _this.polygons;
        for (_i = 0, _len = _ref.length; _i < _len; _i++) {
          poly = _ref[_i];
          poly.invert();
        }
        _ref1 = [_this.divider, _this.front, _this.back];
        for (_j = 0, _len1 = _ref1.length; _j < _len1; _j++) {
          flipper = _ref1[_j];
          if (flipper != null) {
            flipper.invert();
          }
        }
        return _ref2 = [_this.back, _this.front], _this.front = _ref2[0], _this.back = _ref2[1], _ref2;
      });
    };

    Node.prototype.clipPolygons = function(polygons) {
      var back, front, poly, _i, _len;
      if (!this.divider) {
        return polygons.slice();
      }
      front = [];
      back = [];
      for (_i = 0, _len = polygons.length; _i < _len; _i++) {
        poly = polygons[_i];
        this.divider.subdivide(poly, front, back, front, back);
      }
      if (this.front) {
        front = this.front.clipPolygons(front);
      }
      if (this.back) {
        back = this.back.clipPolygons(back);
      }
      return front.concat(this.back ? back : []);
    };

    Node.prototype.clipTo = function(node) {
      var _this = this;
      return returning(this, function() {
        var _ref, _ref1;
        _this.polygons = node.clipPolygons(_this.polygons);
        if ((_ref = _this.front) != null) {
          _ref.clipTo(node);
        }
        return (_ref1 = _this.back) != null ? _ref1.clipTo(node) : void 0;
      });
    };

    return Node;

  })();

  return ThreeBSP;

}