mat/tests/ex259.c

26cec326SBarry Smithstatic char help[] = "Test of setting values in a matrix without preallocation\n\n";
26cec326SBarry Smith
26cec326SBarry Smith#include <petscmat.h>
26cec326SBarry Smith
26cec326SBarry SmithPetscErrorCode ex1_nonsquare_bs1(void)
26cec326SBarry Smith{
26cec326SBarry Smith  Mat       A;
26cec326SBarry Smith  PetscInt  M, N, m, n, bs = 1;
26cec326SBarry Smith  char      type[16];
26cec326SBarry Smith  PetscBool flg;
26cec326SBarry Smith
26cec326SBarry Smith  /*
26cec326SBarry Smith     Create the matrix
26cec326SBarry Smith  */
26cec326SBarry Smith  PetscFunctionBegin;
26cec326SBarry Smith  M = 10;
26cec326SBarry Smith  N = 12;
26cec326SBarry Smith  PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
26cec326SBarry Smith  PetscCall(PetscOptionsGetString(NULL, NULL, "-type", type, sizeof(type), &flg));
26cec326SBarry Smith  if (flg) PetscCall(MatSetType(A, type));
26cec326SBarry Smith  else PetscCall(MatSetType(A, MATAIJ));
26cec326SBarry Smith  PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, M, N));
26cec326SBarry Smith  PetscCall(PetscOptionsGetInt(NULL, NULL, "-bs", &bs, NULL));
26cec326SBarry Smith  PetscCall(MatSetBlockSize(A, bs));
26cec326SBarry Smith  PetscCall(MatSetFromOptions(A));
26cec326SBarry Smith
26cec326SBarry Smith  /*
26cec326SBarry Smith     Get the sizes of the matrix
26cec326SBarry Smith  */
26cec326SBarry Smith  PetscCall(MatGetLocalSize(A, &m, &n));
26cec326SBarry Smith
26cec326SBarry Smith  /*
26cec326SBarry Smith     Insert non-zero pattern (e.g. perform a sweep over the grid).
26cec326SBarry Smith     You can use MatSetValues(), MatSetValuesBlocked() or MatSetValue().
26cec326SBarry Smith  */
26cec326SBarry Smith  {
26cec326SBarry Smith    PetscInt    ii, jj;
26cec326SBarry Smith    PetscScalar vv = 22.0;
26cec326SBarry Smith
26cec326SBarry Smith    ii = 3;
26cec326SBarry Smith    jj = 3;
26cec326SBarry Smith    PetscCall(MatSetValue(A, ii, jj, vv, INSERT_VALUES));
26cec326SBarry Smith
26cec326SBarry Smith    ii = 7;
26cec326SBarry Smith    jj = 4;
26cec326SBarry Smith    PetscCall(MatSetValue(A, ii, jj, vv, INSERT_VALUES));
26cec326SBarry Smith    PetscCall(MatSetValue(A, jj, ii, vv, INSERT_VALUES));
26cec326SBarry Smith
26cec326SBarry Smith    ii = 9;
26cec326SBarry Smith    jj = 7;
26cec326SBarry Smith    PetscCall(MatSetValue(A, ii, jj, vv, INSERT_VALUES));
26cec326SBarry Smith    PetscCall(MatSetValue(A, jj, ii, vv, INSERT_VALUES));
26cec326SBarry Smith  }
26cec326SBarry Smith  PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
26cec326SBarry Smith  PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
26cec326SBarry Smith
26cec326SBarry Smith  PetscCall(PetscViewerPushFormat(PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_ASCII_COMMON));
26cec326SBarry Smith  PetscCall(MatView(A, PETSC_VIEWER_STDOUT_WORLD));
26cec326SBarry Smith
26cec326SBarry Smith  /*
26cec326SBarry Smith     Insert same location non-zero values into A.
26cec326SBarry Smith  */
26cec326SBarry Smith  {
26cec326SBarry Smith    PetscInt    ii, jj;
26cec326SBarry Smith    PetscScalar vv;
26cec326SBarry Smith
26cec326SBarry Smith    ii = 3;
26cec326SBarry Smith    jj = 3;
26cec326SBarry Smith    vv = 0.3;
26cec326SBarry Smith    PetscCall(MatSetValue(A, ii, jj, vv, INSERT_VALUES));
26cec326SBarry Smith
26cec326SBarry Smith    ii = 7;
26cec326SBarry Smith    jj = 4;
26cec326SBarry Smith    vv = 3.3;
26cec326SBarry Smith    PetscCall(MatSetValue(A, ii, jj, vv, INSERT_VALUES));
26cec326SBarry Smith    PetscCall(MatSetValue(A, jj, ii, vv, INSERT_VALUES));
26cec326SBarry Smith
26cec326SBarry Smith    ii = 9;
26cec326SBarry Smith    jj = 7;
26cec326SBarry Smith    vv = 4.3;
26cec326SBarry Smith    PetscCall(MatSetValue(A, ii, jj, vv, INSERT_VALUES));
26cec326SBarry Smith    PetscCall(MatSetValue(A, jj, ii, vv, INSERT_VALUES));
26cec326SBarry Smith  }
26cec326SBarry Smith  PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
26cec326SBarry Smith  PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
26cec326SBarry Smith
26cec326SBarry Smith  PetscCall(MatView(A, PETSC_VIEWER_STDOUT_WORLD));
26cec326SBarry Smith
26cec326SBarry Smith  PetscCall(MatDestroy(&A));
26cec326SBarry Smith  PetscFunctionReturn(PETSC_SUCCESS);
26cec326SBarry Smith}
26cec326SBarry Smith
26cec326SBarry Smithint main(int argc, char **args)
26cec326SBarry Smith{
26cec326SBarry Smith  PetscFunctionBeginUser;
*c8025a54SPierre Jolivet  PetscCall(PetscInitialize(&argc, &args, NULL, help));
26cec326SBarry Smith  PetscCall(ex1_nonsquare_bs1());
26cec326SBarry Smith  PetscCall(PetscFinalize());
26cec326SBarry Smith  return 0;
26cec326SBarry Smith}
26cec326SBarry Smith
26cec326SBarry Smith/*TEST
26cec326SBarry Smith
26cec326SBarry Smith   testset:
26cec326SBarry Smith     args: -bs {{1 2}} -type {{aij baij sbaij}}
26cec326SBarry Smith     filter: grep -v "type:"
26cec326SBarry Smith     test:
26cec326SBarry Smith     test:
26cec326SBarry Smith       suffix: 2
26cec326SBarry Smith       nsize: 2
26cec326SBarry Smith
26cec326SBarry SmithTEST*/