1 2 static char help[] = "Tests MatMult() on MatLoad() matrix \n\n"; 3 4 #include <petscmat.h> 5 6 int main(int argc,char **args) 7 { 8 Mat A; 9 Vec x,b; 10 PetscViewer fd; /* viewer */ 11 char file[PETSC_MAX_PATH_LEN]; /* input file name */ 12 PetscBool flg; 13 14 CHKERRQ(PetscInitialize(&argc,&args,(char*)0,help)); 15 /* Determine file from which we read the matrix A */ 16 CHKERRQ(PetscOptionsGetString(NULL,NULL,"-f",file,sizeof(file),&flg)); 17 PetscCheck(flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -f option"); 18 19 /* Load matrix A */ 20 CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&fd)); 21 CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A)); 22 CHKERRQ(MatLoad(A,fd)); 23 flg = PETSC_FALSE; 24 CHKERRQ(VecCreate(PETSC_COMM_WORLD,&x)); 25 CHKERRQ(PetscOptionsGetString(NULL,NULL,"-vec",file,sizeof(file),&flg)); 26 if (flg) { 27 if (file[0] == '0') { 28 PetscInt m; 29 PetscScalar one = 1.0; 30 CHKERRQ(PetscInfo(0,"Using vector of ones for RHS\n")); 31 CHKERRQ(MatGetLocalSize(A,&m,NULL)); 32 CHKERRQ(VecSetSizes(x,m,PETSC_DECIDE)); 33 CHKERRQ(VecSetFromOptions(x)); 34 CHKERRQ(VecSet(x,one)); 35 } 36 } else { 37 CHKERRQ(VecLoad(x,fd)); 38 CHKERRQ(PetscViewerDestroy(&fd)); 39 } 40 CHKERRQ(VecDuplicate(x,&b)); 41 CHKERRQ(MatMult(A,x,b)); 42 43 /* Print (for testing only) */ 44 CHKERRQ(MatView(A,0)); 45 CHKERRQ(VecView(b,0)); 46 /* Free data structures */ 47 CHKERRQ(MatDestroy(&A)); 48 CHKERRQ(VecDestroy(&x)); 49 CHKERRQ(VecDestroy(&b)); 50 CHKERRQ(PetscFinalize()); 51 return 0; 52 } 53