-
Notifications
You must be signed in to change notification settings - Fork 4
/
index.js
225 lines (194 loc) · 6.51 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
const setPoints = (points, p, i) => {
points[i++] = p[0]
points[i++] = p[1]
points[i++] = p[2] || 0
}
function CSG2Vertices (csg) {
let vLen = 0; let iLen = 0
for (const poly of csg.polygons) {
const len = poly.vertices.length
vLen += len * 3
iLen += 3 * (len - 2)
}
const vertices = new Float32Array(vLen)
const normals = new Float32Array(vLen)
const indices = vLen > 65535 ? new Uint32Array(iLen) : new Uint16Array(iLen)
let vertOffset = 0
let indOffset = 0
let posOffset = 0
let first = 0
for (const poly of csg.polygons) {
const arr = poly.vertices
const normal = calculateNormal(arr)
const len = arr.length
first = posOffset
vertices.set(arr[0], vertOffset)
normals.set(normal, vertOffset)
vertOffset += 3
vertices.set(arr[1], vertOffset)
normals.set(normal, vertOffset)
vertOffset += 3
posOffset += 2
for (let i = 2; i < len; i++) {
vertices.set(arr[i], vertOffset)
normals.set(normal, vertOffset)
indices[indOffset++] = first
indices[indOffset++] = first + i -1
indices[indOffset++] = first + i
vertOffset += 3
posOffset += 1
}
}
return { type: 'mesh', vertices, indices, normals }
}
const calculateNormal = (vertices) => {
const v0 = vertices[0]
const v1 = vertices[1]
const v2 = vertices[2]
const Ax = v1[0] - v0[0]
const Ay = v1[1] - v0[1]
const Az = v1[2] - v0[2]
const Bx = v2[0] - v0[0]
const By = v2[1] - v0[1]
const Bz = v2[2] - v0[2]
const Nx = Ay * Bz - Az * By
const Ny = Az * Bx - Ax * Bz
const Nz = Ax * By - Ay * Bx
const len = Math.hypot(Nx, Ny, Nz)
return [Nx / len, Ny / len, Nz / len]
}
function CSG2LineVertices (csg) {
let vLen = csg.points.length * 3
if (csg.isClosed) vLen += 3
const vertices = new Float32Array(vLen)
csg.points.forEach((p, idx) => setPoints(vertices, p, idx * 3))
if (csg.isClosed) {
setPoints(vertices, csg.points[0], vertices.length - 3)
}
return { type: 'line', vertices }
}
function CSGSides2LineSegmentsVertices (csg) {
const vLen = csg.sides.length * 6
const vertices = new Float32Array(vLen)
csg.sides.forEach((side, idx) => {
const i = idx * 6
setPoints(vertices, side[0], i)
setPoints(vertices, side[1], i + 3)
})
return { type: 'lines', vertices }
}
const CSGOutlines2LineSegmentsVertices = (key) => (csg) => {
const numPoints = csg[key].reduce((acc, outline) => acc + outline.length, 0)
const vLen = numPoints * 6
const vertices = new Float32Array(vLen)
let idx = 0
csg[key].forEach((outline) => {
let prev = outline[outline.length - 1]
outline.forEach((vert) => {
setPoints(vertices, prev, idx * 6)
setPoints(vertices, vert, idx * 6 + 3)
prev = vert
idx++
})
})
return { type: 'lines', vertices }
}
function CSGCached (func, data, cacheKey, transferable, unique, options) {
cacheKey = cacheKey || data
let geo = JscadToCommon.cache.get(cacheKey)
if (!geo) {
geo = func(data, options)
geo.id = JscadToCommon.sequence++
// fill transferable array for postMessage optimization
if (transferable) {
const { vertices, indices, normals } = geo
transferable.push(vertices)
if (indices) transferable.push(indices)
if (normals) transferable.push(normals)
}
JscadToCommon.cache.set(cacheKey, geo)
}
// flll unique map for exports that reuse stuff like 3mf
if (unique) unique.set(geo.id, geo)
return geo
}
/** Prepare lists of geometries gouped by type with format suitable for webgl if possible or type:unknown.
*
* @returns object separating converted geometries by type: line,lines,mesh,instance,unknown
* */
JscadToCommon.prepare = (list, transferable, useInstances) => {
const map = { line: [], lines: [], mesh: [], instance: [], unknown: [], all: [], unique: new Map() }
const instanceMap = new Map()
const add = data => {
map[data.type].push(data)
map.all.push(data)
}
const extract = list => {
list.forEach(csg => {
if (!csg) return
if (csg instanceof Array) {
extract(csg)
} else {
const obj = JscadToCommon(csg, transferable, map.unique)
// transparency in instanced mesh is problematic
// transparent objects need ordering, and that breaks thing for rendering instances
if (useInstances && obj.type === 'mesh' && obj.id && (!csg.color || csg.color.length === 3 || csg.color[3] === 1)) {
let old = instanceMap.get(obj.id)
//console.log('instance', old, obj)
if (!old) {
old = { csg, ...obj, list: [] }
instanceMap.set(obj.id, old)
}
old.list.push(csg)
} else {
add({ csg, ...obj })
}
}
})
}
if (list) extract(list)
instanceMap.forEach(data => {
if (data.list.length === 1) {
delete data.list
} else {
data.type = 'instance'
}
add(data)
})
return map
}
export function JscadToCommon (csg, transferable, unique, options) {
if (csg instanceof Array) return csg.map(csg2 => JscadToCommon(csg2, transferable, unique, options))
if (typeof csg !== 'object') throw new Error('invalid jscad geometry, not an object')
let obj
if (csg.polygons) obj = CSGCached(CSG2Vertices, csg, csg.polygons, transferable, unique, options)
if (csg.sides && !csg.points) obj = CSGCached(CSGSides2LineSegmentsVertices, csg, csg.sides, transferable, unique, options)
if (csg.outlines) obj = CSGCached(CSGOutlines2LineSegmentsVertices('outlines'), csg, csg.outlines, transferable, unique, options)
if (csg.contours) obj = CSGCached(CSGOutlines2LineSegmentsVertices('contours'), csg, csg.contours, transferable, unique, options)
if (csg.points) obj = CSGCached(CSG2LineVertices, csg, csg.points, transferable, unique, options)
if (csg.vertices) { // cover a case where the object already has the format
obj = csg
// avoid filling transferable multiple times
if (!JscadToCommon.cache.get(obj)) {
JscadToCommon.cache.set(obj, obj)
if (transferable) {
transferable.push(obj.vertices)
if (obj.indices) transferable.push(obj.indices)
}
}
}
if(csg.color || csg.transforms) obj = {...obj}
if(csg.color) obj.color = csg.color
if(csg.transforms) obj.transforms = csg.transforms
if (!obj || !obj.type) {
// throw new Error('invalid jscad geometry')
console.error('invalid jscad geometry', csg)
obj = { ...obj, csg, type: 'unknown' }
}
return obj
}
JscadToCommon.clearCache = () => {
JscadToCommon.cache = new WeakMap()
JscadToCommon.sequence = 1
}
JscadToCommon.clearCache()