Changeset 8936 for issm/trunk/src/c/io/Disk/IoModelFetchData.cpp

Timestamp:

07/12/11 17:47:39 (14 years ago)

Author:

Eric.Larour

Message:

Implemented Mat Array for qmu.

File:

• issm/trunk/src/c/io/Disk/IoModelFetchData.cpp (modified) (1 diff)

issm/trunk/src/c/io/Disk/IoModelFetchData.cpp

@@ -215 +215 @@
         *pinteger=integer;
 
+}
+/*}}}*/
+/*FUNCTION IoModelFetchData(double*** pmatrices,int** pmdims,int** pndims, int* pM,FILE* model_handle,int data_enum){{{1*/
+void  IoModelFetchData(double*** pmatrices,int** pmdims,int** pndims, int* pnumrecords,FILE* model_handle,int data_enum){
+
+        int i;
+
+        extern int my_rank;
+        extern int num_procs;
+
+        /*output: */
+        double** matrices=NULL;
+        int*     mdims=NULL;
+        int*     ndims=NULL;
+        int      numrecords=0;
+
+        /*intermediary: */
+        int M,N;
+        double* matrix=NULL;
+        
+        FILE* fid=NULL;
+        
+        /*Set file pointer to beginning of the data: */
+        fid=SetFilePointerToData(model_handle,data_enum);
+        
+        /*Now fetch: */
+        if(my_rank==0){  
+                if(fread(&numrecords,sizeof(int),1,fid)!=1) _error_("could not read number of records in matrix array ");
+        }
+        MPI_Bcast(&numrecords,1,MPI_INT,0,MPI_COMM_WORLD); 
+
+        if(numrecords){
+
+                /*Allocate matrices :*/
+                matrices=(double**)xmalloc(numrecords*sizeof(double*));
+                mdims=(int*)xmalloc(numrecords*sizeof(int));
+                ndims=(int*)xmalloc(numrecords*sizeof(int));
+
+                for(i=0;i<numrecords;i++){
+                        matrices[i]=NULL;
+                        mdims[i]=NULL;
+                        ndims[i]=NULL;
+                }
+
+                /*Loop through records and fetch matrix: */
+                for(i=0;i<numrecords;i++){
+
+                        if(my_rank==0){  
+                                if(fread(&M,sizeof(int),1,fid)!=1) _error_("%s%i%s","could not read number of rows in ",i,"th matrix of matrix array");
+                        }
+                        MPI_Bcast(&M,1,MPI_INT,0,MPI_COMM_WORLD); 
+
+                        if(my_rank==0){  
+                                if(fread(&N,sizeof(int),1,fid)!=1) _error_("%s%i%s","could not read number of columns in ",i,"th matrix of matrix array");
+                        }
+                        MPI_Bcast(&N,1,MPI_INT,0,MPI_COMM_WORLD); 
+
+                        /*Now allocate matrix: */
+                        if(M*N){
+                                matrix=(double*)xmalloc(M*N*sizeof(double));
+
+                                /*Read matrix on node 0, then broadcast: */
+                                if(my_rank==0){  
+                                        if(fread(matrix,M*N*sizeof(double),1,fid)!=1) _error_("could not read matrix ");
+                                }
+
+                                MPI_Bcast(matrix,M*N,MPI_DOUBLE,0,MPI_COMM_WORLD); 
+                        }
+
+                        /*Assign: */
+                        matrices[i]=matrix;
+                        mdims[i]=M;
+                        ndims[i]=N;
+                }
+        }
+
+        /*Assign output pointers: */
+        *pmatrices=matrices;
+        *pmdims=mdims;
+        *pndims=ndims;
+        *pnumrecords=numrecords;
 }
 /*}}}*/