xref: /petsc/include/petscvec.h (revision 93c39befdc2c545b92559b48e3c3f7c5a930faa6)
1*93c39befSBarry Smith /* $Id: vec.h,v 1.64 1997/06/12 21:52:14 bsmith Exp bsmith $ */
22eac72dbSBarry Smith /*
337f753daSBarry Smith     Defines the vector component of PETSc. Vectors generally represent
437f753daSBarry Smith   degrees of freedom for finite element/finite difference functions
584cb2905SBarry Smith   on a grid. They have more mathematical structure then simple arrays.
62eac72dbSBarry Smith */
72eac72dbSBarry Smith 
82eac72dbSBarry Smith #ifndef __VEC_PACKAGE
92eac72dbSBarry Smith #define __VEC_PACKAGE
102eac72dbSBarry Smith #include "is.h"
11bf5bf444SLois Curfman McInnes #include "sys.h"
122eac72dbSBarry Smith 
139e25ed09SBarry Smith #define VEC_COOKIE         PETSC_COOKIE+3
149e25ed09SBarry Smith #define VEC_SCATTER_COOKIE PETSC_COOKIE+4
159e25ed09SBarry Smith 
16b56ba379SLois Curfman McInnes typedef enum {VECSAME=-1, VECSEQ, VECMPI} VecType;
17f0479e8cSBarry Smith 
18f09e8eb9SSatish Balay typedef struct _p_Vec*         Vec;
19f09e8eb9SSatish Balay typedef struct _p_VecScatter*  VecScatter;
202eac72dbSBarry Smith 
21de7da479SBarry Smith extern int VecCreateSeq(MPI_Comm,int,Vec*);
228ed539a5SBarry Smith extern int VecCreateMPI(MPI_Comm,int,int,Vec*);
236469c4f9SBarry Smith extern int VecCreate(MPI_Comm,int,Vec*);
243369ce9aSBarry Smith extern int VecCreateSeqWithArray(MPI_Comm,int,Scalar*,Vec*);
253369ce9aSBarry Smith extern int VecCreateGhost(MPI_Comm,int,int,int,Vec*,Vec*);
262eac72dbSBarry Smith 
274b0e389bSBarry Smith extern int VecDestroy(Vec);
284b0e389bSBarry Smith 
298ed539a5SBarry Smith extern int VecDot(Vec,Vec,Scalar*);
308ed539a5SBarry Smith extern int VecTDot(Vec,Vec,Scalar*);
318ed539a5SBarry Smith extern int VecMDot(int,Vec,Vec*,Scalar*);
328ed539a5SBarry Smith extern int VecMTDot(int,Vec,Vec*,Scalar*);
33cddf8d76SBarry Smith 
34*93c39befSBarry Smith typedef enum {NORM_1=1,NORM_2=2,NORM_FROBENIUS=3,NORM_INFINITY=4,NORM_1_AND_2=5} NormType;
35cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY
36cddf8d76SBarry Smith extern int VecNorm(Vec,NormType,double *);
37ee50ffe9SBarry Smith extern int VecSum(Vec,Scalar*);
387c16e1c9SBarry Smith extern int VecMax(Vec,int*,double*);
397c16e1c9SBarry Smith extern int VecMin(Vec,int*,double*);
408ed539a5SBarry Smith extern int VecScale(Scalar*,Vec);
418ed539a5SBarry Smith extern int VecCopy(Vec,Vec);
4277c4ece6SBarry Smith extern int VecSetRandom(PetscRandom,Vec);
438ed539a5SBarry Smith extern int VecSet(Scalar*,Vec);
448ed539a5SBarry Smith extern int VecSwap(Vec,Vec);
458ed539a5SBarry Smith extern int VecAXPY(Scalar*,Vec,Vec);
4677c4ece6SBarry Smith extern int VecAXPBY(Scalar*,Scalar *,Vec,Vec);
478ed539a5SBarry Smith extern int VecMAXPY(int,Scalar*,Vec,Vec*);
488ed539a5SBarry Smith extern int VecAYPX(Scalar*,Vec,Vec);
498ed539a5SBarry Smith extern int VecWAXPY(Scalar*,Vec,Vec,Vec);
5077c4ece6SBarry Smith extern int VecPointwiseMult(Vec,Vec,Vec);
5177c4ece6SBarry Smith extern int VecPointwiseDivide(Vec,Vec,Vec);
524b0e389bSBarry Smith extern int VecShift(Scalar*,Vec);
534b0e389bSBarry Smith extern int VecReciprocal(Vec);
544b0e389bSBarry Smith extern int VecAbs(Vec);
556469c4f9SBarry Smith extern int VecDuplicate(Vec,Vec*);
56afc8d9b6SBarry Smith extern int VecDuplicateVecs(Vec,int,Vec**);
574b0e389bSBarry Smith extern int VecDestroyVecs(Vec*,int);
582eac72dbSBarry Smith 
5964119d58SLois Curfman McInnes typedef enum {NOT_SET_VALUES, INSERT_VALUES, ADD_VALUES} InsertMode;
6020563c6bSBarry Smith extern int VecSetValues(Vec,int,int*,Scalar*,InsertMode);
61ee50ffe9SBarry Smith extern int VecAssemblyBegin(Vec);
62ee50ffe9SBarry Smith extern int VecAssemblyEnd(Vec);
63b951964fSBarry Smith #define VecSetValue(v,i,va,mode) \
64b951964fSBarry Smith {int _ierr,_row = i; Scalar _va = va; \
65b951964fSBarry Smith   _ierr = VecSetValues(v,1,&_row,&_va,mode);CHKERRQ(_ierr); \
66b951964fSBarry Smith }
678ed539a5SBarry Smith 
68d252947aSBarry Smith typedef enum {SCATTER_FORWARD=0,SCATTER_REVERSE=1,SCATTER_FORWARD_LOCAL=2,
69d252947aSBarry Smith               SCATTER_REVERSE_LOCAL=3,SCATTER_LOCAL=2} ScatterMode;
7084cb2905SBarry Smith extern int VecScatterCreate(Vec,IS,Vec,IS,VecScatter *);
7190f02eecSBarry Smith extern int VecScatterPostRecvs(Vec,Vec,InsertMode,ScatterMode,VecScatter);
7208480c60SBarry Smith extern int VecScatterBegin(Vec,Vec,InsertMode,ScatterMode,VecScatter);
7308480c60SBarry Smith extern int VecScatterEnd(Vec,Vec,InsertMode,ScatterMode,VecScatter);
7408480c60SBarry Smith extern int VecScatterDestroy(VecScatter);
7508480c60SBarry Smith extern int VecScatterCopy(VecScatter,VecScatter *);
7608480c60SBarry Smith extern int VecScatterView(VecScatter,Viewer);
772195c698SBarry Smith extern int VecScatterRemap(VecScatter,int *,int*);
782195c698SBarry Smith 
798ed539a5SBarry Smith extern int VecGetArray(Vec,Scalar**);
806b5873e3SBarry Smith extern int VecRestoreArray(Vec,Scalar**);
81afc8d9b6SBarry Smith extern int VecPlaceArray(Vec,Scalar*);
82707f76d8SBarry Smith extern int VecGetArrays(Vec*,int,Scalar***);
83707f76d8SBarry Smith extern int VecRestoreArrays(Vec*,int,Scalar***);
8484cb2905SBarry Smith 
8577c4ece6SBarry Smith extern int VecValid(Vec,PetscTruth*);
868ed539a5SBarry Smith extern int VecView(Vec,Viewer);
8713054869SSatish Balay extern int VecEqual(Vec,Vec,PetscTruth*);
8884cb2905SBarry Smith extern int VecLoad(Viewer,Vec*);
898ed539a5SBarry Smith 
908ed539a5SBarry Smith extern int VecGetSize(Vec,int*);
911dd0cb96SLois Curfman McInnes extern int VecGetType(Vec,VecType*,char**);
928ed539a5SBarry Smith extern int VecGetLocalSize(Vec,int*);
93d6dfbf8fSBarry Smith extern int VecGetOwnershipRange(Vec,int*,int*);
948ed539a5SBarry Smith 
9590f02eecSBarry Smith extern int VecSetLocalToGlobalMapping(Vec, int,int *);
9690f02eecSBarry Smith extern int VecSetValuesLocal(Vec,int,int*,Scalar*,InsertMode);
9790f02eecSBarry Smith 
9890f02eecSBarry Smith typedef enum {VEC_IGNORE_OFF_PROCESSOR_ENTRIES} VecOption;
9990f02eecSBarry Smith extern int VecSetOption(Vec,VecOption);
10090f02eecSBarry Smith 
1013369ce9aSBarry Smith extern int DrawTensorContour(Draw,int,int,double *,double *,Vec);
10234233285SBarry Smith 
1032eac72dbSBarry Smith #endif
1042eac72dbSBarry Smith 
1052eac72dbSBarry Smith 
1067588ac45SBarry Smith 
107