1c4762a1bSJed Brown static char help[] = "Tests MatComputeOperator() and MatComputeOperatorTranspose()\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscmat.h> 4c4762a1bSJed Brown 59371c9d4SSatish Balay int main(int argc, char **argv) { 6c4762a1bSJed Brown Mat A, Ae, Aet; 7c4762a1bSJed Brown char filename[PETSC_MAX_PATH_LEN]; 8c4762a1bSJed Brown char expltype[128], *etype = NULL; 9c4762a1bSJed Brown PetscInt bs = 1; 10c4762a1bSJed Brown PetscBool flg, check = PETSC_TRUE; 11c4762a1bSJed Brown 12327415f7SBarry Smith PetscFunctionBeginUser; 139566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 14c4762a1bSJed Brown 159566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetString(NULL, NULL, "-expl_type", expltype, sizeof(expltype), &flg)); 16*48a46eb9SPierre Jolivet if (flg) PetscCall(PetscStrallocpy(expltype, &etype)); 179566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetString(NULL, NULL, "-f", filename, sizeof(filename), &flg)); 189566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-bs", &bs, NULL)); 19c4762a1bSJed Brown if (!flg) { 20c4762a1bSJed Brown PetscInt M = 13, N = 6; 21c4762a1bSJed Brown 229566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL)); 239566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL)); 249566063dSJacob Faibussowitsch PetscCall(MatCreateDense(PETSC_COMM_WORLD, PETSC_DECIDE, PETSC_DECIDE, M, N, NULL, &A)); 259566063dSJacob Faibussowitsch PetscCall(MatSetBlockSize(A, bs)); 269566063dSJacob Faibussowitsch PetscCall(MatSetRandom(A, NULL)); 27c4762a1bSJed Brown } else { 28c4762a1bSJed Brown PetscViewer viewer; 29c4762a1bSJed Brown 309566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, filename, FILE_MODE_READ, &viewer)); 319566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &A)); 329566063dSJacob Faibussowitsch PetscCall(MatSetBlockSize(A, bs)); 339566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(A)); 349566063dSJacob Faibussowitsch PetscCall(MatLoad(A, viewer)); 359566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 36c4762a1bSJed Brown } 379566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)A, "Matrix")); 389566063dSJacob Faibussowitsch PetscCall(MatViewFromOptions(A, NULL, "-view_expl")); 39c4762a1bSJed Brown 409566063dSJacob Faibussowitsch PetscCall(MatComputeOperator(A, etype, &Ae)); 419566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)Ae, "Explicit matrix")); 429566063dSJacob Faibussowitsch PetscCall(MatViewFromOptions(Ae, NULL, "-view_expl")); 43c4762a1bSJed Brown 449566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL, NULL, "-check", &check, NULL)); 45c4762a1bSJed Brown if (check) { 46c4762a1bSJed Brown Mat A2; 47c4762a1bSJed Brown PetscReal err, tol = PETSC_SMALL; 48c4762a1bSJed Brown 499566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetReal(NULL, NULL, "-tol", &tol, NULL)); 509566063dSJacob Faibussowitsch PetscCall(MatConvert(A, etype, MAT_INITIAL_MATRIX, &A2)); 519566063dSJacob Faibussowitsch PetscCall(MatAXPY(A2, -1.0, Ae, DIFFERENT_NONZERO_PATTERN)); 529566063dSJacob Faibussowitsch PetscCall(MatNorm(A2, NORM_FROBENIUS, &err)); 53*48a46eb9SPierre Jolivet if (err > tol) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error %g > %g (type %s)\n", (double)err, (double)tol, etype)); 549566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A2)); 55c4762a1bSJed Brown } 56c4762a1bSJed Brown 579566063dSJacob Faibussowitsch PetscCall(MatComputeOperatorTranspose(A, etype, &Aet)); 589566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)Aet, "Explicit matrix transpose")); 599566063dSJacob Faibussowitsch PetscCall(MatViewFromOptions(Aet, NULL, "-view_expl")); 60c4762a1bSJed Brown 619566063dSJacob Faibussowitsch PetscCall(PetscFree(etype)); 629566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Ae)); 639566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Aet)); 649566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 659566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 66b122ec5aSJacob Faibussowitsch return 0; 67c4762a1bSJed Brown } 68c4762a1bSJed Brown 69c4762a1bSJed Brown /*TEST 70c4762a1bSJed Brown 71c4762a1bSJed Brown test: 72c4762a1bSJed Brown output_file: output/ex222_null.out 73c4762a1bSJed Brown 74c4762a1bSJed Brown testset: 75c4762a1bSJed Brown suffix: matexpl_rect 76c4762a1bSJed Brown output_file: output/ex222_null.out 77c4762a1bSJed Brown nsize: {{1 3}} 78c4762a1bSJed Brown args: -expl_type {{dense aij baij}} 79c4762a1bSJed Brown 80c4762a1bSJed Brown testset: 81c4762a1bSJed Brown suffix: matexpl_square 82c4762a1bSJed Brown output_file: output/ex222_null.out 83c4762a1bSJed Brown nsize: {{1 3}} 84c4762a1bSJed Brown args: -bs {{1 2 3}} -M 36 -N 36 -expl_type {{dense aij baij sbaij}} 85c4762a1bSJed Brown 86c4762a1bSJed Brown TEST*/ 87