12eac72dbSBarry Smith /* 237f753daSBarry Smith Defines the vector component of PETSc. Vectors generally represent 337f753daSBarry Smith degrees of freedom for finite element/finite difference functions 484cb2905SBarry Smith on a grid. They have more mathematical structure then simple arrays. 52eac72dbSBarry Smith */ 62eac72dbSBarry Smith 70a835dfdSSatish Balay #ifndef __PETSCVEC_H 80a835dfdSSatish Balay #define __PETSCVEC_H 92c8e378dSBarry Smith #include <petscis.h> 105f5f199fSBarry Smith 112eac72dbSBarry Smith 1209321671SBarry Smith /*S 1309321671SBarry Smith Vec - Abstract PETSc vector object 1409321671SBarry Smith 1509321671SBarry Smith Level: beginner 1609321671SBarry Smith 1709321671SBarry Smith Concepts: field variables, unknowns, arrays 1809321671SBarry Smith 1909321671SBarry Smith .seealso: VecCreate(), VecType, VecSetType() 2009321671SBarry Smith S*/ 21f09e8eb9SSatish Balay typedef struct _p_Vec* Vec; 2209321671SBarry Smith 2309321671SBarry Smith /*S 2409321671SBarry Smith VecScatter - Object used to manage communication of data 2509321671SBarry Smith between vectors in parallel. Manages both scatters and gathers 2609321671SBarry Smith 2709321671SBarry Smith Level: beginner 2809321671SBarry Smith 2909321671SBarry Smith Concepts: scatter 3009321671SBarry Smith 3109321671SBarry Smith .seealso: VecScatterCreate(), VecScatterBegin(), VecScatterEnd() 3209321671SBarry Smith S*/ 33f09e8eb9SSatish Balay typedef struct _p_VecScatter* VecScatter; 3409321671SBarry Smith 3509321671SBarry Smith /*E 36398c84b2SBarry Smith ScatterMode - Determines the direction of a scatter 37398c84b2SBarry Smith 38398c84b2SBarry Smith Level: beginner 39398c84b2SBarry Smith 40398c84b2SBarry Smith .seealso: VecScatter, VecScatterBegin(), VecScatterEnd() 41398c84b2SBarry Smith E*/ 42398c84b2SBarry Smith typedef enum {SCATTER_FORWARD=0, SCATTER_REVERSE=1, SCATTER_FORWARD_LOCAL=2, SCATTER_REVERSE_LOCAL=3, SCATTER_LOCAL=2} ScatterMode; 43398c84b2SBarry Smith 44398c84b2SBarry Smith /*MC 45398c84b2SBarry Smith SCATTER_FORWARD - Scatters the values as dictated by the VecScatterCreate() call 46398c84b2SBarry Smith 47398c84b2SBarry Smith Level: beginner 48398c84b2SBarry Smith 49398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD_LOCAL, 50398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 51398c84b2SBarry Smith 52398c84b2SBarry Smith M*/ 53398c84b2SBarry Smith 54398c84b2SBarry Smith /*MC 55398c84b2SBarry Smith SCATTER_REVERSE - Moves the values in the opposite direction then the directions indicated in 56398c84b2SBarry Smith in the VecScatterCreate() 57398c84b2SBarry Smith 58398c84b2SBarry Smith Level: beginner 59398c84b2SBarry Smith 60398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 61398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 62398c84b2SBarry Smith 63398c84b2SBarry Smith M*/ 64398c84b2SBarry Smith 65398c84b2SBarry Smith /*MC 66398c84b2SBarry Smith SCATTER_FORWARD_LOCAL - Scatters the values as dictated by the VecScatterCreate() call except NO parallel communication 67398c84b2SBarry Smith is done. Any variables that have be moved between processes are ignored 68398c84b2SBarry Smith 69398c84b2SBarry Smith Level: developer 70398c84b2SBarry Smith 71398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD, 72398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 73398c84b2SBarry Smith 74398c84b2SBarry Smith M*/ 75398c84b2SBarry Smith 76398c84b2SBarry Smith /*MC 77398c84b2SBarry Smith SCATTER_REVERSE_LOCAL - Moves the values in the opposite direction then the directions indicated in 78398c84b2SBarry Smith in the VecScatterCreate() except NO parallel communication 79398c84b2SBarry Smith is done. Any variables that have be moved between processes are ignored 80398c84b2SBarry Smith 81398c84b2SBarry Smith Level: developer 82398c84b2SBarry Smith 83398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 84398c84b2SBarry Smith SCATTER_REVERSE 85398c84b2SBarry Smith 86398c84b2SBarry Smith M*/ 87398c84b2SBarry Smith 8876bdecfbSBarry Smith /*J 8909321671SBarry Smith VecType - String with the name of a PETSc vector or the creation function 9009321671SBarry Smith with an optional dynamic library name, for example 9109321671SBarry Smith http://www.mcs.anl.gov/petsc/lib.a:myveccreate() 9209321671SBarry Smith 9309321671SBarry Smith Level: beginner 9409321671SBarry Smith 9509321671SBarry Smith .seealso: VecSetType(), Vec 9676bdecfbSBarry Smith J*/ 97a313700dSBarry Smith #define VecType char* 980676abe4SMatthew Knepley #define VECSEQ "seq" 990676abe4SMatthew Knepley #define VECMPI "mpi" 100f48c50deSBarry Smith #define VECSTANDARD "standard" /* seq on one process and mpi on several */ 1010676abe4SMatthew Knepley #define VECSHARED "shared" 102765467adSMatthew Knepley #define VECSIEVE "sieve" 1038154be41SBarry Smith #define VECSEQCUSP "seqcusp" 1048154be41SBarry Smith #define VECMPICUSP "mpicusp" 1058154be41SBarry Smith #define VECCUSP "cusp" /* seqcusp on one process and mpicusp on several */ 1061aae2881SJed Brown #define VECNEST "nest" 1071bf2b38bSKerry Stevens #define VECSEQPTHREAD "seqpthread" 1084b71561bSShri Abhyankar #define VECMPIPTHREAD "mpipthread" 109bf2c1783SBarry Smith #define VECPTHREAD "pthread" /* seqpthread on one process and mpipthread on several */ 1102eac72dbSBarry Smith 1114b71561bSShri Abhyankar 112fd487807SMatthew Knepley /* Logging support */ 1130700a824SBarry Smith #define VEC_FILE_CLASSID 1211214 114014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_CLASSID; 115014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_SCATTER_CLASSID; 1168ba1e511SMatthew Knepley 11745b63f25SDmitry Karpeev 118014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecInitializePackage(const char[]); 119014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecFinalizePackage(void); 120fd487807SMatthew Knepley 121014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreate(MPI_Comm,Vec*); 122014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeq(MPI_Comm,PetscInt,Vec*); 123014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*); 124014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*); 125014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar[],Vec*); 126014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*); 127014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetFromOptions(Vec); 128014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetUp(Vec); 129014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroy(Vec*); 130014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecZeroEntries(Vec); 131014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOptionsPrefix(Vec,const char[]); 132014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAppendOptionsPrefix(Vec,const char[]); 133014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOptionsPrefix(Vec,const char*[]); 134f69a0ea3SMatthew Knepley 135014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetSizes(Vec,PetscInt,PetscInt); 136fd487807SMatthew Knepley 137014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotNorm2(Vec,Vec,PetscScalar*,PetscScalar*); 138014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDot(Vec,Vec,PetscScalar*); 139014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDot(Vec,Vec,PetscScalar*); 140014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]); 141014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]); 142014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSubVector(Vec,IS,Vec*); 143014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreSubVector(Vec,IS,Vec*); 144cddf8d76SBarry Smith 14509321671SBarry Smith /*E 14609321671SBarry Smith NormType - determines what type of norm to compute 14709321671SBarry Smith 14809321671SBarry Smith Level: beginner 14909321671SBarry Smith 15009321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm() 15109321671SBarry Smith E*/ 1529dcbbd2bSBarry Smith typedef enum {NORM_1=0,NORM_2=1,NORM_FROBENIUS=2,NORM_INFINITY=3,NORM_1_AND_2=4} NormType; 1536a6fc655SJed Brown PETSC_EXTERN const char *const NormTypes[]; 154cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY 15509321671SBarry Smith 1569b250c83SBarry Smith /*MC 1579b250c83SBarry Smith NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum 1589b250c83SBarry Smith 1599b250c83SBarry Smith Level: beginner 1609b250c83SBarry Smith 1619b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS, 1629b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1639b250c83SBarry Smith 1649b250c83SBarry Smith M*/ 1659b250c83SBarry Smith 1669b250c83SBarry Smith /*MC 1679b250c83SBarry Smith NORM_2 - the two norm, ||v|| = sqrt(sum_i (v_i)^2) (vectors only) 1689b250c83SBarry Smith 1699b250c83SBarry Smith Level: beginner 1709b250c83SBarry Smith 1719b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS, 1729b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1739b250c83SBarry Smith 1749b250c83SBarry Smith M*/ 1759b250c83SBarry Smith 1769b250c83SBarry Smith /*MC 1779b250c83SBarry Smith NORM_FROBENIUS - ||A|| = sqrt(sum_ij (A_ij)^2), same as NORM_2 for vectors 1789b250c83SBarry Smith 1799b250c83SBarry Smith Level: beginner 1809b250c83SBarry Smith 1819b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 1829b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1839b250c83SBarry Smith 1849b250c83SBarry Smith M*/ 1859b250c83SBarry Smith 1869b250c83SBarry Smith /*MC 1879b250c83SBarry Smith NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum 1889b250c83SBarry Smith 1899b250c83SBarry Smith Level: beginner 1909b250c83SBarry Smith 1919b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 1929b250c83SBarry Smith NORM_FROBINIUS, NORM_1_AND_2 1939b250c83SBarry Smith 1949b250c83SBarry Smith M*/ 1959b250c83SBarry Smith 1969b250c83SBarry Smith /*MC 1979b250c83SBarry Smith NORM_1_AND_2 - computes both the 1 and 2 norm of a vector 1989b250c83SBarry Smith 1999b250c83SBarry Smith Level: beginner 2009b250c83SBarry Smith 2019b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 2029b250c83SBarry Smith NORM_FROBINIUS, NORM_INFINITY 2039b250c83SBarry Smith 2049b250c83SBarry Smith M*/ 2059b250c83SBarry Smith 2069b250c83SBarry Smith /*MC 2079b250c83SBarry Smith NORM_MAX - see NORM_INFINITY 2089b250c83SBarry Smith 209d41222bbSBarry Smith Level: beginner 210d41222bbSBarry Smith 2119b250c83SBarry Smith M*/ 2129b250c83SBarry Smith 213014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNorm(Vec,NormType,PetscReal *); 214014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *); 215014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormalize(Vec,PetscReal *); 216014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSum(Vec,PetscScalar*); 217014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMax(Vec,PetscInt*,PetscReal *); 218014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMin(Vec,PetscInt*,PetscReal *); 219014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScale(Vec,PetscScalar); 220014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCopy(Vec,Vec); 221014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetRandom(Vec,PetscRandom); 222014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSet(Vec,PetscScalar); 223014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSwap(Vec,Vec); 224014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPY(Vec,PetscScalar,Vec); 225014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBY(Vec,PetscScalar,PetscScalar,Vec); 226014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]); 227014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAYPX(Vec,PetscScalar,Vec); 228014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecWAXPY(Vec,PetscScalar,Vec,Vec); 229014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec); 230014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMax(Vec,Vec,Vec); 231014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMaxAbs(Vec,Vec,Vec); 232014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMin(Vec,Vec,Vec); 233014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMult(Vec,Vec,Vec); 234014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseDivide(Vec,Vec,Vec); 235014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMaxPointwiseDivide(Vec,Vec,PetscReal*); 236014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecShift(Vec,PetscScalar); 237014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReciprocal(Vec); 238014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPermute(Vec, IS, PetscBool ); 239014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSqrtAbs(Vec); 240014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLog(Vec); 241014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecExp(Vec); 242014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAbs(Vec); 243014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicate(Vec,Vec*); 244014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicateVecs(Vec,PetscInt,Vec*[]); 245014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroyVecs(PetscInt, Vec*[]); 246014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNormAll(Vec,NormType,PetscReal[]); 247014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMaxAll(Vec,PetscInt [],PetscReal []); 248014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMinAll(Vec,PetscInt [],PetscReal []); 249014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScaleAll(Vec,const PetscScalar[]); 2504a560884SBarry Smith 251014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNorm(Vec,PetscInt,NormType,PetscReal*); 252014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *); 253014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *); 254014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScale(Vec,PetscInt,PetscScalar); 255014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideSet(Vec,PetscInt,PetscScalar); 256954b3ec6SBarry Smith 257954b3ec6SBarry Smith 258014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGather(Vec,PetscInt,Vec,InsertMode); 259014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatter(Vec,PetscInt,Vec,InsertMode); 260014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGatherAll(Vec,Vec[],InsertMode); 261014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatterAll(Vec[],Vec,InsertMode); 262d2655a18SBarry Smith 263014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 264014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]); 265014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyBegin(Vec); 266014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyEnd(Vec); 267014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashSetInitialSize(Vec,PetscInt,PetscInt); 268014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashView(Vec,PetscViewer); 269014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*); 27062dc5420SSatish Balay 27130de9b25SBarry Smith /*MC 27230de9b25SBarry Smith VecSetValue - Set a single entry into a vector. 27330de9b25SBarry Smith 27430de9b25SBarry Smith Synopsis: 2758122439eSJed Brown PetscErrorCode VecSetValue(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 27630de9b25SBarry Smith 27730de9b25SBarry Smith Not Collective 27830de9b25SBarry Smith 27930de9b25SBarry Smith Input Parameters: 28030de9b25SBarry Smith + v - the vector 28130de9b25SBarry Smith . row - the row location of the entry 28230de9b25SBarry Smith . value - the value to insert 28330de9b25SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 28430de9b25SBarry Smith 28530de9b25SBarry Smith Notes: 28630de9b25SBarry Smith For efficiency one should use VecSetValues() and set several or 28730de9b25SBarry Smith many values simultaneously if possible. 28830de9b25SBarry Smith 2891d73ed98SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 2901d73ed98SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 2911d73ed98SBarry Smith 2921d73ed98SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 2931d73ed98SBarry Smith 2941d73ed98SBarry Smith Level: beginner 2951d73ed98SBarry Smith 2961d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal() 2971d73ed98SBarry Smith M*/ 29850755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);} 2991d73ed98SBarry Smith 30030de9b25SBarry Smith 301014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetBlockSize(Vec,PetscInt); 302014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetBlockSize(Vec,PetscInt*); 303014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 3048ed539a5SBarry Smith 305fd487807SMatthew Knepley /* Dynamic creation and loading functions */ 306014dd563SJed Brown PETSC_EXTERN PetscFList VecList; 307014dd563SJed Brown PETSC_EXTERN PetscBool VecRegisterAllCalled; 308014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetType(Vec, const VecType); 309014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetType(Vec, const VecType *); 310014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRegister(const char[],const char[],const char[],PetscErrorCode (*)(Vec)); 311014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRegisterAll(const char []); 312014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRegisterDestroy(void); 31330de9b25SBarry Smith 31430de9b25SBarry Smith /*MC 31530de9b25SBarry Smith VecRegisterDynamic - Adds a new vector component implementation 31630de9b25SBarry Smith 31730de9b25SBarry Smith Synopsis: 3181890ba74SBarry Smith PetscErrorCode VecRegisterDynamic(const char *name, const char *path, const char *func_name, PetscErrorCode (*create_func)(Vec)) 31930de9b25SBarry Smith 32030de9b25SBarry Smith Not Collective 32130de9b25SBarry Smith 32230de9b25SBarry Smith Input Parameters: 32330de9b25SBarry Smith + name - The name of a new user-defined creation routine 32430de9b25SBarry Smith . path - The path (either absolute or relative) of the library containing this routine 32530de9b25SBarry Smith . func_name - The name of routine to create method context 32630de9b25SBarry Smith - create_func - The creation routine itself 32730de9b25SBarry Smith 32830de9b25SBarry Smith Notes: 32930de9b25SBarry Smith VecRegisterDynamic() may be called multiple times to add several user-defined vectors 33030de9b25SBarry Smith 33130de9b25SBarry Smith If dynamic libraries are used, then the fourth input argument (routine_create) is ignored. 33230de9b25SBarry Smith 33330de9b25SBarry Smith Sample usage: 33430de9b25SBarry Smith .vb 33530de9b25SBarry Smith VecRegisterDynamic("my_vec","/home/username/my_lib/lib/libO/solaris/libmy.a", "MyVectorCreate", MyVectorCreate); 33630de9b25SBarry Smith .ve 33730de9b25SBarry Smith 33830de9b25SBarry Smith Then, your vector type can be chosen with the procedural interface via 33930de9b25SBarry Smith .vb 34030de9b25SBarry Smith VecCreate(MPI_Comm, Vec *); 34130de9b25SBarry Smith VecSetType(Vec,"my_vector_name"); 34230de9b25SBarry Smith .ve 34330de9b25SBarry Smith or at runtime via the option 34430de9b25SBarry Smith .vb 34530de9b25SBarry Smith -vec_type my_vector_name 34630de9b25SBarry Smith .ve 34730de9b25SBarry Smith 348ab901514SBarry Smith Notes: $PETSC_ARCH occuring in pathname will be replaced with appropriate values. 34930de9b25SBarry Smith If your function is not being put into a shared library then use VecRegister() instead 35030de9b25SBarry Smith 35130de9b25SBarry Smith Level: advanced 35230de9b25SBarry Smith 35330de9b25SBarry Smith .keywords: Vec, register 35430de9b25SBarry Smith .seealso: VecRegisterAll(), VecRegisterDestroy(), VecRegister() 35530de9b25SBarry Smith M*/ 356aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 357f1af5d2fSBarry Smith #define VecRegisterDynamic(a,b,c,d) VecRegister(a,b,c,0) 35888d459dfSBarry Smith #else 359f1af5d2fSBarry Smith #define VecRegisterDynamic(a,b,c,d) VecRegister(a,b,c,d) 36088d459dfSBarry Smith #endif 36188d459dfSBarry Smith 36209321671SBarry Smith 363014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreate(Vec,IS,Vec,IS,VecScatter *); 364014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateEmpty(MPI_Comm,VecScatter *); 365014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateLocal(VecScatter,PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt); 366014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode); 367014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode); 368014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterDestroy(VecScatter*); 369014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCopy(VecScatter,VecScatter *); 370014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterView(VecScatter,PetscViewer); 371014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterRemap(VecScatter,PetscInt *,PetscInt*); 372014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterGetMerged(VecScatter,PetscBool *); 3732195c698SBarry Smith 374014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 375014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 376014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 377014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 378014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 379014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 380014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 381014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 382ab360428SBarry Smith 383014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPlaceArray(Vec,const PetscScalar[]); 384014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecResetArray(Vec); 385014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReplaceArray(Vec,const PetscScalar[]); 386014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrays(const Vec[],PetscInt,PetscScalar**[]); 387014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]); 38884cb2905SBarry Smith 389014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecView(Vec,PetscViewer); 390014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecViewFromOptions(Vec, const char *); 391014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecEqual(Vec,Vec,PetscBool *); 392014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLoad(Vec, PetscViewer); 3938ed539a5SBarry Smith 394014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSize(Vec,PetscInt*); 395014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalSize(Vec,PetscInt*); 396014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRange(Vec,PetscInt*,PetscInt*); 397014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRanges(Vec,const PetscInt *[]); 3988ed539a5SBarry Smith 399014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping); 400014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 40188b03592SBarry Smith 40288b03592SBarry Smith /*MC 40388b03592SBarry Smith VecSetValueLocal - Set a single entry into a vector using the local numbering 40488b03592SBarry Smith 40588b03592SBarry Smith Synopsis: 4068122439eSJed Brown PetscErrorCode VecSetValueLocal(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 40788b03592SBarry Smith 40888b03592SBarry Smith Not Collective 40988b03592SBarry Smith 41088b03592SBarry Smith Input Parameters: 41188b03592SBarry Smith + v - the vector 41288b03592SBarry Smith . row - the row location of the entry 41388b03592SBarry Smith . value - the value to insert 41488b03592SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 41588b03592SBarry Smith 41688b03592SBarry Smith Notes: 41788b03592SBarry Smith For efficiency one should use VecSetValues() and set several or 41888b03592SBarry Smith many values simultaneously if possible. 41988b03592SBarry Smith 42088b03592SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 42188b03592SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 42288b03592SBarry Smith 42388b03592SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 42488b03592SBarry Smith 42588b03592SBarry Smith Level: beginner 42688b03592SBarry Smith 42788b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue() 42888b03592SBarry Smith M*/ 42988b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);} 43088b03592SBarry Smith 431014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping); 432014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 433014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping*); 434014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*); 43590f02eecSBarry Smith 436014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotBegin(Vec,Vec,PetscScalar *); 437014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotEnd(Vec,Vec,PetscScalar *); 438014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotBegin(Vec,Vec,PetscScalar *); 439014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotEnd(Vec,Vec,PetscScalar *); 440014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormBegin(Vec,NormType,PetscReal *); 441014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormEnd(Vec,NormType,PetscReal *); 442d3c178dbSBarry Smith 443014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 444014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 445014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 446014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 447014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCommSplitReductionBegin(MPI_Comm); 448a751f32aSSatish Balay 449a751f32aSSatish Balay 4504ec14c9cSBarry Smith #if defined(PETSC_USE_DEBUG) 451ab2fbef6SJed Brown #define VecValidValues(vec,argnum,input) do { \ 4524ec14c9cSBarry Smith PetscErrorCode _ierr; \ 4534ec14c9cSBarry Smith PetscInt _n,_i; \ 4544ec14c9cSBarry Smith const PetscScalar *_x; \ 4554ec14c9cSBarry Smith \ 456ab2fbef6SJed Brown if (vec->petscnative || vec->ops->getarray) { \ 4574ec14c9cSBarry Smith _ierr = VecGetLocalSize(vec,&_n);CHKERRQ(_ierr); \ 4584ec14c9cSBarry Smith _ierr = VecGetArrayRead(vec,&_x);CHKERRQ(_ierr); \ 4594ec14c9cSBarry Smith for (_i=0; _i<_n; _i++) { \ 4604ec14c9cSBarry Smith if (input) { \ 4614ec14c9cSBarry Smith if (PetscIsInfOrNanScalar(_x[_i])) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_FP,"Vec entry at local location %D is not-a-number or infinite at beginning of function: Parameter number %d",_i,argnum); \ 4624ec14c9cSBarry Smith } else { \ 4634ec14c9cSBarry Smith if (PetscIsInfOrNanScalar(_x[_i])) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_FP,"Vec entry at local location %D is not-a-number or infinite at end of function: Parameter number %d",_i,argnum); \ 4644ec14c9cSBarry Smith } \ 4654ec14c9cSBarry Smith } \ 4664ec14c9cSBarry Smith _ierr = VecRestoreArrayRead(vec,&_x);CHKERRQ(_ierr); \ 467ab2fbef6SJed Brown } \ 468ab2fbef6SJed Brown } while (0) 4694ec14c9cSBarry Smith #else 4704ec14c9cSBarry Smith #define VecValidValues(vec,argnum,input) 4714ec14c9cSBarry Smith #endif 4724ec14c9cSBarry Smith 4734ec14c9cSBarry Smith 4749533e83fSBarry Smith typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES} VecOption; 475014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOption(Vec,VecOption,PetscBool ); 47690f02eecSBarry Smith 477ebe3b25bSBarry Smith /* 478ebe3b25bSBarry Smith Expose VecGetArray()/VecRestoreArray() to users. Allows this to work without any function 479ebe3b25bSBarry Smith call overhead on any 'native' Vecs. 480ebe3b25bSBarry Smith */ 481e1fa1e0fSSatish Balay 4822c8e378dSBarry Smith #include <petsc-private/vecimpl.h> 483ebe3b25bSBarry Smith 484014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecContourScale(Vec,PetscReal,PetscReal); 485522c5e43SBarry Smith 48615091d37SBarry Smith /* 48715091d37SBarry Smith These numbers need to match the entries in 4883c94ec11SBarry Smith the function table in vecimpl.h 48915091d37SBarry Smith */ 4902dcb2ebcSBarry Smith typedef enum { VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_DUPLICATE = 0} VecOperation; 491014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOperation(Vec,VecOperation,void(*)(void)); 492b19c1e4cSBarry Smith 493e182c471SBarry Smith /* 494e182c471SBarry Smith Routines for dealing with ghosted vectors: 495e182c471SBarry Smith vectors with ghost elements at the end of the array. 496e182c471SBarry Smith */ 497014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMPISetGhost(Vec,PetscInt,const PetscInt[]); 498014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 499014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 500014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 501014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 502014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostGetLocalForm(Vec,Vec*); 503014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostRestoreLocalForm(Vec,Vec*); 504*3efe6655SBarry Smith PETSC_EXTERN PetscErrorCode VecGhostIsLocalForm(Vec,Vec,PetscBool*); 505014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateBegin(Vec,InsertMode,ScatterMode); 506014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateEnd(Vec,InsertMode,ScatterMode); 507e182c471SBarry Smith 508014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecConjugate(Vec); 50934233285SBarry Smith 510014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToAll(Vec,VecScatter*,Vec*); 511014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToZero(Vec,VecScatter*,Vec*); 512bba1ac68SSatish Balay 513014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaGetVector(PetscViewer, Vec); 514014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaPutVector(PetscViewer, Vec); 5157dbadf16SMatthew Knepley 516d59c15a7SBarry Smith /*S 517d59c15a7SBarry Smith Vecs - Collection of vectors where the data for the vectors is stored in 518759e7b9cSHong Zhang one contiguous memory 519d59c15a7SBarry Smith 520d59c15a7SBarry Smith Level: advanced 521d59c15a7SBarry Smith 522d59c15a7SBarry Smith Notes: 523d59c15a7SBarry Smith Temporary construct for handling multiply right hand side solves 524d59c15a7SBarry Smith 525d59c15a7SBarry Smith This is faked by storing a single vector that has enough array space for 526d59c15a7SBarry Smith n vectors 527d59c15a7SBarry Smith 528d59c15a7SBarry Smith Concepts: parallel decomposition 529d59c15a7SBarry Smith 530d59c15a7SBarry Smith S*/ 53195fbd943SSatish Balay struct _n_Vecs {PetscInt n; Vec v;}; 53295fbd943SSatish Balay typedef struct _n_Vecs* Vecs; 533fcfd50ebSBarry Smith #define VecsDestroy(x) (VecDestroy(&(x)->v) || PetscFree(x)) 53495fbd943SSatish Balay #define VecsCreateSeq(comm,p,m,x) (PetscNew(struct _n_Vecs,x) || VecCreateSeq(comm,p*m,&(*(x))->v) || (-1 == ((*(x))->n = (m)))) 535778a2246SBarry Smith #define VecsCreateSeqWithArray(comm,p,m,a,x) (PetscNew(struct _n_Vecs,x) || VecCreateSeqWithArray(comm,1,p*m,a,&(*(x))->v) || (-1 == ((*(x))->n = (m)))) 53695fbd943SSatish Balay #define VecsDuplicate(x,y) (PetscNew(struct _n_Vecs,y) || VecDuplicate(x->v,&(*(y))->v) || (-1 == ((*(y))->n = (x)->n))) 537e9fa29b7SSatish Balay 5388154be41SBarry Smith #if defined(PETSC_HAVE_CUSP) 539a5891931SBarry Smith typedef struct _p_PetscCUSPIndices* PetscCUSPIndices; 540014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCUSPIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscCUSPIndices*); 541014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCUSPIndicesDestroy(PetscCUSPIndices*); 542014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyToGPUSome_Public(Vec,PetscCUSPIndices); 543014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyFromGPUSome_Public(Vec,PetscCUSPIndices); 54490c53902SBarry Smith 545bd72cc96SPaul Mullowney #if defined(PETSC_HAVE_TXPETSCGPU) 546014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPResetIndexBuffersFlagsGPU_Public(PetscCUSPIndices); 547014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopySomeToContiguousBufferGPU_Public(Vec,PetscCUSPIndices); 548014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopySomeFromContiguousBufferGPU_Public(Vec,PetscCUSPIndices); 549014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec,ScatterMode); 550014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter); 551bd72cc96SPaul Mullowney #endif 552bd72cc96SPaul Mullowney 553014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqCUSP(MPI_Comm,PetscInt,Vec*); 554014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPICUSP(MPI_Comm,PetscInt,PetscInt,Vec*); 555a5891931SBarry Smith #endif 556e9fa29b7SSatish Balay 557014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVecs(Vec,PetscInt*,Vec**); 558014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVec(Vec,PetscInt,Vec*); 559014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*); 560014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVec(Vec,PetscInt,Vec); 561014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*); 562014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSize(Vec,PetscInt*); 56345b63f25SDmitry Karpeev 5642eac72dbSBarry Smith #endif 5652b82984cSSatish Balay 566