72 lines
2.5 KiB
JavaScript
72 lines
2.5 KiB
JavaScript
import assert from "node:assert/strict";
|
|
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import { pathToFileURL } from "node:url";
|
|
import test from "node:test";
|
|
import ts from "typescript";
|
|
|
|
const repoRoot = path.resolve(import.meta.dirname, "../../..");
|
|
const temporary = fs.mkdtempSync(path.join(os.tmpdir(), "weight-paint-unit-"));
|
|
const sourcePath = path.join(repoRoot, "web/protocol/weight-paint.ts");
|
|
const outputPath = path.join(temporary, "weight-paint.mjs");
|
|
const transpiled = ts.transpileModule(fs.readFileSync(sourcePath, "utf8"), {
|
|
compilerOptions: { module: ts.ModuleKind.ES2022, target: ts.ScriptTarget.ES2022 },
|
|
fileName: sourcePath,
|
|
reportDiagnostics: true,
|
|
});
|
|
assert.deepEqual(transpiled.diagnostics, []);
|
|
fs.writeFileSync(outputPath, transpiled.outputText);
|
|
const weightPaint = await import(pathToFileURL(outputPath));
|
|
|
|
const vertices = [
|
|
{ index: 0, position: [-1, 0, 0], influences: [{ group: "Root", weight: 1 }, { group: "Tip", weight: 0.25 }] },
|
|
{ index: 1, position: [1, 0, 0], influences: [{ group: "Root", weight: 1 }, { group: "Tip", weight: 0.25 }] },
|
|
{ index: 2, position: [0, 1, 0], influences: [{ group: "Root", weight: 0.25 }, { group: "Tip", weight: 1 }] },
|
|
];
|
|
|
|
test("M9-12 normalizes after limiting influences", () => {
|
|
const result = weightPaint.applyWeightPaintPatch(vertices, {
|
|
schemaVersion: 1,
|
|
vertexGroup: "WebPaintGroup",
|
|
indices: [2],
|
|
values: [0.5],
|
|
normalize: true,
|
|
limit: 2,
|
|
});
|
|
assert.deepEqual(result[2].influences, [
|
|
{ group: "Tip", weight: 2 / 3 },
|
|
{ group: "WebPaintGroup", weight: 1 / 3 },
|
|
]);
|
|
});
|
|
|
|
test("M9-12 mirrors onto a reciprocal verified coordinate map", () => {
|
|
const result = weightPaint.applyWeightPaintPatch(vertices, {
|
|
schemaVersion: 1,
|
|
vertexGroup: "WebPaintGroup",
|
|
indices: [0],
|
|
values: [0.8],
|
|
mirror: true,
|
|
mirrorAxis: 0,
|
|
mirrorTolerance: 1e-4,
|
|
});
|
|
assert.equal(result[0].influences.at(-1).weight, 0.8);
|
|
assert.equal(result[1].influences.at(-1).weight, 0.8);
|
|
});
|
|
|
|
test("M9-12 rejects an unverified symmetry map and invalid limits", () => {
|
|
assert.throws(() => weightPaint.parseWeightPaintOptions({ limit: 33 }), /PAINT_SCHEMA_INVALID/);
|
|
assert.throws(() => weightPaint.applyWeightPaintPatch([
|
|
...vertices,
|
|
{ index: 3, position: [4, 0, 0], influences: [] },
|
|
], {
|
|
schemaVersion: 1,
|
|
vertexGroup: "WebPaintGroup",
|
|
indices: [0],
|
|
values: [0.8],
|
|
mirror: true,
|
|
mirrorAxis: 0,
|
|
mirrorTolerance: 1e-4,
|
|
}), /WEIGHT_MIRROR_SYMMETRY_UNVERIFIED/);
|
|
});
|