Actual source code: ex37.c

  1: /*$Id: ex37.c,v 1.22 2001/08/07 03:03:07 balay Exp $*/

  3: static char help[] = "Tests MatCopy() and MatStore/RetrieveValues().nn";

 5:  #include petscmat.h

  7: int main(int argc,char **args)
  8: {
  9:   Mat         C,A;
 10:   int         i, n = 10,midx[3],ierr;
 11:   PetscScalar v[3];
 12:   PetscTruth  flg;

 14:   PetscInitialize(&argc,&args,(char *)0,help);
 15:   PetscOptionsGetInt(PETSC_NULL,"-n",&n,PETSC_NULL);

 17:   MatCreate(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,n,n,&C);
 18:   MatSetFromOptions(C);
 19:   MatCreate(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,n,n,&A);
 20:   MatSetFromOptions(A);

 22:   v[0] = -1.; v[1] = 2.; v[2] = -1.;
 23:   for (i=1; i<n-1; i++){
 24:     midx[2] = i-1; midx[1] = i; midx[0] = i+1;
 25:     MatSetValues(C,1,&i,3,midx,v,INSERT_VALUES);
 26:   }
 27:   i = 0; midx[0] = 0; midx[1] = 1;
 28:   v[0] = 2.0; v[1] = -1.;
 29:   MatSetValues(C,1,&i,2,midx,v,INSERT_VALUES);
 30:   i = n-1; midx[0] = n-2; midx[1] = n-1;
 31:   v[0] = -1.0; v[1] = 2.;
 32:   MatSetValues(C,1,&i,2,midx,v,INSERT_VALUES);

 34:   MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);
 35:   MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);

 37:   /* test matrices with different nonzero patterns */
 38:   MatCopy(C,A,DIFFERENT_NONZERO_PATTERN);

 40:   /* Now C and A have the same nonzero pattern */
 41:   MatSetOption(C,MAT_NO_NEW_NONZERO_LOCATIONS);
 42:   MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);
 43:   MatCopy(C,A,SAME_NONZERO_PATTERN);

 45:   MatView(C,PETSC_VIEWER_STDOUT_WORLD);
 46:   MatView(A,PETSC_VIEWER_STDOUT_WORLD);

 48:   MatEqual(A,C,&flg);
 49:   if (flg) {
 50:     PetscPrintf(PETSC_COMM_WORLD,"Matrices are equaln");
 51:   } else {
 52:     SETERRQ(1,"Matrices are NOT equal");
 53:   }

 55:   MatStoreValues(A);
 56:   MatZeroEntries(A);
 57:   MatRetrieveValues(A);
 58:   MatEqual(A,C,&flg);
 59:   if (flg) {
 60:     PetscPrintf(PETSC_COMM_WORLD,"Matrices are equaln");
 61:   } else {
 62:     SETERRQ(1,"Matrices are NOT equal");
 63:   }

 65:   MatDestroy(C);
 66:   MatDestroy(A);

 68:   PetscFinalize();
 69:   return 0;
 70: }

 72: