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