Tree-Generator/worker.js at main · AddisonPrairie/Tree-Generator · GitHub

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
importScripts("./wasm.js");

let module = null;

onmessage = async (e) => {

    const ret = e.data[0];

    switch (ret.type) {
        case "INIT":
            //console.warn("initializing WASM module in worker thread...");

            module = await Module();
            await module.ready;

            postMessage({type: "INITIALIZED"});

            break;

        case "INITIALIZE_TREE":
            //console.warn("initializing tree in worker thread...");

            //await verifyModule();

            initializeTree(ret.args);

            break;

        case "SET_SETTINGS":
            //console.warn("updating settings in worker thread...");

            //await verifyModule();

            setSettings(ret.args)

            break;

        case "STEP":
            //console.warn("calling step in worker thread...")

            //await verifyModule();

            step();

            postMessage({type: "FINISHED_STEP"});

            break;

        case "GET_MESH":
            //console.warn("calling get mesh in worker thread...");

            //await verifyModule();

            const res = getMesh();

            postMessage({type: "MESH", arr: res});

            break;

        default:
            console.error("unrecognized call type: " + ret.type)
    }
};

function initializeTree(argArr) {
    module.ccall("init",
        null,
        ["number", "number", "number"],
        argArr);
}

function setSettings(argArr) {
    module.ccall(
        "setSettings",
        null,
        ["number", "number", "number", "number", "number", "number", "number",
         "number", "number", "number", "number", "number", "number", "number",
         "number", "number"],
        argArr
    );
}

function step() {
    module.ccall("step");
}

function getMesh() {
    const ptr = module.ccall("getTreeMesh");
    const numEdges = module.HEAP32[ptr / 4];

    return module.HEAPF32.slice(
        ptr / 4 + 1, ptr / 4 + 1 + numEdges * 12
    );
}