crs info for repr context

Author: valentin-gauthier-geosiris

energyml-utils/example/attic/misc_test.py

@@ -1,5 +1,6 @@
 import json
 
+from energyml.utils.epc import Epc
 from energyml.utils.epc_utils import as_dor, get_dor_uris_from_obj
 from energyml.utils.introspection import get_obj_uri, search_attribute_matching_type_with_path
 from energyml.utils.serialization import (
@@ -38,10 +39,17 @@ def test_serialize_dor(xml_path: str):
     print(json_dict)
 
     print(get_obj_uri(json_dict))
+    
+
+def load_epc_22dev3(file_path: str):
+    obj = Epc.read_file(file_path, recompute_rels=True)
+    print(obj)
 
 
 if __name__ == "__main__":
+    # Run $env:PYTHONPATH="src" if it fails to be executed from the project root.
     # test_as_uri("rc/ContinuousProperty_1d34249c-4c4f-4705-870e-b5dea9c0d78e.xml")
     # test_as_uri("rc/DiscreteProperty.xml")
-    test_serialize_dor("rc/DiscreteProperty.xml")
+    # test_serialize_dor("rc/DiscreteProperty.xml")
+    load_epc_22dev3("D:/Geosiris/Clients/Egis/Documents/Data/4 MNT Trojena/MNT_Trojena_2024-03-18 _val.epc")
 

energyml-utils/src/energyml/utils/data/representation_context.py

@@ -4,6 +4,7 @@
 import logging
 from typing import Any, Dict, List, Optional
 from energyml.opc.opc import Relationship
+from energyml.utils.data.crs import CrsInfo
 from pydantic import BaseModel, Field, ConfigDict
 
 from energyml.utils.uri import Uri
@@ -58,12 +59,17 @@ class RepresentationContext(BaseModel):
     uri: Uri = Field(default="")
 
     crs: List[Any] = Field(default_factory=list)
+    crs_infos: List[CrsInfo] = Field(default_factory=list)
+    
     rels: List[Relationship] = Field(default_factory=list)
 
     # Graphical information keyed by GraphicalInformationSet uri → list of entries
     graphical_info: dict = Field(default_factory=dict)
 
     time_series: list = Field(default_factory=list)
+    
+    _projected_uom: Optional[str] = None
+    _vertical_uom: Optional[str] = None
 
     def __init__(self, obj: Any, workspace: EnergymlStorageInterface, **data):
         super().__init__(obj=obj, workspace=workspace, uri=get_obj_uri(obj), **data)
@@ -127,13 +133,19 @@ def _collect_properties(self, rels: List[Relationship]):
     def _collect_crs(self):
         # Collect related CRS objects referenced by the representation
         self.crs = []
+        self.crs_infos = []
+        crs_uuids = set()
         crs_dors = search_attribute_matching_name(self.obj, r"\.*Crs", search_in_sub_obj=True, deep_search=False)
         if crs_dors is not None and len(crs_dors) > 0:
             for crs_ref in crs_dors:
                 if crs_ref is not None:
                     crs = self.workspace.get_object(get_obj_uri(crs_ref))
                     if crs is not None:
-                        self.crs.append(crs)
+                        self.crs.append(crs) # always add to keep crs for each patch
+                        crs_uuid = getattr(crs, "uuid", None)
+                        if crs_uuid is not None and crs_uuid not in crs_uuids:
+                            crs_uuids.add(crs_uuid)
+                            self.crs_infos.append(CrsInfo.from_crs_object(crs, self.workspace))
                     else:
                         logging.warning(f"CRS {get_obj_uri(crs_ref)} not found in workspace")
 
@@ -160,6 +172,21 @@ def get_property(self, property_uuid: str) -> Optional[Any]:
         """Return the property object with the given uuid, or None."""
         return self._props.get(property_uuid)
 
+    @property
+    def projected_uom(self) -> Optional[str]:
+        """Return the projected unit of measure (e.g. "m") if available from CRS info, or None."""
+        for ci in self.crs_infos:
+            if ci.projected_uom is not None:
+                return ci.projected_uom
+        return None
+    @property
+    def vertical_uom(self) -> Optional[str]:
+        """Return the vertical unit of measure (e.g. "m") if available from CRS info, or None."""
+        for ci in self.crs_infos:
+            if ci.vertical_uom is not None:
+                return ci.vertical_uom
+        return None
+    
     @property
     def properties(self) -> Dict[str, Any]:
         """Return a dict of all properties keyed by property uuid."""