Jerry
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
"""Selector-manifest entryKind stamping in the GLB STEP_topology writer."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import unittest
|
|
|
|
from cadpy.glb import _GlbBuilder
|
|
from cadpy.glb_topology import STEP_TOPOLOGY_EXTENSION
|
|
from cadpy.selector_types import SelectorBundle
|
|
|
|
|
|
def _decode_view(builder: _GlbBuilder, view_index: int) -> dict[str, object]:
|
|
views = builder.json["bufferViews"]
|
|
assert isinstance(views, list)
|
|
view = views[view_index]
|
|
start = int(view["byteOffset"])
|
|
length = int(view["byteLength"])
|
|
return json.loads(bytes(builder.binary[start : start + length]).decode("utf-8"))
|
|
|
|
|
|
def _bundle_manifest(*, assembly: bool) -> dict[str, object]:
|
|
manifest: dict[str, object] = {
|
|
"cadPath": "tmp/sample",
|
|
"stepPath": "tmp/sample.step",
|
|
"stepHash": "hash-123",
|
|
"sourceKind": "step",
|
|
"sourcePath": "tmp/sample.step",
|
|
"bbox": {"min": [0.0, 0.0, 0.0], "max": [1.0, 1.0, 1.0]},
|
|
"stats": {"occurrenceCount": 3, "leafOccurrenceCount": 2},
|
|
}
|
|
if assembly:
|
|
manifest["assembly"] = {"id": "o1"}
|
|
return manifest
|
|
|
|
|
|
class GlbSelectorManifestEntryKindTests(unittest.TestCase):
|
|
def _write_step_topology(self, manifest: dict[str, object]) -> tuple[dict[str, object], dict[str, object]]:
|
|
builder = _GlbBuilder()
|
|
builder.add_step_topology(SelectorBundle(manifest=manifest), include_selector_topology=True)
|
|
extensions = builder.json["extensions"]
|
|
assert isinstance(extensions, dict)
|
|
extension = extensions[STEP_TOPOLOGY_EXTENSION]
|
|
selector_manifest = _decode_view(builder, extension["selectorView"])
|
|
return extension, selector_manifest
|
|
|
|
def test_assembly_manifest_stamps_selector_entry_kind(self) -> None:
|
|
extension, selector_manifest = self._write_step_topology(_bundle_manifest(assembly=True))
|
|
|
|
self.assertEqual("assembly", extension["entryKind"])
|
|
self.assertEqual("assembly", selector_manifest["entryKind"])
|
|
self.assertNotIn("assembly", selector_manifest)
|
|
|
|
def test_part_manifest_stamps_selector_entry_kind(self) -> None:
|
|
extension, selector_manifest = self._write_step_topology(_bundle_manifest(assembly=False))
|
|
|
|
self.assertEqual("part", extension["entryKind"])
|
|
self.assertEqual("part", selector_manifest["entryKind"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|