| 63 | |
| 64 | @dataclass |
| 65 | class AttributeCodecs: |
| 66 | means: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_png_16bit", "_decompress_png_16bit")) |
| 67 | scales: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_factorized_ans", "_decompress_factorized_ans")) |
| 68 | quats: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_factorized_ans", "_decompress_factorized_ans")) |
| 69 | opacities: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_png", "_decompress_png")) |
| 70 | sh0: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_png", "_decompress_png")) |
| 71 | shN: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_masked_kmeans", "_decompress_masked_kmeans")) |
| 72 | |
| 73 | def to_dict(self) -> Dict[str, Dict[str, str]]: |
| 74 | return { |
| 75 | attr: {"encode": getattr(self, attr).encode, "decode": getattr(self, attr).decode} |
| 76 | for attr in ["means", "scales", "quats", "opacities", "sh0", "shN"] |
| 77 | } |
| 78 | |
| 79 | |
| 80 | def default_qp_values() -> Dict[str, Union[int, Dict[str, Any]]]: |
no test coverage detected