1 static char help[] = "Test MatCreateNest with block sizes.\n"; 2 3 #include <petscmat.h> 4 5 int main(int argc, char **argv) 6 { 7 Mat A, B, C, mats[2]; 8 ISLocalToGlobalMapping cmap, rmap; 9 const PetscInt indices[1] = {0}; 10 PetscMPIInt size; 11 PetscErrorCode ierr; 12 13 ierr = PetscInitialize(&argc, &argv, NULL, help); if (ierr) return ierr; 14 ierr = MPI_Comm_size(PETSC_COMM_WORLD, &size);CHKERRMPI(ierr); 15 if (size > 1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_WRONG,"Only coded for one process"); 16 ierr = MatCreate(PETSC_COMM_WORLD, &A);CHKERRQ(ierr); 17 ierr = MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, 1, 2);CHKERRQ(ierr); 18 ierr = MatSetBlockSizes(A, 1, 2);CHKERRQ(ierr); 19 ierr = MatSetType(A,MATAIJ);CHKERRQ(ierr); 20 ierr = ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD, 2, 1, indices,PETSC_COPY_VALUES, &cmap);CHKERRQ(ierr); 21 ierr = ISLocalToGlobalMappingCreate(PETSC_COMM_WORLD, 1, 1, indices,PETSC_COPY_VALUES, &rmap);CHKERRQ(ierr); 22 ierr = MatSetLocalToGlobalMapping(A, rmap, cmap);CHKERRQ(ierr); 23 ierr = MatCreate(PETSC_COMM_WORLD, &B);CHKERRQ(ierr); 24 ierr = MatSetSizes(B, PETSC_DECIDE, PETSC_DECIDE, 1, 1);CHKERRQ(ierr); 25 ierr = MatSetBlockSizes(A, 1, 1);CHKERRQ(ierr); 26 ierr = MatSetType(B,MATAIJ);CHKERRQ(ierr); 27 ierr = MatSetLocalToGlobalMapping(B, rmap, rmap);CHKERRQ(ierr); 28 ierr = ISLocalToGlobalMappingDestroy(&rmap);CHKERRQ(ierr); 29 ierr = ISLocalToGlobalMappingDestroy(&cmap);CHKERRQ(ierr); 30 ierr = MatSetUp(A);CHKERRQ(ierr); 31 ierr = MatSetUp(B);CHKERRQ(ierr); 32 mats[0] = A; 33 mats[1] = B; 34 ierr = MatCreateNest(PETSC_COMM_WORLD, 1, NULL, 2, NULL, mats,&C);CHKERRQ(ierr); 35 ierr = MatSetUp(C);CHKERRQ(ierr); 36 ierr = MatAssemblyBegin(C, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 37 ierr = MatAssemblyEnd(C, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 38 ierr = MatView(C, NULL);CHKERRQ(ierr); 39 ierr = MatDestroy(&C);CHKERRQ(ierr); 40 ierr = MatDestroy(&B);CHKERRQ(ierr); 41 ierr = MatDestroy(&A);CHKERRQ(ierr); 42 ierr = PetscFinalize(); 43 return ierr; 44 } 45 46 /*TEST 47 48 test: 49 50 TEST*/ 51