Actual source code: ex203.c
2: static char help[] = "Tests incorrect use of MatDiagonalSet() for SHELL matrices\n\n";
4: #include <petscmat.h>
6: typedef struct _n_User *User;
7: struct _n_User {
8: Mat B;
9: };
11: static PetscErrorCode MatGetDiagonal_User(Mat A, Vec X)
12: {
13: User user;
15: PetscFunctionBegin;
16: PetscCall(MatShellGetContext(A, &user));
17: PetscCall(MatGetDiagonal(user->B, X));
18: PetscFunctionReturn(PETSC_SUCCESS);
19: }
21: int main(int argc, char **args)
22: {
23: const PetscScalar xvals[] = {11, 13};
24: const PetscInt inds[] = {0, 1};
25: PetscScalar avals[] = {2, 3, 5, 7};
26: Mat A, S;
27: Vec X, Y;
28: User user;
30: PetscFunctionBeginUser;
31: PetscCall(PetscInitialize(&argc, &args, (char *)0, help));
32: PetscCall(MatCreateSeqAIJ(PETSC_COMM_WORLD, 2, 2, 2, NULL, &A));
33: PetscCall(MatSetUp(A));
34: PetscCall(MatSetValues(A, 2, inds, 2, inds, avals, INSERT_VALUES));
35: PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
36: PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
37: PetscCall(VecCreateSeq(PETSC_COMM_WORLD, 2, &X));
38: PetscCall(VecSetValues(X, 2, inds, xvals, INSERT_VALUES));
39: PetscCall(VecAssemblyBegin(X));
40: PetscCall(VecAssemblyEnd(X));
41: PetscCall(VecDuplicate(X, &Y));
43: PetscCall(PetscNew(&user));
44: user->B = A;
46: PetscCall(MatCreateShell(PETSC_COMM_WORLD, 2, 2, 2, 2, user, &S));
47: PetscCall(MatShellSetOperation(S, MATOP_GET_DIAGONAL, (void (*)(void))MatGetDiagonal_User));
48: PetscCall(MatSetUp(S));
50: PetscCall(MatShift(S, 42));
51: PetscCall(MatGetDiagonal(S, Y));
52: PetscCall(VecView(Y, PETSC_VIEWER_STDOUT_WORLD));
53: PetscCall(MatDiagonalSet(S, X, ADD_VALUES));
54: PetscCall(MatGetDiagonal(S, Y));
55: PetscCall(VecView(Y, PETSC_VIEWER_STDOUT_WORLD));
56: PetscCall(MatScale(S, 42));
57: PetscCall(MatGetDiagonal(S, Y));
58: PetscCall(VecView(Y, PETSC_VIEWER_STDOUT_WORLD));
60: PetscCall(MatDestroy(&A));
61: PetscCall(MatDestroy(&S));
62: PetscCall(VecDestroy(&X));
63: PetscCall(VecDestroy(&Y));
64: PetscCall(PetscFree(user));
65: PetscCall(PetscFinalize());
66: return 0;
67: }
69: /*TEST
71: test:
72: args: -malloc_dump
74: TEST*/