xref: /petsc/src/mat/tests/ex124.c (revision 2c71b3e237ead271e4f3aa1505f92bf476e3413d)
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   PetscErrorCode ierr;
15c4762a1bSJed Brown   PetscMPIInt    size;
16c4762a1bSJed Brown   PetscInt       ma,na,mb,nb;
17c4762a1bSJed Brown 
18c4762a1bSJed Brown   ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr;
19ffc4695bSBarry Smith   ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr);
20*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(size != 1,PETSC_COMM_WORLD,PETSC_ERR_SUP,"This is a uniprocessor example only!");
21c4762a1bSJed Brown 
22c4762a1bSJed Brown   /* read the two matrices, A and B */
23589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-fA",file[0],sizeof(file[0]),&flg);CHKERRQ(ierr);
24*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(!flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -fA options");
25589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-fB",file[1],sizeof(file[1]),&flg);CHKERRQ(ierr);
26*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(!flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -fP options");
27c4762a1bSJed Brown 
28c4762a1bSJed Brown   /* Load matrices */
29c4762a1bSJed Brown   ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,file[0],FILE_MODE_READ,&fd);CHKERRQ(ierr);
30c4762a1bSJed Brown   ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr);
31c4762a1bSJed Brown   ierr = MatLoad(A,fd);CHKERRQ(ierr);
32c4762a1bSJed Brown   ierr = PetscViewerDestroy(&fd);CHKERRQ(ierr);
33c4762a1bSJed Brown   printf("\n A:\n");
34c4762a1bSJed Brown   printf("----------------------\n");
35c4762a1bSJed Brown   ierr = MatView(A,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
36c4762a1bSJed Brown   ierr = MatGetSize(A,&ma,&na);CHKERRQ(ierr);
37c4762a1bSJed Brown 
38c4762a1bSJed Brown   ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,file[1],FILE_MODE_READ,&fd);CHKERRQ(ierr);
39c4762a1bSJed Brown   ierr = MatCreate(PETSC_COMM_WORLD,&B);CHKERRQ(ierr);
40c4762a1bSJed Brown   ierr = MatLoad(B,fd);CHKERRQ(ierr);
41c4762a1bSJed Brown   ierr = PetscViewerDestroy(&fd);CHKERRQ(ierr);
42c4762a1bSJed Brown   printf("\n B:\n");
43c4762a1bSJed Brown   printf("----------------------\n");
44c4762a1bSJed Brown   ierr = MatView(B,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
45c4762a1bSJed Brown   ierr = MatGetSize(B,&mb,&nb);CHKERRQ(ierr);
46c4762a1bSJed Brown 
47c4762a1bSJed Brown   /* Compute B = -A + B */
48*2c71b3e2SJacob Faibussowitsch   PetscCheckFalse(ma != mb || na != nb,PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"nonconforming matrix size");
49c4762a1bSJed Brown   ierr = MatAXPY(B,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
50c4762a1bSJed Brown   printf("\n B - A:\n");
51c4762a1bSJed Brown   printf("----------------------\n");
52c4762a1bSJed Brown   ierr = MatView(B,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
53c4762a1bSJed Brown 
54c4762a1bSJed Brown   ierr = MatDestroy(&B);CHKERRQ(ierr);
55c4762a1bSJed Brown   ierr = MatDestroy(&A);CHKERRQ(ierr);
56c4762a1bSJed Brown   ierr = PetscFinalize();
57c4762a1bSJed Brown   return ierr;
58c4762a1bSJed Brown }
59