1c4762a1bSJed Brown static char help[] = "Tests MatShellTestMult()\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscmat.h> 4c4762a1bSJed Brown 5c4762a1bSJed Brown typedef struct _n_User *User; 6c4762a1bSJed Brown struct _n_User { 7c4762a1bSJed Brown Mat B; 8c4762a1bSJed Brown }; 9c4762a1bSJed Brown 10d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMult_User(Mat A, Vec X, Vec Y) 11d71ae5a4SJacob Faibussowitsch { 12c4762a1bSJed Brown User user; 13c4762a1bSJed Brown 14c4762a1bSJed Brown PetscFunctionBegin; 159566063dSJacob Faibussowitsch PetscCall(MatShellGetContext(A, &user)); 169566063dSJacob Faibussowitsch PetscCall(MatMult(user->B, X, Y)); 173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 18c4762a1bSJed Brown } 19c4762a1bSJed Brown 20d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMultTranspose_User(Mat A, Vec X, Vec Y) 21d71ae5a4SJacob Faibussowitsch { 22c4762a1bSJed Brown User user; 23c4762a1bSJed Brown 24c4762a1bSJed Brown PetscFunctionBegin; 259566063dSJacob Faibussowitsch PetscCall(MatShellGetContext(A, &user)); 269566063dSJacob Faibussowitsch PetscCall(MatMultTranspose(user->B, X, Y)); 273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 28c4762a1bSJed Brown } 29c4762a1bSJed Brown 30*2a8381b2SBarry Smith static PetscErrorCode MyFunction(PetscCtx ctx, Vec x, Vec y) 31d71ae5a4SJacob Faibussowitsch { 32c4762a1bSJed Brown User user = (User)ctx; 33c4762a1bSJed Brown 34c4762a1bSJed Brown PetscFunctionBegin; 359566063dSJacob Faibussowitsch PetscCall(MatMult(user->B, x, y)); 363ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 37c4762a1bSJed Brown } 38c4762a1bSJed Brown 39d71ae5a4SJacob Faibussowitsch int main(int argc, char **args) 40d71ae5a4SJacob Faibussowitsch { 41c4762a1bSJed Brown const PetscInt inds[] = {0, 1}; 42c4762a1bSJed Brown PetscScalar avals[] = {2, 3, 5, 7}; 43c4762a1bSJed Brown Mat S; 44c4762a1bSJed Brown User user; 45c4762a1bSJed Brown Vec base; 46c4762a1bSJed Brown 47327415f7SBarry Smith PetscFunctionBeginUser; 48c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &args, NULL, help)); 499566063dSJacob Faibussowitsch PetscCall(PetscNew(&user)); 509566063dSJacob Faibussowitsch PetscCall(MatCreateSeqAIJ(PETSC_COMM_WORLD, 2, 2, 2, NULL, &user->B)); 519566063dSJacob Faibussowitsch PetscCall(MatSetUp(user->B)); 529566063dSJacob Faibussowitsch PetscCall(MatSetValues(user->B, 2, inds, 2, inds, avals, INSERT_VALUES)); 539566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(user->B, MAT_FINAL_ASSEMBLY)); 549566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(user->B, MAT_FINAL_ASSEMBLY)); 559566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(user->B, &base, NULL)); 569566063dSJacob Faibussowitsch PetscCall(MatCreateShell(PETSC_COMM_WORLD, 2, 2, 2, 2, user, &S)); 579566063dSJacob Faibussowitsch PetscCall(MatSetUp(S)); 5857d50842SBarry Smith PetscCall(MatShellSetOperation(S, MATOP_MULT, (PetscErrorCodeFn *)MatMult_User)); 5957d50842SBarry Smith PetscCall(MatShellSetOperation(S, MATOP_MULT_TRANSPOSE, (PetscErrorCodeFn *)MatMultTranspose_User)); 60c4762a1bSJed Brown 619566063dSJacob Faibussowitsch PetscCall(MatShellTestMult(S, MyFunction, base, user, NULL)); 629566063dSJacob Faibussowitsch PetscCall(MatShellTestMultTranspose(S, MyFunction, base, user, NULL)); 63c4762a1bSJed Brown 649566063dSJacob Faibussowitsch PetscCall(VecDestroy(&base)); 659566063dSJacob Faibussowitsch PetscCall(MatDestroy(&user->B)); 669566063dSJacob Faibussowitsch PetscCall(MatDestroy(&S)); 679566063dSJacob Faibussowitsch PetscCall(PetscFree(user)); 689566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 69b122ec5aSJacob Faibussowitsch return 0; 70c4762a1bSJed Brown } 71c4762a1bSJed Brown 72c4762a1bSJed Brown /*TEST 73c4762a1bSJed Brown 74c4762a1bSJed Brown test: 75c4762a1bSJed Brown args: -mat_shell_test_mult_view -mat_shell_test_mult_transpose_view 76c4762a1bSJed Brown 77c4762a1bSJed Brown TEST*/ 78