xref: /petsc/include/petscvec.h (revision de7da479eb3fb06bd592bb38c249e0b855a0e284)
1*de7da479SBarry Smith /* $Id: vec.h,v 1.32 1995/09/07 22:37:32 bsmith Exp bsmith $ */
22eac72dbSBarry Smith /*
32eac72dbSBarry Smith    This defines the abstract vector component. These are patterned
42eac72dbSBarry Smith    after the Level-1 Blas, but with some additions that have proved
52eac72dbSBarry Smith    useful. These include routines to allocate and free vectors.
62eac72dbSBarry Smith 
72eac72dbSBarry Smith    Note that the routines that are normally thought of as returning a
82eac72dbSBarry Smith    value (e.g., dot, norm) return their value through an argument.
92eac72dbSBarry Smith    This allows these routines to be used with other datatype, such
102eac72dbSBarry Smith    as float and dcomplex.
112eac72dbSBarry Smith 
122eac72dbSBarry Smith    All vectors should be declared as a Vec. All vector routines begin
132eac72dbSBarry Smith    with Vec.
142eac72dbSBarry Smith 
152eac72dbSBarry Smith 
162eac72dbSBarry Smith  */
172eac72dbSBarry Smith 
182eac72dbSBarry Smith #ifndef __VEC_PACKAGE
192eac72dbSBarry Smith #define __VEC_PACKAGE
202eac72dbSBarry Smith #include "is.h"
212eac72dbSBarry Smith 
229e25ed09SBarry Smith #define VEC_COOKIE         PETSC_COOKIE+3
239e25ed09SBarry Smith #define VEC_SCATTER_COOKIE PETSC_COOKIE+4
249e25ed09SBarry Smith 
25b56ba379SLois Curfman McInnes typedef enum { VECSAME=-1, VECSEQ, VECMPI } VecType;
26f0479e8cSBarry Smith 
272eac72dbSBarry Smith typedef struct _Vec*            Vec;
2820563c6bSBarry Smith typedef struct _VecScatterCtx*  VecScatterCtx;
292eac72dbSBarry Smith 
30*de7da479SBarry Smith extern int VecCreateSeq(MPI_Comm,int,Vec *);
318ed539a5SBarry Smith extern int VecCreateMPI(MPI_Comm,int,int,Vec *);
326469c4f9SBarry Smith extern int VecCreate(MPI_Comm,int,Vec *);
332eac72dbSBarry Smith 
348ed539a5SBarry Smith extern int VecDot(Vec, Vec, Scalar*);
358ed539a5SBarry Smith extern int VecTDot(Vec, Vec, Scalar*);
368ed539a5SBarry Smith extern int VecMDot(int,      Vec ,Vec*,Scalar*);
378ed539a5SBarry Smith extern int VecMTDot(int,      Vec ,Vec*,Scalar*);
388ed539a5SBarry Smith extern int VecNorm(Vec, double*);
3920563c6bSBarry Smith extern int VecASum(Vec, double*);
40ee50ffe9SBarry Smith extern int VecSum(Vec,Scalar*);
41fddd07d0SBarry Smith extern int VecAMax(Vec, int *,   double*);
427c16e1c9SBarry Smith extern int VecMax(Vec, int *,    double*);
437c16e1c9SBarry Smith extern int VecMin(Vec, int *,    double*);
448ed539a5SBarry Smith extern int VecScale(Scalar*, Vec);
458ed539a5SBarry Smith extern int VecCopy(Vec, Vec);
468ed539a5SBarry Smith extern int VecSet(Scalar*, Vec);
478ed539a5SBarry Smith extern int VecSwap(Vec, Vec);
488ed539a5SBarry Smith extern int VecAXPY(Scalar*, Vec, Vec);
498ed539a5SBarry Smith extern int VecMAXPY(int,      Scalar*, Vec ,Vec*);
508ed539a5SBarry Smith extern int VecAYPX(Scalar*, Vec, Vec);
518ed539a5SBarry Smith extern int VecWAXPY(Scalar*, Vec, Vec, Vec);
528ed539a5SBarry Smith extern int VecPMult(Vec, Vec, Vec);
538ed539a5SBarry Smith extern int VecPDiv(Vec, Vec, Vec);
546469c4f9SBarry Smith extern int VecDuplicate(Vec,Vec *);
558ed539a5SBarry Smith extern int VecDestroy(Vec);
568ed539a5SBarry Smith extern int VecGetVecs(Vec, int,Vec **);
578ed539a5SBarry Smith extern int VecFreeVecs(Vec*,int);
582eac72dbSBarry Smith 
599d00d63dSBarry Smith typedef enum {NOTSETVALUES, INSERTVALUES, ADDVALUES} InsertMode;
6020563c6bSBarry Smith 
6120563c6bSBarry Smith extern int VecSetValues(Vec, int, int *,Scalar*,InsertMode);
62ee50ffe9SBarry Smith extern int VecAssemblyBegin(Vec);
63ee50ffe9SBarry Smith extern int VecAssemblyEnd(Vec);
648ed539a5SBarry Smith 
659d00d63dSBarry Smith typedef enum {SCATTERREVERSE=1,SCATTERDOWN=2,SCATTERUP=4,SCATTERALL=8,
669d00d63dSBarry Smith               SCATTERALLREVERSE=9} ScatterMode;
67ee50ffe9SBarry Smith 
6806be10caSBarry Smith extern int VecScatterBegin(Vec,Vec,InsertMode,ScatterMode,VecScatterCtx);
6906be10caSBarry Smith extern int VecScatterEnd(Vec,Vec,InsertMode,ScatterMode,VecScatterCtx);
7020563c6bSBarry Smith extern int VecScatterCtxCreate(Vec,IS,Vec,IS,VecScatterCtx *);
7120563c6bSBarry Smith extern int VecScatterCtxDestroy(VecScatterCtx);
72d6dfbf8fSBarry Smith extern int VecScatterCtxCopy(VecScatterCtx,VecScatterCtx *);
73d6dfbf8fSBarry Smith 
749d00d63dSBarry Smith typedef enum {PIPELINEDOWN=0,PIPELINEUP=1} PipelineMode;
75ee50ffe9SBarry Smith 
7606be10caSBarry Smith extern int VecPipelineBegin(Vec,Vec,InsertMode,PipelineMode,VecScatterCtx);
7706be10caSBarry Smith extern int VecPipelineEnd(Vec,Vec,InsertMode,PipelineMode,VecScatterCtx);
788ed539a5SBarry Smith 
79c01c455dSBarry Smith extern int VecShift(Scalar *,Vec);
808ed539a5SBarry Smith extern int VecGetArray(Vec,Scalar**);
816b5873e3SBarry Smith extern int VecRestoreArray(Vec,Scalar**);
82707f76d8SBarry Smith extern int VecGetArrays(Vec*,int,Scalar***);
83707f76d8SBarry Smith extern int VecRestoreArrays(Vec*,int,Scalar***);
848ed539a5SBarry Smith extern int VecValidVector(Vec);
858ed539a5SBarry Smith extern int VecView(Vec,Viewer);
865f52d6abSBarry Smith extern int VecLoad(Viewer,Vec*);
878ed539a5SBarry Smith 
888ed539a5SBarry Smith extern int VecGetSize(Vec,int *);
898ed539a5SBarry Smith extern int VecGetLocalSize(Vec,int *);
90d6dfbf8fSBarry Smith extern int VecGetOwnershipRange(Vec,int*,int*);
918ed539a5SBarry Smith 
928ed539a5SBarry Smith /* utility routines */
938ed539a5SBarry Smith extern int VecReciprocal(Vec);
94bfe6c5c8SBarry Smith extern int VecAbs(Vec);
952eac72dbSBarry Smith 
9634233285SBarry Smith #if defined(__DRAW_PACKAGE)
9734233285SBarry Smith extern int DrawTensorContour(DrawCtx,int,int,double *,double *,Vec);
9834233285SBarry Smith #endif
9934233285SBarry Smith 
1002eac72dbSBarry Smith #endif
1012eac72dbSBarry Smith 
1022eac72dbSBarry Smith 
103