173b6653fSLawrence Mitchell static char help[] = "Test MatCreateSubMatrix with -mat_type nest and block sizes.\n"; 273b6653fSLawrence Mitchell 373b6653fSLawrence Mitchell #include <petscmat.h> 473b6653fSLawrence Mitchell 573b6653fSLawrence Mitchell int main(int argc, char **argv) 673b6653fSLawrence Mitchell { 773b6653fSLawrence Mitchell Mat A, B, C, mats[6]; 873b6653fSLawrence Mitchell IS rows[2]; 973b6653fSLawrence Mitchell ISLocalToGlobalMapping cmap, rmap; 1073b6653fSLawrence Mitchell const PetscInt indices[3] = {0, 1, 2}; 1173b6653fSLawrence Mitchell PetscInt i; 1273b6653fSLawrence Mitchell PetscMPIInt size; 1373b6653fSLawrence Mitchell 14*b122ec5aSJacob Faibussowitsch CHKERRQ(PetscInitialize(&argc,&argv,NULL,help)); 155f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 162c71b3e2SJacob Faibussowitsch PetscCheckFalse(size > 1,PETSC_COMM_WORLD,PETSC_ERR_ARG_WRONG,"Only coded for one process"); 175f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A)); 185f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,2,1)); 195f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetBlockSizes(A,2,1)); 205f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetType(A,MATAIJ)); 215f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD,2,1,indices,PETSC_COPY_VALUES,&rmap)); 225f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD,1,1,indices,PETSC_COPY_VALUES,&cmap)); 235f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetLocalToGlobalMapping(A,rmap,cmap)); 245f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingDestroy(&rmap)); 255f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingDestroy(&cmap)); 265f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD,1,2,indices,PETSC_COPY_VALUES,&rmap)); 275f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD,1,3,indices,PETSC_COPY_VALUES,&cmap)); 285f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreate(PETSC_COMM_WORLD,&B)); 295f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,2,3)); 305f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetBlockSizes(B,1,1)); 315f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetType(B,MATAIJ)); 325f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetLocalToGlobalMapping(B,rmap,cmap)); 335f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingDestroy(&rmap)); 345f80ce2aSJacob Faibussowitsch CHKERRQ(ISLocalToGlobalMappingDestroy(&cmap)); 355f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetUp(A)); 365f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetUp(B)); 3773b6653fSLawrence Mitchell mats[0] = A; 3873b6653fSLawrence Mitchell mats[1] = B; 3973b6653fSLawrence Mitchell mats[2] = A; 4073b6653fSLawrence Mitchell mats[3] = NULL; 4173b6653fSLawrence Mitchell mats[4] = B; 4273b6653fSLawrence Mitchell mats[5] = A; 435f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateNest(PETSC_COMM_WORLD,2,NULL,3,NULL,mats,&C)); 445f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetUp(C)); 455f80ce2aSJacob Faibussowitsch CHKERRQ(MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY)); 465f80ce2aSJacob Faibussowitsch CHKERRQ(MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY)); 475f80ce2aSJacob Faibussowitsch CHKERRQ(MatView(C,NULL)); 485f80ce2aSJacob Faibussowitsch CHKERRQ(MatNestGetISs(C,rows,NULL)); 4973b6653fSLawrence Mitchell for (i=0; i<2; i++) { 5073b6653fSLawrence Mitchell Mat submat; 5173b6653fSLawrence Mitchell IS cols[3]; 5273b6653fSLawrence Mitchell PetscInt j; 535f80ce2aSJacob Faibussowitsch CHKERRQ(ISView(rows[i],NULL)); 545f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreateSubMatrix(C,rows[i],NULL,MAT_INITIAL_MATRIX,&submat)); 555f80ce2aSJacob Faibussowitsch CHKERRQ(MatView(submat,NULL)); 565f80ce2aSJacob Faibussowitsch CHKERRQ(MatNestGetISs(submat,NULL,cols)); 5773b6653fSLawrence Mitchell for (j=0; j<3; j++) { 585f80ce2aSJacob Faibussowitsch CHKERRQ(ISView(cols[j],NULL)); 5973b6653fSLawrence Mitchell } 605f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&submat)); 6173b6653fSLawrence Mitchell } 625f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&C)); 635f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&B)); 645f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&A)); 65*b122ec5aSJacob Faibussowitsch CHKERRQ(PetscFinalize()); 66*b122ec5aSJacob Faibussowitsch return 0; 6773b6653fSLawrence Mitchell } 6873b6653fSLawrence Mitchell 6973b6653fSLawrence Mitchell /*TEST 7073b6653fSLawrence Mitchell 7173b6653fSLawrence Mitchell test: 7273b6653fSLawrence Mitchell 7373b6653fSLawrence Mitchell TEST*/ 74