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 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 89*8f6c3df8SBarry Smith VecType - String with the name of a PETSc vector 9009321671SBarry Smith 9109321671SBarry Smith Level: beginner 9209321671SBarry Smith 93*8f6c3df8SBarry 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" 100765467adSMatthew Knepley #define VECSIEVE "sieve" 1018154be41SBarry Smith #define VECSEQCUSP "seqcusp" 1028154be41SBarry Smith #define VECMPICUSP "mpicusp" 1038154be41SBarry Smith #define VECCUSP "cusp" /* seqcusp on one process and mpicusp on several */ 1041aae2881SJed Brown #define VECNEST "nest" 1051bf2b38bSKerry Stevens #define VECSEQPTHREAD "seqpthread" 1064b71561bSShri Abhyankar #define VECMPIPTHREAD "mpipthread" 107bf2c1783SBarry Smith #define VECPTHREAD "pthread" /* seqpthread on one process and mpipthread on several */ 1082eac72dbSBarry Smith 1094b71561bSShri Abhyankar 110fd487807SMatthew Knepley /* Logging support */ 1110700a824SBarry Smith #define VEC_FILE_CLASSID 1211214 112014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_CLASSID; 113014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_SCATTER_CLASSID; 1148ba1e511SMatthew Knepley 11545b63f25SDmitry Karpeev 116607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecInitializePackage(void); 117014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecFinalizePackage(void); 118fd487807SMatthew Knepley 119014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreate(MPI_Comm,Vec*); 120014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeq(MPI_Comm,PetscInt,Vec*); 121014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*); 122014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*); 123014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar[],Vec*); 124014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*); 125014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetFromOptions(Vec); 126146574abSBarry Smith PETSC_EXTERN PetscErrorCode VecViewFromOptions(Vec,const char[],const char[]); 1270d2bece7SBarry Smith 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 13752b6e47cSBarry Smith PETSC_EXTERN PetscErrorCode VecDotNorm2(Vec,Vec,PetscScalar*,PetscReal*); 138014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDot(Vec,Vec,PetscScalar*); 13967392de3SBarry Smith PETSC_EXTERN PetscErrorCode VecDotRealPart(Vec,Vec,PetscReal*); 140014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDot(Vec,Vec,PetscScalar*); 141014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]); 142014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]); 143014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSubVector(Vec,IS,Vec*); 144014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreSubVector(Vec,IS,Vec*); 145cddf8d76SBarry Smith 14609321671SBarry Smith /*E 14709321671SBarry Smith NormType - determines what type of norm to compute 14809321671SBarry Smith 14909321671SBarry Smith Level: beginner 15009321671SBarry Smith 15109321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm() 15209321671SBarry Smith E*/ 1539dcbbd2bSBarry Smith typedef enum {NORM_1=0,NORM_2=1,NORM_FROBENIUS=2,NORM_INFINITY=3,NORM_1_AND_2=4} NormType; 1546a6fc655SJed Brown PETSC_EXTERN const char *const NormTypes[]; 155cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY 15609321671SBarry Smith 1579b250c83SBarry Smith /*MC 1589b250c83SBarry Smith NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum 1599b250c83SBarry Smith 1609b250c83SBarry Smith Level: beginner 1619b250c83SBarry Smith 1629b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS, 1639b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1649b250c83SBarry Smith 1659b250c83SBarry Smith M*/ 1669b250c83SBarry Smith 1679b250c83SBarry Smith /*MC 1689b250c83SBarry Smith NORM_2 - the two norm, ||v|| = sqrt(sum_i (v_i)^2) (vectors only) 1699b250c83SBarry Smith 1709b250c83SBarry Smith Level: beginner 1719b250c83SBarry Smith 1729b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS, 1739b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1749b250c83SBarry Smith 1759b250c83SBarry Smith M*/ 1769b250c83SBarry Smith 1779b250c83SBarry Smith /*MC 1789b250c83SBarry Smith NORM_FROBENIUS - ||A|| = sqrt(sum_ij (A_ij)^2), same as NORM_2 for vectors 1799b250c83SBarry Smith 1809b250c83SBarry Smith Level: beginner 1819b250c83SBarry Smith 1829b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 1839b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1849b250c83SBarry Smith 1859b250c83SBarry Smith M*/ 1869b250c83SBarry Smith 1879b250c83SBarry Smith /*MC 1889b250c83SBarry Smith NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum 1899b250c83SBarry Smith 1909b250c83SBarry Smith Level: beginner 1919b250c83SBarry Smith 1929b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 1939b250c83SBarry Smith NORM_FROBINIUS, NORM_1_AND_2 1949b250c83SBarry Smith 1959b250c83SBarry Smith M*/ 1969b250c83SBarry Smith 1979b250c83SBarry Smith /*MC 1989b250c83SBarry Smith NORM_1_AND_2 - computes both the 1 and 2 norm of a vector 1999b250c83SBarry Smith 2009b250c83SBarry Smith Level: beginner 2019b250c83SBarry Smith 2029b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 2039b250c83SBarry Smith NORM_FROBINIUS, NORM_INFINITY 2049b250c83SBarry Smith 2059b250c83SBarry Smith M*/ 2069b250c83SBarry Smith 2079b250c83SBarry Smith /*MC 2089b250c83SBarry Smith NORM_MAX - see NORM_INFINITY 2099b250c83SBarry Smith 210d41222bbSBarry Smith Level: beginner 211d41222bbSBarry Smith 2129b250c83SBarry Smith M*/ 2139b250c83SBarry Smith 214014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNorm(Vec,NormType,PetscReal *); 215014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *); 216014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormalize(Vec,PetscReal *); 217014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSum(Vec,PetscScalar*); 218014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMax(Vec,PetscInt*,PetscReal *); 219014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMin(Vec,PetscInt*,PetscReal *); 220014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScale(Vec,PetscScalar); 221014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCopy(Vec,Vec); 222014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetRandom(Vec,PetscRandom); 223014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSet(Vec,PetscScalar); 224014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSwap(Vec,Vec); 225014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPY(Vec,PetscScalar,Vec); 226014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBY(Vec,PetscScalar,PetscScalar,Vec); 227014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]); 228014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAYPX(Vec,PetscScalar,Vec); 229014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecWAXPY(Vec,PetscScalar,Vec,Vec); 230014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec); 231014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMax(Vec,Vec,Vec); 232014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMaxAbs(Vec,Vec,Vec); 233014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMin(Vec,Vec,Vec); 234014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMult(Vec,Vec,Vec); 235014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseDivide(Vec,Vec,Vec); 236014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMaxPointwiseDivide(Vec,Vec,PetscReal*); 237014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecShift(Vec,PetscScalar); 238014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReciprocal(Vec); 239014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPermute(Vec, IS, PetscBool ); 240014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSqrtAbs(Vec); 241014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLog(Vec); 242014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecExp(Vec); 243014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAbs(Vec); 244014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicate(Vec,Vec*); 245014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicateVecs(Vec,PetscInt,Vec*[]); 246014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroyVecs(PetscInt, Vec*[]); 247014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNormAll(Vec,NormType,PetscReal[]); 248014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMaxAll(Vec,PetscInt [],PetscReal []); 249014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMinAll(Vec,PetscInt [],PetscReal []); 250014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScaleAll(Vec,const PetscScalar[]); 2514a560884SBarry Smith 252014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNorm(Vec,PetscInt,NormType,PetscReal*); 253014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *); 254014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *); 255014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScale(Vec,PetscInt,PetscScalar); 256014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideSet(Vec,PetscInt,PetscScalar); 257954b3ec6SBarry Smith 258954b3ec6SBarry Smith 259014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGather(Vec,PetscInt,Vec,InsertMode); 260014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatter(Vec,PetscInt,Vec,InsertMode); 261014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGatherAll(Vec,Vec[],InsertMode); 262014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatterAll(Vec[],Vec,InsertMode); 263d2655a18SBarry Smith 264014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 265014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]); 266014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyBegin(Vec); 267014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyEnd(Vec); 268014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashSetInitialSize(Vec,PetscInt,PetscInt); 269014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashView(Vec,PetscViewer); 270014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*); 27162dc5420SSatish Balay 27230de9b25SBarry Smith /*MC 27330de9b25SBarry Smith VecSetValue - Set a single entry into a vector. 27430de9b25SBarry Smith 27530de9b25SBarry Smith Synopsis: 276f2ba6396SBarry Smith #include "petscvec.h" 2778122439eSJed Brown PetscErrorCode VecSetValue(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 27830de9b25SBarry Smith 27930de9b25SBarry Smith Not Collective 28030de9b25SBarry Smith 28130de9b25SBarry Smith Input Parameters: 28230de9b25SBarry Smith + v - the vector 28330de9b25SBarry Smith . row - the row location of the entry 28430de9b25SBarry Smith . value - the value to insert 28530de9b25SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 28630de9b25SBarry Smith 28730de9b25SBarry Smith Notes: 28830de9b25SBarry Smith For efficiency one should use VecSetValues() and set several or 28930de9b25SBarry Smith many values simultaneously if possible. 29030de9b25SBarry Smith 2911d73ed98SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 2921d73ed98SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 2931d73ed98SBarry Smith 2941d73ed98SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 2951d73ed98SBarry Smith 2961d73ed98SBarry Smith Level: beginner 2971d73ed98SBarry Smith 2981d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal() 2991d73ed98SBarry Smith M*/ 30050755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);} 3011d73ed98SBarry Smith 30230de9b25SBarry Smith 303014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetBlockSize(Vec,PetscInt); 304014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetBlockSize(Vec,PetscInt*); 305014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 3068ed539a5SBarry Smith 307fd487807SMatthew Knepley /* Dynamic creation and loading functions */ 308140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList VecList; 309014dd563SJed Brown PETSC_EXTERN PetscBool VecRegisterAllCalled; 31019fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecSetType(Vec, VecType); 31119fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecGetType(Vec, VecType *); 312bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode VecRegister(const char[],PetscErrorCode (*)(Vec)); 313607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecRegisterAll(void); 31430de9b25SBarry Smith 315014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreate(Vec,IS,Vec,IS,VecScatter *); 316014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateEmpty(MPI_Comm,VecScatter *); 317014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateLocal(VecScatter,PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt); 318014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode); 319014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode); 320014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterDestroy(VecScatter*); 321014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCopy(VecScatter,VecScatter *); 322014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterView(VecScatter,PetscViewer); 323014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterRemap(VecScatter,PetscInt *,PetscInt*); 324014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterGetMerged(VecScatter,PetscBool *); 3252195c698SBarry Smith 326014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 327014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 328014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 329014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 330014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 331014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 332014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 333014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 334ab360428SBarry Smith 335014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPlaceArray(Vec,const PetscScalar[]); 336014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecResetArray(Vec); 337014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReplaceArray(Vec,const PetscScalar[]); 338014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrays(const Vec[],PetscInt,PetscScalar**[]); 339014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]); 34084cb2905SBarry Smith 341014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecView(Vec,PetscViewer); 342014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecEqual(Vec,Vec,PetscBool *); 343014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLoad(Vec, PetscViewer); 3448ed539a5SBarry Smith 345014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSize(Vec,PetscInt*); 346014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalSize(Vec,PetscInt*); 347014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRange(Vec,PetscInt*,PetscInt*); 348014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRanges(Vec,const PetscInt *[]); 3498ed539a5SBarry Smith 350014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping); 351014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 35288b03592SBarry Smith 35388b03592SBarry Smith /*MC 35488b03592SBarry Smith VecSetValueLocal - Set a single entry into a vector using the local numbering 35588b03592SBarry Smith 35688b03592SBarry Smith Synopsis: 357f2ba6396SBarry Smith #include "petscvec.h" 3588122439eSJed Brown PetscErrorCode VecSetValueLocal(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 35988b03592SBarry Smith 36088b03592SBarry Smith Not Collective 36188b03592SBarry Smith 36288b03592SBarry Smith Input Parameters: 36388b03592SBarry Smith + v - the vector 36488b03592SBarry Smith . row - the row location of the entry 36588b03592SBarry Smith . value - the value to insert 36688b03592SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 36788b03592SBarry Smith 36888b03592SBarry Smith Notes: 36988b03592SBarry Smith For efficiency one should use VecSetValues() and set several or 37088b03592SBarry Smith many values simultaneously if possible. 37188b03592SBarry Smith 37288b03592SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 37388b03592SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 37488b03592SBarry Smith 37588b03592SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 37688b03592SBarry Smith 37788b03592SBarry Smith Level: beginner 37888b03592SBarry Smith 37988b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue() 38088b03592SBarry Smith M*/ 38188b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);} 38288b03592SBarry Smith 383014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping); 384014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 385014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping*); 386014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*); 38790f02eecSBarry Smith 388014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotBegin(Vec,Vec,PetscScalar *); 389014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotEnd(Vec,Vec,PetscScalar *); 390014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotBegin(Vec,Vec,PetscScalar *); 391014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotEnd(Vec,Vec,PetscScalar *); 392014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormBegin(Vec,NormType,PetscReal *); 393014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormEnd(Vec,NormType,PetscReal *); 394d3c178dbSBarry Smith 395014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 396014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 397014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 398014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 399014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCommSplitReductionBegin(MPI_Comm); 400a751f32aSSatish Balay 401a751f32aSSatish Balay 4029533e83fSBarry Smith typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES} VecOption; 403014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOption(Vec,VecOption,PetscBool ); 40490f02eecSBarry Smith 4055c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray(Vec,PetscScalar**); 4065c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrayRead(Vec,const PetscScalar**); 4075c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray(Vec,PetscScalar**); 4085c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrayRead(Vec,const PetscScalar**); 409e1fa1e0fSSatish Balay 41062796dfbSBarry Smith PETSC_EXTERN PetscErrorCode VecValidValues(Vec,PetscInt,PetscBool); 411f7ecc322SBarry Smith 412014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecContourScale(Vec,PetscReal,PetscReal); 413522c5e43SBarry Smith 41415091d37SBarry Smith /* 41515091d37SBarry Smith These numbers need to match the entries in 4163c94ec11SBarry Smith the function table in vecimpl.h 41715091d37SBarry Smith */ 4182dcb2ebcSBarry Smith typedef enum { VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_DUPLICATE = 0} VecOperation; 419014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOperation(Vec,VecOperation,void(*)(void)); 420b19c1e4cSBarry Smith 421e182c471SBarry Smith /* 422e182c471SBarry Smith Routines for dealing with ghosted vectors: 423e182c471SBarry Smith vectors with ghost elements at the end of the array. 424e182c471SBarry Smith */ 425014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMPISetGhost(Vec,PetscInt,const PetscInt[]); 426014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 427014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 428014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 429014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 430014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostGetLocalForm(Vec,Vec*); 431014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostRestoreLocalForm(Vec,Vec*); 4323efe6655SBarry Smith PETSC_EXTERN PetscErrorCode VecGhostIsLocalForm(Vec,Vec,PetscBool*); 433014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateBegin(Vec,InsertMode,ScatterMode); 434014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateEnd(Vec,InsertMode,ScatterMode); 435e182c471SBarry Smith 436014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecConjugate(Vec); 43734233285SBarry Smith 438014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToAll(Vec,VecScatter*,Vec*); 439014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToZero(Vec,VecScatter*,Vec*); 440bba1ac68SSatish Balay 441014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaGetVector(PetscViewer, Vec); 442014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaPutVector(PetscViewer, Vec); 4437dbadf16SMatthew Knepley 444d59c15a7SBarry Smith /*S 445d59c15a7SBarry Smith Vecs - Collection of vectors where the data for the vectors is stored in 446759e7b9cSHong Zhang one contiguous memory 447d59c15a7SBarry Smith 448d59c15a7SBarry Smith Level: advanced 449d59c15a7SBarry Smith 450d59c15a7SBarry Smith Notes: 451d59c15a7SBarry Smith Temporary construct for handling multiply right hand side solves 452d59c15a7SBarry Smith 453d59c15a7SBarry Smith This is faked by storing a single vector that has enough array space for 454d59c15a7SBarry Smith n vectors 455d59c15a7SBarry Smith 456d59c15a7SBarry Smith Concepts: parallel decomposition 457d59c15a7SBarry Smith 458d59c15a7SBarry Smith S*/ 45995fbd943SSatish Balay struct _n_Vecs {PetscInt n; Vec v;}; 46095fbd943SSatish Balay typedef struct _n_Vecs* Vecs; 461e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDestroy(Vecs); 462e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeq(MPI_Comm,PetscInt,PetscInt,Vecs*); 463e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,PetscScalar*,Vecs*); 464e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDuplicate(Vecs,Vecs*); 465e9fa29b7SSatish Balay 4668154be41SBarry Smith #if defined(PETSC_HAVE_CUSP) 467a5891931SBarry Smith typedef struct _p_PetscCUSPIndices* PetscCUSPIndices; 468014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCUSPIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscCUSPIndices*); 469014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCUSPIndicesDestroy(PetscCUSPIndices*); 470014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyToGPUSome_Public(Vec,PetscCUSPIndices); 471014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyFromGPUSome_Public(Vec,PetscCUSPIndices); 47290c53902SBarry Smith 473bd72cc96SPaul Mullowney #if defined(PETSC_HAVE_TXPETSCGPU) 474014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPResetIndexBuffersFlagsGPU_Public(PetscCUSPIndices); 475014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopySomeToContiguousBufferGPU_Public(Vec,PetscCUSPIndices); 476014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopySomeFromContiguousBufferGPU_Public(Vec,PetscCUSPIndices); 477014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec,ScatterMode); 478014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter); 479bd72cc96SPaul Mullowney #endif 480bd72cc96SPaul Mullowney 481014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqCUSP(MPI_Comm,PetscInt,Vec*); 482014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPICUSP(MPI_Comm,PetscInt,PetscInt,Vec*); 483a5891931SBarry Smith #endif 484e9fa29b7SSatish Balay 485014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVecs(Vec,PetscInt*,Vec**); 486014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVec(Vec,PetscInt,Vec*); 487014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*); 488014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVec(Vec,PetscInt,Vec); 489014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*); 490014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSize(Vec,PetscInt*); 49145b63f25SDmitry Karpeev 492ab352700SBarry Smith PETSC_EXTERN PetscErrorCode PetscOptionsVec(const char[],const char[],const char[],Vec,PetscBool*); 4934325cce7SMatthew G Knepley PETSC_EXTERN PetscErrorCode VecChop(Vec,PetscReal); 494ab352700SBarry Smith 495077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecGetLayout(Vec,PetscLayout*); 496077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecSetLayout(Vec,PetscLayout); 4979a6d0b0bSJed Brown 4989a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode PetscSectionVecView(PetscSection, Vec, PetscViewer); 4999a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecGetValuesSection(Vec, PetscSection, PetscInt, PetscScalar **); 5009a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesSection(Vec, PetscSection, PetscInt, PetscScalar [], InsertMode); 5019a6d0b0bSJed Brown 5022eac72dbSBarry Smith #endif 5032b82984cSSatish Balay 504