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> 10665c2dedSJed Brown #include <petscviewer.h> 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 3135928de7SBarry Smith .seealso: VecScatterCreateWithData(), 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 4535928de7SBarry Smith SCATTER_FORWARD - Scatters the values as dictated by the VecScatterCreateWithData() call 46398c84b2SBarry Smith 47398c84b2SBarry Smith Level: beginner 48398c84b2SBarry Smith 4935928de7SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreateWithData(), 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 5635928de7SBarry Smith in the VecScatterCreateWithData() 57398c84b2SBarry Smith 58398c84b2SBarry Smith Level: beginner 59398c84b2SBarry Smith 6035928de7SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreateWithData(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 61398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 62398c84b2SBarry Smith 63398c84b2SBarry Smith M*/ 64398c84b2SBarry Smith 65398c84b2SBarry Smith /*MC 6635928de7SBarry Smith SCATTER_FORWARD_LOCAL - Scatters the values as dictated by the VecScatterCreateWithData() 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 7135928de7SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreateWithData(), 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 7835928de7SBarry Smith in the VecScatterCreateWithData() 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 8335928de7SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreateWithData(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 84398c84b2SBarry Smith SCATTER_REVERSE 85398c84b2SBarry Smith 86398c84b2SBarry Smith M*/ 87398c84b2SBarry Smith 8876bdecfbSBarry Smith /*J 898f6c3df8SBarry Smith VecType - String with the name of a PETSc vector 9009321671SBarry Smith 9109321671SBarry Smith Level: beginner 9209321671SBarry Smith 938f6c3df8SBarry Smith .seealso: VecSetType(), Vec, VecCreate(), VecDestroy() 9476bdecfbSBarry Smith J*/ 9519fd82e9SBarry Smith typedef const char* VecType; 960676abe4SMatthew Knepley #define VECSEQ "seq" 970676abe4SMatthew Knepley #define VECMPI "mpi" 98f48c50deSBarry Smith #define VECSTANDARD "standard" /* seq on one process and mpi on several */ 990676abe4SMatthew Knepley #define VECSHARED "shared" 100b17c682bSKarl Rupp #define VECSEQVIENNACL "seqviennacl" 101b17c682bSKarl Rupp #define VECMPIVIENNACL "mpiviennacl" 102b17c682bSKarl Rupp #define VECVIENNACL "viennacl" /* seqviennacl on one process and mpiviennacl on several */ 10382f73ecaSAlejandro Lamas Daviña #define VECSEQCUDA "seqcuda" 10482f73ecaSAlejandro Lamas Daviña #define VECMPICUDA "mpicuda" 10582f73ecaSAlejandro Lamas Daviña #define VECCUDA "cuda" /* seqcuda on one process and mpicuda on several */ 1061aae2881SJed Brown #define VECNEST "nest" 107803a1b88SHong Zhang #define VECNODE "node" /* use on-node shared memory */ 1082eac72dbSBarry Smith 109803a1b88SHong Zhang /*J 110803a1b88SHong Zhang VecScatterType - String with the name of a PETSc vector scatter type 111803a1b88SHong Zhang 112803a1b88SHong Zhang Level: beginner 113803a1b88SHong Zhang 11435928de7SBarry Smith .seealso: VecScatterSetType(), VecScatter, VecScatterCreateWithData(), VecScatterDestroy() 115803a1b88SHong Zhang J*/ 116803a1b88SHong Zhang typedef const char* VecScatterType; 117803a1b88SHong Zhang #define VECSCATTERSEQ "seq" 118803a1b88SHong Zhang #define VECSCATTERMPI1 "mpi1" 119803a1b88SHong Zhang #define VECSCATTERMPI3 "mpi3" /* use MPI3 on-node shared memory */ 120803a1b88SHong Zhang #define VECSCATTERMPI3NODE "mpi3node" /* use MPI3 on-node shared memory for vector type VECNODE */ 121803a1b88SHong Zhang 122803a1b88SHong Zhang /* Dynamic creation and loading functions */ 123803a1b88SHong Zhang PETSC_EXTERN PetscFunctionList VecScatterList; 124803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterSetType(VecScatter, VecScatterType); 125803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterGetType(VecScatter, VecScatterType *); 126803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterSetFromOptions(VecScatter); 127803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterRegister(const char[],PetscErrorCode (*)(VecScatter)); 12835928de7SBarry Smith PETSC_EXTERN PetscErrorCode VecScatterCreateWithData(Vec,IS,Vec,IS,VecScatter*); 129803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterInitializePackage(void); 130803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterFinalizePackage(void); 1314b71561bSShri Abhyankar 132fd487807SMatthew Knepley /* Logging support */ 133450a797fSBarry Smith #define REAL_FILE_CLASSID 1211213 1340700a824SBarry Smith #define VEC_FILE_CLASSID 1211214 135014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_CLASSID; 136014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_SCATTER_CLASSID; 1378ba1e511SMatthew Knepley 13845b63f25SDmitry Karpeev 139607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecInitializePackage(void); 140014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecFinalizePackage(void); 141fd487807SMatthew Knepley 142014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreate(MPI_Comm,Vec*); 143014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeq(MPI_Comm,PetscInt,Vec*); 144014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*); 145014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*); 146014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar[],Vec*); 147014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*); 148803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecCreateNode(MPI_Comm,PetscInt,PetscInt,Vec*); 149803a1b88SHong Zhang 150014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetFromOptions(Vec); 151685405a1SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecViewFromOptions(Vec A,PetscObject B,const char name[]) {return PetscObjectViewFromOptions((PetscObject)A,B,name);} 1520d2bece7SBarry Smith 153014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetUp(Vec); 154014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroy(Vec*); 155014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecZeroEntries(Vec); 156014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOptionsPrefix(Vec,const char[]); 157014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAppendOptionsPrefix(Vec,const char[]); 158014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOptionsPrefix(Vec,const char*[]); 159f69a0ea3SMatthew Knepley 160014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetSizes(Vec,PetscInt,PetscInt); 161fd487807SMatthew Knepley 16252b6e47cSBarry Smith PETSC_EXTERN PetscErrorCode VecDotNorm2(Vec,Vec,PetscScalar*,PetscReal*); 163014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDot(Vec,Vec,PetscScalar*); 16467392de3SBarry Smith PETSC_EXTERN PetscErrorCode VecDotRealPart(Vec,Vec,PetscReal*); 165014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDot(Vec,Vec,PetscScalar*); 166014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]); 167014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]); 168014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSubVector(Vec,IS,Vec*); 169014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreSubVector(Vec,IS,Vec*); 170cddf8d76SBarry Smith 17109321671SBarry Smith /*E 17209321671SBarry Smith NormType - determines what type of norm to compute 17309321671SBarry Smith 17409321671SBarry Smith Level: beginner 17509321671SBarry Smith 17609321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm() 17709321671SBarry Smith E*/ 1789dcbbd2bSBarry Smith typedef enum {NORM_1=0,NORM_2=1,NORM_FROBENIUS=2,NORM_INFINITY=3,NORM_1_AND_2=4} NormType; 1796a6fc655SJed Brown PETSC_EXTERN const char *const NormTypes[]; 180cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY 18109321671SBarry Smith 1829b250c83SBarry Smith /*MC 1839b250c83SBarry Smith NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum 1849b250c83SBarry Smith 1859b250c83SBarry Smith Level: beginner 1869b250c83SBarry Smith 1879b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS, 1889b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1899b250c83SBarry Smith 1909b250c83SBarry Smith M*/ 1919b250c83SBarry Smith 1929b250c83SBarry Smith /*MC 193df54d55cSMatthias Liertzer NORM_2 - the two norm, ||v|| = sqrt(sum_i |v_i|^2) (vectors only) 1949b250c83SBarry Smith 1959b250c83SBarry Smith Level: beginner 1969b250c83SBarry Smith 1979b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS, 1989b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1999b250c83SBarry Smith 2009b250c83SBarry Smith M*/ 2019b250c83SBarry Smith 2029b250c83SBarry Smith /*MC 203df54d55cSMatthias Liertzer NORM_FROBENIUS - ||A|| = sqrt(sum_ij |A_ij|^2), same as NORM_2 for vectors 2049b250c83SBarry Smith 2059b250c83SBarry Smith Level: beginner 2069b250c83SBarry Smith 2079b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 2089b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 2099b250c83SBarry Smith 2109b250c83SBarry Smith M*/ 2119b250c83SBarry Smith 2129b250c83SBarry Smith /*MC 2139b250c83SBarry Smith NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum 2149b250c83SBarry Smith 2159b250c83SBarry Smith Level: beginner 2169b250c83SBarry Smith 2179b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 2189b250c83SBarry Smith NORM_FROBINIUS, NORM_1_AND_2 2199b250c83SBarry Smith 2209b250c83SBarry Smith M*/ 2219b250c83SBarry Smith 2229b250c83SBarry Smith /*MC 2239b250c83SBarry Smith NORM_1_AND_2 - computes both the 1 and 2 norm of a vector 2249b250c83SBarry Smith 2259b250c83SBarry Smith Level: beginner 2269b250c83SBarry Smith 2279b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 2289b250c83SBarry Smith NORM_FROBINIUS, NORM_INFINITY 2299b250c83SBarry Smith 2309b250c83SBarry Smith M*/ 2319b250c83SBarry Smith 2329b250c83SBarry Smith /*MC 2339b250c83SBarry Smith NORM_MAX - see NORM_INFINITY 2349b250c83SBarry Smith 235d41222bbSBarry Smith Level: beginner 236d41222bbSBarry Smith 2379b250c83SBarry Smith M*/ 2389b250c83SBarry Smith 239014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNorm(Vec,NormType,PetscReal *); 240014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *); 241014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormalize(Vec,PetscReal *); 242014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSum(Vec,PetscScalar*); 243014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMax(Vec,PetscInt*,PetscReal *); 244014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMin(Vec,PetscInt*,PetscReal *); 245014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScale(Vec,PetscScalar); 246014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCopy(Vec,Vec); 247014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetRandom(Vec,PetscRandom); 248014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSet(Vec,PetscScalar); 249422a814eSBarry Smith PETSC_EXTERN PetscErrorCode VecSetInf(Vec); 250014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSwap(Vec,Vec); 251014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPY(Vec,PetscScalar,Vec); 252014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBY(Vec,PetscScalar,PetscScalar,Vec); 253014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]); 254014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAYPX(Vec,PetscScalar,Vec); 255014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecWAXPY(Vec,PetscScalar,Vec,Vec); 256014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec); 257014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMax(Vec,Vec,Vec); 258014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMaxAbs(Vec,Vec,Vec); 259014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMin(Vec,Vec,Vec); 260014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMult(Vec,Vec,Vec); 261014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseDivide(Vec,Vec,Vec); 262014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMaxPointwiseDivide(Vec,Vec,PetscReal*); 263014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecShift(Vec,PetscScalar); 264014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReciprocal(Vec); 265014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPermute(Vec, IS, PetscBool ); 266014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSqrtAbs(Vec); 267014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLog(Vec); 268014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecExp(Vec); 269014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAbs(Vec); 270014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicate(Vec,Vec*); 271014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicateVecs(Vec,PetscInt,Vec*[]); 272014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroyVecs(PetscInt, Vec*[]); 273014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNormAll(Vec,NormType,PetscReal[]); 274014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMaxAll(Vec,PetscInt [],PetscReal []); 275014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMinAll(Vec,PetscInt [],PetscReal []); 276014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScaleAll(Vec,const PetscScalar[]); 277849b11cfSMatthew G. Knepley PETSC_EXTERN PetscErrorCode VecUniqueEntries(Vec,PetscInt*,PetscScalar**); 2784a560884SBarry Smith 279014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNorm(Vec,PetscInt,NormType,PetscReal*); 280014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *); 281014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *); 282014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScale(Vec,PetscInt,PetscScalar); 283014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideSet(Vec,PetscInt,PetscScalar); 284954b3ec6SBarry Smith 285954b3ec6SBarry Smith 286014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGather(Vec,PetscInt,Vec,InsertMode); 287014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatter(Vec,PetscInt,Vec,InsertMode); 288014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGatherAll(Vec,Vec[],InsertMode); 289014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatterAll(Vec[],Vec,InsertMode); 290d2655a18SBarry Smith 291bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetScatter(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode); 292bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetGather(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode); 293bdeb6c88SBarry Smith 294014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 295014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]); 296014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyBegin(Vec); 297014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyEnd(Vec); 298014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashSetInitialSize(Vec,PetscInt,PetscInt); 299014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashView(Vec,PetscViewer); 300685405a1SBarry Smith PETSC_EXTERN PetscErrorCode VecStashViewFromOptions(Vec,PetscObject,const char[]); 301014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*); 30262dc5420SSatish Balay 30330de9b25SBarry Smith /*MC 30430de9b25SBarry Smith VecSetValue - Set a single entry into a vector. 30530de9b25SBarry Smith 30630de9b25SBarry Smith Synopsis: 307aaa7dc30SBarry Smith #include <petscvec.h> 3088122439eSJed Brown PetscErrorCode VecSetValue(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 30930de9b25SBarry Smith 31030de9b25SBarry Smith Not Collective 31130de9b25SBarry Smith 31230de9b25SBarry Smith Input Parameters: 31330de9b25SBarry Smith + v - the vector 31430de9b25SBarry Smith . row - the row location of the entry 31530de9b25SBarry Smith . value - the value to insert 31630de9b25SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 31730de9b25SBarry Smith 31830de9b25SBarry Smith Notes: 31930de9b25SBarry Smith For efficiency one should use VecSetValues() and set several or 32030de9b25SBarry Smith many values simultaneously if possible. 32130de9b25SBarry Smith 3221d73ed98SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 323879a43d7SBarry Smith MUST be called after all calls to VecSetValue() have been completed. 3241d73ed98SBarry Smith 325879a43d7SBarry Smith VecSetValue() uses 0-based indices in Fortran as well as in C. 3261d73ed98SBarry Smith 3271d73ed98SBarry Smith Level: beginner 3281d73ed98SBarry Smith 3291d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal() 3301d73ed98SBarry Smith M*/ 33150755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);} 3321d73ed98SBarry Smith 33330de9b25SBarry Smith 334014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetBlockSize(Vec,PetscInt); 335014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetBlockSize(Vec,PetscInt*); 336014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 3378ed539a5SBarry Smith 338fd487807SMatthew Knepley /* Dynamic creation and loading functions */ 339140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList VecList; 34019fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecSetType(Vec, VecType); 34119fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecGetType(Vec, VecType *); 342bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode VecRegister(const char[],PetscErrorCode (*)(Vec)); 34330de9b25SBarry Smith 34435928de7SBarry Smith PETSC_EXTERN PetscErrorCode VecScatterSetData(VecScatter,Vec,IS,Vec,IS); 34535928de7SBarry Smith PETSC_EXTERN PetscErrorCode VecScatterCreate(MPI_Comm,VecScatter *); 346014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode); 347014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode); 348014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterDestroy(VecScatter*); 349246626dcSBarry Smith PETSC_EXTERN PetscErrorCode VecScatterSetUp(VecScatter); 350014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCopy(VecScatter,VecScatter *); 351014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterView(VecScatter,PetscViewer); 352685405a1SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecScatterViewFromOptions(VecScatter A,PetscObject obj,const char name[]) {return PetscObjectViewFromOptions((PetscObject)A,obj,name);} 353014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterRemap(VecScatter,PetscInt *,PetscInt*); 354014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterGetMerged(VecScatter,PetscBool *); 3552195c698SBarry Smith 356014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 357014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 358014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 359014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 360014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 361014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 362014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 363014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 364ab360428SBarry Smith 3655edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 3665edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 3675edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 3685edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 3695edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 3705edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 3715edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray1dRead(Vec,PetscInt,PetscInt,PetscScalar *[]); 3725edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray1dRead(Vec,PetscInt,PetscInt,PetscScalar *[]); 3735edff71fSBarry Smith 374014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPlaceArray(Vec,const PetscScalar[]); 375014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecResetArray(Vec); 376014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReplaceArray(Vec,const PetscScalar[]); 377014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrays(const Vec[],PetscInt,PetscScalar**[]); 378014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]); 37984cb2905SBarry Smith 380014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecView(Vec,PetscViewer); 381014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecEqual(Vec,Vec,PetscBool *); 382014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLoad(Vec, PetscViewer); 3838ed539a5SBarry Smith 384014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSize(Vec,PetscInt*); 385014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalSize(Vec,PetscInt*); 386014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRange(Vec,PetscInt*,PetscInt*); 387014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRanges(Vec,const PetscInt *[]); 3888ed539a5SBarry Smith 389014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping); 390014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 39188b03592SBarry Smith 39288b03592SBarry Smith /*MC 39388b03592SBarry Smith VecSetValueLocal - Set a single entry into a vector using the local numbering 39488b03592SBarry Smith 39588b03592SBarry Smith Synopsis: 396aaa7dc30SBarry Smith #include <petscvec.h> 3978122439eSJed Brown PetscErrorCode VecSetValueLocal(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 39888b03592SBarry Smith 39988b03592SBarry Smith Not Collective 40088b03592SBarry Smith 40188b03592SBarry Smith Input Parameters: 40288b03592SBarry Smith + v - the vector 40388b03592SBarry Smith . row - the row location of the entry 40488b03592SBarry Smith . value - the value to insert 40588b03592SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 40688b03592SBarry Smith 40788b03592SBarry Smith Notes: 40888b03592SBarry Smith For efficiency one should use VecSetValues() and set several or 40988b03592SBarry Smith many values simultaneously if possible. 41088b03592SBarry Smith 41188b03592SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 41288b03592SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 41388b03592SBarry Smith 41488b03592SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 41588b03592SBarry Smith 41688b03592SBarry Smith Level: beginner 41788b03592SBarry Smith 41888b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue() 41988b03592SBarry Smith M*/ 42088b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);} 42188b03592SBarry Smith 422014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 423014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*); 42490f02eecSBarry Smith 425014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotBegin(Vec,Vec,PetscScalar *); 426014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotEnd(Vec,Vec,PetscScalar *); 427014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotBegin(Vec,Vec,PetscScalar *); 428014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotEnd(Vec,Vec,PetscScalar *); 429014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormBegin(Vec,NormType,PetscReal *); 430014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormEnd(Vec,NormType,PetscReal *); 431d3c178dbSBarry Smith 432014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 433014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 434014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 435014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 436014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCommSplitReductionBegin(MPI_Comm); 437a751f32aSSatish Balay 438a751f32aSSatish Balay 439bda0d2a0SJed Brown typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES,VEC_SUBSET_OFF_PROC_ENTRIES} VecOption; 440014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOption(Vec,VecOption,PetscBool ); 44190f02eecSBarry Smith 4425c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray(Vec,PetscScalar**); 4435c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrayRead(Vec,const PetscScalar**); 4445c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray(Vec,PetscScalar**); 4455c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrayRead(Vec,const PetscScalar**); 446046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVector(Vec,Vec); 447046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVector(Vec,Vec); 448046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVectorRead(Vec,Vec); 449046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVectorRead(Vec,Vec); 4509521ec69SBarry Smith 4519521ec69SBarry Smith /*@C 4529521ec69SBarry Smith VecGetArrayPair - Accesses a pair of pointers for two vectors that may be common. When not common the first is read only 4539521ec69SBarry Smith 4549521ec69SBarry Smith Logically Collective on Vec 4559521ec69SBarry Smith 456f5f57ec0SBarry Smith Input Parameters: 4579521ec69SBarry Smith + x - the vector 4589521ec69SBarry Smith - y - the second vector 4599521ec69SBarry Smith 460f5f57ec0SBarry Smith Output Parameters: 4619521ec69SBarry Smith + xv - location to put pointer to the first array 4629521ec69SBarry Smith - yv - location to put pointer to the second array 4639521ec69SBarry Smith 4649521ec69SBarry Smith Level: developer 4659521ec69SBarry Smith 466f5f57ec0SBarry Smith Not available from Fortran 467f5f57ec0SBarry Smith 4689521ec69SBarry Smith .seealso: VecGetArray(), VecGetArrayRead(), VecRestoreArrayPair() 4699521ec69SBarry Smith 4709521ec69SBarry Smith @*/ 471d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecGetArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv) 472d9ca1df4SBarry Smith { 473d9ca1df4SBarry Smith PetscErrorCode ierr; 474d9ca1df4SBarry Smith 475d9ca1df4SBarry Smith PetscFunctionBegin; 476d9ca1df4SBarry Smith ierr = VecGetArray(y,yv);CHKERRQ(ierr); 477d9ca1df4SBarry Smith if (x != y) { 478d9ca1df4SBarry Smith ierr = VecGetArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr); 479d9ca1df4SBarry Smith } else { 480d9ca1df4SBarry Smith *xv = *yv; 481d9ca1df4SBarry Smith } 482d9ca1df4SBarry Smith PetscFunctionReturn(0); 483d9ca1df4SBarry Smith } 4849521ec69SBarry Smith 4859521ec69SBarry Smith /*@C 4869521ec69SBarry Smith VecRestoreArrayPair - Returns a pair of pointers for two vectors that may be common. When not common the first is read only 4879521ec69SBarry Smith 4889521ec69SBarry Smith Logically Collective on Vec 4899521ec69SBarry Smith 490f5f57ec0SBarry Smith Input Parameters: 4919521ec69SBarry Smith + x - the vector 4929521ec69SBarry Smith - y - the second vector 4939521ec69SBarry Smith 494f5f57ec0SBarry Smith Output Parameters: 4959521ec69SBarry Smith + xv - location to put pointer to the first array 4969521ec69SBarry Smith - yv - location to put pointer to the second array 4979521ec69SBarry Smith 4989521ec69SBarry Smith Level: developer 4999521ec69SBarry Smith 500f5f57ec0SBarry Smith Not available from Fortran 501f5f57ec0SBarry Smith 5029521ec69SBarry Smith .seealso: VecGetArray(), VecGetArrayRead(), VecGetArrayPair() 5039521ec69SBarry Smith 5049521ec69SBarry Smith @*/ 505d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecRestoreArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv) 506d9ca1df4SBarry Smith { 507d9ca1df4SBarry Smith PetscErrorCode ierr; 508d9ca1df4SBarry Smith 509d9ca1df4SBarry Smith PetscFunctionBegin; 510d9ca1df4SBarry Smith ierr = VecRestoreArray(y,yv);CHKERRQ(ierr); 511d9ca1df4SBarry Smith if (x != y) { 512d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr); 513d9ca1df4SBarry Smith } 514d9ca1df4SBarry Smith PetscFunctionReturn(0); 515d9ca1df4SBarry Smith } 516d9ca1df4SBarry Smith 517d9ca1df4SBarry Smith #if defined(PETSC_USE_DEBUG) 518d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockGet(Vec,PetscInt*); 519d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockPush(Vec); 520d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockPop(Vec); 521d9ca1df4SBarry Smith #define VecLocked(x,arg) do {PetscInt _st; PetscErrorCode __ierr = VecLockGet(x,&_st); CHKERRQ(__ierr); if (_st > 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE," Vec is locked read only, argument # %d",arg);} while (0) 522d9ca1df4SBarry Smith #else 523e5a281bcSBarry Smith #define VecLockGet(x,arg) *(arg) = 0 524e5a281bcSBarry Smith #define VecLockPush(x) 0 525e5a281bcSBarry Smith #define VecLockPop(x) 0 526e5a281bcSBarry Smith #define VecLocked(x,arg) 527d9ca1df4SBarry Smith #endif 528e1fa1e0fSSatish Balay 52962796dfbSBarry Smith PETSC_EXTERN PetscErrorCode VecValidValues(Vec,PetscInt,PetscBool); 530f7ecc322SBarry Smith 53115091d37SBarry Smith /* 53215091d37SBarry Smith These numbers need to match the entries in 5333c94ec11SBarry Smith the function table in vecimpl.h 53415091d37SBarry Smith */ 53592d03f5aSMatthew G. Knepley typedef enum { VECOP_DUPLICATE = 0, VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_VIEWNATIVE = 68, VECOP_LOADNATIVE = 69 } VecOperation; 536014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOperation(Vec,VecOperation,void(*)(void)); 537b19c1e4cSBarry Smith 538e182c471SBarry Smith /* 539e182c471SBarry Smith Routines for dealing with ghosted vectors: 540e182c471SBarry Smith vectors with ghost elements at the end of the array. 541e182c471SBarry Smith */ 542014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMPISetGhost(Vec,PetscInt,const PetscInt[]); 543014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 544014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 545014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 546014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 547014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostGetLocalForm(Vec,Vec*); 548014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostRestoreLocalForm(Vec,Vec*); 5493efe6655SBarry Smith PETSC_EXTERN PetscErrorCode VecGhostIsLocalForm(Vec,Vec,PetscBool*); 550014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateBegin(Vec,InsertMode,ScatterMode); 551014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateEnd(Vec,InsertMode,ScatterMode); 552e182c471SBarry Smith 553014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecConjugate(Vec); 55434233285SBarry Smith 555014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToAll(Vec,VecScatter*,Vec*); 556014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToZero(Vec,VecScatter*,Vec*); 557bba1ac68SSatish Balay 5584473680cSBarry Smith PETSC_EXTERN PetscErrorCode ISComplementVec(IS,Vec,IS*); 5594473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecPow(Vec, PetscScalar); 5604473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecMedian(Vec, Vec, Vec, Vec); 561ce902467SBarry Smith PETSC_EXTERN PetscErrorCode VecWhichInactive(Vec, Vec, Vec, Vec, PetscBool, IS *); 5624473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetween(Vec, Vec, Vec, IS *); 5634473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetweenOrEqual(Vec, Vec, Vec, IS *); 5644473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichGreaterThan(Vec, Vec, IS * ); 5654473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichLessThan(Vec, Vec, IS *); 5664473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichEqual(Vec, Vec, IS *); 5674473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISAXPY(Vec, IS, PetscScalar,Vec); 5688883cd90SMatthew G. Knepley PETSC_EXTERN PetscErrorCode VecISCopy(Vec, IS, ScatterMode, Vec); 5694473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISSet(Vec,IS, PetscScalar); 5704473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecBoundGradientProjection(Vec, Vec, Vec, Vec, Vec); 5714473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepBoundInfo(Vec,Vec,Vec,Vec,PetscReal*, PetscReal*,PetscReal*); 5724473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMax(Vec, Vec, PetscReal *); 5734473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMaxBounded(Vec,Vec,Vec,Vec,PetscReal*); 5744473680cSBarry Smith 575014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaGetVector(PetscViewer, Vec); 576014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaPutVector(PetscViewer, Vec); 5777dbadf16SMatthew Knepley 578d59c15a7SBarry Smith /*S 579d59c15a7SBarry Smith Vecs - Collection of vectors where the data for the vectors is stored in 580759e7b9cSHong Zhang one contiguous memory 581d59c15a7SBarry Smith 582d59c15a7SBarry Smith Level: advanced 583d59c15a7SBarry Smith 584d59c15a7SBarry Smith Notes: 585d59c15a7SBarry Smith Temporary construct for handling multiply right hand side solves 586d59c15a7SBarry Smith 587d59c15a7SBarry Smith This is faked by storing a single vector that has enough array space for 588d59c15a7SBarry Smith n vectors 589d59c15a7SBarry Smith 590d59c15a7SBarry Smith Concepts: parallel decomposition 591d59c15a7SBarry Smith 592d59c15a7SBarry Smith S*/ 59395fbd943SSatish Balay struct _n_Vecs {PetscInt n; Vec v;}; 59495fbd943SSatish Balay typedef struct _n_Vecs* Vecs; 595e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDestroy(Vecs); 596e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeq(MPI_Comm,PetscInt,PetscInt,Vecs*); 597e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,PetscScalar*,Vecs*); 598e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDuplicate(Vecs,Vecs*); 599e9fa29b7SSatish Balay 600e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL) 601b17c682bSKarl Rupp typedef struct _p_PetscViennaCLIndices* PetscViennaCLIndices; 602b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscViennaCLIndices*); 603b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesDestroy(PetscViennaCLIndices*); 604b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyToGPUSome_Public(Vec,PetscViennaCLIndices); 605b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyFromGPUSome_Public(Vec,PetscViennaCLIndices); 606b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateSeqViennaCL(MPI_Comm,PetscInt,Vec*); 607b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateMPIViennaCL(MPI_Comm,PetscInt,PetscInt,Vec*); 608b8ced49eSKarl Rupp #endif 609*fd314934SBarry Smith #if defined(PETSC_HAVE_CUDA) 61082f73ecaSAlejandro Lamas Daviña typedef struct _p_PetscCUDAIndices* PetscCUDAIndices; 61182f73ecaSAlejandro Lamas Daviña typedef struct _p_VecScatterCUDAIndices_StoS* VecScatterCUDAIndices_StoS; 61282f73ecaSAlejandro Lamas Daviña typedef struct _p_VecScatterCUDAIndices_PtoP* VecScatterCUDAIndices_PtoP; 61382f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCUDACopyToGPUSome_Public(Vec,PetscCUDAIndices); 61482f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCUDACopyFromGPUSome_Public(Vec,PetscCUDAIndices); 61582f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec,ScatterMode); 61682f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter); 61782f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateSeqCUDA(MPI_Comm,PetscInt,Vec*); 61876d6722aSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateSeqCUDAWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar*,Vec*); 61982f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateMPICUDA(MPI_Comm,PetscInt,PetscInt,Vec*); 62076d6722aSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateMPICUDAWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar*,Vec*); 621b17c682bSKarl Rupp #endif 622b17c682bSKarl Rupp 623014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVecs(Vec,PetscInt*,Vec**); 624014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVec(Vec,PetscInt,Vec*); 625014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*); 626014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVec(Vec,PetscInt,Vec); 627014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*); 628014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSize(Vec,PetscInt*); 62945b63f25SDmitry Karpeev 630c5929fdfSBarry Smith PETSC_EXTERN PetscErrorCode PetscOptionsGetVec(PetscOptions,const char[],const char[],Vec,PetscBool*); 6314325cce7SMatthew G Knepley PETSC_EXTERN PetscErrorCode VecChop(Vec,PetscReal); 632ab352700SBarry Smith 633077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecGetLayout(Vec,PetscLayout*); 634077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecSetLayout(Vec,PetscLayout); 6359a6d0b0bSJed Brown 6369a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode PetscSectionVecView(PetscSection, Vec, PetscViewer); 6379a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecGetValuesSection(Vec, PetscSection, PetscInt, PetscScalar **); 6389a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesSection(Vec, PetscSection, PetscInt, PetscScalar [], InsertMode); 63922d96a0cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscSectionVecNorm(PetscSection, PetscSection, Vec, NormType, PetscReal []); 6409a6d0b0bSJed Brown 641ad59f098SAta Mesgarnejad PETSC_EXTERN PetscErrorCode PetscSFCreateFromZero(MPI_Comm,Vec,PetscSF*); 642ad59f098SAta Mesgarnejad 6435543aaa0SToby Isaac /*S 6445543aaa0SToby Isaac VecTagger - Object used to manage the tagging of a subset of indices based on the values of a vector. The 6455543aaa0SToby Isaac motivating application is the selection of cells for refinement or coarsening based on vector containing 6465543aaa0SToby Isaac the values in an error indicator metric. 6475543aaa0SToby Isaac 6485543aaa0SToby Isaac Level: advanced 6495543aaa0SToby Isaac S*/ 65061bbf837SToby Isaac typedef struct _p_VecTagger *VecTagger; 6515543aaa0SToby Isaac 6525543aaa0SToby Isaac /*J 6535543aaa0SToby Isaac VecTaggerType - String with the name of a VecTagger type 6545543aaa0SToby Isaac 6555543aaa0SToby Isaac Level: advanced 6565543aaa0SToby Isaac J*/ 6575543aaa0SToby Isaac typedef const char* VecTaggerType; 6584178c1b7SToby Isaac /* tag where the vector values are in a box of explicitly defined values */ 659cab77d58SToby Isaac #define VECTAGGERABSOLUTE "absolute" 6604178c1b7SToby Isaac /* tag where the vector values are in a box of values relative to the set of all values in the vector */ 6615543aaa0SToby Isaac #define VECTAGGERRELATIVE "relative" 6625543aaa0SToby Isaac /* tag where the vector values are in a relative range of the *cumulative distribution* of values in the vector */ 6631d1d0dd8SToby Isaac #define VECTAGGERCDF "cdf" 6645543aaa0SToby Isaac /* tag a vector as the union of other tags */ 6655543aaa0SToby Isaac #define VECTAGGEROR "or" 6665543aaa0SToby Isaac /* tag a vector as the intersection of other tags */ 6675543aaa0SToby Isaac #define VECTAGGERAND "and" 6685543aaa0SToby Isaac 6695543aaa0SToby Isaac PETSC_EXTERN PetscClassId VEC_TAGGER_CLASSID; 67061bbf837SToby Isaac PETSC_EXTERN PetscFunctionList VecTaggerList; 67161bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRegister(const char[],PetscErrorCode (*) (VecTagger)); 6725543aaa0SToby Isaac 6735543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCreate(MPI_Comm,VecTagger *); 674cab77d58SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetBlockSize(VecTagger,PetscInt); 675cab77d58SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetBlockSize(VecTagger,PetscInt*); 6765543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetType(VecTagger,VecTaggerType); 6775543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetType(VecTagger,VecTaggerType *); 6785543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetInvert(VecTagger,PetscBool); 6795543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetInvert(VecTagger,PetscBool*); 6805543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetFromOptions(VecTagger); 6815543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetUp(VecTagger); 6825543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerView(VecTagger,PetscViewer); 6835543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerComputeIS(VecTagger,Vec,IS *); 6845543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerDestroy(VecTagger *); 6855543aaa0SToby Isaac 6864178c1b7SToby Isaac /*S 6874178c1b7SToby Isaac VecTaggerBox - A box range used to tag values. For real scalars, this is just a closed interval; for complex scalars, the box is the closed region in the complex plane 6884178c1b7SToby Isaac such that real(min) <= real(z) <= real(max) and imag(min) <= imag(z) <= imag(max). INF is an acceptable endpoint. 68961bbf837SToby Isaac 6904178c1b7SToby Isaac Level: beginner 69161bbf837SToby Isaac 6924178c1b7SToby Isaac .seealso: VecTaggerComputeIntervals() 6934178c1b7SToby Isaac S*/ 6944178c1b7SToby Isaac typedef struct { 6954178c1b7SToby Isaac PetscScalar min; 6964178c1b7SToby Isaac PetscScalar max; 6974178c1b7SToby Isaac } VecTaggerBox; 6984178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerComputeBoxes(VecTagger,Vec,PetscInt *,VecTaggerBox **); 6994178c1b7SToby Isaac 7004178c1b7SToby Isaac 7014178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAbsoluteSetBox(VecTagger,VecTaggerBox *); 7024178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAbsoluteGetBox(VecTagger,const VecTaggerBox **); 7034178c1b7SToby Isaac 7044178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRelativeSetBox(VecTagger,VecTaggerBox *); 7054178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRelativeGetBox(VecTagger,const VecTaggerBox **); 7064178c1b7SToby Isaac 7074178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFSetBox(VecTagger,VecTaggerBox *); 7084178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFGetBox(VecTagger,const VecTaggerBox **); 7091d1d0dd8SToby Isaac 7101d1d0dd8SToby Isaac /*E 7111d1d0dd8SToby Isaac VecTaggerCDFMethod - Determines what method is used to compute absolute values from cumulative distribution values (e.g., what value is the preimage of .95 in the cdf). Relevant only in parallel: in serial it is directly computed. 7121d1d0dd8SToby Isaac 7131d1d0dd8SToby Isaac Level: advanced 7141d1d0dd8SToby Isaac .seealso: VecTaggerCDFSetMethod(), VecTaggerCDFMethods 7151d1d0dd8SToby Isaac E*/ 7161d1d0dd8SToby Isaac typedef enum {VECTAGGER_CDF_GATHER,VECTAGGER_CDF_ITERATIVE,VECTAGGER_CDF_NUM_METHODS} VecTaggerCDFMethod; 7171d1d0dd8SToby Isaac PETSC_EXTERN const char *const VecTaggerCDFMethods[]; 7181d1d0dd8SToby Isaac 7191d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFSetMethod(VecTagger,VecTaggerCDFMethod); 7201d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFGetMethod(VecTagger,VecTaggerCDFMethod*); 7211d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFIterativeSetTolerances(VecTagger,PetscInt,PetscReal,PetscReal); 7221d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFIterativeGetTolerances(VecTagger,PetscInt*,PetscReal*,PetscReal*); 72361bbf837SToby Isaac 724497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerOrSetSubs(VecTagger,PetscInt,VecTagger*,PetscCopyMode); 725497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerOrGetSubs(VecTagger,PetscInt*,VecTagger**); 72661bbf837SToby Isaac 727497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAndSetSubs(VecTagger,PetscInt,VecTagger*,PetscCopyMode); 728497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAndGetSubs(VecTagger,PetscInt*,VecTagger**); 72961bbf837SToby Isaac 73061bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerInitializePackage(void); 73161bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerFinalizePackage(void); 73261bbf837SToby Isaac 7332eac72dbSBarry Smith #endif 734