static char help[] = "Tests MatMPI{AIJ,BAIJ,SBAIJ}SetPreallocationCSR\n\n";

#include <petscmat.h>

int main(int argc, char **argv)
{
  PetscInt    ia[3]   = {0, 2, 4};
  PetscInt    ja[4]   = {0, 1, 0, 1};
  PetscScalar c[16]   = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
  PetscInt    ia2[5]  = {0, 4, 8, 12, 16};
  PetscInt    ja2[16] = {0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3};
  PetscScalar c2[16]  = {0, 1, 4, 5, 2, 3, 6, 7, 8, 9, 12, 13, 10, 11, 14, 15};
  PetscMPIInt size, rank;
  Mat         ssbaij;
  PetscBool   rect = PETSC_FALSE;

  PetscFunctionBeginUser;
  PetscCall(PetscInitialize(&argc, &argv, NULL, help));
  PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
  PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
  PetscCheck(size > 1, PETSC_COMM_WORLD, PETSC_ERR_WRONG_MPI_SIZE, "This is an example with more than one process");
  if (rank) {
    PetscInt i;
    for (i = 0; i < 3; i++) ia[i] = 0;
    for (i = 0; i < 5; i++) ia2[i] = 0;
  }
  PetscCall(PetscOptionsGetBool(NULL, NULL, "-rect", &rect, NULL));
  PetscCall(MatCreate(PETSC_COMM_WORLD, &ssbaij));
  PetscCall(MatSetBlockSize(ssbaij, 2));
  if (rect) {
    PetscCall(MatSetType(ssbaij, MATMPIBAIJ));
    PetscCall(MatSetSizes(ssbaij, 4, 6, PETSC_DECIDE, PETSC_DECIDE));
  } else {
    PetscCall(MatSetType(ssbaij, MATMPISBAIJ));
    PetscCall(MatSetSizes(ssbaij, 4, 4, PETSC_DECIDE, PETSC_DECIDE));
  }
  PetscCall(MatSetFromOptions(ssbaij));
  PetscCall(MatMPIAIJSetPreallocationCSR(ssbaij, ia2, ja2, c2));
  PetscCall(MatMPIBAIJSetPreallocationCSR(ssbaij, 2, ia, ja, c));
  PetscCall(MatMPISBAIJSetPreallocationCSR(ssbaij, 2, ia, ja, c));
  PetscCall(MatViewFromOptions(ssbaij, NULL, "-view"));
  PetscCall(MatDestroy(&ssbaij));
  PetscCall(PetscFinalize());
  return 0;
}

/*TEST

  test:
    filter: grep -v type | sed -e "s/\.//g"
    suffix: aijbaij_csr
    nsize: 2
    args: -mat_type {{aij baij}} -view -rect {{0 1}}

  test:
    filter: sed -e "s/\.//g"
    suffix: sbaij_csr
    nsize: 2
    args: -mat_type sbaij -view -rect {{0 1}}

TEST*/