Changeset 23602 for issm/trunk-jpl/src/c/toolkits/petsc/objects/PetscVec.cpp

Timestamp:

01/05/19 22:36:12 (6 years ago)

Author:

Mathieu Morlighem

Message:

NEW: Mergesolution from f to g is now working without MPI calls, for speed and efficiency, and cleaned up many things in the code'

File:

• issm/trunk-jpl/src/c/toolkits/petsc/objects/PetscVec.cpp (modified) (1 diff)

issm/trunk-jpl/src/c/toolkits/petsc/objects/PetscVec.cpp

@@ -113 +113 @@
         _assert_(this->vector);
         VecGetSize(this->vector,pM);
-
-}
-/*}}}*/
-void PetscVec::GetLocalSize(int* pM){/*{{{*/
-
-        _assert_(this->vector);
-        VecGetLocalSize(this->vector,pM);
-
-}
-/*}}}*/
+}
+/*}}}*/
+void PetscVec::GetLocalVector(double** pvector,int** pindices){/*{{{*/
+
+        _assert_(this->vector);
+
+        /*First, check that vector size is not 0*/
+        int vector_size;
+        this->GetSize(&vector_size);
+        if(vector_size==0){
+                *pvector=NULL;
+                *pindices=NULL;
+                return;
+        }
+
+        /*Get Ownership range*/
+        PetscInt lower_row,upper_row; 
+        VecGetOwnershipRange(this->vector,&lower_row,&upper_row);
+        int range=upper_row-lower_row;    
+
+        /*return NULL if no range*/
+        if(range==0){
+                *pvector=NULL;
+                *pindices=NULL;
+                return;
+        }
+
+        /*Build indices*/
+        int* indices=xNew<int>(range); 
+        for(int i=0;i<range;i++) indices[i]=lower_row+i;
+        /*Get vector*/
+        IssmDouble* values =xNew<double>(range);
+        VecGetValues(this->vector,range,indices,values); 
+
+        *pvector  = values;
+        *pindices = indices;
+} /*}}}*/
 PetscVec* PetscVec::Duplicate(void){/*{{{*/