Actual source code: matio.c
petsc-3.13.6 2020-09-29
1: #include <petscviewer.h>
2: #include <petsc/private/matimpl.h>
4: PetscErrorCode MatView_Binary_BlockSizes(Mat mat,PetscViewer viewer)
5: {
6: FILE *info;
7: PetscMPIInt rank;
8: PetscInt rbs,cbs;
12: MatGetBlockSizes(mat,&rbs,&cbs);
13: PetscViewerBinaryGetInfoPointer(viewer,&info);
14: MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);
15: if (!rank && info) {
16: if (rbs != cbs) {PetscFPrintf(PETSC_COMM_SELF,info,"-matload_block_size %D,%D\n",rbs,cbs);}
17: else {PetscFPrintf(PETSC_COMM_SELF,info,"-matload_block_size %D\n",rbs);}
18: }
19: return(0);
20: }
22: PetscErrorCode MatLoad_Binary_BlockSizes(Mat mat,PetscViewer viewer)
23: {
24: PetscInt rbs,cbs,bs[2],n = 2;
25: PetscBool set;
29: /* get current block sizes */
30: MatGetBlockSizes(mat,&rbs,&cbs);
31: bs[0] = rbs; bs[1] = cbs;
32: /* get block sizes from the options database */
33: PetscOptionsBegin(PetscObjectComm((PetscObject)viewer),NULL,"Options for loading matrix block size","Mat");
34: PetscOptionsIntArray("-matload_block_size","Set the block size used to store the matrix","MatLoad",bs,&n,&set);
35: PetscOptionsEnd();
36: if (!set) return(0);
37: if (n == 1) bs[1] = bs[0]; /* to support -matload_block_size <bs> */
38: /* set matrix block sizes */
39: if (bs[0] > 0) rbs = bs[0];
40: if (bs[1] > 0) cbs = bs[1];
41: MatSetBlockSizes(mat,rbs,cbs);
42: return(0);
43: }