PetscErrorCode DMCreateFieldDecomposition_Composite(DM dm, PetscInt *len,char ***namelist, IS **islist, DM** dmlist)
{
PetscInt nDM;
PetscInt i;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = DMCreateFieldIS_Composite(dm, len, namelist, islist);CHKERRQ(ierr);
if (dmlist) {
ierr = DMCompositeGetNumberDM(dm, &nDM);CHKERRQ(ierr);
ierr = PetscMalloc(nDM*sizeof(DM), dmlist);CHKERRQ(ierr);
ierr = DMCompositeGetEntriesArray(dm, *dmlist);CHKERRQ(ierr);
for (i=0; i<nDM; i++) {
ierr = PetscObjectReference((PetscObject)((*dmlist)[i]));CHKERRQ(ierr);
}
}
PetscFunctionReturn(0);
}
PetscErrorCode DMCreateFieldIS_Composite(DM dm, PetscInt *numFields,char ***fieldNames, IS **fields)
{
PetscInt nDM;
DM *dms;
PetscInt i;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = DMCompositeGetNumberDM(dm, &nDM);CHKERRQ(ierr);
if (numFields) {*numFields = nDM;}
ierr = DMCompositeGetGlobalISs(dm, fields);CHKERRQ(ierr);
if (fieldNames) {
ierr = PetscMalloc(nDM*sizeof(DM), &dms);CHKERRQ(ierr);
ierr = PetscMalloc(nDM*sizeof(const char *), fieldNames);CHKERRQ(ierr);
ierr = DMCompositeGetEntriesArray(dm, dms);CHKERRQ(ierr);
for (i=0; i<nDM; i++) {
char buf[256];
const char *splitname;
/* Split naming precedence: object name, prefix, number */
splitname = ((PetscObject) dm)->name;
if (!splitname) {
ierr = PetscObjectGetOptionsPrefix((PetscObject)dms[i],&splitname);CHKERRQ(ierr);
if (splitname) {
size_t len;
ierr = PetscStrncpy(buf,splitname,sizeof(buf));CHKERRQ(ierr);
buf[sizeof(buf) - 1] = 0;
ierr = PetscStrlen(buf,&len);CHKERRQ(ierr);
if (buf[len-1] == '_') buf[len-1] = 0; /* Remove trailing underscore if it was used */
splitname = buf;
}
}
if (!splitname) {
ierr = PetscSNPrintf(buf,sizeof(buf),"%D",i);CHKERRQ(ierr);
splitname = buf;
}
ierr = PetscStrallocpy(splitname,&(*fieldNames)[i]);CHKERRQ(ierr);
}
ierr = PetscFree(dms);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
PETSC_EXTERN void PETSC_STDCALL dmcompositegetnumberdm_(DM dm,PetscInt *nDM, int *__ierr ){
*__ierr = DMCompositeGetNumberDM(
(DM)PetscToPointer((dm) ),nDM);
}
static PetscErrorCode SNESMultiblockSetDefaults(SNES snes)
{
SNES_Multiblock *mb = (SNES_Multiblock *) snes->data;
BlockDesc blocks = mb->blocks;
PetscInt i;
PetscErrorCode ierr;
PetscFunctionBegin;
if (!blocks) {
if (snes->dm) {
PetscBool dmcomposite;
ierr = PetscObjectTypeCompare((PetscObject) snes->dm, DMCOMPOSITE, &dmcomposite);CHKERRQ(ierr);
if (dmcomposite) {
PetscInt nDM;
IS *fields;
ierr = PetscInfo(snes,"Setting up physics based multiblock solver using the embedded DM\n");CHKERRQ(ierr);
ierr = DMCompositeGetNumberDM(snes->dm, &nDM);CHKERRQ(ierr);
ierr = DMCompositeGetGlobalISs(snes->dm, &fields);CHKERRQ(ierr);
for (i = 0; i < nDM; ++i) {
char name[8];
ierr = PetscSNPrintf(name, sizeof(name), "%D", i);CHKERRQ(ierr);
ierr = SNESMultiblockSetIS(snes, name, fields[i]);CHKERRQ(ierr);
ierr = ISDestroy(&fields[i]);CHKERRQ(ierr);
}
ierr = PetscFree(fields);CHKERRQ(ierr);
}
} else {
PetscBool flg = PETSC_FALSE;
PetscBool stokes = PETSC_FALSE;
if (mb->bs <= 0) {
if (snes->jacobian_pre) {
ierr = MatGetBlockSize(snes->jacobian_pre, &mb->bs);CHKERRQ(ierr);
} else {
mb->bs = 1;
}
}
ierr = PetscOptionsGetBool(((PetscObject) snes)->prefix, "-snes_multiblock_default", &flg, PETSC_NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetBool(((PetscObject) snes)->prefix, "-snes_multiblock_detect_saddle_point", &stokes, PETSC_NULL);CHKERRQ(ierr);
if (stokes) {
IS zerodiags, rest;
PetscInt nmin, nmax;
ierr = MatGetOwnershipRange(snes->jacobian_pre, &nmin, &nmax);CHKERRQ(ierr);
ierr = MatFindZeroDiagonals(snes->jacobian_pre, &zerodiags);CHKERRQ(ierr);
ierr = ISComplement(zerodiags, nmin, nmax, &rest);CHKERRQ(ierr);
ierr = SNESMultiblockSetIS(snes, "0", rest);CHKERRQ(ierr);
ierr = SNESMultiblockSetIS(snes, "1", zerodiags);CHKERRQ(ierr);
ierr = ISDestroy(&zerodiags);CHKERRQ(ierr);
ierr = ISDestroy(&rest);CHKERRQ(ierr);
} else {
if (!flg) {
/* Allow user to set fields from command line, if bs was known at the time of SNESSetFromOptions_Multiblock()
then it is set there. This is not ideal because we should only have options set in XXSetFromOptions(). */
ierr = SNESMultiblockSetFieldsRuntime_Private(snes);CHKERRQ(ierr);
if (mb->defined) {ierr = PetscInfo(snes, "Blocks defined using the options database\n");CHKERRQ(ierr);}
}
if (flg || !mb->defined) {
ierr = PetscInfo(snes, "Using default splitting of fields\n");CHKERRQ(ierr);
for (i = 0; i < mb->bs; ++i) {
char name[8];
ierr = PetscSNPrintf(name, sizeof(name), "%D", i);CHKERRQ(ierr);
ierr = SNESMultiblockSetFields(snes, name, 1, &i);CHKERRQ(ierr);
}
mb->defaultblocks = PETSC_TRUE;
}
}
}
} else if (mb->numBlocks == 1) {
if (blocks->is) {
IS is2;
PetscInt nmin, nmax;
ierr = MatGetOwnershipRange(snes->jacobian_pre, &nmin, &nmax);CHKERRQ(ierr);
ierr = ISComplement(blocks->is, nmin, nmax, &is2);CHKERRQ(ierr);
ierr = SNESMultiblockSetIS(snes, "1", is2);CHKERRQ(ierr);
ierr = ISDestroy(&is2);CHKERRQ(ierr);
} else {
SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Must provide at least two sets of fields to SNES multiblock");
}
}
if (mb->numBlocks < 2) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unhandled case, must have at least two blocks");
PetscFunctionReturn(0);
}
