xref: /petsc/src/mat/tests/ex167.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)

static char help[] = "Extract submatrices using unsorted indices. For SEQSBAIJ either sort both rows and columns, or sort none.\n\n";
/*
   Take a 4x4 grid and form a 5-point stencil graph Laplacian over it.
   Partition the grid into two subdomains by splitting into two in the j-direction (slowest varying).
   Impose an overlap of 1 and order the subdomains with the j-direction varying fastest.
   Extract the subdomain submatrices, one per rank.
*/
/* Results:
    Sequential:
    - seqaij:   will error out, if rows or columns are unsorted
    - seqbaij:  will extract submatrices correctly even for unsorted row or column indices
    - seqsbaij: will extract submatrices correctly even for unsorted row and column indices (both must be sorted or not);
                CANNOT automatically report inversions, because MatGetRow is not available.
    MPI:
    - mpiaij:   will error out, if columns are unsorted
    - mpibaij:  will error out, if columns are unsorted.
    - mpisbaij: will error out, if columns are unsorted; even with unsorted rows will produce correct submatrices;
                CANNOT automatically report inversions, because MatGetRow is not available.
*/

#include <petscmat.h>
#include <petscis.h>

int main(int argc,char **args)
{
  Mat            A, *S;
  IS             rowis[2], colis[2];
  PetscInt       n,N,i,j,k,l,nsub,Jlow[2] = {0,1}, *jlow, Jhigh[2] = {3,4}, *jhigh, row, col, *subindices, ncols;
  const PetscInt *cols;
  PetscScalar    v;
  PetscMPIInt    rank, size, p, inversions, total_inversions;
  PetscBool      sort_rows, sort_cols, show_inversions;

  CHKERRQ(PetscInitialize(&argc,&args,(char*)0,help));
  CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank));
  CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size));
  PetscCheckFalse(size>2,PETSC_COMM_WORLD,PETSC_ERR_ARG_WRONG, "A uniprocessor or two-processor example only.");

  CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A));
  if (size > 1) {
    n = 8; N = 16;
  } else {
    n = 16; N = 16;
  }
  CHKERRQ(MatSetSizes(A,n,n,N,N));
  CHKERRQ(MatSetFromOptions(A));
  CHKERRQ(MatSetUp(A));

  /* Don't care if the entries are set multiple times by different procs. */
  for (i=0; i<4; ++i) {
    for (j = 0; j<4; ++j) {
      row = j*4+i;
      v   = -1.0;
      if (i>0) {
        col =  row-1; CHKERRQ(MatSetValues(A,1,&row,1,&col,&v,INSERT_VALUES));
      }
      if (i<3) {
        col = row+1; CHKERRQ(MatSetValues(A,1,&row,1,&col,&v,INSERT_VALUES));
      }
      if (j>0) {
        col = row-4; CHKERRQ(MatSetValues(A,1,&row,1,&col,&v,INSERT_VALUES));
      }
      if (j<3) {
        col = row+4; CHKERRQ(MatSetValues(A,1,&row,1,&col,&v,INSERT_VALUES));
      }
      v    = 4.0;
      CHKERRQ(MatSetValues(A,1,&row,1,&row,&v,INSERT_VALUES));
    }
  }
  CHKERRQ(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY));
  CHKERRQ(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY));
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD, "Original matrix\n"));
  CHKERRQ(MatView(A,PETSC_VIEWER_STDOUT_WORLD));

  if (size > 1) {
    nsub = 1; /* one subdomain per rank */
  }
  else {
    nsub = 2; /* both subdomains on rank 0 */
  }
  if (rank) {
    jlow = Jlow+1; jhigh = Jhigh+1;
  }
  else {
    jlow = Jlow; jhigh = Jhigh;
  }
  sort_rows = PETSC_FALSE;
  CHKERRQ(PetscOptionsGetBool(NULL,NULL, "-sort_rows", &sort_rows, NULL));
  sort_cols = PETSC_FALSE;
  CHKERRQ(PetscOptionsGetBool(NULL,NULL, "-sort_cols", &sort_cols, NULL));
  for (l = 0; l < nsub; ++l) {
    CHKERRQ(PetscMalloc1(12, &subindices));
    k    = 0;
    for (i = 0; i < 4; ++i) {
      for (j = jlow[l]; j < jhigh[l]; ++j) {
        subindices[k] = j*4+i;
        k++;
      }
    }
    CHKERRQ(ISCreateGeneral(PETSC_COMM_SELF, 12, subindices, PETSC_OWN_POINTER, rowis+l));
    if ((sort_rows && !sort_cols) || (!sort_rows && sort_cols)) {
      CHKERRQ(ISDuplicate(rowis[l],colis+l));
    } else {
      CHKERRQ(PetscObjectReference((PetscObject)rowis[l]));
      colis[l] = rowis[l];
    }
    if (sort_rows) {
      CHKERRQ(ISSort(rowis[l]));
    }
    if (sort_cols) {
      CHKERRQ(ISSort(colis[l]));
    }
  }

  CHKERRQ(MatCreateSubMatrices(A,nsub,rowis,colis,MAT_INITIAL_MATRIX, &S));

  show_inversions = PETSC_FALSE;

  CHKERRQ(PetscOptionsGetBool(NULL,NULL, "-show_inversions", &show_inversions, NULL));

  inversions = 0;
  for (p = 0; p < size; ++p) {
    if (p == rank) {
      CHKERRQ(PetscPrintf(PETSC_COMM_SELF, "[%" PetscInt_FMT ":%" PetscInt_FMT "]: Number of subdomains: %" PetscInt_FMT ":\n", rank, size, nsub));
      for (l = 0; l < nsub; ++l) {
        PetscInt i0, i1;
        CHKERRQ(PetscPrintf(PETSC_COMM_SELF, "[%" PetscInt_FMT ":%" PetscInt_FMT "]: Subdomain row IS %" PetscInt_FMT ":\n", rank, size, l));
        CHKERRQ(ISView(rowis[l],PETSC_VIEWER_STDOUT_SELF));
        CHKERRQ(PetscPrintf(PETSC_COMM_SELF, "[%" PetscInt_FMT ":%" PetscInt_FMT "]: Subdomain col IS %" PetscInt_FMT ":\n", rank, size, l));
        CHKERRQ(ISView(colis[l],PETSC_VIEWER_STDOUT_SELF));
        CHKERRQ(PetscPrintf(PETSC_COMM_SELF, "[%" PetscInt_FMT ":%" PetscInt_FMT "]: Submatrix %" PetscInt_FMT ":\n", rank, size, l));
        CHKERRQ(MatView(S[l],PETSC_VIEWER_STDOUT_SELF));
        if (show_inversions) {
          CHKERRQ(MatGetOwnershipRange(S[l], &i0,&i1));
          for (i = i0; i < i1; ++i) {
            CHKERRQ(MatGetRow(S[l], i, &ncols, &cols, NULL));
            for (j = 1; j < ncols; ++j) {
              if (cols[j] < cols[j-1]) {
                CHKERRQ(PetscPrintf(PETSC_COMM_SELF, "***Inversion in row %" PetscInt_FMT ": col[%" PetscInt_FMT "] = %" PetscInt_FMT " < %" PetscInt_FMT " = col[%" PetscInt_FMT "]\n", i, j, cols[j], cols[j-1], j-1));
                inversions++;
              }
            }
            CHKERRQ(MatRestoreRow(S[l], i, &ncols, &cols, NULL));
          }
        }
      }
    }
    CHKERRMPI(MPI_Barrier(PETSC_COMM_WORLD));
  }
  if (show_inversions) {
    CHKERRMPI(MPI_Reduce(&inversions,&total_inversions,1,MPIU_INT, MPI_SUM,0,PETSC_COMM_WORLD));
    CHKERRQ(PetscPrintf(PETSC_COMM_WORLD, "*Total inversions: %" PetscInt_FMT "\n", total_inversions));
  }
  CHKERRQ(MatDestroy(&A));

  for (l = 0; l < nsub; ++l) {
    CHKERRQ(ISDestroy(&(rowis[l])));
    CHKERRQ(ISDestroy(&(colis[l])));
  }
  CHKERRQ(MatDestroySubMatrices(nsub,&S));
  CHKERRQ(PetscFinalize());
  return 0;
}