Add VecDohpGetFS()

jedbrown · Jun 17, 2011 · ac4d9ba · ac4d9ba
1 parent 12ec27f
commit ac4d9ba
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/include/dohpfs.h b/include/dohpfs.h
@@ -160,6 +160,7 @@ extern dErr dFSGetNodalCoordinatesExpanded(dFS,Vec*);
 extern dErr dFSGetNodalCoordinatesGlobal(dFS,Vec*);
 
 extern dErr dFSRedimension(dFS,dInt,dFSClosureMode,dFS*);
+extern dErr VecDohpGetFS(Vec,dFS*);
 
 /* The normal MatAXPY() does not work for arguments of mixed type, so we redeclare this "private" function here. */
 extern PetscErrorCode MatAXPY_Basic(Mat Y,PetscScalar a,Mat X,MatStructure str);

diff --git a/src/fs/interface/fs.c b/src/fs/interface/fs.c
@@ -885,3 +885,18 @@ dErr dFSMatSetValuesBlockedExpanded(dFS fs,Mat A,dInt m,const dInt idxm[],dInt n
 #endif
   dFunctionReturn(0);
 }
+
+// Retrieve the FS composed with a VecDohp, return NULL if there is none
+dErr VecDohpGetFS(Vec X,dFS *fs)
+{
+  dErr err;
+  dBool isdohp;
+
+  dFunctionBegin;
+  dValidHeader(X,VEC_CLASSID,1);
+  dValidPointer(fs,2);
+  err = PetscTypeCompare((PetscObject)X,VECDOHP,&isdohp);dCHK(err);
+  if (!isdohp) dERROR(((PetscObject)X)->comm,PETSC_ERR_ARG_WRONG,"Vector is not of type DOHP");
+  err = PetscObjectQuery((PetscObject)X,"dFS",(PetscObject*)fs);dCHK(err);
+  dFunctionReturn(0);
+}