Actual source code: ex58.c


  2: static char help[] = "Tests MatTranspose() and MatEqual() for MPIAIJ matrices.\n\n";

  4: #include <petscmat.h>

  6: int main(int argc,char **argv)
  7: {
  8:   Mat            A,B;
  9:   PetscInt       m = 7,n,i,rstart,rend,cols[3];
 10:   PetscScalar    v[3];
 11:   PetscBool      equal;
 12:   const char     *eq[2];

 14:   PetscInitialize(&argc,&argv,(char*)0,help);
 15:   PetscOptionsGetInt(NULL,NULL,"-m",&m,NULL);
 16:   n    = m;

 18:   /* ------- Assemble matrix, --------- */

 20:   MatCreateAIJ(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,m,n,0,0,0,0,&A);
 21:   MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_FALSE);
 22:   MatGetOwnershipRange(A,&rstart,&rend);
 23:   if (!rstart) {
 24:     cols[0] = 0;
 25:     cols[1] = 1;
 26:     v[0]    = 2.0; v[1] = -1.0;
 27:     MatSetValues(A,1,&rstart,2,cols,v,INSERT_VALUES);
 28:     rstart++;
 29:   }
 30:   if (rend == m) {
 31:     rend--;
 32:     cols[0] = rend-1;
 33:     cols[1] = rend;
 34:     v[0]    = -1.0; v[1] = 2.0;
 35:     MatSetValues(A,1,&rend,2,cols,v,INSERT_VALUES);
 36:   }
 37:   v[0] = -1.0; v[1] = 2.0; v[2] = -1.0;
 38:   for (i=rstart; i<rend; i++) {
 39:     cols[0] = i-1;
 40:     cols[1] = i;
 41:     cols[2] = i+1;
 42:     MatSetValues(A,1,&i,3,cols,v,INSERT_VALUES);
 43:   }
 44:   MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
 45:   MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);

 47:   MatTranspose(A,MAT_INITIAL_MATRIX,&B);

 49:   MatEqual(A,B,&equal);

 51:   eq[0] = "not equal";
 52:   eq[1] = "equal";
 53:   PetscPrintf(PETSC_COMM_WORLD,"Matrices are %s\n",eq[equal]);

 55:   MatTranspose(A,MAT_REUSE_MATRIX,&B);
 56:   MatEqual(A,B,&equal);
 57:   if (!equal) PetscPrintf(PETSC_COMM_WORLD,"MatTranspose with MAT_REUSE_MATRIX failed");

 59:   /* Free data structures */
 60:   MatDestroy(&A);
 61:   MatDestroy(&B);

 63:   PetscFinalize();
 64:   return 0;
 65: }

 67: /*TEST

 69:     test:

 71:     test:
 72:       suffix: 2
 73:       nsize: 2
 74:       output_file: output/ex58_1.out

 76: TEST*/