| |
|
|
| import logging |
| from collections import UserDict |
| from dataclasses import dataclass |
| from typing import Iterable, Optional |
|
|
| from ..utils import maybe_prepend_base_path |
|
|
|
|
| @dataclass |
| class MeshInfo: |
| name: str |
| data: str |
| geodists: Optional[str] = None |
| symmetry: Optional[str] = None |
| texcoords: Optional[str] = None |
|
|
|
|
| class _MeshCatalog(UserDict): |
| def __init__(self, *args, **kwargs): |
| super().__init__(*args, **kwargs) |
| self.mesh_ids = {} |
| self.mesh_names = {} |
| self.max_mesh_id = -1 |
|
|
| def __setitem__(self, key, value): |
| if key in self: |
| logger = logging.getLogger(__name__) |
| logger.warning( |
| f"Overwriting mesh catalog entry '{key}': old value {self[key]}" |
| f", new value {value}" |
| ) |
| mesh_id = self.mesh_ids[key] |
| else: |
| self.max_mesh_id += 1 |
| mesh_id = self.max_mesh_id |
| super().__setitem__(key, value) |
| self.mesh_ids[key] = mesh_id |
| self.mesh_names[mesh_id] = key |
|
|
| def get_mesh_id(self, shape_name: str) -> int: |
| return self.mesh_ids[shape_name] |
|
|
| def get_mesh_name(self, mesh_id: int) -> str: |
| return self.mesh_names[mesh_id] |
|
|
|
|
| MeshCatalog = _MeshCatalog() |
|
|
|
|
| def register_mesh(mesh_info: MeshInfo, base_path: Optional[str]) -> None: |
| geodists, symmetry, texcoords = mesh_info.geodists, mesh_info.symmetry, mesh_info.texcoords |
| if geodists: |
| geodists = maybe_prepend_base_path(base_path, geodists) |
| if symmetry: |
| symmetry = maybe_prepend_base_path(base_path, symmetry) |
| if texcoords: |
| texcoords = maybe_prepend_base_path(base_path, texcoords) |
| MeshCatalog[mesh_info.name] = MeshInfo( |
| name=mesh_info.name, |
| data=maybe_prepend_base_path(base_path, mesh_info.data), |
| geodists=geodists, |
| symmetry=symmetry, |
| texcoords=texcoords, |
| ) |
|
|
|
|
| def register_meshes(mesh_infos: Iterable[MeshInfo], base_path: Optional[str]) -> None: |
| for mesh_info in mesh_infos: |
| register_mesh(mesh_info, base_path) |
|
|
