xref: /petsc/src/mat/tests/ex124.c (revision e00437b97e7e022a766c323f69bd5255763f3ff2)
1c4762a1bSJed Brown static char help[] = "Check the difference of the two matrices \n\
2c4762a1bSJed Brown Reads PETSc matrix A and B, then check B=A-B \n\
3c4762a1bSJed Brown Input parameters include\n\
4c4762a1bSJed Brown   -fA <input_file> -fB <input_file> \n\n";
5c4762a1bSJed Brown 
6c4762a1bSJed Brown #include <petscmat.h>
7c4762a1bSJed Brown 
8ad0e0ea3SStefano Zampini int main(int argc,char **args)
9c4762a1bSJed Brown {
10c4762a1bSJed Brown   Mat            A,B;
11c4762a1bSJed Brown   PetscViewer    fd;
12c4762a1bSJed Brown   char           file[2][PETSC_MAX_PATH_LEN];
13c4762a1bSJed Brown   PetscBool      flg;
14c4762a1bSJed Brown   PetscMPIInt    size;
15c4762a1bSJed Brown   PetscInt       ma,na,mb,nb;
16c4762a1bSJed Brown 
179566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc,&args,(char*)0,help));
189566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size));
19be096a46SBarry Smith   PetscCheck(size == 1,PETSC_COMM_WORLD,PETSC_ERR_WRONG_MPI_SIZE,"This is a uniprocessor example only!");
20c4762a1bSJed Brown 
21c4762a1bSJed Brown   /* read the two matrices, A and B */
229566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetString(NULL,NULL,"-fA",file[0],sizeof(file[0]),&flg));
2328b400f6SJacob Faibussowitsch   PetscCheck(flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -fA options");
249566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetString(NULL,NULL,"-fB",file[1],sizeof(file[1]),&flg));
2528b400f6SJacob Faibussowitsch   PetscCheck(flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -fP options");
26c4762a1bSJed Brown 
27c4762a1bSJed Brown   /* Load matrices */
289566063dSJacob Faibussowitsch   PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file[0],FILE_MODE_READ,&fd));
299566063dSJacob Faibussowitsch   PetscCall(MatCreate(PETSC_COMM_WORLD,&A));
309566063dSJacob Faibussowitsch   PetscCall(MatLoad(A,fd));
319566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&fd));
32c4762a1bSJed Brown   printf("\n A:\n");
33c4762a1bSJed Brown   printf("----------------------\n");
349566063dSJacob Faibussowitsch   PetscCall(MatView(A,PETSC_VIEWER_STDOUT_WORLD));
359566063dSJacob Faibussowitsch   PetscCall(MatGetSize(A,&ma,&na));
36c4762a1bSJed Brown 
379566063dSJacob Faibussowitsch   PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file[1],FILE_MODE_READ,&fd));
389566063dSJacob Faibussowitsch   PetscCall(MatCreate(PETSC_COMM_WORLD,&B));
399566063dSJacob Faibussowitsch   PetscCall(MatLoad(B,fd));
409566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&fd));
41c4762a1bSJed Brown   printf("\n B:\n");
42c4762a1bSJed Brown   printf("----------------------\n");
439566063dSJacob Faibussowitsch   PetscCall(MatView(B,PETSC_VIEWER_STDOUT_WORLD));
449566063dSJacob Faibussowitsch   PetscCall(MatGetSize(B,&mb,&nb));
45c4762a1bSJed Brown 
46c4762a1bSJed Brown   /* Compute B = -A + B */
47*e00437b9SBarry Smith   PetscCheck(ma == mb && na == nb,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"nonconforming matrix size");
489566063dSJacob Faibussowitsch   PetscCall(MatAXPY(B,-1.0,A,DIFFERENT_NONZERO_PATTERN));
49c4762a1bSJed Brown   printf("\n B - A:\n");
50c4762a1bSJed Brown   printf("----------------------\n");
519566063dSJacob Faibussowitsch   PetscCall(MatView(B,PETSC_VIEWER_STDOUT_WORLD));
52c4762a1bSJed Brown 
539566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&B));
549566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&A));
559566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
56b122ec5aSJacob Faibussowitsch   return 0;
57c4762a1bSJed Brown }
58