not working

master
howard 2021-03-24 10:49:40 -07:00
parent e23c38c651
commit 0d67c75d64
4 changed files with 116 additions and 103 deletions

2
dist/bundle.js vendored

File diff suppressed because one or more lines are too long

BIN
dist/solver.wasm vendored

Binary file not shown.

View File

@ -70,6 +70,8 @@ export class Sketcher extends THREE.Group {
this.colorPt = new THREE.Color('white') this.colorPt = new THREE.Color('white')
this.selected = new Set() this.selected = new Set()
this.geomGroup = new THREE.Group();
scene.add(this.geomGroup);
this.lineMaterial = new THREE.LineBasicMaterial({ this.lineMaterial = new THREE.LineBasicMaterial({
linewidth: 3, linewidth: 3,
@ -113,13 +115,13 @@ export class Sketcher extends THREE.Group {
this.objIdx = new Map() this.objIdx = new Map()
this.max_pts = 1000 this.max_pts = 1000
this.ptsBuf = new Float32Array(this.max_pts * 2) this.ptsBuf = new Float32Array(this.max_pts * 2).fill(NaN)
this.max_links = 1000 this.max_links = 1000
this.linksBuf = new Int32Array(this.max_links * 5) // [0]:type, [1]:pt1, [2]:pt2, [3]:pt3, [4]:pt4 this.linksBuf = new Float32Array(this.max_links * 5).fill(NaN) // [0]:type, [1]:pt1, [2]:pt2, [3]:pt3, [4]:pt4
this.max_constraints = 1000 this.max_constraints = 1000
this.constraintsBuf = new Float32Array(this.max_constraints * 6) // [0]:type, [1]:pt1, [2]:pt2, [3]:lk1, [4]:lk2, [5]:val this.constraintsBuf = new Float32Array(this.max_constraints * 6).fill(NaN) // [0]:type, [1]:val, [2]:pt1, [3]:pt2, [4]:lk1, [5]:lk2
this.subsequent = false; this.subsequent = false;
this.ptsBufPt = 0; this.ptsBufPt = 0;
@ -194,7 +196,7 @@ export class Sketcher extends THREE.Group {
this.camera this.camera
); );
const hoverPts = this.raycaster.intersectObjects(this.children) const hoverPts = this.raycaster.intersectObjects(this.geomGroup.children)
if (hoverPts.length) { if (hoverPts.length) {
let minDist = Infinity; let minDist = Infinity;
@ -228,7 +230,7 @@ export class Sketcher extends THREE.Group {
if (this.hovered) { if (this.hovered) {
this.selected.add(this.hovered) this.selected.add(this.hovered)
if (this.hovered.type === "Points") { if (this.hovered.type === "Points") {
this.grabPtIdx = this.children.indexOf( this.grabPtIdx = this.geomGroup.children.indexOf(
this.hovered this.hovered
) )
this.domElement.addEventListener('pointermove', this.onDrag); this.domElement.addEventListener('pointermove', this.onDrag);
@ -245,7 +247,7 @@ export class Sketcher extends THREE.Group {
onDrag(e) { onDrag(e) {
const mouseLoc = this.getLocation(e); const mouseLoc = this.getLocation(e);
this.children[this.grabPtIdx].geometry.attributes.position.set(mouseLoc); this.geomGroup.children[this.grabPtIdx].geometry.attributes.position.set(mouseLoc);
this.solve() this.solve()
this.dispatchEvent({ type: 'change' }) this.dispatchEvent({ type: 'change' })
} }
@ -254,22 +256,22 @@ export class Sketcher extends THREE.Group {
onRelease() { onRelease() {
this.domElement.removeEventListener('pointermove', this.onDrag) this.domElement.removeEventListener('pointermove', this.onDrag)
this.domElement.removeEventListener('pointerup', this.onRelease) this.domElement.removeEventListener('pointerup', this.onRelease)
this.children[this.grabPtIdx].geometry.computeBoundingSphere() this.geomGroup.children[this.grabPtIdx].geometry.computeBoundingSphere()
// this.grabbedObject = null // this.grabbedObject = null
} }
deleteSelected() { deleteSelected() {
let minI = this.children.length; let minI = this.geomGroup.children.length;
for (let obj of this.selected) { for (let obj of this.selected) {
minI = Math.min(minI, this.delete(obj)) minI = Math.min(minI, this.delete(obj))
} }
this.updateBuffer(minI) this.updatePointsBuffer(minI)
this.resetConstraints() this.updateOtherBuffers()
this.selected.clear() this.selected.clear()
this.dispatchEvent({ type: 'change' }) this.dispatchEvent({ type: 'change' })
@ -283,13 +285,13 @@ export class Sketcher extends THREE.Group {
this.constraints.delete(c_id) this.constraints.delete(c_id)
} }
resetConstraints() { updateOtherBuffers() {
let i = 0 let i = 0
for (let [key,obj] of this.constraints) { for (let [key, obj] of this.constraints) {
this.constraintsBuf.set( this.constraintsBuf.set(
[ [
this.contraintNum[obj[1]], obj[2],
...obj[0].map(ele => this.objIdx.get(ele.id) || -1), ...obj[0].map(ele => this.objIdx.get(ele.id) || -1),
this.contraintNum[obj[4]], obj[5]
], ],
(i) * 6 (i) * 6
) )
@ -297,14 +299,16 @@ export class Sketcher extends THREE.Group {
} }
i = 0; i = 0;
for (let [key,obj] of this.linkedObjs) { for (let [key, obj] of this.linkedObjs) {
console.log(obj[0])
this.linksBuf.set( this.linksBuf.set(
[ [
...obj[0].map(ele => this.objIdx.get(ele.id) || -1), this.linkNum[obj[1]],
this.linkNum[obj[4]] ...obj[0].map(ele => this.objIdx.get(ele.id) ?? -1),
], ],
(i) * 5 (i) * 5
) )
console.log(this.linksBuf)
i++ i++
} }
@ -314,11 +318,11 @@ export class Sketcher extends THREE.Group {
const link = this.linkedObjs.get(obj.l_id) const link = this.linkedObjs.get(obj.l_id)
let i = this.children.indexOf(link[0][0]) let i = this.geomGroup.children.indexOf(link[0][0])
if (i == -1) return Infinity if (i == -1) return Infinity
for (let j = 0; j < link[0].length; j++) { for (let j = 0; j < link[0].length; j++) {
const obj = this.children[i + j] const obj = this.geomGroup.children[i + j]
obj.geometry.dispose() obj.geometry.dispose()
obj.material.dispose() obj.material.dispose()
@ -327,16 +331,16 @@ export class Sketcher extends THREE.Group {
} }
} }
this.children.splice(i, link[0].length) this.geomGroup.children.splice(i, link[0].length)
this.linkedObjs.delete(obj.l_id) this.linkedObjs.delete(obj.l_id)
return i return i
} }
updateBuffer(startingIdx) { updatePointsBuffer(startingIdx=0) {
for (let i = startingIdx; i < this.children.length; i++) { for (let i = startingIdx; i < this.geomGroup.children.length; i++) {
const obj = this.children[i] const obj = this.geomGroup.children[i]
this.objIdx.set(obj.id, i) this.objIdx.set(obj.id, i)
if (obj.type == "Points") { if (obj.type == "Points") {
this.ptsBuf[2 * i] = obj.geometry.attributes.position.array[0] this.ptsBuf[2 * i] = obj.geometry.attributes.position.array[0]
@ -354,7 +358,7 @@ export class Sketcher extends THREE.Group {
this.domElement.removeEventListener('pointermove', this.beforeClick_2); this.domElement.removeEventListener('pointermove', this.beforeClick_2);
this.domElement.removeEventListener('pointerdown', this.onClick_2); this.domElement.removeEventListener('pointerdown', this.onClick_2);
this.delete(this.children[this.children.length - 1]) this.delete(this.geomGroup.children[this.geomGroup.children.length - 1])
this.dispatchEvent({ type: 'change' }) this.dispatchEvent({ type: 'change' })
this.subsequent = false this.subsequent = false
@ -413,23 +417,15 @@ export class Sketcher extends THREE.Group {
if (this.subsequent) { if (this.subsequent) {
this.constraintsBuf.set(
[
this.children.length - 2, this.children.length, -1, -1,
this.contraintNum['coincident'], -1
],
(this.constraints.size) * 6
)
this.constraints.set(this.c_id, this.constraints.set(this.c_id,
[ [
[this.children[this.children.length - 2], this.p1, -1, -1], [this.geomGroup.children[this.geomGroup.children.length - 2], this.p1, -1, -1],
'coincident', -1 'coincident', -1
] ]
) )
this.p1.constraints.add(this.c_id) this.p1.constraints.add(this.c_id)
this.children[this.children.length - 2].constraints.add(this.c_id) this.geomGroup.children[this.geomGroup.children.length - 2].constraints.add(this.c_id)
this.c_id += 1 this.c_id += 1
} }
@ -455,21 +451,8 @@ export class Sketcher extends THREE.Group {
} }
let ptr = this.children.length this.updatePoint = this.geomGroup.children.length
this.add(...toPush) this.geomGroup.add(...toPush)
this.updateBuffer(ptr)
const link_entry = new Array(5).fill(-1)
for (let i = ptr, j = 0; j < toPush.length - 1;) {
link_entry[j++] = i++
}
link_entry[link_entry.length - 1] = this.linkNum[this.mode]
this.linksBuf.set(
link_entry,
(this.linkedObjs.size) * 5
)
this.linkedObjs.set(this.l_id, [toPush, this.mode]) this.linkedObjs.set(this.l_id, [toPush, this.mode])
@ -479,7 +462,6 @@ export class Sketcher extends THREE.Group {
this.l_id += 1 this.l_id += 1
this.domElement.removeEventListener('pointerdown', this.onClick_1) this.domElement.removeEventListener('pointerdown', this.onClick_1)
this.domElement.addEventListener('pointermove', this.beforeClick_2) this.domElement.addEventListener('pointermove', this.beforeClick_2)
this.domElement.addEventListener('pointerdown', this.onClick_2) this.domElement.addEventListener('pointerdown', this.onClick_2)
@ -518,13 +500,18 @@ export class Sketcher extends THREE.Group {
if (e.buttons !== 1) return; if (e.buttons !== 1) return;
this.domElement.removeEventListener('pointermove', this.beforeClick_2); this.domElement.removeEventListener('pointermove', this.beforeClick_2);
this.domElement.removeEventListener('pointerdown', this.onClick_2); this.domElement.removeEventListener('pointerdown', this.onClick_2);
if (this.mode == "line") { if (this.mode == "line") {
this.updatePointsBuffer(this.updatePoint)
this.updateOtherBuffers()
this.subsequent = true this.subsequent = true
this.onClick_1(e) this.onClick_1(e)
} else if (this.mode == "arc") { } else if (this.mode == "arc") {
// this.domElement.addEventListener('pointermove', this.beforeClick_3) // this.domElement.addEventListener('pointermove', this.beforeClick_3)
} }
} }
@ -539,48 +526,45 @@ export class Sketcher extends THREE.Group {
solve() { solve() {
// for (let i = 0, p = 0; i < this.geomGroup.children.length; i++) {
// this.ptsBuf[p++] = this.geomGroup.children[i].geometry.attributes.position.array[0]
// this.ptsBuf[p++] = this.geomGroup.children[i].geometry.attributes.position.array[1]
// }
this.updatePointsBuffer()
for (let i = 0, p = 0; i < this.children.length; i++) { const pts_buffer = Module._malloc(this.ptsBuf.length * this.ptsBuf.BYTES_PER_ELEMENT)
this.ptsBuf[p++] = this.children[i].geometry.attributes.position.array[0] Module.HEAPF32.set(this.ptsBuf, pts_buffer >> 2)
this.ptsBuf[p++] = this.children[i].geometry.attributes.position.array[1]
}
buffer = Module._malloc(this.ptsBuf.length * this.ptsBuf.BYTES_PER_ELEMENT) const constraints_buffer = Module._malloc(this.constraintsBuf.length * this.constraintsBuf.BYTES_PER_ELEMENT)
Module.HEAPF32.set(this.ptsBuf, buffer >> 2) Module.HEAPF32.set(this.constraintsBuf, constraints_buffer >> 2)
const links_buffer = Module._malloc(this.linksBuf.length * this.linksBuf.BYTES_PER_ELEMENT)
Module.HEAPF32.set(this.linksBuf, links_buffer >> 2)
Module["_solver"](this.geomGroup.children.length, pts_buffer, this.constraints.size, constraints_buffer, this.linkedObjs.size, links_buffer)
let ptr = pts_buffer >> 2;
Module["_solver"](this.children.length / 2, buffer) for (let i = 0; i < this.geomGroup.children.length; i += 1) {
const pos = this.geomGroup.children[i].geometry.attributes.position;
if (isNaN(Module.HEAPF32[ptr])) {
pos.array[0] = Module.HEAPF32[ptr-4]
pos.array[1] = Module.HEAPF32[ptr-3]
pos.array[3] = Module.HEAPF32[ptr-2]
pos.array[4] = Module.HEAPF32[ptr-1]
} else {
pos.array[0] = Module.HEAPF32[ptr++]
pos.array[1] = Module.HEAPF32[ptr++]
}
let ptr = buffer >> 2; pos.needsUpdate = true;
for (let i = 0; i < this.linkedObjs.lengthxx; i += 1) {
const pt1_pos = this.children[i >> 1].geometry.attributes.position;
const pt2_pos = this.children[(i >> 1) + 1].geometry.attributes.position;
const line_pos = this.linesArr[i >> 2].geometry.attributes.position;
pt1_pos.array[0] = Module.HEAPF32[ptr]
line_pos.array[0] = Module.HEAPF32[ptr++]
pt1_pos.array[1] = Module.HEAPF32[ptr]
line_pos.array[1] = Module.HEAPF32[ptr++]
pt2_pos.array[0] = Module.HEAPF32[ptr]
line_pos.array[3] = Module.HEAPF32[ptr++]
pt2_pos.array[1] = Module.HEAPF32[ptr]
line_pos.array[4] = Module.HEAPF32[ptr++]
pt1_pos.needsUpdate = true;
pt2_pos.needsUpdate = true;
line_pos.needsUpdate = true;
} }
this.dispatchEvent({ type: 'change' }) this.dispatchEvent({ type: 'change' })
Module._free(buffer) Module._free(pts_buffer)
} }
} }

View File

@ -12,7 +12,7 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <stdint.h> #include <stdint.h>
#include <math.h>
#include "slvs.h" #include "slvs.h"
static Slvs_System sys; static Slvs_System sys;
@ -33,11 +33,9 @@ static void *CheckMalloc(size_t n)
* entities and constraints. * entities and constraints.
*---------------------------------------------------------------------------*/ *---------------------------------------------------------------------------*/
int solver(int nLines, float *ptr) int solver(int nPts, float *p_ptr, int nConst, float *c_ptr, int nLinks, float *l_ptr)
{ {
// printf("first %i \n", (int)*(l_ptr + 1));
float *buf_pt_start = ptr;
Slvs_hGroup g; Slvs_hGroup g;
double qw, qx, qy, qz; double qw, qx, qy, qz;
@ -67,46 +65,77 @@ int solver(int nLines, float *ptr)
/* These points are represented by their coordinates (u v) within the /* These points are represented by their coordinates (u v) within the
* workplane, so they need only two parameters each. */ * workplane, so they need only two parameters each. */
int p_start = sys.params; Slvs_hParam ph = 11;
Slvs_hParam ph = 11, ph_s = 11;
Slvs_hParam vh = 301, vh_s = 301; Slvs_hParam vh = 301, vh_s = 301;
Slvs_hParam lh = 400, lh_s = 400; Slvs_hParam lh = 400, lh_s = 400;
Slvs_hParam con_id = 1; Slvs_hParam con_id = 1;
for (int i = 0; i < nLines * 2; i++) float *buf_pt_start = p_ptr;
int p_start = sys.params;
for (int i = 0; i < nPts; i++)
{ {
printf("i: %i %f %f \n", i,(float)*p_ptr,(float)*(p_ptr+1));
if (isnan((float)*p_ptr))
{
p_ptr+=2;
continue;
}
sys.param[sys.params++] = Slvs_MakeParam(ph++, g, (float)*p_ptr++);
sys.param[sys.params++] = Slvs_MakeParam(ph++, g, (float)*p_ptr++);
sys.entity[sys.entities++] = Slvs_MakePoint2d(i, g, 200, ph - 1, ph - 2);
}
sys.param[sys.params++] = Slvs_MakeParam(ph++, g, (float)*ptr++); for (int i = 0; i < nLinks; i++)
sys.param[sys.params++] = Slvs_MakeParam(ph++, g, (float)*ptr++); {
sys.entity[sys.entities++] = Slvs_MakePoint2d(vh++, g, 200, ph - 1, ph - 2); if (*l_ptr++ == 0)
if (i % 2 == 1)
{ {
sys.entity[sys.entities++] = Slvs_MakeLineSegment(lh++, g, sys.entity[sys.entities++] = Slvs_MakeLineSegment(lh++, g,
200, vh - 1, vh - 2); 200, (int)*l_ptr++, (int)*l_ptr++);
l_ptr += 2;
} else {
l_ptr += 4;
} }
else if (i > 0) }
printf("nconst: %i \n", nConst);
for (int i = 0; i < nConst; i++)
{
if ((int)*c_ptr == 0)
{ {
c_ptr+=2;
printf("const: %i %i \n", (int)*c_ptr, (int)*(c_ptr+1));
sys.constraint[sys.constraints++] = Slvs_MakeConstraint( sys.constraint[sys.constraints++] = Slvs_MakeConstraint(
con_id++, g, con_id++, g,
SLVS_C_POINTS_COINCIDENT, SLVS_C_POINTS_COINCIDENT,
200, 200,
0.0, 0.0,
vh - 2, vh - 1, 0, 0); (int)*c_ptr++, (int)*c_ptr++, 0, 0);
c_ptr += 2;
} else {
c_ptr += 6;
} }
} }
/* And solve. */ /* And solve. */
Slvs_Solve(&sys, g); Slvs_Solve(&sys, g);
printf("npts: %i \n", nPts);
if (sys.result == SLVS_RESULT_OKAY) if (sys.result == SLVS_RESULT_OKAY)
{ {
// printf("solved okay\n"); // printf("solved okay\n");
for (int i = 0; i < nLines * 4; i++) for (int i = 0; i < nPts; i++)
{ {
if (isnan((float)*buf_pt_start))
{
buf_pt_start+=2;
continue;
}
printf("res: %i %f %f \n",i, (float)sys.param[p_start].val, (float)sys.param[p_start+1].val);
*buf_pt_start++ = (float)sys.param[p_start++].val;
*buf_pt_start++ = (float)sys.param[p_start++].val; *buf_pt_start++ = (float)sys.param[p_start++].val;
} }
} }