Date: 2013-07-02T14:44:18+00:00

Actual source code: ex131.c
petsc-3.4.2 2013-07-02

      2: static char help[] = "Tests MatMult() on MatLoad() matrix \n\n";

      4: #include <petscmat.h>

      8: int main(int argc,char **args)
      9: {
     10:   Mat            A;
     11:   Vec            x,b;
     13:   PetscViewer    fd;              /* viewer */
     14:   char           file[PETSC_MAX_PATH_LEN]; /* input file name */
     15:   PetscBool      flg;

     17:   PetscInitialize(&argc,&args,(char*)0,help);

     19:   /* Determine file from which we read the matrix A */
     20:   PetscOptionsGetString(NULL,"-f",file,PETSC_MAX_PATH_LEN,&flg);
     21:   if (!flg) SETERRQ(PETSC_COMM_WORLD,1,"Must indicate binary file with the -f option");

     23:   /* Load matrix A */
     24:   PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&fd);
     25:   MatCreate(PETSC_COMM_WORLD,&A);
     26:   MatLoad(A,fd);
     27:   flg  = PETSC_FALSE;
     28:   VecCreate(PETSC_COMM_WORLD,&x);
     29:   PetscOptionsGetString(NULL,"-vec",file,PETSC_MAX_PATH_LEN,&flg);
     30:   if (flg) {
     31:     if (file[0] == '0') {
     32:       PetscInt    m;
     33:       PetscScalar one = 1.0;
     34:       PetscInfo(0,"Using vector of ones for RHS\n");
     35:       MatGetLocalSize(A,&m,NULL);
     36:       VecSetSizes(x,m,PETSC_DECIDE);
     37:       VecSetFromOptions(x);
     38:       VecSet(x,one);
     39:     }
     40:   } else {
     41:     VecLoad(x,fd);
     42:     PetscViewerDestroy(&fd);
     43:   }
     44:   VecDuplicate(x,&b);
     45:   MatMult(A,x,b);

     47:   /* Print (for testing only) */
     48:   MatView(A,0);
     49:   VecView(b,0);
     50:   /* Free data structures */
     51:   MatDestroy(&A);
     52:   VecDestroy(&x);
     53:   VecDestroy(&b);
     54:   PetscFinalize();
     55:   return 0;
     56: }