poly-extrude 0.13.0 → 0.14.0

package/src/plane.ts

@@ -0,0 +1,60 @@
+import { ResultType } from "./type";
+
+export function plane(width: number, height: number, devideW: number, devideH: number): ResultType {
+    devideW = Math.max(1, devideW);
+    devideH = Math.max(1, devideH);
+    const dx = width / devideW, dy = height / devideH;
+    const minX = -width / 2, maxY = height / 2, minY = -height / 2;
+    const len = (devideW + 1) * (devideH + 1);
+    const position = new Float32Array(len * 3), uv = new Float32Array(len * 2), normal = new Float32Array(len * 3), indices = new Uint32Array(len * 10);
+    let index = 0, uIndex = 0, iIndex = 0;
+    for (let j = 0; j <= devideH; j++) {
+        for (let i = 0; i <= devideW; i++) {
+            const x = minX + dx * i;
+            const y = maxY - dy * j;
+            position[index] = x;
+            position[index + 1] = y;
+            position[index + 2] = 0;
+
+            normal[index] = 0;
+            normal[index + 1] = 0;
+            normal[index + 2] = 1;
+            // position.push(x, y, 0);
+            // normal.push(0, 0, 1);
+            const uvx = (x - minX) / width, uvy = (y - minY) / height;
+            // uv.push(uvx, uvy);
+            uv[uIndex] = uvx;
+            uv[uIndex + 1] = uvy;
+
+            index += 3;
+            uIndex += 2;
+            if (i < devideW && j < devideH) {
+                const a = j * (devideW + 1) + i, b = a + 1, c = (devideW + 1) * (j + 1) + i, d = c + 1;
+                indices[iIndex] = a;
+                indices[iIndex + 1] = c;
+                indices[iIndex + 2] = b;
+                indices[iIndex + 3] = c;
+                indices[iIndex + 4] = d;
+                indices[iIndex + 5] = b;
+                iIndex += 6;
+                // indexs.push(a, c, b, c, d, b);
+            }
+        }
+    }
+    const indexArray = new Uint32Array(iIndex);
+    for (let i = 0, len = indexArray.length; i < len; i++) {
+        indexArray[i] = indices[i];
+    }
+    // for (let j = 0; j < devideH; j++) {
+    //     for (let i = 0; i < devideW; i++) {
+    //         const a = j * (devideW + 1) + i, b = a + 1, c = (devideW + 1) * (j + 1) + i, d = c + 1;
+    //         indexs.push(a, c, b, c, d, b);
+    //     }
+    // }
+    return {
+        position,
+        uv,
+        normal,
+        indices: indexArray
+    };
+}

package/src/polygon.ts

@@ -0,0 +1,189 @@
+
+import earcut from 'earcut';
+import { generateNormal, generateSideWallUV, isClockwise, merge } from './util';
+import { PolylineType, PolygonType, ResultType } from './type';
+
+type PolygonsOptions = {
+    depth?: number
+}
+
+type PolygonsResult = ResultType & {
+    polygons: Array<PolygonType>;
+}
+
+
+export function extrudePolygons(polygons: Array<PolygonType>, options?: PolygonsOptions): PolygonsResult {
+    options = Object.assign({}, { depth: 2 }, options);
+    const results = polygons.map(polygon => {
+        for (let i = 0, len = polygon.length; i < len; i++) {
+            const ring = polygon[i];
+            validateRing(ring);
+            if (i === 0) {
+                if (!isClockwise(ring)) {
+                    polygon[i] = ring.reverse();
+                }
+            } else if (isClockwise(ring)) {
+                polygon[i] = ring.reverse();
+            }
+            if (isClosedRing(ring)) {
+                ring.splice(ring.length - 1, 1);
+            }
+        }
+        const result = flatVertices(polygon, options) as Record<string, any>;
+        result.polygon = polygon;
+        const triangles = earcut(result.flatVertices, result.holes, 2);
+        generateTopAndBottom(result, triangles);
+        generateSides(result, options);
+        result.position = new Float32Array(result.points);
+        result.indices = new Uint32Array(result.indices);
+        result.uv = new Float32Array(result.uv);
+        result.normal = generateNormal(result.indices, result.position);
+        return result;
+    });
+    const result = merge(results) as PolygonsResult;
+    result.polygons = polygons;
+    return result;
+
+}
+
+function generateTopAndBottom(result, triangles) {
+    const indices: number[] = [];
+    const { count } = result;
+    for (let i = 0, len = triangles.length; i < len; i += 3) {
+        // top
+        const a = triangles[i], b = triangles[i + 1], c = triangles[i + 2];
+        indices[i] = a;
+        indices[i + 1] = b;
+        indices[i + 2] = c;
+        // bottom
+        const idx = len + i;
+        const a1 = count + a, b1 = count + b, c1 = count + c;
+        indices[idx] = a1;
+        indices[idx + 1] = b1;
+        indices[idx + 2] = c1;
+    }
+    result.indices = indices;
+}
+
+function generateSides(result, options) {
+    const { points, indices, polygon, uv } = result;
+    const depth = options.depth;
+    let pIndex = points.length - 1;
+    let iIndex = indices.length - 1;
+    for (let i = 0, len = polygon.length; i < len; i++) {
+        const ring = polygon[i];
+        let j = 0;
+        const len1 = ring.length;
+        while (j < len1) {
+            const v1 = ring[j];
+            let v2 = ring[j + 1];
+            if (j === len1 - 1) {
+                v2 = ring[0];
+            }
+            const idx = points.length / 3;
+            const x1 = v1[0], y1 = v1[1], z1 = v1[2] || 0, x2 = v2[0], y2 = v2[1], z2 = v2[2] || 0;
+            points[++pIndex] = x1;
+            points[++pIndex] = y1;
+            points[++pIndex] = z1 + depth;
+            points[++pIndex] = x2;
+            points[++pIndex] = y2;
+            points[++pIndex] = z2 + depth;
+            points[++pIndex] = x1;
+            points[++pIndex] = y1;
+            points[++pIndex] = z1;
+            points[++pIndex] = x2;
+            points[++pIndex] = y2;
+            points[++pIndex] = z2;
+            // points.push(x1, y1, z, x2, y2, z, x1, y1, 0, x2, y2, 0);
+            const a = idx + 2, b = idx + 3, c = idx, d = idx + 1;
+            // points.push(p3, p4, p1, p2);
+            // index.push(a, c, b, c, d, b);
+            indices[++iIndex] = a;
+            indices[++iIndex] = c;
+            indices[++iIndex] = b;
+            indices[++iIndex] = c;
+            indices[++iIndex] = d;
+            indices[++iIndex] = b;
+            // index.push(c, d, b);
+
+            generateSideWallUV(uv, points, a, b, c, d);
+            j++;
+        }
+    }
+}
+
+function calPolygonPointsCount(polygon) {
+    let count = 0;
+    let i = 0;
+    const len = polygon.length;
+    while (i < len) {
+        count += (polygon[i].length);
+        i++;
+    }
+    return count;
+}
+
+function flatVertices(polygon, options) {
+    const count = calPolygonPointsCount(polygon);
+    const len = polygon.length;
+    const holes: number[] = [], flatVertices = new Float32Array(count * 2), points: number[] = [], uv: number[] = [];
+    const pOffset = count * 3, uOffset = count * 2;
+    const depth = options.depth;
+
+    let idx0 = 0, idx1 = 0, idx2 = 0;
+    for (let i = 0; i < len; i++) {
+        const ring = polygon[i];
+        if (i > 0) {
+            holes.push(idx0 / 2);
+        }
+        let j = 0;
+        const len1 = ring.length;
+        while (j < len1) {
+            const c = ring[j];
+            const x = c[0], y = c[1], z = c[2] || 0;
+
+            flatVertices[idx0++] = x;
+            flatVertices[idx0++] = y;
+
+            // top vertices
+            points[idx1] = x;
+            points[idx1 + 1] = y;
+            points[idx1 + 2] = depth + z;
+
+            // bottom vertices
+            points[pOffset + idx1] = x;
+            points[pOffset + idx1 + 1] = y;
+            points[pOffset + idx1 + 2] = z;
+
+            uv[idx2] = x;
+            uv[idx2 + 1] = y;
+
+            uv[uOffset + idx2] = x;
+            uv[uOffset + idx2 + 1] = y;
+
+            idx1 += 3;
+            idx2 += 2;
+            j++;
+        }
+    }
+    return {
+        flatVertices,
+        holes,
+        points,
+        count,
+        uv
+    };
+
+}
+
+function validateRing(ring: PolylineType) {
+    if (!isClosedRing(ring)) {
+        ring.push(ring[0]);
+    }
+}
+
+function isClosedRing(ring: PolylineType) {
+    const len = ring.length;
+    const [x1, y1] = ring[0], [x2, y2] = ring[len - 1];
+    return (x1 === x2 && y1 === y2);
+}