xref: /petsc/src/mat/impls/is/matis.c (revision d8d19677bbccf95218448bee62e6b87f4513e133)
1b4319ba4SBarry Smith /*
2b4319ba4SBarry Smith     Creates a matrix class for using the Neumann-Neumann type preconditioners.
3b4319ba4SBarry Smith     This stores the matrices in globally unassembled form. Each processor
4b4319ba4SBarry Smith     assembles only its local Neumann problem and the parallel matrix vector
5b4319ba4SBarry Smith     product is handled "implicitly".
6b4319ba4SBarry Smith 
7b4319ba4SBarry Smith     Currently this allows for only one subdomain per processor.
8b4319ba4SBarry Smith */
9b4319ba4SBarry Smith 
10c6db04a5SJed Brown #include <../src/mat/impls/is/matis.h>      /*I "petscmat.h" I*/
114f2d7cafSStefano Zampini #include <petsc/private/sfimpl.h>
12a72d46e8SStefano Zampini #include <petsc/private/vecimpl.h>
1328f4e0baSStefano Zampini 
14f26d0771SStefano Zampini #define MATIS_MAX_ENTRIES_INSERTION 2048
15b4f971dfSStefano Zampini static PetscErrorCode MatSetValuesLocal_IS(Mat,PetscInt,const PetscInt*,PetscInt,const PetscInt*,const PetscScalar*,InsertMode);
16b4f971dfSStefano Zampini static PetscErrorCode MatSetValuesBlockedLocal_IS(Mat,PetscInt,const PetscInt*,PetscInt,const PetscInt*,const PetscScalar*,InsertMode);
178546b261SStefano Zampini static PetscErrorCode MatISSetUpScatters_Private(Mat);
18f26d0771SStefano Zampini 
1975d48cdbSStefano Zampini static PetscErrorCode MatISContainerDestroyPtAP_Private(void *ptr)
2075d48cdbSStefano Zampini {
2175d48cdbSStefano Zampini   MatISPtAP      ptap = (MatISPtAP)ptr;
2275d48cdbSStefano Zampini   PetscErrorCode ierr;
2375d48cdbSStefano Zampini 
2475d48cdbSStefano Zampini   PetscFunctionBegin;
2575d48cdbSStefano Zampini   ierr = MatDestroySubMatrices(ptap->ris1 ? 2 : 1,&ptap->lP);CHKERRQ(ierr);
2675d48cdbSStefano Zampini   ierr = ISDestroy(&ptap->cis0);CHKERRQ(ierr);
2775d48cdbSStefano Zampini   ierr = ISDestroy(&ptap->cis1);CHKERRQ(ierr);
2875d48cdbSStefano Zampini   ierr = ISDestroy(&ptap->ris0);CHKERRQ(ierr);
2975d48cdbSStefano Zampini   ierr = ISDestroy(&ptap->ris1);CHKERRQ(ierr);
3075d48cdbSStefano Zampini   ierr = PetscFree(ptap);CHKERRQ(ierr);
3175d48cdbSStefano Zampini   PetscFunctionReturn(0);
3275d48cdbSStefano Zampini }
3375d48cdbSStefano Zampini 
3475d48cdbSStefano Zampini static PetscErrorCode MatPtAPNumeric_IS_XAIJ(Mat A, Mat P, Mat C)
3575d48cdbSStefano Zampini {
3675d48cdbSStefano Zampini   MatISPtAP      ptap;
3775d48cdbSStefano Zampini   Mat_IS         *matis = (Mat_IS*)A->data;
3875d48cdbSStefano Zampini   Mat            lA,lC;
3975d48cdbSStefano Zampini   MatReuse       reuse;
4075d48cdbSStefano Zampini   IS             ris[2],cis[2];
4175d48cdbSStefano Zampini   PetscContainer c;
4275d48cdbSStefano Zampini   PetscInt       n;
4375d48cdbSStefano Zampini   PetscErrorCode ierr;
4475d48cdbSStefano Zampini 
4575d48cdbSStefano Zampini   PetscFunctionBegin;
466afe12f5SStefano Zampini   ierr = PetscObjectQuery((PetscObject)C,"_MatIS_PtAP",(PetscObject*)&c);CHKERRQ(ierr);
4775d48cdbSStefano Zampini   if (!c) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_PLIB,"Missing PtAP information");
4875d48cdbSStefano Zampini   ierr   = PetscContainerGetPointer(c,(void**)&ptap);CHKERRQ(ierr);
4975d48cdbSStefano Zampini   ris[0] = ptap->ris0;
5075d48cdbSStefano Zampini   ris[1] = ptap->ris1;
5175d48cdbSStefano Zampini   cis[0] = ptap->cis0;
5275d48cdbSStefano Zampini   cis[1] = ptap->cis1;
5375d48cdbSStefano Zampini   n      = ptap->ris1 ? 2 : 1;
5475d48cdbSStefano Zampini   reuse  = ptap->lP ? MAT_REUSE_MATRIX : MAT_INITIAL_MATRIX;
5575d48cdbSStefano Zampini   ierr   = MatCreateSubMatrices(P,n,ris,cis,reuse,&ptap->lP);CHKERRQ(ierr);
5675d48cdbSStefano Zampini 
5775d48cdbSStefano Zampini   ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr);
5875d48cdbSStefano Zampini   ierr = MatISGetLocalMat(C,&lC);CHKERRQ(ierr);
5975d48cdbSStefano Zampini   if (ptap->ris1) { /* unsymmetric A mapping */
6075d48cdbSStefano Zampini     Mat lPt;
6175d48cdbSStefano Zampini 
6275d48cdbSStefano Zampini     ierr = MatTranspose(ptap->lP[1],MAT_INITIAL_MATRIX,&lPt);CHKERRQ(ierr);
6375d48cdbSStefano Zampini     ierr = MatMatMatMult(lPt,lA,ptap->lP[0],reuse,ptap->fill,&lC);CHKERRQ(ierr);
6475d48cdbSStefano Zampini     if (matis->storel2l) {
6575d48cdbSStefano Zampini       ierr = PetscObjectCompose((PetscObject)(A),"_MatIS_PtAP_l2l",(PetscObject)lPt);CHKERRQ(ierr);
6675d48cdbSStefano Zampini     }
6775d48cdbSStefano Zampini     ierr = MatDestroy(&lPt);CHKERRQ(ierr);
6875d48cdbSStefano Zampini   } else {
6975d48cdbSStefano Zampini     ierr = MatPtAP(lA,ptap->lP[0],reuse,ptap->fill,&lC);CHKERRQ(ierr);
7075d48cdbSStefano Zampini     if (matis->storel2l) {
714222ddf1SHong Zhang      ierr = PetscObjectCompose((PetscObject)C,"_MatIS_PtAP_l2l",(PetscObject)ptap->lP[0]);CHKERRQ(ierr);
7275d48cdbSStefano Zampini     }
7375d48cdbSStefano Zampini   }
7475d48cdbSStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
7575d48cdbSStefano Zampini     ierr = MatISSetLocalMat(C,lC);CHKERRQ(ierr);
7675d48cdbSStefano Zampini     ierr = MatDestroy(&lC);CHKERRQ(ierr);
7775d48cdbSStefano Zampini   }
7875d48cdbSStefano Zampini   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7975d48cdbSStefano Zampini   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
8075d48cdbSStefano Zampini   PetscFunctionReturn(0);
8175d48cdbSStefano Zampini }
8275d48cdbSStefano Zampini 
8375d48cdbSStefano Zampini static PetscErrorCode MatGetNonzeroColumnsLocal_Private(Mat PT,IS *cis)
8475d48cdbSStefano Zampini {
8575d48cdbSStefano Zampini   Mat            Po,Pd;
8675d48cdbSStefano Zampini   IS             zd,zo;
8775d48cdbSStefano Zampini   const PetscInt *garray;
8875d48cdbSStefano Zampini   PetscInt       *aux,i,bs;
8975d48cdbSStefano Zampini   PetscInt       dc,stc,oc,ctd,cto;
9075d48cdbSStefano Zampini   PetscBool      ismpiaij,ismpibaij,isseqaij,isseqbaij;
9175d48cdbSStefano Zampini   MPI_Comm       comm;
9275d48cdbSStefano Zampini   PetscErrorCode ierr;
9375d48cdbSStefano Zampini 
9475d48cdbSStefano Zampini   PetscFunctionBegin;
9575d48cdbSStefano Zampini   PetscValidHeaderSpecific(PT,MAT_CLASSID,1);
9675d48cdbSStefano Zampini   PetscValidPointer(cis,2);
9775d48cdbSStefano Zampini   ierr = PetscObjectGetComm((PetscObject)PT,&comm);CHKERRQ(ierr);
9875d48cdbSStefano Zampini   bs   = 1;
99b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)PT,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr);
100b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)PT,MATMPIBAIJ,&ismpibaij);CHKERRQ(ierr);
10104637862SRichard Tran Mills   ierr = PetscObjectBaseTypeCompare((PetscObject)PT,MATSEQAIJ,&isseqaij);CHKERRQ(ierr);
10275d48cdbSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)PT,MATSEQBAIJ,&isseqbaij);CHKERRQ(ierr);
10375d48cdbSStefano Zampini   if (isseqaij || isseqbaij) {
10475d48cdbSStefano Zampini     Pd = PT;
10575d48cdbSStefano Zampini     Po = NULL;
10675d48cdbSStefano Zampini     garray = NULL;
10775d48cdbSStefano Zampini   } else if (ismpiaij) {
10875d48cdbSStefano Zampini     ierr = MatMPIAIJGetSeqAIJ(PT,&Pd,&Po,&garray);CHKERRQ(ierr);
10975d48cdbSStefano Zampini   } else if (ismpibaij) {
11075d48cdbSStefano Zampini     ierr = MatMPIBAIJGetSeqBAIJ(PT,&Pd,&Po,&garray);CHKERRQ(ierr);
11175d48cdbSStefano Zampini     ierr = MatGetBlockSize(PT,&bs);CHKERRQ(ierr);
11275d48cdbSStefano Zampini   } else SETERRQ1(comm,PETSC_ERR_SUP,"Not for matrix type %s",((PetscObject)(PT))->type_name);
11375d48cdbSStefano Zampini 
11475d48cdbSStefano Zampini   /* identify any null columns in Pd or Po */
11522f7620eSStefano Zampini   /* We use a tolerance comparison since it may happen that, with geometric multigrid,
11622f7620eSStefano Zampini      some of the columns are not really zero, but very close to */
11775d48cdbSStefano Zampini   zo = zd = NULL;
11875d48cdbSStefano Zampini   if (Po) {
11922f7620eSStefano Zampini     ierr = MatFindNonzeroRowsOrCols_Basic(Po,PETSC_TRUE,PETSC_SMALL,&zo);CHKERRQ(ierr);
12075d48cdbSStefano Zampini   }
12122f7620eSStefano Zampini   ierr = MatFindNonzeroRowsOrCols_Basic(Pd,PETSC_TRUE,PETSC_SMALL,&zd);CHKERRQ(ierr);
12275d48cdbSStefano Zampini 
12375d48cdbSStefano Zampini   ierr = MatGetLocalSize(PT,NULL,&dc);CHKERRQ(ierr);
12475d48cdbSStefano Zampini   ierr = MatGetOwnershipRangeColumn(PT,&stc,NULL);CHKERRQ(ierr);
12575d48cdbSStefano Zampini   if (Po) { ierr = MatGetLocalSize(Po,NULL,&oc);CHKERRQ(ierr); }
12675d48cdbSStefano Zampini   else oc = 0;
12775d48cdbSStefano Zampini   ierr = PetscMalloc1((dc+oc)/bs,&aux);CHKERRQ(ierr);
12875d48cdbSStefano Zampini   if (zd) {
12975d48cdbSStefano Zampini     const PetscInt *idxs;
13075d48cdbSStefano Zampini     PetscInt       nz;
13175d48cdbSStefano Zampini 
13275d48cdbSStefano Zampini     /* this will throw an error if bs is not valid */
13375d48cdbSStefano Zampini     ierr = ISSetBlockSize(zd,bs);CHKERRQ(ierr);
13475d48cdbSStefano Zampini     ierr = ISGetLocalSize(zd,&nz);CHKERRQ(ierr);
13575d48cdbSStefano Zampini     ierr = ISGetIndices(zd,&idxs);CHKERRQ(ierr);
13675d48cdbSStefano Zampini     ctd  = nz/bs;
13775d48cdbSStefano Zampini     for (i=0; i<ctd; i++) aux[i] = (idxs[bs*i]+stc)/bs;
13875d48cdbSStefano Zampini     ierr = ISRestoreIndices(zd,&idxs);CHKERRQ(ierr);
13975d48cdbSStefano Zampini   } else {
14075d48cdbSStefano Zampini     ctd = dc/bs;
14175d48cdbSStefano Zampini     for (i=0; i<ctd; i++) aux[i] = i+stc/bs;
14275d48cdbSStefano Zampini   }
14375d48cdbSStefano Zampini   if (zo) {
14475d48cdbSStefano Zampini     const PetscInt *idxs;
14575d48cdbSStefano Zampini     PetscInt       nz;
14675d48cdbSStefano Zampini 
14775d48cdbSStefano Zampini     /* this will throw an error if bs is not valid */
14875d48cdbSStefano Zampini     ierr = ISSetBlockSize(zo,bs);CHKERRQ(ierr);
14975d48cdbSStefano Zampini     ierr = ISGetLocalSize(zo,&nz);CHKERRQ(ierr);
15075d48cdbSStefano Zampini     ierr = ISGetIndices(zo,&idxs);CHKERRQ(ierr);
15175d48cdbSStefano Zampini     cto  = nz/bs;
15275d48cdbSStefano Zampini     for (i=0; i<cto; i++) aux[i+ctd] = garray[idxs[bs*i]/bs];
15375d48cdbSStefano Zampini     ierr = ISRestoreIndices(zo,&idxs);CHKERRQ(ierr);
15475d48cdbSStefano Zampini   } else {
15575d48cdbSStefano Zampini     cto = oc/bs;
15675d48cdbSStefano Zampini     for (i=0; i<cto; i++) aux[i+ctd] = garray[i];
15775d48cdbSStefano Zampini   }
15875d48cdbSStefano Zampini   ierr = ISCreateBlock(comm,bs,ctd+cto,aux,PETSC_OWN_POINTER,cis);CHKERRQ(ierr);
15975d48cdbSStefano Zampini   ierr = ISDestroy(&zd);CHKERRQ(ierr);
16075d48cdbSStefano Zampini   ierr = ISDestroy(&zo);CHKERRQ(ierr);
16175d48cdbSStefano Zampini   PetscFunctionReturn(0);
16275d48cdbSStefano Zampini }
16375d48cdbSStefano Zampini 
1644222ddf1SHong Zhang static PetscErrorCode MatPtAPSymbolic_IS_XAIJ(Mat A,Mat P,PetscReal fill,Mat C)
16575d48cdbSStefano Zampini {
1668546b261SStefano Zampini   Mat                    PT,lA;
16775d48cdbSStefano Zampini   MatISPtAP              ptap;
16875d48cdbSStefano Zampini   ISLocalToGlobalMapping Crl2g,Ccl2g,rl2g,cl2g;
16975d48cdbSStefano Zampini   PetscContainer         c;
1708546b261SStefano Zampini   MatType                lmtype;
17175d48cdbSStefano Zampini   const PetscInt         *garray;
17275d48cdbSStefano Zampini   PetscInt               ibs,N,dc;
17375d48cdbSStefano Zampini   MPI_Comm               comm;
17475d48cdbSStefano Zampini   PetscErrorCode         ierr;
17575d48cdbSStefano Zampini 
17675d48cdbSStefano Zampini   PetscFunctionBegin;
17775d48cdbSStefano Zampini   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
1784222ddf1SHong Zhang   ierr = MatSetType(C,MATIS);CHKERRQ(ierr);
1798546b261SStefano Zampini   ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr);
1808546b261SStefano Zampini   ierr = MatGetType(lA,&lmtype);CHKERRQ(ierr);
1814222ddf1SHong Zhang   ierr = MatISSetLocalMatType(C,lmtype);CHKERRQ(ierr);
18275d48cdbSStefano Zampini   ierr = MatGetSize(P,NULL,&N);CHKERRQ(ierr);
18375d48cdbSStefano Zampini   ierr = MatGetLocalSize(P,NULL,&dc);CHKERRQ(ierr);
1844222ddf1SHong Zhang   ierr = MatSetSizes(C,dc,dc,N,N);CHKERRQ(ierr);
18575d48cdbSStefano Zampini /* Not sure about this
18675d48cdbSStefano Zampini   ierr = MatGetBlockSizes(P,NULL,&ibs);CHKERRQ(ierr);
18775d48cdbSStefano Zampini   ierr = MatSetBlockSize(*C,ibs);CHKERRQ(ierr);
18875d48cdbSStefano Zampini */
18975d48cdbSStefano Zampini 
19075d48cdbSStefano Zampini   ierr = PetscNew(&ptap);CHKERRQ(ierr);
19175d48cdbSStefano Zampini   ierr = PetscContainerCreate(PETSC_COMM_SELF,&c);CHKERRQ(ierr);
19275d48cdbSStefano Zampini   ierr = PetscContainerSetPointer(c,ptap);CHKERRQ(ierr);
19375d48cdbSStefano Zampini   ierr = PetscContainerSetUserDestroy(c,MatISContainerDestroyPtAP_Private);CHKERRQ(ierr);
1944222ddf1SHong Zhang   ierr = PetscObjectCompose((PetscObject)C,"_MatIS_PtAP",(PetscObject)c);CHKERRQ(ierr);
19575d48cdbSStefano Zampini   ierr = PetscContainerDestroy(&c);CHKERRQ(ierr);
19675d48cdbSStefano Zampini   ptap->fill = fill;
19775d48cdbSStefano Zampini 
19875d48cdbSStefano Zampini   ierr = MatGetLocalToGlobalMapping(A,&rl2g,&cl2g);CHKERRQ(ierr);
19975d48cdbSStefano Zampini 
20075d48cdbSStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(cl2g,&ibs);CHKERRQ(ierr);
20175d48cdbSStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(cl2g,&N);CHKERRQ(ierr);
20275d48cdbSStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockIndices(cl2g,&garray);CHKERRQ(ierr);
20375d48cdbSStefano Zampini   ierr = ISCreateBlock(comm,ibs,N/ibs,garray,PETSC_COPY_VALUES,&ptap->ris0);CHKERRQ(ierr);
20475d48cdbSStefano Zampini   ierr = ISLocalToGlobalMappingRestoreBlockIndices(cl2g,&garray);CHKERRQ(ierr);
20575d48cdbSStefano Zampini 
20675d48cdbSStefano Zampini   ierr = MatCreateSubMatrix(P,ptap->ris0,NULL,MAT_INITIAL_MATRIX,&PT);CHKERRQ(ierr);
20775d48cdbSStefano Zampini   ierr = MatGetNonzeroColumnsLocal_Private(PT,&ptap->cis0);CHKERRQ(ierr);
20875d48cdbSStefano Zampini   ierr = ISLocalToGlobalMappingCreateIS(ptap->cis0,&Ccl2g);CHKERRQ(ierr);
20975d48cdbSStefano Zampini   ierr = MatDestroy(&PT);CHKERRQ(ierr);
21075d48cdbSStefano Zampini 
21175d48cdbSStefano Zampini   Crl2g = NULL;
21275d48cdbSStefano Zampini   if (rl2g != cl2g) { /* unsymmetric A mapping */
21375d48cdbSStefano Zampini     PetscBool same,lsame = PETSC_FALSE;
21475d48cdbSStefano Zampini     PetscInt  N1,ibs1;
21575d48cdbSStefano Zampini 
21675d48cdbSStefano Zampini     ierr = ISLocalToGlobalMappingGetSize(rl2g,&N1);CHKERRQ(ierr);
21775d48cdbSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockSize(rl2g,&ibs1);CHKERRQ(ierr);
21875d48cdbSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockIndices(rl2g,&garray);CHKERRQ(ierr);
21975d48cdbSStefano Zampini     ierr = ISCreateBlock(comm,ibs,N/ibs,garray,PETSC_COPY_VALUES,&ptap->ris1);CHKERRQ(ierr);
22075d48cdbSStefano Zampini     ierr = ISLocalToGlobalMappingRestoreBlockIndices(rl2g,&garray);CHKERRQ(ierr);
22175d48cdbSStefano Zampini     if (ibs1 == ibs && N1 == N) { /* check if the l2gmaps are the same */
22275d48cdbSStefano Zampini       const PetscInt *i1,*i2;
22375d48cdbSStefano Zampini 
22475d48cdbSStefano Zampini       ierr = ISBlockGetIndices(ptap->ris0,&i1);CHKERRQ(ierr);
22575d48cdbSStefano Zampini       ierr = ISBlockGetIndices(ptap->ris1,&i2);CHKERRQ(ierr);
226580bdb30SBarry Smith       ierr = PetscArraycmp(i1,i2,N,&lsame);CHKERRQ(ierr);
22775d48cdbSStefano Zampini     }
228820f2d46SBarry Smith     ierr = MPIU_Allreduce(&lsame,&same,1,MPIU_BOOL,MPI_LAND,comm);CHKERRMPI(ierr);
22975d48cdbSStefano Zampini     if (same) {
23075d48cdbSStefano Zampini       ierr = ISDestroy(&ptap->ris1);CHKERRQ(ierr);
23175d48cdbSStefano Zampini     } else {
23275d48cdbSStefano Zampini       ierr = MatCreateSubMatrix(P,ptap->ris1,NULL,MAT_INITIAL_MATRIX,&PT);CHKERRQ(ierr);
23375d48cdbSStefano Zampini       ierr = MatGetNonzeroColumnsLocal_Private(PT,&ptap->cis1);CHKERRQ(ierr);
23475d48cdbSStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(ptap->cis1,&Crl2g);CHKERRQ(ierr);
23575d48cdbSStefano Zampini       ierr = MatDestroy(&PT);CHKERRQ(ierr);
23675d48cdbSStefano Zampini     }
23775d48cdbSStefano Zampini   }
23875d48cdbSStefano Zampini /* Not sure about this
23975d48cdbSStefano Zampini   if (!Crl2g) {
2404222ddf1SHong Zhang     ierr = MatGetBlockSize(C,&ibs);CHKERRQ(ierr);
24175d48cdbSStefano Zampini     ierr = ISLocalToGlobalMappingSetBlockSize(Ccl2g,ibs);CHKERRQ(ierr);
24275d48cdbSStefano Zampini   }
24375d48cdbSStefano Zampini */
2444222ddf1SHong Zhang   ierr = MatSetLocalToGlobalMapping(C,Crl2g ? Crl2g : Ccl2g,Ccl2g);CHKERRQ(ierr);
24575d48cdbSStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&Crl2g);CHKERRQ(ierr);
24675d48cdbSStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&Ccl2g);CHKERRQ(ierr);
24775d48cdbSStefano Zampini 
2484222ddf1SHong Zhang   C->ops->ptapnumeric = MatPtAPNumeric_IS_XAIJ;
24975d48cdbSStefano Zampini   PetscFunctionReturn(0);
25075d48cdbSStefano Zampini }
25175d48cdbSStefano Zampini 
2524222ddf1SHong Zhang /* ----------------------------------------- */
2534222ddf1SHong Zhang static PetscErrorCode MatProductSymbolic_PtAP_IS_XAIJ(Mat C)
25475d48cdbSStefano Zampini {
25575d48cdbSStefano Zampini   PetscErrorCode ierr;
2564222ddf1SHong Zhang   Mat_Product    *product = C->product;
2574222ddf1SHong Zhang   Mat            A=product->A,P=product->B;
2584222ddf1SHong Zhang   PetscReal      fill=product->fill;
25975d48cdbSStefano Zampini 
26075d48cdbSStefano Zampini   PetscFunctionBegin;
26175d48cdbSStefano Zampini   ierr = MatPtAPSymbolic_IS_XAIJ(A,P,fill,C);CHKERRQ(ierr);
2624222ddf1SHong Zhang   C->ops->productnumeric = MatProductNumeric_PtAP;
26375d48cdbSStefano Zampini   PetscFunctionReturn(0);
26475d48cdbSStefano Zampini }
26575d48cdbSStefano Zampini 
2664222ddf1SHong Zhang static PetscErrorCode MatProductSetFromOptions_IS_XAIJ_PtAP(Mat C)
2674222ddf1SHong Zhang {
2684222ddf1SHong Zhang   PetscFunctionBegin;
2694222ddf1SHong Zhang   C->ops->productsymbolic = MatProductSymbolic_PtAP_IS_XAIJ;
2704222ddf1SHong Zhang   PetscFunctionReturn(0);
2714222ddf1SHong Zhang }
2724222ddf1SHong Zhang 
2734222ddf1SHong Zhang PETSC_INTERN PetscErrorCode MatProductSetFromOptions_IS_XAIJ(Mat C)
2744222ddf1SHong Zhang {
2754222ddf1SHong Zhang   PetscErrorCode ierr;
2764222ddf1SHong Zhang   Mat_Product    *product = C->product;
2774222ddf1SHong Zhang 
2784222ddf1SHong Zhang   PetscFunctionBegin;
2794222ddf1SHong Zhang   if (product->type == MATPRODUCT_PtAP) {
2804222ddf1SHong Zhang     ierr = MatProductSetFromOptions_IS_XAIJ_PtAP(C);CHKERRQ(ierr);
2816718818eSStefano Zampini   }
2824222ddf1SHong Zhang   PetscFunctionReturn(0);
2834222ddf1SHong Zhang }
2844222ddf1SHong Zhang 
2854222ddf1SHong Zhang /* ----------------------------------------- */
2865b003df0Sstefano_zampini static PetscErrorCode MatISContainerDestroyFields_Private(void *ptr)
2875b003df0Sstefano_zampini {
2885b003df0Sstefano_zampini   MatISLocalFields lf = (MatISLocalFields)ptr;
2895b003df0Sstefano_zampini   PetscInt         i;
2905b003df0Sstefano_zampini   PetscErrorCode   ierr;
2915b003df0Sstefano_zampini 
292ab4d48faSStefano Zampini   PetscFunctionBegin;
2935b003df0Sstefano_zampini   for (i=0;i<lf->nr;i++) {
2945b003df0Sstefano_zampini     ierr = ISDestroy(&lf->rf[i]);CHKERRQ(ierr);
2955b003df0Sstefano_zampini   }
2965b003df0Sstefano_zampini   for (i=0;i<lf->nc;i++) {
2975b003df0Sstefano_zampini     ierr = ISDestroy(&lf->cf[i]);CHKERRQ(ierr);
2985b003df0Sstefano_zampini   }
2995b003df0Sstefano_zampini   ierr = PetscFree2(lf->rf,lf->cf);CHKERRQ(ierr);
3005b003df0Sstefano_zampini   ierr = PetscFree(lf);CHKERRQ(ierr);
3015b003df0Sstefano_zampini   PetscFunctionReturn(0);
3025b003df0Sstefano_zampini }
303a72627d2SStefano Zampini 
304c9225affSStefano Zampini static PetscErrorCode MatConvert_SeqXAIJ_IS(Mat A,MatType type,MatReuse reuse,Mat *newmat)
3056989cf23SStefano Zampini {
306c9225affSStefano Zampini   Mat            B,lB;
307c9225affSStefano Zampini   PetscErrorCode ierr;
308c9225affSStefano Zampini 
309c9225affSStefano Zampini   PetscFunctionBegin;
310c9225affSStefano Zampini   if (reuse != MAT_REUSE_MATRIX) {
311c9225affSStefano Zampini     ISLocalToGlobalMapping rl2g,cl2g;
312c9225affSStefano Zampini     PetscInt               bs;
313c9225affSStefano Zampini     IS                     is;
314c9225affSStefano Zampini 
315c9225affSStefano Zampini     ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
316c9225affSStefano Zampini     ierr = ISCreateStride(PetscObjectComm((PetscObject)A),A->rmap->n/bs,0,1,&is);CHKERRQ(ierr);
317c9225affSStefano Zampini     if (bs > 1) {
318c9225affSStefano Zampini       IS       is2;
319c9225affSStefano Zampini       PetscInt i,*aux;
320c9225affSStefano Zampini 
321c9225affSStefano Zampini       ierr = ISGetLocalSize(is,&i);CHKERRQ(ierr);
322c9225affSStefano Zampini       ierr = ISGetIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr);
323c9225affSStefano Zampini       ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),bs,i,aux,PETSC_COPY_VALUES,&is2);CHKERRQ(ierr);
324c9225affSStefano Zampini       ierr = ISRestoreIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr);
325c9225affSStefano Zampini       ierr = ISDestroy(&is);CHKERRQ(ierr);
326c9225affSStefano Zampini       is   = is2;
327c9225affSStefano Zampini     }
328c9225affSStefano Zampini     ierr = ISSetIdentity(is);CHKERRQ(ierr);
329c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr);
330c9225affSStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
331c9225affSStefano Zampini     ierr = ISCreateStride(PetscObjectComm((PetscObject)A),A->cmap->n/bs,0,1,&is);CHKERRQ(ierr);
332c9225affSStefano Zampini     if (bs > 1) {
333c9225affSStefano Zampini       IS       is2;
334c9225affSStefano Zampini       PetscInt i,*aux;
335c9225affSStefano Zampini 
336c9225affSStefano Zampini       ierr = ISGetLocalSize(is,&i);CHKERRQ(ierr);
337c9225affSStefano Zampini       ierr = ISGetIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr);
338c9225affSStefano Zampini       ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),bs,i,aux,PETSC_COPY_VALUES,&is2);CHKERRQ(ierr);
339c9225affSStefano Zampini       ierr = ISRestoreIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr);
340c9225affSStefano Zampini       ierr = ISDestroy(&is);CHKERRQ(ierr);
341c9225affSStefano Zampini       is   = is2;
342c9225affSStefano Zampini     }
343c9225affSStefano Zampini     ierr = ISSetIdentity(is);CHKERRQ(ierr);
344c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr);
345c9225affSStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
346c9225affSStefano Zampini     ierr = MatCreateIS(PetscObjectComm((PetscObject)A),bs,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N,rl2g,cl2g,&B);CHKERRQ(ierr);
347c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
348c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
349c9225affSStefano Zampini     ierr = MatDuplicate(A,MAT_COPY_VALUES,&lB);CHKERRQ(ierr);
350c9225affSStefano Zampini     if (reuse == MAT_INITIAL_MATRIX) *newmat = B;
351c9225affSStefano Zampini   } else {
352c9225affSStefano Zampini     B    = *newmat;
353c9225affSStefano Zampini     ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
354c9225affSStefano Zampini     lB   = A;
355c9225affSStefano Zampini   }
356c9225affSStefano Zampini   ierr = MatISSetLocalMat(B,lB);CHKERRQ(ierr);
357c9225affSStefano Zampini   ierr = MatDestroy(&lB);CHKERRQ(ierr);
358c9225affSStefano Zampini   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
359c9225affSStefano Zampini   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
360c9225affSStefano Zampini   if (reuse == MAT_INPLACE_MATRIX) {
361c9225affSStefano Zampini     ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr);
362c9225affSStefano Zampini   }
363c9225affSStefano Zampini   PetscFunctionReturn(0);
364c9225affSStefano Zampini }
365c9225affSStefano Zampini 
366c9225affSStefano Zampini static PetscErrorCode MatISScaleDisassembling_Private(Mat A)
367c9225affSStefano Zampini {
368c9225affSStefano Zampini   Mat_IS         *matis = (Mat_IS*)(A->data);
369c9225affSStefano Zampini   PetscScalar    *aa;
370c9225affSStefano Zampini   const PetscInt *ii,*jj;
371c9225affSStefano Zampini   PetscInt       i,n,m;
372fabe8965SStefano Zampini   PetscInt       *ecount,**eneighs;
373c9225affSStefano Zampini   PetscBool      flg;
374c9225affSStefano Zampini   PetscErrorCode ierr;
375c9225affSStefano Zampini 
376c9225affSStefano Zampini   PetscFunctionBegin;
377c9225affSStefano Zampini   ierr = MatGetRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&m,&ii,&jj,&flg);CHKERRQ(ierr);
378c9225affSStefano Zampini   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure");
379fabe8965SStefano Zampini   ierr = ISLocalToGlobalMappingGetNodeInfo(A->rmap->mapping,&n,&ecount,&eneighs);CHKERRQ(ierr);
380c9225affSStefano Zampini   if (m != n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected %D != %D",m,n);
381c9225affSStefano Zampini   ierr = MatSeqAIJGetArray(matis->A,&aa);CHKERRQ(ierr);
382c9225affSStefano Zampini   for (i=0;i<n;i++) {
383fabe8965SStefano Zampini     if (ecount[i] > 1) {
384c9225affSStefano Zampini       PetscInt j;
385c9225affSStefano Zampini 
386c9225affSStefano Zampini       for (j=ii[i];j<ii[i+1];j++) {
387c9225affSStefano Zampini         PetscInt    i2 = jj[j],p,p2;
388fabe8965SStefano Zampini         PetscReal   scal = 0.0;
389c9225affSStefano Zampini 
390c9225affSStefano Zampini         for (p=0;p<ecount[i];p++) {
391c9225affSStefano Zampini           for (p2=0;p2<ecount[i2];p2++) {
392c9225affSStefano Zampini             if (eneighs[i][p] == eneighs[i2][p2]) { scal += 1.0; break; }
393c9225affSStefano Zampini           }
394c9225affSStefano Zampini         }
395fabe8965SStefano Zampini         if (scal) aa[j] /= scal;
396c9225affSStefano Zampini       }
397c9225affSStefano Zampini     }
398c9225affSStefano Zampini   }
399fabe8965SStefano Zampini   ierr = ISLocalToGlobalMappingRestoreNodeInfo(A->rmap->mapping,&n,&ecount,&eneighs);CHKERRQ(ierr);
400c9225affSStefano Zampini   ierr = MatSeqAIJRestoreArray(matis->A,&aa);CHKERRQ(ierr);
401c9225affSStefano Zampini   ierr = MatRestoreRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&m,&ii,&jj,&flg);CHKERRQ(ierr);
402c9225affSStefano Zampini   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot restore IJ structure");
403c9225affSStefano Zampini   PetscFunctionReturn(0);
404c9225affSStefano Zampini }
405c9225affSStefano Zampini 
406fabe8965SStefano Zampini typedef enum {MAT_IS_DISASSEMBLE_L2G_NATURAL,MAT_IS_DISASSEMBLE_L2G_MAT, MAT_IS_DISASSEMBLE_L2G_ND} MatISDisassemblel2gType;
407fabe8965SStefano Zampini 
408c9225affSStefano Zampini static PetscErrorCode MatMPIXAIJComputeLocalToGlobalMapping_Private(Mat A, ISLocalToGlobalMapping *l2g)
409c9225affSStefano Zampini {
410fabe8965SStefano Zampini   Mat                     Ad,Ao;
411fabe8965SStefano Zampini   IS                      is,ndmap,ndsub;
412c9225affSStefano Zampini   MPI_Comm                comm;
413fabe8965SStefano Zampini   const PetscInt          *garray,*ndmapi;
414fabe8965SStefano Zampini   PetscInt                bs,i,cnt,nl,*ncount,*ndmapc;
415fabe8965SStefano Zampini   PetscBool               ismpiaij,ismpibaij;
416f4259b30SLisandro Dalcin   const char *const       MatISDisassemblel2gTypes[] = {"NATURAL","MAT","ND","MatISDisassemblel2gType","MAT_IS_DISASSEMBLE_L2G_",NULL};
417fabe8965SStefano Zampini   MatISDisassemblel2gType mode = MAT_IS_DISASSEMBLE_L2G_NATURAL;
418fabe8965SStefano Zampini   MatPartitioning         part;
419fabe8965SStefano Zampini   PetscSF                 sf;
420c9225affSStefano Zampini   PetscErrorCode          ierr;
421c9225affSStefano Zampini 
422c9225affSStefano Zampini   PetscFunctionBegin;
423fabe8965SStefano Zampini   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"MatIS l2g disassembling options","Mat");CHKERRQ(ierr);
424fabe8965SStefano Zampini   ierr = PetscOptionsEnum("-mat_is_disassemble_l2g_type","Type of local-to-global mapping to be used for disassembling","MatISDisassemblel2gType",MatISDisassemblel2gTypes,(PetscEnum)mode,(PetscEnum*)&mode,NULL);CHKERRQ(ierr);
425fabe8965SStefano Zampini   ierr = PetscOptionsEnd();CHKERRQ(ierr);
426fabe8965SStefano Zampini   if (mode == MAT_IS_DISASSEMBLE_L2G_MAT) {
427c9225affSStefano Zampini     ierr = MatGetLocalToGlobalMapping(A,l2g,NULL);CHKERRQ(ierr);
428c9225affSStefano Zampini     PetscFunctionReturn(0);
429c9225affSStefano Zampini   }
430c9225affSStefano Zampini   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
431b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIAIJ ,&ismpiaij);CHKERRQ(ierr);
432b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIBAIJ,&ismpibaij);CHKERRQ(ierr);
433c9225affSStefano Zampini   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
434fabe8965SStefano Zampini   switch (mode) {
435fabe8965SStefano Zampini   case MAT_IS_DISASSEMBLE_L2G_ND:
436fabe8965SStefano Zampini     ierr = MatPartitioningCreate(comm,&part);CHKERRQ(ierr);
437fabe8965SStefano Zampini     ierr = MatPartitioningSetAdjacency(part,A);CHKERRQ(ierr);
438fabe8965SStefano Zampini     ierr = PetscObjectSetOptionsPrefix((PetscObject)part,((PetscObject)A)->prefix);CHKERRQ(ierr);
439fabe8965SStefano Zampini     ierr = MatPartitioningSetFromOptions(part);CHKERRQ(ierr);
440fabe8965SStefano Zampini     ierr = MatPartitioningApplyND(part,&ndmap);CHKERRQ(ierr);
441fabe8965SStefano Zampini     ierr = MatPartitioningDestroy(&part);CHKERRQ(ierr);
442fabe8965SStefano Zampini     ierr = ISBuildTwoSided(ndmap,NULL,&ndsub);CHKERRQ(ierr);
443fabe8965SStefano Zampini     ierr = MatMPIAIJSetUseScalableIncreaseOverlap(A,PETSC_TRUE);CHKERRQ(ierr);
444fabe8965SStefano Zampini     ierr = MatIncreaseOverlap(A,1,&ndsub,1);CHKERRQ(ierr);
445fabe8965SStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(ndsub,l2g);CHKERRQ(ierr);
446fabe8965SStefano Zampini 
447fabe8965SStefano Zampini     /* it may happen that a separator node is not properly shared */
448fabe8965SStefano Zampini     ierr = ISLocalToGlobalMappingGetNodeInfo(*l2g,&nl,&ncount,NULL);CHKERRQ(ierr);
449fabe8965SStefano Zampini     ierr = PetscSFCreate(comm,&sf);CHKERRQ(ierr);
450fabe8965SStefano Zampini     ierr = ISLocalToGlobalMappingGetIndices(*l2g,&garray);CHKERRQ(ierr);
451fabe8965SStefano Zampini     ierr = PetscSFSetGraphLayout(sf,A->rmap,nl,NULL,PETSC_OWN_POINTER,garray);CHKERRQ(ierr);
452fabe8965SStefano Zampini     ierr = ISLocalToGlobalMappingRestoreIndices(*l2g,&garray);CHKERRQ(ierr);
453fabe8965SStefano Zampini     ierr = PetscCalloc1(A->rmap->n,&ndmapc);CHKERRQ(ierr);
45483df288dSJunchao Zhang     ierr = PetscSFReduceBegin(sf,MPIU_INT,ncount,ndmapc,MPI_REPLACE);CHKERRQ(ierr);
45583df288dSJunchao Zhang     ierr = PetscSFReduceEnd(sf,MPIU_INT,ncount,ndmapc,MPI_REPLACE);CHKERRQ(ierr);
456fabe8965SStefano Zampini     ierr = ISLocalToGlobalMappingRestoreNodeInfo(*l2g,NULL,&ncount,NULL);CHKERRQ(ierr);
457fabe8965SStefano Zampini     ierr = ISGetIndices(ndmap,&ndmapi);CHKERRQ(ierr);
458fabe8965SStefano Zampini     for (i = 0, cnt = 0; i < A->rmap->n; i++)
459fabe8965SStefano Zampini       if (ndmapi[i] < 0 && ndmapc[i] < 2)
460fabe8965SStefano Zampini         cnt++;
461fabe8965SStefano Zampini 
462820f2d46SBarry Smith     ierr = MPIU_Allreduce(&cnt,&i,1,MPIU_INT,MPI_MAX,comm);CHKERRMPI(ierr);
463fabe8965SStefano Zampini     if (i) { /* we detected isolated separator nodes */
464fabe8965SStefano Zampini       Mat                    A2,A3;
465fabe8965SStefano Zampini       IS                     *workis,is2;
466fabe8965SStefano Zampini       PetscScalar            *vals;
467fabe8965SStefano Zampini       PetscInt               gcnt = i,*dnz,*onz,j,*lndmapi;
468fabe8965SStefano Zampini       ISLocalToGlobalMapping ll2g;
469fabe8965SStefano Zampini       PetscBool              flg;
470fabe8965SStefano Zampini       const PetscInt         *ii,*jj;
471fabe8965SStefano Zampini 
472fabe8965SStefano Zampini       /* communicate global id of separators */
473fabe8965SStefano Zampini       ierr = MatPreallocateInitialize(comm,A->rmap->n,A->cmap->n,dnz,onz);CHKERRQ(ierr);
474fabe8965SStefano Zampini       for (i = 0, cnt = 0; i < A->rmap->n; i++)
475fabe8965SStefano Zampini         dnz[i] = ndmapi[i] < 0 ? i + A->rmap->rstart : -1;
476fabe8965SStefano Zampini 
477fabe8965SStefano Zampini       ierr = PetscMalloc1(nl,&lndmapi);CHKERRQ(ierr);
478ad227feaSJunchao Zhang       ierr = PetscSFBcastBegin(sf,MPIU_INT,dnz,lndmapi,MPI_REPLACE);CHKERRQ(ierr);
479fabe8965SStefano Zampini 
480fabe8965SStefano Zampini       /* compute adjacency of isolated separators node */
481fabe8965SStefano Zampini       ierr = PetscMalloc1(gcnt,&workis);CHKERRQ(ierr);
482fabe8965SStefano Zampini       for (i = 0, cnt = 0; i < A->rmap->n; i++) {
483fabe8965SStefano Zampini         if (ndmapi[i] < 0 && ndmapc[i] < 2) {
484fabe8965SStefano Zampini           ierr = ISCreateStride(comm,1,i+A->rmap->rstart,1,&workis[cnt++]);CHKERRQ(ierr);
485fabe8965SStefano Zampini         }
486fabe8965SStefano Zampini       }
487fabe8965SStefano Zampini       for (i = cnt; i < gcnt; i++) {
488fabe8965SStefano Zampini         ierr = ISCreateStride(comm,0,0,1,&workis[i]);CHKERRQ(ierr);
489fabe8965SStefano Zampini       }
490fabe8965SStefano Zampini       for (i = 0; i < gcnt; i++) {
491fabe8965SStefano Zampini         ierr = PetscObjectSetName((PetscObject)workis[i],"ISOLATED");CHKERRQ(ierr);
492fabe8965SStefano Zampini         ierr = ISViewFromOptions(workis[i],NULL,"-view_isolated_separators");CHKERRQ(ierr);
493fabe8965SStefano Zampini       }
494fabe8965SStefano Zampini 
495fabe8965SStefano Zampini       /* no communications since all the ISes correspond to locally owned rows */
496fabe8965SStefano Zampini       ierr = MatIncreaseOverlap(A,gcnt,workis,1);CHKERRQ(ierr);
497fabe8965SStefano Zampini 
498fabe8965SStefano Zampini       /* end communicate global id of separators */
499ad227feaSJunchao Zhang       ierr = PetscSFBcastEnd(sf,MPIU_INT,dnz,lndmapi,MPI_REPLACE);CHKERRQ(ierr);
500fabe8965SStefano Zampini 
501fabe8965SStefano Zampini       /* communicate new layers : create a matrix and transpose it */
502580bdb30SBarry Smith       ierr = PetscArrayzero(dnz,A->rmap->n);CHKERRQ(ierr);
503580bdb30SBarry Smith       ierr = PetscArrayzero(onz,A->rmap->n);CHKERRQ(ierr);
504fabe8965SStefano Zampini       for (i = 0, j = 0; i < A->rmap->n; i++) {
505fabe8965SStefano Zampini         if (ndmapi[i] < 0 && ndmapc[i] < 2) {
506fabe8965SStefano Zampini           const PetscInt* idxs;
507fabe8965SStefano Zampini           PetscInt        s;
508fabe8965SStefano Zampini 
509fabe8965SStefano Zampini           ierr = ISGetLocalSize(workis[j],&s);CHKERRQ(ierr);
510fabe8965SStefano Zampini           ierr = ISGetIndices(workis[j],&idxs);CHKERRQ(ierr);
511fabe8965SStefano Zampini           ierr = MatPreallocateSet(i+A->rmap->rstart,s,idxs,dnz,onz);CHKERRQ(ierr);
512fabe8965SStefano Zampini           j++;
513fabe8965SStefano Zampini         }
514fabe8965SStefano Zampini       }
515fabe8965SStefano Zampini       if (j != cnt) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected local count %D != %D",j,cnt);
516fabe8965SStefano Zampini 
517fabe8965SStefano Zampini       for (i = 0; i < gcnt; i++) {
518fabe8965SStefano Zampini         ierr = PetscObjectSetName((PetscObject)workis[i],"EXTENDED");CHKERRQ(ierr);
519fabe8965SStefano Zampini         ierr = ISViewFromOptions(workis[i],NULL,"-view_isolated_separators");CHKERRQ(ierr);
520fabe8965SStefano Zampini       }
521fabe8965SStefano Zampini 
522fabe8965SStefano Zampini       for (i = 0, j = 0; i < A->rmap->n; i++) j = PetscMax(j,dnz[i]+onz[i]);
523fabe8965SStefano Zampini       ierr = PetscMalloc1(j,&vals);CHKERRQ(ierr);
524fabe8965SStefano Zampini       for (i = 0; i < j; i++) vals[i] = 1.0;
525fabe8965SStefano Zampini 
526fabe8965SStefano Zampini       ierr = MatCreate(comm,&A2);CHKERRQ(ierr);
527fabe8965SStefano Zampini       ierr = MatSetType(A2,MATMPIAIJ);CHKERRQ(ierr);
528fabe8965SStefano Zampini       ierr = MatSetSizes(A2,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
529fabe8965SStefano Zampini       ierr = MatMPIAIJSetPreallocation(A2,0,dnz,0,onz);CHKERRQ(ierr);
530fabe8965SStefano Zampini       ierr = MatSetOption(A2,MAT_NO_OFF_PROC_ENTRIES,PETSC_TRUE);CHKERRQ(ierr);
531fabe8965SStefano Zampini       for (i = 0, j = 0; i < A2->rmap->n; i++) {
532fabe8965SStefano Zampini         PetscInt        row = i+A2->rmap->rstart,s = dnz[i] + onz[i];
533fabe8965SStefano Zampini         const PetscInt* idxs;
534fabe8965SStefano Zampini 
535fabe8965SStefano Zampini         if (s) {
536fabe8965SStefano Zampini           ierr = ISGetIndices(workis[j],&idxs);CHKERRQ(ierr);
537fabe8965SStefano Zampini           ierr = MatSetValues(A2,1,&row,s,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
538fabe8965SStefano Zampini           ierr = ISRestoreIndices(workis[j],&idxs);CHKERRQ(ierr);
539fabe8965SStefano Zampini           j++;
540fabe8965SStefano Zampini         }
541fabe8965SStefano Zampini       }
542fabe8965SStefano Zampini       if (j != cnt) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected local count %D != %D",j,cnt);
543fabe8965SStefano Zampini       ierr = PetscFree(vals);CHKERRQ(ierr);
544fabe8965SStefano Zampini       ierr = MatAssemblyBegin(A2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
545fabe8965SStefano Zampini       ierr = MatAssemblyEnd(A2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
546fabe8965SStefano Zampini       ierr = MatTranspose(A2,MAT_INPLACE_MATRIX,&A2);CHKERRQ(ierr);
547fabe8965SStefano Zampini 
548fabe8965SStefano Zampini       /* extract submatrix corresponding to the coupling "owned separators" x "isolated separators" */
549fabe8965SStefano Zampini       for (i = 0, j = 0; i < nl; i++)
550fabe8965SStefano Zampini         if (lndmapi[i] >= 0) lndmapi[j++] = lndmapi[i];
551fabe8965SStefano Zampini       ierr = ISCreateGeneral(comm,j,lndmapi,PETSC_USE_POINTER,&is);CHKERRQ(ierr);
552fabe8965SStefano Zampini       ierr = MatMPIAIJGetLocalMatCondensed(A2,MAT_INITIAL_MATRIX,&is,NULL,&A3);CHKERRQ(ierr);
553fabe8965SStefano Zampini       ierr = ISDestroy(&is);CHKERRQ(ierr);
554fabe8965SStefano Zampini       ierr = MatDestroy(&A2);CHKERRQ(ierr);
555fabe8965SStefano Zampini 
556fabe8965SStefano Zampini       /* extend local to global map to include connected isolated separators */
557fabe8965SStefano Zampini       ierr = PetscObjectQuery((PetscObject)A3,"_petsc_GetLocalMatCondensed_iscol",(PetscObject*)&is);CHKERRQ(ierr);
558fabe8965SStefano Zampini       if (!is) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Missing column map");
559fabe8965SStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(is,&ll2g);CHKERRQ(ierr);
560fabe8965SStefano Zampini       ierr = MatGetRowIJ(A3,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&flg);CHKERRQ(ierr);
561fabe8965SStefano Zampini       if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure");
562fabe8965SStefano Zampini       ierr = ISCreateGeneral(PETSC_COMM_SELF,ii[i],jj,PETSC_COPY_VALUES,&is);CHKERRQ(ierr);
563fabe8965SStefano Zampini       ierr = MatRestoreRowIJ(A3,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&flg);CHKERRQ(ierr);
564fabe8965SStefano Zampini       if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure");
565fabe8965SStefano Zampini       ierr = ISLocalToGlobalMappingApplyIS(ll2g,is,&is2);CHKERRQ(ierr);
566fabe8965SStefano Zampini       ierr = ISDestroy(&is);CHKERRQ(ierr);
567fabe8965SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&ll2g);CHKERRQ(ierr);
568fabe8965SStefano Zampini 
569fabe8965SStefano Zampini       /* add new nodes to the local-to-global map */
570fabe8965SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(l2g);CHKERRQ(ierr);
571fabe8965SStefano Zampini       ierr = ISExpand(ndsub,is2,&is);CHKERRQ(ierr);
572fabe8965SStefano Zampini       ierr = ISDestroy(&is2);CHKERRQ(ierr);
573fabe8965SStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(is,l2g);CHKERRQ(ierr);
574fabe8965SStefano Zampini       ierr = ISDestroy(&is);CHKERRQ(ierr);
575fabe8965SStefano Zampini 
576fabe8965SStefano Zampini       ierr = MatDestroy(&A3);CHKERRQ(ierr);
577fabe8965SStefano Zampini       ierr = PetscFree(lndmapi);CHKERRQ(ierr);
578fabe8965SStefano Zampini       ierr = MatPreallocateFinalize(dnz,onz);CHKERRQ(ierr);
579fabe8965SStefano Zampini       for (i = 0; i < gcnt; i++) {
580fabe8965SStefano Zampini         ierr = ISDestroy(&workis[i]);CHKERRQ(ierr);
581fabe8965SStefano Zampini       }
582fabe8965SStefano Zampini       ierr = PetscFree(workis);CHKERRQ(ierr);
583fabe8965SStefano Zampini     }
584fabe8965SStefano Zampini     ierr = ISRestoreIndices(ndmap,&ndmapi);CHKERRQ(ierr);
585fabe8965SStefano Zampini     ierr = PetscSFDestroy(&sf);CHKERRQ(ierr);
586fabe8965SStefano Zampini     ierr = PetscFree(ndmapc);CHKERRQ(ierr);
587fabe8965SStefano Zampini     ierr = ISDestroy(&ndmap);CHKERRQ(ierr);
588fabe8965SStefano Zampini     ierr = ISDestroy(&ndsub);CHKERRQ(ierr);
589fabe8965SStefano Zampini     ierr = ISLocalToGlobalMappingSetBlockSize(*l2g,bs);CHKERRQ(ierr);
590fabe8965SStefano Zampini     ierr = ISLocalToGlobalMappingViewFromOptions(*l2g,NULL,"-matis_nd_l2g_view");CHKERRQ(ierr);
591fabe8965SStefano Zampini     break;
592fabe8965SStefano Zampini   case MAT_IS_DISASSEMBLE_L2G_NATURAL:
593fabe8965SStefano Zampini     if (ismpiaij) {
594fabe8965SStefano Zampini       ierr = MatMPIAIJGetSeqAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr);
595fabe8965SStefano Zampini     } else if (ismpibaij) {
596fabe8965SStefano Zampini       ierr = MatMPIBAIJGetSeqBAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr);
597c9225affSStefano Zampini     } else SETERRQ1(comm,PETSC_ERR_SUP,"Type %s",((PetscObject)A)->type_name);
598c9225affSStefano Zampini     if (!garray) SETERRQ(comm,PETSC_ERR_ARG_WRONGSTATE,"garray not present");
599c9225affSStefano Zampini     if (A->rmap->n) {
600fabe8965SStefano Zampini       PetscInt dc,oc,stc,*aux;
601c9225affSStefano Zampini 
602c9225affSStefano Zampini       ierr = MatGetLocalSize(A,NULL,&dc);CHKERRQ(ierr);
603c9225affSStefano Zampini       ierr = MatGetLocalSize(Ao,NULL,&oc);CHKERRQ(ierr);
604c9225affSStefano Zampini       ierr = MatGetOwnershipRangeColumn(A,&stc,NULL);CHKERRQ(ierr);
605c9225affSStefano Zampini       ierr = PetscMalloc1((dc+oc)/bs,&aux);CHKERRQ(ierr);
606c9225affSStefano Zampini       for (i=0; i<dc/bs; i++) aux[i]       = i+stc/bs;
607c9225affSStefano Zampini       for (i=0; i<oc/bs; i++) aux[i+dc/bs] = garray[i];
608c9225affSStefano Zampini       ierr = ISCreateBlock(comm,bs,(dc+oc)/bs,aux,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
609c9225affSStefano Zampini     } else {
610c9225affSStefano Zampini       ierr = ISCreateBlock(comm,1,0,NULL,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
611c9225affSStefano Zampini     }
612c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(is,l2g);CHKERRQ(ierr);
613c9225affSStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
614fabe8965SStefano Zampini     break;
615fabe8965SStefano Zampini   default:
616fabe8965SStefano Zampini     SETERRQ1(comm,PETSC_ERR_ARG_WRONG,"Unsupported l2g disassembling type %D",mode);
617c9225affSStefano Zampini   }
618c9225affSStefano Zampini   PetscFunctionReturn(0);
619c9225affSStefano Zampini }
620c9225affSStefano Zampini 
621c9225affSStefano Zampini PETSC_INTERN PetscErrorCode MatConvert_XAIJ_IS(Mat A,MatType type,MatReuse reuse,Mat *newmat)
622c9225affSStefano Zampini {
623c9225affSStefano Zampini   Mat                    lA,Ad,Ao,B = NULL;
6246989cf23SStefano Zampini   ISLocalToGlobalMapping rl2g,cl2g;
6256989cf23SStefano Zampini   IS                     is;
6266989cf23SStefano Zampini   MPI_Comm               comm;
6276989cf23SStefano Zampini   void                   *ptrs[2];
6286989cf23SStefano Zampini   const char             *names[2] = {"_convert_csr_aux","_convert_csr_data"};
629c9225affSStefano Zampini   const PetscInt         *garray;
6306989cf23SStefano Zampini   PetscScalar            *dd,*od,*aa,*data;
631c9225affSStefano Zampini   const PetscInt         *di,*dj,*oi,*oj;
632c9225affSStefano Zampini   const PetscInt         *odi,*odj,*ooi,*ooj;
6336989cf23SStefano Zampini   PetscInt               *aux,*ii,*jj;
634c9225affSStefano Zampini   PetscInt               bs,lc,dr,dc,oc,str,stc,nnz,i,jd,jo,cum;
635c9225affSStefano Zampini   PetscBool              flg,ismpiaij,ismpibaij,was_inplace = PETSC_FALSE;
636c9225affSStefano Zampini   PetscMPIInt            size;
6376989cf23SStefano Zampini   PetscErrorCode         ierr;
6386989cf23SStefano Zampini 
639ab4d48faSStefano Zampini   PetscFunctionBegin;
6406989cf23SStefano Zampini   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
641ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr);
642c9225affSStefano Zampini   if (size == 1) {
643c9225affSStefano Zampini     ierr = MatConvert_SeqXAIJ_IS(A,type,reuse,newmat);CHKERRQ(ierr);
644c9225affSStefano Zampini     PetscFunctionReturn(0);
645c9225affSStefano Zampini   }
646c9225affSStefano Zampini   if (reuse != MAT_REUSE_MATRIX && A->cmap->N == A->rmap->N) {
647c9225affSStefano Zampini     ierr = MatMPIXAIJComputeLocalToGlobalMapping_Private(A,&rl2g);CHKERRQ(ierr);
648c9225affSStefano Zampini     ierr = MatCreate(comm,&B);CHKERRQ(ierr);
649c9225affSStefano Zampini     ierr = MatSetType(B,MATIS);CHKERRQ(ierr);
650c9225affSStefano Zampini     ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
651c9225affSStefano Zampini     ierr = MatSetLocalToGlobalMapping(B,rl2g,rl2g);CHKERRQ(ierr);
652c9225affSStefano Zampini     ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
653c9225affSStefano Zampini     ierr = MatSetBlockSize(B,bs);CHKERRQ(ierr);
654c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
655c9225affSStefano Zampini     if (reuse == MAT_INPLACE_MATRIX) was_inplace = PETSC_TRUE;
656c9225affSStefano Zampini     reuse = MAT_REUSE_MATRIX;
657c9225affSStefano Zampini   }
658c9225affSStefano Zampini   if (reuse == MAT_REUSE_MATRIX) {
659c9225affSStefano Zampini     Mat            *newlA, lA;
660c9225affSStefano Zampini     IS             rows, cols;
661c9225affSStefano Zampini     const PetscInt *ridx, *cidx;
662c9225affSStefano Zampini     PetscInt       rbs, cbs, nr, nc;
663c9225affSStefano Zampini 
664c9225affSStefano Zampini     if (!B) B = *newmat;
665c9225affSStefano Zampini     ierr = MatGetLocalToGlobalMapping(B,&rl2g,&cl2g);CHKERRQ(ierr);
666c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockIndices(rl2g,&ridx);CHKERRQ(ierr);
667c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockIndices(cl2g,&cidx);CHKERRQ(ierr);
668c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingGetSize(rl2g,&nr);CHKERRQ(ierr);
669c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingGetSize(cl2g,&nc);CHKERRQ(ierr);
670c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockSize(rl2g,&rbs);CHKERRQ(ierr);
671c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockSize(cl2g,&cbs);CHKERRQ(ierr);
672c9225affSStefano Zampini     ierr = ISCreateBlock(comm,rbs,nr/rbs,ridx,PETSC_USE_POINTER,&rows);CHKERRQ(ierr);
673c9225affSStefano Zampini     if (rl2g != cl2g) {
674c9225affSStefano Zampini       ierr = ISCreateBlock(comm,cbs,nc/cbs,cidx,PETSC_USE_POINTER,&cols);CHKERRQ(ierr);
675c9225affSStefano Zampini     } else {
676c9225affSStefano Zampini       ierr = PetscObjectReference((PetscObject)rows);CHKERRQ(ierr);
677c9225affSStefano Zampini       cols = rows;
678c9225affSStefano Zampini     }
679c9225affSStefano Zampini     ierr = MatISGetLocalMat(B,&lA);CHKERRQ(ierr);
680c9225affSStefano Zampini     ierr = MatCreateSubMatrices(A,1,&rows,&cols,MAT_INITIAL_MATRIX,&newlA);CHKERRQ(ierr);
681c9225affSStefano Zampini     ierr = MatConvert(newlA[0],MATSEQAIJ,MAT_INPLACE_MATRIX,&newlA[0]);CHKERRQ(ierr);
682c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingRestoreBlockIndices(rl2g,&ridx);CHKERRQ(ierr);
683c9225affSStefano Zampini     ierr = ISLocalToGlobalMappingRestoreBlockIndices(cl2g,&cidx);CHKERRQ(ierr);
684c9225affSStefano Zampini     ierr = ISDestroy(&rows);CHKERRQ(ierr);
685c9225affSStefano Zampini     ierr = ISDestroy(&cols);CHKERRQ(ierr);
686c9225affSStefano Zampini     if (!lA->preallocated) { /* first time */
687c9225affSStefano Zampini       ierr = MatDuplicate(newlA[0],MAT_COPY_VALUES,&lA);CHKERRQ(ierr);
688c9225affSStefano Zampini       ierr = MatISSetLocalMat(B,lA);CHKERRQ(ierr);
689c9225affSStefano Zampini       ierr = PetscObjectDereference((PetscObject)lA);CHKERRQ(ierr);
690c9225affSStefano Zampini     }
691c9225affSStefano Zampini     ierr = MatCopy(newlA[0],lA,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
692c9225affSStefano Zampini     ierr = MatDestroySubMatrices(1,&newlA);CHKERRQ(ierr);
693c9225affSStefano Zampini     ierr = MatISScaleDisassembling_Private(B);CHKERRQ(ierr);
694c9225affSStefano Zampini     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
695c9225affSStefano Zampini     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
696c9225affSStefano Zampini     if (was_inplace) { ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); }
697c9225affSStefano Zampini     else *newmat = B;
698c9225affSStefano Zampini     PetscFunctionReturn(0);
699c9225affSStefano Zampini   }
700c9225affSStefano Zampini   /* rectangular case, just compress out the column space */
701b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIAIJ ,&ismpiaij);CHKERRQ(ierr);
702b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIBAIJ,&ismpibaij);CHKERRQ(ierr);
703c9225affSStefano Zampini   if (ismpiaij) {
704c9225affSStefano Zampini     bs   = 1;
705c9225affSStefano Zampini     ierr = MatMPIAIJGetSeqAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr);
706c9225affSStefano Zampini   } else if (ismpibaij) {
707c9225affSStefano Zampini     ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
708c9225affSStefano Zampini     ierr = MatMPIBAIJGetSeqBAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr);
709c9225affSStefano Zampini     ierr = MatConvert(Ad,MATSEQAIJ,MAT_INITIAL_MATRIX,&Ad);CHKERRQ(ierr);
710c9225affSStefano Zampini     ierr = MatConvert(Ao,MATSEQAIJ,MAT_INITIAL_MATRIX,&Ao);CHKERRQ(ierr);
711c9225affSStefano Zampini   } else SETERRQ1(comm,PETSC_ERR_SUP,"Type %s",((PetscObject)A)->type_name);
712c9225affSStefano Zampini   ierr = MatSeqAIJGetArray(Ad,&dd);CHKERRQ(ierr);
713c9225affSStefano Zampini   ierr = MatSeqAIJGetArray(Ao,&od);CHKERRQ(ierr);
714c9225affSStefano Zampini   if (!garray) SETERRQ(comm,PETSC_ERR_ARG_WRONGSTATE,"garray not present");
7156989cf23SStefano Zampini 
7166989cf23SStefano Zampini   /* access relevant information from MPIAIJ */
7176989cf23SStefano Zampini   ierr = MatGetOwnershipRange(A,&str,NULL);CHKERRQ(ierr);
7186989cf23SStefano Zampini   ierr = MatGetOwnershipRangeColumn(A,&stc,NULL);CHKERRQ(ierr);
7196989cf23SStefano Zampini   ierr = MatGetLocalSize(A,&dr,&dc);CHKERRQ(ierr);
720c9225affSStefano Zampini   ierr = MatGetLocalSize(Ao,NULL,&oc);CHKERRQ(ierr);
721c9225affSStefano Zampini   ierr = MatGetRowIJ(Ad,0,PETSC_FALSE,PETSC_FALSE,&i,&di,&dj,&flg);CHKERRQ(ierr);
722c9225affSStefano Zampini   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure");
723c9225affSStefano Zampini   ierr = MatGetRowIJ(Ao,0,PETSC_FALSE,PETSC_FALSE,&i,&oi,&oj,&flg);CHKERRQ(ierr);
724c9225affSStefano Zampini   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure");
725c9225affSStefano Zampini   nnz = di[dr] + oi[dr];
726c9225affSStefano Zampini   /* store original pointers to be restored later */
727c9225affSStefano Zampini   odi = di; odj = dj; ooi = oi; ooj = oj;
7286989cf23SStefano Zampini 
7296989cf23SStefano Zampini   /* generate l2g maps for rows and cols */
730c9225affSStefano Zampini   ierr = ISCreateStride(comm,dr/bs,str/bs,1,&is);CHKERRQ(ierr);
731c9225affSStefano Zampini   if (bs > 1) {
732c9225affSStefano Zampini     IS is2;
733c9225affSStefano Zampini 
734c9225affSStefano Zampini     ierr = ISGetLocalSize(is,&i);CHKERRQ(ierr);
735c9225affSStefano Zampini     ierr = ISGetIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr);
736c9225affSStefano Zampini     ierr = ISCreateBlock(comm,bs,i,aux,PETSC_COPY_VALUES,&is2);CHKERRQ(ierr);
737c9225affSStefano Zampini     ierr = ISRestoreIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr);
738c9225affSStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
739c9225affSStefano Zampini     is   = is2;
740c9225affSStefano Zampini   }
7416989cf23SStefano Zampini   ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr);
7426989cf23SStefano Zampini   ierr = ISDestroy(&is);CHKERRQ(ierr);
743e363d98aSStefano Zampini   if (dr) {
744c9225affSStefano Zampini     ierr = PetscMalloc1((dc+oc)/bs,&aux);CHKERRQ(ierr);
745c9225affSStefano Zampini     for (i=0; i<dc/bs; i++) aux[i]       = i+stc/bs;
746c9225affSStefano Zampini     for (i=0; i<oc/bs; i++) aux[i+dc/bs] = garray[i];
747c9225affSStefano Zampini     ierr = ISCreateBlock(comm,bs,(dc+oc)/bs,aux,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
748e363d98aSStefano Zampini     lc   = dc+oc;
749e363d98aSStefano Zampini   } else {
750c9225affSStefano Zampini     ierr = ISCreateBlock(comm,bs,0,NULL,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
751e363d98aSStefano Zampini     lc   = 0;
752e363d98aSStefano Zampini   }
7536989cf23SStefano Zampini   ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr);
7546989cf23SStefano Zampini   ierr = ISDestroy(&is);CHKERRQ(ierr);
7556989cf23SStefano Zampini 
7566989cf23SStefano Zampini   /* create MATIS object */
757c9225affSStefano Zampini   ierr = MatCreate(comm,&B);CHKERRQ(ierr);
758c9225affSStefano Zampini   ierr = MatSetSizes(B,dr,dc,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
759c9225affSStefano Zampini   ierr = MatSetType(B,MATIS);CHKERRQ(ierr);
760c9225affSStefano Zampini   ierr = MatSetBlockSize(B,bs);CHKERRQ(ierr);
761c9225affSStefano Zampini   ierr = MatSetLocalToGlobalMapping(B,rl2g,cl2g);CHKERRQ(ierr);
7626989cf23SStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
7636989cf23SStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
7646989cf23SStefano Zampini 
7656989cf23SStefano Zampini   /* merge local matrices */
7666989cf23SStefano Zampini   ierr = PetscMalloc1(nnz+dr+1,&aux);CHKERRQ(ierr);
7676989cf23SStefano Zampini   ierr = PetscMalloc1(nnz,&data);CHKERRQ(ierr);
7686989cf23SStefano Zampini   ii   = aux;
7696989cf23SStefano Zampini   jj   = aux+dr+1;
7706989cf23SStefano Zampini   aa   = data;
7716989cf23SStefano Zampini   *ii  = *(di++) + *(oi++);
7726989cf23SStefano Zampini   for (jd=0,jo=0,cum=0;*ii<nnz;cum++)
7736989cf23SStefano Zampini   {
7746989cf23SStefano Zampini      for (;jd<*di;jd++) { *jj++ = *dj++;      *aa++ = *dd++; }
7756989cf23SStefano Zampini      for (;jo<*oi;jo++) { *jj++ = *oj++ + dc; *aa++ = *od++; }
7766989cf23SStefano Zampini      *(++ii) = *(di++) + *(oi++);
7776989cf23SStefano Zampini   }
7786989cf23SStefano Zampini   for (;cum<dr;cum++) *(++ii) = nnz;
779c9225affSStefano Zampini 
780c9225affSStefano Zampini   ierr = MatRestoreRowIJ(Ad,0,PETSC_FALSE,PETSC_FALSE,&i,&odi,&odj,&flg);CHKERRQ(ierr);
781c9225affSStefano Zampini   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot restore IJ structure");
782c9225affSStefano Zampini   ierr = MatRestoreRowIJ(Ao,0,PETSC_FALSE,PETSC_FALSE,&i,&ooi,&ooj,&flg);CHKERRQ(ierr);
783c9225affSStefano Zampini   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot restore IJ structure");
784c9225affSStefano Zampini   ierr = MatSeqAIJRestoreArray(Ad,&dd);CHKERRQ(ierr);
785c9225affSStefano Zampini   ierr = MatSeqAIJRestoreArray(Ao,&od);CHKERRQ(ierr);
786c9225affSStefano Zampini 
7876989cf23SStefano Zampini   ii   = aux;
7886989cf23SStefano Zampini   jj   = aux+dr+1;
7896989cf23SStefano Zampini   aa   = data;
790e363d98aSStefano Zampini   ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,dr,lc,ii,jj,aa,&lA);CHKERRQ(ierr);
7916989cf23SStefano Zampini 
7926989cf23SStefano Zampini   /* create containers to destroy the data */
7936989cf23SStefano Zampini   ptrs[0] = aux;
7946989cf23SStefano Zampini   ptrs[1] = data;
7956989cf23SStefano Zampini   for (i=0; i<2; i++) {
7966989cf23SStefano Zampini     PetscContainer c;
7976989cf23SStefano Zampini 
7986989cf23SStefano Zampini     ierr = PetscContainerCreate(PETSC_COMM_SELF,&c);CHKERRQ(ierr);
7996989cf23SStefano Zampini     ierr = PetscContainerSetPointer(c,ptrs[i]);CHKERRQ(ierr);
800b81c21eeSStefano Zampini     ierr = PetscContainerSetUserDestroy(c,PetscContainerUserDestroyDefault);CHKERRQ(ierr);
8016989cf23SStefano Zampini     ierr = PetscObjectCompose((PetscObject)lA,names[i],(PetscObject)c);CHKERRQ(ierr);
8026989cf23SStefano Zampini     ierr = PetscContainerDestroy(&c);CHKERRQ(ierr);
8036989cf23SStefano Zampini   }
804c9225affSStefano Zampini   if (ismpibaij) { /* destroy converted local matrices */
805c9225affSStefano Zampini     ierr = MatDestroy(&Ad);CHKERRQ(ierr);
806c9225affSStefano Zampini     ierr = MatDestroy(&Ao);CHKERRQ(ierr);
807c9225affSStefano Zampini   }
8086989cf23SStefano Zampini 
8096989cf23SStefano Zampini   /* finalize matrix */
810c9225affSStefano Zampini   ierr = MatISSetLocalMat(B,lA);CHKERRQ(ierr);
8116989cf23SStefano Zampini   ierr = MatDestroy(&lA);CHKERRQ(ierr);
812c9225affSStefano Zampini   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
813c9225affSStefano Zampini   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
814c9225affSStefano Zampini   if (reuse == MAT_INPLACE_MATRIX) {
815c9225affSStefano Zampini     ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr);
816c9225affSStefano Zampini   } else *newmat = B;
8176989cf23SStefano Zampini   PetscFunctionReturn(0);
8186989cf23SStefano Zampini }
8196989cf23SStefano Zampini 
8205e3038f0Sstefano_zampini PETSC_INTERN PetscErrorCode MatConvert_Nest_IS(Mat A,MatType type,MatReuse reuse,Mat *newmat)
8215e3038f0Sstefano_zampini {
8225e3038f0Sstefano_zampini   Mat                    **nest,*snest,**rnest,lA,B;
8235e3038f0Sstefano_zampini   IS                     *iscol,*isrow,*islrow,*islcol;
8245e3038f0Sstefano_zampini   ISLocalToGlobalMapping rl2g,cl2g;
8255e3038f0Sstefano_zampini   MPI_Comm               comm;
8265b003df0Sstefano_zampini   PetscInt               *lr,*lc,*l2gidxs;
8275b003df0Sstefano_zampini   PetscInt               i,j,nr,nc,rbs,cbs;
8289e7b2b25Sstefano_zampini   PetscBool              convert,lreuse,*istrans;
8295e3038f0Sstefano_zampini   PetscErrorCode         ierr;
8305e3038f0Sstefano_zampini 
831ab4d48faSStefano Zampini   PetscFunctionBegin;
8325e3038f0Sstefano_zampini   ierr   = MatNestGetSubMats(A,&nr,&nc,&nest);CHKERRQ(ierr);
8335e3038f0Sstefano_zampini   lreuse = PETSC_FALSE;
8345e3038f0Sstefano_zampini   rnest  = NULL;
8355e3038f0Sstefano_zampini   if (reuse == MAT_REUSE_MATRIX) {
8365e3038f0Sstefano_zampini     PetscBool ismatis,isnest;
8375e3038f0Sstefano_zampini 
8385e3038f0Sstefano_zampini     ierr = PetscObjectTypeCompare((PetscObject)*newmat,MATIS,&ismatis);CHKERRQ(ierr);
839a32993e3SJed Brown     if (!ismatis) SETERRQ1(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_USER,"Cannot reuse matrix of type %s",((PetscObject)(*newmat))->type_name);
8405e3038f0Sstefano_zampini     ierr = MatISGetLocalMat(*newmat,&lA);CHKERRQ(ierr);
8415e3038f0Sstefano_zampini     ierr = PetscObjectTypeCompare((PetscObject)lA,MATNEST,&isnest);CHKERRQ(ierr);
8425e3038f0Sstefano_zampini     if (isnest) {
8435e3038f0Sstefano_zampini       ierr   = MatNestGetSubMats(lA,&i,&j,&rnest);CHKERRQ(ierr);
8445e3038f0Sstefano_zampini       lreuse = (PetscBool)(i == nr && j == nc);
8455e3038f0Sstefano_zampini       if (!lreuse) rnest = NULL;
8465e3038f0Sstefano_zampini     }
8475e3038f0Sstefano_zampini   }
8485e3038f0Sstefano_zampini   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
8495b003df0Sstefano_zampini   ierr = PetscCalloc2(nr,&lr,nc,&lc);CHKERRQ(ierr);
850071fcb05SBarry Smith   ierr = PetscCalloc6(nr,&isrow,nc,&iscol,nr,&islrow,nc,&islcol,nr*nc,&snest,nr*nc,&istrans);CHKERRQ(ierr);
8515e3038f0Sstefano_zampini   ierr = MatNestGetISs(A,isrow,iscol);CHKERRQ(ierr);
8525e3038f0Sstefano_zampini   for (i=0;i<nr;i++) {
8535e3038f0Sstefano_zampini     for (j=0;j<nc;j++) {
8545e3038f0Sstefano_zampini       PetscBool ismatis;
8559e7b2b25Sstefano_zampini       PetscInt  l1,l2,lb1,lb2,ij=i*nc+j;
8565e3038f0Sstefano_zampini 
8575e3038f0Sstefano_zampini       /* Null matrix pointers are allowed in MATNEST */
8585e3038f0Sstefano_zampini       if (!nest[i][j]) continue;
8595e3038f0Sstefano_zampini 
8605e3038f0Sstefano_zampini       /* Nested matrices should be of type MATIS */
8619e7b2b25Sstefano_zampini       ierr = PetscObjectTypeCompare((PetscObject)nest[i][j],MATTRANSPOSEMAT,&istrans[ij]);CHKERRQ(ierr);
8629e7b2b25Sstefano_zampini       if (istrans[ij]) {
8639e7b2b25Sstefano_zampini         Mat T,lT;
8649e7b2b25Sstefano_zampini         ierr = MatTransposeGetMat(nest[i][j],&T);CHKERRQ(ierr);
8659e7b2b25Sstefano_zampini         ierr = PetscObjectTypeCompare((PetscObject)T,MATIS,&ismatis);CHKERRQ(ierr);
8669e7b2b25Sstefano_zampini         if (!ismatis) SETERRQ2(comm,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) (transposed) is not of type MATIS",i,j);
8679e7b2b25Sstefano_zampini         ierr = MatISGetLocalMat(T,&lT);CHKERRQ(ierr);
8689e7b2b25Sstefano_zampini         ierr = MatCreateTranspose(lT,&snest[ij]);CHKERRQ(ierr);
8699e7b2b25Sstefano_zampini       } else {
8705e3038f0Sstefano_zampini         ierr = PetscObjectTypeCompare((PetscObject)nest[i][j],MATIS,&ismatis);CHKERRQ(ierr);
8715e3038f0Sstefano_zampini         if (!ismatis) SETERRQ2(comm,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) is not of type MATIS",i,j);
8729e7b2b25Sstefano_zampini         ierr = MatISGetLocalMat(nest[i][j],&snest[ij]);CHKERRQ(ierr);
8739e7b2b25Sstefano_zampini       }
8745e3038f0Sstefano_zampini 
8755e3038f0Sstefano_zampini       /* Check compatibility of local sizes */
8765e3038f0Sstefano_zampini       ierr = MatGetSize(snest[ij],&l1,&l2);CHKERRQ(ierr);
8779e7b2b25Sstefano_zampini       ierr = MatGetBlockSizes(snest[ij],&lb1,&lb2);CHKERRQ(ierr);
8785e3038f0Sstefano_zampini       if (!l1 || !l2) continue;
8795e3038f0Sstefano_zampini       if (lr[i] && l1 != lr[i]) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid local size %D != %D",i,j,lr[i],l1);
8805e3038f0Sstefano_zampini       if (lc[j] && l2 != lc[j]) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid local size %D != %D",i,j,lc[j],l2);
8815e3038f0Sstefano_zampini       lr[i] = l1;
8825e3038f0Sstefano_zampini       lc[j] = l2;
8835e3038f0Sstefano_zampini 
8845e3038f0Sstefano_zampini       /* check compatibilty for local matrix reusage */
8855e3038f0Sstefano_zampini       if (rnest && !rnest[i][j] != !snest[ij]) lreuse = PETSC_FALSE;
8865e3038f0Sstefano_zampini     }
8875e3038f0Sstefano_zampini   }
8885e3038f0Sstefano_zampini 
88976bd3646SJed Brown   if (PetscDefined (USE_DEBUG)) {
8905e3038f0Sstefano_zampini     /* Check compatibility of l2g maps for rows */
8915e3038f0Sstefano_zampini     for (i=0;i<nr;i++) {
8925e3038f0Sstefano_zampini       rl2g = NULL;
8935e3038f0Sstefano_zampini       for (j=0;j<nc;j++) {
8945e3038f0Sstefano_zampini         PetscInt n1,n2;
8955e3038f0Sstefano_zampini 
8965e3038f0Sstefano_zampini         if (!nest[i][j]) continue;
8979e7b2b25Sstefano_zampini         if (istrans[i*nc+j]) {
8989e7b2b25Sstefano_zampini           Mat T;
8999e7b2b25Sstefano_zampini 
9009e7b2b25Sstefano_zampini           ierr = MatTransposeGetMat(nest[i][j],&T);CHKERRQ(ierr);
9019e7b2b25Sstefano_zampini           ierr = MatGetLocalToGlobalMapping(T,NULL,&cl2g);CHKERRQ(ierr);
9029e7b2b25Sstefano_zampini         } else {
9035e3038f0Sstefano_zampini           ierr = MatGetLocalToGlobalMapping(nest[i][j],&cl2g,NULL);CHKERRQ(ierr);
9049e7b2b25Sstefano_zampini         }
9055e3038f0Sstefano_zampini         ierr = ISLocalToGlobalMappingGetSize(cl2g,&n1);CHKERRQ(ierr);
9065e3038f0Sstefano_zampini         if (!n1) continue;
9075e3038f0Sstefano_zampini         if (!rl2g) {
9085e3038f0Sstefano_zampini           rl2g = cl2g;
9095e3038f0Sstefano_zampini         } else {
9105e3038f0Sstefano_zampini           const PetscInt *idxs1,*idxs2;
9115e3038f0Sstefano_zampini           PetscBool      same;
9125e3038f0Sstefano_zampini 
9135e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingGetSize(rl2g,&n2);CHKERRQ(ierr);
9145e3038f0Sstefano_zampini           if (n1 != n2) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid row l2gmap size %D != %D",i,j,n1,n2);
9155e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingGetIndices(cl2g,&idxs1);CHKERRQ(ierr);
9165e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingGetIndices(rl2g,&idxs2);CHKERRQ(ierr);
917580bdb30SBarry Smith           ierr = PetscArraycmp(idxs1,idxs2,n1,&same);CHKERRQ(ierr);
9185e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingRestoreIndices(cl2g,&idxs1);CHKERRQ(ierr);
9195e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingRestoreIndices(rl2g,&idxs2);CHKERRQ(ierr);
9205e3038f0Sstefano_zampini           if (!same) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid row l2gmap",i,j);
9215e3038f0Sstefano_zampini         }
9225e3038f0Sstefano_zampini       }
9235e3038f0Sstefano_zampini     }
9245e3038f0Sstefano_zampini     /* Check compatibility of l2g maps for columns */
9255e3038f0Sstefano_zampini     for (i=0;i<nc;i++) {
9265e3038f0Sstefano_zampini       rl2g = NULL;
9275e3038f0Sstefano_zampini       for (j=0;j<nr;j++) {
9285e3038f0Sstefano_zampini         PetscInt n1,n2;
9295e3038f0Sstefano_zampini 
9305e3038f0Sstefano_zampini         if (!nest[j][i]) continue;
9319e7b2b25Sstefano_zampini         if (istrans[j*nc+i]) {
9329e7b2b25Sstefano_zampini           Mat T;
9339e7b2b25Sstefano_zampini 
9349e7b2b25Sstefano_zampini           ierr = MatTransposeGetMat(nest[j][i],&T);CHKERRQ(ierr);
9359e7b2b25Sstefano_zampini           ierr = MatGetLocalToGlobalMapping(T,&cl2g,NULL);CHKERRQ(ierr);
9369e7b2b25Sstefano_zampini         } else {
9375e3038f0Sstefano_zampini           ierr = MatGetLocalToGlobalMapping(nest[j][i],NULL,&cl2g);CHKERRQ(ierr);
9389e7b2b25Sstefano_zampini         }
9395e3038f0Sstefano_zampini         ierr = ISLocalToGlobalMappingGetSize(cl2g,&n1);CHKERRQ(ierr);
9405e3038f0Sstefano_zampini         if (!n1) continue;
9415e3038f0Sstefano_zampini         if (!rl2g) {
9425e3038f0Sstefano_zampini           rl2g = cl2g;
9435e3038f0Sstefano_zampini         } else {
9445e3038f0Sstefano_zampini           const PetscInt *idxs1,*idxs2;
9455e3038f0Sstefano_zampini           PetscBool      same;
9465e3038f0Sstefano_zampini 
9475e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingGetSize(rl2g,&n2);CHKERRQ(ierr);
9485e3038f0Sstefano_zampini           if (n1 != n2) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid column l2gmap size %D != %D",j,i,n1,n2);
9495e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingGetIndices(cl2g,&idxs1);CHKERRQ(ierr);
9505e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingGetIndices(rl2g,&idxs2);CHKERRQ(ierr);
951580bdb30SBarry Smith           ierr = PetscArraycmp(idxs1,idxs2,n1,&same);CHKERRQ(ierr);
9525e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingRestoreIndices(cl2g,&idxs1);CHKERRQ(ierr);
9535e3038f0Sstefano_zampini           ierr = ISLocalToGlobalMappingRestoreIndices(rl2g,&idxs2);CHKERRQ(ierr);
9545e3038f0Sstefano_zampini           if (!same) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid column l2gmap",j,i);
9555e3038f0Sstefano_zampini         }
9565e3038f0Sstefano_zampini       }
9575e3038f0Sstefano_zampini     }
95876bd3646SJed Brown   }
9595e3038f0Sstefano_zampini 
9605e3038f0Sstefano_zampini   B = NULL;
9615e3038f0Sstefano_zampini   if (reuse != MAT_REUSE_MATRIX) {
9625b003df0Sstefano_zampini     PetscInt stl;
9635b003df0Sstefano_zampini 
9645e3038f0Sstefano_zampini     /* Create l2g map for the rows of the new matrix and index sets for the local MATNEST */
9655e3038f0Sstefano_zampini     for (i=0,stl=0;i<nr;i++) stl += lr[i];
9665e3038f0Sstefano_zampini     ierr = PetscMalloc1(stl,&l2gidxs);CHKERRQ(ierr);
9675b003df0Sstefano_zampini     for (i=0,stl=0;i<nr;i++) {
9685e3038f0Sstefano_zampini       Mat            usedmat;
9695e3038f0Sstefano_zampini       Mat_IS         *matis;
9705e3038f0Sstefano_zampini       const PetscInt *idxs;
9715e3038f0Sstefano_zampini 
9725e3038f0Sstefano_zampini       /* local IS for local NEST */
9735b003df0Sstefano_zampini       ierr  = ISCreateStride(PETSC_COMM_SELF,lr[i],stl,1,&islrow[i]);CHKERRQ(ierr);
9745e3038f0Sstefano_zampini 
9755e3038f0Sstefano_zampini       /* l2gmap */
9765e3038f0Sstefano_zampini       j = 0;
9775e3038f0Sstefano_zampini       usedmat = nest[i][j];
9789e7b2b25Sstefano_zampini       while (!usedmat && j < nc-1) usedmat = nest[i][++j];
9799e7b2b25Sstefano_zampini       if (!usedmat) SETERRQ(comm,PETSC_ERR_SUP,"Cannot find valid row mat");
9809e7b2b25Sstefano_zampini 
9819e7b2b25Sstefano_zampini       if (istrans[i*nc+j]) {
9829e7b2b25Sstefano_zampini         Mat T;
9839e7b2b25Sstefano_zampini         ierr    = MatTransposeGetMat(usedmat,&T);CHKERRQ(ierr);
9849e7b2b25Sstefano_zampini         usedmat = T;
9859e7b2b25Sstefano_zampini       }
9865e3038f0Sstefano_zampini       matis = (Mat_IS*)(usedmat->data);
9875e3038f0Sstefano_zampini       ierr  = ISGetIndices(isrow[i],&idxs);CHKERRQ(ierr);
9889e7b2b25Sstefano_zampini       if (istrans[i*nc+j]) {
989ad227feaSJunchao Zhang         ierr = PetscSFBcastBegin(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
990ad227feaSJunchao Zhang         ierr = PetscSFBcastEnd(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
9919e7b2b25Sstefano_zampini       } else {
992ad227feaSJunchao Zhang         ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
993ad227feaSJunchao Zhang         ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
9949e7b2b25Sstefano_zampini       }
9955e3038f0Sstefano_zampini       ierr = ISRestoreIndices(isrow[i],&idxs);CHKERRQ(ierr);
9965e3038f0Sstefano_zampini       stl += lr[i];
9975e3038f0Sstefano_zampini     }
9985e3038f0Sstefano_zampini     ierr = ISLocalToGlobalMappingCreate(comm,1,stl,l2gidxs,PETSC_OWN_POINTER,&rl2g);CHKERRQ(ierr);
9995e3038f0Sstefano_zampini 
10005e3038f0Sstefano_zampini     /* Create l2g map for columns of the new matrix and index sets for the local MATNEST */
10015e3038f0Sstefano_zampini     for (i=0,stl=0;i<nc;i++) stl += lc[i];
10025e3038f0Sstefano_zampini     ierr = PetscMalloc1(stl,&l2gidxs);CHKERRQ(ierr);
10035b003df0Sstefano_zampini     for (i=0,stl=0;i<nc;i++) {
10045e3038f0Sstefano_zampini       Mat            usedmat;
10055e3038f0Sstefano_zampini       Mat_IS         *matis;
10065e3038f0Sstefano_zampini       const PetscInt *idxs;
10075e3038f0Sstefano_zampini 
10085e3038f0Sstefano_zampini       /* local IS for local NEST */
10095b003df0Sstefano_zampini       ierr  = ISCreateStride(PETSC_COMM_SELF,lc[i],stl,1,&islcol[i]);CHKERRQ(ierr);
10105e3038f0Sstefano_zampini 
10115e3038f0Sstefano_zampini       /* l2gmap */
10125e3038f0Sstefano_zampini       j = 0;
10135e3038f0Sstefano_zampini       usedmat = nest[j][i];
10149e7b2b25Sstefano_zampini       while (!usedmat && j < nr-1) usedmat = nest[++j][i];
10159e7b2b25Sstefano_zampini       if (!usedmat) SETERRQ(comm,PETSC_ERR_SUP,"Cannot find valid column mat");
10169e7b2b25Sstefano_zampini       if (istrans[j*nc+i]) {
10179e7b2b25Sstefano_zampini         Mat T;
10189e7b2b25Sstefano_zampini         ierr    = MatTransposeGetMat(usedmat,&T);CHKERRQ(ierr);
10199e7b2b25Sstefano_zampini         usedmat = T;
10209e7b2b25Sstefano_zampini       }
10215e3038f0Sstefano_zampini       matis = (Mat_IS*)(usedmat->data);
10225e3038f0Sstefano_zampini       ierr  = ISGetIndices(iscol[i],&idxs);CHKERRQ(ierr);
10239e7b2b25Sstefano_zampini       if (istrans[j*nc+i]) {
1024ad227feaSJunchao Zhang         ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
1025ad227feaSJunchao Zhang         ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
10269e7b2b25Sstefano_zampini       } else {
1027ad227feaSJunchao Zhang         ierr = PetscSFBcastBegin(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
1028ad227feaSJunchao Zhang         ierr = PetscSFBcastEnd(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr);
10299e7b2b25Sstefano_zampini       }
10305e3038f0Sstefano_zampini       ierr = ISRestoreIndices(iscol[i],&idxs);CHKERRQ(ierr);
10315e3038f0Sstefano_zampini       stl += lc[i];
10325e3038f0Sstefano_zampini     }
10335e3038f0Sstefano_zampini     ierr = ISLocalToGlobalMappingCreate(comm,1,stl,l2gidxs,PETSC_OWN_POINTER,&cl2g);CHKERRQ(ierr);
10345e3038f0Sstefano_zampini 
10355e3038f0Sstefano_zampini     /* Create MATIS */
10365e3038f0Sstefano_zampini     ierr = MatCreate(comm,&B);CHKERRQ(ierr);
10375e3038f0Sstefano_zampini     ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
10385e3038f0Sstefano_zampini     ierr = MatGetBlockSizes(A,&rbs,&cbs);CHKERRQ(ierr);
10395e3038f0Sstefano_zampini     ierr = MatSetBlockSizes(B,rbs,cbs);CHKERRQ(ierr);
10405e3038f0Sstefano_zampini     ierr = MatSetType(B,MATIS);CHKERRQ(ierr);
10418546b261SStefano Zampini     ierr = MatISSetLocalMatType(B,MATNEST);CHKERRQ(ierr);
10428546b261SStefano Zampini     { /* hack : avoid setup of scatters */
10438546b261SStefano Zampini       Mat_IS *matis = (Mat_IS*)(B->data);
10448546b261SStefano Zampini       matis->islocalref = PETSC_TRUE;
10458546b261SStefano Zampini     }
10465e3038f0Sstefano_zampini     ierr = MatSetLocalToGlobalMapping(B,rl2g,cl2g);CHKERRQ(ierr);
10475e3038f0Sstefano_zampini     ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
10485e3038f0Sstefano_zampini     ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
10495e3038f0Sstefano_zampini     ierr = MatCreateNest(PETSC_COMM_SELF,nr,islrow,nc,islcol,snest,&lA);CHKERRQ(ierr);
10508546b261SStefano Zampini     ierr = MatNestSetVecType(lA,VECNEST);CHKERRQ(ierr);
10519e7b2b25Sstefano_zampini     for (i=0;i<nr*nc;i++) {
10529e7b2b25Sstefano_zampini       if (istrans[i]) {
10539e7b2b25Sstefano_zampini         ierr = MatDestroy(&snest[i]);CHKERRQ(ierr);
10549e7b2b25Sstefano_zampini       }
10559e7b2b25Sstefano_zampini     }
10565e3038f0Sstefano_zampini     ierr = MatISSetLocalMat(B,lA);CHKERRQ(ierr);
10575e3038f0Sstefano_zampini     ierr = MatDestroy(&lA);CHKERRQ(ierr);
10588546b261SStefano Zampini     { /* hack : setup of scatters done here */
10598546b261SStefano Zampini       Mat_IS *matis = (Mat_IS*)(B->data);
10608546b261SStefano Zampini 
10618546b261SStefano Zampini       matis->islocalref = PETSC_FALSE;
10628546b261SStefano Zampini       ierr = MatISSetUpScatters_Private(B);CHKERRQ(ierr);
10638546b261SStefano Zampini     }
10645e3038f0Sstefano_zampini     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10655e3038f0Sstefano_zampini     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10665e3038f0Sstefano_zampini     if (reuse == MAT_INPLACE_MATRIX) {
10675e3038f0Sstefano_zampini       ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr);
10685e3038f0Sstefano_zampini     } else {
10695e3038f0Sstefano_zampini       *newmat = B;
10705e3038f0Sstefano_zampini     }
10715e3038f0Sstefano_zampini   } else {
10725e3038f0Sstefano_zampini     if (lreuse) {
10735e3038f0Sstefano_zampini       ierr = MatISGetLocalMat(*newmat,&lA);CHKERRQ(ierr);
10745e3038f0Sstefano_zampini       for (i=0;i<nr;i++) {
10755e3038f0Sstefano_zampini         for (j=0;j<nc;j++) {
10765e3038f0Sstefano_zampini           if (snest[i*nc+j]) {
10775e3038f0Sstefano_zampini             ierr = MatNestSetSubMat(lA,i,j,snest[i*nc+j]);CHKERRQ(ierr);
10789e7b2b25Sstefano_zampini             if (istrans[i*nc+j]) {
10799e7b2b25Sstefano_zampini               ierr = MatDestroy(&snest[i*nc+j]);CHKERRQ(ierr);
10809e7b2b25Sstefano_zampini             }
10815e3038f0Sstefano_zampini           }
10825e3038f0Sstefano_zampini         }
10835e3038f0Sstefano_zampini       }
10845e3038f0Sstefano_zampini     } else {
10855b003df0Sstefano_zampini       PetscInt stl;
10865b003df0Sstefano_zampini       for (i=0,stl=0;i<nr;i++) {
10875b003df0Sstefano_zampini         ierr  = ISCreateStride(PETSC_COMM_SELF,lr[i],stl,1,&islrow[i]);CHKERRQ(ierr);
10885b003df0Sstefano_zampini         stl  += lr[i];
10895e3038f0Sstefano_zampini       }
10905b003df0Sstefano_zampini       for (i=0,stl=0;i<nc;i++) {
10915b003df0Sstefano_zampini         ierr  = ISCreateStride(PETSC_COMM_SELF,lc[i],stl,1,&islcol[i]);CHKERRQ(ierr);
10925b003df0Sstefano_zampini         stl  += lc[i];
10935e3038f0Sstefano_zampini       }
10945e3038f0Sstefano_zampini       ierr = MatCreateNest(PETSC_COMM_SELF,nr,islrow,nc,islcol,snest,&lA);CHKERRQ(ierr);
1095ab4d48faSStefano Zampini       for (i=0;i<nr*nc;i++) {
10969e7b2b25Sstefano_zampini         if (istrans[i]) {
10979e7b2b25Sstefano_zampini           ierr = MatDestroy(&snest[i]);CHKERRQ(ierr);
10989e7b2b25Sstefano_zampini         }
1099ab4d48faSStefano Zampini       }
11005e3038f0Sstefano_zampini       ierr = MatISSetLocalMat(*newmat,lA);CHKERRQ(ierr);
11015e3038f0Sstefano_zampini       ierr = MatDestroy(&lA);CHKERRQ(ierr);
11025e3038f0Sstefano_zampini     }
11035e3038f0Sstefano_zampini     ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11045e3038f0Sstefano_zampini     ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11055e3038f0Sstefano_zampini   }
11065e3038f0Sstefano_zampini 
11075b003df0Sstefano_zampini   /* Create local matrix in MATNEST format */
11085b003df0Sstefano_zampini   convert = PETSC_FALSE;
11095b003df0Sstefano_zampini   ierr = PetscOptionsGetBool(NULL,((PetscObject)A)->prefix,"-matis_convert_local_nest",&convert,NULL);CHKERRQ(ierr);
11105b003df0Sstefano_zampini   if (convert) {
11115b003df0Sstefano_zampini     Mat              M;
11125b003df0Sstefano_zampini     MatISLocalFields lf;
11135b003df0Sstefano_zampini     PetscContainer   c;
11145b003df0Sstefano_zampini 
11155b003df0Sstefano_zampini     ierr = MatISGetLocalMat(*newmat,&lA);CHKERRQ(ierr);
11165b003df0Sstefano_zampini     ierr = MatConvert(lA,MATAIJ,MAT_INITIAL_MATRIX,&M);CHKERRQ(ierr);
11175b003df0Sstefano_zampini     ierr = MatISSetLocalMat(*newmat,M);CHKERRQ(ierr);
11185b003df0Sstefano_zampini     ierr = MatDestroy(&M);CHKERRQ(ierr);
11195b003df0Sstefano_zampini 
11205b003df0Sstefano_zampini     /* attach local fields to the matrix */
11215b003df0Sstefano_zampini     ierr = PetscNew(&lf);CHKERRQ(ierr);
1122071fcb05SBarry Smith     ierr = PetscMalloc2(nr,&lf->rf,nc,&lf->cf);CHKERRQ(ierr);
11235b003df0Sstefano_zampini     for (i=0;i<nr;i++) {
11245b003df0Sstefano_zampini       PetscInt n,st;
11255b003df0Sstefano_zampini 
11265b003df0Sstefano_zampini       ierr = ISGetLocalSize(islrow[i],&n);CHKERRQ(ierr);
11275b003df0Sstefano_zampini       ierr = ISStrideGetInfo(islrow[i],&st,NULL);CHKERRQ(ierr);
11285b003df0Sstefano_zampini       ierr = ISCreateStride(comm,n,st,1,&lf->rf[i]);CHKERRQ(ierr);
11295b003df0Sstefano_zampini     }
11305b003df0Sstefano_zampini     for (i=0;i<nc;i++) {
11315b003df0Sstefano_zampini       PetscInt n,st;
11325b003df0Sstefano_zampini 
11335b003df0Sstefano_zampini       ierr = ISGetLocalSize(islcol[i],&n);CHKERRQ(ierr);
11345b003df0Sstefano_zampini       ierr = ISStrideGetInfo(islcol[i],&st,NULL);CHKERRQ(ierr);
11355b003df0Sstefano_zampini       ierr = ISCreateStride(comm,n,st,1,&lf->cf[i]);CHKERRQ(ierr);
11365b003df0Sstefano_zampini     }
11375b003df0Sstefano_zampini     lf->nr = nr;
11385b003df0Sstefano_zampini     lf->nc = nc;
11395b003df0Sstefano_zampini     ierr = PetscContainerCreate(PetscObjectComm((PetscObject)(*newmat)),&c);CHKERRQ(ierr);
11405b003df0Sstefano_zampini     ierr = PetscContainerSetPointer(c,lf);CHKERRQ(ierr);
11415b003df0Sstefano_zampini     ierr = PetscContainerSetUserDestroy(c,MatISContainerDestroyFields_Private);CHKERRQ(ierr);
11425b003df0Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)(*newmat),"_convert_nest_lfields",(PetscObject)c);CHKERRQ(ierr);
11435b003df0Sstefano_zampini     ierr = PetscContainerDestroy(&c);CHKERRQ(ierr);
11445b003df0Sstefano_zampini   }
11455b003df0Sstefano_zampini 
11465e3038f0Sstefano_zampini   /* Free workspace */
11475e3038f0Sstefano_zampini   for (i=0;i<nr;i++) {
11485e3038f0Sstefano_zampini     ierr = ISDestroy(&islrow[i]);CHKERRQ(ierr);
11495e3038f0Sstefano_zampini   }
11505e3038f0Sstefano_zampini   for (i=0;i<nc;i++) {
11515e3038f0Sstefano_zampini     ierr = ISDestroy(&islcol[i]);CHKERRQ(ierr);
11525e3038f0Sstefano_zampini   }
11539e7b2b25Sstefano_zampini   ierr = PetscFree6(isrow,iscol,islrow,islcol,snest,istrans);CHKERRQ(ierr);
11545b003df0Sstefano_zampini   ierr = PetscFree2(lr,lc);CHKERRQ(ierr);
11555e3038f0Sstefano_zampini   PetscFunctionReturn(0);
11565e3038f0Sstefano_zampini }
11575e3038f0Sstefano_zampini 
1158ad219c80Sstefano_zampini static PetscErrorCode MatDiagonalScale_IS(Mat A, Vec l, Vec r)
1159ad219c80Sstefano_zampini {
1160ad219c80Sstefano_zampini   Mat_IS            *matis = (Mat_IS*)A->data;
1161ad219c80Sstefano_zampini   Vec               ll,rr;
1162ad219c80Sstefano_zampini   const PetscScalar *Y,*X;
1163ad219c80Sstefano_zampini   PetscScalar       *x,*y;
1164ad219c80Sstefano_zampini   PetscErrorCode    ierr;
1165ad219c80Sstefano_zampini 
1166ad219c80Sstefano_zampini   PetscFunctionBegin;
1167ad219c80Sstefano_zampini   if (l) {
1168ad219c80Sstefano_zampini     ll   = matis->y;
1169ad219c80Sstefano_zampini     ierr = VecGetArrayRead(l,&Y);CHKERRQ(ierr);
1170ad219c80Sstefano_zampini     ierr = VecGetArray(ll,&y);CHKERRQ(ierr);
1171ad227feaSJunchao Zhang     ierr = PetscSFBcastBegin(matis->sf,MPIU_SCALAR,Y,y,MPI_REPLACE);CHKERRQ(ierr);
1172ad219c80Sstefano_zampini   } else {
1173ad219c80Sstefano_zampini     ll = NULL;
1174ad219c80Sstefano_zampini   }
1175ad219c80Sstefano_zampini   if (r) {
1176ad219c80Sstefano_zampini     rr   = matis->x;
1177ad219c80Sstefano_zampini     ierr = VecGetArrayRead(r,&X);CHKERRQ(ierr);
1178ad219c80Sstefano_zampini     ierr = VecGetArray(rr,&x);CHKERRQ(ierr);
1179ad227feaSJunchao Zhang     ierr = PetscSFBcastBegin(matis->csf,MPIU_SCALAR,X,x,MPI_REPLACE);CHKERRQ(ierr);
1180ad219c80Sstefano_zampini   } else {
1181ad219c80Sstefano_zampini     rr = NULL;
1182ad219c80Sstefano_zampini   }
1183ad219c80Sstefano_zampini   if (ll) {
1184ad227feaSJunchao Zhang     ierr = PetscSFBcastEnd(matis->sf,MPIU_SCALAR,Y,y,MPI_REPLACE);CHKERRQ(ierr);
1185ad219c80Sstefano_zampini     ierr = VecRestoreArrayRead(l,&Y);CHKERRQ(ierr);
1186ad219c80Sstefano_zampini     ierr = VecRestoreArray(ll,&y);CHKERRQ(ierr);
1187ad219c80Sstefano_zampini   }
1188ad219c80Sstefano_zampini   if (rr) {
1189ad227feaSJunchao Zhang     ierr = PetscSFBcastEnd(matis->csf,MPIU_SCALAR,X,x,MPI_REPLACE);CHKERRQ(ierr);
1190ad219c80Sstefano_zampini     ierr = VecRestoreArrayRead(r,&X);CHKERRQ(ierr);
1191ad219c80Sstefano_zampini     ierr = VecRestoreArray(rr,&x);CHKERRQ(ierr);
1192ad219c80Sstefano_zampini   }
1193ad219c80Sstefano_zampini   ierr = MatDiagonalScale(matis->A,ll,rr);CHKERRQ(ierr);
1194ad219c80Sstefano_zampini   PetscFunctionReturn(0);
1195ad219c80Sstefano_zampini }
1196ad219c80Sstefano_zampini 
11977fa8f2d3SStefano Zampini static PetscErrorCode MatGetInfo_IS(Mat A,MatInfoType flag,MatInfo *ginfo)
11987fa8f2d3SStefano Zampini {
11997fa8f2d3SStefano Zampini   Mat_IS         *matis = (Mat_IS*)A->data;
12007fa8f2d3SStefano Zampini   MatInfo        info;
12013966268fSBarry Smith   PetscLogDouble isend[6],irecv[6];
12027fa8f2d3SStefano Zampini   PetscInt       bs;
12037fa8f2d3SStefano Zampini   PetscErrorCode ierr;
12047fa8f2d3SStefano Zampini 
12057fa8f2d3SStefano Zampini   PetscFunctionBegin;
12067fa8f2d3SStefano Zampini   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
1207a2ccb5f9Sstefano_zampini   if (matis->A->ops->getinfo) {
12087fa8f2d3SStefano Zampini     ierr     = MatGetInfo(matis->A,MAT_LOCAL,&info);CHKERRQ(ierr);
12097fa8f2d3SStefano Zampini     isend[0] = info.nz_used;
12107fa8f2d3SStefano Zampini     isend[1] = info.nz_allocated;
12117fa8f2d3SStefano Zampini     isend[2] = info.nz_unneeded;
12127fa8f2d3SStefano Zampini     isend[3] = info.memory;
12137fa8f2d3SStefano Zampini     isend[4] = info.mallocs;
1214a2ccb5f9Sstefano_zampini   } else {
1215a2ccb5f9Sstefano_zampini     isend[0] = 0.;
1216a2ccb5f9Sstefano_zampini     isend[1] = 0.;
1217a2ccb5f9Sstefano_zampini     isend[2] = 0.;
1218a2ccb5f9Sstefano_zampini     isend[3] = 0.;
1219a2ccb5f9Sstefano_zampini     isend[4] = 0.;
1220a2ccb5f9Sstefano_zampini   }
1221314ce898Sstefano_zampini   isend[5] = matis->A->num_ass;
12227fa8f2d3SStefano Zampini   if (flag == MAT_LOCAL) {
12237fa8f2d3SStefano Zampini     ginfo->nz_used      = isend[0];
12247fa8f2d3SStefano Zampini     ginfo->nz_allocated = isend[1];
12257fa8f2d3SStefano Zampini     ginfo->nz_unneeded  = isend[2];
12267fa8f2d3SStefano Zampini     ginfo->memory       = isend[3];
12277fa8f2d3SStefano Zampini     ginfo->mallocs      = isend[4];
1228314ce898Sstefano_zampini     ginfo->assemblies   = isend[5];
12297fa8f2d3SStefano Zampini   } else if (flag == MAT_GLOBAL_MAX) {
1230820f2d46SBarry Smith     ierr = MPIU_Allreduce(isend,irecv,6,MPIU_PETSCLOGDOUBLE,MPI_MAX,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
12317fa8f2d3SStefano Zampini 
12327fa8f2d3SStefano Zampini     ginfo->nz_used      = irecv[0];
12337fa8f2d3SStefano Zampini     ginfo->nz_allocated = irecv[1];
12347fa8f2d3SStefano Zampini     ginfo->nz_unneeded  = irecv[2];
12357fa8f2d3SStefano Zampini     ginfo->memory       = irecv[3];
12367fa8f2d3SStefano Zampini     ginfo->mallocs      = irecv[4];
1237314ce898Sstefano_zampini     ginfo->assemblies   = irecv[5];
12387fa8f2d3SStefano Zampini   } else if (flag == MAT_GLOBAL_SUM) {
1239820f2d46SBarry Smith     ierr = MPIU_Allreduce(isend,irecv,5,MPIU_PETSCLOGDOUBLE,MPI_SUM,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
12407fa8f2d3SStefano Zampini 
12417fa8f2d3SStefano Zampini     ginfo->nz_used      = irecv[0];
12427fa8f2d3SStefano Zampini     ginfo->nz_allocated = irecv[1];
12437fa8f2d3SStefano Zampini     ginfo->nz_unneeded  = irecv[2];
12447fa8f2d3SStefano Zampini     ginfo->memory       = irecv[3];
12457fa8f2d3SStefano Zampini     ginfo->mallocs      = irecv[4];
12467fa8f2d3SStefano Zampini     ginfo->assemblies   = A->num_ass;
12477fa8f2d3SStefano Zampini   }
12487fa8f2d3SStefano Zampini   ginfo->block_size        = bs;
12497fa8f2d3SStefano Zampini   ginfo->fill_ratio_given  = 0;
12507fa8f2d3SStefano Zampini   ginfo->fill_ratio_needed = 0;
12517fa8f2d3SStefano Zampini   ginfo->factor_mallocs    = 0;
12525e3038f0Sstefano_zampini   PetscFunctionReturn(0);
12535e3038f0Sstefano_zampini }
12545e3038f0Sstefano_zampini 
1255d7f69cd0SStefano Zampini PetscErrorCode MatTranspose_IS(Mat A,MatReuse reuse,Mat *B)
1256d7f69cd0SStefano Zampini {
1257d7f69cd0SStefano Zampini   Mat                    C,lC,lA;
1258d7f69cd0SStefano Zampini   PetscErrorCode         ierr;
1259d7f69cd0SStefano Zampini 
1260d7f69cd0SStefano Zampini   PetscFunctionBegin;
1261cf37664fSBarry Smith   if (reuse == MAT_INITIAL_MATRIX || reuse == MAT_INPLACE_MATRIX) {
1262cf37664fSBarry Smith     ISLocalToGlobalMapping rl2g,cl2g;
1263d7f69cd0SStefano Zampini     ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
1264d7f69cd0SStefano Zampini     ierr = MatSetSizes(C,A->cmap->n,A->rmap->n,A->cmap->N,A->rmap->N);CHKERRQ(ierr);
1265d7f69cd0SStefano Zampini     ierr = MatSetBlockSizes(C,PetscAbs(A->cmap->bs),PetscAbs(A->rmap->bs));CHKERRQ(ierr);
1266d7f69cd0SStefano Zampini     ierr = MatSetType(C,MATIS);CHKERRQ(ierr);
1267d7f69cd0SStefano Zampini     ierr = MatGetLocalToGlobalMapping(A,&rl2g,&cl2g);CHKERRQ(ierr);
1268d7f69cd0SStefano Zampini     ierr = MatSetLocalToGlobalMapping(C,cl2g,rl2g);CHKERRQ(ierr);
1269cf37664fSBarry Smith   } else {
1270cf37664fSBarry Smith     C = *B;
1271d7f69cd0SStefano Zampini   }
1272d7f69cd0SStefano Zampini 
1273d7f69cd0SStefano Zampini   /* perform local transposition */
1274d7f69cd0SStefano Zampini   ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr);
1275d7f69cd0SStefano Zampini   ierr = MatTranspose(lA,MAT_INITIAL_MATRIX,&lC);CHKERRQ(ierr);
1276d7f69cd0SStefano Zampini   ierr = MatISSetLocalMat(C,lC);CHKERRQ(ierr);
1277d7f69cd0SStefano Zampini   ierr = MatDestroy(&lC);CHKERRQ(ierr);
1278d7f69cd0SStefano Zampini 
1279cf37664fSBarry Smith   if (reuse == MAT_INITIAL_MATRIX || reuse == MAT_REUSE_MATRIX) {
1280d7f69cd0SStefano Zampini     *B = C;
1281d7f69cd0SStefano Zampini   } else {
1282d7f69cd0SStefano Zampini     ierr = MatHeaderMerge(A,&C);CHKERRQ(ierr);
1283d7f69cd0SStefano Zampini   }
12847aa7aec5Sstefano_zampini   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
12857aa7aec5Sstefano_zampini   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1286d7f69cd0SStefano Zampini   PetscFunctionReturn(0);
1287d7f69cd0SStefano Zampini }
1288d7f69cd0SStefano Zampini 
12893fd1c9e7SStefano Zampini PetscErrorCode  MatDiagonalSet_IS(Mat A,Vec D,InsertMode insmode)
12903fd1c9e7SStefano Zampini {
12913fd1c9e7SStefano Zampini   Mat_IS         *is = (Mat_IS*)A->data;
12923fd1c9e7SStefano Zampini   PetscErrorCode ierr;
12933fd1c9e7SStefano Zampini 
12943fd1c9e7SStefano Zampini   PetscFunctionBegin;
12954b89b9cdSStefano Zampini   if (D) { /* MatShift_IS pass D = NULL */
12963fd1c9e7SStefano Zampini     ierr = VecScatterBegin(is->rctx,D,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12973fd1c9e7SStefano Zampini     ierr = VecScatterEnd(is->rctx,D,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12983fd1c9e7SStefano Zampini   }
12993fd1c9e7SStefano Zampini   ierr = VecPointwiseDivide(is->y,is->y,is->counter);CHKERRQ(ierr);
13003fd1c9e7SStefano Zampini   ierr = MatDiagonalSet(is->A,is->y,insmode);CHKERRQ(ierr);
13013fd1c9e7SStefano Zampini   PetscFunctionReturn(0);
13023fd1c9e7SStefano Zampini }
13033fd1c9e7SStefano Zampini 
13043fd1c9e7SStefano Zampini PetscErrorCode  MatShift_IS(Mat A,PetscScalar a)
13053fd1c9e7SStefano Zampini {
13064b89b9cdSStefano Zampini   Mat_IS         *is = (Mat_IS*)A->data;
13073fd1c9e7SStefano Zampini   PetscErrorCode ierr;
13083fd1c9e7SStefano Zampini 
13093fd1c9e7SStefano Zampini   PetscFunctionBegin;
13104b89b9cdSStefano Zampini   ierr = VecSet(is->y,a);CHKERRQ(ierr);
13113fd1c9e7SStefano Zampini   ierr = MatDiagonalSet_IS(A,NULL,ADD_VALUES);CHKERRQ(ierr);
13123fd1c9e7SStefano Zampini   PetscFunctionReturn(0);
13133fd1c9e7SStefano Zampini }
13143fd1c9e7SStefano Zampini 
1315f26d0771SStefano Zampini static PetscErrorCode MatSetValuesLocal_SubMat_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv)
1316f26d0771SStefano Zampini {
1317f26d0771SStefano Zampini   PetscErrorCode ierr;
1318f26d0771SStefano Zampini   PetscInt       rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION];
1319f26d0771SStefano Zampini 
1320f26d0771SStefano Zampini   PetscFunctionBegin;
132176bd3646SJed Brown   if (PetscUnlikely(m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION)) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n);
1322f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingApply(A->rmap->mapping,m,rows,rows_l);CHKERRQ(ierr);
1323f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingApply(A->cmap->mapping,n,cols,cols_l);CHKERRQ(ierr);
1324b4f971dfSStefano Zampini   ierr = MatSetValuesLocal_IS(A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr);
1325f26d0771SStefano Zampini   PetscFunctionReturn(0);
1326f26d0771SStefano Zampini }
1327f26d0771SStefano Zampini 
1328f26d0771SStefano Zampini static PetscErrorCode MatSetValuesBlockedLocal_SubMat_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv)
1329f26d0771SStefano Zampini {
1330f26d0771SStefano Zampini   PetscErrorCode ierr;
1331f26d0771SStefano Zampini   PetscInt       rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION];
1332f26d0771SStefano Zampini 
1333f26d0771SStefano Zampini   PetscFunctionBegin;
133476bd3646SJed Brown   if (PetscUnlikely(m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION)) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column block indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n);
1335f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingApplyBlock(A->rmap->mapping,m,rows,rows_l);CHKERRQ(ierr);
1336f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingApplyBlock(A->cmap->mapping,n,cols,cols_l);CHKERRQ(ierr);
1337b4f971dfSStefano Zampini   ierr = MatSetValuesBlockedLocal_IS(A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr);
1338f26d0771SStefano Zampini   PetscFunctionReturn(0);
1339f26d0771SStefano Zampini }
1340f26d0771SStefano Zampini 
13417dae84e0SHong Zhang static PetscErrorCode MatCreateSubMatrix_IS(Mat mat,IS irow,IS icol,MatReuse scall,Mat *newmat)
1342a8116848SStefano Zampini {
1343a8116848SStefano Zampini   Mat               locmat,newlocmat;
1344a8116848SStefano Zampini   Mat_IS            *newmatis;
1345a8116848SStefano Zampini   const PetscInt    *idxs;
1346a8116848SStefano Zampini   PetscInt          i,m,n;
1347a8116848SStefano Zampini   PetscErrorCode    ierr;
1348a8116848SStefano Zampini 
1349a8116848SStefano Zampini   PetscFunctionBegin;
1350a8116848SStefano Zampini   if (scall == MAT_REUSE_MATRIX) {
1351a8116848SStefano Zampini     PetscBool ismatis;
1352a8116848SStefano Zampini 
1353a8116848SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)*newmat,MATIS,&ismatis);CHKERRQ(ierr);
1354a8116848SStefano Zampini     if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_ARG_WRONG,"Cannot reuse matrix! Not of MATIS type");
1355a8116848SStefano Zampini     newmatis = (Mat_IS*)(*newmat)->data;
1356a8116848SStefano Zampini     if (!newmatis->getsub_ris) SETERRQ(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_ARG_WRONG,"Cannot reuse matrix! Misses local row IS");
1357a8116848SStefano Zampini     if (!newmatis->getsub_cis) SETERRQ(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_ARG_WRONG,"Cannot reuse matrix! Misses local col IS");
1358a8116848SStefano Zampini   }
1359a8116848SStefano Zampini   /* irow and icol may not have duplicate entries */
136076bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
136176bd3646SJed Brown     Vec               rtest,ltest;
136276bd3646SJed Brown     const PetscScalar *array;
136376bd3646SJed Brown 
1364a8116848SStefano Zampini     ierr = MatCreateVecs(mat,&ltest,&rtest);CHKERRQ(ierr);
1365a8116848SStefano Zampini     ierr = ISGetLocalSize(irow,&n);CHKERRQ(ierr);
1366a8116848SStefano Zampini     ierr = ISGetIndices(irow,&idxs);CHKERRQ(ierr);
1367a8116848SStefano Zampini     for (i=0;i<n;i++) {
1368a8116848SStefano Zampini       ierr = VecSetValue(rtest,idxs[i],1.0,ADD_VALUES);CHKERRQ(ierr);
1369a8116848SStefano Zampini     }
1370a8116848SStefano Zampini     ierr = VecAssemblyBegin(rtest);CHKERRQ(ierr);
1371a8116848SStefano Zampini     ierr = VecAssemblyEnd(rtest);CHKERRQ(ierr);
1372a8116848SStefano Zampini     ierr = VecGetLocalSize(rtest,&n);CHKERRQ(ierr);
1373a8116848SStefano Zampini     ierr = VecGetOwnershipRange(rtest,&m,NULL);CHKERRQ(ierr);
1374a8116848SStefano Zampini     ierr = VecGetArrayRead(rtest,&array);CHKERRQ(ierr);
1375fd479f66SMatthew G. Knepley     for (i=0;i<n;i++) if (array[i] != 0. && array[i] != 1.) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Index %D counted %D times! Irow may not have duplicate entries",i+m,(PetscInt)PetscRealPart(array[i]));
1376a8116848SStefano Zampini     ierr = VecRestoreArrayRead(rtest,&array);CHKERRQ(ierr);
1377a8116848SStefano Zampini     ierr = ISRestoreIndices(irow,&idxs);CHKERRQ(ierr);
1378a8116848SStefano Zampini     ierr = ISGetLocalSize(icol,&n);CHKERRQ(ierr);
1379a8116848SStefano Zampini     ierr = ISGetIndices(icol,&idxs);CHKERRQ(ierr);
1380a8116848SStefano Zampini     for (i=0;i<n;i++) {
1381a8116848SStefano Zampini       ierr = VecSetValue(ltest,idxs[i],1.0,ADD_VALUES);CHKERRQ(ierr);
1382a8116848SStefano Zampini     }
1383a8116848SStefano Zampini     ierr = VecAssemblyBegin(ltest);CHKERRQ(ierr);
1384a8116848SStefano Zampini     ierr = VecAssemblyEnd(ltest);CHKERRQ(ierr);
1385a8116848SStefano Zampini     ierr = VecGetLocalSize(ltest,&n);CHKERRQ(ierr);
1386a8116848SStefano Zampini     ierr = VecGetOwnershipRange(ltest,&m,NULL);CHKERRQ(ierr);
1387a8116848SStefano Zampini     ierr = VecGetArrayRead(ltest,&array);CHKERRQ(ierr);
1388fd479f66SMatthew G. Knepley     for (i=0;i<n;i++) if (array[i] != 0. && array[i] != 1.) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Index %D counted %D times! Icol may not have duplicate entries",i+m,(PetscInt)PetscRealPart(array[i]));
1389a8116848SStefano Zampini     ierr = VecRestoreArrayRead(ltest,&array);CHKERRQ(ierr);
1390a8116848SStefano Zampini     ierr = ISRestoreIndices(icol,&idxs);CHKERRQ(ierr);
1391a8116848SStefano Zampini     ierr = VecDestroy(&rtest);CHKERRQ(ierr);
1392a8116848SStefano Zampini     ierr = VecDestroy(&ltest);CHKERRQ(ierr);
139376bd3646SJed Brown   }
1394a8116848SStefano Zampini   if (scall == MAT_INITIAL_MATRIX) {
1395a8116848SStefano Zampini     Mat_IS                 *matis = (Mat_IS*)mat->data;
1396a8116848SStefano Zampini     ISLocalToGlobalMapping rl2g;
1397a8116848SStefano Zampini     IS                     is;
1398a8116848SStefano Zampini     PetscInt               *lidxs,*lgidxs,*newgidxs;
1399306cf5c7SStefano Zampini     PetscInt               ll,newloc,irbs,icbs,arbs,acbs,rbs,cbs;
140094342113SStefano Zampini     PetscBool              cong;
1401a8116848SStefano Zampini     MPI_Comm               comm;
1402a8116848SStefano Zampini 
1403a8116848SStefano Zampini     ierr = PetscObjectGetComm((PetscObject)mat,&comm);CHKERRQ(ierr);
1404306cf5c7SStefano Zampini     ierr = MatGetBlockSizes(mat,&arbs,&acbs);CHKERRQ(ierr);
1405306cf5c7SStefano Zampini     ierr = ISGetBlockSize(irow,&irbs);CHKERRQ(ierr);
1406306cf5c7SStefano Zampini     ierr = ISGetBlockSize(icol,&icbs);CHKERRQ(ierr);
1407306cf5c7SStefano Zampini     rbs  = arbs == irbs ? irbs : 1;
1408306cf5c7SStefano Zampini     cbs  = acbs == icbs ? icbs : 1;
1409a8116848SStefano Zampini     ierr = ISGetLocalSize(irow,&m);CHKERRQ(ierr);
1410a8116848SStefano Zampini     ierr = ISGetLocalSize(icol,&n);CHKERRQ(ierr);
1411a8116848SStefano Zampini     ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1412a8116848SStefano Zampini     ierr = MatSetType(*newmat,MATIS);CHKERRQ(ierr);
1413a8116848SStefano Zampini     ierr = MatSetSizes(*newmat,m,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
1414306cf5c7SStefano Zampini     ierr = MatSetBlockSizes(*newmat,rbs,cbs);CHKERRQ(ierr);
1415a8116848SStefano Zampini     /* communicate irow to their owners in the layout */
1416a8116848SStefano Zampini     ierr = ISGetIndices(irow,&idxs);CHKERRQ(ierr);
1417a72d46e8SStefano Zampini     ierr = PetscLayoutMapLocal(mat->rmap,m,idxs,&ll,&lidxs,&lgidxs);CHKERRQ(ierr);
1418a8116848SStefano Zampini     ierr = ISRestoreIndices(irow,&idxs);CHKERRQ(ierr);
1419580bdb30SBarry Smith     ierr = PetscArrayzero(matis->sf_rootdata,matis->sf->nroots);CHKERRQ(ierr);
1420a8116848SStefano Zampini     for (i=0;i<ll;i++) matis->sf_rootdata[lidxs[i]] = lgidxs[i]+1;
1421a8116848SStefano Zampini     ierr = PetscFree(lidxs);CHKERRQ(ierr);
1422a8116848SStefano Zampini     ierr = PetscFree(lgidxs);CHKERRQ(ierr);
1423ad227feaSJunchao Zhang     ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
1424ad227feaSJunchao Zhang     ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
14253d996552SStefano Zampini     for (i=0,newloc=0;i<matis->sf->nleaves;i++) if (matis->sf_leafdata[i]) newloc++;
1426a8116848SStefano Zampini     ierr = PetscMalloc1(newloc,&newgidxs);CHKERRQ(ierr);
1427a8116848SStefano Zampini     ierr = PetscMalloc1(newloc,&lidxs);CHKERRQ(ierr);
14283d996552SStefano Zampini     for (i=0,newloc=0;i<matis->sf->nleaves;i++)
1429a8116848SStefano Zampini       if (matis->sf_leafdata[i]) {
1430a8116848SStefano Zampini         lidxs[newloc] = i;
1431a8116848SStefano Zampini         newgidxs[newloc++] = matis->sf_leafdata[i]-1;
1432a8116848SStefano Zampini       }
1433a8116848SStefano Zampini     ierr = ISCreateGeneral(comm,newloc,newgidxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
1434a8116848SStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr);
1435306cf5c7SStefano Zampini     ierr = ISLocalToGlobalMappingSetBlockSize(rl2g,rbs);CHKERRQ(ierr);
1436a8116848SStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
1437a8116848SStefano Zampini     /* local is to extract local submatrix */
1438a8116848SStefano Zampini     newmatis = (Mat_IS*)(*newmat)->data;
1439a8116848SStefano Zampini     ierr = ISCreateGeneral(comm,newloc,lidxs,PETSC_OWN_POINTER,&newmatis->getsub_ris);CHKERRQ(ierr);
144094342113SStefano Zampini     ierr = MatHasCongruentLayouts(mat,&cong);CHKERRQ(ierr);
144194342113SStefano Zampini     if (cong && irow == icol && matis->csf == matis->sf) {
1442a8116848SStefano Zampini       ierr = MatSetLocalToGlobalMapping(*newmat,rl2g,rl2g);CHKERRQ(ierr);
1443a8116848SStefano Zampini       ierr = PetscObjectReference((PetscObject)newmatis->getsub_ris);CHKERRQ(ierr);
1444a8116848SStefano Zampini       newmatis->getsub_cis = newmatis->getsub_ris;
1445a8116848SStefano Zampini     } else {
1446a8116848SStefano Zampini       ISLocalToGlobalMapping cl2g;
1447a8116848SStefano Zampini 
1448a8116848SStefano Zampini       /* communicate icol to their owners in the layout */
1449a8116848SStefano Zampini       ierr = ISGetIndices(icol,&idxs);CHKERRQ(ierr);
1450a72d46e8SStefano Zampini       ierr = PetscLayoutMapLocal(mat->cmap,n,idxs,&ll,&lidxs,&lgidxs);CHKERRQ(ierr);
1451a8116848SStefano Zampini       ierr = ISRestoreIndices(icol,&idxs);CHKERRQ(ierr);
1452580bdb30SBarry Smith       ierr = PetscArrayzero(matis->csf_rootdata,matis->csf->nroots);CHKERRQ(ierr);
1453a8116848SStefano Zampini       for (i=0;i<ll;i++) matis->csf_rootdata[lidxs[i]] = lgidxs[i]+1;
1454a8116848SStefano Zampini       ierr = PetscFree(lidxs);CHKERRQ(ierr);
1455a8116848SStefano Zampini       ierr = PetscFree(lgidxs);CHKERRQ(ierr);
1456ad227feaSJunchao Zhang       ierr = PetscSFBcastBegin(matis->csf,MPIU_INT,matis->csf_rootdata,matis->csf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
1457ad227feaSJunchao Zhang       ierr = PetscSFBcastEnd(matis->csf,MPIU_INT,matis->csf_rootdata,matis->csf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
14583d996552SStefano Zampini       for (i=0,newloc=0;i<matis->csf->nleaves;i++) if (matis->csf_leafdata[i]) newloc++;
1459a8116848SStefano Zampini       ierr = PetscMalloc1(newloc,&newgidxs);CHKERRQ(ierr);
1460a8116848SStefano Zampini       ierr = PetscMalloc1(newloc,&lidxs);CHKERRQ(ierr);
14613d996552SStefano Zampini       for (i=0,newloc=0;i<matis->csf->nleaves;i++)
1462a8116848SStefano Zampini         if (matis->csf_leafdata[i]) {
1463a8116848SStefano Zampini           lidxs[newloc] = i;
1464a8116848SStefano Zampini           newgidxs[newloc++] = matis->csf_leafdata[i]-1;
1465a8116848SStefano Zampini         }
1466a8116848SStefano Zampini       ierr = ISCreateGeneral(comm,newloc,newgidxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
1467a8116848SStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr);
1468306cf5c7SStefano Zampini       ierr = ISLocalToGlobalMappingSetBlockSize(cl2g,cbs);CHKERRQ(ierr);
1469a8116848SStefano Zampini       ierr = ISDestroy(&is);CHKERRQ(ierr);
1470a8116848SStefano Zampini       /* local is to extract local submatrix */
1471a8116848SStefano Zampini       ierr = ISCreateGeneral(comm,newloc,lidxs,PETSC_OWN_POINTER,&newmatis->getsub_cis);CHKERRQ(ierr);
1472a8116848SStefano Zampini       ierr = MatSetLocalToGlobalMapping(*newmat,rl2g,cl2g);CHKERRQ(ierr);
1473a8116848SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
1474a8116848SStefano Zampini     }
1475a8116848SStefano Zampini     ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
1476a8116848SStefano Zampini   } else {
1477a8116848SStefano Zampini     ierr = MatISGetLocalMat(*newmat,&newlocmat);CHKERRQ(ierr);
1478a8116848SStefano Zampini   }
1479a8116848SStefano Zampini   ierr = MatISGetLocalMat(mat,&locmat);CHKERRQ(ierr);
1480a8116848SStefano Zampini   newmatis = (Mat_IS*)(*newmat)->data;
14817dae84e0SHong Zhang   ierr = MatCreateSubMatrix(locmat,newmatis->getsub_ris,newmatis->getsub_cis,scall,&newlocmat);CHKERRQ(ierr);
1482a8116848SStefano Zampini   if (scall == MAT_INITIAL_MATRIX) {
1483a8116848SStefano Zampini     ierr = MatISSetLocalMat(*newmat,newlocmat);CHKERRQ(ierr);
1484a8116848SStefano Zampini     ierr = MatDestroy(&newlocmat);CHKERRQ(ierr);
1485a8116848SStefano Zampini   }
1486a8116848SStefano Zampini   ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1487a8116848SStefano Zampini   ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1488a8116848SStefano Zampini   PetscFunctionReturn(0);
1489a8116848SStefano Zampini }
1490a8116848SStefano Zampini 
1491a8116848SStefano Zampini static PetscErrorCode MatCopy_IS(Mat A,Mat B,MatStructure str)
14922b404112SStefano Zampini {
14932b404112SStefano Zampini   Mat_IS         *a = (Mat_IS*)A->data,*b;
14942b404112SStefano Zampini   PetscBool      ismatis;
14952b404112SStefano Zampini   PetscErrorCode ierr;
14962b404112SStefano Zampini 
14972b404112SStefano Zampini   PetscFunctionBegin;
14982b404112SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)B,MATIS,&ismatis);CHKERRQ(ierr);
14992b404112SStefano Zampini   if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"Need to be implemented");
15002b404112SStefano Zampini   b = (Mat_IS*)B->data;
15012b404112SStefano Zampini   ierr = MatCopy(a->A,b->A,str);CHKERRQ(ierr);
1502cdc753b6SBarry Smith   ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr);
15032b404112SStefano Zampini   PetscFunctionReturn(0);
15042b404112SStefano Zampini }
15052b404112SStefano Zampini 
1506a8116848SStefano Zampini static PetscErrorCode MatMissingDiagonal_IS(Mat A,PetscBool  *missing,PetscInt *d)
15076bd84002SStefano Zampini {
1508527b2640SStefano Zampini   Vec               v;
1509527b2640SStefano Zampini   const PetscScalar *array;
1510527b2640SStefano Zampini   PetscInt          i,n;
15116bd84002SStefano Zampini   PetscErrorCode    ierr;
15126bd84002SStefano Zampini 
15136bd84002SStefano Zampini   PetscFunctionBegin;
1514527b2640SStefano Zampini   *missing = PETSC_FALSE;
1515527b2640SStefano Zampini   ierr = MatCreateVecs(A,NULL,&v);CHKERRQ(ierr);
1516527b2640SStefano Zampini   ierr = MatGetDiagonal(A,v);CHKERRQ(ierr);
1517527b2640SStefano Zampini   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
1518527b2640SStefano Zampini   ierr = VecGetArrayRead(v,&array);CHKERRQ(ierr);
1519527b2640SStefano Zampini   for (i=0;i<n;i++) if (array[i] == 0.) break;
1520527b2640SStefano Zampini   ierr = VecRestoreArrayRead(v,&array);CHKERRQ(ierr);
1521527b2640SStefano Zampini   ierr = VecDestroy(&v);CHKERRQ(ierr);
1522527b2640SStefano Zampini   if (i != n) *missing = PETSC_TRUE;
1523527b2640SStefano Zampini   if (d) {
1524527b2640SStefano Zampini     *d = -1;
1525527b2640SStefano Zampini     if (*missing) {
1526527b2640SStefano Zampini       PetscInt rstart;
1527527b2640SStefano Zampini       ierr = MatGetOwnershipRange(A,&rstart,NULL);CHKERRQ(ierr);
1528527b2640SStefano Zampini       *d = i+rstart;
1529527b2640SStefano Zampini     }
1530527b2640SStefano Zampini   }
15316bd84002SStefano Zampini   PetscFunctionReturn(0);
15326bd84002SStefano Zampini }
15336bd84002SStefano Zampini 
1534cf0a3239SStefano Zampini static PetscErrorCode MatISSetUpSF_IS(Mat B)
153528f4e0baSStefano Zampini {
153628f4e0baSStefano Zampini   Mat_IS         *matis = (Mat_IS*)(B->data);
153728f4e0baSStefano Zampini   const PetscInt *gidxs;
15384f2d7cafSStefano Zampini   PetscInt       nleaves;
153928f4e0baSStefano Zampini   PetscErrorCode ierr;
154028f4e0baSStefano Zampini 
154128f4e0baSStefano Zampini   PetscFunctionBegin;
15424f2d7cafSStefano Zampini   if (matis->sf) PetscFunctionReturn(0);
154328f4e0baSStefano Zampini   ierr = PetscSFCreate(PetscObjectComm((PetscObject)B),&matis->sf);CHKERRQ(ierr);
15443bbff08aSStefano Zampini   ierr = ISLocalToGlobalMappingGetIndices(B->rmap->mapping,&gidxs);CHKERRQ(ierr);
15454f2d7cafSStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(B->rmap->mapping,&nleaves);CHKERRQ(ierr);
15464f2d7cafSStefano Zampini   ierr = PetscSFSetGraphLayout(matis->sf,B->rmap,nleaves,NULL,PETSC_OWN_POINTER,gidxs);CHKERRQ(ierr);
15473bbff08aSStefano Zampini   ierr = ISLocalToGlobalMappingRestoreIndices(B->rmap->mapping,&gidxs);CHKERRQ(ierr);
15484f2d7cafSStefano Zampini   ierr = PetscMalloc2(matis->sf->nroots,&matis->sf_rootdata,matis->sf->nleaves,&matis->sf_leafdata);CHKERRQ(ierr);
1549a8116848SStefano Zampini   if (B->rmap->mapping != B->cmap->mapping) { /* setup SF for columns */
15503d996552SStefano Zampini     ierr = ISLocalToGlobalMappingGetSize(B->cmap->mapping,&nleaves);CHKERRQ(ierr);
1551a8116848SStefano Zampini     ierr = PetscSFCreate(PetscObjectComm((PetscObject)B),&matis->csf);CHKERRQ(ierr);
1552a8116848SStefano Zampini     ierr = ISLocalToGlobalMappingGetIndices(B->cmap->mapping,&gidxs);CHKERRQ(ierr);
15533d996552SStefano Zampini     ierr = PetscSFSetGraphLayout(matis->csf,B->cmap,nleaves,NULL,PETSC_OWN_POINTER,gidxs);CHKERRQ(ierr);
1554a8116848SStefano Zampini     ierr = ISLocalToGlobalMappingRestoreIndices(B->cmap->mapping,&gidxs);CHKERRQ(ierr);
15553d996552SStefano Zampini     ierr = PetscMalloc2(matis->csf->nroots,&matis->csf_rootdata,matis->csf->nleaves,&matis->csf_leafdata);CHKERRQ(ierr);
1556a8116848SStefano Zampini   } else {
1557a8116848SStefano Zampini     matis->csf = matis->sf;
1558a8116848SStefano Zampini     matis->csf_leafdata = matis->sf_leafdata;
1559a8116848SStefano Zampini     matis->csf_rootdata = matis->sf_rootdata;
1560a8116848SStefano Zampini   }
156128f4e0baSStefano Zampini   PetscFunctionReturn(0);
156228f4e0baSStefano Zampini }
15632e1947a5SStefano Zampini 
1564eb82efa4SStefano Zampini /*@
156575d48cdbSStefano Zampini    MatISStoreL2L - Store local-to-local operators during the Galerkin process of MatPtAP.
156675d48cdbSStefano Zampini 
1567d083f849SBarry Smith    Collective
156875d48cdbSStefano Zampini 
156975d48cdbSStefano Zampini    Input Parameters:
157075d48cdbSStefano Zampini +  A - the matrix
157175d48cdbSStefano Zampini -  store - the boolean flag
157275d48cdbSStefano Zampini 
157375d48cdbSStefano Zampini    Level: advanced
157475d48cdbSStefano Zampini 
157575d48cdbSStefano Zampini    Notes:
157675d48cdbSStefano Zampini 
157775d48cdbSStefano Zampini .seealso: MatCreate(), MatCreateIS(), MatISSetPreallocation(), MatPtAP()
157875d48cdbSStefano Zampini @*/
157975d48cdbSStefano Zampini PetscErrorCode  MatISStoreL2L(Mat A, PetscBool store)
158075d48cdbSStefano Zampini {
158175d48cdbSStefano Zampini   PetscErrorCode ierr;
158275d48cdbSStefano Zampini 
158375d48cdbSStefano Zampini   PetscFunctionBegin;
158475d48cdbSStefano Zampini   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
158575d48cdbSStefano Zampini   PetscValidType(A,1);
158675d48cdbSStefano Zampini   PetscValidLogicalCollectiveBool(A,store,2);
158775d48cdbSStefano Zampini   ierr = PetscTryMethod(A,"MatISStoreL2L_C",(Mat,PetscBool),(A,store));CHKERRQ(ierr);
158875d48cdbSStefano Zampini   PetscFunctionReturn(0);
158975d48cdbSStefano Zampini }
159075d48cdbSStefano Zampini 
159175d48cdbSStefano Zampini static PetscErrorCode MatISStoreL2L_IS(Mat A, PetscBool store)
159275d48cdbSStefano Zampini {
159375d48cdbSStefano Zampini   Mat_IS         *matis = (Mat_IS*)(A->data);
159475d48cdbSStefano Zampini   PetscErrorCode ierr;
159575d48cdbSStefano Zampini 
159675d48cdbSStefano Zampini   PetscFunctionBegin;
159775d48cdbSStefano Zampini   matis->storel2l = store;
159875d48cdbSStefano Zampini   if (!store) {
159975d48cdbSStefano Zampini     ierr = PetscObjectCompose((PetscObject)(A),"_MatIS_PtAP_l2l",NULL);CHKERRQ(ierr);
160075d48cdbSStefano Zampini   }
160175d48cdbSStefano Zampini   PetscFunctionReturn(0);
160275d48cdbSStefano Zampini }
160375d48cdbSStefano Zampini 
160475d48cdbSStefano Zampini /*@
1605f03112d0SStefano Zampini    MatISFixLocalEmpty - Compress out zero local rows from the local matrices
1606f03112d0SStefano Zampini 
1607d083f849SBarry Smith    Collective
1608f03112d0SStefano Zampini 
1609f03112d0SStefano Zampini    Input Parameters:
1610f03112d0SStefano Zampini +  A - the matrix
1611f03112d0SStefano Zampini -  fix - the boolean flag
1612f03112d0SStefano Zampini 
1613f03112d0SStefano Zampini    Level: advanced
1614f03112d0SStefano Zampini 
1615f03112d0SStefano Zampini    Notes: When fix is true, new local matrices and l2g maps are generated during the final assembly process.
1616f03112d0SStefano Zampini 
1617f03112d0SStefano Zampini .seealso: MatCreate(), MatCreateIS(), MatISSetPreallocation(), MatAssemblyEnd(), MAT_FINAL_ASSEMBLY
1618f03112d0SStefano Zampini @*/
1619f03112d0SStefano Zampini PetscErrorCode  MatISFixLocalEmpty(Mat A, PetscBool fix)
1620f03112d0SStefano Zampini {
1621f03112d0SStefano Zampini   PetscErrorCode ierr;
1622f03112d0SStefano Zampini 
1623f03112d0SStefano Zampini   PetscFunctionBegin;
1624f03112d0SStefano Zampini   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
1625f03112d0SStefano Zampini   PetscValidType(A,1);
1626f03112d0SStefano Zampini   PetscValidLogicalCollectiveBool(A,fix,2);
1627f03112d0SStefano Zampini   ierr = PetscTryMethod(A,"MatISFixLocalEmpty_C",(Mat,PetscBool),(A,fix));CHKERRQ(ierr);
1628f03112d0SStefano Zampini   PetscFunctionReturn(0);
1629f03112d0SStefano Zampini }
1630f03112d0SStefano Zampini 
1631f03112d0SStefano Zampini static PetscErrorCode MatISFixLocalEmpty_IS(Mat A, PetscBool fix)
1632f03112d0SStefano Zampini {
1633f03112d0SStefano Zampini   Mat_IS *matis = (Mat_IS*)(A->data);
1634f03112d0SStefano Zampini 
1635f03112d0SStefano Zampini   PetscFunctionBegin;
1636f03112d0SStefano Zampini   matis->locempty = fix;
1637f03112d0SStefano Zampini   PetscFunctionReturn(0);
1638f03112d0SStefano Zampini }
1639f03112d0SStefano Zampini 
1640f03112d0SStefano Zampini /*@
1641a88811baSStefano Zampini    MatISSetPreallocation - Preallocates memory for a MATIS parallel matrix.
1642a88811baSStefano Zampini 
1643d083f849SBarry Smith    Collective
1644a88811baSStefano Zampini 
1645a88811baSStefano Zampini    Input Parameters:
1646a88811baSStefano Zampini +  B - the matrix
1647a88811baSStefano Zampini .  d_nz  - number of nonzeros per row in DIAGONAL portion of local submatrix
1648a88811baSStefano Zampini            (same value is used for all local rows)
1649a88811baSStefano Zampini .  d_nnz - array containing the number of nonzeros in the various rows of the
1650a88811baSStefano Zampini            DIAGONAL portion of the local submatrix (possibly different for each row)
1651a88811baSStefano Zampini            or NULL, if d_nz is used to specify the nonzero structure.
1652a88811baSStefano Zampini            The size of this array is equal to the number of local rows, i.e 'm'.
1653a88811baSStefano Zampini            For matrices that will be factored, you must leave room for (and set)
1654a88811baSStefano Zampini            the diagonal entry even if it is zero.
1655a88811baSStefano Zampini .  o_nz  - number of nonzeros per row in the OFF-DIAGONAL portion of local
1656a88811baSStefano Zampini            submatrix (same value is used for all local rows).
1657a88811baSStefano Zampini -  o_nnz - array containing the number of nonzeros in the various rows of the
1658a88811baSStefano Zampini            OFF-DIAGONAL portion of the local submatrix (possibly different for
1659a88811baSStefano Zampini            each row) or NULL, if o_nz is used to specify the nonzero
1660a88811baSStefano Zampini            structure. The size of this array is equal to the number
1661a88811baSStefano Zampini            of local rows, i.e 'm'.
1662a88811baSStefano Zampini 
1663a88811baSStefano Zampini    If the *_nnz parameter is given then the *_nz parameter is ignored
1664a88811baSStefano Zampini 
1665a88811baSStefano Zampini    Level: intermediate
1666a88811baSStefano Zampini 
166795452b02SPatrick Sanan    Notes:
166895452b02SPatrick Sanan     This function has the same interface as the MPIAIJ preallocation routine in order to simplify the transition
1669a88811baSStefano Zampini           from the asssembled format to the unassembled one. It overestimates the preallocation of MATIS local
1670a88811baSStefano Zampini           matrices; for exact preallocation, the user should set the preallocation directly on local matrix objects.
1671a88811baSStefano Zampini 
16723c212e90SHong Zhang .seealso: MatCreate(), MatCreateIS(), MatMPIAIJSetPreallocation(), MatISGetLocalMat(), MATIS
1673a88811baSStefano Zampini @*/
16742e1947a5SStefano Zampini PetscErrorCode  MatISSetPreallocation(Mat B,PetscInt d_nz,const PetscInt d_nnz[],PetscInt o_nz,const PetscInt o_nnz[])
16752e1947a5SStefano Zampini {
16762e1947a5SStefano Zampini   PetscErrorCode ierr;
16772e1947a5SStefano Zampini 
16782e1947a5SStefano Zampini   PetscFunctionBegin;
16792e1947a5SStefano Zampini   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
16802e1947a5SStefano Zampini   PetscValidType(B,1);
16812e1947a5SStefano Zampini   ierr = PetscTryMethod(B,"MatISSetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(B,d_nz,d_nnz,o_nz,o_nnz));CHKERRQ(ierr);
16822e1947a5SStefano Zampini   PetscFunctionReturn(0);
16832e1947a5SStefano Zampini }
16842e1947a5SStefano Zampini 
1685844bd0d7SStefano Zampini /* this is used by DMDA */
1686844bd0d7SStefano Zampini PETSC_EXTERN PetscErrorCode  MatISSetPreallocation_IS(Mat B,PetscInt d_nz,const PetscInt d_nnz[],PetscInt o_nz,const PetscInt o_nnz[])
16872e1947a5SStefano Zampini {
16882e1947a5SStefano Zampini   Mat_IS         *matis = (Mat_IS*)(B->data);
168928f4e0baSStefano Zampini   PetscInt       bs,i,nlocalcols;
16902e1947a5SStefano Zampini   PetscErrorCode ierr;
16912e1947a5SStefano Zampini 
16922e1947a5SStefano Zampini   PetscFunctionBegin;
16936c4ed002SBarry Smith   if (!matis->A) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"You should first call MatSetLocalToGlobalMapping");
16944f2d7cafSStefano Zampini 
16954f2d7cafSStefano Zampini   if (!d_nnz) for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] = d_nz;
16964f2d7cafSStefano Zampini   else for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] = d_nnz[i];
16974f2d7cafSStefano Zampini 
16984f2d7cafSStefano Zampini   if (!o_nnz) for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] += o_nz;
16994f2d7cafSStefano Zampini   else for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] += o_nnz[i];
17004f2d7cafSStefano Zampini 
1701ad227feaSJunchao Zhang   ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
170228f4e0baSStefano Zampini   ierr = MatGetSize(matis->A,NULL,&nlocalcols);CHKERRQ(ierr);
170328f4e0baSStefano Zampini   ierr = MatGetBlockSize(matis->A,&bs);CHKERRQ(ierr);
1704ad227feaSJunchao Zhang   ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
17054f2d7cafSStefano Zampini 
17064f2d7cafSStefano Zampini   for (i=0;i<matis->sf->nleaves;i++) matis->sf_leafdata[i] = PetscMin(matis->sf_leafdata[i],nlocalcols);
170728f4e0baSStefano Zampini   ierr = MatSeqAIJSetPreallocation(matis->A,0,matis->sf_leafdata);CHKERRQ(ierr);
17080f2f62c7SStefano Zampini #if defined(PETSC_HAVE_HYPRE)
17090f2f62c7SStefano Zampini   ierr = MatHYPRESetPreallocation(matis->A,0,matis->sf_leafdata,0,NULL);CHKERRQ(ierr);
17100f2f62c7SStefano Zampini #endif
17114f2d7cafSStefano Zampini 
17124f2d7cafSStefano Zampini   for (i=0;i<matis->sf->nleaves/bs;i++) matis->sf_leafdata[i] = matis->sf_leafdata[i*bs]/bs;
171328f4e0baSStefano Zampini   ierr = MatSeqBAIJSetPreallocation(matis->A,bs,0,matis->sf_leafdata);CHKERRQ(ierr);
17144f2d7cafSStefano Zampini 
171500a59248SStefano Zampini   nlocalcols /= bs;
171600a59248SStefano Zampini   for (i=0;i<matis->sf->nleaves/bs;i++) matis->sf_leafdata[i] = PetscMin(matis->sf_leafdata[i],nlocalcols - i);
171728f4e0baSStefano Zampini   ierr = MatSeqSBAIJSetPreallocation(matis->A,bs,0,matis->sf_leafdata);CHKERRQ(ierr);
17180f2f62c7SStefano Zampini 
17190f2f62c7SStefano Zampini   /* for other matrix types */
17200f2f62c7SStefano Zampini   ierr = MatSetUp(matis->A);CHKERRQ(ierr);
17212e1947a5SStefano Zampini   PetscFunctionReturn(0);
17222e1947a5SStefano Zampini }
1723b4319ba4SBarry Smith 
17243927de2eSStefano Zampini PETSC_EXTERN PetscErrorCode MatISSetMPIXAIJPreallocation_Private(Mat A, Mat B, PetscBool maxreduce)
17253927de2eSStefano Zampini {
17263927de2eSStefano Zampini   Mat_IS          *matis = (Mat_IS*)(A->data);
17273927de2eSStefano Zampini   PetscInt        *my_dnz,*my_onz,*dnz,*onz,*mat_ranges,*row_ownership;
1728ecf5a873SStefano Zampini   const PetscInt  *global_indices_r,*global_indices_c;
17293927de2eSStefano Zampini   PetscInt        i,j,bs,rows,cols;
17303927de2eSStefano Zampini   PetscInt        lrows,lcols;
17313927de2eSStefano Zampini   PetscInt        local_rows,local_cols;
1732f03112d0SStefano Zampini   PetscMPIInt     size;
17333927de2eSStefano Zampini   PetscBool       isdense,issbaij;
17343927de2eSStefano Zampini   PetscErrorCode  ierr;
17353927de2eSStefano Zampini 
17363927de2eSStefano Zampini   PetscFunctionBegin;
1737ffc4695bSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRMPI(ierr);
17383927de2eSStefano Zampini   ierr = MatGetSize(A,&rows,&cols);CHKERRQ(ierr);
17393927de2eSStefano Zampini   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
17403927de2eSStefano Zampini   ierr = MatGetSize(matis->A,&local_rows,&local_cols);CHKERRQ(ierr);
1741b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQDENSE,&isdense);CHKERRQ(ierr);
1742b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQSBAIJ,&issbaij);CHKERRQ(ierr);
1743ecf5a873SStefano Zampini   ierr = ISLocalToGlobalMappingGetIndices(A->rmap->mapping,&global_indices_r);CHKERRQ(ierr);
1744ecf5a873SStefano Zampini   if (A->rmap->mapping != A->cmap->mapping) {
17457230de76SStefano Zampini     ierr = ISLocalToGlobalMappingGetIndices(A->cmap->mapping,&global_indices_c);CHKERRQ(ierr);
1746ecf5a873SStefano Zampini   } else {
1747ecf5a873SStefano Zampini     global_indices_c = global_indices_r;
1748ecf5a873SStefano Zampini   }
1749ecf5a873SStefano Zampini 
17503927de2eSStefano Zampini   if (issbaij) {
17513927de2eSStefano Zampini     ierr = MatGetRowUpperTriangular(matis->A);CHKERRQ(ierr);
17523927de2eSStefano Zampini   }
17533927de2eSStefano Zampini   /*
1754ecf5a873SStefano Zampini      An SF reduce is needed to sum up properly on shared rows.
17553927de2eSStefano Zampini      Note that generally preallocation is not exact, since it overestimates nonzeros
17563927de2eSStefano Zampini   */
17573927de2eSStefano Zampini   ierr = MatGetLocalSize(A,&lrows,&lcols);CHKERRQ(ierr);
17583927de2eSStefano Zampini   ierr = MatPreallocateInitialize(PetscObjectComm((PetscObject)A),lrows,lcols,dnz,onz);CHKERRQ(ierr);
17593927de2eSStefano Zampini   /* All processes need to compute entire row ownership */
17603927de2eSStefano Zampini   ierr = PetscMalloc1(rows,&row_ownership);CHKERRQ(ierr);
17613927de2eSStefano Zampini   ierr = MatGetOwnershipRanges(A,(const PetscInt**)&mat_ranges);CHKERRQ(ierr);
1762f03112d0SStefano Zampini   for (i=0;i<size;i++) {
17633927de2eSStefano Zampini     for (j=mat_ranges[i];j<mat_ranges[i+1];j++) {
17643927de2eSStefano Zampini       row_ownership[j] = i;
17653927de2eSStefano Zampini     }
17663927de2eSStefano Zampini   }
17677230de76SStefano Zampini   ierr = MatGetOwnershipRangesColumn(A,(const PetscInt**)&mat_ranges);CHKERRQ(ierr);
17683927de2eSStefano Zampini 
17693927de2eSStefano Zampini   /*
17703927de2eSStefano Zampini      my_dnz and my_onz contains exact contribution to preallocation from each local mat
17713927de2eSStefano Zampini      then, they will be summed up properly. This way, preallocation is always sufficient
17723927de2eSStefano Zampini   */
17733927de2eSStefano Zampini   ierr = PetscCalloc2(local_rows,&my_dnz,local_rows,&my_onz);CHKERRQ(ierr);
17743927de2eSStefano Zampini   /* preallocation as a MATAIJ */
17753927de2eSStefano Zampini   if (isdense) { /* special case for dense local matrices */
17763927de2eSStefano Zampini     for (i=0;i<local_rows;i++) {
177712dfadf8SStefano Zampini       PetscInt owner = row_ownership[global_indices_r[i]];
177812dfadf8SStefano Zampini       for (j=0;j<local_cols;j++) {
1779ecf5a873SStefano Zampini         PetscInt index_col = global_indices_c[j];
17803927de2eSStefano Zampini         if (index_col > mat_ranges[owner]-1 && index_col < mat_ranges[owner+1]) { /* diag block */
17813927de2eSStefano Zampini           my_dnz[i] += 1;
17823927de2eSStefano Zampini         } else { /* offdiag block */
17833927de2eSStefano Zampini           my_onz[i] += 1;
17843927de2eSStefano Zampini         }
17853927de2eSStefano Zampini       }
17863927de2eSStefano Zampini     }
1787bb1015c3SStefano Zampini   } else if (matis->A->ops->getrowij) {
1788bb1015c3SStefano Zampini     const PetscInt *ii,*jj,*jptr;
1789bb1015c3SStefano Zampini     PetscBool      done;
1790bb1015c3SStefano Zampini     ierr = MatGetRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&local_rows,&ii,&jj,&done);CHKERRQ(ierr);
1791938e1ff8SStefano Zampini     if (!done) SETERRQ(PetscObjectComm((PetscObject)(matis->A)),PETSC_ERR_PLIB,"Error in MatGetRowIJ");
1792bb1015c3SStefano Zampini     jptr = jj;
1793bb1015c3SStefano Zampini     for (i=0;i<local_rows;i++) {
1794bb1015c3SStefano Zampini       PetscInt index_row = global_indices_r[i];
1795bb1015c3SStefano Zampini       for (j=0;j<ii[i+1]-ii[i];j++,jptr++) {
1796bb1015c3SStefano Zampini         PetscInt owner = row_ownership[index_row];
1797bb1015c3SStefano Zampini         PetscInt index_col = global_indices_c[*jptr];
1798bb1015c3SStefano Zampini         if (index_col > mat_ranges[owner]-1 && index_col < mat_ranges[owner+1]) { /* diag block */
1799bb1015c3SStefano Zampini           my_dnz[i] += 1;
1800bb1015c3SStefano Zampini         } else { /* offdiag block */
1801bb1015c3SStefano Zampini           my_onz[i] += 1;
1802bb1015c3SStefano Zampini         }
1803bb1015c3SStefano Zampini         /* same as before, interchanging rows and cols */
1804bb1015c3SStefano Zampini         if (issbaij && index_col != index_row) {
1805bb1015c3SStefano Zampini           owner = row_ownership[index_col];
1806bb1015c3SStefano Zampini           if (index_row > mat_ranges[owner]-1 && index_row < mat_ranges[owner+1]) {
1807bb1015c3SStefano Zampini             my_dnz[*jptr] += 1;
1808bb1015c3SStefano Zampini           } else {
1809bb1015c3SStefano Zampini             my_onz[*jptr] += 1;
1810bb1015c3SStefano Zampini           }
1811bb1015c3SStefano Zampini         }
1812bb1015c3SStefano Zampini       }
1813bb1015c3SStefano Zampini     }
1814bb1015c3SStefano Zampini     ierr = MatRestoreRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&local_rows,&ii,&jj,&done);CHKERRQ(ierr);
1815938e1ff8SStefano Zampini     if (!done) SETERRQ(PetscObjectComm((PetscObject)(matis->A)),PETSC_ERR_PLIB,"Error in MatRestoreRowIJ");
1816bb1015c3SStefano Zampini   } else { /* loop over rows and use MatGetRow */
18173927de2eSStefano Zampini     for (i=0;i<local_rows;i++) {
18183927de2eSStefano Zampini       const PetscInt *cols;
1819ecf5a873SStefano Zampini       PetscInt       ncols,index_row = global_indices_r[i];
18203927de2eSStefano Zampini       ierr = MatGetRow(matis->A,i,&ncols,&cols,NULL);CHKERRQ(ierr);
18213927de2eSStefano Zampini       for (j=0;j<ncols;j++) {
18223927de2eSStefano Zampini         PetscInt owner = row_ownership[index_row];
1823ecf5a873SStefano Zampini         PetscInt index_col = global_indices_c[cols[j]];
18243927de2eSStefano Zampini         if (index_col > mat_ranges[owner]-1 && index_col < mat_ranges[owner+1]) { /* diag block */
18253927de2eSStefano Zampini           my_dnz[i] += 1;
18263927de2eSStefano Zampini         } else { /* offdiag block */
18273927de2eSStefano Zampini           my_onz[i] += 1;
18283927de2eSStefano Zampini         }
18293927de2eSStefano Zampini         /* same as before, interchanging rows and cols */
1830d9a9e74cSStefano Zampini         if (issbaij && index_col != index_row) {
18313927de2eSStefano Zampini           owner = row_ownership[index_col];
18323927de2eSStefano Zampini           if (index_row > mat_ranges[owner]-1 && index_row < mat_ranges[owner+1]) {
1833d9a9e74cSStefano Zampini             my_dnz[cols[j]] += 1;
18343927de2eSStefano Zampini           } else {
1835d9a9e74cSStefano Zampini             my_onz[cols[j]] += 1;
18363927de2eSStefano Zampini           }
18373927de2eSStefano Zampini         }
18383927de2eSStefano Zampini       }
18393927de2eSStefano Zampini       ierr = MatRestoreRow(matis->A,i,&ncols,&cols,NULL);CHKERRQ(ierr);
18403927de2eSStefano Zampini     }
18413927de2eSStefano Zampini   }
1842ecf5a873SStefano Zampini   if (global_indices_c != global_indices_r) {
18437230de76SStefano Zampini     ierr = ISLocalToGlobalMappingRestoreIndices(A->cmap->mapping,&global_indices_c);CHKERRQ(ierr);
1844ecf5a873SStefano Zampini   }
18454f619741Sstefano_zampini   ierr = ISLocalToGlobalMappingRestoreIndices(A->rmap->mapping,&global_indices_r);CHKERRQ(ierr);
18463927de2eSStefano Zampini   ierr = PetscFree(row_ownership);CHKERRQ(ierr);
1847ecf5a873SStefano Zampini 
1848ecf5a873SStefano Zampini   /* Reduce my_dnz and my_onz */
18493927de2eSStefano Zampini   if (maxreduce) {
18503927de2eSStefano Zampini     ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_dnz,dnz,MPI_MAX);CHKERRQ(ierr);
18513927de2eSStefano Zampini     ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_onz,onz,MPI_MAX);CHKERRQ(ierr);
1852bb1015c3SStefano Zampini     ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_dnz,dnz,MPI_MAX);CHKERRQ(ierr);
18533927de2eSStefano Zampini     ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_onz,onz,MPI_MAX);CHKERRQ(ierr);
18543927de2eSStefano Zampini   } else {
18553927de2eSStefano Zampini     ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_dnz,dnz,MPI_SUM);CHKERRQ(ierr);
18563927de2eSStefano Zampini     ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_onz,onz,MPI_SUM);CHKERRQ(ierr);
1857bb1015c3SStefano Zampini     ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_dnz,dnz,MPI_SUM);CHKERRQ(ierr);
18583927de2eSStefano Zampini     ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_onz,onz,MPI_SUM);CHKERRQ(ierr);
18593927de2eSStefano Zampini   }
18603927de2eSStefano Zampini   ierr = PetscFree2(my_dnz,my_onz);CHKERRQ(ierr);
18613927de2eSStefano Zampini 
18623927de2eSStefano Zampini   /* Resize preallocation if overestimated */
18633927de2eSStefano Zampini   for (i=0;i<lrows;i++) {
18643927de2eSStefano Zampini     dnz[i] = PetscMin(dnz[i],lcols);
18653927de2eSStefano Zampini     onz[i] = PetscMin(onz[i],cols-lcols);
18663927de2eSStefano Zampini   }
18671670daf9Sstefano_zampini 
18681670daf9Sstefano_zampini   /* Set preallocation */
1869268753edSStefano Zampini   ierr = MatSeqAIJSetPreallocation(B,0,dnz);CHKERRQ(ierr);
18703927de2eSStefano Zampini   ierr = MatMPIAIJSetPreallocation(B,0,dnz,0,onz);CHKERRQ(ierr);
187153b44cf5SStefano Zampini   for (i=0;i<lrows;i+=bs) {
187253b44cf5SStefano Zampini     PetscInt b, d = dnz[i],o = onz[i];
187353b44cf5SStefano Zampini 
187453b44cf5SStefano Zampini     for (b=1;b<bs;b++) {
187553b44cf5SStefano Zampini       d = PetscMax(d,dnz[i+b]);
187653b44cf5SStefano Zampini       o = PetscMax(o,onz[i+b]);
187753b44cf5SStefano Zampini     }
187853b44cf5SStefano Zampini     dnz[i/bs] = PetscMin(d/bs + d%bs,lcols/bs);
187953b44cf5SStefano Zampini     onz[i/bs] = PetscMin(o/bs + o%bs,(cols-lcols)/bs);
18803927de2eSStefano Zampini   }
1881268753edSStefano Zampini   ierr = MatSeqBAIJSetPreallocation(B,bs,0,dnz);CHKERRQ(ierr);
18823927de2eSStefano Zampini   ierr = MatMPIBAIJSetPreallocation(B,bs,0,dnz,0,onz);CHKERRQ(ierr);
18833927de2eSStefano Zampini   ierr = MatMPISBAIJSetPreallocation(B,bs,0,dnz,0,onz);CHKERRQ(ierr);
18843927de2eSStefano Zampini   ierr = MatPreallocateFinalize(dnz,onz);CHKERRQ(ierr);
18853927de2eSStefano Zampini   if (issbaij) {
18863927de2eSStefano Zampini     ierr = MatRestoreRowUpperTriangular(matis->A);CHKERRQ(ierr);
18873927de2eSStefano Zampini   }
18889be90c3fSStefano Zampini   ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
18893927de2eSStefano Zampini   PetscFunctionReturn(0);
18903927de2eSStefano Zampini }
18913927de2eSStefano Zampini 
1892487b449aSStefano Zampini PETSC_INTERN PetscErrorCode MatConvert_IS_XAIJ(Mat mat, MatType mtype, MatReuse reuse, Mat *M)
1893b7ce53b6SStefano Zampini {
1894b7ce53b6SStefano Zampini   Mat_IS            *matis = (Mat_IS*)(mat->data);
1895487b449aSStefano Zampini   Mat               local_mat,MT;
189653b44cf5SStefano Zampini   PetscInt          rbs,cbs,rows,cols,lrows,lcols;
1897b7ce53b6SStefano Zampini   PetscInt          local_rows,local_cols;
1898b9ed4604SStefano Zampini   PetscBool         isseqdense,isseqsbaij,isseqaij,isseqbaij;
1899f03112d0SStefano Zampini   PetscMPIInt       size;
19001683a169SBarry Smith   const PetscScalar *array;
1901b7ce53b6SStefano Zampini   PetscErrorCode    ierr;
1902b7ce53b6SStefano Zampini 
1903b7ce53b6SStefano Zampini   PetscFunctionBegin;
1904ffc4695bSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRMPI(ierr);
1905f03112d0SStefano Zampini   if (size == 1 && mat->rmap->N == matis->A->rmap->N && mat->cmap->N == matis->A->cmap->N) {
19061670daf9Sstefano_zampini     Mat      B;
190753b44cf5SStefano Zampini     IS       irows = NULL,icols = NULL;
1908487b449aSStefano Zampini     PetscInt rbs,cbs;
19091670daf9Sstefano_zampini 
1910487b449aSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockSize(mat->rmap->mapping,&rbs);CHKERRQ(ierr);
1911487b449aSStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockSize(mat->cmap->mapping,&cbs);CHKERRQ(ierr);
191253b44cf5SStefano Zampini     if (reuse != MAT_REUSE_MATRIX) { /* check if l2g maps are one-to-one */
191353b44cf5SStefano Zampini       IS             rows,cols;
191453b44cf5SStefano Zampini       const PetscInt *ridxs,*cidxs;
191553b44cf5SStefano Zampini       PetscInt       i,nw,*work;
191653b44cf5SStefano Zampini 
191753b44cf5SStefano Zampini       ierr = ISLocalToGlobalMappingGetBlockIndices(mat->rmap->mapping,&ridxs);CHKERRQ(ierr);
191853b44cf5SStefano Zampini       ierr = ISLocalToGlobalMappingGetSize(mat->rmap->mapping,&nw);CHKERRQ(ierr);
191953b44cf5SStefano Zampini       nw   = nw/rbs;
192053b44cf5SStefano Zampini       ierr = PetscCalloc1(nw,&work);CHKERRQ(ierr);
192153b44cf5SStefano Zampini       for (i=0;i<nw;i++) work[ridxs[i]] += 1;
192253b44cf5SStefano Zampini       for (i=0;i<nw;i++) if (!work[i] || work[i] > 1) break;
192353b44cf5SStefano Zampini       if (i == nw) {
192453b44cf5SStefano Zampini         ierr = ISCreateBlock(PETSC_COMM_SELF,rbs,nw,ridxs,PETSC_USE_POINTER,&rows);CHKERRQ(ierr);
1925acdf38a7Sstefano_zampini         ierr = ISSetPermutation(rows);CHKERRQ(ierr);
192653b44cf5SStefano Zampini         ierr = ISInvertPermutation(rows,PETSC_DECIDE,&irows);CHKERRQ(ierr);
1927acdf38a7Sstefano_zampini         ierr = ISDestroy(&rows);CHKERRQ(ierr);
192853b44cf5SStefano Zampini       }
192953b44cf5SStefano Zampini       ierr = ISLocalToGlobalMappingRestoreBlockIndices(mat->rmap->mapping,&ridxs);CHKERRQ(ierr);
193053b44cf5SStefano Zampini       ierr = PetscFree(work);CHKERRQ(ierr);
193153b44cf5SStefano Zampini       if (irows && mat->rmap->mapping != mat->cmap->mapping) {
193253b44cf5SStefano Zampini         ierr = ISLocalToGlobalMappingGetBlockIndices(mat->cmap->mapping,&cidxs);CHKERRQ(ierr);
193353b44cf5SStefano Zampini         ierr = ISLocalToGlobalMappingGetSize(mat->cmap->mapping,&nw);CHKERRQ(ierr);
193453b44cf5SStefano Zampini         nw   = nw/cbs;
193553b44cf5SStefano Zampini         ierr = PetscCalloc1(nw,&work);CHKERRQ(ierr);
193653b44cf5SStefano Zampini         for (i=0;i<nw;i++) work[cidxs[i]] += 1;
193753b44cf5SStefano Zampini         for (i=0;i<nw;i++) if (!work[i] || work[i] > 1) break;
193853b44cf5SStefano Zampini         if (i == nw) {
193953b44cf5SStefano Zampini           ierr = ISCreateBlock(PETSC_COMM_SELF,cbs,nw,cidxs,PETSC_USE_POINTER,&cols);CHKERRQ(ierr);
194053b44cf5SStefano Zampini           ierr = ISSetPermutation(cols);CHKERRQ(ierr);
194153b44cf5SStefano Zampini           ierr = ISInvertPermutation(cols,PETSC_DECIDE,&icols);CHKERRQ(ierr);
194253b44cf5SStefano Zampini           ierr = ISDestroy(&cols);CHKERRQ(ierr);
194353b44cf5SStefano Zampini         }
194453b44cf5SStefano Zampini         ierr = ISLocalToGlobalMappingRestoreBlockIndices(mat->cmap->mapping,&cidxs);CHKERRQ(ierr);
194553b44cf5SStefano Zampini         ierr = PetscFree(work);CHKERRQ(ierr);
194653b44cf5SStefano Zampini       } else if (irows) {
194753b44cf5SStefano Zampini         ierr  = PetscObjectReference((PetscObject)irows);CHKERRQ(ierr);
194853b44cf5SStefano Zampini         icols = irows;
194953b44cf5SStefano Zampini       }
195053b44cf5SStefano Zampini     } else {
195153b44cf5SStefano Zampini       ierr = PetscObjectQuery((PetscObject)(*M),"_MatIS_IS_XAIJ_irows",(PetscObject*)&irows);CHKERRQ(ierr);
195253b44cf5SStefano Zampini       ierr = PetscObjectQuery((PetscObject)(*M),"_MatIS_IS_XAIJ_icols",(PetscObject*)&icols);CHKERRQ(ierr);
195353b44cf5SStefano Zampini       if (irows) { ierr = PetscObjectReference((PetscObject)irows);CHKERRQ(ierr); }
195453b44cf5SStefano Zampini       if (icols) { ierr = PetscObjectReference((PetscObject)icols);CHKERRQ(ierr); }
195553b44cf5SStefano Zampini     }
195653b44cf5SStefano Zampini     if (!irows || !icols) {
195753b44cf5SStefano Zampini       ierr = ISDestroy(&icols);CHKERRQ(ierr);
195853b44cf5SStefano Zampini       ierr = ISDestroy(&irows);CHKERRQ(ierr);
195953b44cf5SStefano Zampini       goto general_assembly;
196053b44cf5SStefano Zampini     }
1961d8e18881SStefano Zampini     ierr = MatConvert(matis->A,mtype,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
1962487b449aSStefano Zampini     if (reuse != MAT_INPLACE_MATRIX) {
19637dae84e0SHong Zhang       ierr = MatCreateSubMatrix(B,irows,icols,reuse,M);CHKERRQ(ierr);
196453b44cf5SStefano Zampini       ierr = PetscObjectCompose((PetscObject)(*M),"_MatIS_IS_XAIJ_irows",(PetscObject)irows);CHKERRQ(ierr);
196553b44cf5SStefano Zampini       ierr = PetscObjectCompose((PetscObject)(*M),"_MatIS_IS_XAIJ_icols",(PetscObject)icols);CHKERRQ(ierr);
1966487b449aSStefano Zampini     } else {
1967487b449aSStefano Zampini       Mat C;
1968487b449aSStefano Zampini 
1969487b449aSStefano Zampini       ierr = MatCreateSubMatrix(B,irows,icols,MAT_INITIAL_MATRIX,&C);CHKERRQ(ierr);
1970487b449aSStefano Zampini       ierr = MatHeaderReplace(mat,&C);CHKERRQ(ierr);
1971487b449aSStefano Zampini     }
1972acdf38a7Sstefano_zampini     ierr = MatDestroy(&B);CHKERRQ(ierr);
1973acdf38a7Sstefano_zampini     ierr = ISDestroy(&icols);CHKERRQ(ierr);
1974acdf38a7Sstefano_zampini     ierr = ISDestroy(&irows);CHKERRQ(ierr);
19757c03b4e8SStefano Zampini     PetscFunctionReturn(0);
19767c03b4e8SStefano Zampini   }
197753b44cf5SStefano Zampini general_assembly:
1978b7ce53b6SStefano Zampini   ierr = MatGetSize(mat,&rows,&cols);CHKERRQ(ierr);
197953b44cf5SStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(mat->rmap->mapping,&rbs);CHKERRQ(ierr);
198053b44cf5SStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(mat->cmap->mapping,&cbs);CHKERRQ(ierr);
19813cfa4ea4SStefano Zampini   ierr = MatGetLocalSize(mat,&lrows,&lcols);CHKERRQ(ierr);
1982b7ce53b6SStefano Zampini   ierr = MatGetSize(matis->A,&local_rows,&local_cols);CHKERRQ(ierr);
1983b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQDENSE,&isseqdense);CHKERRQ(ierr);
19844099cc6bSBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQAIJ,&isseqaij);CHKERRQ(ierr);
1985b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQBAIJ,&isseqbaij);CHKERRQ(ierr);
1986b9e7e5c1SBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQSBAIJ,&isseqsbaij);CHKERRQ(ierr);
1987b9ed4604SStefano Zampini   if (!isseqdense && !isseqaij && !isseqbaij && !isseqsbaij) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not for matrix type %s",((PetscObject)(matis->A))->type_name);
198876bd3646SJed Brown   if (PetscDefined (USE_DEBUG)) {
198976bd3646SJed Brown     PetscBool         lb[4],bb[4];
199076bd3646SJed Brown 
1991b9ed4604SStefano Zampini     lb[0] = isseqdense;
1992b9ed4604SStefano Zampini     lb[1] = isseqaij;
1993b9ed4604SStefano Zampini     lb[2] = isseqbaij;
1994b9ed4604SStefano Zampini     lb[3] = isseqsbaij;
1995820f2d46SBarry Smith     ierr = MPIU_Allreduce(lb,bb,4,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)mat));CHKERRMPI(ierr);
1996b9ed4604SStefano Zampini     if (!bb[0] && !bb[1] && !bb[2] && !bb[3]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Local matrices must have the same type");
199776bd3646SJed Brown   }
1998b7ce53b6SStefano Zampini 
1999487b449aSStefano Zampini   if (reuse != MAT_REUSE_MATRIX) {
2000487b449aSStefano Zampini     ierr = MatCreate(PetscObjectComm((PetscObject)mat),&MT);CHKERRQ(ierr);
2001487b449aSStefano Zampini     ierr = MatSetSizes(MT,lrows,lcols,rows,cols);CHKERRQ(ierr);
2002487b449aSStefano Zampini     ierr = MatSetType(MT,mtype);CHKERRQ(ierr);
200353b44cf5SStefano Zampini     ierr = MatSetBlockSizes(MT,rbs,cbs);CHKERRQ(ierr);
2004487b449aSStefano Zampini     ierr = MatISSetMPIXAIJPreallocation_Private(mat,MT,PETSC_FALSE);CHKERRQ(ierr);
2005b7ce53b6SStefano Zampini   } else {
200653b44cf5SStefano Zampini     PetscInt mrbs,mcbs,mrows,mcols,mlrows,mlcols;
2007487b449aSStefano Zampini 
2008b7ce53b6SStefano Zampini     /* some checks */
2009487b449aSStefano Zampini     MT   = *M;
201053b44cf5SStefano Zampini     ierr = MatGetBlockSizes(MT,&mrbs,&mcbs);CHKERRQ(ierr);
2011487b449aSStefano Zampini     ierr = MatGetSize(MT,&mrows,&mcols);CHKERRQ(ierr);
2012487b449aSStefano Zampini     ierr = MatGetLocalSize(MT,&mlrows,&mlcols);CHKERRQ(ierr);
20136c4ed002SBarry Smith     if (mrows != rows) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of rows (%d != %d)",rows,mrows);
20146c4ed002SBarry Smith     if (mcols != cols) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of cols (%d != %d)",cols,mcols);
20156c4ed002SBarry Smith     if (mlrows != lrows) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of local rows (%d != %d)",lrows,mlrows);
20166c4ed002SBarry Smith     if (mlcols != lcols) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of local cols (%d != %d)",lcols,mlcols);
201753b44cf5SStefano Zampini     if (mrbs != rbs) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong row block size (%d != %d)",rbs,mrbs);
201853b44cf5SStefano Zampini     if (mcbs != cbs) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong col block size (%d != %d)",cbs,mcbs);
2019487b449aSStefano Zampini     ierr = MatZeroEntries(MT);CHKERRQ(ierr);
2020b7ce53b6SStefano Zampini   }
2021d9a9e74cSStefano Zampini 
20228546b261SStefano Zampini   if (isseqsbaij || isseqbaij) {
20238546b261SStefano Zampini     ierr = MatConvert(matis->A,MATSEQAIJ,MAT_INITIAL_MATRIX,&local_mat);CHKERRQ(ierr);
20248546b261SStefano Zampini     isseqaij = PETSC_TRUE;
2025d9a9e74cSStefano Zampini   } else {
2026d9a9e74cSStefano Zampini     ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
2027d9a9e74cSStefano Zampini     local_mat = matis->A;
2028d9a9e74cSStefano Zampini   }
2029686e3a49SStefano Zampini 
2030b7ce53b6SStefano Zampini   /* Set values */
2031487b449aSStefano Zampini   ierr = MatSetLocalToGlobalMapping(MT,mat->rmap->mapping,mat->cmap->mapping);CHKERRQ(ierr);
2032b9ed4604SStefano Zampini   if (isseqdense) { /* special case for dense local matrices */
203365066ba5SStefano Zampini     PetscInt          i,*dummy;
2034ecf5a873SStefano Zampini 
203565066ba5SStefano Zampini     ierr = PetscMalloc1(PetscMax(local_rows,local_cols),&dummy);CHKERRQ(ierr);
203665066ba5SStefano Zampini     for (i=0;i<PetscMax(local_rows,local_cols);i++) dummy[i] = i;
2037487b449aSStefano Zampini     ierr = MatSetOption(MT,MAT_ROW_ORIENTED,PETSC_FALSE);CHKERRQ(ierr);
20381683a169SBarry Smith     ierr = MatDenseGetArrayRead(local_mat,&array);CHKERRQ(ierr);
2039487b449aSStefano Zampini     ierr = MatSetValuesLocal(MT,local_rows,dummy,local_cols,dummy,array,ADD_VALUES);CHKERRQ(ierr);
20401683a169SBarry Smith     ierr = MatDenseRestoreArrayRead(local_mat,&array);CHKERRQ(ierr);
204165066ba5SStefano Zampini     ierr = PetscFree(dummy);CHKERRQ(ierr);
2042686e3a49SStefano Zampini   } else if (isseqaij) {
20436afe12f5SStefano Zampini     const PetscInt *blocks;
20446afe12f5SStefano Zampini     PetscInt       i,nvtxs,*xadj,*adjncy, nb;
2045686e3a49SStefano Zampini     PetscBool      done;
20461683a169SBarry Smith     PetscScalar    *sarray;
2047686e3a49SStefano Zampini 
2048d9a9e74cSStefano Zampini     ierr = MatGetRowIJ(local_mat,0,PETSC_FALSE,PETSC_FALSE,&nvtxs,(const PetscInt**)&xadj,(const PetscInt**)&adjncy,&done);CHKERRQ(ierr);
2049938e1ff8SStefano Zampini     if (!done) SETERRQ(PetscObjectComm((PetscObject)local_mat),PETSC_ERR_PLIB,"Error in MatGetRowIJ");
20501683a169SBarry Smith     ierr = MatSeqAIJGetArray(local_mat,&sarray);CHKERRQ(ierr);
20516afe12f5SStefano Zampini     ierr = MatGetVariableBlockSizes(local_mat,&nb,&blocks);CHKERRQ(ierr);
20526afe12f5SStefano Zampini     if (nb) { /* speed up assembly for special blocked matrices (used by BDDC) */
20536afe12f5SStefano Zampini       PetscInt sum;
20546afe12f5SStefano Zampini 
20556afe12f5SStefano Zampini       for (i=0,sum=0;i<nb;i++) sum += blocks[i];
20566afe12f5SStefano Zampini       if (sum == nvtxs) {
20576afe12f5SStefano Zampini         PetscInt r;
20586afe12f5SStefano Zampini 
20596afe12f5SStefano Zampini         for (i=0,r=0;i<nb;i++) {
2060cf9c20a2SJed Brown           if (PetscUnlikelyDebug(blocks[i] != xadj[r+1] - xadj[r])) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Invalid block sizes prescribed for block %D: expected %D, got %D",i,blocks[i],xadj[r+1] - xadj[r]);
20611683a169SBarry Smith           ierr = MatSetValuesLocal(MT,blocks[i],adjncy+xadj[r],blocks[i],adjncy+xadj[r],sarray+xadj[r],ADD_VALUES);CHKERRQ(ierr);
20626afe12f5SStefano Zampini           r   += blocks[i];
20636afe12f5SStefano Zampini         }
20646afe12f5SStefano Zampini       } else {
2065686e3a49SStefano Zampini         for (i=0;i<nvtxs;i++) {
20661683a169SBarry Smith           ierr = MatSetValuesLocal(MT,1,&i,xadj[i+1]-xadj[i],adjncy+xadj[i],sarray+xadj[i],ADD_VALUES);CHKERRQ(ierr);
2067686e3a49SStefano Zampini         }
20686afe12f5SStefano Zampini       }
20696afe12f5SStefano Zampini     } else {
20706afe12f5SStefano Zampini       for (i=0;i<nvtxs;i++) {
20711683a169SBarry Smith         ierr = MatSetValuesLocal(MT,1,&i,xadj[i+1]-xadj[i],adjncy+xadj[i],sarray+xadj[i],ADD_VALUES);CHKERRQ(ierr);
20726afe12f5SStefano Zampini       }
20736afe12f5SStefano Zampini     }
2074d9a9e74cSStefano Zampini     ierr = MatRestoreRowIJ(local_mat,0,PETSC_FALSE,PETSC_FALSE,&nvtxs,(const PetscInt**)&xadj,(const PetscInt**)&adjncy,&done);CHKERRQ(ierr);
2075938e1ff8SStefano Zampini     if (!done) SETERRQ(PetscObjectComm((PetscObject)local_mat),PETSC_ERR_PLIB,"Error in MatRestoreRowIJ");
20761683a169SBarry Smith     ierr = MatSeqAIJRestoreArray(local_mat,&sarray);CHKERRQ(ierr);
2077686e3a49SStefano Zampini   } else { /* very basic values insertion for all other matrix types */
2078ecf5a873SStefano Zampini     PetscInt i;
2079c0962df8SStefano Zampini 
2080686e3a49SStefano Zampini     for (i=0;i<local_rows;i++) {
2081686e3a49SStefano Zampini       PetscInt       j;
2082ecf5a873SStefano Zampini       const PetscInt *local_indices_cols;
2083686e3a49SStefano Zampini 
20841683a169SBarry Smith       ierr = MatGetRow(local_mat,i,&j,&local_indices_cols,&array);CHKERRQ(ierr);
2085487b449aSStefano Zampini       ierr = MatSetValuesLocal(MT,1,&i,j,local_indices_cols,array,ADD_VALUES);CHKERRQ(ierr);
20861683a169SBarry Smith       ierr = MatRestoreRow(local_mat,i,&j,&local_indices_cols,&array);CHKERRQ(ierr);
2087686e3a49SStefano Zampini     }
2088b7ce53b6SStefano Zampini   }
2089487b449aSStefano Zampini   ierr = MatAssemblyBegin(MT,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2090d9a9e74cSStefano Zampini   ierr = MatDestroy(&local_mat);CHKERRQ(ierr);
2091487b449aSStefano Zampini   ierr = MatAssemblyEnd(MT,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2092b9ed4604SStefano Zampini   if (isseqdense) {
2093487b449aSStefano Zampini     ierr = MatSetOption(MT,MAT_ROW_ORIENTED,PETSC_TRUE);CHKERRQ(ierr);
2094487b449aSStefano Zampini   }
2095487b449aSStefano Zampini   if (reuse == MAT_INPLACE_MATRIX) {
2096487b449aSStefano Zampini     ierr = MatHeaderReplace(mat,&MT);CHKERRQ(ierr);
2097487b449aSStefano Zampini   } else if (reuse == MAT_INITIAL_MATRIX) {
2098487b449aSStefano Zampini     *M = MT;
2099b7ce53b6SStefano Zampini   }
2100b7ce53b6SStefano Zampini   PetscFunctionReturn(0);
2101b7ce53b6SStefano Zampini }
2102b7ce53b6SStefano Zampini 
2103b7ce53b6SStefano Zampini /*@
2104b7ce53b6SStefano Zampini     MatISGetMPIXAIJ - Converts MATIS matrix into a parallel AIJ format
2105b7ce53b6SStefano Zampini 
2106*d8d19677SJose E. Roman   Input Parameters:
2107a2b725a8SWilliam Gropp +  mat - the matrix (should be of type MATIS)
2108a2b725a8SWilliam Gropp -  reuse - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
2109b7ce53b6SStefano Zampini 
2110b7ce53b6SStefano Zampini   Output Parameter:
2111b7ce53b6SStefano Zampini .  newmat - the matrix in AIJ format
2112b7ce53b6SStefano Zampini 
2113b7ce53b6SStefano Zampini   Level: developer
2114b7ce53b6SStefano Zampini 
211595452b02SPatrick Sanan   Notes:
2116487b449aSStefano Zampini     This function has been deprecated and it will be removed in future releases. Update your code to use the MatConvert() interface.
2117b7ce53b6SStefano Zampini 
2118487b449aSStefano Zampini .seealso: MATIS, MatConvert()
2119b7ce53b6SStefano Zampini @*/
2120b7ce53b6SStefano Zampini PetscErrorCode MatISGetMPIXAIJ(Mat mat, MatReuse reuse, Mat *newmat)
2121b7ce53b6SStefano Zampini {
2122b7ce53b6SStefano Zampini   PetscErrorCode ierr;
2123b7ce53b6SStefano Zampini 
2124b7ce53b6SStefano Zampini   PetscFunctionBegin;
2125b7ce53b6SStefano Zampini   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2126b7ce53b6SStefano Zampini   PetscValidLogicalCollectiveEnum(mat,reuse,2);
2127b7ce53b6SStefano Zampini   PetscValidPointer(newmat,3);
2128487b449aSStefano Zampini   if (reuse == MAT_REUSE_MATRIX) {
2129b7ce53b6SStefano Zampini     PetscValidHeaderSpecific(*newmat,MAT_CLASSID,3);
2130b7ce53b6SStefano Zampini     PetscCheckSameComm(mat,1,*newmat,3);
21316c4ed002SBarry Smith     if (mat == *newmat) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse the same matrix");
2132b7ce53b6SStefano Zampini   }
2133487b449aSStefano Zampini   ierr = PetscUseMethod(mat,"MatISGetMPIXAIJ_C",(Mat,MatType,MatReuse,Mat*),(mat,MATAIJ,reuse,newmat));CHKERRQ(ierr);
2134b7ce53b6SStefano Zampini   PetscFunctionReturn(0);
2135b7ce53b6SStefano Zampini }
2136b7ce53b6SStefano Zampini 
2137ad6194a2SStefano Zampini PetscErrorCode MatDuplicate_IS(Mat mat,MatDuplicateOption op,Mat *newmat)
2138ad6194a2SStefano Zampini {
2139ad6194a2SStefano Zampini   PetscErrorCode ierr;
2140ad6194a2SStefano Zampini   Mat_IS         *matis = (Mat_IS*)(mat->data);
2141c9225affSStefano Zampini   PetscInt       rbs,cbs,m,n,M,N;
2142ad6194a2SStefano Zampini   Mat            B,localmat;
2143ad6194a2SStefano Zampini 
2144ad6194a2SStefano Zampini   PetscFunctionBegin;
2145c9225affSStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(mat->rmap->mapping,&rbs);CHKERRQ(ierr);
2146c9225affSStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(mat->cmap->mapping,&cbs);CHKERRQ(ierr);
2147ad6194a2SStefano Zampini   ierr = MatGetSize(mat,&M,&N);CHKERRQ(ierr);
2148ad6194a2SStefano Zampini   ierr = MatGetLocalSize(mat,&m,&n);CHKERRQ(ierr);
21498546b261SStefano Zampini   ierr = MatCreate(PetscObjectComm((PetscObject)mat),&B);CHKERRQ(ierr);
21508546b261SStefano Zampini   ierr = MatSetSizes(B,m,n,M,N);CHKERRQ(ierr);
21518546b261SStefano Zampini   ierr = MatSetBlockSize(B,rbs == cbs ? rbs : 1);CHKERRQ(ierr);
21528546b261SStefano Zampini   ierr = MatSetType(B,MATIS);CHKERRQ(ierr);
21538546b261SStefano Zampini   ierr = MatISSetLocalMatType(B,matis->lmattype);CHKERRQ(ierr);
21548546b261SStefano Zampini   ierr = MatSetLocalToGlobalMapping(B,mat->rmap->mapping,mat->cmap->mapping);CHKERRQ(ierr);
2155ad6194a2SStefano Zampini   ierr = MatDuplicate(matis->A,op,&localmat);CHKERRQ(ierr);
2156ad6194a2SStefano Zampini   ierr = MatISSetLocalMat(B,localmat);CHKERRQ(ierr);
2157b3317aa8SStefano Zampini   ierr = MatDestroy(&localmat);CHKERRQ(ierr);
2158ad6194a2SStefano Zampini   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2159ad6194a2SStefano Zampini   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2160ad6194a2SStefano Zampini   *newmat = B;
2161ad6194a2SStefano Zampini   PetscFunctionReturn(0);
2162ad6194a2SStefano Zampini }
2163ad6194a2SStefano Zampini 
2164a8116848SStefano Zampini static PetscErrorCode MatIsHermitian_IS(Mat A,PetscReal tol,PetscBool  *flg)
216569796d55SStefano Zampini {
216669796d55SStefano Zampini   PetscErrorCode ierr;
216769796d55SStefano Zampini   Mat_IS         *matis = (Mat_IS*)A->data;
216869796d55SStefano Zampini   PetscBool      local_sym;
216969796d55SStefano Zampini 
217069796d55SStefano Zampini   PetscFunctionBegin;
217169796d55SStefano Zampini   ierr = MatIsHermitian(matis->A,tol,&local_sym);CHKERRQ(ierr);
2172820f2d46SBarry Smith   ierr = MPIU_Allreduce(&local_sym,flg,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
217369796d55SStefano Zampini   PetscFunctionReturn(0);
217469796d55SStefano Zampini }
217569796d55SStefano Zampini 
2176a8116848SStefano Zampini static PetscErrorCode MatIsSymmetric_IS(Mat A,PetscReal tol,PetscBool  *flg)
217769796d55SStefano Zampini {
217869796d55SStefano Zampini   PetscErrorCode ierr;
217969796d55SStefano Zampini   Mat_IS         *matis = (Mat_IS*)A->data;
218069796d55SStefano Zampini   PetscBool      local_sym;
218169796d55SStefano Zampini 
218269796d55SStefano Zampini   PetscFunctionBegin;
218369796d55SStefano Zampini   ierr = MatIsSymmetric(matis->A,tol,&local_sym);CHKERRQ(ierr);
2184820f2d46SBarry Smith   ierr = MPIU_Allreduce(&local_sym,flg,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
218569796d55SStefano Zampini   PetscFunctionReturn(0);
218669796d55SStefano Zampini }
218769796d55SStefano Zampini 
218845471136SStefano Zampini static PetscErrorCode MatIsStructurallySymmetric_IS(Mat A,PetscBool  *flg)
218945471136SStefano Zampini {
219045471136SStefano Zampini   PetscErrorCode ierr;
219145471136SStefano Zampini   Mat_IS         *matis = (Mat_IS*)A->data;
219245471136SStefano Zampini   PetscBool      local_sym;
219345471136SStefano Zampini 
219445471136SStefano Zampini   PetscFunctionBegin;
219545471136SStefano Zampini   if (A->rmap->mapping != A->cmap->mapping) {
219645471136SStefano Zampini     *flg = PETSC_FALSE;
219745471136SStefano Zampini     PetscFunctionReturn(0);
219845471136SStefano Zampini   }
219945471136SStefano Zampini   ierr = MatIsStructurallySymmetric(matis->A,&local_sym);CHKERRQ(ierr);
2200820f2d46SBarry Smith   ierr = MPIU_Allreduce(&local_sym,flg,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
220145471136SStefano Zampini   PetscFunctionReturn(0);
220245471136SStefano Zampini }
220345471136SStefano Zampini 
2204a8116848SStefano Zampini static PetscErrorCode MatDestroy_IS(Mat A)
2205b4319ba4SBarry Smith {
2206dfbe8321SBarry Smith   PetscErrorCode ierr;
2207b4319ba4SBarry Smith   Mat_IS         *b = (Mat_IS*)A->data;
2208b4319ba4SBarry Smith 
2209b4319ba4SBarry Smith   PetscFunctionBegin;
2210b89f26deSStefano Zampini   ierr = PetscFree(b->bdiag);CHKERRQ(ierr);
22118546b261SStefano Zampini   ierr = PetscFree(b->lmattype);CHKERRQ(ierr);
22126bf464f9SBarry Smith   ierr = MatDestroy(&b->A);CHKERRQ(ierr);
2213e176bc59SStefano Zampini   ierr = VecScatterDestroy(&b->cctx);CHKERRQ(ierr);
2214e176bc59SStefano Zampini   ierr = VecScatterDestroy(&b->rctx);CHKERRQ(ierr);
22156bf464f9SBarry Smith   ierr = VecDestroy(&b->x);CHKERRQ(ierr);
22166bf464f9SBarry Smith   ierr = VecDestroy(&b->y);CHKERRQ(ierr);
22173fd1c9e7SStefano Zampini   ierr = VecDestroy(&b->counter);CHKERRQ(ierr);
2218a8116848SStefano Zampini   ierr = ISDestroy(&b->getsub_ris);CHKERRQ(ierr);
2219a8116848SStefano Zampini   ierr = ISDestroy(&b->getsub_cis);CHKERRQ(ierr);
2220a8116848SStefano Zampini   if (b->sf != b->csf) {
2221a8116848SStefano Zampini     ierr = PetscSFDestroy(&b->csf);CHKERRQ(ierr);
2222a8116848SStefano Zampini     ierr = PetscFree2(b->csf_rootdata,b->csf_leafdata);CHKERRQ(ierr);
2223f03112d0SStefano Zampini   } else b->csf = NULL;
222428f4e0baSStefano Zampini   ierr = PetscSFDestroy(&b->sf);CHKERRQ(ierr);
222528f4e0baSStefano Zampini   ierr = PetscFree2(b->sf_rootdata,b->sf_leafdata);CHKERRQ(ierr);
2226bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
2227f4259b30SLisandro Dalcin   ierr = PetscObjectChangeTypeName((PetscObject)A,NULL);CHKERRQ(ierr);
22288546b261SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMatType_C",NULL);CHKERRQ(ierr);
2229bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetLocalMat_C",NULL);CHKERRQ(ierr);
2230b7ce53b6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMat_C",NULL);CHKERRQ(ierr);
2231b7ce53b6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetMPIXAIJ_C",NULL);CHKERRQ(ierr);
22322e1947a5SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetPreallocation_C",NULL);CHKERRQ(ierr);
223375d48cdbSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISStoreL2L_C",NULL);CHKERRQ(ierr);
2234f03112d0SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISFixLocalEmpty_C",NULL);CHKERRQ(ierr);
2235487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpiaij_C",NULL);CHKERRQ(ierr);
2236487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpibaij_C",NULL);CHKERRQ(ierr);
2237487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpisbaij_C",NULL);CHKERRQ(ierr);
2238487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqaij_C",NULL);CHKERRQ(ierr);
2239487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqbaij_C",NULL);CHKERRQ(ierr);
2240487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqsbaij_C",NULL);CHKERRQ(ierr);
2241487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_aij_C",NULL);CHKERRQ(ierr);
2242b4319ba4SBarry Smith   PetscFunctionReturn(0);
2243b4319ba4SBarry Smith }
2244b4319ba4SBarry Smith 
2245a8116848SStefano Zampini static PetscErrorCode MatMult_IS(Mat A,Vec x,Vec y)
2246b4319ba4SBarry Smith {
2247dfbe8321SBarry Smith   PetscErrorCode ierr;
2248b4319ba4SBarry Smith   Mat_IS         *is  = (Mat_IS*)A->data;
2249b4319ba4SBarry Smith   PetscScalar    zero = 0.0;
2250b4319ba4SBarry Smith 
2251b4319ba4SBarry Smith   PetscFunctionBegin;
2252b4319ba4SBarry Smith   /*  scatter the global vector x into the local work vector */
2253e176bc59SStefano Zampini   ierr = VecScatterBegin(is->cctx,x,is->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2254e176bc59SStefano Zampini   ierr = VecScatterEnd(is->cctx,x,is->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2255b4319ba4SBarry Smith 
2256b4319ba4SBarry Smith   /* multiply the local matrix */
2257b4319ba4SBarry Smith   ierr = MatMult(is->A,is->x,is->y);CHKERRQ(ierr);
2258b4319ba4SBarry Smith 
2259b4319ba4SBarry Smith   /* scatter product back into global memory */
22602dcb1b2aSMatthew Knepley   ierr = VecSet(y,zero);CHKERRQ(ierr);
2261e176bc59SStefano Zampini   ierr = VecScatterBegin(is->rctx,is->y,y,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2262e176bc59SStefano Zampini   ierr = VecScatterEnd(is->rctx,is->y,y,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2263b4319ba4SBarry Smith   PetscFunctionReturn(0);
2264b4319ba4SBarry Smith }
2265b4319ba4SBarry Smith 
2266a8116848SStefano Zampini static PetscErrorCode MatMultAdd_IS(Mat A,Vec v1,Vec v2,Vec v3)
22672e74eeadSLisandro Dalcin {
2268650997f4SStefano Zampini   Vec            temp_vec;
22692e74eeadSLisandro Dalcin   PetscErrorCode ierr;
22702e74eeadSLisandro Dalcin 
22712e74eeadSLisandro Dalcin   PetscFunctionBegin; /*  v3 = v2 + A * v1.*/
2272650997f4SStefano Zampini   if (v3 != v2) {
2273650997f4SStefano Zampini     ierr = MatMult(A,v1,v3);CHKERRQ(ierr);
2274650997f4SStefano Zampini     ierr = VecAXPY(v3,1.0,v2);CHKERRQ(ierr);
2275650997f4SStefano Zampini   } else {
2276650997f4SStefano Zampini     ierr = VecDuplicate(v2,&temp_vec);CHKERRQ(ierr);
2277650997f4SStefano Zampini     ierr = MatMult(A,v1,temp_vec);CHKERRQ(ierr);
2278650997f4SStefano Zampini     ierr = VecAXPY(temp_vec,1.0,v2);CHKERRQ(ierr);
2279650997f4SStefano Zampini     ierr = VecCopy(temp_vec,v3);CHKERRQ(ierr);
2280650997f4SStefano Zampini     ierr = VecDestroy(&temp_vec);CHKERRQ(ierr);
2281650997f4SStefano Zampini   }
22822e74eeadSLisandro Dalcin   PetscFunctionReturn(0);
22832e74eeadSLisandro Dalcin }
22842e74eeadSLisandro Dalcin 
2285a8116848SStefano Zampini static PetscErrorCode MatMultTranspose_IS(Mat A,Vec y,Vec x)
22862e74eeadSLisandro Dalcin {
22872e74eeadSLisandro Dalcin   Mat_IS         *is = (Mat_IS*)A->data;
22882e74eeadSLisandro Dalcin   PetscErrorCode ierr;
22892e74eeadSLisandro Dalcin 
2290e176bc59SStefano Zampini   PetscFunctionBegin;
22912e74eeadSLisandro Dalcin   /*  scatter the global vector x into the local work vector */
2292e176bc59SStefano Zampini   ierr = VecScatterBegin(is->rctx,y,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2293e176bc59SStefano Zampini   ierr = VecScatterEnd(is->rctx,y,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
22942e74eeadSLisandro Dalcin 
22952e74eeadSLisandro Dalcin   /* multiply the local matrix */
2296e176bc59SStefano Zampini   ierr = MatMultTranspose(is->A,is->y,is->x);CHKERRQ(ierr);
22972e74eeadSLisandro Dalcin 
22982e74eeadSLisandro Dalcin   /* scatter product back into global vector */
2299e176bc59SStefano Zampini   ierr = VecSet(x,0);CHKERRQ(ierr);
2300e176bc59SStefano Zampini   ierr = VecScatterBegin(is->cctx,is->x,x,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2301e176bc59SStefano Zampini   ierr = VecScatterEnd(is->cctx,is->x,x,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
23022e74eeadSLisandro Dalcin   PetscFunctionReturn(0);
23032e74eeadSLisandro Dalcin }
23042e74eeadSLisandro Dalcin 
2305a8116848SStefano Zampini static PetscErrorCode MatMultTransposeAdd_IS(Mat A,Vec v1,Vec v2,Vec v3)
23062e74eeadSLisandro Dalcin {
2307650997f4SStefano Zampini   Vec            temp_vec;
23082e74eeadSLisandro Dalcin   PetscErrorCode ierr;
23092e74eeadSLisandro Dalcin 
23102e74eeadSLisandro Dalcin   PetscFunctionBegin; /*  v3 = v2 + A' * v1.*/
2311650997f4SStefano Zampini   if (v3 != v2) {
2312650997f4SStefano Zampini     ierr = MatMultTranspose(A,v1,v3);CHKERRQ(ierr);
2313650997f4SStefano Zampini     ierr = VecAXPY(v3,1.0,v2);CHKERRQ(ierr);
2314650997f4SStefano Zampini   } else {
2315650997f4SStefano Zampini     ierr = VecDuplicate(v2,&temp_vec);CHKERRQ(ierr);
2316650997f4SStefano Zampini     ierr = MatMultTranspose(A,v1,temp_vec);CHKERRQ(ierr);
2317650997f4SStefano Zampini     ierr = VecAXPY(temp_vec,1.0,v2);CHKERRQ(ierr);
2318650997f4SStefano Zampini     ierr = VecCopy(temp_vec,v3);CHKERRQ(ierr);
2319650997f4SStefano Zampini     ierr = VecDestroy(&temp_vec);CHKERRQ(ierr);
2320650997f4SStefano Zampini   }
23212e74eeadSLisandro Dalcin   PetscFunctionReturn(0);
23222e74eeadSLisandro Dalcin }
23232e74eeadSLisandro Dalcin 
2324a8116848SStefano Zampini static PetscErrorCode MatView_IS(Mat A,PetscViewer viewer)
2325b4319ba4SBarry Smith {
2326b4319ba4SBarry Smith   Mat_IS         *a = (Mat_IS*)A->data;
2327dfbe8321SBarry Smith   PetscErrorCode ierr;
2328b4319ba4SBarry Smith   PetscViewer    sviewer;
2329ee2491ecSStefano Zampini   PetscBool      isascii,view = PETSC_TRUE;
2330b4319ba4SBarry Smith 
2331b4319ba4SBarry Smith   PetscFunctionBegin;
2332ee2491ecSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
2333ee2491ecSStefano Zampini   if (isascii)  {
2334ee2491ecSStefano Zampini     PetscViewerFormat format;
2335ee2491ecSStefano Zampini 
2336ee2491ecSStefano Zampini     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
2337ee2491ecSStefano Zampini     if (format == PETSC_VIEWER_ASCII_INFO) view = PETSC_FALSE;
2338ee2491ecSStefano Zampini   }
2339ee2491ecSStefano Zampini   if (!view) PetscFunctionReturn(0);
23403f08860eSBarry Smith   ierr = PetscViewerGetSubViewer(viewer,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr);
2341b4319ba4SBarry Smith   ierr = MatView(a->A,sviewer);CHKERRQ(ierr);
23423f08860eSBarry Smith   ierr = PetscViewerRestoreSubViewer(viewer,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr);
23436e520ac8SStefano Zampini   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
2344b4319ba4SBarry Smith   PetscFunctionReturn(0);
2345b4319ba4SBarry Smith }
2346b4319ba4SBarry Smith 
2347b89f26deSStefano Zampini static PetscErrorCode MatInvertBlockDiagonal_IS(Mat mat,const PetscScalar **values)
2348b89f26deSStefano Zampini {
2349b89f26deSStefano Zampini   Mat_IS            *is = (Mat_IS*)mat->data;
2350b89f26deSStefano Zampini   MPI_Datatype      nodeType;
2351b89f26deSStefano Zampini   const PetscScalar *lv;
2352b89f26deSStefano Zampini   PetscInt          bs;
2353b89f26deSStefano Zampini   PetscErrorCode    ierr;
2354b89f26deSStefano Zampini 
2355b89f26deSStefano Zampini   PetscFunctionBegin;
2356b89f26deSStefano Zampini   ierr = MatGetBlockSize(mat,&bs);CHKERRQ(ierr);
2357b89f26deSStefano Zampini   ierr = MatSetBlockSize(is->A,bs);CHKERRQ(ierr);
2358b89f26deSStefano Zampini   ierr = MatInvertBlockDiagonal(is->A,&lv);CHKERRQ(ierr);
2359b89f26deSStefano Zampini   if (!is->bdiag) {
2360b89f26deSStefano Zampini     ierr = PetscMalloc1(bs*mat->rmap->n,&is->bdiag);CHKERRQ(ierr);
2361b89f26deSStefano Zampini   }
2362ffc4695bSBarry Smith   ierr = MPI_Type_contiguous(bs,MPIU_SCALAR,&nodeType);CHKERRMPI(ierr);
2363ffc4695bSBarry Smith   ierr = MPI_Type_commit(&nodeType);CHKERRMPI(ierr);
236483df288dSJunchao Zhang   ierr = PetscSFReduceBegin(is->sf,nodeType,lv,is->bdiag,MPI_REPLACE);CHKERRQ(ierr);
236583df288dSJunchao Zhang   ierr = PetscSFReduceEnd(is->sf,nodeType,lv,is->bdiag,MPI_REPLACE);CHKERRQ(ierr);
2366ffc4695bSBarry Smith   ierr = MPI_Type_free(&nodeType);CHKERRMPI(ierr);
2367b89f26deSStefano Zampini   if (values) *values = is->bdiag;
2368b89f26deSStefano Zampini   PetscFunctionReturn(0);
2369b89f26deSStefano Zampini }
2370b89f26deSStefano Zampini 
23718546b261SStefano Zampini static PetscErrorCode MatISSetUpScatters_Private(Mat A)
2372b4319ba4SBarry Smith {
2373e176bc59SStefano Zampini   Vec            cglobal,rglobal;
23748546b261SStefano Zampini   IS             from;
23758546b261SStefano Zampini   Mat_IS         *is = (Mat_IS*)A->data;
2376b89f26deSStefano Zampini   PetscScalar    sum;
23778546b261SStefano Zampini   const PetscInt *garray;
23788546b261SStefano Zampini   PetscInt       nr,rbs,nc,cbs;
23798546b261SStefano Zampini   PetscBool      iscuda;
23808546b261SStefano Zampini   PetscErrorCode ierr;
2381b4319ba4SBarry Smith 
2382b4319ba4SBarry Smith   PetscFunctionBegin;
23838546b261SStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(A->rmap->mapping,&nr);CHKERRQ(ierr);
23848546b261SStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(A->rmap->mapping,&rbs);CHKERRQ(ierr);
23858546b261SStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(A->cmap->mapping,&nc);CHKERRQ(ierr);
23868546b261SStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(A->cmap->mapping,&cbs);CHKERRQ(ierr);
238770cf5478SStefano Zampini   ierr = VecDestroy(&is->x);CHKERRQ(ierr);
238870cf5478SStefano Zampini   ierr = VecDestroy(&is->y);CHKERRQ(ierr);
23893fd1c9e7SStefano Zampini   ierr = VecDestroy(&is->counter);CHKERRQ(ierr);
2390e176bc59SStefano Zampini   ierr = VecScatterDestroy(&is->rctx);CHKERRQ(ierr);
2391e176bc59SStefano Zampini   ierr = VecScatterDestroy(&is->cctx);CHKERRQ(ierr);
23928546b261SStefano Zampini   ierr = MatCreateVecs(is->A,&is->x,&is->y);CHKERRQ(ierr);
2393b470e4b4SRichard Tran Mills   ierr = VecBindToCPU(is->y,PETSC_TRUE);CHKERRQ(ierr);
23948546b261SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)is->y,VECSEQCUDA,&iscuda);CHKERRQ(ierr);
23958546b261SStefano Zampini   if (iscuda) {
23968546b261SStefano Zampini     ierr = PetscFree(A->defaultvectype);CHKERRQ(ierr);
23978546b261SStefano Zampini     ierr = PetscStrallocpy(VECCUDA,&A->defaultvectype);CHKERRQ(ierr);
23988546b261SStefano Zampini   }
23998546b261SStefano Zampini   ierr = MatCreateVecs(A,&cglobal,&rglobal);CHKERRQ(ierr);
2400b470e4b4SRichard Tran Mills   ierr = VecBindToCPU(rglobal,PETSC_TRUE);CHKERRQ(ierr);
24018546b261SStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockIndices(A->rmap->mapping,&garray);CHKERRQ(ierr);
24028546b261SStefano Zampini   ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),rbs,nr/rbs,garray,PETSC_USE_POINTER,&from);CHKERRQ(ierr);
24039448b7f1SJunchao Zhang   ierr = VecScatterCreate(rglobal,from,is->y,NULL,&is->rctx);CHKERRQ(ierr);
24048546b261SStefano Zampini   ierr = ISLocalToGlobalMappingRestoreBlockIndices(A->rmap->mapping,&garray);CHKERRQ(ierr);
24058546b261SStefano Zampini   ierr = ISDestroy(&from);CHKERRQ(ierr);
24068546b261SStefano Zampini   if (A->rmap->mapping != A->cmap->mapping) {
24078546b261SStefano Zampini     ierr = ISLocalToGlobalMappingGetBlockIndices(A->cmap->mapping,&garray);CHKERRQ(ierr);
24088546b261SStefano Zampini     ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),cbs,nc/cbs,garray,PETSC_USE_POINTER,&from);CHKERRQ(ierr);
24099448b7f1SJunchao Zhang     ierr = VecScatterCreate(cglobal,from,is->x,NULL,&is->cctx);CHKERRQ(ierr);
24108546b261SStefano Zampini     ierr = ISLocalToGlobalMappingRestoreBlockIndices(A->cmap->mapping,&garray);CHKERRQ(ierr);
24118546b261SStefano Zampini     ierr = ISDestroy(&from);CHKERRQ(ierr);
24128546b261SStefano Zampini   } else {
24138546b261SStefano Zampini     ierr = PetscObjectReference((PetscObject)is->rctx);CHKERRQ(ierr);
24148546b261SStefano Zampini     is->cctx = is->rctx;
24158546b261SStefano Zampini   }
2416b89f26deSStefano Zampini   ierr = VecDestroy(&cglobal);CHKERRQ(ierr);
2417b89f26deSStefano Zampini 
24188546b261SStefano Zampini   /* interface counter vector (local) */
24198546b261SStefano Zampini   ierr = VecDuplicate(is->y,&is->counter);CHKERRQ(ierr);
2420b470e4b4SRichard Tran Mills   ierr = VecBindToCPU(is->counter,PETSC_TRUE);CHKERRQ(ierr);
24218546b261SStefano Zampini   ierr = VecSet(is->y,1.);CHKERRQ(ierr);
24228546b261SStefano Zampini   ierr = VecScatterBegin(is->rctx,is->y,rglobal,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
24238546b261SStefano Zampini   ierr = VecScatterEnd(is->rctx,is->y,rglobal,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
24248546b261SStefano Zampini   ierr = VecScatterBegin(is->rctx,rglobal,is->counter,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
24258546b261SStefano Zampini   ierr = VecScatterEnd(is->rctx,rglobal,is->counter,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2426b470e4b4SRichard Tran Mills   ierr = VecBindToCPU(is->y,PETSC_FALSE);CHKERRQ(ierr);
2427b470e4b4SRichard Tran Mills   ierr = VecBindToCPU(is->counter,PETSC_FALSE);CHKERRQ(ierr);
2428b89f26deSStefano Zampini 
2429b89f26deSStefano Zampini   /* special functions for block-diagonal matrices */
2430b89f26deSStefano Zampini   ierr = VecSum(rglobal,&sum);CHKERRQ(ierr);
2431b89f26deSStefano Zampini   if ((PetscInt)(PetscRealPart(sum)) == A->rmap->N && A->rmap->N == A->cmap->N && A->rmap->mapping == A->cmap->mapping) {
2432b89f26deSStefano Zampini     A->ops->invertblockdiagonal = MatInvertBlockDiagonal_IS;
2433b89f26deSStefano Zampini   } else {
2434b89f26deSStefano Zampini     A->ops->invertblockdiagonal = NULL;
2435b89f26deSStefano Zampini   }
24368546b261SStefano Zampini   ierr = VecDestroy(&rglobal);CHKERRQ(ierr);
2437b0cc1f67SStefano Zampini 
2438b0cc1f67SStefano Zampini   /* setup SF for general purpose shared indices based communications */
2439b0cc1f67SStefano Zampini   ierr = MatISSetUpSF_IS(A);CHKERRQ(ierr);
24408546b261SStefano Zampini   PetscFunctionReturn(0);
24418546b261SStefano Zampini }
24428546b261SStefano Zampini 
24438546b261SStefano Zampini static PetscErrorCode MatSetLocalToGlobalMapping_IS(Mat A,ISLocalToGlobalMapping rmapping,ISLocalToGlobalMapping cmapping)
24448546b261SStefano Zampini {
24458546b261SStefano Zampini   PetscErrorCode ierr;
24468546b261SStefano Zampini   PetscInt       nr,rbs,nc,cbs;
24478546b261SStefano Zampini   Mat_IS         *is = (Mat_IS*)A->data;
24488546b261SStefano Zampini 
24498546b261SStefano Zampini   PetscFunctionBegin;
24508546b261SStefano Zampini   PetscCheckSameComm(A,1,rmapping,2);
24518546b261SStefano Zampini   PetscCheckSameComm(A,1,cmapping,3);
24521c47cb0fSStefano Zampini   ierr = MatDestroy(&is->A);CHKERRQ(ierr);
2453872cf891SStefano Zampini   if (is->csf != is->sf) {
2454872cf891SStefano Zampini     ierr = PetscSFDestroy(&is->csf);CHKERRQ(ierr);
2455872cf891SStefano Zampini     ierr = PetscFree2(is->csf_rootdata,is->csf_leafdata);CHKERRQ(ierr);
2456f03112d0SStefano Zampini   } else is->csf = NULL;
245728f4e0baSStefano Zampini   ierr = PetscSFDestroy(&is->sf);CHKERRQ(ierr);
245828f4e0baSStefano Zampini   ierr = PetscFree2(is->sf_rootdata,is->sf_leafdata);CHKERRQ(ierr);
2459b89f26deSStefano Zampini   ierr = PetscFree(is->bdiag);CHKERRQ(ierr);
24603bbff08aSStefano Zampini 
24613bbff08aSStefano Zampini   /* Setup Layout and set local to global maps */
2462fc27028aSStefano Zampini   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
2463fc27028aSStefano Zampini   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
2464e176bc59SStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(rmapping,&nr);CHKERRQ(ierr);
2465e176bc59SStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(rmapping,&rbs);CHKERRQ(ierr);
2466e176bc59SStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(cmapping,&nc);CHKERRQ(ierr);
2467e176bc59SStefano Zampini   ierr = ISLocalToGlobalMappingGetBlockSize(cmapping,&cbs);CHKERRQ(ierr);
24686625354bSStefano Zampini   /* check if the two mappings are actually the same for square matrices (DOLFIN passes 2 different objects) */
24696625354bSStefano Zampini   if (rmapping != cmapping && A->rmap->N == A->cmap->N) {
24706625354bSStefano Zampini     PetscBool same,gsame;
24716625354bSStefano Zampini 
24726625354bSStefano Zampini     same = PETSC_FALSE;
24736625354bSStefano Zampini     if (nr == nc && cbs == rbs) {
24746625354bSStefano Zampini       const PetscInt *idxs1,*idxs2;
24756625354bSStefano Zampini 
24766625354bSStefano Zampini       ierr = ISLocalToGlobalMappingGetBlockIndices(rmapping,&idxs1);CHKERRQ(ierr);
24776625354bSStefano Zampini       ierr = ISLocalToGlobalMappingGetBlockIndices(cmapping,&idxs2);CHKERRQ(ierr);
2478580bdb30SBarry Smith       ierr = PetscArraycmp(idxs1,idxs2,nr/rbs,&same);CHKERRQ(ierr);
24796625354bSStefano Zampini       ierr = ISLocalToGlobalMappingRestoreBlockIndices(rmapping,&idxs1);CHKERRQ(ierr);
24806625354bSStefano Zampini       ierr = ISLocalToGlobalMappingRestoreBlockIndices(cmapping,&idxs2);CHKERRQ(ierr);
24816625354bSStefano Zampini     }
2482820f2d46SBarry Smith     ierr = MPIU_Allreduce(&same,&gsame,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
24836625354bSStefano Zampini     if (gsame) cmapping = rmapping;
24846625354bSStefano Zampini   }
248553b44cf5SStefano Zampini   ierr = PetscLayoutSetBlockSize(A->rmap,rbs);CHKERRQ(ierr);
248653b44cf5SStefano Zampini   ierr = PetscLayoutSetBlockSize(A->cmap,cbs);CHKERRQ(ierr);
24876625354bSStefano Zampini   ierr = PetscLayoutSetISLocalToGlobalMapping(A->rmap,rmapping);CHKERRQ(ierr);
24886625354bSStefano Zampini   ierr = PetscLayoutSetISLocalToGlobalMapping(A->cmap,cmapping);CHKERRQ(ierr);
24896625354bSStefano Zampini 
24906625354bSStefano Zampini   /* Create the local matrix A */
2491f69a0ea3SMatthew Knepley   ierr = MatCreate(PETSC_COMM_SELF,&is->A);CHKERRQ(ierr);
24928546b261SStefano Zampini   ierr = MatSetType(is->A,is->lmattype);CHKERRQ(ierr);
2493e176bc59SStefano Zampini   ierr = MatSetSizes(is->A,nr,nc,nr,nc);CHKERRQ(ierr);
2494e176bc59SStefano Zampini   ierr = MatSetBlockSizes(is->A,rbs,cbs);CHKERRQ(ierr);
24958546b261SStefano Zampini   ierr = MatSetOptionsPrefix(is->A,"is_");CHKERRQ(ierr);
24968546b261SStefano Zampini   ierr = MatAppendOptionsPrefix(is->A,((PetscObject)A)->prefix);CHKERRQ(ierr);
2497c77832edSStefano Zampini   ierr = PetscLayoutSetUp(is->A->rmap);CHKERRQ(ierr);
2498c77832edSStefano Zampini   ierr = PetscLayoutSetUp(is->A->cmap);CHKERRQ(ierr);
2499b4319ba4SBarry Smith 
2500f26d0771SStefano Zampini   if (!is->islocalref) { /* setup scatters and local vectors for MatMult */
25018546b261SStefano Zampini     ierr = MatISSetUpScatters_Private(A);CHKERRQ(ierr);
2502f26d0771SStefano Zampini   }
250348ff6bf3SStefano Zampini   ierr = MatSetUp(A);CHKERRQ(ierr);
2504b4319ba4SBarry Smith   PetscFunctionReturn(0);
2505b4319ba4SBarry Smith }
2506b4319ba4SBarry Smith 
2507a8116848SStefano Zampini static PetscErrorCode MatSetValues_IS(Mat mat, PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols, const PetscScalar *values, InsertMode addv)
25082e74eeadSLisandro Dalcin {
25092e74eeadSLisandro Dalcin   Mat_IS         *is = (Mat_IS*)mat->data;
25102e74eeadSLisandro Dalcin   PetscErrorCode ierr;
251197563a80SStefano Zampini   PetscInt       i,zm,zn;
2512f26d0771SStefano Zampini   PetscInt       rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION];
25132e74eeadSLisandro Dalcin 
25142e74eeadSLisandro Dalcin   PetscFunctionBegin;
251576bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
2516f26d0771SStefano Zampini     if (m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n);
251797563a80SStefano Zampini     /* count negative indices */
251897563a80SStefano Zampini     for (i=0,zm=0;i<m;i++) if (rows[i] < 0) zm++;
251997563a80SStefano Zampini     for (i=0,zn=0;i<n;i++) if (cols[i] < 0) zn++;
252076bd3646SJed Brown   }
252197563a80SStefano Zampini   ierr = ISGlobalToLocalMappingApply(mat->rmap->mapping,IS_GTOLM_MASK,m,rows,&m,rows_l);CHKERRQ(ierr);
252297563a80SStefano Zampini   ierr = ISGlobalToLocalMappingApply(mat->cmap->mapping,IS_GTOLM_MASK,n,cols,&n,cols_l);CHKERRQ(ierr);
252376bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
252497563a80SStefano Zampini     /* count negative indices (should be the same as before) */
252597563a80SStefano Zampini     for (i=0;i<m;i++) if (rows_l[i] < 0) zm--;
252697563a80SStefano Zampini     for (i=0;i<n;i++) if (cols_l[i] < 0) zn--;
2527b4f971dfSStefano Zampini     if (!is->A->rmap->mapping && zm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the row indices can not be mapped! Maybe you should not use MATIS");
2528b4f971dfSStefano Zampini     if (!is->A->cmap->mapping && zn) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the column indices can not be mapped! Maybe you should not use MATIS");
252976bd3646SJed Brown   }
25302e74eeadSLisandro Dalcin   ierr = MatSetValues(is->A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr);
25312e74eeadSLisandro Dalcin   PetscFunctionReturn(0);
25322e74eeadSLisandro Dalcin }
25332e74eeadSLisandro Dalcin 
2534a8116848SStefano Zampini static PetscErrorCode MatSetValuesBlocked_IS(Mat mat, PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols, const PetscScalar *values, InsertMode addv)
253597563a80SStefano Zampini {
253697563a80SStefano Zampini   Mat_IS         *is = (Mat_IS*)mat->data;
253797563a80SStefano Zampini   PetscErrorCode ierr;
253897563a80SStefano Zampini   PetscInt       i,zm,zn;
2539f26d0771SStefano Zampini   PetscInt       rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION];
254097563a80SStefano Zampini 
254197563a80SStefano Zampini   PetscFunctionBegin;
254276bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
2543f26d0771SStefano Zampini     if (m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column block indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n);
254497563a80SStefano Zampini     /* count negative indices */
254597563a80SStefano Zampini     for (i=0,zm=0;i<m;i++) if (rows[i] < 0) zm++;
254697563a80SStefano Zampini     for (i=0,zn=0;i<n;i++) if (cols[i] < 0) zn++;
254776bd3646SJed Brown   }
254897563a80SStefano Zampini   ierr = ISGlobalToLocalMappingApplyBlock(mat->rmap->mapping,IS_GTOLM_MASK,m,rows,&m,rows_l);CHKERRQ(ierr);
254997563a80SStefano Zampini   ierr = ISGlobalToLocalMappingApplyBlock(mat->cmap->mapping,IS_GTOLM_MASK,n,cols,&n,cols_l);CHKERRQ(ierr);
255076bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
255197563a80SStefano Zampini     /* count negative indices (should be the same as before) */
255297563a80SStefano Zampini     for (i=0;i<m;i++) if (rows_l[i] < 0) zm--;
255397563a80SStefano Zampini     for (i=0;i<n;i++) if (cols_l[i] < 0) zn--;
2554b4f971dfSStefano Zampini     if (!is->A->rmap->mapping && zm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the row indices can not be mapped! Maybe you should not use MATIS");
2555b4f971dfSStefano Zampini     if (!is->A->cmap->mapping && zn) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the column indices can not be mapped! Maybe you should not use MATIS");
255676bd3646SJed Brown   }
2557d59cf9ebSStefano Zampini   ierr = MatSetValuesBlocked(is->A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr);
255897563a80SStefano Zampini   PetscFunctionReturn(0);
255997563a80SStefano Zampini }
256097563a80SStefano Zampini 
2561a8116848SStefano Zampini static PetscErrorCode MatSetValuesLocal_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv)
2562b4319ba4SBarry Smith {
2563dfbe8321SBarry Smith   PetscErrorCode ierr;
2564b4319ba4SBarry Smith   Mat_IS         *is = (Mat_IS*)A->data;
2565b4319ba4SBarry Smith 
2566b4319ba4SBarry Smith   PetscFunctionBegin;
2567b4f971dfSStefano Zampini   if (is->A->rmap->mapping) {
2568872cf891SStefano Zampini     ierr = MatSetValuesLocal(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr);
2569872cf891SStefano Zampini   } else {
2570b4319ba4SBarry Smith     ierr = MatSetValues(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr);
2571872cf891SStefano Zampini   }
2572b4319ba4SBarry Smith   PetscFunctionReturn(0);
2573b4319ba4SBarry Smith }
2574b4319ba4SBarry Smith 
2575a8116848SStefano Zampini static PetscErrorCode MatSetValuesBlockedLocal_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv)
2576f0006bf2SLisandro Dalcin {
2577f0006bf2SLisandro Dalcin   PetscErrorCode ierr;
2578f0006bf2SLisandro Dalcin   Mat_IS         *is = (Mat_IS*)A->data;
2579f0006bf2SLisandro Dalcin 
2580f0006bf2SLisandro Dalcin   PetscFunctionBegin;
2581b4f971dfSStefano Zampini   if (is->A->rmap->mapping) {
258276bd3646SJed Brown     if (PetscDefined(USE_DEBUG)) {
2583b4f971dfSStefano Zampini       PetscInt ibs,bs;
2584b4f971dfSStefano Zampini 
2585b4f971dfSStefano Zampini       ierr = ISLocalToGlobalMappingGetBlockSize(is->A->rmap->mapping,&ibs);CHKERRQ(ierr);
2586b4f971dfSStefano Zampini       ierr = MatGetBlockSize(is->A,&bs);CHKERRQ(ierr);
2587b4f971dfSStefano Zampini       if (ibs != bs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Different block sizes! local mat %D, local l2g map %D",bs,ibs);
258876bd3646SJed Brown     }
2589b4f971dfSStefano Zampini     ierr = MatSetValuesBlockedLocal(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr);
2590b4f971dfSStefano Zampini   } else {
2591f0006bf2SLisandro Dalcin     ierr = MatSetValuesBlocked(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr);
2592b4f971dfSStefano Zampini   }
2593f0006bf2SLisandro Dalcin   PetscFunctionReturn(0);
2594f0006bf2SLisandro Dalcin }
2595f0006bf2SLisandro Dalcin 
2596f0ae7da4SStefano Zampini static PetscErrorCode MatISZeroRowsColumnsLocal_Private(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,PetscBool columns)
2597f0ae7da4SStefano Zampini {
2598f0ae7da4SStefano Zampini   Mat_IS         *is = (Mat_IS*)A->data;
2599f0ae7da4SStefano Zampini   PetscErrorCode ierr;
2600f0ae7da4SStefano Zampini 
2601f0ae7da4SStefano Zampini   PetscFunctionBegin;
2602f0ae7da4SStefano Zampini   if (!n) {
2603f0ae7da4SStefano Zampini     is->pure_neumann = PETSC_TRUE;
2604f0ae7da4SStefano Zampini   } else {
2605f0ae7da4SStefano Zampini     PetscInt i;
2606f0ae7da4SStefano Zampini     is->pure_neumann = PETSC_FALSE;
2607f0ae7da4SStefano Zampini 
2608f0ae7da4SStefano Zampini     if (columns) {
2609f4259b30SLisandro Dalcin       ierr = MatZeroRowsColumns(is->A,n,rows,diag,NULL,NULL);CHKERRQ(ierr);
2610f0ae7da4SStefano Zampini     } else {
2611f4259b30SLisandro Dalcin       ierr = MatZeroRows(is->A,n,rows,diag,NULL,NULL);CHKERRQ(ierr);
2612f0ae7da4SStefano Zampini     }
2613f0ae7da4SStefano Zampini     if (diag != 0.) {
2614f0ae7da4SStefano Zampini       const PetscScalar *array;
2615f0ae7da4SStefano Zampini       ierr = VecGetArrayRead(is->counter,&array);CHKERRQ(ierr);
2616f0ae7da4SStefano Zampini       for (i=0; i<n; i++) {
2617f0ae7da4SStefano Zampini         ierr = MatSetValue(is->A,rows[i],rows[i],diag/(array[rows[i]]),INSERT_VALUES);CHKERRQ(ierr);
2618f0ae7da4SStefano Zampini       }
2619f0ae7da4SStefano Zampini       ierr = VecRestoreArrayRead(is->counter,&array);CHKERRQ(ierr);
2620f0ae7da4SStefano Zampini     }
2621f0ae7da4SStefano Zampini     ierr = MatAssemblyBegin(is->A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2622f0ae7da4SStefano Zampini     ierr = MatAssemblyEnd(is->A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2623f0ae7da4SStefano Zampini   }
2624f0ae7da4SStefano Zampini   PetscFunctionReturn(0);
2625f0ae7da4SStefano Zampini }
2626f0ae7da4SStefano Zampini 
2627f0ae7da4SStefano Zampini static PetscErrorCode MatZeroRowsColumns_Private_IS(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,Vec x,Vec b,PetscBool columns)
26282e74eeadSLisandro Dalcin {
26296e520ac8SStefano Zampini   Mat_IS         *matis = (Mat_IS*)A->data;
26306e520ac8SStefano Zampini   PetscInt       nr,nl,len,i;
26316e520ac8SStefano Zampini   PetscInt       *lrows;
26322e74eeadSLisandro Dalcin   PetscErrorCode ierr;
26332e74eeadSLisandro Dalcin 
26342e74eeadSLisandro Dalcin   PetscFunctionBegin;
2635cf9c20a2SJed Brown   if (PetscUnlikelyDebug(columns || diag != 0. || (x && b))) {
2636f0ae7da4SStefano Zampini     PetscBool cong;
263726b0207aSStefano Zampini 
2638f0ae7da4SStefano Zampini     ierr = PetscLayoutCompare(A->rmap,A->cmap,&cong);CHKERRQ(ierr);
263926b0207aSStefano Zampini     cong = (PetscBool)(cong && matis->sf == matis->csf);
2640268753edSStefano Zampini     if (!cong && columns) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Columns can be zeroed if and only if A->rmap and A->cmap are congruent and the l2g maps are the same for MATIS");
2641268753edSStefano Zampini     if (!cong && diag != 0.) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Nonzero diagonal value supported if and only if A->rmap and A->cmap are congruent and the l2g maps are the same for MATIS");
2642268753edSStefano Zampini     if (!cong && x && b) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"A->rmap and A->cmap need to be congruent, and the l2g maps be the same");
2643f0ae7da4SStefano Zampini   }
26446e520ac8SStefano Zampini   /* get locally owned rows */
2645a72d46e8SStefano Zampini   ierr = PetscLayoutMapLocal(A->rmap,n,rows,&len,&lrows,NULL);CHKERRQ(ierr);
26466e520ac8SStefano Zampini   /* fix right hand side if needed */
26476e520ac8SStefano Zampini   if (x && b) {
26486e520ac8SStefano Zampini     const PetscScalar *xx;
26496e520ac8SStefano Zampini     PetscScalar       *bb;
26506e520ac8SStefano Zampini 
26516e520ac8SStefano Zampini     ierr = VecGetArrayRead(x, &xx);CHKERRQ(ierr);
26526e520ac8SStefano Zampini     ierr = VecGetArray(b, &bb);CHKERRQ(ierr);
26536e520ac8SStefano Zampini     for (i=0;i<len;++i) bb[lrows[i]] = diag*xx[lrows[i]];
26546e520ac8SStefano Zampini     ierr = VecRestoreArrayRead(x, &xx);CHKERRQ(ierr);
26556e520ac8SStefano Zampini     ierr = VecRestoreArray(b, &bb);CHKERRQ(ierr);
26562e74eeadSLisandro Dalcin   }
26576e520ac8SStefano Zampini   /* get rows associated to the local matrices */
26586e520ac8SStefano Zampini   ierr = MatGetSize(matis->A,&nl,NULL);CHKERRQ(ierr);
2659580bdb30SBarry Smith   ierr = PetscArrayzero(matis->sf_leafdata,nl);CHKERRQ(ierr);
2660580bdb30SBarry Smith   ierr = PetscArrayzero(matis->sf_rootdata,A->rmap->n);CHKERRQ(ierr);
26616e520ac8SStefano Zampini   for (i=0;i<len;i++) matis->sf_rootdata[lrows[i]] = 1;
26626e520ac8SStefano Zampini   ierr = PetscFree(lrows);CHKERRQ(ierr);
2663ad227feaSJunchao Zhang   ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
2664ad227feaSJunchao Zhang   ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr);
26656e520ac8SStefano Zampini   ierr = PetscMalloc1(nl,&lrows);CHKERRQ(ierr);
26666e520ac8SStefano Zampini   for (i=0,nr=0;i<nl;i++) if (matis->sf_leafdata[i]) lrows[nr++] = i;
2667f0ae7da4SStefano Zampini   ierr = MatISZeroRowsColumnsLocal_Private(A,nr,lrows,diag,columns);CHKERRQ(ierr);
26686e520ac8SStefano Zampini   ierr = PetscFree(lrows);CHKERRQ(ierr);
26692e74eeadSLisandro Dalcin   PetscFunctionReturn(0);
26702e74eeadSLisandro Dalcin }
26712e74eeadSLisandro Dalcin 
2672f0ae7da4SStefano Zampini static PetscErrorCode MatZeroRows_IS(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
2673b4319ba4SBarry Smith {
2674dfbe8321SBarry Smith   PetscErrorCode ierr;
2675b4319ba4SBarry Smith 
2676b4319ba4SBarry Smith   PetscFunctionBegin;
2677f0ae7da4SStefano Zampini   ierr = MatZeroRowsColumns_Private_IS(A,n,rows,diag,x,b,PETSC_FALSE);CHKERRQ(ierr);
2678f0ae7da4SStefano Zampini   PetscFunctionReturn(0);
2679f0ae7da4SStefano Zampini }
26802205254eSKarl Rupp 
2681f0ae7da4SStefano Zampini static PetscErrorCode MatZeroRowsColumns_IS(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
2682f0ae7da4SStefano Zampini {
2683f0ae7da4SStefano Zampini   PetscErrorCode ierr;
2684f0ae7da4SStefano Zampini 
2685f0ae7da4SStefano Zampini   PetscFunctionBegin;
2686f0ae7da4SStefano Zampini   ierr = MatZeroRowsColumns_Private_IS(A,n,rows,diag,x,b,PETSC_TRUE);CHKERRQ(ierr);
2687b4319ba4SBarry Smith   PetscFunctionReturn(0);
2688b4319ba4SBarry Smith }
2689b4319ba4SBarry Smith 
2690a8116848SStefano Zampini static PetscErrorCode MatAssemblyBegin_IS(Mat A,MatAssemblyType type)
2691b4319ba4SBarry Smith {
2692b4319ba4SBarry Smith   Mat_IS         *is = (Mat_IS*)A->data;
2693dfbe8321SBarry Smith   PetscErrorCode ierr;
2694b4319ba4SBarry Smith 
2695b4319ba4SBarry Smith   PetscFunctionBegin;
2696b4319ba4SBarry Smith   ierr = MatAssemblyBegin(is->A,type);CHKERRQ(ierr);
2697b4319ba4SBarry Smith   PetscFunctionReturn(0);
2698b4319ba4SBarry Smith }
2699b4319ba4SBarry Smith 
2700a8116848SStefano Zampini static PetscErrorCode MatAssemblyEnd_IS(Mat A,MatAssemblyType type)
2701b4319ba4SBarry Smith {
2702b4319ba4SBarry Smith   Mat_IS         *is = (Mat_IS*)A->data;
2703dfbe8321SBarry Smith   PetscErrorCode ierr;
2704b4319ba4SBarry Smith 
2705b4319ba4SBarry Smith   PetscFunctionBegin;
2706b4319ba4SBarry Smith   ierr = MatAssemblyEnd(is->A,type);CHKERRQ(ierr);
2707872cf891SStefano Zampini   /* fix for local empty rows/cols */
2708872cf891SStefano Zampini   if (is->locempty && type == MAT_FINAL_ASSEMBLY) {
2709872cf891SStefano Zampini     Mat                    newlA;
2710f03112d0SStefano Zampini     ISLocalToGlobalMapping rl2g,cl2g;
2711f03112d0SStefano Zampini     IS                     nzr,nzc;
2712f03112d0SStefano Zampini     PetscInt               nr,nc,nnzr,nnzc;
2713f03112d0SStefano Zampini     PetscBool              lnewl2g,newl2g;
2714872cf891SStefano Zampini 
2715f03112d0SStefano Zampini     ierr = MatGetSize(is->A,&nr,&nc);CHKERRQ(ierr);
2716f03112d0SStefano Zampini     ierr = MatFindNonzeroRowsOrCols_Basic(is->A,PETSC_FALSE,PETSC_SMALL,&nzr);CHKERRQ(ierr);
2717f03112d0SStefano Zampini     if (!nzr) {
2718f03112d0SStefano Zampini       ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nr,0,1,&nzr);CHKERRQ(ierr);
2719872cf891SStefano Zampini     }
2720f03112d0SStefano Zampini     ierr = MatFindNonzeroRowsOrCols_Basic(is->A,PETSC_TRUE,PETSC_SMALL,&nzc);CHKERRQ(ierr);
2721f03112d0SStefano Zampini     if (!nzc) {
2722f03112d0SStefano Zampini       ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nc,0,1,&nzc);CHKERRQ(ierr);
2723872cf891SStefano Zampini     }
2724f03112d0SStefano Zampini     ierr = ISGetSize(nzr,&nnzr);CHKERRQ(ierr);
2725f03112d0SStefano Zampini     ierr = ISGetSize(nzc,&nnzc);CHKERRQ(ierr);
2726f03112d0SStefano Zampini     if (nnzr != nr || nnzc != nc) {
2727f03112d0SStefano Zampini       ISLocalToGlobalMapping l2g;
2728f03112d0SStefano Zampini       IS                     is1,is2;
2729f03112d0SStefano Zampini 
2730f03112d0SStefano Zampini       /* need new global l2g map */
2731f03112d0SStefano Zampini       lnewl2g = PETSC_TRUE;
273255b25c41SPierre Jolivet       ierr    = MPI_Allreduce(&lnewl2g,&newl2g,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
2733f03112d0SStefano Zampini 
2734872cf891SStefano Zampini       /* extract valid submatrix */
2735f03112d0SStefano Zampini       ierr = MatCreateSubMatrix(is->A,nzr,nzc,MAT_INITIAL_MATRIX,&newlA);CHKERRQ(ierr);
2736f03112d0SStefano Zampini 
2737f03112d0SStefano Zampini       /* attach local l2g maps for successive calls of MatSetValues on the local matrix */
2738f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(nzr,&l2g);CHKERRQ(ierr);
2739f03112d0SStefano Zampini       ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nr,0,1,&is1);CHKERRQ(ierr);
2740f03112d0SStefano Zampini       ierr = ISGlobalToLocalMappingApplyIS(l2g,IS_GTOLM_MASK,is1,&is2);CHKERRQ(ierr);
2741872cf891SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&l2g);CHKERRQ(ierr);
274272ed36d8SStefano Zampini       if (is->A->rmap->mapping) { /* the matrix has a local-to-local map already attached (probably DMDA generated) */
274372ed36d8SStefano Zampini         const PetscInt *idxs1,*idxs2;
274472ed36d8SStefano Zampini         PetscInt       j,i,nl,*tidxs;
274572ed36d8SStefano Zampini 
274672ed36d8SStefano Zampini         ierr = ISLocalToGlobalMappingGetSize(is->A->rmap->mapping,&nl);CHKERRQ(ierr);
274772ed36d8SStefano Zampini         ierr = ISLocalToGlobalMappingGetIndices(is->A->rmap->mapping,&idxs1);CHKERRQ(ierr);
274872ed36d8SStefano Zampini         ierr = PetscMalloc1(nl,&tidxs);CHKERRQ(ierr);
274972ed36d8SStefano Zampini         ierr = ISGetIndices(is2,&idxs2);CHKERRQ(ierr);
275072ed36d8SStefano Zampini         for (i=0,j=0;i<nl;i++) tidxs[i] = idxs1[i] < 0 ? -1 : idxs2[j++];
275172ed36d8SStefano Zampini         if (j != nr) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected count %D != %D",j,nr);
275272ed36d8SStefano Zampini         ierr = ISLocalToGlobalMappingRestoreIndices(is->A->rmap->mapping,&idxs1);CHKERRQ(ierr);
275372ed36d8SStefano Zampini         ierr = ISRestoreIndices(is2,&idxs2);CHKERRQ(ierr);
275472ed36d8SStefano Zampini         ierr = ISDestroy(&is2);CHKERRQ(ierr);
275572ed36d8SStefano Zampini         ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is->A->rmap->mapping),nl,tidxs,PETSC_OWN_POINTER,&is2);CHKERRQ(ierr);
275672ed36d8SStefano Zampini       }
2757f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(is2,&rl2g);CHKERRQ(ierr);
2758f03112d0SStefano Zampini       ierr = ISDestroy(&is1);CHKERRQ(ierr);
2759f03112d0SStefano Zampini       ierr = ISDestroy(&is2);CHKERRQ(ierr);
276072ed36d8SStefano Zampini 
2761f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(nzc,&l2g);CHKERRQ(ierr);
2762f03112d0SStefano Zampini       ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nc,0,1,&is1);CHKERRQ(ierr);
2763f03112d0SStefano Zampini       ierr = ISGlobalToLocalMappingApplyIS(l2g,IS_GTOLM_MASK,is1,&is2);CHKERRQ(ierr);
2764f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&l2g);CHKERRQ(ierr);
276572ed36d8SStefano Zampini       if (is->A->cmap->mapping) { /* the matrix has a local-to-local map already attached (probably DMDA generated) */
276672ed36d8SStefano Zampini         const PetscInt *idxs1,*idxs2;
276772ed36d8SStefano Zampini         PetscInt       j,i,nl,*tidxs;
276872ed36d8SStefano Zampini 
276972ed36d8SStefano Zampini         ierr = ISLocalToGlobalMappingGetSize(is->A->cmap->mapping,&nl);CHKERRQ(ierr);
277072ed36d8SStefano Zampini         ierr = ISLocalToGlobalMappingGetIndices(is->A->cmap->mapping,&idxs1);CHKERRQ(ierr);
277172ed36d8SStefano Zampini         ierr = PetscMalloc1(nl,&tidxs);CHKERRQ(ierr);
277272ed36d8SStefano Zampini         ierr = ISGetIndices(is2,&idxs2);CHKERRQ(ierr);
277372ed36d8SStefano Zampini         for (i=0,j=0;i<nl;i++) tidxs[i] = idxs1[i] < 0 ? -1 : idxs2[j++];
277472ed36d8SStefano Zampini         if (j != nc) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected count %D != %D",j,nc);
277572ed36d8SStefano Zampini         ierr = ISLocalToGlobalMappingRestoreIndices(is->A->cmap->mapping,&idxs1);CHKERRQ(ierr);
277672ed36d8SStefano Zampini         ierr = ISRestoreIndices(is2,&idxs2);CHKERRQ(ierr);
277772ed36d8SStefano Zampini         ierr = ISDestroy(&is2);CHKERRQ(ierr);
277872ed36d8SStefano Zampini         ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is->A->cmap->mapping),nl,tidxs,PETSC_OWN_POINTER,&is2);CHKERRQ(ierr);
277972ed36d8SStefano Zampini       }
2780f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingCreateIS(is2,&cl2g);CHKERRQ(ierr);
2781f03112d0SStefano Zampini       ierr = ISDestroy(&is1);CHKERRQ(ierr);
2782f03112d0SStefano Zampini       ierr = ISDestroy(&is2);CHKERRQ(ierr);
278372ed36d8SStefano Zampini 
2784f03112d0SStefano Zampini       ierr = MatSetLocalToGlobalMapping(newlA,rl2g,cl2g);CHKERRQ(ierr);
278572ed36d8SStefano Zampini 
2786f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
2787f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
2788f03112d0SStefano Zampini     } else { /* local matrix fully populated */
2789f03112d0SStefano Zampini       lnewl2g = PETSC_FALSE;
279055b25c41SPierre Jolivet       ierr    = MPI_Allreduce(&lnewl2g,&newl2g,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr);
2791f03112d0SStefano Zampini       ierr    = PetscObjectReference((PetscObject)is->A);CHKERRQ(ierr);
2792f03112d0SStefano Zampini       newlA   = is->A;
2793f03112d0SStefano Zampini     }
2794f03112d0SStefano Zampini     /* attach new global l2g map if needed */
2795f03112d0SStefano Zampini     if (newl2g) {
2796f03112d0SStefano Zampini       IS             gnzr,gnzc;
2797f03112d0SStefano Zampini       const PetscInt *grid,*gcid;
2798f03112d0SStefano Zampini 
2799f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingApplyIS(A->rmap->mapping,nzr,&gnzr);CHKERRQ(ierr);
2800f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingApplyIS(A->cmap->mapping,nzc,&gnzc);CHKERRQ(ierr);
2801f03112d0SStefano Zampini       ierr = ISGetIndices(gnzr,&grid);CHKERRQ(ierr);
2802f03112d0SStefano Zampini       ierr = ISGetIndices(gnzc,&gcid);CHKERRQ(ierr);
2803f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)A),1,nnzr,grid,PETSC_COPY_VALUES,&rl2g);CHKERRQ(ierr);
2804f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)A),1,nnzc,gcid,PETSC_COPY_VALUES,&cl2g);CHKERRQ(ierr);
2805f03112d0SStefano Zampini       ierr = ISRestoreIndices(gnzr,&grid);CHKERRQ(ierr);
2806f03112d0SStefano Zampini       ierr = ISRestoreIndices(gnzc,&gcid);CHKERRQ(ierr);
2807f03112d0SStefano Zampini       ierr = ISDestroy(&gnzr);CHKERRQ(ierr);
2808f03112d0SStefano Zampini       ierr = ISDestroy(&gnzc);CHKERRQ(ierr);
2809f03112d0SStefano Zampini       ierr = MatSetLocalToGlobalMapping(A,rl2g,cl2g);CHKERRQ(ierr);
2810f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
2811f03112d0SStefano Zampini       ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
2812f03112d0SStefano Zampini     }
2813872cf891SStefano Zampini     ierr = MatISSetLocalMat(A,newlA);CHKERRQ(ierr);
2814872cf891SStefano Zampini     ierr = MatDestroy(&newlA);CHKERRQ(ierr);
2815f03112d0SStefano Zampini     ierr = ISDestroy(&nzr);CHKERRQ(ierr);
2816f03112d0SStefano Zampini     ierr = ISDestroy(&nzc);CHKERRQ(ierr);
2817872cf891SStefano Zampini     is->locempty = PETSC_FALSE;
2818f03112d0SStefano Zampini   }
2819b4319ba4SBarry Smith   PetscFunctionReturn(0);
2820b4319ba4SBarry Smith }
2821b4319ba4SBarry Smith 
2822a8116848SStefano Zampini static PetscErrorCode MatISGetLocalMat_IS(Mat mat,Mat *local)
2823b4319ba4SBarry Smith {
2824b4319ba4SBarry Smith   Mat_IS *is = (Mat_IS*)mat->data;
2825b4319ba4SBarry Smith 
2826b4319ba4SBarry Smith   PetscFunctionBegin;
2827b4319ba4SBarry Smith   *local = is->A;
2828b4319ba4SBarry Smith   PetscFunctionReturn(0);
2829b4319ba4SBarry Smith }
2830b4319ba4SBarry Smith 
28313b3b1effSJed Brown static PetscErrorCode MatISRestoreLocalMat_IS(Mat mat,Mat *local)
28323b3b1effSJed Brown {
28333b3b1effSJed Brown   PetscFunctionBegin;
28343b3b1effSJed Brown   *local = NULL;
28353b3b1effSJed Brown   PetscFunctionReturn(0);
28363b3b1effSJed Brown }
28373b3b1effSJed Brown 
2838b4319ba4SBarry Smith /*@
2839b4319ba4SBarry Smith     MatISGetLocalMat - Gets the local matrix stored inside a MATIS matrix.
2840b4319ba4SBarry Smith 
2841b4319ba4SBarry Smith   Input Parameter:
2842b4319ba4SBarry Smith .  mat - the matrix
2843b4319ba4SBarry Smith 
2844b4319ba4SBarry Smith   Output Parameter:
2845eb82efa4SStefano Zampini .  local - the local matrix
2846b4319ba4SBarry Smith 
2847b4319ba4SBarry Smith   Level: advanced
2848b4319ba4SBarry Smith 
2849b4319ba4SBarry Smith   Notes:
2850b4319ba4SBarry Smith     This can be called if you have precomputed the nonzero structure of the
2851b4319ba4SBarry Smith   matrix and want to provide it to the inner matrix object to improve the performance
2852b4319ba4SBarry Smith   of the MatSetValues() operation.
2853b4319ba4SBarry Smith 
28543b3b1effSJed Brown   Call MatISRestoreLocalMat() when finished with the local matrix.
285596a6f129SJed Brown 
2856b4319ba4SBarry Smith .seealso: MATIS
2857b4319ba4SBarry Smith @*/
28587087cfbeSBarry Smith PetscErrorCode MatISGetLocalMat(Mat mat,Mat *local)
2859b4319ba4SBarry Smith {
28604ac538c5SBarry Smith   PetscErrorCode ierr;
2861b4319ba4SBarry Smith 
2862b4319ba4SBarry Smith   PetscFunctionBegin;
28630700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2864b4319ba4SBarry Smith   PetscValidPointer(local,2);
28654ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatISGetLocalMat_C",(Mat,Mat*),(mat,local));CHKERRQ(ierr);
2866b4319ba4SBarry Smith   PetscFunctionReturn(0);
2867b4319ba4SBarry Smith }
2868b4319ba4SBarry Smith 
28693b3b1effSJed Brown /*@
28703b3b1effSJed Brown     MatISRestoreLocalMat - Restores the local matrix obtained with MatISGetLocalMat()
28713b3b1effSJed Brown 
28723b3b1effSJed Brown   Input Parameter:
28733b3b1effSJed Brown .  mat - the matrix
28743b3b1effSJed Brown 
28753b3b1effSJed Brown   Output Parameter:
28763b3b1effSJed Brown .  local - the local matrix
28773b3b1effSJed Brown 
28783b3b1effSJed Brown   Level: advanced
28793b3b1effSJed Brown 
28803b3b1effSJed Brown .seealso: MATIS
28813b3b1effSJed Brown @*/
28823b3b1effSJed Brown PetscErrorCode MatISRestoreLocalMat(Mat mat,Mat *local)
28833b3b1effSJed Brown {
28843b3b1effSJed Brown   PetscErrorCode ierr;
28853b3b1effSJed Brown 
28863b3b1effSJed Brown   PetscFunctionBegin;
28873b3b1effSJed Brown   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
28883b3b1effSJed Brown   PetscValidPointer(local,2);
28893b3b1effSJed Brown   ierr = PetscUseMethod(mat,"MatISRestoreLocalMat_C",(Mat,Mat*),(mat,local));CHKERRQ(ierr);
28903b3b1effSJed Brown   PetscFunctionReturn(0);
28913b3b1effSJed Brown }
28923b3b1effSJed Brown 
28938546b261SStefano Zampini static PetscErrorCode MatISSetLocalMatType_IS(Mat mat,MatType mtype)
28948546b261SStefano Zampini {
28958546b261SStefano Zampini   Mat_IS         *is = (Mat_IS*)mat->data;
28968546b261SStefano Zampini   PetscErrorCode ierr;
28978546b261SStefano Zampini 
28988546b261SStefano Zampini   PetscFunctionBegin;
28998546b261SStefano Zampini   if (is->A) {
29008546b261SStefano Zampini     ierr = MatSetType(is->A,mtype);CHKERRQ(ierr);
29018546b261SStefano Zampini   }
29028546b261SStefano Zampini   ierr = PetscFree(is->lmattype);CHKERRQ(ierr);
29038546b261SStefano Zampini   ierr = PetscStrallocpy(mtype,&is->lmattype);CHKERRQ(ierr);
29048546b261SStefano Zampini   PetscFunctionReturn(0);
29058546b261SStefano Zampini }
29068546b261SStefano Zampini 
29078546b261SStefano Zampini /*@
29088546b261SStefano Zampini     MatISSetLocalMatType - Specifies the type of local matrix
29098546b261SStefano Zampini 
2910*d8d19677SJose E. Roman   Input Parameters:
2911a2b725a8SWilliam Gropp +  mat - the matrix
2912a2b725a8SWilliam Gropp -  mtype - the local matrix type
29138546b261SStefano Zampini 
29148546b261SStefano Zampini   Output Parameter:
29158546b261SStefano Zampini 
29168546b261SStefano Zampini   Level: advanced
29178546b261SStefano Zampini 
29188546b261SStefano Zampini .seealso: MATIS, MatSetType(), MatType
29198546b261SStefano Zampini @*/
29208546b261SStefano Zampini PetscErrorCode MatISSetLocalMatType(Mat mat,MatType mtype)
29218546b261SStefano Zampini {
29228546b261SStefano Zampini   PetscErrorCode ierr;
29238546b261SStefano Zampini 
29248546b261SStefano Zampini   PetscFunctionBegin;
29258546b261SStefano Zampini   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
29268546b261SStefano Zampini   ierr = PetscUseMethod(mat,"MatISSetLocalMatType_C",(Mat,MatType),(mat,mtype));CHKERRQ(ierr);
29278546b261SStefano Zampini   PetscFunctionReturn(0);
29288546b261SStefano Zampini }
29298546b261SStefano Zampini 
2930a8116848SStefano Zampini static PetscErrorCode MatISSetLocalMat_IS(Mat mat,Mat local)
29313b03a366Sstefano_zampini {
29323b03a366Sstefano_zampini   Mat_IS         *is = (Mat_IS*)mat->data;
29333b03a366Sstefano_zampini   PetscInt       nrows,ncols,orows,ocols;
29343b03a366Sstefano_zampini   PetscErrorCode ierr;
29358546b261SStefano Zampini   MatType        mtype,otype;
29368546b261SStefano Zampini   PetscBool      sametype = PETSC_TRUE;
29373b03a366Sstefano_zampini 
29383b03a366Sstefano_zampini   PetscFunctionBegin;
29394e4c7dbeSStefano Zampini   if (is->A) {
29403b03a366Sstefano_zampini     ierr = MatGetSize(is->A,&orows,&ocols);CHKERRQ(ierr);
29413b03a366Sstefano_zampini     ierr = MatGetSize(local,&nrows,&ncols);CHKERRQ(ierr);
2942f0ae7da4SStefano Zampini     if (orows != nrows || ocols != ncols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Local MATIS matrix should be of size %Dx%D (you passed a %Dx%D matrix)",orows,ocols,nrows,ncols);
29438546b261SStefano Zampini     ierr = MatGetType(local,&mtype);CHKERRQ(ierr);
29448546b261SStefano Zampini     ierr = MatGetType(is->A,&otype);CHKERRQ(ierr);
29458546b261SStefano Zampini     ierr = PetscStrcmp(mtype,otype,&sametype);CHKERRQ(ierr);
29464e4c7dbeSStefano Zampini   }
29473b03a366Sstefano_zampini   ierr  = PetscObjectReference((PetscObject)local);CHKERRQ(ierr);
29483b03a366Sstefano_zampini   ierr  = MatDestroy(&is->A);CHKERRQ(ierr);
29493b03a366Sstefano_zampini   is->A = local;
29508546b261SStefano Zampini   ierr  = MatGetType(is->A,&mtype);CHKERRQ(ierr);
29518546b261SStefano Zampini   ierr  = MatISSetLocalMatType(mat,mtype);CHKERRQ(ierr);
29528546b261SStefano Zampini   if (!sametype && !is->islocalref) {
29538546b261SStefano Zampini     ierr = MatISSetUpScatters_Private(mat);CHKERRQ(ierr);
29548546b261SStefano Zampini   }
29553b03a366Sstefano_zampini   PetscFunctionReturn(0);
29563b03a366Sstefano_zampini }
29573b03a366Sstefano_zampini 
29583b03a366Sstefano_zampini /*@
2959eb82efa4SStefano Zampini     MatISSetLocalMat - Replace the local matrix stored inside a MATIS object.
29603b03a366Sstefano_zampini 
29618546b261SStefano Zampini   Collective on Mat
29628546b261SStefano Zampini 
2963*d8d19677SJose E. Roman   Input Parameters:
2964a2b725a8SWilliam Gropp +  mat - the matrix
2965a2b725a8SWilliam Gropp -  local - the local matrix
29663b03a366Sstefano_zampini 
29673b03a366Sstefano_zampini   Output Parameter:
29683b03a366Sstefano_zampini 
29693b03a366Sstefano_zampini   Level: advanced
29703b03a366Sstefano_zampini 
29713b03a366Sstefano_zampini   Notes:
29723b03a366Sstefano_zampini     This can be called if you have precomputed the local matrix and
29733b03a366Sstefano_zampini   want to provide it to the matrix object MATIS.
29743b03a366Sstefano_zampini 
29753b03a366Sstefano_zampini .seealso: MATIS
29763b03a366Sstefano_zampini @*/
29773b03a366Sstefano_zampini PetscErrorCode MatISSetLocalMat(Mat mat,Mat local)
29783b03a366Sstefano_zampini {
29793b03a366Sstefano_zampini   PetscErrorCode ierr;
29803b03a366Sstefano_zampini 
29813b03a366Sstefano_zampini   PetscFunctionBegin;
29823b03a366Sstefano_zampini   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2983b7ce53b6SStefano Zampini   PetscValidHeaderSpecific(local,MAT_CLASSID,2);
29843b03a366Sstefano_zampini   ierr = PetscUseMethod(mat,"MatISSetLocalMat_C",(Mat,Mat),(mat,local));CHKERRQ(ierr);
29853b03a366Sstefano_zampini   PetscFunctionReturn(0);
29863b03a366Sstefano_zampini }
29873b03a366Sstefano_zampini 
2988a8116848SStefano Zampini static PetscErrorCode MatZeroEntries_IS(Mat A)
29896726f965SBarry Smith {
29906726f965SBarry Smith   Mat_IS         *a = (Mat_IS*)A->data;
29916726f965SBarry Smith   PetscErrorCode ierr;
29926726f965SBarry Smith 
29936726f965SBarry Smith   PetscFunctionBegin;
29946726f965SBarry Smith   ierr = MatZeroEntries(a->A);CHKERRQ(ierr);
29956726f965SBarry Smith   PetscFunctionReturn(0);
29966726f965SBarry Smith }
29976726f965SBarry Smith 
2998a8116848SStefano Zampini static PetscErrorCode MatScale_IS(Mat A,PetscScalar a)
29992e74eeadSLisandro Dalcin {
30002e74eeadSLisandro Dalcin   Mat_IS         *is = (Mat_IS*)A->data;
30012e74eeadSLisandro Dalcin   PetscErrorCode ierr;
30022e74eeadSLisandro Dalcin 
30032e74eeadSLisandro Dalcin   PetscFunctionBegin;
30042e74eeadSLisandro Dalcin   ierr = MatScale(is->A,a);CHKERRQ(ierr);
30052e74eeadSLisandro Dalcin   PetscFunctionReturn(0);
30062e74eeadSLisandro Dalcin }
30072e74eeadSLisandro Dalcin 
3008a8116848SStefano Zampini static PetscErrorCode MatGetDiagonal_IS(Mat A, Vec v)
30092e74eeadSLisandro Dalcin {
30102e74eeadSLisandro Dalcin   Mat_IS         *is = (Mat_IS*)A->data;
30112e74eeadSLisandro Dalcin   PetscErrorCode ierr;
30122e74eeadSLisandro Dalcin 
30132e74eeadSLisandro Dalcin   PetscFunctionBegin;
30142e74eeadSLisandro Dalcin   /* get diagonal of the local matrix */
3015e176bc59SStefano Zampini   ierr = MatGetDiagonal(is->A,is->y);CHKERRQ(ierr);
30162e74eeadSLisandro Dalcin 
30172e74eeadSLisandro Dalcin   /* scatter diagonal back into global vector */
30182e74eeadSLisandro Dalcin   ierr = VecSet(v,0);CHKERRQ(ierr);
3019e176bc59SStefano Zampini   ierr = VecScatterBegin(is->rctx,is->y,v,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
3020e176bc59SStefano Zampini   ierr = VecScatterEnd(is->rctx,is->y,v,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
30212e74eeadSLisandro Dalcin   PetscFunctionReturn(0);
30222e74eeadSLisandro Dalcin }
30232e74eeadSLisandro Dalcin 
3024a8116848SStefano Zampini static PetscErrorCode MatSetOption_IS(Mat A,MatOption op,PetscBool flg)
30256726f965SBarry Smith {
30266726f965SBarry Smith   Mat_IS         *a = (Mat_IS*)A->data;
30276726f965SBarry Smith   PetscErrorCode ierr;
30286726f965SBarry Smith 
30296726f965SBarry Smith   PetscFunctionBegin;
30304e0d8c25SBarry Smith   ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
30316726f965SBarry Smith   PetscFunctionReturn(0);
30326726f965SBarry Smith }
30336726f965SBarry Smith 
3034f26d0771SStefano Zampini static PetscErrorCode MatAXPY_IS(Mat Y,PetscScalar a,Mat X,MatStructure str)
3035f26d0771SStefano Zampini {
3036f26d0771SStefano Zampini   Mat_IS         *y = (Mat_IS*)Y->data;
3037f26d0771SStefano Zampini   Mat_IS         *x;
3038f26d0771SStefano Zampini   PetscErrorCode ierr;
3039f26d0771SStefano Zampini 
3040f26d0771SStefano Zampini   PetscFunctionBegin;
304176bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
304276bd3646SJed Brown     PetscBool      ismatis;
3043f26d0771SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)X,MATIS,&ismatis);CHKERRQ(ierr);
3044f26d0771SStefano Zampini     if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)Y),PETSC_ERR_SUP,"Cannot call MatAXPY(Y,a,X,str) with X not of type MATIS");
304576bd3646SJed Brown   }
3046f26d0771SStefano Zampini   x = (Mat_IS*)X->data;
3047f26d0771SStefano Zampini   ierr = MatAXPY(y->A,a,x->A,str);CHKERRQ(ierr);
3048f26d0771SStefano Zampini   PetscFunctionReturn(0);
3049f26d0771SStefano Zampini }
3050f26d0771SStefano Zampini 
3051f26d0771SStefano Zampini static PetscErrorCode MatGetLocalSubMatrix_IS(Mat A,IS row,IS col,Mat *submat)
3052f26d0771SStefano Zampini {
3053f26d0771SStefano Zampini   Mat                    lA;
3054f26d0771SStefano Zampini   Mat_IS                 *matis;
3055f26d0771SStefano Zampini   ISLocalToGlobalMapping rl2g,cl2g;
3056f26d0771SStefano Zampini   IS                     is;
3057f26d0771SStefano Zampini   const PetscInt         *rg,*rl;
3058f26d0771SStefano Zampini   PetscInt               nrg;
3059f26d0771SStefano Zampini   PetscInt               N,M,nrl,i,*idxs;
3060f26d0771SStefano Zampini   PetscErrorCode         ierr;
3061f26d0771SStefano Zampini 
3062f26d0771SStefano Zampini   PetscFunctionBegin;
3063f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingGetIndices(A->rmap->mapping,&rg);CHKERRQ(ierr);
3064f26d0771SStefano Zampini   ierr = ISGetLocalSize(row,&nrl);CHKERRQ(ierr);
3065f26d0771SStefano Zampini   ierr = ISGetIndices(row,&rl);CHKERRQ(ierr);
3066f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(A->rmap->mapping,&nrg);CHKERRQ(ierr);
306776bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
3068249c8a71SStefano Zampini     for (i=0; i<nrl; i++) if (rl[i]>=nrg) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Local row index %D -> %D greater then maximum possible %D",i,rl[i],nrg);
306976bd3646SJed Brown   }
3070f26d0771SStefano Zampini   ierr = PetscMalloc1(nrg,&idxs);CHKERRQ(ierr);
3071f26d0771SStefano Zampini   /* map from [0,nrl) to row */
3072f26d0771SStefano Zampini   for (i=0;i<nrl;i++) idxs[i] = rl[i];
3073f26d0771SStefano Zampini   for (i=nrl;i<nrg;i++) idxs[i] = -1;
3074f26d0771SStefano Zampini   ierr = ISRestoreIndices(row,&rl);CHKERRQ(ierr);
3075f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingRestoreIndices(A->rmap->mapping,&rg);CHKERRQ(ierr);
3076f26d0771SStefano Zampini   ierr = ISCreateGeneral(PetscObjectComm((PetscObject)A),nrg,idxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
3077f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr);
3078f26d0771SStefano Zampini   ierr = ISDestroy(&is);CHKERRQ(ierr);
3079f26d0771SStefano Zampini   /* compute new l2g map for columns */
3080f26d0771SStefano Zampini   if (col != row || A->rmap->mapping != A->cmap->mapping) {
3081f26d0771SStefano Zampini     const PetscInt *cg,*cl;
3082f26d0771SStefano Zampini     PetscInt       ncg;
3083f26d0771SStefano Zampini     PetscInt       ncl;
3084f26d0771SStefano Zampini 
3085f26d0771SStefano Zampini     ierr = ISLocalToGlobalMappingGetIndices(A->cmap->mapping,&cg);CHKERRQ(ierr);
3086f26d0771SStefano Zampini     ierr = ISGetLocalSize(col,&ncl);CHKERRQ(ierr);
3087f26d0771SStefano Zampini     ierr = ISGetIndices(col,&cl);CHKERRQ(ierr);
3088f26d0771SStefano Zampini     ierr = ISLocalToGlobalMappingGetSize(A->cmap->mapping,&ncg);CHKERRQ(ierr);
308976bd3646SJed Brown     if (PetscDefined(USE_DEBUG)) {
3090249c8a71SStefano Zampini       for (i=0; i<ncl; i++) if (cl[i]>=ncg) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Local column index %D -> %D greater then maximum possible %D",i,cl[i],ncg);
309176bd3646SJed Brown     }
3092f26d0771SStefano Zampini     ierr = PetscMalloc1(ncg,&idxs);CHKERRQ(ierr);
3093f26d0771SStefano Zampini     /* map from [0,ncl) to col */
3094f26d0771SStefano Zampini     for (i=0;i<ncl;i++) idxs[i] = cl[i];
3095f26d0771SStefano Zampini     for (i=ncl;i<ncg;i++) idxs[i] = -1;
3096f26d0771SStefano Zampini     ierr = ISRestoreIndices(col,&cl);CHKERRQ(ierr);
3097f26d0771SStefano Zampini     ierr = ISLocalToGlobalMappingRestoreIndices(A->cmap->mapping,&cg);CHKERRQ(ierr);
3098f26d0771SStefano Zampini     ierr = ISCreateGeneral(PetscObjectComm((PetscObject)A),ncg,idxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
3099f26d0771SStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr);
3100f26d0771SStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
3101f26d0771SStefano Zampini   } else {
3102f26d0771SStefano Zampini     ierr = PetscObjectReference((PetscObject)rl2g);CHKERRQ(ierr);
3103f26d0771SStefano Zampini     cl2g = rl2g;
3104f26d0771SStefano Zampini   }
3105f26d0771SStefano Zampini   /* create the MATIS submatrix */
3106f26d0771SStefano Zampini   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
3107f26d0771SStefano Zampini   ierr = MatCreate(PetscObjectComm((PetscObject)A),submat);CHKERRQ(ierr);
3108f26d0771SStefano Zampini   ierr = MatSetSizes(*submat,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr);
3109f26d0771SStefano Zampini   ierr = MatSetType(*submat,MATIS);CHKERRQ(ierr);
3110b0aa3428SStefano Zampini   matis = (Mat_IS*)((*submat)->data);
3111f26d0771SStefano Zampini   matis->islocalref = PETSC_TRUE;
3112f26d0771SStefano Zampini   ierr = MatSetLocalToGlobalMapping(*submat,rl2g,cl2g);CHKERRQ(ierr);
3113f26d0771SStefano Zampini   ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr);
3114f26d0771SStefano Zampini   ierr = MatISSetLocalMat(*submat,lA);CHKERRQ(ierr);
3115f26d0771SStefano Zampini   ierr = MatSetUp(*submat);CHKERRQ(ierr);
3116f26d0771SStefano Zampini   ierr = MatAssemblyBegin(*submat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3117f26d0771SStefano Zampini   ierr = MatAssemblyEnd(*submat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3118f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
3119f26d0771SStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
3120f26d0771SStefano Zampini   /* remove unsupported ops */
3121f26d0771SStefano Zampini   ierr = PetscMemzero((*submat)->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
3122f26d0771SStefano Zampini   (*submat)->ops->destroy               = MatDestroy_IS;
3123f26d0771SStefano Zampini   (*submat)->ops->setvalueslocal        = MatSetValuesLocal_SubMat_IS;
3124f26d0771SStefano Zampini   (*submat)->ops->setvaluesblockedlocal = MatSetValuesBlockedLocal_SubMat_IS;
3125f26d0771SStefano Zampini   (*submat)->ops->assemblybegin         = MatAssemblyBegin_IS;
3126f26d0771SStefano Zampini   (*submat)->ops->assemblyend           = MatAssemblyEnd_IS;
3127f26d0771SStefano Zampini   PetscFunctionReturn(0);
3128f26d0771SStefano Zampini }
3129f26d0771SStefano Zampini 
3130872cf891SStefano Zampini static PetscErrorCode MatSetFromOptions_IS(PetscOptionItems *PetscOptionsObject, Mat A)
3131872cf891SStefano Zampini {
3132872cf891SStefano Zampini   Mat_IS         *a = (Mat_IS*)A->data;
31338546b261SStefano Zampini   char           type[256];
31348546b261SStefano Zampini   PetscBool      flg;
3135872cf891SStefano Zampini   PetscErrorCode ierr;
3136872cf891SStefano Zampini 
3137872cf891SStefano Zampini   PetscFunctionBegin;
3138872cf891SStefano Zampini   ierr = PetscOptionsHead(PetscOptionsObject,"MATIS options");CHKERRQ(ierr);
3139f03112d0SStefano Zampini   ierr = PetscOptionsBool("-matis_fixempty","Fix local matrices in case of empty local rows/columns","MatISFixLocalEmpty",a->locempty,&a->locempty,NULL);CHKERRQ(ierr);
314075d48cdbSStefano Zampini   ierr = PetscOptionsBool("-matis_storel2l","Store local-to-local matrices generated from PtAP operations","MatISStoreL2L",a->storel2l,&a->storel2l,NULL);CHKERRQ(ierr);
31418546b261SStefano Zampini   ierr = PetscOptionsFList("-matis_localmat_type","Matrix type","MatISSetLocalMatType",MatList,a->lmattype,type,256,&flg);CHKERRQ(ierr);
31428546b261SStefano Zampini   if (flg) {
31438546b261SStefano Zampini     ierr = MatISSetLocalMatType(A,type);CHKERRQ(ierr);
31448546b261SStefano Zampini   }
31458546b261SStefano Zampini   if (a->A) {
31468546b261SStefano Zampini     ierr = MatSetFromOptions(a->A);CHKERRQ(ierr);
31478546b261SStefano Zampini   }
31480af67c1bSStefano Zampini   ierr = PetscOptionsTail();CHKERRQ(ierr);
3149872cf891SStefano Zampini   PetscFunctionReturn(0);
3150872cf891SStefano Zampini }
3151872cf891SStefano Zampini 
3152284134d9SBarry Smith /*@
31533c212e90SHong Zhang     MatCreateIS - Creates a "process" unassembled matrix, assembled on each
3154284134d9SBarry Smith        process but not across processes.
3155284134d9SBarry Smith 
3156284134d9SBarry Smith    Input Parameters:
3157284134d9SBarry Smith +     comm    - MPI communicator that will share the matrix
3158e176bc59SStefano Zampini .     bs      - block size of the matrix
3159df3898eeSBarry Smith .     m,n,M,N - local and/or global sizes of the left and right vector used in matrix vector products
3160e176bc59SStefano Zampini .     rmap    - local to global map for rows
3161e176bc59SStefano Zampini -     cmap    - local to global map for cols
3162284134d9SBarry Smith 
3163284134d9SBarry Smith    Output Parameter:
3164284134d9SBarry Smith .    A - the resulting matrix
3165284134d9SBarry Smith 
31668e6c10adSSatish Balay    Level: advanced
31678e6c10adSSatish Balay 
316895452b02SPatrick Sanan    Notes:
316995452b02SPatrick Sanan     See MATIS for more details.
31706fdf41d1SStefano Zampini           m and n are NOT related to the size of the map; they represent the size of the local parts of the vectors
31716fdf41d1SStefano Zampini           used in MatMult operations. The sizes of rmap and cmap define the size of the local matrices.
31723c212e90SHong Zhang           If either rmap or cmap are NULL, then the matrix is assumed to be square.
3173284134d9SBarry Smith 
3174284134d9SBarry Smith .seealso: MATIS, MatSetLocalToGlobalMapping()
3175284134d9SBarry Smith @*/
3176e176bc59SStefano Zampini PetscErrorCode  MatCreateIS(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt M,PetscInt N,ISLocalToGlobalMapping rmap,ISLocalToGlobalMapping cmap,Mat *A)
3177284134d9SBarry Smith {
3178284134d9SBarry Smith   PetscErrorCode ierr;
3179284134d9SBarry Smith 
3180284134d9SBarry Smith   PetscFunctionBegin;
31816fdf41d1SStefano Zampini   if (!rmap && !cmap) SETERRQ(comm,PETSC_ERR_USER,"You need to provide at least one of the mappings");
3182284134d9SBarry Smith   ierr = MatCreate(comm,A);CHKERRQ(ierr);
3183284134d9SBarry Smith   ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr);
31846fdf41d1SStefano Zampini   if (bs > 0) {
3185284134d9SBarry Smith     ierr = MatSetBlockSize(*A,bs);CHKERRQ(ierr);
31866fdf41d1SStefano Zampini   }
3187284134d9SBarry Smith   ierr = MatSetType(*A,MATIS);CHKERRQ(ierr);
3188e176bc59SStefano Zampini   if (rmap && cmap) {
3189e176bc59SStefano Zampini     ierr = MatSetLocalToGlobalMapping(*A,rmap,cmap);CHKERRQ(ierr);
3190e176bc59SStefano Zampini   } else if (!rmap) {
3191e176bc59SStefano Zampini     ierr = MatSetLocalToGlobalMapping(*A,cmap,cmap);CHKERRQ(ierr);
3192e176bc59SStefano Zampini   } else {
3193e176bc59SStefano Zampini     ierr = MatSetLocalToGlobalMapping(*A,rmap,rmap);CHKERRQ(ierr);
3194e176bc59SStefano Zampini   }
3195284134d9SBarry Smith   PetscFunctionReturn(0);
3196284134d9SBarry Smith }
3197284134d9SBarry Smith 
3198b4319ba4SBarry Smith /*MC
3199f26d0771SStefano Zampini    MATIS - MATIS = "is" - A matrix type to be used for using the non-overlapping domain decomposition methods (e.g. PCBDDC or KSPFETIDP).
3200b89f26deSStefano Zampini    This stores the matrices in globally unassembled form. Each processor assembles only its local Neumann problem and the parallel matrix vector
3201b4319ba4SBarry Smith    product is handled "implicitly".
3202b4319ba4SBarry Smith 
3203b4319ba4SBarry Smith    Options Database Keys:
320475d48cdbSStefano Zampini + -mat_type is - sets the matrix type to "is" during a call to MatSetFromOptions()
320575d48cdbSStefano Zampini . -matis_fixempty - Fixes local matrices in case of empty local rows/columns.
320675d48cdbSStefano Zampini - -matis_storel2l - stores the local-to-local operators generated by the Galerkin process of MatPtAP().
3207b4319ba4SBarry Smith 
320895452b02SPatrick Sanan    Notes:
320995452b02SPatrick Sanan     Options prefix for the inner matrix are given by -is_mat_xxx
3210b4319ba4SBarry Smith 
3211b4319ba4SBarry Smith           You must call MatSetLocalToGlobalMapping() before using this matrix type.
3212b4319ba4SBarry Smith 
3213b4319ba4SBarry Smith           You can do matrix preallocation on the local matrix after you obtain it with
3214eb82efa4SStefano Zampini           MatISGetLocalMat(); otherwise, you could use MatISSetPreallocation()
3215b4319ba4SBarry Smith 
3216b4319ba4SBarry Smith   Level: advanced
3217b4319ba4SBarry Smith 
3218f26d0771SStefano Zampini .seealso: Mat, MatISGetLocalMat(), MatSetLocalToGlobalMapping(), MatISSetPreallocation(), MatCreateIS(), PCBDDC, KSPFETIDP
3219b4319ba4SBarry Smith 
3220b4319ba4SBarry Smith M*/
3221b4319ba4SBarry Smith 
32228cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_IS(Mat A)
3223b4319ba4SBarry Smith {
3224dfbe8321SBarry Smith   PetscErrorCode ierr;
3225b4319ba4SBarry Smith   Mat_IS         *b;
3226b4319ba4SBarry Smith 
3227b4319ba4SBarry Smith   PetscFunctionBegin;
3228b00a9115SJed Brown   ierr    = PetscNewLog(A,&b);CHKERRQ(ierr);
32298546b261SStefano Zampini   ierr    = PetscStrallocpy(MATAIJ,&b->lmattype);CHKERRQ(ierr);
3230b4319ba4SBarry Smith   A->data = (void*)b;
3231b4319ba4SBarry Smith 
3232e176bc59SStefano Zampini   /* matrix ops */
3233e176bc59SStefano Zampini   ierr = PetscMemzero(A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
3234b4319ba4SBarry Smith   A->ops->mult                    = MatMult_IS;
32352e74eeadSLisandro Dalcin   A->ops->multadd                 = MatMultAdd_IS;
32362e74eeadSLisandro Dalcin   A->ops->multtranspose           = MatMultTranspose_IS;
32372e74eeadSLisandro Dalcin   A->ops->multtransposeadd        = MatMultTransposeAdd_IS;
3238b4319ba4SBarry Smith   A->ops->destroy                 = MatDestroy_IS;
3239b4319ba4SBarry Smith   A->ops->setlocaltoglobalmapping = MatSetLocalToGlobalMapping_IS;
32402e74eeadSLisandro Dalcin   A->ops->setvalues               = MatSetValues_IS;
324198921651SStefano Zampini   A->ops->setvaluesblocked        = MatSetValuesBlocked_IS;
3242b4319ba4SBarry Smith   A->ops->setvalueslocal          = MatSetValuesLocal_IS;
3243f0006bf2SLisandro Dalcin   A->ops->setvaluesblockedlocal   = MatSetValuesBlockedLocal_IS;
32442e74eeadSLisandro Dalcin   A->ops->zerorows                = MatZeroRows_IS;
3245f0ae7da4SStefano Zampini   A->ops->zerorowscolumns         = MatZeroRowsColumns_IS;
3246b4319ba4SBarry Smith   A->ops->assemblybegin           = MatAssemblyBegin_IS;
3247b4319ba4SBarry Smith   A->ops->assemblyend             = MatAssemblyEnd_IS;
3248b4319ba4SBarry Smith   A->ops->view                    = MatView_IS;
32496726f965SBarry Smith   A->ops->zeroentries             = MatZeroEntries_IS;
32502e74eeadSLisandro Dalcin   A->ops->scale                   = MatScale_IS;
32512e74eeadSLisandro Dalcin   A->ops->getdiagonal             = MatGetDiagonal_IS;
32526726f965SBarry Smith   A->ops->setoption               = MatSetOption_IS;
325369796d55SStefano Zampini   A->ops->ishermitian             = MatIsHermitian_IS;
325469796d55SStefano Zampini   A->ops->issymmetric             = MatIsSymmetric_IS;
325545471136SStefano Zampini   A->ops->isstructurallysymmetric = MatIsStructurallySymmetric_IS;
3256ad6194a2SStefano Zampini   A->ops->duplicate               = MatDuplicate_IS;
32576bd84002SStefano Zampini   A->ops->missingdiagonal         = MatMissingDiagonal_IS;
32582b404112SStefano Zampini   A->ops->copy                    = MatCopy_IS;
3259659959c5SStefano Zampini   A->ops->getlocalsubmatrix       = MatGetLocalSubMatrix_IS;
32607dae84e0SHong Zhang   A->ops->createsubmatrix         = MatCreateSubMatrix_IS;
3261f26d0771SStefano Zampini   A->ops->axpy                    = MatAXPY_IS;
32623fd1c9e7SStefano Zampini   A->ops->diagonalset             = MatDiagonalSet_IS;
32633fd1c9e7SStefano Zampini   A->ops->shift                   = MatShift_IS;
3264d7f69cd0SStefano Zampini   A->ops->transpose               = MatTranspose_IS;
32657fa8f2d3SStefano Zampini   A->ops->getinfo                 = MatGetInfo_IS;
3266ad219c80Sstefano_zampini   A->ops->diagonalscale           = MatDiagonalScale_IS;
3267872cf891SStefano Zampini   A->ops->setfromoptions          = MatSetFromOptions_IS;
3268b4319ba4SBarry Smith 
3269b7ce53b6SStefano Zampini   /* special MATIS functions */
32708546b261SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMatType_C",MatISSetLocalMatType_IS);CHKERRQ(ierr);
3271bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetLocalMat_C",MatISGetLocalMat_IS);CHKERRQ(ierr);
32723b3b1effSJed Brown   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISRestoreLocalMat_C",MatISRestoreLocalMat_IS);CHKERRQ(ierr);
3273bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMat_C",MatISSetLocalMat_IS);CHKERRQ(ierr);
3274487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetMPIXAIJ_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
32752e1947a5SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetPreallocation_C",MatISSetPreallocation_IS);CHKERRQ(ierr);
327675d48cdbSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISStoreL2L_C",MatISStoreL2L_IS);CHKERRQ(ierr);
3277f03112d0SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatISFixLocalEmpty_C",MatISFixLocalEmpty_IS);CHKERRQ(ierr);
3278487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpiaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
3279487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpibaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
3280487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpisbaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
3281487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
3282487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqbaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
3283487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqsbaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
3284487b449aSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_aij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr);
328517667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)A,MATIS);CHKERRQ(ierr);
3286b4319ba4SBarry Smith   PetscFunctionReturn(0);
3287b4319ba4SBarry Smith }
3288