feat(api) : working on read_hydro_matrices

AntaresSimulatorTeam · Jan 10, 2025 · 0b7ba89 · 0b7ba89
1 parent b0c1045
commit 0b7ba89
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 0 deletions.
diff --git a/src/antares/craft/model/area.py b/src/antares/craft/model/area.py
@@ -375,3 +375,18 @@ def read_hydro(
         self,
     ) -> Hydro:
         return self._area_service.read_hydro(self.id)
+
+    def get_maxpower(self) -> pd.DataFrame:
+        return self._area_service.get_reservoir()
+
+    def get_reservoir(self) -> pd.DataFrame:
+        return self._area_service.get_reservoir()
+
+    def get_inflow_pattern(self) -> pd.DataFrame:
+        return self._area_service.get_inflow_pattern()
+
+    def get_credit_modulations(self) -> pd.DataFrame:
+        return self._area_service.get_credit_modulations()
+
+    def get_water_values(self) -> pd.DataFrame:
+        return self._area_service.get_water_values()
diff --git a/src/antares/craft/service/api_services/area_api.py b/src/antares/craft/service/api_services/area_api.py
@@ -605,3 +605,18 @@ def read_areas(self) -> List[Area]:
         # sort area list to ensure reproducibility
         area_list.sort(key=lambda area: area.id)
         return area_list
+
+    def get_maxpower(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_reservoir(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_inflow_pattern(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_water_values(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_credit_modulations(self) -> pd.DataFrame:
+        raise NotImplementedError
diff --git a/src/antares/craft/service/base_services.py b/src/antares/craft/service/base_services.py
@@ -314,6 +314,26 @@ def read_areas(self) -> list[Area]:
         """
         pass
 
+    @abstractmethod
+    def get_maxpower(self) -> pd.DataFrame:
+        pass
+
+    @abstractmethod
+    def get_reservoir(self) -> pd.DataFrame:
+        pass
+
+    @abstractmethod
+    def get_inflow_pattern(self) -> pd.DataFrame:
+        pass
+
+    @abstractmethod
+    def get_credit_modulations(self) -> pd.DataFrame:
+        pass
+
+    @abstractmethod
+    def get_water_values(self) -> pd.DataFrame:
+        pass
+
 
 class BaseLinkService(ABC):
     @abstractmethod

diff --git a/src/antares/craft/service/local_services/area_local.py b/src/antares/craft/service/local_services/area_local.py
@@ -388,3 +388,18 @@ def read_areas(self) -> List[Area]:
                     )
                 )
         return areas
+
+    def get_maxpower(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_reservoir(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_inflow_pattern(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_water_values(self) -> pd.DataFrame:
+        raise NotImplementedError
+
+    def get_credit_modulations(self) -> pd.DataFrame:
+        raise NotImplementedError