xref: /petsc/include/petscvec.h (revision 45fccdb2ddc5ea077b57a4649930c46446c1fc46)
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 */
121ef6f2af5SJunchao Zhang #define VECSCATTERSF        "sf"       /* use StarForest */
122803a1b88SHong Zhang 
123803a1b88SHong Zhang /* Dynamic creation and loading functions */
124803a1b88SHong Zhang PETSC_EXTERN PetscFunctionList VecScatterList;
125803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterSetType(VecScatter, VecScatterType);
126803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterGetType(VecScatter, VecScatterType *);
127803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterSetFromOptions(VecScatter);
128803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterRegister(const char[],PetscErrorCode (*)(VecScatter));
12935928de7SBarry Smith PETSC_EXTERN PetscErrorCode VecScatterCreateWithData(Vec,IS,Vec,IS,VecScatter*);
130803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterInitializePackage(void);
131803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterFinalizePackage(void);
1324b71561bSShri Abhyankar 
133fd487807SMatthew Knepley /* Logging support */
134450a797fSBarry Smith #define    REAL_FILE_CLASSID 1211213
1350700a824SBarry Smith #define    VEC_FILE_CLASSID 1211214
136014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_CLASSID;
137014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_SCATTER_CLASSID;
1388ba1e511SMatthew Knepley 
13945b63f25SDmitry Karpeev 
140607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecInitializePackage(void);
141014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecFinalizePackage(void);
142fd487807SMatthew Knepley 
143014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreate(MPI_Comm,Vec*);
144014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeq(MPI_Comm,PetscInt,Vec*);
145014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*);
146014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*);
147014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar[],Vec*);
148014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*);
149803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecCreateNode(MPI_Comm,PetscInt,PetscInt,Vec*);
150803a1b88SHong Zhang 
151014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetFromOptions(Vec);
152685405a1SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecViewFromOptions(Vec A,PetscObject B,const char name[]) {return PetscObjectViewFromOptions((PetscObject)A,B,name);}
1530d2bece7SBarry Smith 
154014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetUp(Vec);
155014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroy(Vec*);
156014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecZeroEntries(Vec);
157014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOptionsPrefix(Vec,const char[]);
158014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAppendOptionsPrefix(Vec,const char[]);
159014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOptionsPrefix(Vec,const char*[]);
160f69a0ea3SMatthew Knepley 
161014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetSizes(Vec,PetscInt,PetscInt);
162fd487807SMatthew Knepley 
16352b6e47cSBarry Smith PETSC_EXTERN PetscErrorCode VecDotNorm2(Vec,Vec,PetscScalar*,PetscReal*);
164014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDot(Vec,Vec,PetscScalar*);
16567392de3SBarry Smith PETSC_EXTERN PetscErrorCode VecDotRealPart(Vec,Vec,PetscReal*);
166014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDot(Vec,Vec,PetscScalar*);
167014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]);
168014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]);
169014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSubVector(Vec,IS,Vec*);
170014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreSubVector(Vec,IS,Vec*);
171cddf8d76SBarry Smith 
17209321671SBarry Smith /*E
17309321671SBarry Smith     NormType - determines what type of norm to compute
17409321671SBarry Smith 
17509321671SBarry Smith     Level: beginner
17609321671SBarry Smith 
17709321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm()
17809321671SBarry Smith E*/
1799dcbbd2bSBarry Smith typedef enum {NORM_1=0,NORM_2=1,NORM_FROBENIUS=2,NORM_INFINITY=3,NORM_1_AND_2=4} NormType;
1806a6fc655SJed Brown PETSC_EXTERN const char *const NormTypes[];
181cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY
18209321671SBarry Smith 
1839b250c83SBarry Smith /*MC
1849b250c83SBarry Smith      NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum
1859b250c83SBarry Smith 
1869b250c83SBarry Smith    Level: beginner
1879b250c83SBarry Smith 
1889b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS,
1899b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
1909b250c83SBarry Smith 
1919b250c83SBarry Smith M*/
1929b250c83SBarry Smith 
1939b250c83SBarry Smith /*MC
194df54d55cSMatthias Liertzer      NORM_2 - the two norm, ||v|| = sqrt(sum_i |v_i|^2) (vectors only)
1959b250c83SBarry Smith 
1969b250c83SBarry Smith    Level: beginner
1979b250c83SBarry Smith 
1989b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS,
1999b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
2009b250c83SBarry Smith 
2019b250c83SBarry Smith M*/
2029b250c83SBarry Smith 
2039b250c83SBarry Smith /*MC
204df54d55cSMatthias Liertzer      NORM_FROBENIUS - ||A|| = sqrt(sum_ij |A_ij|^2), same as NORM_2 for vectors
2059b250c83SBarry Smith 
2069b250c83SBarry Smith    Level: beginner
2079b250c83SBarry Smith 
2089b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
2099b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
2109b250c83SBarry Smith 
2119b250c83SBarry Smith M*/
2129b250c83SBarry Smith 
2139b250c83SBarry Smith /*MC
2149b250c83SBarry Smith      NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum
2159b250c83SBarry Smith 
2169b250c83SBarry Smith    Level: beginner
2179b250c83SBarry Smith 
2189b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
2199b250c83SBarry Smith            NORM_FROBINIUS, NORM_1_AND_2
2209b250c83SBarry Smith 
2219b250c83SBarry Smith M*/
2229b250c83SBarry Smith 
2239b250c83SBarry Smith /*MC
2249b250c83SBarry Smith      NORM_1_AND_2 - computes both the 1 and 2 norm of a vector
2259b250c83SBarry Smith 
2269b250c83SBarry Smith    Level: beginner
2279b250c83SBarry Smith 
2289b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
2299b250c83SBarry Smith            NORM_FROBINIUS, NORM_INFINITY
2309b250c83SBarry Smith 
2319b250c83SBarry Smith M*/
2329b250c83SBarry Smith 
2339b250c83SBarry Smith /*MC
2349b250c83SBarry Smith      NORM_MAX - see NORM_INFINITY
2359b250c83SBarry Smith 
236d41222bbSBarry Smith    Level: beginner
237d41222bbSBarry Smith 
2389b250c83SBarry Smith M*/
2399b250c83SBarry Smith 
240014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNorm(Vec,NormType,PetscReal *);
241014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *);
242014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormalize(Vec,PetscReal *);
243014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSum(Vec,PetscScalar*);
244014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMax(Vec,PetscInt*,PetscReal *);
245014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMin(Vec,PetscInt*,PetscReal *);
246014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScale(Vec,PetscScalar);
247014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCopy(Vec,Vec);
248014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetRandom(Vec,PetscRandom);
249014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSet(Vec,PetscScalar);
250422a814eSBarry Smith PETSC_EXTERN PetscErrorCode VecSetInf(Vec);
251014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSwap(Vec,Vec);
252014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPY(Vec,PetscScalar,Vec);
253014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBY(Vec,PetscScalar,PetscScalar,Vec);
254014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]);
255014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAYPX(Vec,PetscScalar,Vec);
256014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecWAXPY(Vec,PetscScalar,Vec,Vec);
257014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec);
258014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMax(Vec,Vec,Vec);
259014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMaxAbs(Vec,Vec,Vec);
260014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMin(Vec,Vec,Vec);
261014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMult(Vec,Vec,Vec);
262014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseDivide(Vec,Vec,Vec);
263014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMaxPointwiseDivide(Vec,Vec,PetscReal*);
264014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecShift(Vec,PetscScalar);
265014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReciprocal(Vec);
266014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPermute(Vec, IS, PetscBool );
267014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSqrtAbs(Vec);
268014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLog(Vec);
269014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecExp(Vec);
270014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAbs(Vec);
271014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicate(Vec,Vec*);
272014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicateVecs(Vec,PetscInt,Vec*[]);
273014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroyVecs(PetscInt, Vec*[]);
274014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNormAll(Vec,NormType,PetscReal[]);
275014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMaxAll(Vec,PetscInt [],PetscReal []);
276014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMinAll(Vec,PetscInt [],PetscReal []);
277014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScaleAll(Vec,const PetscScalar[]);
278849b11cfSMatthew G. Knepley PETSC_EXTERN PetscErrorCode VecUniqueEntries(Vec,PetscInt*,PetscScalar**);
2794a560884SBarry Smith 
280014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNorm(Vec,PetscInt,NormType,PetscReal*);
281014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *);
282014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *);
283014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScale(Vec,PetscInt,PetscScalar);
284014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideSet(Vec,PetscInt,PetscScalar);
285954b3ec6SBarry Smith 
286954b3ec6SBarry Smith 
287014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGather(Vec,PetscInt,Vec,InsertMode);
288014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatter(Vec,PetscInt,Vec,InsertMode);
289014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGatherAll(Vec,Vec[],InsertMode);
290014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatterAll(Vec[],Vec,InsertMode);
291d2655a18SBarry Smith 
292bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetScatter(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode);
293bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetGather(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode);
294bdeb6c88SBarry Smith 
295014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
296014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]);
297014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyBegin(Vec);
298014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyEnd(Vec);
299014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashSetInitialSize(Vec,PetscInt,PetscInt);
300014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashView(Vec,PetscViewer);
301685405a1SBarry Smith PETSC_EXTERN PetscErrorCode VecStashViewFromOptions(Vec,PetscObject,const char[]);
302014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*);
30362dc5420SSatish Balay 
30430de9b25SBarry Smith /*MC
30530de9b25SBarry Smith    VecSetValue - Set a single entry into a vector.
30630de9b25SBarry Smith 
30730de9b25SBarry Smith    Synopsis:
308aaa7dc30SBarry Smith    #include <petscvec.h>
3098122439eSJed Brown    PetscErrorCode VecSetValue(Vec v,PetscInt row,PetscScalar value, InsertMode mode);
31030de9b25SBarry Smith 
31130de9b25SBarry Smith    Not Collective
31230de9b25SBarry Smith 
31330de9b25SBarry Smith    Input Parameters:
31430de9b25SBarry Smith +  v - the vector
31530de9b25SBarry Smith .  row - the row location of the entry
31630de9b25SBarry Smith .  value - the value to insert
31730de9b25SBarry Smith -  mode - either INSERT_VALUES or ADD_VALUES
31830de9b25SBarry Smith 
31930de9b25SBarry Smith    Notes:
32030de9b25SBarry Smith    For efficiency one should use VecSetValues() and set several or
32130de9b25SBarry Smith    many values simultaneously if possible.
32230de9b25SBarry Smith 
3231d73ed98SBarry Smith    These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd()
324879a43d7SBarry Smith    MUST be called after all calls to VecSetValue() have been completed.
3251d73ed98SBarry Smith 
326879a43d7SBarry Smith    VecSetValue() uses 0-based indices in Fortran as well as in C.
3271d73ed98SBarry Smith 
3281d73ed98SBarry Smith    Level: beginner
3291d73ed98SBarry Smith 
3301d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal()
3311d73ed98SBarry Smith M*/
33250755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);}
3331d73ed98SBarry Smith 
33430de9b25SBarry Smith 
335014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetBlockSize(Vec,PetscInt);
336014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetBlockSize(Vec,PetscInt*);
337014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
3388ed539a5SBarry Smith 
339fd487807SMatthew Knepley /* Dynamic creation and loading functions */
340140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList VecList;
34119fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecSetType(Vec, VecType);
34219fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecGetType(Vec, VecType *);
343bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode VecRegister(const char[],PetscErrorCode (*)(Vec));
34430de9b25SBarry Smith 
34535928de7SBarry Smith PETSC_EXTERN PetscErrorCode VecScatterSetData(VecScatter,Vec,IS,Vec,IS);
34635928de7SBarry Smith PETSC_EXTERN PetscErrorCode VecScatterCreate(MPI_Comm,VecScatter *);
347014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode);
348014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode);
349014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterDestroy(VecScatter*);
350246626dcSBarry Smith PETSC_EXTERN PetscErrorCode VecScatterSetUp(VecScatter);
351014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCopy(VecScatter,VecScatter *);
352014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterView(VecScatter,PetscViewer);
353685405a1SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecScatterViewFromOptions(VecScatter A,PetscObject obj,const char name[]) {return PetscObjectViewFromOptions((PetscObject)A,obj,name);}
354d27e6124SSatish Balay PETSC_EXTERN PetscErrorCode VecScatterRemap(VecScatter,PetscInt[],PetscInt[]);
355014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterGetMerged(VecScatter,PetscBool *);
3562195c698SBarry Smith 
357014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
358014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
359014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
360014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
361014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
362014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
363014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]);
364014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]);
365ab360428SBarry Smith 
3665edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
3675edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
3685edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
3695edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
3705edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
3715edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
3725edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray1dRead(Vec,PetscInt,PetscInt,PetscScalar *[]);
3735edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray1dRead(Vec,PetscInt,PetscInt,PetscScalar *[]);
3745edff71fSBarry Smith 
375014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPlaceArray(Vec,const PetscScalar[]);
376014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecResetArray(Vec);
377014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReplaceArray(Vec,const PetscScalar[]);
378014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrays(const Vec[],PetscInt,PetscScalar**[]);
379014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]);
38084cb2905SBarry Smith 
381014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecView(Vec,PetscViewer);
382014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecEqual(Vec,Vec,PetscBool *);
383014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLoad(Vec, PetscViewer);
3848ed539a5SBarry Smith 
385014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSize(Vec,PetscInt*);
386014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalSize(Vec,PetscInt*);
387014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRange(Vec,PetscInt*,PetscInt*);
388014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRanges(Vec,const PetscInt *[]);
3898ed539a5SBarry Smith 
390014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping);
391014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
39288b03592SBarry Smith 
39388b03592SBarry Smith /*MC
39488b03592SBarry Smith    VecSetValueLocal - Set a single entry into a vector using the local numbering
39588b03592SBarry Smith 
39688b03592SBarry Smith    Synopsis:
397aaa7dc30SBarry Smith    #include <petscvec.h>
3988122439eSJed Brown    PetscErrorCode VecSetValueLocal(Vec v,PetscInt row,PetscScalar value, InsertMode mode);
39988b03592SBarry Smith 
40088b03592SBarry Smith    Not Collective
40188b03592SBarry Smith 
40288b03592SBarry Smith    Input Parameters:
40388b03592SBarry Smith +  v - the vector
40488b03592SBarry Smith .  row - the row location of the entry
40588b03592SBarry Smith .  value - the value to insert
40688b03592SBarry Smith -  mode - either INSERT_VALUES or ADD_VALUES
40788b03592SBarry Smith 
40888b03592SBarry Smith    Notes:
40988b03592SBarry Smith    For efficiency one should use VecSetValues() and set several or
41088b03592SBarry Smith    many values simultaneously if possible.
41188b03592SBarry Smith 
41288b03592SBarry Smith    These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd()
41388b03592SBarry Smith    MUST be called after all calls to VecSetValues() have been completed.
41488b03592SBarry Smith 
41588b03592SBarry Smith    VecSetValues() uses 0-based indices in Fortran as well as in C.
41688b03592SBarry Smith 
41788b03592SBarry Smith    Level: beginner
41888b03592SBarry Smith 
41988b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue()
42088b03592SBarry Smith M*/
42188b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);}
42288b03592SBarry Smith 
423014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
424014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*);
42590f02eecSBarry Smith 
426014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotBegin(Vec,Vec,PetscScalar *);
427014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotEnd(Vec,Vec,PetscScalar *);
428014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotBegin(Vec,Vec,PetscScalar *);
429014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotEnd(Vec,Vec,PetscScalar *);
430014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormBegin(Vec,NormType,PetscReal *);
431014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormEnd(Vec,NormType,PetscReal *);
432d3c178dbSBarry Smith 
433014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]);
434014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]);
435014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]);
436014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]);
437014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCommSplitReductionBegin(MPI_Comm);
438a751f32aSSatish Balay 
439a751f32aSSatish Balay 
440bda0d2a0SJed Brown typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES,VEC_SUBSET_OFF_PROC_ENTRIES} VecOption;
441014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOption(Vec,VecOption,PetscBool );
44290f02eecSBarry Smith 
4435c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray(Vec,PetscScalar**);
4445c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrayRead(Vec,const PetscScalar**);
4455c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray(Vec,PetscScalar**);
4465c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrayRead(Vec,const PetscScalar**);
447046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVector(Vec,Vec);
448046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVector(Vec,Vec);
449046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVectorRead(Vec,Vec);
450046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVectorRead(Vec,Vec);
4519521ec69SBarry Smith 
4529521ec69SBarry Smith /*@C
4539521ec69SBarry Smith    VecGetArrayPair - Accesses a pair of pointers for two vectors that may be common. When not common the first is read only
4549521ec69SBarry Smith 
4559521ec69SBarry Smith    Logically Collective on Vec
4569521ec69SBarry Smith 
457f5f57ec0SBarry Smith    Input Parameters:
4589521ec69SBarry Smith +  x - the vector
4599521ec69SBarry Smith -  y - the second vector
4609521ec69SBarry Smith 
461f5f57ec0SBarry Smith    Output Parameters:
4629521ec69SBarry Smith +  xv - location to put pointer to the first array
4639521ec69SBarry Smith -  yv - location to put pointer to the second array
4649521ec69SBarry Smith 
4659521ec69SBarry Smith    Level: developer
4669521ec69SBarry Smith 
467f5f57ec0SBarry Smith    Not available from Fortran
468f5f57ec0SBarry Smith 
4699521ec69SBarry Smith .seealso: VecGetArray(), VecGetArrayRead(), VecRestoreArrayPair()
4709521ec69SBarry Smith 
4719521ec69SBarry Smith @*/
472d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecGetArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv)
473d9ca1df4SBarry Smith {
474d9ca1df4SBarry Smith   PetscErrorCode ierr;
475d9ca1df4SBarry Smith 
476d9ca1df4SBarry Smith   PetscFunctionBegin;
477d9ca1df4SBarry Smith   ierr = VecGetArray(y,yv);CHKERRQ(ierr);
478d9ca1df4SBarry Smith   if (x != y) {
479d9ca1df4SBarry Smith     ierr = VecGetArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr);
480d9ca1df4SBarry Smith   } else {
481d9ca1df4SBarry Smith     *xv = *yv;
482d9ca1df4SBarry Smith   }
483d9ca1df4SBarry Smith   PetscFunctionReturn(0);
484d9ca1df4SBarry Smith }
4859521ec69SBarry Smith 
4869521ec69SBarry Smith /*@C
4879521ec69SBarry Smith    VecRestoreArrayPair - Returns a pair of pointers for two vectors that may be common. When not common the first is read only
4889521ec69SBarry Smith 
4899521ec69SBarry Smith    Logically Collective on Vec
4909521ec69SBarry Smith 
491f5f57ec0SBarry Smith    Input Parameters:
4929521ec69SBarry Smith +  x - the vector
4939521ec69SBarry Smith -  y - the second vector
4949521ec69SBarry Smith 
495f5f57ec0SBarry Smith    Output Parameters:
4969521ec69SBarry Smith +  xv - location to put pointer to the first array
4979521ec69SBarry Smith -  yv - location to put pointer to the second array
4989521ec69SBarry Smith 
4999521ec69SBarry Smith    Level: developer
5009521ec69SBarry Smith 
501f5f57ec0SBarry Smith    Not available from Fortran
502f5f57ec0SBarry Smith 
5039521ec69SBarry Smith .seealso: VecGetArray(), VecGetArrayRead(), VecGetArrayPair()
5049521ec69SBarry Smith 
5059521ec69SBarry Smith @*/
506d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecRestoreArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv)
507d9ca1df4SBarry Smith {
508d9ca1df4SBarry Smith   PetscErrorCode ierr;
509d9ca1df4SBarry Smith 
510d9ca1df4SBarry Smith   PetscFunctionBegin;
511d9ca1df4SBarry Smith   ierr = VecRestoreArray(y,yv);CHKERRQ(ierr);
512d9ca1df4SBarry Smith   if (x != y) {
513d9ca1df4SBarry Smith     ierr = VecRestoreArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr);
514d9ca1df4SBarry Smith   }
515d9ca1df4SBarry Smith   PetscFunctionReturn(0);
516d9ca1df4SBarry Smith }
517d9ca1df4SBarry Smith 
518d9ca1df4SBarry Smith #if defined(PETSC_USE_DEBUG)
51938fecf7cSJunchao Zhang PETSC_EXTERN PetscErrorCode VecLockReadPush(Vec);
52038fecf7cSJunchao Zhang PETSC_EXTERN PetscErrorCode VecLockReadPop(Vec);
52138fecf7cSJunchao Zhang /* We also have a non-public VecLockWriteSet_Private() in vecimpl.h */
522d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockGet(Vec,PetscInt*);
52338fecf7cSJunchao Zhang #define VecSetErrorIfLocked(x,arg) \
5248e89e9e2SJunchao Zhang   do { \
5258e89e9e2SJunchao Zhang     PetscInt       __st; \
5268e89e9e2SJunchao Zhang     PetscErrorCode __ierr; \
5278e89e9e2SJunchao Zhang     __ierr = VecLockGet(x,&__st);CHKERRQ(__ierr); \
5288e89e9e2SJunchao Zhang     if (__st != 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE," Vec is already locked for read-only or read/write access, argument # %d",arg); \
5298e89e9e2SJunchao Zhang   } while (0)
53038fecf7cSJunchao Zhang 
53138fecf7cSJunchao Zhang /* The three are deprecated */
53238fecf7cSJunchao Zhang PETSC_DEPRECATED("Use VecLockReadPush (since v3.11)") PetscErrorCode VecLockPush(Vec);
53338fecf7cSJunchao Zhang PETSC_DEPRECATED("Use VecLockReadPop (since v3.11)")  PetscErrorCode VecLockPop(Vec);
53438fecf7cSJunchao Zhang #define VecLocked(x,arg) VecSetErrorIfLocked(x,arg)
535d9ca1df4SBarry Smith #else
536*45fccdb2SJunchao Zhang #define VecLockReadPush(x)           0
537*45fccdb2SJunchao Zhang #define VecLockReadPop(x)            0
53838fecf7cSJunchao Zhang #define VecLockGet(x,s)              *(s) = 0
539*45fccdb2SJunchao Zhang #define VecSetErrorIfLocked(x,arg)   0
54038fecf7cSJunchao Zhang /* The three are deprecated */
541*45fccdb2SJunchao Zhang #define VecLockPush(x)               0
542*45fccdb2SJunchao Zhang #define VecLockPop(x)                0
543*45fccdb2SJunchao Zhang #define VecLocked(x,arg)             0
544d9ca1df4SBarry Smith #endif
545e1fa1e0fSSatish Balay 
54662796dfbSBarry Smith PETSC_EXTERN PetscErrorCode VecValidValues(Vec,PetscInt,PetscBool);
547f7ecc322SBarry Smith 
54815091d37SBarry Smith /*
54915091d37SBarry Smith     These numbers need to match the entries in
5503c94ec11SBarry Smith   the function table in vecimpl.h
55115091d37SBarry Smith */
55292d03f5aSMatthew G. Knepley typedef enum { VECOP_DUPLICATE = 0, VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_VIEWNATIVE = 68, VECOP_LOADNATIVE = 69 } VecOperation;
553014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOperation(Vec,VecOperation,void(*)(void));
554b19c1e4cSBarry Smith 
555e182c471SBarry Smith /*
556e182c471SBarry Smith      Routines for dealing with ghosted vectors:
557e182c471SBarry Smith   vectors with ghost elements at the end of the array.
558e182c471SBarry Smith */
559014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMPISetGhost(Vec,PetscInt,const PetscInt[]);
560014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*);
561014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*);
562014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*);
563014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*);
564014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostGetLocalForm(Vec,Vec*);
565014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostRestoreLocalForm(Vec,Vec*);
5663efe6655SBarry Smith PETSC_EXTERN PetscErrorCode VecGhostIsLocalForm(Vec,Vec,PetscBool*);
567014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateBegin(Vec,InsertMode,ScatterMode);
568014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateEnd(Vec,InsertMode,ScatterMode);
569e182c471SBarry Smith 
570014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecConjugate(Vec);
571c92e3469SBarry Smith PETSC_EXTERN PetscErrorCode VecImaginaryPart(Vec);
572c92e3469SBarry Smith PETSC_EXTERN PetscErrorCode VecRealPart(Vec);
57334233285SBarry Smith 
574014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToAll(Vec,VecScatter*,Vec*);
575014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToZero(Vec,VecScatter*,Vec*);
576bba1ac68SSatish Balay 
5774473680cSBarry Smith PETSC_EXTERN PetscErrorCode ISComplementVec(IS,Vec,IS*);
5784473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecPow(Vec, PetscScalar);
5794473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecMedian(Vec, Vec, Vec, Vec);
580ce902467SBarry Smith PETSC_EXTERN PetscErrorCode VecWhichInactive(Vec, Vec, Vec, Vec, PetscBool, IS *);
5814473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetween(Vec, Vec, Vec, IS *);
5824473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetweenOrEqual(Vec, Vec, Vec, IS *);
5834473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichGreaterThan(Vec, Vec, IS * );
5844473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichLessThan(Vec, Vec, IS *);
5854473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichEqual(Vec, Vec, IS *);
5864473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISAXPY(Vec, IS, PetscScalar,Vec);
5878883cd90SMatthew G. Knepley PETSC_EXTERN PetscErrorCode VecISCopy(Vec, IS, ScatterMode, Vec);
5884473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISSet(Vec,IS, PetscScalar);
5894473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecBoundGradientProjection(Vec, Vec, Vec, Vec, Vec);
5904473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepBoundInfo(Vec,Vec,Vec,Vec,PetscReal*, PetscReal*,PetscReal*);
5914473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMax(Vec, Vec, PetscReal *);
5924473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMaxBounded(Vec,Vec,Vec,Vec,PetscReal*);
5934473680cSBarry Smith 
594014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaGetVector(PetscViewer, Vec);
595014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaPutVector(PetscViewer, Vec);
5967dbadf16SMatthew Knepley 
597d59c15a7SBarry Smith /*S
598d59c15a7SBarry Smith      Vecs - Collection of vectors where the data for the vectors is stored in
599759e7b9cSHong Zhang             one contiguous memory
600d59c15a7SBarry Smith 
601d59c15a7SBarry Smith    Level: advanced
602d59c15a7SBarry Smith 
603d59c15a7SBarry Smith    Notes:
604d59c15a7SBarry Smith     Temporary construct for handling multiply right hand side solves
605d59c15a7SBarry Smith 
606d59c15a7SBarry Smith     This is faked by storing a single vector that has enough array space for
607d59c15a7SBarry Smith     n vectors
608d59c15a7SBarry Smith 
609d59c15a7SBarry Smith   Concepts: parallel decomposition
610d59c15a7SBarry Smith 
611d59c15a7SBarry Smith S*/
61295fbd943SSatish Balay         struct _n_Vecs  {PetscInt n; Vec v;};
61395fbd943SSatish Balay typedef struct _n_Vecs* Vecs;
614e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDestroy(Vecs);
615e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeq(MPI_Comm,PetscInt,PetscInt,Vecs*);
616e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,PetscScalar*,Vecs*);
617e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDuplicate(Vecs,Vecs*);
618e9fa29b7SSatish Balay 
619e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
620b17c682bSKarl Rupp typedef struct _p_PetscViennaCLIndices* PetscViennaCLIndices;
621b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscViennaCLIndices*);
622b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesDestroy(PetscViennaCLIndices*);
623b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyToGPUSome_Public(Vec,PetscViennaCLIndices);
624b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyFromGPUSome_Public(Vec,PetscViennaCLIndices);
625b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateSeqViennaCL(MPI_Comm,PetscInt,Vec*);
626b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateMPIViennaCL(MPI_Comm,PetscInt,PetscInt,Vec*);
627b8ced49eSKarl Rupp #endif
628fd314934SBarry Smith #if defined(PETSC_HAVE_CUDA)
62982f73ecaSAlejandro Lamas Daviña typedef struct _p_PetscCUDAIndices* PetscCUDAIndices;
63082f73ecaSAlejandro Lamas Daviña typedef struct _p_VecScatterCUDAIndices_StoS* VecScatterCUDAIndices_StoS;
63182f73ecaSAlejandro Lamas Daviña typedef struct _p_VecScatterCUDAIndices_PtoP* VecScatterCUDAIndices_PtoP;
632959dcdf5SJunchao Zhang PETSC_EXTERN PetscErrorCode VecCUDACopyToGPUSome_Public(Vec,PetscCUDAIndices,ScatterMode);
633959dcdf5SJunchao Zhang PETSC_EXTERN PetscErrorCode VecCUDACopyFromGPUSome_Public(Vec,PetscCUDAIndices,ScatterMode);
634959dcdf5SJunchao Zhang PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec);
63582f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter);
63682f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateSeqCUDA(MPI_Comm,PetscInt,Vec*);
63776d6722aSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateSeqCUDAWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar*,Vec*);
63882f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateMPICUDA(MPI_Comm,PetscInt,PetscInt,Vec*);
63976d6722aSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateMPICUDAWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar*,Vec*);
640b17c682bSKarl Rupp #endif
641b17c682bSKarl Rupp 
642014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVecs(Vec,PetscInt*,Vec**);
643014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVec(Vec,PetscInt,Vec*);
644014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*);
645014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVec(Vec,PetscInt,Vec);
646014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*);
647014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSize(Vec,PetscInt*);
64845b63f25SDmitry Karpeev 
649c5929fdfSBarry Smith PETSC_EXTERN PetscErrorCode PetscOptionsGetVec(PetscOptions,const char[],const char[],Vec,PetscBool*);
6504325cce7SMatthew G Knepley PETSC_EXTERN PetscErrorCode VecChop(Vec,PetscReal);
651ab352700SBarry Smith 
652077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecGetLayout(Vec,PetscLayout*);
653077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecSetLayout(Vec,PetscLayout);
6549a6d0b0bSJed Brown 
6559a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode PetscSectionVecView(PetscSection, Vec, PetscViewer);
6569a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecGetValuesSection(Vec, PetscSection, PetscInt, PetscScalar **);
6579a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesSection(Vec, PetscSection, PetscInt, PetscScalar [], InsertMode);
65822d96a0cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscSectionVecNorm(PetscSection, PetscSection, Vec, NormType, PetscReal []);
6599a6d0b0bSJed Brown 
660ad59f098SAta Mesgarnejad PETSC_EXTERN PetscErrorCode PetscSFCreateFromZero(MPI_Comm,Vec,PetscSF*);
661ad59f098SAta Mesgarnejad 
6625543aaa0SToby Isaac /*S
6635543aaa0SToby Isaac   VecTagger - Object used to manage the tagging of a subset of indices based on the values of a vector.  The
6645543aaa0SToby Isaac               motivating application is the selection of cells for refinement or coarsening based on vector containing
6655543aaa0SToby Isaac               the values in an error indicator metric.
6665543aaa0SToby Isaac 
6675543aaa0SToby Isaac   Level: advanced
6685543aaa0SToby Isaac S*/
66961bbf837SToby Isaac typedef struct _p_VecTagger *VecTagger;
6705543aaa0SToby Isaac 
6715543aaa0SToby Isaac /*J
6725543aaa0SToby Isaac   VecTaggerType - String with the name of a VecTagger type
6735543aaa0SToby Isaac 
6745543aaa0SToby Isaac   Level: advanced
6755543aaa0SToby Isaac J*/
6765543aaa0SToby Isaac typedef const char* VecTaggerType;
6774178c1b7SToby Isaac /* tag where the vector values are in a box of explicitly defined values */
678cab77d58SToby Isaac #define VECTAGGERABSOLUTE   "absolute"
6794178c1b7SToby Isaac /* tag where the vector values are in a box of values relative to the set of all values in the vector */
6805543aaa0SToby Isaac #define VECTAGGERRELATIVE   "relative"
6815543aaa0SToby Isaac /* tag where the vector values are in a relative range of the *cumulative distribution* of values in the vector */
6821d1d0dd8SToby Isaac #define VECTAGGERCDF        "cdf"
6835543aaa0SToby Isaac /* tag a vector as the union of other tags */
6845543aaa0SToby Isaac #define VECTAGGEROR         "or"
6855543aaa0SToby Isaac /* tag a vector as the intersection of other tags */
6865543aaa0SToby Isaac #define VECTAGGERAND        "and"
6875543aaa0SToby Isaac 
6885543aaa0SToby Isaac PETSC_EXTERN PetscClassId VEC_TAGGER_CLASSID;
68961bbf837SToby Isaac PETSC_EXTERN PetscFunctionList VecTaggerList;
69061bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRegister(const char[],PetscErrorCode (*) (VecTagger));
6915543aaa0SToby Isaac 
6925543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCreate(MPI_Comm,VecTagger *);
693cab77d58SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetBlockSize(VecTagger,PetscInt);
694cab77d58SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetBlockSize(VecTagger,PetscInt*);
6955543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetType(VecTagger,VecTaggerType);
6965543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetType(VecTagger,VecTaggerType *);
6975543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetInvert(VecTagger,PetscBool);
6985543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetInvert(VecTagger,PetscBool*);
6995543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetFromOptions(VecTagger);
7005543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetUp(VecTagger);
7015543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerView(VecTagger,PetscViewer);
7025543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerComputeIS(VecTagger,Vec,IS *);
7035543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerDestroy(VecTagger *);
7045543aaa0SToby Isaac 
7054178c1b7SToby Isaac /*S
7064178c1b7SToby 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
7074178c1b7SToby Isaac    such that real(min) <= real(z) <= real(max) and imag(min) <= imag(z) <= imag(max).  INF is an acceptable endpoint.
70861bbf837SToby Isaac 
7094178c1b7SToby Isaac    Level: beginner
71061bbf837SToby Isaac 
7114178c1b7SToby Isaac .seealso: VecTaggerComputeIntervals()
7124178c1b7SToby Isaac S*/
7134178c1b7SToby Isaac typedef struct {
7144178c1b7SToby Isaac   PetscScalar min;
7154178c1b7SToby Isaac   PetscScalar max;
7164178c1b7SToby Isaac } VecTaggerBox;
7174178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerComputeBoxes(VecTagger,Vec,PetscInt *,VecTaggerBox **);
7184178c1b7SToby Isaac 
7194178c1b7SToby Isaac 
7204178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAbsoluteSetBox(VecTagger,VecTaggerBox *);
7214178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAbsoluteGetBox(VecTagger,const VecTaggerBox **);
7224178c1b7SToby Isaac 
7234178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRelativeSetBox(VecTagger,VecTaggerBox *);
7244178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRelativeGetBox(VecTagger,const VecTaggerBox **);
7254178c1b7SToby Isaac 
7264178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFSetBox(VecTagger,VecTaggerBox *);
7274178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFGetBox(VecTagger,const VecTaggerBox **);
7281d1d0dd8SToby Isaac 
7291d1d0dd8SToby Isaac /*E
7301d1d0dd8SToby 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.
7311d1d0dd8SToby Isaac 
7321d1d0dd8SToby Isaac   Level: advanced
7331d1d0dd8SToby Isaac .seealso: VecTaggerCDFSetMethod(), VecTaggerCDFMethods
7341d1d0dd8SToby Isaac E*/
7351d1d0dd8SToby Isaac typedef enum {VECTAGGER_CDF_GATHER,VECTAGGER_CDF_ITERATIVE,VECTAGGER_CDF_NUM_METHODS} VecTaggerCDFMethod;
7361d1d0dd8SToby Isaac PETSC_EXTERN const char *const VecTaggerCDFMethods[];
7371d1d0dd8SToby Isaac 
7381d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFSetMethod(VecTagger,VecTaggerCDFMethod);
7391d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFGetMethod(VecTagger,VecTaggerCDFMethod*);
7401d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFIterativeSetTolerances(VecTagger,PetscInt,PetscReal,PetscReal);
7411d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFIterativeGetTolerances(VecTagger,PetscInt*,PetscReal*,PetscReal*);
74261bbf837SToby Isaac 
743497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerOrSetSubs(VecTagger,PetscInt,VecTagger*,PetscCopyMode);
744497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerOrGetSubs(VecTagger,PetscInt*,VecTagger**);
74561bbf837SToby Isaac 
746497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAndSetSubs(VecTagger,PetscInt,VecTagger*,PetscCopyMode);
747497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAndGetSubs(VecTagger,PetscInt*,VecTagger**);
74861bbf837SToby Isaac 
74961bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerInitializePackage(void);
75061bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerFinalizePackage(void);
75161bbf837SToby Isaac 
7522eac72dbSBarry Smith #endif
753