1c4762a1bSJed Brown static char help[] = "Tests late MatSetBlockSizes.\n\n";
2c4762a1bSJed Brown
3c4762a1bSJed Brown #include <petscmat.h>
4c4762a1bSJed Brown
main(int argc,char ** args)5d71ae5a4SJacob Faibussowitsch int main(int argc, char **args)
6d71ae5a4SJacob Faibussowitsch {
7c4762a1bSJed Brown Mat A;
8c4762a1bSJed Brown Vec x[4];
9c4762a1bSJed Brown IS is;
10c4762a1bSJed Brown ISLocalToGlobalMapping rmap, cmap;
11c4762a1bSJed Brown PetscInt bs[4], l2gbs[4], rbs, cbs, l2grbs, l2gcbs, i;
12c4762a1bSJed Brown
13327415f7SBarry Smith PetscFunctionBeginUser;
14*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &args, NULL, help));
159566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
169566063dSJacob Faibussowitsch PetscCall(MatSetSizes(A, 12, 12, PETSC_DECIDE, PETSC_DECIDE));
179566063dSJacob Faibussowitsch PetscCall(MatSetType(A, MATAIJ));
189566063dSJacob Faibussowitsch PetscCall(ISCreateStride(PETSC_COMM_WORLD, 12, 0, 1, &is));
199566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingCreateIS(is, &rmap));
209566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingSetBlockSize(rmap, 2));
219566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingCreateIS(is, &cmap));
229566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingSetBlockSize(cmap, 2));
23c4762a1bSJed Brown
249566063dSJacob Faibussowitsch PetscCall(MatSetLocalToGlobalMapping(A, rmap, cmap));
259566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingDestroy(&rmap));
269566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingDestroy(&cmap));
279566063dSJacob Faibussowitsch PetscCall(ISDestroy(&is));
289566063dSJacob Faibussowitsch PetscCall(MatSetUp(A));
29c4762a1bSJed Brown
309566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(A, &x[1], &x[0]));
319566063dSJacob Faibussowitsch PetscCall(MatSetBlockSizes(A, 6, 3));
329566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(A, &x[3], &x[2]));
33c4762a1bSJed Brown for (i = 0; i < 4; i++) {
34c4762a1bSJed Brown ISLocalToGlobalMapping l2g;
35c4762a1bSJed Brown
369566063dSJacob Faibussowitsch PetscCall(VecGetBlockSize(x[i], &bs[i]));
379566063dSJacob Faibussowitsch PetscCall(VecGetLocalToGlobalMapping(x[i], &l2g));
389566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingGetBlockSize(l2g, &l2gbs[i]));
399566063dSJacob Faibussowitsch PetscCall(VecDestroy(&x[i]));
40c4762a1bSJed Brown }
419566063dSJacob Faibussowitsch PetscCall(MatGetBlockSizes(A, &rbs, &cbs));
429566063dSJacob Faibussowitsch PetscCall(MatGetLocalToGlobalMapping(A, &rmap, &cmap));
439566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingGetBlockSize(rmap, &l2grbs));
449566063dSJacob Faibussowitsch PetscCall(ISLocalToGlobalMappingGetBlockSize(cmap, &l2gcbs));
459566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A));
469566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Mat Block sizes: %" PetscInt_FMT " %" PetscInt_FMT " (l2g %" PetscInt_FMT " %" PetscInt_FMT ")\n", rbs, cbs, l2grbs, l2gcbs));
479566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Vec Block sizes: %" PetscInt_FMT " %" PetscInt_FMT " (l2g %" PetscInt_FMT " %" PetscInt_FMT ")\n", bs[0], bs[1], l2gbs[0], l2gbs[1]));
489566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Vec Block sizes: %" PetscInt_FMT " %" PetscInt_FMT " (l2g %" PetscInt_FMT " %" PetscInt_FMT ")\n", bs[2], bs[3], l2gbs[2], l2gbs[3]));
499566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
50b122ec5aSJacob Faibussowitsch return 0;
51c4762a1bSJed Brown }
52c4762a1bSJed Brown
53c4762a1bSJed Brown /*TEST
54c4762a1bSJed Brown
55c4762a1bSJed Brown test:
56c4762a1bSJed Brown nsize: 2
57c4762a1bSJed Brown
58c4762a1bSJed Brown TEST*/
59