wolfhece 2.2.38__py3-none-any.whl → 2.2.40__py3-none-any.whl

wolfhece/hydrology/Catchment.py

@@ -193,10 +193,7 @@ class Catchment:
             sys.exit()
 
         # Get the number of subbasins
-        try:
-            self.nbSubBasin = int(self.paramsInput.myparams['Semi distributed model']['How many?'][key_Param.VALUE]) + 1 # +1 because the outlet is also counted
-        except:
-            self.nbSubBasin = int(self.paramsInput.myparams_default['Semi distributed model']['How many?'][key_Param.VALUE]) + 1 # +1 because the outlet is also counted
+        self.nbSubBasin = int(self.paramsInput[('Semi distributed model', 'How many?')]) + 1 # +1 because the outlet is also counted
 
         # Fill the dictionary containing the id of the sorted subbasin returned by the Fortran code
         self.init_dictIdConversion(self.workingDir)
@@ -217,7 +214,7 @@ class Catchment:
         # TO DO: Check how the rain is read for the first time
 
         # Get the hydrology model used (1-linear reservoir, 2-VHM, 3-Unit Hydrograph)
-        self.myModel = int(self.paramsInput.myparams['Model Type']['Type of hydrological model'][key_Param.VALUE])
+        self.myModel = self.paramsInput[('Model Type', 'Type of hydrological model')]
 
         # Save the stations SPW characteristics in a dictionnary
         self.read_measuring_stations_SPW()
@@ -225,123 +222,128 @@ class Catchment:
 
 
 
+        try:
 
+            # Construction of the Catchment
+            # ------------------------------
+
+            # 1) 1st Iteration: Object creation
+
+            # Iterate through the Input params dictionnary
+            self.create_ObjectsInCatchment()
+
+            self.charact_watrshd.set_names_subbasins([(cur.iDSorted, cur.name) for cur in self.subBasinDict.values()])
+
+            # self.add_hyetoToDict()
+
+
+            # 2) 2nd Iteration: Link between objects
+            self.link_objects()   # This procedure also creates the first layer of the topo tree by identifying the source ss-basins
+
+            """
+            The topo tree is organised by level:
+            - The first level contains should only contain subbasins which don't have any input flows.
+            Therefore, they already contains all the information to build their hydrograph
+            - The second and upper levels can contain either RB or subbasins with input flows.
+
+            """
+
+            # 3) 3rd Iteration: Complete the tree
+            self.complete_topoDict()
+            if(not(self.plotNothing)):
+                flowchart = graphviz.Digraph("Test")
+                flowchart.format = 'png'
+                self.draw_flowChart(flowchart)
+                # flowchart.view()
+                flowchart.save(directory=self.workingDir)
+                flowchart.render(os.path.join(self.workingDir,"Topology"), view=False)
+                #  Hello! To DO !!! the following lines problem
+                # topologyImage = Image.open(BytesIO(flowchart.pipe(format="png")))
+                # # topologyImage = Image.open(os.path.join(self.working
+                # # Dir,"Test.png"))
+
+                # # self.topoMapViewer = WolfMapViewer(None, "Topology", treewidth=0)
+                # ratio = topologyImage.width/topologyImage.height
+                # # self.topoMapViewer = WolfMapViewer(None, "Topology",w=topologyImage.width,h=topologyImage.height,treewidth=0)
+                # self.topoMapViewer = WolfMapViewer(None, "Topology",w=4000*ratio,treewidth=0)
+                # self.topoMapViewer.add_object('Other', newobj=genericImagetexture('Other','Topology',self.topoMapViewer,xmin=0, xmax=topologyImage.width,
+                #                     ymin=0, ymax=topologyImage.height, width=topologyImage.width, height=topologyImage.height,imageObj=topologyImage),
+                #                     ToCheck=False, id='Topology')
+                # self.topoMapViewer.Autoscale()
+                # self.topoMapViewer.OnPaint(None)
+                # self.topoMapViewer.Show()
+                pass
+
+                # isNotCorrect = True
+                # while(isNotCorrect):
+                #     print("Is this Flowchart ok?")
+                #     print("Y-Yes, N-No")
+                #     answer = input("Your answer:")
+                #     if(answer=="N" or answer=="No"):
+                #         print("The postprocess was stopped by the user!")
+                #         sys.exit()
+                #     elif(answer=="Y" or answer=="Yes"):
+                #         isNotCorrect = False
+                #     else:
+                #         print("ERROR: Please enter the correct answer!")
+
+                # r = wx.MessageDialog(None, "Is this Flowchart ok?", "Topology verification", wx.YES_NO | wx.NO_DEFAULT | wx.ICON_QUESTION).ShowModal()
+
+                # if r != wx.ID_YES:
+                #     print("The postprocess was stopped by the user!")
+                #     sys.exit()
+
+            # Definition of junctionOut by default the last junction/element in the topology tree
+            self.junctionOut = self.get_lastJunctionKey()
 
-        # Construction of the Catchment
-        # ------------------------------
-
-        # 1) 1st Iteration: Object creation
-
-        # Iterate through the Input params dictionnary
-        self.create_ObjectsInCatchment()
-
-        self.charact_watrshd.set_names_subbasins([(cur.iDSorted, cur.name) for cur in self.subBasinDict.values()])
-
-        # self.add_hyetoToDict()
-
-
-        # 2) 2nd Iteration: Link between objects
-        self.link_objects()   # This procedure also creates the first layer of the topo tree by identifying the source ss-basins
+            # Associate objects linked to timeDelay in all RB objects
+            tmp = self.find_all_timeDelayObj()
 
-        """
-        The topo tree is organised by level:
-        - The first level contains should only contain subbasins which don't have any input flows.
-          Therefore, they already contains all the information to build their hydrograph
-        - The second and upper levels can contain either RB or subbasins with input flows.
+            # ===============================
+            # Computation of the hydrographs
+            self.construct_hydro()
 
-        """
+            # Reading of the rain for each subbasin
+            try:
+                self.add_rainToAllObjects()
+            except:
+                print("ERROR! The rain couldn't be created. This might be induced by the lack of the dbf file!")
+                pass
 
-        # 3) 3rd Iteration: Complete the tree
-        self.complete_topoDict()
-        if(not(self.plotNothing)):
-            flowchart = graphviz.Digraph("Test")
-            flowchart.format = 'png'
-            self.draw_flowChart(flowchart)
-            # flowchart.view()
-            flowchart.save(directory=self.workingDir)
-            flowchart.render(os.path.join(self.workingDir,"Topology"), view=False)
-            #  Hello! To DO !!! the following lines problem
-            # topologyImage = Image.open(BytesIO(flowchart.pipe(format="png")))
-            # # topologyImage = Image.open(os.path.join(self.working
-            # # Dir,"Test.png"))
-
-            # # self.topoMapViewer = WolfMapViewer(None, "Topology", treewidth=0)
-            # ratio = topologyImage.width/topologyImage.height
-            # # self.topoMapViewer = WolfMapViewer(None, "Topology",w=topologyImage.width,h=topologyImage.height,treewidth=0)
-            # self.topoMapViewer = WolfMapViewer(None, "Topology",w=4000*ratio,treewidth=0)
-            # self.topoMapViewer.add_object('Other', newobj=genericImagetexture('Other','Topology',self.topoMapViewer,xmin=0, xmax=topologyImage.width,
-            #                     ymin=0, ymax=topologyImage.height, width=topologyImage.width, height=topologyImage.height,imageObj=topologyImage),
-            #                     ToCheck=False, id='Topology')
-            # self.topoMapViewer.Autoscale()
-            # self.topoMapViewer.OnPaint(None)
-            # self.topoMapViewer.Show()
-            pass
-
-            # isNotCorrect = True
-            # while(isNotCorrect):
-            #     print("Is this Flowchart ok?")
-            #     print("Y-Yes, N-No")
-            #     answer = input("Your answer:")
-            #     if(answer=="N" or answer=="No"):
-            #         print("The postprocess was stopped by the user!")
-            #         sys.exit()
-            #     elif(answer=="Y" or answer=="Yes"):
-            #         isNotCorrect = False
-            #     else:
-            #         print("ERROR: Please enter the correct answer!")
-
-            # r = wx.MessageDialog(None, "Is this Flowchart ok?", "Topology verification", wx.YES_NO | wx.NO_DEFAULT | wx.ICON_QUESTION).ShowModal()
-
-            # if r != wx.ID_YES:
-            #     print("The postprocess was stopped by the user!")
-            #     sys.exit()
-
-        # Definition of junctionOut by default the last junction/element in the topology tree
-        self.junctionOut = self.get_lastJunctionKey()
-
-        # Associate objects linked to timeDelay in all RB objects
-        tmp = self.find_all_timeDelayObj()
-
-        # ===============================
-        # Computation of the hydrographs
-        self.construct_hydro()
-
-        # Reading of the rain for each subbasin
-        try:
-            self.add_rainToAllObjects()
-        except:
-            print("ERROR! The rain couldn't be created. This might be induced by the lack of the dbf file!")
-            pass
+            # Contruct the effective subbasins
+            self.get_eff_subBasin()
 
-        # Contruct the effective subbasins
-        self.get_eff_subBasin()
+            # Construct the surface drained by anthropogenic modules
+            self.construct_surfaceDrainedHydro_RB()
 
-        # Construct the surface drained by anthropogenic modules
-        self.construct_surfaceDrainedHydro_RB()
+            # Read the landuses of all subbasins
+            self.read_all_landuses()
 
-        # Read the landuses of all subbasins
-        self.read_all_landuses()
+            # ==============================
+            # Save in excel file all the hydrographs
 
-        # ==============================
-        # Save in excel file all the hydrographs
+            # Hello! To uncomment!!!!
+            # self.save_ExcelFile()
+            # self.save_ExcelFile_noLagTime()
+            # self.save_ExcelFile_V2()
+            # self.save_characteristics()
 
-        # Hello! To uncomment!!!!
-        # self.save_ExcelFile()
-        # self.save_ExcelFile_noLagTime()
-        # self.save_ExcelFile_V2()
-        # self.save_characteristics()
+            # Plot the of the subbasin or RB with level above 1 in the topo tree
+            self.plotNothing = True
+            if(not(self.plotNothing)):
+                self.plot_intersection()
+            # Plot all the subbasin hydrographs and hyetograph
+            if(self.plotAllSub):
+                self.plot_allSub()
 
-        # Plot the of the subbasin or RB with level above 1 in the topo tree
-        self.plotNothing = True
-        if(not(self.plotNothing)):
-            self.plot_intersection()
-        # Plot all the subbasin hydrographs and hyetograph
-        if(self.plotAllSub):
-            self.plot_allSub()
+            # self.charact_watrshd.impose_sorted_index_subbasins([cur.iDSorted for cur in self.subBasinDict.values()])
 
-        # self.charact_watrshd.impose_sorted_index_subbasins([cur.iDSorted for cur in self.subBasinDict.values()])
+            self._fill_cloud_retentionbasin()
 
-        self._fill_cloud_retentionbasin()
+        except:
+            logging.error(_("An error occured during the creation of the Catchment object."))
+            logging.info(_("If you are in a preprocessing step, it could be a normal issue."))
+            return
 
     def get_subBasin(self, id_sorted_or_name:int | str) -> SubBasin:
         """
@@ -461,6 +463,10 @@ class Catchment:
         """
         fileNameInteriorPoints = os.path.join(workingDir,'simul_sorted_interior_points.txt')
 
+        if not os.path.exists(fileNameInteriorPoints):
+            logging.error("The file simul_sorted_interior_points.txt is not present in the working directory!")
+            return
+
         # !!! Add the case the file is not present.
         with open(fileNameInteriorPoints) as fileID:
             data_reader = csv.reader(fileID, delimiter='\t')
@@ -1415,21 +1421,21 @@ class Catchment:
         # Extract the column names according to their sorted subbasin indices
         col_time = ["Time [s]"]
         col_subs = [cur_sub.name for cur_sub in sorted(self.subBasinDict.values(), key=lambda sub: sub.iDSorted)]
-        col_anth = [" : ".join([cur_anth.name, name]) 
-                    for cur_anth in self.retentionBasinDict.values() 
+        col_anth = [" : ".join([cur_anth.name, name])
+                    for cur_anth in self.retentionBasinDict.values()
                     for name in cur_anth.get_outFlow_names()]
-        
-        
+
+
         all_data = np.concatenate((time, np.array(data_subs).T, np.array(data_anth).T), axis=1)
         all_columns = "\t".join(col_time+col_subs+col_anth)
-        
+
         # Save the data in a text file
         np.savetxt(cur_file, all_data, delimiter='\t', newline="\n", header=all_columns, fmt=format)
 
 
     def save_own_hydro_for_2D(self, fileName:str="HydrosSub_2_simul2D.txt", directory:str="", format:str='%1.5e'):
         """
-        Saves subbasins' hydrographs from their own drained surface only (not taking into account the 
+        Saves subbasins' hydrographs from their own drained surface only (not taking into account the
         surface drained by the inlets or upstream elements) to a text file that can be read and used in a 2D model.
 
         Args:
@@ -1451,11 +1457,11 @@ class Catchment:
         # Extract the column names according to their sorted subbasin indices
         col_time = ["Time [s]"]
         col_subs = [cur_sub.name for cur_sub in sorted(self.subBasinDict.values(), key=lambda sub: sub.iDSorted)]
-        
-        
+
+
         all_data = np.concatenate((time, np.array(data_subs).T), axis=1)
         all_columns = "\t".join(col_time+col_subs)
-        
+
         # Save the data in a text file
         np.savetxt(cur_file, all_data, delimiter='\t', newline="\n", header=all_columns, fmt=format)
 
@@ -1525,34 +1531,40 @@ class Catchment:
 
         # Read the DBF file to save all the "Ordered To Nb" in the dictionnary
         fileName = os.path.join(self.workingDir, "Whole_basin/Rain_basin_geom.vec.dbf")
-        if(os.path.exists(fileName)):
-            dbfDict = self.read_dbfFile(fileName)
-            self.hyetoDict['Ordered To Nb'] = {}
-            self.hyetoDict['Hyetos'] = {}
-
-            for i in range(len(dbfDict.records)):
-                iDsorted = i + 1
-                iDHyeto = dbfDict.records[i]['data']
-                self.hyetoDict['Ordered To Nb'][iDsorted] = iDHyeto
-
-            # Read all the .hyeto file to save the time and rain arrays
-            beginFileName = os.path.join(self.workingDir,"Whole_basin/")
-            endFileName = "rain.hyeto"
-            for element in self.hyetoDict['Ordered To Nb']:
-                nbToRead = self.hyetoDict['Ordered To Nb'][element]
-                fileName = os.path.join(beginFileName, nbToRead+ endFileName)
-                isOk, fileName = check_path(fileName, applyCWD=True)
-                if isOk<0:
-                    # print("WARNING: could not find any dbf file! ")
-                    time_mod.sleep(.5)
-                    return
-                [time, rain] = self.get_hyeto(fileName)
-                self.hyetoDict['Hyetos'][element] = {}
-                self.hyetoDict['Hyetos'][element]['time'] = time
-                self.hyetoDict['Hyetos'][element]['rain'] = rain
-        else:
-            print("WARNING: could not find any dbf file! ")
-            time_mod.sleep(.5)
+
+        if self.type_of_rain == cst.source_municipality_unit_hyeto:
+            try:
+                if(os.path.exists(fileName)):
+                    dbfDict = self.read_dbfFile(fileName)
+                    self.hyetoDict['Ordered To Nb'] = {}
+                    self.hyetoDict['Hyetos'] = {}
+
+                    for i in range(len(dbfDict.records)):
+                        iDsorted = i + 1
+                        iDHyeto = dbfDict.records[i]['data']
+                        self.hyetoDict['Ordered To Nb'][iDsorted] = iDHyeto
+
+                    # Read all the .hyeto file to save the time and rain arrays
+                    beginFileName = os.path.join(self.workingDir,"Whole_basin/")
+                    endFileName = "rain.hyeto"
+                    for element in self.hyetoDict['Ordered To Nb']:
+                        nbToRead = self.hyetoDict['Ordered To Nb'][element]
+                        fileName = os.path.join(beginFileName, nbToRead+ endFileName)
+                        isOk, fileName = check_path(fileName, applyCWD=True)
+                        if isOk<0:
+                            # print("WARNING: could not find any dbf file! ")
+                            time_mod.sleep(.5)
+                            return
+                        [time, rain] = self.get_hyeto(fileName)
+                        self.hyetoDict['Hyetos'][element] = {}
+                        self.hyetoDict['Hyetos'][element]['time'] = time
+                        self.hyetoDict['Hyetos'][element]['rain'] = rain
+            except:
+                print("WARNING: problem in some dbf file! ")
+                time_mod.sleep(.5)
+            else:
+                print("WARNING: could not find any dbf file! ")
+                time_mod.sleep(.5)
 
 
     def add_rainToAllObjects(self):
@@ -1568,6 +1580,11 @@ class Catchment:
             indexName = txt + str(i)
             for element in self.topologyDict[indexName]:
                 timeTest = self.topologyDict[indexName][element].add_rain(self.workingDir, tzDelta=datetime.timedelta(hours=self.tz))
+
+                if timeTest is None:
+                    print(f"WARNING: No rain data found for {element} in {indexName}.")
+                    continue
+
                 if not(np.array_equal(timeTest,self.time)):
                     print("ERROR: the time arrays are different!")
                 if type(self.topologyDict[indexName][element]) == SubBasin:
@@ -1711,7 +1728,22 @@ class Catchment:
                         downName += ' [sub'+sortNodeID+']'
                     flowchart.edge(nodeName, downName)
 
+    def save_flow_chart(self, filename="Topology", directory="") -> str:
+        """ This procedure saves the flowchart representing the topo tree
+
+        :param filename: Name of the file to save the flowchart as.
+        :param directory: Directory where the flowchart will be saved. If not provided, it will be saved in the working directory.
+        """
+
+        flowchart = graphviz.Digraph("Test")
+        flowchart.format = 'png'
+        self.draw_flowChart(flowchart)
+        # flowchart.view()
+        flowchart.save(directory=self.workingDir)
+        flowchart.render(os.path.join(self.workingDir,"Topology"), view=False)
 
+        logging.info("Flowchart saved as: " + os.path.join(self.workingDir,"Topology.png"))
+        return os.path.join(self.workingDir,"Topology.png")
 
     def make_stat_distributionOfslope(self):
         """ This procedure plot the stat distribution of slopes.
@@ -1927,7 +1959,7 @@ class Catchment:
                 # paramsInput.ApplytoMemory(None)
                 paramsInput.SavetoFile(None)
 
-    
+
     def _correct_Umax_from_old_model(self, adapt_with_rain:bool=True, k_opt:float=0.0, U_max_opt=0.0):
         fileName = "simul_soil.param"
         which="Umax"
@@ -1940,7 +1972,7 @@ class Catchment:
 
             paramsInput = Wolf_Param(to_read=False,toShow=False)
             paramsInput.ReadFile(fileToModif)
-            
+
             if U_max_opt>0.0:
                 maxRain = U_max_opt
             elif adapt_with_rain:
@@ -1963,7 +1995,7 @@ class Catchment:
                     continue
             else:
                 k = k_opt
-            
+
             U_max = maxRain/k
 
             paramsInput.change_param("Distributed production model parameters", which, U_max)
@@ -2065,16 +2097,16 @@ class Catchment:
 
     def get_eff_subBasin(self):
 
-        nbIP = int(self.paramsInput.get_param("Semi distributed model", "How many?"))
+        nbIP = self.paramsInput[("Semi distributed model", "How many?")]
         try:
-            allSub = int(self.paramsInput.get_param("Semi distributed model", "Compute all?"))
+            allSub = int(self.paramsInput[("Semi distributed model", "Compute all?")])
         except:
             allSub = 1
 
         if(allSub==0 or allSub==-1):
             for i in range(1,nbIP+1):
-                isActive = self.paramsInput.get_param("Interior point "+str(i), "Active")
-                if(isActive=="1"):
+                isActive = self.paramsInput[(f"Interior point {i}", "Active")]
+                if(isActive==1):
                     self.myEffSortSubBasins.append(self.dictIdConversion[i])
                     self.myEffSubBasins.append(i)
         else:
@@ -2597,14 +2629,20 @@ class Catchment:
 
     def set_eff_outlet_coord(self):
 
-        all_nodes = [self.charact_watrshd.find_rivers(whichsub=ii+1)[0][0] for ii in range(self.nbSubBasin)]
-        for el in all_nodes:
-            el:Node_Watershed
-            mysubxy=wolfvertex(el.x,el.y)
-            self.subBasinCloud.add_vertex(mysubxy)
-
-        return 0
-
+        try:
+            all_nodes = [self.charact_watrshd.find_rivers(whichsub=ii+1) for ii in range(self.nbSubBasin)]
+            if len(all_nodes) == 0:
+                logging.warning("No effective outlet coordinates found in the watershed. Please check the watershed data.")
+                return -1
+            all_nodes = [sublist[0][0] for sublist in all_nodes if len(sublist[0]) > 0]
+            for el in all_nodes:
+                el:Node_Watershed
+                mysubxy=wolfvertex(el.x,el.y)
+                self.subBasinCloud.add_vertex(mysubxy)
+            return 0
+        except Exception as e:
+            logging.error(f"Error in setting effective outlet coordinates: {e}")
+            return -1
 
     def update_charact_watrshd(self):
         if self.charact_watrshd.to_update_times:
@@ -2934,7 +2972,35 @@ class Catchment:
             paramsInput.change_param(group, key, values[index])
             paramsInput.SavetoFile(None)
             paramsInput.Reload(None)
-        
+
+
+    def get_outflow(self, station_out:str="") -> np.ndarray:
+        """
+        Set the initial conditions for the outflow of the outlet defined by junctionOut.
+
+        Args:
+            keys (list[str]): List of sub-basin names.
+            values (np.ndarray): Corresponding outflow values for each sub-basin.
+        """
+        if station_out == "":
+            key_module = self.junctionOut
+        else:
+            key_module = self.get_key_catchmentDict(station_out)
+            if station_out is None:
+                logging.error("ERROR : The station name given in 'get_outflow' is not correct ! " + station_out)
+                return None
+        cur_module = self.catchmentDict[key_module]
+        return cur_module.get_outFlow()
+
+
+    @property
+    def type_of_rain(self) -> int:
+        rain = self.paramsInput.get_param("Atmospheric data", "Type of rainfall")
+        if rain == cst.source_custom:
+            rain = self.paramsInput.get_param("Custom inputs", "Rain data")
+
+        return rain
+
 
     def make_nd_array(self, c_pointer, shape, dtype=np.float64, order='C', own_data=True,readonly=False):
         arr_size = np.prod(shape[:]) * np.dtype(dtype).itemsize

wolfhece/hydrology/Internal_variables.py

@@ -26,6 +26,7 @@ class Internal_Variable:
     file:str
     type_of_var:int
     linked_param:int
+    id:int
 
 
     def get_time_serie(self, directory, prefix_file:str="", 
@@ -135,6 +136,7 @@ class Param_to_Activate:
         """
         Activate the parameters for the internal variables.
         """
+        # If the key is None, it means that it either always written or it is just not an internal variable but directly an exit
         if self.key is None or self.group is None:
             return
         
@@ -155,13 +157,16 @@ class Param_to_Activate:
             param_file.change_param(self.group, self.key, 1)
             param_file.SavetoFile(None)
             param_file.Reload(None)
-        else:
-            self.deactivate(directory, prefix_file)
+
 
     def deactivate(self, directory:str, prefix_file:str=""):
         """
         Deactivate the parameters for the internal variables.
         """
+        # If the key is None, it means that it either always written or it is just not an internal variable but directly an exit
+        if self.key is None or self.group is None:
+            return
+        
         new_prefix = self._build_prefix(prefix_file)
         filename = ".".join([new_prefix,"param"])
         param_filename = join(directory, filename)
@@ -281,3 +286,39 @@ class Group_to_Activate:
             all_linked_params.update(param.get_linked_params())
 
         return all_linked_params
+    
+    def get_all_variables_names_from_ids(self, ids:list[int], 
+                                         type_of_var:int=ALL_VAR) -> tuple[list[str], list[int]]:
+        """
+        Get the names of the internal variables from their IDs.
+
+        :param ids: List of IDs of the internal variables.
+        :return: List of names of the internal variables.
+        """
+        all_names = []
+        kept_indices = []
+        for param in self.all_params:
+            for var in param.all_variables:
+                if var.id in ids and (var.type_of_var == type_of_var or type_of_var == ALL_VAR):
+                    all_names.append(var.name)
+                    kept_indices.append(ids.index(var.id))
+
+        return all_names, kept_indices
+    
+    def get_dict_from_matrix_and_ids(self, matrix:np.ndarray, ids:list[int], 
+                                     type_of_var:int=ALL_VAR) -> dict[str, np.ndarray]:
+        """
+        Get a dictionary from a matrix and a list of IDs.
+
+        :param matrix: Matrix containing the data.
+        :param ids: List of IDs corresponding to the data.
+        :return: Dictionary with IDs as keys and data as values.
+        """
+        # Check the dimensions of the matrix and the length of the IDs
+        if matrix.shape[1] != len(ids):
+            logging.error("Matrix length does not match IDs length!")
+            return {}
+        # Extract the the names of the internal variables from the IDs
+        names, kept_indices = self.get_all_variables_names_from_ids(ids, type_of_var=type_of_var)
+        
+        return {key: matrix[:,i] for key, i in zip(names, kept_indices)}