1 /* $Id: petscvec.h,v 1.110 2000/06/23 15:41:47 buschelm Exp bsmith $ */ 2 /* 3 Defines the vector component of PETSc. Vectors generally represent 4 degrees of freedom for finite element/finite difference functions 5 on a grid. They have more mathematical structure then simple arrays. 6 */ 7 8 #ifndef __PETSCVEC_H 9 #define __PETSCVEC_H 10 #include "petscis.h" 11 #include "petscsys.h" 12 13 #define VEC_COOKIE PETSC_COOKIE+3 14 #define MAP_COOKIE PETSC_COOKIE+22 15 #define VEC_SCATTER_COOKIE PETSC_COOKIE+4 16 17 typedef struct _p_Map* Map; 18 typedef struct _p_Vec* Vec; 19 typedef struct _p_VecScatter* VecScatter; 20 #define VEC_SEQ "seq" 21 #define VEC_MPI "mpi" 22 #define VEC_FETI "feti" 23 #define VEC_SHARED "shared" 24 typedef char* VecType; 25 26 EXTERN int VecCreateSeq(MPI_Comm,int,Vec*); 27 EXTERN int MapCreateMPI(MPI_Comm,int,int,Map*); 28 EXTERN int VecCreateMPI(MPI_Comm,int,int,Vec*); 29 EXTERN int VecCreateSeqWithArray(MPI_Comm,int,const Scalar[],Vec*); 30 EXTERN int VecCreateMPIWithArray(MPI_Comm,int,int,const Scalar[],Vec*); 31 EXTERN int VecCreateShared(MPI_Comm,int,int,Vec*); 32 EXTERN int VecCreate(MPI_Comm,int,int,Vec*); 33 EXTERN int VecSetType(Vec,VecType); 34 EXTERN int VecSetFromOptions(Vec); 35 EXTERN int VecPrintHelp(Vec); 36 37 EXTERN int VecDestroy(Vec); 38 39 EXTERN int MapDestroy(Map); 40 EXTERN int MapGetLocalSize(Map,int *); 41 EXTERN int MapGetSize(Map,int *); 42 EXTERN int MapGetLocalRange(Map,int *,int *); 43 EXTERN int MapGetGlobalRange(Map,int *[]); 44 45 EXTERN int VecDot(Vec,Vec,Scalar*); 46 EXTERN int VecTDot(Vec,Vec,Scalar*); 47 EXTERN int VecMDot(int,Vec,const Vec[],Scalar*); 48 EXTERN int VecMTDot(int,Vec,const Vec[],Scalar*); 49 50 typedef enum {NORM_1=1,NORM_2=2,NORM_FROBENIUS=3,NORM_INFINITY=4,NORM_1_AND_2=5} NormType; 51 #define NORM_MAX NORM_INFINITY 52 EXTERN int VecNorm(Vec,NormType,double *); 53 EXTERN int VecSum(Vec,Scalar*); 54 EXTERN int VecMax(Vec,int*,double*); 55 EXTERN int VecMin(Vec,int*,double*); 56 EXTERN int VecScale(const Scalar*,Vec); 57 EXTERN int VecCopy(Vec,Vec); 58 EXTERN int VecSetRandom(PetscRandom,Vec); 59 EXTERN int VecSet(const Scalar*,Vec); 60 EXTERN int VecSwap(Vec,Vec); 61 EXTERN int VecAXPY(const Scalar*,Vec,Vec); 62 EXTERN int VecAXPBY(const Scalar*,const Scalar *,Vec,Vec); 63 EXTERN int VecMAXPY(int,const Scalar*,Vec,Vec*); 64 EXTERN int VecAYPX(const Scalar*,Vec,Vec); 65 EXTERN int VecWAXPY(const Scalar*,Vec,Vec,Vec); 66 EXTERN int VecPointwiseMult(Vec,Vec,Vec); 67 EXTERN int VecPointwiseDivide(Vec,Vec,Vec); 68 EXTERN int VecShift(const Scalar*,Vec); 69 EXTERN int VecReciprocal(Vec); 70 EXTERN int VecAbs(Vec); 71 EXTERN int VecDuplicate(Vec,Vec*); 72 EXTERN int VecDuplicateVecs(Vec,int,Vec*[]); 73 EXTERN int VecDestroyVecs(const Vec[],int); 74 EXTERN int VecGetMap(Vec,Map*); 75 76 typedef enum {NOT_SET_VALUES,INSERT_VALUES,ADD_VALUES,MAX_VALUES} InsertMode; 77 78 EXTERN int VecStrideNorm(Vec,int,NormType,double*); 79 EXTERN int VecStrideGather(Vec,int,Vec,InsertMode); 80 EXTERN int VecStrideScatter(Vec,int,Vec,InsertMode); 81 EXTERN int VecStrideMax(Vec,int,int *,double *); 82 EXTERN int VecStrideMin(Vec,int,int *,double *); 83 EXTERN int VecStrideGatherAll(Vec,Vec*,InsertMode); 84 EXTERN int VecStrideScatterAll(Vec*,Vec,InsertMode); 85 86 EXTERN int VecSetValues(Vec,int,const int[],const Scalar[],InsertMode); 87 EXTERN int VecAssemblyBegin(Vec); 88 EXTERN int VecAssemblyEnd(Vec); 89 EXTERN int VecSetStashInitialSize(Vec,int,int); 90 EXTERN int VecStashView(Vec,Viewer); 91 92 #define VecSetValue(v,i,va,mode) \ 93 {int _ierr,_row = i; Scalar _va = va; \ 94 _ierr = VecSetValues(v,1,&_row,&_va,mode);CHKERRQ(_ierr); \ 95 } 96 EXTERN int VecSetBlockSize(Vec,int); 97 EXTERN int VecGetBlockSize(Vec,int*); 98 EXTERN int VecSetValuesBlocked(Vec,int,const int[],const Scalar[],InsertMode); 99 100 extern PetscTruth VecRegisterAllCalled; 101 EXTERN int VecRegisterAll(const char []); 102 EXTERN int VecRegister(const char[],const char[],const char[],int(*)(Vec)); 103 #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 104 #define VecRegisterDynamic(a,b,c,d) VecRegister(a,b,c,0) 105 #else 106 #define VecRegisterDynamic(a,b,c,d) VecRegister(a,b,c,d) 107 #endif 108 109 typedef enum {SCATTER_FORWARD=0,SCATTER_REVERSE=1,SCATTER_FORWARD_LOCAL=2, 110 SCATTER_REVERSE_LOCAL=3,SCATTER_LOCAL=2} ScatterMode; 111 EXTERN int VecScatterCreate(Vec,IS,Vec,IS,VecScatter *); 112 EXTERN int VecScatterPostRecvs(Vec,Vec,InsertMode,ScatterMode,VecScatter); 113 EXTERN int VecScatterBegin(Vec,Vec,InsertMode,ScatterMode,VecScatter); 114 EXTERN int VecScatterEnd(Vec,Vec,InsertMode,ScatterMode,VecScatter); 115 EXTERN int VecScatterDestroy(VecScatter); 116 EXTERN int VecScatterCopy(VecScatter,VecScatter *); 117 EXTERN int VecScatterView(VecScatter,Viewer); 118 EXTERN int VecScatterRemap(VecScatter,int *,int*); 119 120 typedef enum {PIPELINE_DOWN=0,PIPELINE_UP=1} PipelineDirection; 121 typedef enum {PIPELINE_NONE=1,PIPELINE_SEQUENTIAL=2, 122 PIPELINE_REDBLACK=3,PIPELINE_MULTICOLOR=4} PipelineType; 123 124 typedef struct _p_VecPipeline* VecPipeline; 125 126 EXTERN int VecPipelineCreate(MPI_Comm,Vec,IS,Vec,IS,VecPipeline *); 127 EXTERN int VecPipelineSetType(VecPipeline,PipelineType,PetscObject); 128 EXTERN int VecPipelineSetup(VecPipeline); 129 EXTERN int VecPipelineBegin(Vec,Vec,InsertMode,ScatterMode,PipelineDirection,VecPipeline); 130 EXTERN int VecPipelineEnd(Vec,Vec,InsertMode,ScatterMode,PipelineDirection,VecPipeline); 131 EXTERN int VecPipelineView(VecPipeline,Viewer); 132 EXTERN int VecPipelineDestroy(VecPipeline); 133 134 EXTERN int VecGetArray(Vec,Scalar*[]); 135 EXTERN int VecRestoreArray(Vec,Scalar*[]); 136 EXTERN int VecGetArray4d(Vec,int,int,int,int,int,int,int,int,Scalar**[]); 137 EXTERN int VecRestoreArray4d(Vec,int,int,int,int,int,int,int,int,Scalar**[]); 138 EXTERN int VecGetArray3d(Vec,int,int,int,int,int,int,Scalar**[]); 139 EXTERN int VecRestoreArray3d(Vec,int,int,int,int,int,int,Scalar**[]); 140 EXTERN int VecGetArray2d(Vec,int,int,int,int,Scalar**[]); 141 EXTERN int VecRestoreArray2d(Vec,int,int,int,int,Scalar**[]); 142 EXTERN int VecGetArray1d(Vec,int,int,Scalar *[]); 143 EXTERN int VecRestoreArray1d(Vec,int,int,Scalar *[]); 144 145 EXTERN int VecPlaceArray(Vec,const Scalar[]); 146 EXTERN int VecReplaceArray(Vec,const Scalar[]); 147 EXTERN int VecGetArrays(const Vec[],int,Scalar**[]); 148 EXTERN int VecRestoreArrays(const Vec[],int,Scalar**[]); 149 150 EXTERN int VecValid(Vec,PetscTruth*); 151 EXTERN int VecView(Vec,Viewer); 152 EXTERN int VecEqual(Vec,Vec,PetscTruth*); 153 EXTERN int VecLoad(Viewer,Vec*); 154 EXTERN int VecLoadIntoVector(Viewer,Vec); 155 156 EXTERN int VecGetSize(Vec,int*); 157 EXTERN int VecGetType(Vec,VecType*); 158 EXTERN int VecGetLocalSize(Vec,int*); 159 EXTERN int VecGetOwnershipRange(Vec,int*,int*); 160 161 EXTERN int VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping); 162 EXTERN int VecSetValuesLocal(Vec,int,const int[],const Scalar[],InsertMode); 163 EXTERN int VecSetLocalToGlobalMappingBlocked(Vec,ISLocalToGlobalMapping); 164 EXTERN int VecSetValuesBlockedLocal(Vec,int,const int[],const Scalar[],InsertMode); 165 166 EXTERN int VecDotBegin(Vec,Vec,Scalar *); 167 EXTERN int VecDotEnd(Vec,Vec,Scalar *); 168 EXTERN int VecTDotBegin(Vec,Vec,Scalar *); 169 EXTERN int VecTDotEnd(Vec,Vec,Scalar *); 170 EXTERN int VecNormBegin(Vec,NormType,double *); 171 EXTERN int VecNormEnd(Vec,NormType,double *); 172 173 typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES} VecOption; 174 EXTERN int VecSetOption(Vec,VecOption); 175 176 EXTERN int VecContourScale(Vec,double,double); 177 178 /* 179 These numbers need to match the entries in 180 the function table in src/vec/vecimpl.h 181 */ 182 typedef enum { VECOP_VIEW = 33, 183 VECOP_LOADINTOVECTOR = 40 184 } VecOperation; 185 EXTERN int VecSetOperation(Vec,VecOperation,void*); /* */ 186 187 /* 188 Routines for dealing with ghosted vectors: 189 vectors with ghost elements at the end of the array. 190 */ 191 EXTERN int VecCreateGhost(MPI_Comm,int,int,int,const int[],Vec*); 192 EXTERN int VecCreateGhostWithArray(MPI_Comm,int,int,int,const int[],const Scalar[],Vec*); 193 EXTERN int VecCreateGhostBlock(MPI_Comm,int,int,int,int,const int[],Vec*); 194 EXTERN int VecCreateGhostBlockWithArray(MPI_Comm,int,int,int,int,const int[],const Scalar[],Vec*); 195 EXTERN int VecGhostGetLocalForm(Vec,Vec*); 196 EXTERN int VecGhostRestoreLocalForm(Vec,Vec*); 197 EXTERN int VecGhostUpdateBegin(Vec,InsertMode,ScatterMode); 198 EXTERN int VecGhostUpdateEnd(Vec,InsertMode,ScatterMode); 199 200 EXTERN int VecConjugate(Vec); 201 202 #endif 203 204 205 206