petsc/include/petscvec.h

2eac72dbSBarry Smith/*
2eac72dbSBarry Smith   This defines the abstract vector component. These are patterned
2eac72dbSBarry Smith   after the Level-1 Blas, but with some additions that have proved
2eac72dbSBarry Smith   useful. These include routines to allocate and free vectors.
2eac72dbSBarry Smith
2eac72dbSBarry Smith   Note that the routines that are normally thought of as returning a
2eac72dbSBarry Smith   value (e.g., dot, norm) return their value through an argument.
2eac72dbSBarry Smith   This allows these routines to be used with other datatype, such
2eac72dbSBarry Smith   as float and dcomplex.
2eac72dbSBarry Smith
2eac72dbSBarry Smith   All vectors should be declared as a Vec. All vector routines begin
2eac72dbSBarry Smith   with Vec.
2eac72dbSBarry Smith
2eac72dbSBarry Smith
2eac72dbSBarry Smith */
2eac72dbSBarry Smith
2eac72dbSBarry Smith#ifndef __VEC_PACKAGE
2eac72dbSBarry Smith#define __VEC_PACKAGE
2eac72dbSBarry Smith#include "is.h"
2eac72dbSBarry Smith
9e25ed09SBarry Smith#define VEC_COOKIE         PETSC_COOKIE+3
9e25ed09SBarry Smith#define VEC_SCATTER_COOKIE PETSC_COOKIE+4
9e25ed09SBarry Smith
f0479e8cSBarry Smith
2eac72dbSBarry Smithtypedef struct _Vec*            Vec;
20563c6bSBarry Smithtypedef struct _VecScatterCtx*  VecScatterCtx;
2eac72dbSBarry Smith
6b5873e3SBarry Smithextern int VecCreateSequential(MPI_Comm,int,Vec *);
8ed539a5SBarry Smithextern int VecCreateMPI(MPI_Comm,int,int,Vec *);
6b5873e3SBarry Smithextern int VecCreateInitialVector(MPI_Comm,int,Vec *);
2eac72dbSBarry Smith
8ed539a5SBarry Smithextern int VecDot(Vec, Vec, Scalar*);
8ed539a5SBarry Smithextern int VecTDot(Vec, Vec, Scalar*);
8ed539a5SBarry Smithextern int VecMDot(int,      Vec ,Vec*,Scalar*);
8ed539a5SBarry Smithextern int VecMTDot(int,      Vec ,Vec*,Scalar*);
8ed539a5SBarry Smithextern int VecNorm(Vec, double*);
20563c6bSBarry Smithextern int VecASum(Vec, double*);
ee50ffe9SBarry Smithextern int VecSum(Vec,Scalar*);
fddd07d0SBarry Smithextern int VecAMax(Vec, int *,   double*);
7c16e1c9SBarry Smithextern int VecMax(Vec, int *,    double*);
7c16e1c9SBarry Smithextern int VecMin(Vec, int *,    double*);
8ed539a5SBarry Smithextern int VecScale(Scalar*, Vec);
8ed539a5SBarry Smithextern int VecCopy(Vec, Vec);
8ed539a5SBarry Smithextern int VecSet(Scalar*, Vec);
8ed539a5SBarry Smithextern int VecSwap(Vec, Vec);
8ed539a5SBarry Smithextern int VecAXPY(Scalar*, Vec, Vec);
8ed539a5SBarry Smithextern int VecMAXPY(int,      Scalar*, Vec ,Vec*);
8ed539a5SBarry Smithextern int VecAYPX(Scalar*, Vec, Vec);
8ed539a5SBarry Smithextern int VecWAXPY(Scalar*, Vec, Vec, Vec);
8ed539a5SBarry Smithextern int VecPMult(Vec, Vec, Vec);
8ed539a5SBarry Smithextern int VecPDiv(Vec, Vec, Vec);
8ed539a5SBarry Smithextern int VecCreate(Vec,Vec *);
8ed539a5SBarry Smithextern int VecDestroy(Vec);
8ed539a5SBarry Smithextern int VecGetVecs(Vec, int,Vec **);
8ed539a5SBarry Smithextern int VecFreeVecs(Vec*,int);
2eac72dbSBarry Smith
*9d00d63dSBarry Smithtypedef enum {NOTSETVALUES, INSERTVALUES, ADDVALUES} InsertMode;
20563c6bSBarry Smith
20563c6bSBarry Smithextern int VecSetValues(Vec, int, int *,Scalar*,InsertMode);
ee50ffe9SBarry Smithextern int VecAssemblyBegin(Vec);
ee50ffe9SBarry Smithextern int VecAssemblyEnd(Vec);
8ed539a5SBarry Smith
*9d00d63dSBarry Smithtypedef enum {SCATTERREVERSE=1,SCATTERDOWN=2,SCATTERUP=4,SCATTERALL=8,
*9d00d63dSBarry Smith              SCATTERALLREVERSE=9} ScatterMode;
ee50ffe9SBarry Smith
ee50ffe9SBarry Smithextern int VecScatterBegin(Vec,IS,Vec,IS,InsertMode,ScatterMode,VecScatterCtx);
ee50ffe9SBarry Smithextern int VecScatterEnd(Vec,IS,Vec,IS,InsertMode,ScatterMode,VecScatterCtx);
20563c6bSBarry Smithextern int VecScatterCtxCreate(Vec,IS,Vec,IS,VecScatterCtx *);
20563c6bSBarry Smithextern int VecScatterCtxDestroy(VecScatterCtx);
d6dfbf8fSBarry Smithextern int VecScatterCtxCopy(VecScatterCtx,VecScatterCtx *);
d6dfbf8fSBarry Smith
*9d00d63dSBarry Smithtypedef enum {PIPELINEDOWN=0,PIPELINEUP=1} PipelineMode;
ee50ffe9SBarry Smith
ca9b4cbeSLois Curfman McInnesextern int VecPipelineBegin(Vec,IS,Vec,IS,InsertMode,PipelineMode,VecScatterCtx);
ca9b4cbeSLois Curfman McInnesextern int VecPipelineEnd(Vec,IS,Vec,IS,InsertMode,PipelineMode,VecScatterCtx);
8ed539a5SBarry Smith
8ed539a5SBarry Smithextern int VecGetArray(Vec,Scalar**);
6b5873e3SBarry Smithextern int VecRestoreArray(Vec,Scalar**);
8ed539a5SBarry Smithextern int VecValidVector(Vec);
8ed539a5SBarry Smithextern int VecView(Vec, Viewer);
8ed539a5SBarry Smith
8ed539a5SBarry Smithextern int VecGetSize(Vec,int *);
8ed539a5SBarry Smithextern int VecGetLocalSize(Vec,int *);
d6dfbf8fSBarry Smithextern int VecGetOwnershipRange(Vec,int*,int*);
8ed539a5SBarry Smith
8ed539a5SBarry Smith/* utility routines */
8ed539a5SBarry Smithextern int VecReciprocal(Vec);
2eac72dbSBarry Smith
2eac72dbSBarry Smith#endif
2eac72dbSBarry Smith
2eac72dbSBarry Smith