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