1 static char help[] = "Tests MatCreateConstantDiagonal().\n" 2 "\n"; 3 4 #include <petscmat.h> 5 6 int main(int argc, char **args) 7 { 8 Vec X, Y; 9 Mat A, Adup, B, Af; 10 PetscBool flg; 11 PetscReal xnorm, ynorm, anorm; 12 13 PetscFunctionBeginUser; 14 PetscCall(PetscInitialize(&argc, &args, (char *)0, help)); 15 16 PetscCall(MatCreateConstantDiagonal(PETSC_COMM_WORLD, PETSC_DETERMINE, PETSC_DETERMINE, 20, 20, 3.0, &A)); 17 PetscCall(MatCreateVecs(A, &X, &Y)); 18 PetscCall(MatView(A, PETSC_VIEWER_STDOUT_WORLD)); 19 20 PetscCall(VecSetRandom(X, NULL)); 21 PetscCall(VecNorm(X, NORM_2, &xnorm)); 22 PetscCall(MatMult(A, X, Y)); 23 PetscCall(VecNorm(Y, NORM_2, &ynorm)); 24 PetscCheck(PetscAbsReal(ynorm - 3 * xnorm) <= PETSC_SMALL, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Expected norm %g actual norm %g", (double)(3 * xnorm), (double)ynorm); 25 PetscCall(MatShift(A, 5.0)); 26 PetscCall(MatScale(A, .5)); 27 PetscCall(MatView(A, PETSC_VIEWER_STDOUT_WORLD)); 28 PetscCall(MatNorm(A, NORM_FROBENIUS, &anorm)); 29 PetscCheck(PetscAbsReal(anorm - 4.0) <= PETSC_SMALL, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Expected norm 4.0 actual norm %g", (double)anorm); 30 31 /* Convert to AIJ (exercises MatGetRow/MatRestoreRow) */ 32 PetscCall(MatConvert(A, MATAIJ, MAT_INITIAL_MATRIX, &B)); 33 PetscCall(MatMultEqual(A, B, 10, &flg)); 34 if (!flg) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error MatMult\n")); 35 PetscCall(MatMultAddEqual(A, B, 10, &flg)); 36 if (!flg) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error MatMultAdd\n")); 37 PetscCall(MatMultTransposeEqual(A, B, 10, &flg)); 38 if (!flg) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error MatMultTranspose\n")); 39 PetscCall(MatMultTransposeAddEqual(A, B, 10, &flg)); 40 if (!flg) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error MatMultTransposeAdd\n")); 41 PetscCall(MatMultHermitianTransposeEqual(A, B, 10, &flg)); 42 if (!flg) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error MatMultHermitianTranspose\n")); 43 PetscCall(MatMultHermitianTransposeAddEqual(A, B, 10, &flg)); 44 if (!flg) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Error MatMultHermitianTransposeAdd\n")); 45 46 PetscCall(MatGetDiagonal(A, Y)); 47 PetscCall(MatGetFactor(A, MATSOLVERPETSC, MAT_FACTOR_LU, &Af)); 48 PetscCall(MatLUFactorSymbolic(Af, A, NULL, NULL, NULL)); 49 PetscCall(MatLUFactorNumeric(Af, A, NULL)); 50 PetscCall(MatSolve(Af, X, Y)); 51 PetscCall(VecNorm(Y, NORM_2, &ynorm)); 52 PetscCheck(PetscAbsReal(ynorm - xnorm / 4) <= PETSC_SMALL, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Expected norm %g actual norm %g", (double)(.25 * xnorm), (double)ynorm); 53 54 // Solve can be called without factorization 55 PetscCall(MatSolve(A, X, Y)); 56 PetscCall(VecNorm(Y, NORM_2, &ynorm)); 57 PetscCheck(PetscAbsReal(ynorm - xnorm / 4) <= PETSC_SMALL, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Expected norm %g actual norm %g", (double)(.25 * xnorm), (double)ynorm); 58 59 // For a scalar multiple of the identity smoothing is equivalent to solving 60 PetscCall(MatSOR(A, X, 1.5, SOR_FORWARD_SWEEP, 0.0, 1, 1, Y)); 61 PetscCall(VecNorm(Y, NORM_2, &ynorm)); 62 PetscCheck(PetscAbsReal(ynorm - xnorm / 4) <= PETSC_SMALL, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Expected norm %g actual norm %g", (double)(.25 * xnorm), (double)ynorm); 63 64 PetscCall(MatDuplicate(A, MAT_COPY_VALUES, &Adup)); 65 PetscCall(MatEqual(A, Adup, &flg)); 66 PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "MatEqual after copy failure"); 67 68 PetscCall(MatDestroy(&Adup)); 69 PetscCall(MatDestroy(&A)); 70 PetscCall(MatDestroy(&B)); 71 PetscCall(MatDestroy(&Af)); 72 PetscCall(VecDestroy(&X)); 73 PetscCall(VecDestroy(&Y)); 74 75 PetscCall(PetscFinalize()); 76 return 0; 77 } 78 79 /*TEST 80 81 test: 82 nsize: 2 83 84 TEST*/ 85