Actual source code: matreg.c

  1: #define PETSCMAT_DLL

  3: /*
  4:      Mechanism for register PETSc matrix types
  5: */
 6:  #include src/mat/matimpl.h
 7:  #include petscsys.h

  9: PetscTruth MatRegisterAllCalled = PETSC_FALSE;

 11: /*
 12:    Contains the list of registered Mat routines
 13: */
 14: PetscFList MatList = 0;

 18: /*@C
 19:    MatSetType - Builds matrix object for a particular matrix type

 21:    Collective on Mat

 23:    Input Parameters:
 24: +  mat      - the matrix object
 25: -  matype   - matrix type

 27:    Options Database Key:
 28: .  -mat_type  <method> - Sets the type; use -help for a list 
 29:     of available methods (for instance, seqaij)

 31:    Notes:  
 32:    See "${PETSC_DIR}/include/petscmat.h" for available methods

 34:   Level: intermediate

 36: .keywords: Mat, MatType, set, method

 38: .seealso: PCSetType(), VecSetType(), MatCreate(), MatType, Mat
 39: @*/
 40: PetscErrorCode PETSCMAT_DLLEXPORT MatSetType(Mat mat, MatType matype)
 41: {
 42:   PetscErrorCode ierr,(*r)(Mat);
 43:   PetscTruth     sametype;

 47:   if (mat->rmap.n < 0 && mat->rmap.N < 0 && mat->cmap.n < 0 && mat->cmap.N < 0) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call MatSetSizes() first");
 48:   PetscTypeCompare((PetscObject)mat,matype,&sametype);
 49:   if (!sametype) {
 50:     /* Get the function pointers for the matrix requested */
 51:     if (!MatRegisterAllCalled) {MatRegisterAll(PETSC_NULL);}
 52:      PetscFListFind(mat->comm,MatList,matype,(void(**)(void))&r);
 53:     if (!r) SETERRQ1(PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown Mat type given: %s",matype);

 55:     /* free the old data structure if it existed */
 56:     if (mat->ops->destroy) {
 57:       MatPreallocated(mat);
 58:       (*mat->ops->destroy)(mat);
 59:       mat->ops->destroy = PETSC_NULL;
 60:       mat->preallocated = PETSC_FALSE;
 61:     }

 63:     /* create the new data structure */
 64:     (*r)(mat);
 65:     PetscObjectChangeTypeName((PetscObject)mat,matype);
 66:   }
 67:   PetscPublishAll(mat);
 68:   return(0);
 69: }


 74: /*@C
 75:    MatRegisterDestroy - Frees the list of matrix types that were
 76:    registered by MatRegister()/MatRegisterDynamic().

 78:    Not Collective

 80:    Level: advanced

 82: .keywords: Mat, register, destroy

 84: .seealso: MatRegister(), MatRegisterAll(), MatRegisterDynamic()
 85: @*/
 86: PetscErrorCode PETSCMAT_DLLEXPORT MatRegisterDestroy(void)
 87: {

 91:   if (MatList) {
 92:     PetscFListDestroy(&MatList);
 93:     MatList = 0;
 94:   }
 95:   MatRegisterAllCalled = PETSC_FALSE;
 96:   return(0);
 97: }

101: /*@C
102:    MatGetType - Gets the matrix type as a string from the matrix object.

104:    Not Collective

106:    Input Parameter:
107: .  mat - the matrix

109:    Output Parameter:
110: .  name - name of matrix type

112:    Level: intermediate

114: .keywords: Mat, MatType, get, method, name

116: .seealso: MatSetType()
117: @*/
118: PetscErrorCode PETSCMAT_DLLEXPORT MatGetType(Mat mat,MatType *type)
119: {
121:   *type = mat->type_name;
122:   return(0);
123: }


128: /*@C
129:   MatRegister - See MatRegisterDynamic()

131:   Level: advanced
132: @*/
133: PetscErrorCode PETSCMAT_DLLEXPORT MatRegister(const char sname[],const char path[],const char name[],PetscErrorCode (*function)(Mat))
134: {
136:   char           fullname[PETSC_MAX_PATH_LEN];

139:   PetscFListConcat(path,name,fullname);
140:   PetscFListAdd(&MatList,sname,fullname,(void (*)(void))function);
141:   return(0);
142: }