geode_surface_mesh2d.py

# Standard library imports
from __future__ import annotations

# Third party imports
import opengeode as og
import opengeode_geosciences as og_geosciences
import opengeode_inspector as og_inspector

# Local application imports
from .geode_vertex_set import GeodeVertexSet
from .types import ViewerElementsType


class GeodeSurfaceMesh2D(GeodeVertexSet):
    surface_mesh: og.SurfaceMesh2D

    def __init__(self, surface_mesh: og.SurfaceMesh2D | None = None) -> None:
        self.surface_mesh = (
            surface_mesh if surface_mesh is not None else og.SurfaceMesh2D.create()
        )
        super().__init__(self.surface_mesh)

    @classmethod
    def is_3D(cls) -> bool:
        return False

    @classmethod
    def is_viewable(cls) -> bool:
        return True

    @classmethod
    def viewer_elements_type(cls) -> ViewerElementsType:
        return "polygons"

    def builder(self) -> og.SurfaceMeshBuilder2D:
        return og.SurfaceMeshBuilder2D.create(self.surface_mesh)

    def inspect(self) -> og_inspector.SurfaceInspectionResult:
        return og_inspector.inspect_surface2D(self.surface_mesh)

    def assign_crs(
        self, crs_name: str, info: og_geosciences.GeographicCoordinateSystemInfo
    ) -> None:
        builder = self.builder()
        og_geosciences.assign_surface_mesh_geographic_coordinate_system_info2D(
            self.surface_mesh, builder, crs_name, info
        )

    def convert_crs(
        self, crs_name: str, info: og_geosciences.GeographicCoordinateSystemInfo
    ) -> None:
        builder = self.builder()
        og_geosciences.convert_surface_mesh_coordinate_reference_system2D(
            self.surface_mesh, builder, crs_name, info
        )

    def create_crs(
        self, crs_name: str, input: og.CoordinateSystem2D, output: og.CoordinateSystem2D
    ) -> None:
        builder = self.builder()
        og.create_surface_mesh_coordinate_system2D(
            self.surface_mesh, builder, crs_name, input, output
        )

    def polygon_attribute_manager(self) -> og.AttributeManager:
        return self.surface_mesh.polygon_attribute_manager()

    def texture_manager(self) -> og.TextureManager2D:
        return self.surface_mesh.texture_manager()