1c4762a1bSJed Brown static char help[] = "Tests MatComputeOperator() and MatComputeOperatorTranspose()\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscmat.h> 4c4762a1bSJed Brown 5d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv) 6d71ae5a4SJacob Faibussowitsch { 7c4762a1bSJed Brown Mat A, Ae, Aet; 8c4762a1bSJed Brown char filename[PETSC_MAX_PATH_LEN]; 9c4762a1bSJed Brown char expltype[128], *etype = NULL; 10c4762a1bSJed Brown PetscInt bs = 1; 11c4762a1bSJed Brown PetscBool flg, check = PETSC_TRUE; 12c4762a1bSJed Brown 13327415f7SBarry Smith PetscFunctionBeginUser; 149566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 15c4762a1bSJed Brown 169566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetString(NULL, NULL, "-expl_type", expltype, sizeof(expltype), &flg)); 1748a46eb9SPierre Jolivet if (flg) PetscCall(PetscStrallocpy(expltype, &etype)); 189566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetString(NULL, NULL, "-f", filename, sizeof(filename), &flg)); 199566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-bs", &bs, NULL)); 20c4762a1bSJed Brown if (!flg) { 21c4762a1bSJed Brown PetscInt M = 13, N = 6; 22*26cec326SBarry Smith PetscBool sbaij; 23c4762a1bSJed Brown 249566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL)); 259566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL)); 269566063dSJacob Faibussowitsch PetscCall(MatCreateDense(PETSC_COMM_WORLD, PETSC_DECIDE, PETSC_DECIDE, M, N, NULL, &A)); 279566063dSJacob Faibussowitsch PetscCall(MatSetBlockSize(A, bs)); 289566063dSJacob Faibussowitsch PetscCall(MatSetRandom(A, NULL)); 29*26cec326SBarry Smith PetscCall(PetscStrcmp(etype, MATSBAIJ, &sbaij)); 30*26cec326SBarry Smith if (sbaij) { 31*26cec326SBarry Smith Mat At; 32*26cec326SBarry Smith PetscCall(MatTranspose(A, MAT_INITIAL_MATRIX, &At)); 33*26cec326SBarry Smith PetscCall(MatAXPY(A, 1.0, At, DIFFERENT_NONZERO_PATTERN)); 34*26cec326SBarry Smith PetscCall(MatDestroy(&At)); 35*26cec326SBarry Smith } 36c4762a1bSJed Brown } else { 37c4762a1bSJed Brown PetscViewer viewer; 38c4762a1bSJed Brown 399566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, filename, FILE_MODE_READ, &viewer)); 409566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &A)); 419566063dSJacob Faibussowitsch PetscCall(MatSetBlockSize(A, bs)); 429566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(A)); 439566063dSJacob Faibussowitsch PetscCall(MatLoad(A, viewer)); 449566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 45c4762a1bSJed Brown } 469566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)A, "Matrix")); 479566063dSJacob Faibussowitsch PetscCall(MatViewFromOptions(A, NULL, "-view_expl")); 48c4762a1bSJed Brown 499566063dSJacob Faibussowitsch PetscCall(MatComputeOperator(A, etype, &Ae)); 509566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)Ae, "Explicit matrix")); 519566063dSJacob Faibussowitsch PetscCall(MatViewFromOptions(Ae, NULL, "-view_expl")); 52c4762a1bSJed Brown 539566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL, NULL, "-check", &check, NULL)); 54c4762a1bSJed Brown if (check) { 55c4762a1bSJed Brown Mat A2; 56c4762a1bSJed Brown PetscReal err, tol = PETSC_SMALL; 57c4762a1bSJed Brown 589566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetReal(NULL, NULL, "-tol", &tol, NULL)); 599566063dSJacob Faibussowitsch PetscCall(MatConvert(A, etype, MAT_INITIAL_MATRIX, &A2)); 609566063dSJacob Faibussowitsch PetscCall(MatAXPY(A2, -1.0, Ae, DIFFERENT_NONZERO_PATTERN)); 619566063dSJacob Faibussowitsch PetscCall(MatNorm(A2, NORM_FROBENIUS, &err)); 6248a46eb9SPierre Jolivet if (err > tol) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error %g > %g (type %s)\n", (double)err, (double)tol, etype)); 639566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A2)); 64c4762a1bSJed Brown } 65c4762a1bSJed Brown 669566063dSJacob Faibussowitsch PetscCall(MatComputeOperatorTranspose(A, etype, &Aet)); 679566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)Aet, "Explicit matrix transpose")); 689566063dSJacob Faibussowitsch PetscCall(MatViewFromOptions(Aet, NULL, "-view_expl")); 69c4762a1bSJed Brown 709566063dSJacob Faibussowitsch PetscCall(PetscFree(etype)); 719566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Ae)); 729566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Aet)); 739566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 749566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 75b122ec5aSJacob Faibussowitsch return 0; 76c4762a1bSJed Brown } 77c4762a1bSJed Brown 78c4762a1bSJed Brown /*TEST 79c4762a1bSJed Brown 80c4762a1bSJed Brown test: 81c4762a1bSJed Brown output_file: output/ex222_null.out 82c4762a1bSJed Brown 83c4762a1bSJed Brown testset: 84c4762a1bSJed Brown suffix: matexpl_rect 85c4762a1bSJed Brown output_file: output/ex222_null.out 86c4762a1bSJed Brown nsize: {{1 3}} 87c4762a1bSJed Brown args: -expl_type {{dense aij baij}} 88c4762a1bSJed Brown 89c4762a1bSJed Brown testset: 90c4762a1bSJed Brown suffix: matexpl_square 91c4762a1bSJed Brown output_file: output/ex222_null.out 92c4762a1bSJed Brown nsize: {{1 3}} 93c4762a1bSJed Brown args: -bs {{1 2 3}} -M 36 -N 36 -expl_type {{dense aij baij sbaij}} 94c4762a1bSJed Brown 95c4762a1bSJed Brown TEST*/ 96