ObjToSchematic/tests/normal-corrected-ray-voxeliser.test.ts

import fs from 'fs';
import path from 'path';

import { RGBAColours } from '../src/colour';
import { ObjImporter } from '../src/importers/obj_importer';
import { ASSERT } from '../src/util/error_util';
import { Vector3 } from '../src/vector';
import { NormalCorrectedRayVoxeliser } from '../src/voxelisers/normal-corrected-ray-voxeliser';
import { TEST_PREAMBLE } from './preamble';

test('Voxelise solid 2x2 cube', () => {
    TEST_PREAMBLE();

    const obj = fs.readFileSync(path.join(__dirname, './data/cube.obj'), 'utf-8');

    const importer = new ObjImporter();
    importer.parse(obj);
    const mesh = importer.toMesh();
    mesh.processMesh(0, 0, 0);

    const voxeliser = new NormalCorrectedRayVoxeliser();
    const voxelMesh = voxeliser.voxelise(mesh, {
        constraintAxis: 'y',
        size: 2,
        useMultisampleColouring: false,
        enableAmbientOcclusion: false,
        voxelOverlapRule: 'average',
        voxeliser: 'ncrb',
    });

    const expectedVoxels = [
        {
            position: new Vector3(0, 1, 0),
            colour: RGBAColours.BLUE,
        },
        {
            position: new Vector3(1, 1, 0),
            colour: RGBAColours.GREEN,
        },
        {
            position: new Vector3(1, 1, 1),
            colour: RGBAColours.RED,
        },
        {
            position: new Vector3(0, 1, 1),
            colour: RGBAColours.WHITE,
        },
        {
            position: new Vector3(1, 0, 1),
            colour: RGBAColours.YELLOW,
        },
        {
            position: new Vector3(1, 0, 0),
            colour: RGBAColours.CYAN,
        },
        {
            position: new Vector3(0, 0, 1),
            colour: RGBAColours.BLACK,
        },
        {
            position: new Vector3(0, 0, 0),
            colour: RGBAColours.MAGENTA,
        },
    ];

    for (const expected of expectedVoxels) {
        expect(voxelMesh.isVoxelAt(expected.position)).toBe(true);
        const voxel = voxelMesh.getVoxelAt(expected.position);
        expect(voxel).toBeDefined(); ASSERT(voxel);
        expect(voxel.colour.r).toBeCloseTo(expected.colour.r);
        expect(voxel.colour.g).toBeCloseTo(expected.colour.g);
        expect(voxel.colour.b).toBeCloseTo(expected.colour.b);
        expect(voxel.colour.a).toBeCloseTo(expected.colour.a);
    }
});