1 2 static char help[] = "Tests late MatSetBlockSizes.\n\n"; 3 4 #include <petscmat.h> 5 6 int main(int argc,char **args) 7 { 8 Mat A; 9 Vec x[4]; 10 IS is; 11 ISLocalToGlobalMapping rmap,cmap; 12 PetscInt bs[4],l2gbs[4],rbs,cbs,l2grbs,l2gcbs,i; 13 PetscErrorCode ierr; 14 15 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 16 ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr); 17 ierr = MatSetSizes(A,12,12,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 18 ierr = MatSetType(A,MATAIJ);CHKERRQ(ierr); 19 ierr = ISCreateStride(PETSC_COMM_WORLD,12,0,1,&is);CHKERRQ(ierr); 20 ierr = ISLocalToGlobalMappingCreateIS(is,&rmap);CHKERRQ(ierr); 21 ierr = ISLocalToGlobalMappingSetBlockSize(rmap,2);CHKERRQ(ierr); 22 ierr = ISLocalToGlobalMappingCreateIS(is,&cmap);CHKERRQ(ierr); 23 ierr = ISLocalToGlobalMappingSetBlockSize(cmap,2);CHKERRQ(ierr); 24 25 ierr = MatSetLocalToGlobalMapping(A,rmap,cmap);CHKERRQ(ierr); 26 ierr = ISLocalToGlobalMappingDestroy(&rmap);CHKERRQ(ierr); 27 ierr = ISLocalToGlobalMappingDestroy(&cmap);CHKERRQ(ierr); 28 ierr = ISDestroy(&is);CHKERRQ(ierr); 29 ierr = MatSetUp(A);CHKERRQ(ierr); 30 31 ierr = MatCreateVecs(A,&x[1],&x[0]);CHKERRQ(ierr); 32 ierr = MatSetBlockSizes(A,6,3);CHKERRQ(ierr); 33 ierr = MatCreateVecs(A,&x[3],&x[2]);CHKERRQ(ierr); 34 for (i=0;i<4;i++) { 35 ISLocalToGlobalMapping l2g; 36 37 ierr = VecGetBlockSize(x[i],&bs[i]);CHKERRQ(ierr); 38 ierr = VecGetLocalToGlobalMapping(x[i],&l2g);CHKERRQ(ierr); 39 ierr = ISLocalToGlobalMappingGetBlockSize(l2g,&l2gbs[i]);CHKERRQ(ierr); 40 ierr = VecDestroy(&x[i]);CHKERRQ(ierr); 41 } 42 ierr = MatGetBlockSizes(A,&rbs,&cbs);CHKERRQ(ierr); 43 ierr = MatGetLocalToGlobalMapping(A,&rmap,&cmap);CHKERRQ(ierr); 44 ierr = ISLocalToGlobalMappingGetBlockSize(rmap,&l2grbs);CHKERRQ(ierr); 45 ierr = ISLocalToGlobalMappingGetBlockSize(cmap,&l2gcbs);CHKERRQ(ierr); 46 ierr = MatDestroy(&A);CHKERRQ(ierr); 47 ierr = PetscPrintf(PETSC_COMM_WORLD,"Mat Block sizes: %D %D (l2g %D %D)\n",rbs,cbs,l2grbs,l2gcbs);CHKERRQ(ierr); 48 ierr = PetscPrintf(PETSC_COMM_WORLD,"Vec Block sizes: %D %D (l2g %D %D)\n",bs[0],bs[1],l2gbs[0],l2gbs[1]);CHKERRQ(ierr); 49 ierr = PetscPrintf(PETSC_COMM_WORLD,"Vec Block sizes: %D %D (l2g %D %D)\n",bs[2],bs[3],l2gbs[2],l2gbs[3]);CHKERRQ(ierr); 50 ierr = PetscFinalize(); 51 return ierr; 52 } 53 54 55 56 /*TEST 57 58 test: 59 nsize: 2 60 61 TEST*/ 62