PETSC_EXTERN void PETSC_STDCALL dmdagetnonoverlappingregion_(DM da,PetscInt *xs,PetscInt *ys,PetscInt *zs,PetscInt *xm,PetscInt *ym,PetscInt *zm, int *__ierr ){
*__ierr = DMDAGetNonOverlappingRegion(
(DM)PetscToPointer((da) ),xs,ys,zs,xm,ym,zm);
}
/*
Fills the local vector problem on the subdomain from the global problem.
Right now this assumes one subdomain per processor.
*/
PetscErrorCode DMCreateDomainDecompositionScatters_DA(DM dm,PetscInt nsubdms,DM *subdms,VecScatter **iscat,VecScatter **oscat, VecScatter **lscat)
{
PetscErrorCode ierr;
DMDALocalInfo info,subinfo;
DM subdm;
MatStencil upper,lower;
IS idis,isis,odis,osis,gdis;
Vec svec,dvec,slvec;
PetscInt xm,ym,zm,xs,ys,zs;
PetscInt i;
PetscFunctionBegin;
/* allocate the arrays of scatters */
if (iscat) {ierr = PetscMalloc1(nsubdms,iscat);CHKERRQ(ierr);}
if (oscat) {ierr = PetscMalloc1(nsubdms,oscat);CHKERRQ(ierr);}
if (lscat) {ierr = PetscMalloc1(nsubdms,lscat);CHKERRQ(ierr);}
ierr = DMDAGetLocalInfo(dm,&info);CHKERRQ(ierr);
for (i = 0; i < nsubdms; i++) {
subdm = subdms[i];
ierr = DMDAGetLocalInfo(subdm,&subinfo);CHKERRQ(ierr);
ierr = DMDAGetNonOverlappingRegion(subdm,&xs,&ys,&zs,&xm,&ym,&zm);CHKERRQ(ierr);
/* create the global and subdomain index sets for the inner domain */
lower.i = xs;
lower.j = ys;
lower.k = zs;
upper.i = xs+xm;
upper.j = ys+ym;
upper.k = zs+zm;
ierr = DMDACreatePatchIS(dm,&lower,&upper,&idis);CHKERRQ(ierr);
ierr = DMDACreatePatchIS(subdm,&lower,&upper,&isis);CHKERRQ(ierr);
/* create the global and subdomain index sets for the outer subdomain */
lower.i = subinfo.xs;
lower.j = subinfo.ys;
lower.k = subinfo.zs;
upper.i = subinfo.xs+subinfo.xm;
upper.j = subinfo.ys+subinfo.ym;
upper.k = subinfo.zs+subinfo.zm;
ierr = DMDACreatePatchIS(dm,&lower,&upper,&odis);CHKERRQ(ierr);
ierr = DMDACreatePatchIS(subdm,&lower,&upper,&osis);CHKERRQ(ierr);
/* global and subdomain ISes for the local indices of the subdomain */
/* todo - make this not loop over at nonperiodic boundaries, which will be more involved */
lower.i = subinfo.gxs;
lower.j = subinfo.gys;
lower.k = subinfo.gzs;
upper.i = subinfo.gxs+subinfo.gxm;
upper.j = subinfo.gys+subinfo.gym;
upper.k = subinfo.gzs+subinfo.gzm;
ierr = DMDACreatePatchIS(dm,&lower,&upper,&gdis);CHKERRQ(ierr);
/* form the scatter */
ierr = DMGetGlobalVector(dm,&dvec);CHKERRQ(ierr);
ierr = DMGetGlobalVector(subdm,&svec);CHKERRQ(ierr);
ierr = DMGetLocalVector(subdm,&slvec);CHKERRQ(ierr);
if (iscat) {ierr = VecScatterCreate(dvec,idis,svec,isis,&(*iscat)[i]);CHKERRQ(ierr);}
if (oscat) {ierr = VecScatterCreate(dvec,odis,svec,osis,&(*oscat)[i]);CHKERRQ(ierr);}
if (lscat) {ierr = VecScatterCreate(dvec,gdis,slvec,NULL,&(*lscat)[i]);CHKERRQ(ierr);}
ierr = DMRestoreGlobalVector(dm,&dvec);CHKERRQ(ierr);
ierr = DMRestoreGlobalVector(subdm,&svec);CHKERRQ(ierr);
ierr = DMRestoreLocalVector(subdm,&slvec);CHKERRQ(ierr);
ierr = ISDestroy(&idis);CHKERRQ(ierr);
ierr = ISDestroy(&isis);CHKERRQ(ierr);
ierr = ISDestroy(&odis);CHKERRQ(ierr);
ierr = ISDestroy(&osis);CHKERRQ(ierr);
ierr = ISDestroy(&gdis);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
