1 2 static char help[] = "Tests MatDiagonalSet() 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; 10 PetscErrorCode ierr; 11 PetscViewer fd; /* viewer */ 12 char file[PETSC_MAX_PATH_LEN]; /* input file name */ 13 PetscReal norm; 14 PetscBool flg; 15 16 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 17 /* Determine file from which we read the matrix A */ 18 CHKERRQ(PetscOptionsGetString(NULL,NULL,"-f",file,sizeof(file),&flg)); 19 PetscCheckFalse(!flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -f option"); 20 21 /* Load matrix A */ 22 CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&fd)); 23 CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A)); 24 CHKERRQ(MatLoad(A,fd)); 25 CHKERRQ(PetscViewerDestroy(&fd)); 26 CHKERRQ(MatCreateVecs(A,&x,NULL)); 27 CHKERRQ(MatGetDiagonal(A,x)); 28 CHKERRQ(VecScale(x,-1.0)); 29 CHKERRQ(MatDiagonalSet(A,x,ADD_VALUES)); 30 CHKERRQ(MatGetDiagonal(A,x)); 31 CHKERRQ(VecNorm(x,NORM_2,&norm)); 32 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"Norm %g\n",(double)norm)); 33 34 /* Free data structures */ 35 CHKERRQ(MatDestroy(&A)); 36 CHKERRQ(VecDestroy(&x)); 37 ierr = PetscFinalize(); 38 return ierr; 39 } 40 41 /*TEST 42 43 test: 44 nsize: 4 45 requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES) 46 args: -f ${wPETSC_DIR}/share/petsc/datafiles/matrices/ns-real-int32-float64 -malloc_dump 47 48 TEST*/ 49