xref: /petsc/include/petscvec.h (revision ab2fbef6e007314ebd3fec67b0583297ffc3cc6d)
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
90a835dfdSSatish Balay #include "petscis.h"
105f5f199fSBarry Smith 
11e9fa29b7SSatish Balay PETSC_EXTERN_CXX_BEGIN
122eac72dbSBarry Smith 
1309321671SBarry Smith /*S
1409321671SBarry Smith      Vec - Abstract PETSc vector object
1509321671SBarry Smith 
1609321671SBarry Smith    Level: beginner
1709321671SBarry Smith 
1809321671SBarry Smith   Concepts: field variables, unknowns, arrays
1909321671SBarry Smith 
2009321671SBarry Smith .seealso:  VecCreate(), VecType, VecSetType()
2109321671SBarry Smith S*/
22f09e8eb9SSatish Balay typedef struct _p_Vec*         Vec;
2309321671SBarry Smith 
2409321671SBarry Smith /*S
2509321671SBarry Smith      VecScatter - Object used to manage communication of data
2609321671SBarry Smith        between vectors in parallel. Manages both scatters and gathers
2709321671SBarry Smith 
2809321671SBarry Smith    Level: beginner
2909321671SBarry Smith 
3009321671SBarry Smith   Concepts: scatter
3109321671SBarry Smith 
3209321671SBarry Smith .seealso:  VecScatterCreate(), VecScatterBegin(), VecScatterEnd()
3309321671SBarry Smith S*/
34f09e8eb9SSatish Balay typedef struct _p_VecScatter*  VecScatter;
3509321671SBarry Smith 
3609321671SBarry Smith /*E
37398c84b2SBarry Smith   ScatterMode - Determines the direction of a scatter
38398c84b2SBarry Smith 
39398c84b2SBarry Smith   Level: beginner
40398c84b2SBarry Smith 
41398c84b2SBarry Smith .seealso: VecScatter, VecScatterBegin(), VecScatterEnd()
42398c84b2SBarry Smith E*/
43398c84b2SBarry Smith typedef enum {SCATTER_FORWARD=0, SCATTER_REVERSE=1, SCATTER_FORWARD_LOCAL=2, SCATTER_REVERSE_LOCAL=3, SCATTER_LOCAL=2} ScatterMode;
44398c84b2SBarry Smith 
45398c84b2SBarry Smith /*MC
46398c84b2SBarry Smith     SCATTER_FORWARD - Scatters the values as dictated by the VecScatterCreate() call
47398c84b2SBarry Smith 
48398c84b2SBarry Smith     Level: beginner
49398c84b2SBarry Smith 
50398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD_LOCAL,
51398c84b2SBarry Smith           SCATTER_REVERSE_LOCAL
52398c84b2SBarry Smith 
53398c84b2SBarry Smith M*/
54398c84b2SBarry Smith 
55398c84b2SBarry Smith /*MC
56398c84b2SBarry Smith     SCATTER_REVERSE - Moves the values in the opposite direction then the directions indicated in
57398c84b2SBarry Smith          in the VecScatterCreate()
58398c84b2SBarry Smith 
59398c84b2SBarry Smith     Level: beginner
60398c84b2SBarry Smith 
61398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL,
62398c84b2SBarry Smith           SCATTER_REVERSE_LOCAL
63398c84b2SBarry Smith 
64398c84b2SBarry Smith M*/
65398c84b2SBarry Smith 
66398c84b2SBarry Smith /*MC
67398c84b2SBarry Smith     SCATTER_FORWARD_LOCAL - Scatters the values as dictated by the VecScatterCreate() call except NO parallel communication
68398c84b2SBarry Smith        is done. Any variables that have be moved between processes are ignored
69398c84b2SBarry Smith 
70398c84b2SBarry Smith     Level: developer
71398c84b2SBarry Smith 
72398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD,
73398c84b2SBarry Smith           SCATTER_REVERSE_LOCAL
74398c84b2SBarry Smith 
75398c84b2SBarry Smith M*/
76398c84b2SBarry Smith 
77398c84b2SBarry Smith /*MC
78398c84b2SBarry Smith     SCATTER_REVERSE_LOCAL - Moves the values in the opposite direction then the directions indicated in
79398c84b2SBarry Smith          in the VecScatterCreate()  except NO parallel communication
80398c84b2SBarry Smith        is done. Any variables that have be moved between processes are ignored
81398c84b2SBarry Smith 
82398c84b2SBarry Smith     Level: developer
83398c84b2SBarry Smith 
84398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL,
85398c84b2SBarry Smith           SCATTER_REVERSE
86398c84b2SBarry Smith 
87398c84b2SBarry Smith M*/
88398c84b2SBarry Smith 
8976bdecfbSBarry Smith /*J
9009321671SBarry Smith     VecType - String with the name of a PETSc vector or the creation function
9109321671SBarry Smith        with an optional dynamic library name, for example
9209321671SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:myveccreate()
9309321671SBarry Smith 
9409321671SBarry Smith    Level: beginner
9509321671SBarry Smith 
9609321671SBarry Smith .seealso: VecSetType(), Vec
9776bdecfbSBarry Smith J*/
98a313700dSBarry Smith #define VecType char*
990676abe4SMatthew Knepley #define VECSEQ         "seq"
1000676abe4SMatthew Knepley #define VECMPI         "mpi"
101f48c50deSBarry Smith #define VECSTANDARD    "standard"   /* seq on one process and mpi on several */
1020676abe4SMatthew Knepley #define VECSHARED      "shared"
103765467adSMatthew Knepley #define VECSIEVE       "sieve"
1048154be41SBarry Smith #define VECSEQCUSP     "seqcusp"
1058154be41SBarry Smith #define VECMPICUSP     "mpicusp"
1068154be41SBarry Smith #define VECCUSP        "cusp"       /* seqcusp on one process and mpicusp on several */
1071aae2881SJed Brown #define VECNEST        "nest"
1081bf2b38bSKerry Stevens #define VECSEQPTHREAD  "seqpthread"
1094b71561bSShri Abhyankar #define VECMPIPTHREAD  "mpipthread"
110bf2c1783SBarry Smith #define VECPTHREAD     "pthread"    /* seqpthread on one process and mpipthread on several */
1112eac72dbSBarry Smith 
1124b71561bSShri Abhyankar 
113fd487807SMatthew Knepley /* Logging support */
1140700a824SBarry Smith #define    VEC_FILE_CLASSID 1211214
1157087cfbeSBarry Smith extern  PetscClassId VEC_CLASSID;
1167087cfbeSBarry Smith extern  PetscClassId VEC_SCATTER_CLASSID;
1178ba1e511SMatthew Knepley 
11845b63f25SDmitry Karpeev 
1197087cfbeSBarry Smith extern PetscErrorCode  VecInitializePackage(const char[]);
1207087cfbeSBarry Smith extern PetscErrorCode  VecFinalizePackage(void);
121fd487807SMatthew Knepley 
1227087cfbeSBarry Smith extern PetscErrorCode  VecCreate(MPI_Comm,Vec*);
123045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecCreate,(Vec *x),(PETSC_COMM_SELF,x))
1247087cfbeSBarry Smith extern PetscErrorCode  VecCreateSeq(MPI_Comm,PetscInt,Vec*);
125045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecCreateSeq,(PetscInt n,Vec *x),(PETSC_COMM_SELF,n,x))
1267087cfbeSBarry Smith extern PetscErrorCode  VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*);
127045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecCreateMPI,(PetscInt n,PetscInt N,Vec *x),(PETSC_COMM_WORLD,n,N,x))
1287087cfbeSBarry Smith extern PetscErrorCode  VecCreateSeqWithArray(MPI_Comm,PetscInt,const PetscScalar[],Vec*);
129045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecCreateSeqWithArray,(PetscInt n,PetscScalar s[],Vec *x),(PETSC_COMM_SELF,n,s,x))
1307087cfbeSBarry Smith extern PetscErrorCode  VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*);
131045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecCreateMPIWithArray,(PetscInt n,PetscInt N,PetscScalar s[],Vec *x),(PETSC_COMM_WORLD,n,N,s,x))
1327087cfbeSBarry Smith extern PetscErrorCode  VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*);
1337087cfbeSBarry Smith extern PetscErrorCode  VecSetFromOptions(Vec);
1347087cfbeSBarry Smith extern PetscErrorCode  VecSetUp(Vec);
1356bf464f9SBarry Smith extern PetscErrorCode  VecDestroy(Vec*);
1367087cfbeSBarry Smith extern PetscErrorCode  VecZeroEntries(Vec);
1377087cfbeSBarry Smith extern PetscErrorCode  VecSetOptionsPrefix(Vec,const char[]);
1387087cfbeSBarry Smith extern PetscErrorCode  VecAppendOptionsPrefix(Vec,const char[]);
1397087cfbeSBarry Smith extern PetscErrorCode  VecGetOptionsPrefix(Vec,const char*[]);
140f69a0ea3SMatthew Knepley 
1417087cfbeSBarry Smith extern PetscErrorCode  VecSetSizes(Vec,PetscInt,PetscInt);
142fd487807SMatthew Knepley 
1437087cfbeSBarry Smith extern PetscErrorCode  VecDotNorm2(Vec,Vec,PetscScalar*,PetscScalar*);
1447087cfbeSBarry Smith extern PetscErrorCode  VecDot(Vec,Vec,PetscScalar*);
145fcd5dd21SSatish Balay PetscPolymorphicFunction(VecDot,(Vec x,Vec y),(x,y,&s),PetscScalar,s)
1467087cfbeSBarry Smith extern PetscErrorCode  VecTDot(Vec,Vec,PetscScalar*);
147fcd5dd21SSatish Balay PetscPolymorphicFunction(VecTDot,(Vec x,Vec y),(x,y,&s),PetscScalar,s)
1487087cfbeSBarry Smith extern PetscErrorCode  VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]);
1497087cfbeSBarry Smith extern PetscErrorCode  VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]);
1507087cfbeSBarry Smith extern PetscErrorCode  VecGetSubVector(Vec,IS,Vec*);
1517087cfbeSBarry Smith extern PetscErrorCode  VecRestoreSubVector(Vec,IS,Vec*);
152cddf8d76SBarry Smith 
15309321671SBarry Smith /*E
15409321671SBarry Smith     NormType - determines what type of norm to compute
15509321671SBarry Smith 
15609321671SBarry Smith     Level: beginner
15709321671SBarry Smith 
15809321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm()
15909321671SBarry Smith E*/
1609dcbbd2bSBarry Smith typedef enum {NORM_1=0,NORM_2=1,NORM_FROBENIUS=2,NORM_INFINITY=3,NORM_1_AND_2=4} NormType;
1619dcbbd2bSBarry Smith extern const char *NormTypes[];
162cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY
16309321671SBarry Smith 
1649b250c83SBarry Smith /*MC
1659b250c83SBarry Smith      NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum
1669b250c83SBarry Smith 
1679b250c83SBarry Smith    Level: beginner
1689b250c83SBarry Smith 
1699b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS,
1709b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
1719b250c83SBarry Smith 
1729b250c83SBarry Smith M*/
1739b250c83SBarry Smith 
1749b250c83SBarry Smith /*MC
1759b250c83SBarry Smith      NORM_2 - the two norm, ||v|| = sqrt(sum_i (v_i)^2) (vectors only)
1769b250c83SBarry Smith 
1779b250c83SBarry Smith    Level: beginner
1789b250c83SBarry Smith 
1799b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS,
1809b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
1819b250c83SBarry Smith 
1829b250c83SBarry Smith M*/
1839b250c83SBarry Smith 
1849b250c83SBarry Smith /*MC
1859b250c83SBarry Smith      NORM_FROBENIUS - ||A|| = sqrt(sum_ij (A_ij)^2), same as NORM_2 for vectors
1869b250c83SBarry Smith 
1879b250c83SBarry Smith    Level: beginner
1889b250c83SBarry Smith 
1899b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
1909b250c83SBarry Smith            NORM_INFINITY, NORM_1_AND_2
1919b250c83SBarry Smith 
1929b250c83SBarry Smith M*/
1939b250c83SBarry Smith 
1949b250c83SBarry Smith /*MC
1959b250c83SBarry Smith      NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum
1969b250c83SBarry Smith 
1979b250c83SBarry Smith    Level: beginner
1989b250c83SBarry Smith 
1999b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
2009b250c83SBarry Smith            NORM_FROBINIUS, NORM_1_AND_2
2019b250c83SBarry Smith 
2029b250c83SBarry Smith M*/
2039b250c83SBarry Smith 
2049b250c83SBarry Smith /*MC
2059b250c83SBarry Smith      NORM_1_AND_2 - computes both the 1 and 2 norm of a vector
2069b250c83SBarry Smith 
2079b250c83SBarry Smith    Level: beginner
2089b250c83SBarry Smith 
2099b250c83SBarry Smith .seealso:  NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2,
2109b250c83SBarry Smith            NORM_FROBINIUS, NORM_INFINITY
2119b250c83SBarry Smith 
2129b250c83SBarry Smith M*/
2139b250c83SBarry Smith 
2149b250c83SBarry Smith /*MC
2159b250c83SBarry Smith      NORM_MAX - see NORM_INFINITY
2169b250c83SBarry Smith 
217d41222bbSBarry Smith    Level: beginner
218d41222bbSBarry Smith 
2199b250c83SBarry Smith M*/
2209b250c83SBarry Smith 
2217087cfbeSBarry Smith extern PetscErrorCode  VecNorm(Vec,NormType,PetscReal *);
2227087cfbeSBarry Smith extern PetscErrorCode  VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *);
223045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecNorm,(Vec x,PetscReal *r),(x,NORM_2,r))
224c714d2d0SBarry Smith PetscPolymorphicFunction(VecNorm,(Vec x,NormType t),(x,t,&r),PetscReal,r)
225c714d2d0SBarry Smith PetscPolymorphicFunction(VecNorm,(Vec x),(x,NORM_2,&r),PetscReal,r)
2267087cfbeSBarry Smith extern PetscErrorCode  VecNormalize(Vec,PetscReal *);
2277087cfbeSBarry Smith extern PetscErrorCode  VecSum(Vec,PetscScalar*);
2287087cfbeSBarry Smith extern PetscErrorCode  VecMax(Vec,PetscInt*,PetscReal *);
229045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecMax,(Vec x,PetscReal *r),(x,PETSC_NULL,r))
2307087cfbeSBarry Smith extern PetscErrorCode  VecMin(Vec,PetscInt*,PetscReal *);
231045ff3e0SBarry Smith PetscPolymorphicSubroutine(VecMin,(Vec x,PetscReal *r),(x,PETSC_NULL,r))
2327087cfbeSBarry Smith extern PetscErrorCode  VecScale(Vec,PetscScalar);
2337087cfbeSBarry Smith extern PetscErrorCode  VecCopy(Vec,Vec);
2347087cfbeSBarry Smith extern PetscErrorCode  VecSetRandom(Vec,PetscRandom);
2357087cfbeSBarry Smith extern PetscErrorCode  VecSet(Vec,PetscScalar);
2367087cfbeSBarry Smith extern PetscErrorCode  VecSwap(Vec,Vec);
2377087cfbeSBarry Smith extern PetscErrorCode  VecAXPY(Vec,PetscScalar,Vec);
2387087cfbeSBarry Smith extern PetscErrorCode  VecAXPBY(Vec,PetscScalar,PetscScalar,Vec);
2397087cfbeSBarry Smith extern PetscErrorCode  VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]);
2407087cfbeSBarry Smith extern PetscErrorCode  VecAYPX(Vec,PetscScalar,Vec);
2417087cfbeSBarry Smith extern PetscErrorCode  VecWAXPY(Vec,PetscScalar,Vec,Vec);
2427087cfbeSBarry Smith extern PetscErrorCode  VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec);
2437087cfbeSBarry Smith extern PetscErrorCode  VecPointwiseMax(Vec,Vec,Vec);
2440190745aSSatish Balay PetscPolymorphicSubroutine(VecPointwiseMax,(Vec x,Vec y),(x,y,y))
2457087cfbeSBarry Smith extern PetscErrorCode  VecPointwiseMaxAbs(Vec,Vec,Vec);
2460190745aSSatish Balay PetscPolymorphicSubroutine(VecPointwiseMaxAbs,(Vec x,Vec y),(x,y,y))
2477087cfbeSBarry Smith extern PetscErrorCode  VecPointwiseMin(Vec,Vec,Vec);
2480190745aSSatish Balay PetscPolymorphicSubroutine(VecPointwiseMin,(Vec x,Vec y),(x,y,y))
2497087cfbeSBarry Smith extern PetscErrorCode  VecPointwiseMult(Vec,Vec,Vec);
250b2587007SBarry Smith PetscPolymorphicSubroutine(VecPointwiseMult,(Vec x,Vec y),(x,x,y))
2517087cfbeSBarry Smith extern PetscErrorCode  VecPointwiseDivide(Vec,Vec,Vec);
252b2587007SBarry Smith PetscPolymorphicSubroutine(VecPointwiseDivide,(Vec x,Vec y),(x,x,y))
2537087cfbeSBarry Smith extern PetscErrorCode  VecMaxPointwiseDivide(Vec,Vec,PetscReal*);
2547087cfbeSBarry Smith extern PetscErrorCode  VecShift(Vec,PetscScalar);
2557087cfbeSBarry Smith extern PetscErrorCode  VecReciprocal(Vec);
2567087cfbeSBarry Smith extern PetscErrorCode  VecPermute(Vec, IS, PetscBool );
2577087cfbeSBarry Smith extern PetscErrorCode  VecSqrtAbs(Vec);
2587087cfbeSBarry Smith extern PetscErrorCode  VecLog(Vec);
2597087cfbeSBarry Smith extern PetscErrorCode  VecExp(Vec);
2607087cfbeSBarry Smith extern PetscErrorCode  VecAbs(Vec);
2617087cfbeSBarry Smith extern PetscErrorCode  VecDuplicate(Vec,Vec*);
2627087cfbeSBarry Smith extern PetscErrorCode  VecDuplicateVecs(Vec,PetscInt,Vec*[]);
263574deadeSBarry Smith extern PetscErrorCode  VecDestroyVecs(PetscInt, Vec*[]);
2647087cfbeSBarry Smith extern PetscErrorCode  VecStrideNormAll(Vec,NormType,PetscReal[]);
2657087cfbeSBarry Smith extern PetscErrorCode  VecStrideMaxAll(Vec,PetscInt [],PetscReal []);
2667087cfbeSBarry Smith extern PetscErrorCode  VecStrideMinAll(Vec,PetscInt [],PetscReal []);
26789981eeeSJed Brown extern PetscErrorCode  VecStrideScaleAll(Vec,const PetscScalar[]);
2684a560884SBarry Smith 
2697087cfbeSBarry Smith extern PetscErrorCode  VecStrideNorm(Vec,PetscInt,NormType,PetscReal*);
270fcd5dd21SSatish Balay PetscPolymorphicFunction(VecStrideNorm,(Vec x,PetscInt i),(x,i,NORM_2,&r),PetscReal,r)
271fcd5dd21SSatish Balay PetscPolymorphicFunction(VecStrideNorm,(Vec x,PetscInt i,NormType t),(x,i,t,&r),PetscReal,r)
2727087cfbeSBarry Smith extern PetscErrorCode  VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *);
273fcd5dd21SSatish Balay PetscPolymorphicFunction(VecStrideMax,(Vec x,PetscInt i),(x,i,PETSC_NULL,&r),PetscReal,r)
2747087cfbeSBarry Smith extern PetscErrorCode  VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *);
275fcd5dd21SSatish Balay PetscPolymorphicFunction(VecStrideMin,(Vec x,PetscInt i),(x,i,PETSC_NULL,&r),PetscReal,r)
2767087cfbeSBarry Smith extern PetscErrorCode  VecStrideScale(Vec,PetscInt,PetscScalar);
277717e37fcSBarry Smith extern PetscErrorCode  VecStrideSet(Vec,PetscInt,PetscScalar);
278954b3ec6SBarry Smith 
279954b3ec6SBarry Smith 
2807087cfbeSBarry Smith extern PetscErrorCode  VecStrideGather(Vec,PetscInt,Vec,InsertMode);
2817087cfbeSBarry Smith extern PetscErrorCode  VecStrideScatter(Vec,PetscInt,Vec,InsertMode);
2827087cfbeSBarry Smith extern PetscErrorCode  VecStrideGatherAll(Vec,Vec[],InsertMode);
2837087cfbeSBarry Smith extern PetscErrorCode  VecStrideScatterAll(Vec[],Vec,InsertMode);
284d2655a18SBarry Smith 
2857087cfbeSBarry Smith extern PetscErrorCode  VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
2867087cfbeSBarry Smith extern PetscErrorCode  VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]);
2877087cfbeSBarry Smith extern PetscErrorCode  VecAssemblyBegin(Vec);
2887087cfbeSBarry Smith extern PetscErrorCode  VecAssemblyEnd(Vec);
2897087cfbeSBarry Smith extern PetscErrorCode  VecStashSetInitialSize(Vec,PetscInt,PetscInt);
2907087cfbeSBarry Smith extern PetscErrorCode  VecStashView(Vec,PetscViewer);
2917087cfbeSBarry Smith extern PetscErrorCode  VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*);
29262dc5420SSatish Balay 
29330de9b25SBarry Smith /*MC
29430de9b25SBarry Smith    VecSetValue - Set a single entry into a vector.
29530de9b25SBarry Smith 
29630de9b25SBarry Smith    Synopsis:
297d360dc6fSBarry Smith    PetscErrorCode VecSetValue(Vec v,int row,PetscScalar value, InsertMode mode);
29830de9b25SBarry Smith 
29930de9b25SBarry Smith    Not Collective
30030de9b25SBarry Smith 
30130de9b25SBarry Smith    Input Parameters:
30230de9b25SBarry Smith +  v - the vector
30330de9b25SBarry Smith .  row - the row location of the entry
30430de9b25SBarry Smith .  value - the value to insert
30530de9b25SBarry Smith -  mode - either INSERT_VALUES or ADD_VALUES
30630de9b25SBarry Smith 
30730de9b25SBarry Smith    Notes:
30830de9b25SBarry Smith    For efficiency one should use VecSetValues() and set several or
30930de9b25SBarry Smith    many values simultaneously if possible.
31030de9b25SBarry Smith 
3111d73ed98SBarry Smith    These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd()
3121d73ed98SBarry Smith    MUST be called after all calls to VecSetValues() have been completed.
3131d73ed98SBarry Smith 
3141d73ed98SBarry Smith    VecSetValues() uses 0-based indices in Fortran as well as in C.
3151d73ed98SBarry Smith 
3161d73ed98SBarry Smith    Level: beginner
3171d73ed98SBarry Smith 
3181d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal()
3191d73ed98SBarry Smith M*/
32050755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);}
3211d73ed98SBarry Smith 
32230de9b25SBarry Smith 
3237087cfbeSBarry Smith extern PetscErrorCode  VecSetBlockSize(Vec,PetscInt);
3247087cfbeSBarry Smith extern PetscErrorCode  VecGetBlockSize(Vec,PetscInt*);
325fcd5dd21SSatish Balay PetscPolymorphicFunction(VecGetBlockSize,(Vec x),(x,&i),PetscInt,i)
3267087cfbeSBarry Smith extern PetscErrorCode  VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
3278ed539a5SBarry Smith 
328fd487807SMatthew Knepley /* Dynamic creation and loading functions */
329fd487807SMatthew Knepley extern PetscFList VecList;
330ace3abfcSBarry Smith extern PetscBool  VecRegisterAllCalled;
3317087cfbeSBarry Smith extern PetscErrorCode  VecSetType(Vec, const VecType);
3327087cfbeSBarry Smith extern PetscErrorCode  VecGetType(Vec, const VecType *);
3337087cfbeSBarry Smith extern PetscErrorCode  VecRegister(const char[],const char[],const char[],PetscErrorCode (*)(Vec));
3347087cfbeSBarry Smith extern PetscErrorCode  VecRegisterAll(const char []);
3357087cfbeSBarry Smith extern PetscErrorCode  VecRegisterDestroy(void);
33630de9b25SBarry Smith 
33730de9b25SBarry Smith /*MC
33830de9b25SBarry Smith   VecRegisterDynamic - Adds a new vector component implementation
33930de9b25SBarry Smith 
34030de9b25SBarry Smith   Synopsis:
3411890ba74SBarry Smith   PetscErrorCode VecRegisterDynamic(const char *name, const char *path, const char *func_name, PetscErrorCode (*create_func)(Vec))
34230de9b25SBarry Smith 
34330de9b25SBarry Smith   Not Collective
34430de9b25SBarry Smith 
34530de9b25SBarry Smith   Input Parameters:
34630de9b25SBarry Smith + name        - The name of a new user-defined creation routine
34730de9b25SBarry Smith . path        - The path (either absolute or relative) of the library containing this routine
34830de9b25SBarry Smith . func_name   - The name of routine to create method context
34930de9b25SBarry Smith - create_func - The creation routine itself
35030de9b25SBarry Smith 
35130de9b25SBarry Smith   Notes:
35230de9b25SBarry Smith   VecRegisterDynamic() may be called multiple times to add several user-defined vectors
35330de9b25SBarry Smith 
35430de9b25SBarry Smith   If dynamic libraries are used, then the fourth input argument (routine_create) is ignored.
35530de9b25SBarry Smith 
35630de9b25SBarry Smith   Sample usage:
35730de9b25SBarry Smith .vb
35830de9b25SBarry Smith     VecRegisterDynamic("my_vec","/home/username/my_lib/lib/libO/solaris/libmy.a", "MyVectorCreate", MyVectorCreate);
35930de9b25SBarry Smith .ve
36030de9b25SBarry Smith 
36130de9b25SBarry Smith   Then, your vector type can be chosen with the procedural interface via
36230de9b25SBarry Smith .vb
36330de9b25SBarry Smith     VecCreate(MPI_Comm, Vec *);
36430de9b25SBarry Smith     VecSetType(Vec,"my_vector_name");
36530de9b25SBarry Smith .ve
36630de9b25SBarry Smith    or at runtime via the option
36730de9b25SBarry Smith .vb
36830de9b25SBarry Smith     -vec_type my_vector_name
36930de9b25SBarry Smith .ve
37030de9b25SBarry Smith 
371ab901514SBarry Smith   Notes: $PETSC_ARCH occuring in pathname will be replaced with appropriate values.
37230de9b25SBarry Smith          If your function is not being put into a shared library then use VecRegister() instead
37330de9b25SBarry Smith 
37430de9b25SBarry Smith   Level: advanced
37530de9b25SBarry Smith 
37630de9b25SBarry Smith .keywords: Vec, register
37730de9b25SBarry Smith .seealso: VecRegisterAll(), VecRegisterDestroy(), VecRegister()
37830de9b25SBarry Smith M*/
379aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
380f1af5d2fSBarry Smith #define VecRegisterDynamic(a,b,c,d) VecRegister(a,b,c,0)
38188d459dfSBarry Smith #else
382f1af5d2fSBarry Smith #define VecRegisterDynamic(a,b,c,d) VecRegister(a,b,c,d)
38388d459dfSBarry Smith #endif
38488d459dfSBarry Smith 
38509321671SBarry Smith 
3867087cfbeSBarry Smith extern PetscErrorCode  VecScatterCreate(Vec,IS,Vec,IS,VecScatter *);
387fcd5dd21SSatish Balay PetscPolymorphicFunction(VecScatterCreate,(Vec x,IS is1,Vec y,IS is2),(x,is1,y,is2,&s),VecScatter,s)
3880190745aSSatish Balay PetscPolymorphicSubroutine(VecScatterCreate,(Vec x,IS is,Vec y,VecScatter *s),(x,is,y,PETSC_NULL,s))
389fcd5dd21SSatish Balay PetscPolymorphicFunction(VecScatterCreate,(Vec x,IS is,Vec y),(x,is,y,PETSC_NULL,&s),VecScatter,s)
3900190745aSSatish Balay PetscPolymorphicSubroutine(VecScatterCreate,(Vec x,Vec y,IS is,VecScatter *s),(x,PETSC_NULL,y,is,s))
391fcd5dd21SSatish Balay PetscPolymorphicFunction(VecScatterCreate,(Vec x,Vec y,IS is),(x,PETSC_NULL,y,is,&s),VecScatter,s)
3927087cfbeSBarry Smith extern PetscErrorCode  VecScatterCreateEmpty(MPI_Comm,VecScatter *);
3937087cfbeSBarry Smith extern PetscErrorCode  VecScatterCreateLocal(VecScatter,PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt);
3947087cfbeSBarry Smith extern PetscErrorCode  VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode);
3957087cfbeSBarry Smith extern PetscErrorCode  VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode);
396fcfd50ebSBarry Smith extern PetscErrorCode  VecScatterDestroy(VecScatter*);
3977087cfbeSBarry Smith extern PetscErrorCode  VecScatterCopy(VecScatter,VecScatter *);
3987087cfbeSBarry Smith extern PetscErrorCode  VecScatterView(VecScatter,PetscViewer);
3997087cfbeSBarry Smith extern PetscErrorCode  VecScatterRemap(VecScatter,PetscInt *,PetscInt*);
4007087cfbeSBarry Smith extern PetscErrorCode  VecScatterGetMerged(VecScatter,PetscBool *);
4012195c698SBarry Smith 
4027087cfbeSBarry Smith extern PetscErrorCode  VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
4037087cfbeSBarry Smith extern PetscErrorCode  VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]);
4047087cfbeSBarry Smith extern PetscErrorCode  VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
4057087cfbeSBarry Smith extern PetscErrorCode  VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]);
4067087cfbeSBarry Smith extern PetscErrorCode  VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
4077087cfbeSBarry Smith extern PetscErrorCode  VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]);
4087087cfbeSBarry Smith extern PetscErrorCode  VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]);
4097087cfbeSBarry Smith extern PetscErrorCode  VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]);
410ab360428SBarry Smith 
4117087cfbeSBarry Smith extern PetscErrorCode  VecPlaceArray(Vec,const PetscScalar[]);
4127087cfbeSBarry Smith extern PetscErrorCode  VecResetArray(Vec);
4137087cfbeSBarry Smith extern PetscErrorCode  VecReplaceArray(Vec,const PetscScalar[]);
4147087cfbeSBarry Smith extern PetscErrorCode  VecGetArrays(const Vec[],PetscInt,PetscScalar**[]);
4157087cfbeSBarry Smith extern PetscErrorCode  VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]);
41684cb2905SBarry Smith 
4177087cfbeSBarry Smith extern PetscErrorCode  VecView(Vec,PetscViewer);
4187087cfbeSBarry Smith extern PetscErrorCode  VecViewFromOptions(Vec, const char *);
4197087cfbeSBarry Smith extern PetscErrorCode  VecEqual(Vec,Vec,PetscBool *);
420ace3abfcSBarry Smith PetscPolymorphicFunction(VecEqual,(Vec x,Vec y),(x,y,&s),PetscBool ,s)
4217087cfbeSBarry Smith extern PetscErrorCode  VecLoad(Vec, PetscViewer);
4228ed539a5SBarry Smith 
4237087cfbeSBarry Smith extern PetscErrorCode  VecGetSize(Vec,PetscInt*);
424fcd5dd21SSatish Balay PetscPolymorphicFunction(VecGetSize,(Vec x),(x,&s),PetscInt,s)
4257087cfbeSBarry Smith extern PetscErrorCode  VecGetLocalSize(Vec,PetscInt*);
426fcd5dd21SSatish Balay PetscPolymorphicFunction(VecGetLocalSize,(Vec x),(x,&s),PetscInt,s)
4277087cfbeSBarry Smith extern PetscErrorCode  VecGetOwnershipRange(Vec,PetscInt*,PetscInt*);
4287087cfbeSBarry Smith extern PetscErrorCode  VecGetOwnershipRanges(Vec,const PetscInt *[]);
4298ed539a5SBarry Smith 
4307087cfbeSBarry Smith extern PetscErrorCode  VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping);
4317087cfbeSBarry Smith extern PetscErrorCode  VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
43288b03592SBarry Smith 
43388b03592SBarry Smith /*MC
43488b03592SBarry Smith    VecSetValueLocal - Set a single entry into a vector using the local numbering
43588b03592SBarry Smith 
43688b03592SBarry Smith    Synopsis:
43788b03592SBarry Smith    PetscErrorCode VecSetValueLocal(Vec v,int row,PetscScalar value, InsertMode mode);
43888b03592SBarry Smith 
43988b03592SBarry Smith    Not Collective
44088b03592SBarry Smith 
44188b03592SBarry Smith    Input Parameters:
44288b03592SBarry Smith +  v - the vector
44388b03592SBarry Smith .  row - the row location of the entry
44488b03592SBarry Smith .  value - the value to insert
44588b03592SBarry Smith -  mode - either INSERT_VALUES or ADD_VALUES
44688b03592SBarry Smith 
44788b03592SBarry Smith    Notes:
44888b03592SBarry Smith    For efficiency one should use VecSetValues() and set several or
44988b03592SBarry Smith    many values simultaneously if possible.
45088b03592SBarry Smith 
45188b03592SBarry Smith    These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd()
45288b03592SBarry Smith    MUST be called after all calls to VecSetValues() have been completed.
45388b03592SBarry Smith 
45488b03592SBarry Smith    VecSetValues() uses 0-based indices in Fortran as well as in C.
45588b03592SBarry Smith 
45688b03592SBarry Smith    Level: beginner
45788b03592SBarry Smith 
45888b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue()
45988b03592SBarry Smith M*/
46088b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);}
46188b03592SBarry Smith 
4627087cfbeSBarry Smith extern PetscErrorCode  VecSetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping);
4637087cfbeSBarry Smith extern PetscErrorCode  VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
464dcb12a5bSJed Brown extern PetscErrorCode  VecGetLocalToGlobalMappingBlock(Vec,ISLocalToGlobalMapping*);
465dcb12a5bSJed Brown extern PetscErrorCode  VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*);
46690f02eecSBarry Smith 
4677087cfbeSBarry Smith extern PetscErrorCode  VecDotBegin(Vec,Vec,PetscScalar *);
468ba966639SSatish Balay PetscPolymorphicSubroutine(VecDotBegin,(Vec x,Vec y),(x,y,PETSC_NULL))
4697087cfbeSBarry Smith extern PetscErrorCode  VecDotEnd(Vec,Vec,PetscScalar *);
470fcd5dd21SSatish Balay PetscPolymorphicFunction(VecDotEnd,(Vec x,Vec y),(x,y,&s),PetscScalar,s)
4717087cfbeSBarry Smith extern PetscErrorCode  VecTDotBegin(Vec,Vec,PetscScalar *);
472ba966639SSatish Balay PetscPolymorphicSubroutine(VecTDotBegin,(Vec x,Vec y),(x,y,PETSC_NULL))
4737087cfbeSBarry Smith extern PetscErrorCode  VecTDotEnd(Vec,Vec,PetscScalar *);
474fcd5dd21SSatish Balay PetscPolymorphicFunction(VecTDotEnd,(Vec x,Vec y),(x,y,&s),PetscScalar,s)
4757087cfbeSBarry Smith extern PetscErrorCode  VecNormBegin(Vec,NormType,PetscReal *);
476ba966639SSatish Balay PetscPolymorphicSubroutine(VecNormBegin,(Vec x,NormType t),(x,t,PETSC_NULL))
477ba966639SSatish Balay PetscPolymorphicSubroutine(VecNormBegin,(Vec x),(x,NORM_2,PETSC_NULL))
4787087cfbeSBarry Smith extern PetscErrorCode  VecNormEnd(Vec,NormType,PetscReal *);
479fcd5dd21SSatish Balay PetscPolymorphicFunction(VecNormEnd,(Vec x,NormType t),(x,t,&s),PetscReal,s)
480fcd5dd21SSatish Balay PetscPolymorphicFunction(VecNormEnd,(Vec x),(x,NORM_2,&s),PetscReal,s)
481d3c178dbSBarry Smith 
4827087cfbeSBarry Smith extern PetscErrorCode  VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]);
4837087cfbeSBarry Smith extern PetscErrorCode  VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]);
4847087cfbeSBarry Smith extern PetscErrorCode  VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]);
4857087cfbeSBarry Smith extern PetscErrorCode  VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]);
486a751f32aSSatish Balay 
487a751f32aSSatish Balay 
4884ec14c9cSBarry Smith #if defined(PETSC_USE_DEBUG)
489ab2fbef6SJed Brown #define VecValidValues(vec,argnum,input) do {                           \
4904ec14c9cSBarry Smith     PetscErrorCode     _ierr;                                           \
4914ec14c9cSBarry Smith     PetscInt          _n,_i;                                            \
4924ec14c9cSBarry Smith     const PetscScalar *_x;                                              \
4934ec14c9cSBarry Smith                                                                         \
494ab2fbef6SJed Brown     if (vec->petscnative || vec->ops->getarray) {                       \
4954ec14c9cSBarry Smith       _ierr = VecGetLocalSize(vec,&_n);CHKERRQ(_ierr);                  \
4964ec14c9cSBarry Smith       _ierr = VecGetArrayRead(vec,&_x);CHKERRQ(_ierr);                  \
4974ec14c9cSBarry Smith       for (_i=0; _i<_n; _i++) {                                         \
4984ec14c9cSBarry Smith         if (input) {                                                    \
4994ec14c9cSBarry Smith           if (PetscIsInfOrNanScalar(_x[_i])) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_FP,"Vec entry at local location %D is not-a-number or infinite at beginning of function: Parameter number %d",_i,argnum); \
5004ec14c9cSBarry Smith         } else {                                                        \
5014ec14c9cSBarry Smith           if (PetscIsInfOrNanScalar(_x[_i])) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_FP,"Vec entry at local location %D is not-a-number or infinite at end of function: Parameter number %d",_i,argnum); \
5024ec14c9cSBarry Smith         }                                                               \
5034ec14c9cSBarry Smith       }                                                                 \
5044ec14c9cSBarry Smith       _ierr = VecRestoreArrayRead(vec,&_x);CHKERRQ(_ierr);              \
505ab2fbef6SJed Brown     }                                                                   \
506ab2fbef6SJed Brown   } while (0)
5074ec14c9cSBarry Smith #else
5084ec14c9cSBarry Smith #define VecValidValues(vec,argnum,input)
5094ec14c9cSBarry Smith #endif
5104ec14c9cSBarry Smith 
5114ec14c9cSBarry Smith 
5129533e83fSBarry Smith typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES} VecOption;
5137087cfbeSBarry Smith extern PetscErrorCode  VecSetOption(Vec,VecOption,PetscBool );
51490f02eecSBarry Smith 
515ebe3b25bSBarry Smith /*
516ebe3b25bSBarry Smith    Expose VecGetArray()/VecRestoreArray() to users. Allows this to work without any function
517ebe3b25bSBarry Smith    call overhead on any 'native' Vecs.
518ebe3b25bSBarry Smith */
519e1fa1e0fSSatish Balay 
5201d8d5f9aSSatish Balay #include "private/vecimpl.h"
521ebe3b25bSBarry Smith 
5227087cfbeSBarry Smith extern PetscErrorCode  VecContourScale(Vec,PetscReal,PetscReal);
523522c5e43SBarry Smith 
52415091d37SBarry Smith /*
52515091d37SBarry Smith     These numbers need to match the entries in
5263c94ec11SBarry Smith   the function table in vecimpl.h
52715091d37SBarry Smith */
5282dcb2ebcSBarry Smith typedef enum { VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_DUPLICATE = 0} VecOperation;
5297087cfbeSBarry Smith extern PetscErrorCode  VecSetOperation(Vec,VecOperation,void(*)(void));
530b19c1e4cSBarry Smith 
531e182c471SBarry Smith /*
532e182c471SBarry Smith      Routines for dealing with ghosted vectors:
533e182c471SBarry Smith   vectors with ghost elements at the end of the array.
534e182c471SBarry Smith */
535*63bec58dSBarry Smith extern PetscErrorCode  VecMPISetGhost(Vec,PetscInt,const PetscInt[]);
5367087cfbeSBarry Smith extern PetscErrorCode  VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*);
5377087cfbeSBarry Smith extern PetscErrorCode  VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*);
5387087cfbeSBarry Smith extern PetscErrorCode  VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*);
5397087cfbeSBarry Smith extern PetscErrorCode  VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*);
5407087cfbeSBarry Smith extern PetscErrorCode  VecGhostGetLocalForm(Vec,Vec*);
541ba966639SSatish Balay PetscPolymorphicFunction(VecGhostGetLocalForm,(Vec x),(x,&s),Vec,s)
5427087cfbeSBarry Smith extern PetscErrorCode  VecGhostRestoreLocalForm(Vec,Vec*);
5437087cfbeSBarry Smith extern PetscErrorCode  VecGhostUpdateBegin(Vec,InsertMode,ScatterMode);
5447087cfbeSBarry Smith extern PetscErrorCode  VecGhostUpdateEnd(Vec,InsertMode,ScatterMode);
545e182c471SBarry Smith 
5467087cfbeSBarry Smith extern PetscErrorCode  VecConjugate(Vec);
54734233285SBarry Smith 
5487087cfbeSBarry Smith extern PetscErrorCode  VecScatterCreateToAll(Vec,VecScatter*,Vec*);
5497087cfbeSBarry Smith extern PetscErrorCode  VecScatterCreateToZero(Vec,VecScatter*,Vec*);
550bba1ac68SSatish Balay 
5517087cfbeSBarry Smith extern PetscErrorCode  PetscViewerMathematicaGetVector(PetscViewer, Vec);
5527087cfbeSBarry Smith extern PetscErrorCode  PetscViewerMathematicaPutVector(PetscViewer, Vec);
5537dbadf16SMatthew Knepley 
554d59c15a7SBarry Smith /*S
555d59c15a7SBarry Smith      Vecs - Collection of vectors where the data for the vectors is stored in
556759e7b9cSHong Zhang             one contiguous memory
557d59c15a7SBarry Smith 
558d59c15a7SBarry Smith    Level: advanced
559d59c15a7SBarry Smith 
560d59c15a7SBarry Smith    Notes:
561d59c15a7SBarry Smith     Temporary construct for handling multiply right hand side solves
562d59c15a7SBarry Smith 
563d59c15a7SBarry Smith     This is faked by storing a single vector that has enough array space for
564d59c15a7SBarry Smith     n vectors
565d59c15a7SBarry Smith 
566d59c15a7SBarry Smith   Concepts: parallel decomposition
567d59c15a7SBarry Smith 
568d59c15a7SBarry Smith S*/
56995fbd943SSatish Balay         struct _n_Vecs  {PetscInt n; Vec v;};
57095fbd943SSatish Balay typedef struct _n_Vecs* Vecs;
571fcfd50ebSBarry Smith #define VecsDestroy(x)            (VecDestroy(&(x)->v)         || PetscFree(x))
57295fbd943SSatish Balay #define VecsCreateSeq(comm,p,m,x) (PetscNew(struct _n_Vecs,x) || VecCreateSeq(comm,p*m,&(*(x))->v) || (-1 == ((*(x))->n = (m))))
57395fbd943SSatish Balay #define VecsCreateSeqWithArray(comm,p,m,a,x) (PetscNew(struct _n_Vecs,x) || VecCreateSeqWithArray(comm,p*m,a,&(*(x))->v) || (-1 == ((*(x))->n = (m))))
57495fbd943SSatish Balay #define VecsDuplicate(x,y)        (PetscNew(struct _n_Vecs,y) || VecDuplicate(x->v,&(*(y))->v) || (-1 == ((*(y))->n = (x)->n)))
575e9fa29b7SSatish Balay 
5768154be41SBarry Smith #if defined(PETSC_HAVE_CUSP)
577a5891931SBarry Smith typedef struct _p_PetscCUSPIndices* PetscCUSPIndices;
578a5891931SBarry Smith extern PetscErrorCode PetscCUSPIndicesCreate(PetscInt,const PetscInt*,PetscCUSPIndices*);
579fcfd50ebSBarry Smith extern PetscErrorCode PetscCUSPIndicesDestroy(PetscCUSPIndices*);
5808154be41SBarry Smith extern PetscErrorCode VecCUSPCopyToGPUSome_Public(Vec,PetscCUSPIndices);
5818154be41SBarry Smith extern PetscErrorCode VecCUSPCopyFromGPUSome_Public(Vec,PetscCUSPIndices);
58290c53902SBarry Smith 
58390c53902SBarry Smith extern PetscErrorCode  VecCreateSeqCUSP(MPI_Comm,PetscInt,Vec*);
58490c53902SBarry Smith extern PetscErrorCode  VecCreateMPICUSP(MPI_Comm,PetscInt,PetscInt,Vec*);
585a5891931SBarry Smith #endif
586e9fa29b7SSatish Balay 
587054b7196SShri Abhyankar #if defined(PETSC_HAVE_PTHREADCLASSES)
5886dc86e2bSShri Abhyankar extern PetscErrorCode VecPThreadSetNThreads(Vec,PetscInt);
589ffee5b81SShri Abhyankar extern PetscErrorCode VecPThreadGetNThreads(Vec,PetscInt*);
590ffee5b81SShri Abhyankar extern PetscErrorCode VecPThreadSetThreadAffinities(Vec,const PetscInt[]);
591be9be048SShri Abhyankar extern PetscErrorCode VecCreateSeqPThread(MPI_Comm,PetscInt,PetscInt,PetscInt[],Vec*);
592be9be048SShri Abhyankar extern PetscErrorCode VecCreateMPIPThread(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt[],Vec*);
593054b7196SShri Abhyankar #endif
594054b7196SShri Abhyankar 
5957087cfbeSBarry Smith extern PetscErrorCode  VecNestGetSubVecs(Vec,PetscInt*,Vec**);
5967087cfbeSBarry Smith extern PetscErrorCode  VecNestGetSubVec(Vec,PetscInt,Vec*);
5970091908dSmahadevan extern PetscErrorCode  VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*);
5980091908dSmahadevan extern PetscErrorCode  VecNestSetSubVec(Vec,PetscInt,Vec);
5997087cfbeSBarry Smith extern PetscErrorCode  VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*);
6007087cfbeSBarry Smith extern PetscErrorCode  VecNestGetSize(Vec,PetscInt*);
60145b63f25SDmitry Karpeev 
602e9fa29b7SSatish Balay PETSC_EXTERN_CXX_END
6032eac72dbSBarry Smith #endif
6042b82984cSSatish Balay 
605