xref: /petsc/include/petscvec.h (revision b17c682b0f8f1f82c493daa72b7676b957659c0f)
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
8909321671SBarry Smith     VecType - String with the name of a PETSc vector or the creation function
9009321671SBarry Smith        with an optional dynamic library name, for example
9109321671SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:myveccreate()
9209321671SBarry Smith 
9309321671SBarry Smith    Level: beginner
9409321671SBarry Smith 
9509321671SBarry Smith .seealso: VecSetType(), Vec
9676bdecfbSBarry Smith J*/
9719fd82e9SBarry Smith typedef const char* VecType;
980676abe4SMatthew Knepley #define VECSEQ         "seq"
990676abe4SMatthew Knepley #define VECMPI         "mpi"
100f48c50deSBarry Smith #define VECSTANDARD    "standard"   /* seq on one process and mpi on several */
1010676abe4SMatthew Knepley #define VECSHARED      "shared"
102765467adSMatthew Knepley #define VECSIEVE       "sieve"
1038154be41SBarry Smith #define VECSEQCUSP     "seqcusp"
1048154be41SBarry Smith #define VECMPICUSP     "mpicusp"
1058154be41SBarry Smith #define VECCUSP        "cusp"       /* seqcusp on one process and mpicusp on several */
106*b17c682bSKarl Rupp #define VECSEQVIENNACL "seqviennacl"
107*b17c682bSKarl Rupp #define VECMPIVIENNACL "mpiviennacl"
108*b17c682bSKarl Rupp #define VECVIENNACL    "viennacl"   /* seqviennacl on one process and mpiviennacl on several */
1091aae2881SJed Brown #define VECNEST        "nest"
1101bf2b38bSKerry Stevens #define VECSEQPTHREAD  "seqpthread"
1114b71561bSShri Abhyankar #define VECMPIPTHREAD  "mpipthread"
112bf2c1783SBarry Smith #define VECPTHREAD     "pthread"    /* seqpthread on one process and mpipthread on several */
1132eac72dbSBarry Smith 
1144b71561bSShri Abhyankar 
115fd487807SMatthew Knepley /* Logging support */
1160700a824SBarry Smith #define    VEC_FILE_CLASSID 1211214
117014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_CLASSID;
118014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_SCATTER_CLASSID;
1198ba1e511SMatthew Knepley 
12045b63f25SDmitry Karpeev 
121607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecInitializePackage(void);
122014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecFinalizePackage(void);
123fd487807SMatthew Knepley 
124014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreate(MPI_Comm,Vec*);
125014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeq(MPI_Comm,PetscInt,Vec*);
126014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*);
127014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*);
128014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar[],Vec*);
129014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*);
130014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetFromOptions(Vec);
131146574abSBarry Smith PETSC_EXTERN PetscErrorCode VecViewFromOptions(Vec,const char[],const char[]);
1320d2bece7SBarry Smith 
133014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetUp(Vec);
134014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroy(Vec*);
135014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecZeroEntries(Vec);
136014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOptionsPrefix(Vec,const char[]);
137014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAppendOptionsPrefix(Vec,const char[]);
138014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOptionsPrefix(Vec,const char*[]);
139f69a0ea3SMatthew Knepley 
140014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetSizes(Vec,PetscInt,PetscInt);
141fd487807SMatthew Knepley 
14252b6e47cSBarry Smith PETSC_EXTERN PetscErrorCode VecDotNorm2(Vec,Vec,PetscScalar*,PetscReal*);
143014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDot(Vec,Vec,PetscScalar*);
14467392de3SBarry Smith PETSC_EXTERN PetscErrorCode VecDotRealPart(Vec,Vec,PetscReal*);
145014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDot(Vec,Vec,PetscScalar*);
146014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]);
147014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]);
148014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSubVector(Vec,IS,Vec*);
149014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreSubVector(Vec,IS,Vec*);
150cddf8d76SBarry Smith 
15109321671SBarry Smith /*E
15209321671SBarry Smith     NormType - determines what type of norm to compute
15309321671SBarry Smith 
15409321671SBarry Smith     Level: beginner
15509321671SBarry Smith 
15609321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm()
15709321671SBarry Smith E*/
1589dcbbd2bSBarry Smith typedef enum {NORM_1=0,NORM_2=1,NORM_FROBENIUS=2,NORM_INFINITY=3,NORM_1_AND_2=4} NormType;
1596a6fc655SJed Brown PETSC_EXTERN const char *const NormTypes[];
160cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY
16109321671SBarry Smith 
1629b250c83SBarry Smith /*MC
1639b250c83SBarry Smith      NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum
1649b250c83SBarry Smith 
1659b250c83SBarry Smith    Level: beginner
1669b250c83SBarry Smith 
1679b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS,
1689b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
1699b250c83SBarry Smith 
1709b250c83SBarry Smith M*/
1719b250c83SBarry Smith 
1729b250c83SBarry Smith /*MC
1739b250c83SBarry Smith      NORM_2 - the two norm, ||v|| = sqrt(sum_i (v_i)^2) (vectors only)
1749b250c83SBarry Smith 
1759b250c83SBarry Smith    Level: beginner
1769b250c83SBarry Smith 
1779b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS,
1789b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
1799b250c83SBarry Smith 
1809b250c83SBarry Smith M*/
1819b250c83SBarry Smith 
1829b250c83SBarry Smith /*MC
1839b250c83SBarry Smith      NORM_FROBENIUS - ||A|| = sqrt(sum_ij (A_ij)^2), same as NORM_2 for vectors
1849b250c83SBarry Smith 
1859b250c83SBarry Smith    Level: beginner
1869b250c83SBarry Smith 
1879b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
1889b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
1899b250c83SBarry Smith 
1909b250c83SBarry Smith M*/
1919b250c83SBarry Smith 
1929b250c83SBarry Smith /*MC
1939b250c83SBarry Smith      NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum
1949b250c83SBarry Smith 
1959b250c83SBarry Smith    Level: beginner
1969b250c83SBarry Smith 
1979b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
1989b250c83SBarry Smith            NORM_FROBINIUS, NORM_1_AND_2
1999b250c83SBarry Smith 
2009b250c83SBarry Smith M*/
2019b250c83SBarry Smith 
2029b250c83SBarry Smith /*MC
2039b250c83SBarry Smith      NORM_1_AND_2 - computes both the 1 and 2 norm of a vector
2049b250c83SBarry Smith 
2059b250c83SBarry Smith    Level: beginner
2069b250c83SBarry Smith 
2079b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
2089b250c83SBarry Smith            NORM_FROBINIUS, NORM_INFINITY
2099b250c83SBarry Smith 
2109b250c83SBarry Smith M*/
2119b250c83SBarry Smith 
2129b250c83SBarry Smith /*MC
2139b250c83SBarry Smith      NORM_MAX - see NORM_INFINITY
2149b250c83SBarry Smith 
215d41222bbSBarry Smith    Level: beginner
216d41222bbSBarry Smith 
2179b250c83SBarry Smith M*/
2189b250c83SBarry Smith 
219014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNorm(Vec,NormType,PetscReal *);
220014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *);
221014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormalize(Vec,PetscReal *);
222014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSum(Vec,PetscScalar*);
223014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMax(Vec,PetscInt*,PetscReal *);
224014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMin(Vec,PetscInt*,PetscReal *);
225014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScale(Vec,PetscScalar);
226014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCopy(Vec,Vec);
227014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetRandom(Vec,PetscRandom);
228014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSet(Vec,PetscScalar);
229014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSwap(Vec,Vec);
230014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPY(Vec,PetscScalar,Vec);
231014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBY(Vec,PetscScalar,PetscScalar,Vec);
232014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]);
233014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAYPX(Vec,PetscScalar,Vec);
234014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecWAXPY(Vec,PetscScalar,Vec,Vec);
235014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec);
236014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMax(Vec,Vec,Vec);
237014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMaxAbs(Vec,Vec,Vec);
238014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMin(Vec,Vec,Vec);
239014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMult(Vec,Vec,Vec);
240014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseDivide(Vec,Vec,Vec);
241014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMaxPointwiseDivide(Vec,Vec,PetscReal*);
242014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecShift(Vec,PetscScalar);
243014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReciprocal(Vec);
244014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPermute(Vec, IS, PetscBool );
245014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSqrtAbs(Vec);
246014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLog(Vec);
247014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecExp(Vec);
248014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAbs(Vec);
249014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicate(Vec,Vec*);
250014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicateVecs(Vec,PetscInt,Vec*[]);
251014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroyVecs(PetscInt, Vec*[]);
252014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNormAll(Vec,NormType,PetscReal[]);
253014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMaxAll(Vec,PetscInt [],PetscReal []);
254014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMinAll(Vec,PetscInt [],PetscReal []);
255014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScaleAll(Vec,const PetscScalar[]);
2564a560884SBarry Smith 
257014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNorm(Vec,PetscInt,NormType,PetscReal*);
258014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *);
259014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *);
260014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScale(Vec,PetscInt,PetscScalar);
261014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideSet(Vec,PetscInt,PetscScalar);
262954b3ec6SBarry Smith 
263954b3ec6SBarry Smith 
264014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGather(Vec,PetscInt,Vec,InsertMode);
265014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatter(Vec,PetscInt,Vec,InsertMode);
266014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGatherAll(Vec,Vec[],InsertMode);
267014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatterAll(Vec[],Vec,InsertMode);
268d2655a18SBarry Smith 
269014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
270014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]);
271014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyBegin(Vec);
272014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyEnd(Vec);
273014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashSetInitialSize(Vec,PetscInt,PetscInt);
274014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashView(Vec,PetscViewer);
275014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*);
27662dc5420SSatish Balay 
27730de9b25SBarry Smith /*MC
27830de9b25SBarry Smith    VecSetValue - Set a single entry into a vector.
27930de9b25SBarry Smith 
28030de9b25SBarry Smith    Synopsis:
281f2ba6396SBarry Smith    #include "petscvec.h"
2828122439eSJed Brown    PetscErrorCode VecSetValue(Vec v,PetscInt row,PetscScalar value, InsertMode mode);
28330de9b25SBarry Smith 
28430de9b25SBarry Smith    Not Collective
28530de9b25SBarry Smith 
28630de9b25SBarry Smith    Input Parameters:
28730de9b25SBarry Smith +  v - the vector
28830de9b25SBarry Smith .  row - the row location of the entry
28930de9b25SBarry Smith .  value - the value to insert
29030de9b25SBarry Smith -  mode - either INSERT_VALUES or ADD_VALUES
29130de9b25SBarry Smith 
29230de9b25SBarry Smith    Notes:
29330de9b25SBarry Smith    For efficiency one should use VecSetValues() and set several or
29430de9b25SBarry Smith    many values simultaneously if possible.
29530de9b25SBarry Smith 
2961d73ed98SBarry Smith    These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd()
2971d73ed98SBarry Smith    MUST be called after all calls to VecSetValues() have been completed.
2981d73ed98SBarry Smith 
2991d73ed98SBarry Smith    VecSetValues() uses 0-based indices in Fortran as well as in C.
3001d73ed98SBarry Smith 
3011d73ed98SBarry Smith    Level: beginner
3021d73ed98SBarry Smith 
3031d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal()
3041d73ed98SBarry Smith M*/
30550755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);}
3061d73ed98SBarry Smith 
30730de9b25SBarry Smith 
308014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetBlockSize(Vec,PetscInt);
309014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetBlockSize(Vec,PetscInt*);
310014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
3118ed539a5SBarry Smith 
312fd487807SMatthew Knepley /* Dynamic creation and loading functions */
313140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList VecList;
314014dd563SJed Brown PETSC_EXTERN PetscBool         VecRegisterAllCalled;
31519fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecSetType(Vec, VecType);
31619fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecGetType(Vec, VecType *);
317bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode VecRegister(const char[],PetscErrorCode (*)(Vec));
318607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecRegisterAll(void);
31930de9b25SBarry Smith 
320014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreate(Vec,IS,Vec,IS,VecScatter *);
321014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateEmpty(MPI_Comm,VecScatter *);
322014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateLocal(VecScatter,PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt);
323014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode);
324014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode);
325014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterDestroy(VecScatter*);
326014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCopy(VecScatter,VecScatter *);
327014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterView(VecScatter,PetscViewer);
328014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterRemap(VecScatter,PetscInt *,PetscInt*);
329014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterGetMerged(VecScatter,PetscBool *);
3302195c698SBarry Smith 
331014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
332014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
333014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
334014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
335014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
336014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
337014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]);
338014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]);
339ab360428SBarry Smith 
340014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPlaceArray(Vec,const PetscScalar[]);
341014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecResetArray(Vec);
342014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReplaceArray(Vec,const PetscScalar[]);
343014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrays(const Vec[],PetscInt,PetscScalar**[]);
344014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]);
34584cb2905SBarry Smith 
346014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecView(Vec,PetscViewer);
347014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecEqual(Vec,Vec,PetscBool *);
348014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLoad(Vec, PetscViewer);
3498ed539a5SBarry Smith 
350014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSize(Vec,PetscInt*);
351014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalSize(Vec,PetscInt*);
352014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRange(Vec,PetscInt*,PetscInt*);
353014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRanges(Vec,const PetscInt *[]);
3548ed539a5SBarry Smith 
355014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping);
356014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
35788b03592SBarry Smith 
35888b03592SBarry Smith /*MC
35988b03592SBarry Smith    VecSetValueLocal - Set a single entry into a vector using the local numbering
36088b03592SBarry Smith 
36188b03592SBarry Smith    Synopsis:
362f2ba6396SBarry Smith    #include "petscvec.h"
3638122439eSJed Brown    PetscErrorCode VecSetValueLocal(Vec v,PetscInt row,PetscScalar value, InsertMode mode);
36488b03592SBarry Smith 
36588b03592SBarry Smith    Not Collective
36688b03592SBarry Smith 
36788b03592SBarry Smith    Input Parameters:
36888b03592SBarry Smith +  v - the vector
36988b03592SBarry Smith .  row - the row location of the entry
37088b03592SBarry Smith .  value - the value to insert
37188b03592SBarry Smith -  mode - either INSERT_VALUES or ADD_VALUES
37288b03592SBarry Smith 
37388b03592SBarry Smith    Notes:
37488b03592SBarry Smith    For efficiency one should use VecSetValues() and set several or
37588b03592SBarry Smith    many values simultaneously if possible.
37688b03592SBarry Smith 
37788b03592SBarry Smith    These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd()
37888b03592SBarry Smith    MUST be called after all calls to VecSetValues() have been completed.
37988b03592SBarry Smith 
38088b03592SBarry Smith    VecSetValues() uses 0-based indices in Fortran as well as in C.
38188b03592SBarry Smith 
38288b03592SBarry Smith    Level: beginner
38388b03592SBarry Smith 
38488b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue()
38588b03592SBarry Smith M*/
38688b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);}
38788b03592SBarry Smith 
388014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping);
389014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
390014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping*);
391014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*);
39290f02eecSBarry Smith 
393014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotBegin(Vec,Vec,PetscScalar *);
394014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotEnd(Vec,Vec,PetscScalar *);
395014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotBegin(Vec,Vec,PetscScalar *);
396014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotEnd(Vec,Vec,PetscScalar *);
397014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormBegin(Vec,NormType,PetscReal *);
398014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormEnd(Vec,NormType,PetscReal *);
399d3c178dbSBarry Smith 
400014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]);
401014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]);
402014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]);
403014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]);
404014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCommSplitReductionBegin(MPI_Comm);
405a751f32aSSatish Balay 
406a751f32aSSatish Balay 
4079533e83fSBarry Smith typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES} VecOption;
408014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOption(Vec,VecOption,PetscBool );
40990f02eecSBarry Smith 
4105c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray(Vec,PetscScalar**);
4115c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrayRead(Vec,const PetscScalar**);
4125c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray(Vec,PetscScalar**);
4135c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrayRead(Vec,const PetscScalar**);
414e1fa1e0fSSatish Balay 
41562796dfbSBarry Smith PETSC_EXTERN PetscErrorCode VecValidValues(Vec,PetscInt,PetscBool);
416f7ecc322SBarry Smith 
417014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecContourScale(Vec,PetscReal,PetscReal);
418522c5e43SBarry Smith 
41915091d37SBarry Smith /*
42015091d37SBarry Smith     These numbers need to match the entries in
4213c94ec11SBarry Smith   the function table in vecimpl.h
42215091d37SBarry Smith */
4232dcb2ebcSBarry Smith typedef enum { VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_DUPLICATE = 0} VecOperation;
424014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOperation(Vec,VecOperation,void(*)(void));
425b19c1e4cSBarry Smith 
426e182c471SBarry Smith /*
427e182c471SBarry Smith      Routines for dealing with ghosted vectors:
428e182c471SBarry Smith   vectors with ghost elements at the end of the array.
429e182c471SBarry Smith */
430014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMPISetGhost(Vec,PetscInt,const PetscInt[]);
431014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*);
432014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*);
433014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*);
434014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*);
435014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostGetLocalForm(Vec,Vec*);
436014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostRestoreLocalForm(Vec,Vec*);
4373efe6655SBarry Smith PETSC_EXTERN PetscErrorCode VecGhostIsLocalForm(Vec,Vec,PetscBool*);
438014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateBegin(Vec,InsertMode,ScatterMode);
439014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateEnd(Vec,InsertMode,ScatterMode);
440e182c471SBarry Smith 
441014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecConjugate(Vec);
44234233285SBarry Smith 
443014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToAll(Vec,VecScatter*,Vec*);
444014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToZero(Vec,VecScatter*,Vec*);
445bba1ac68SSatish Balay 
446014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaGetVector(PetscViewer, Vec);
447014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaPutVector(PetscViewer, Vec);
4487dbadf16SMatthew Knepley 
449d59c15a7SBarry Smith /*S
450d59c15a7SBarry Smith      Vecs - Collection of vectors where the data for the vectors is stored in
451759e7b9cSHong Zhang             one contiguous memory
452d59c15a7SBarry Smith 
453d59c15a7SBarry Smith    Level: advanced
454d59c15a7SBarry Smith 
455d59c15a7SBarry Smith    Notes:
456d59c15a7SBarry Smith     Temporary construct for handling multiply right hand side solves
457d59c15a7SBarry Smith 
458d59c15a7SBarry Smith     This is faked by storing a single vector that has enough array space for
459d59c15a7SBarry Smith     n vectors
460d59c15a7SBarry Smith 
461d59c15a7SBarry Smith   Concepts: parallel decomposition
462d59c15a7SBarry Smith 
463d59c15a7SBarry Smith S*/
46495fbd943SSatish Balay         struct _n_Vecs  {PetscInt n; Vec v;};
46595fbd943SSatish Balay typedef struct _n_Vecs* Vecs;
466e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDestroy(Vecs);
467e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeq(MPI_Comm,PetscInt,PetscInt,Vecs*);
468e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,PetscScalar*,Vecs*);
469e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDuplicate(Vecs,Vecs*);
470e9fa29b7SSatish Balay 
4718154be41SBarry Smith #if defined(PETSC_HAVE_CUSP)
472a5891931SBarry Smith typedef struct _p_PetscCUSPIndices* PetscCUSPIndices;
473014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCUSPIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscCUSPIndices*);
474014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCUSPIndicesDestroy(PetscCUSPIndices*);
475014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyToGPUSome_Public(Vec,PetscCUSPIndices);
476014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyFromGPUSome_Public(Vec,PetscCUSPIndices);
47790c53902SBarry Smith 
478bd72cc96SPaul Mullowney #if defined(PETSC_HAVE_TXPETSCGPU)
479014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPResetIndexBuffersFlagsGPU_Public(PetscCUSPIndices);
480014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopySomeToContiguousBufferGPU_Public(Vec,PetscCUSPIndices);
481014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopySomeFromContiguousBufferGPU_Public(Vec,PetscCUSPIndices);
482014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec,ScatterMode);
483014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter);
484bd72cc96SPaul Mullowney #endif
485bd72cc96SPaul Mullowney 
486014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqCUSP(MPI_Comm,PetscInt,Vec*);
487014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPICUSP(MPI_Comm,PetscInt,PetscInt,Vec*);
488a5891931SBarry Smith #endif
489e9fa29b7SSatish Balay 
490*b17c682bSKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
491*b17c682bSKarl Rupp typedef struct _p_PetscViennaCLIndices* PetscViennaCLIndices;
492*b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscViennaCLIndices*);
493*b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesDestroy(PetscViennaCLIndices*);
494*b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyToGPUSome_Public(Vec,PetscViennaCLIndices);
495*b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyFromGPUSome_Public(Vec,PetscViennaCLIndices);
496*b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateSeqViennaCL(MPI_Comm,PetscInt,Vec*);
497*b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateMPIViennaCL(MPI_Comm,PetscInt,PetscInt,Vec*);
498*b17c682bSKarl Rupp #endif
499*b17c682bSKarl Rupp 
500014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVecs(Vec,PetscInt*,Vec**);
501014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVec(Vec,PetscInt,Vec*);
502014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*);
503014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVec(Vec,PetscInt,Vec);
504014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*);
505014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSize(Vec,PetscInt*);
50645b63f25SDmitry Karpeev 
507ab352700SBarry Smith PETSC_EXTERN PetscErrorCode PetscOptionsVec(const char[],const char[],const char[],Vec,PetscBool*);
5084325cce7SMatthew G Knepley PETSC_EXTERN PetscErrorCode VecChop(Vec,PetscReal);
509ab352700SBarry Smith 
510077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecGetLayout(Vec,PetscLayout*);
511077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecSetLayout(Vec,PetscLayout);
5129a6d0b0bSJed Brown 
5139a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode PetscSectionVecView(PetscSection, Vec, PetscViewer);
5149a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecGetValuesSection(Vec, PetscSection, PetscInt, PetscScalar **);
5159a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesSection(Vec, PetscSection, PetscInt, PetscScalar [], InsertMode);
5169a6d0b0bSJed Brown 
5172eac72dbSBarry Smith #endif
5182b82984cSSatish Balay 
519