Files
openfoam/src/parallel/decompose/decompositionMethods/decompositionMethod/decompositionMethod.C
Mark Olesen 86f627b9e6 ENH: consolidate decomposition model, constructors for decomposition methods
- make regionName an optional constructor parameter, which eliminates
  a separate set of constructors and construction tables. Adjust
  internals to treat a missing/empty regionName as a no-op.

- pass in fallback dictionary content via new IOdictionary constructor
  with a pointer

ENH: further relax check for matching number of processor dirs

- if the "numberOfSubdomains" entry is missing (or even zero)
  ignore checks of processor dirs as meaningless.
2021-04-27 09:14:48 +02:00

1346 lines
34 KiB
C

/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration |
\\ / A nd | www.openfoam.com
\\/ M anipulation |
-------------------------------------------------------------------------------
Copyright (C) 2011-2016 OpenFOAM Foundation
Copyright (C) 2015-2021 OpenCFD Ltd.
-------------------------------------------------------------------------------
License
This file is part of OpenFOAM.
OpenFOAM is free software: you can redistribute it and/or modify it
under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License
along with OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
\*---------------------------------------------------------------------------*/
#include "decompositionMethod.H"
#include "globalIndex.H"
#include "syncTools.H"
#include "Tuple2.H"
#include "faceSet.H"
#include "regionSplit.H"
#include "localPointRegion.H"
#include "minData.H"
#include "BitOps.H"
#include "FaceCellWave.H"
// Compatibility (MAY-2014)
#include "preserveBafflesConstraint.H"
#include "preservePatchesConstraint.H"
#include "preserveFaceZonesConstraint.H"
#include "singleProcessorFaceSetsConstraint.H"
// * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
namespace Foam
{
defineTypeNameAndDebug(decompositionMethod, 0);
defineRunTimeSelectionTable(decompositionMethod, dictionary);
} // End namespace Foam
// * * * * * * * * * * * * * * * Local Functions * * * * * * * * * * * * * * //
namespace Foam
{
// Find named coefficents dictionary, or use default "coeffs"
static inline const dictionary* cfindCoeffsDict
(
const dictionary& dict,
const word& coeffsName,
const bool allowDefault
)
{
const dictionary* dictptr = dict.findDict(coeffsName);
if (!dictptr && allowDefault)
{
dictptr = dict.findDict("coeffs");
}
return dictptr;
}
} // End namespace Foam
// * * * * * * * * * * * * * Static Member Functions * * * * * * * * * * * * //
Foam::label Foam::decompositionMethod::nDomains
(
const dictionary& decompDict,
const word& regionName
)
{
const label nDomainsGlobal = decompDict.get<label>("numberOfSubdomains");
if (!regionName.empty())
{
const dictionary& regionDict =
optionalRegionDict(decompDict, regionName);
label nDomainsRegion;
if (regionDict.readIfPresent("numberOfSubdomains", nDomainsRegion))
{
if (nDomainsRegion >= 1 && nDomainsRegion <= nDomainsGlobal)
{
return nDomainsRegion;
}
WarningInFunction
<< "ignoring out of range numberOfSubdomains "
<< nDomainsRegion << " for region " << regionName
<< nl << nl
<< endl;
}
}
return nDomainsGlobal;
}
const Foam::dictionary& Foam::decompositionMethod::optionalRegionDict
(
const dictionary& decompDict,
const word& regionName
)
{
const dictionary* dictptr = nullptr;
if
(
!regionName.empty()
&& (dictptr = decompDict.findDict("regions")) != nullptr
)
{
dictptr = dictptr->findDict(regionName);
}
return (dictptr ? *dictptr : dictionary::null);
}
// * * * * * * * * * * * * * Private Member Functions * * * * * * * * * * * //
bool Foam::decompositionMethod::constraintCompat(const word& modelType) const
{
bool usable = decompDict_.found(modelType);
if (!usable)
{
return false;
}
for (const auto& item : constraints_)
{
if (modelType == item.type())
{
usable = false;
break;
}
}
if (usable)
{
Warning
<< nl << " Using '" << modelType
<< "' constraint specification." << nl;
}
else
{
Warning
<< nl << " Ignoring '" << modelType
<< "' constraint specification - was already specified." << nl;
}
// The syntax changed MAY-2014
error::warnAboutAge("constraint keyword", 1406);
return usable;
}
void Foam::decompositionMethod::readConstraints()
{
constraints_.clear();
const dictionary* dictptr = decompDict_.findDict("constraints");
if (dictptr)
{
for (const entry& dEntry : *dictptr)
{
if (!dEntry.isDict()) // safety
{
// Ignore or warn
continue;
}
const dictionary& dict = dEntry.dict();
if (dict.getOrDefault("enabled", true))
{
constraints_.append(decompositionConstraint::New(dict));
}
}
}
// Backwards compatibility (MAY-2014)
if (constraintCompat("preserveBaffles"))
{
constraints_.append
(
new decompositionConstraints::preserveBaffles()
);
}
if (constraintCompat("preservePatches"))
{
constraints_.append
(
new decompositionConstraints::preservePatches
(
decompDict_.get<wordRes>("preservePatches")
)
);
}
if (constraintCompat("preserveFaceZones"))
{
constraints_.append
(
new decompositionConstraints::preserveFaceZones
(
decompDict_.get<wordRes>("preserveFaceZones")
)
);
}
if (constraintCompat("singleProcessorFaceSets"))
{
constraints_.append
(
new decompositionConstraints::singleProcessorFaceSets
(
decompDict_.lookup("singleProcessorFaceSets")
)
);
}
}
// * * * * * * * * * * * * Protected Member Functions * * * * * * * * * * * //
const Foam::dictionary& Foam::decompositionMethod::findCoeffsDict
(
const dictionary& dict,
const word& coeffsName,
int select
)
{
const bool allowDefault = !(select & selectionType::EXACT);
const dictionary* dictptr =
cfindCoeffsDict(dict, coeffsName, allowDefault);
if (dictptr)
{
return *dictptr;
}
// Not found
if (select & selectionType::MANDATORY)
{
FatalIOError
<< "'" << coeffsName << "' dictionary not found in dictionary "
<< dict.name() << endl
<< abort(FatalIOError);
}
if (select & selectionType::NULL_DICT)
{
return dictionary::null;
}
return dict; // Return original dictionary
}
const Foam::dictionary& Foam::decompositionMethod::findCoeffsDict
(
const word& coeffsName,
int select
) const
{
const bool allowDefault = !(select & selectionType::EXACT);
const dictionary* dictptr = nullptr;
if (!decompRegionDict_.empty())
{
// Region-specific dictionary
dictptr = cfindCoeffsDict(decompRegionDict_, coeffsName, allowDefault);
}
if (!dictptr)
{
// General
dictptr = cfindCoeffsDict(decompDict_, coeffsName, allowDefault);
}
if (dictptr)
{
return *dictptr;
}
// Not found
if (select & selectionType::MANDATORY)
{
FatalIOError
<< "'" << coeffsName << "' dictionary not found in dictionary "
<< decompDict_.name() << endl
<< abort(FatalIOError);
}
if (select & selectionType::NULL_DICT)
{
return dictionary::null;
}
return decompDict_; // Return general dictionary
}
// * * * * * * * * * * * * * * * * Constructors * * * * * * * * * * * * * * //
Foam::decompositionMethod::decompositionMethod
(
const dictionary& decompDict,
const word& regionName
)
:
decompDict_(decompDict),
decompRegionDict_
(
optionalRegionDict(decompDict_, regionName)
),
nDomains_(nDomains(decompDict, regionName))
{
readConstraints();
}
// * * * * * * * * * * * * * * * Member Functions * * * * * * * * * * * * * //
Foam::autoPtr<Foam::decompositionMethod> Foam::decompositionMethod::New
(
const dictionary& decompDict,
const word& regionName
)
{
word methodType(decompDict.get<word>("method"));
const dictionary& regionDict = optionalRegionDict(decompDict, regionName);
regionDict.readIfPresent("method", methodType);
auto cstrIter = dictionaryConstructorTablePtr_->cfind(methodType);
if (!cstrIter.found())
{
FatalIOErrorInLookup
(
decompDict,
"decompositionMethod",
methodType,
*dictionaryConstructorTablePtr_
) << exit(FatalIOError);
}
// verbose
{
Info<< "Decomposition method " << methodType
<< " [" << (nDomains(decompDict, regionName)) << ']';
if (!regionName.empty())
{
Info<< " (region " << regionName << ')';
}
Info<< endl;
}
return autoPtr<decompositionMethod>(cstrIter()(decompDict, regionName));
}
Foam::labelList Foam::decompositionMethod::decompose
(
const polyMesh& mesh,
const pointField& points
) const
{
scalarField weights(points.size(), scalar(1));
return decompose(mesh, points, weights);
}
Foam::labelList Foam::decompositionMethod::decompose
(
const polyMesh& mesh,
const labelList& fineToCoarse,
const pointField& coarsePoints,
const scalarField& coarseWeights
) const
{
CompactListList<label> coarseCellCells;
calcCellCells
(
mesh,
fineToCoarse,
coarsePoints.size(),
true, // use global cell labels
coarseCellCells
);
// Decompose based on agglomerated points
labelList coarseDistribution
(
decompose
(
coarseCellCells(),
coarsePoints,
coarseWeights
)
);
// Rework back into decomposition for original mesh_
labelList fineDistribution(fineToCoarse.size());
forAll(fineDistribution, i)
{
fineDistribution[i] = coarseDistribution[fineToCoarse[i]];
}
return fineDistribution;
}
Foam::labelList Foam::decompositionMethod::decompose
(
const polyMesh& mesh,
const labelList& fineToCoarse,
const pointField& coarsePoints
) const
{
scalarField weights(coarsePoints.size(), scalar(1));
return decompose
(
mesh,
fineToCoarse,
coarsePoints,
weights
);
}
Foam::labelList Foam::decompositionMethod::decompose
(
const labelListList& globalCellCells,
const pointField& cc
) const
{
scalarField weights(cc.size(), scalar(1));
return decompose(globalCellCells, cc, weights);
}
void Foam::decompositionMethod::calcCellCells
(
const polyMesh& mesh,
const labelList& agglom,
const label nLocalCoarse,
const bool parallel,
CompactListList<label>& cellCells
)
{
const labelList& faceOwner = mesh.faceOwner();
const labelList& faceNeighbour = mesh.faceNeighbour();
const polyBoundaryMesh& patches = mesh.boundaryMesh();
// Create global cell numbers
// ~~~~~~~~~~~~~~~~~~~~~~~~~~
globalIndex globalAgglom
(
nLocalCoarse,
Pstream::msgType(),
Pstream::worldComm,
parallel
);
// Get agglomerate owner on other side of coupled faces
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
labelList globalNeighbour(mesh.nBoundaryFaces());
for (const polyPatch& pp : patches)
{
if (pp.coupled() && (parallel || !isA<processorPolyPatch>(pp)))
{
label facei = pp.start();
label bFacei = pp.start() - mesh.nInternalFaces();
forAll(pp, i)
{
globalNeighbour[bFacei] = globalAgglom.toGlobal
(
agglom[faceOwner[facei]]
);
++facei;
++bFacei;
}
}
}
// Get the cell on the other side of coupled patches
syncTools::swapBoundaryFaceList(mesh, globalNeighbour);
// Count number of faces (internal + coupled)
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
// Number of faces per coarse cell
labelList nFacesPerCell(nLocalCoarse, Zero);
for (label facei = 0; facei < mesh.nInternalFaces(); ++facei)
{
const label own = agglom[faceOwner[facei]];
const label nei = agglom[faceNeighbour[facei]];
nFacesPerCell[own]++;
nFacesPerCell[nei]++;
}
for (const polyPatch& pp : patches)
{
if (pp.coupled() && (parallel || !isA<processorPolyPatch>(pp)))
{
label facei = pp.start();
label bFacei = pp.start()-mesh.nInternalFaces();
forAll(pp, i)
{
const label own = agglom[faceOwner[facei]];
const label globalNei = globalNeighbour[bFacei];
if
(
!globalAgglom.isLocal(globalNei)
|| globalAgglom.toLocal(globalNei) != own
)
{
nFacesPerCell[own]++;
}
++facei;
++bFacei;
}
}
}
// Fill in offset and data
// ~~~~~~~~~~~~~~~~~~~~~~~
cellCells.setSize(nFacesPerCell);
nFacesPerCell = 0;
labelList& m = cellCells.m();
const labelList& offsets = cellCells.offsets();
// For internal faces is just offsetted owner and neighbour
for (label facei = 0; facei < mesh.nInternalFaces(); ++facei)
{
const label own = agglom[faceOwner[facei]];
const label nei = agglom[faceNeighbour[facei]];
m[offsets[own] + nFacesPerCell[own]++] = globalAgglom.toGlobal(nei);
m[offsets[nei] + nFacesPerCell[nei]++] = globalAgglom.toGlobal(own);
}
// For boundary faces is offsetted coupled neighbour
for (const polyPatch& pp : patches)
{
if (pp.coupled() && (parallel || !isA<processorPolyPatch>(pp)))
{
label facei = pp.start();
label bFacei = pp.start()-mesh.nInternalFaces();
forAll(pp, i)
{
const label own = agglom[faceOwner[facei]];
const label globalNei = globalNeighbour[bFacei];
if
(
!globalAgglom.isLocal(globalNei)
|| globalAgglom.toLocal(globalNei) != own
)
{
m[offsets[own] + nFacesPerCell[own]++] = globalNei;
}
++facei;
++bFacei;
}
}
}
// Check for duplicates connections between cells
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
// Done as postprocessing step since we now have cellCells.
if (cellCells.size() == 0)
{
return;
}
label newIndex = 0;
labelHashSet nbrCells;
label startIndex = cellCells.offsets()[0];
forAll(cellCells, celli)
{
nbrCells.clear();
nbrCells.insert(globalAgglom.toGlobal(celli));
const label endIndex = cellCells.offsets()[celli+1];
for (label i = startIndex; i < endIndex; ++i)
{
if (nbrCells.insert(cellCells.m()[i]))
{
cellCells.m()[newIndex++] = cellCells.m()[i];
}
}
startIndex = endIndex;
cellCells.offsets()[celli+1] = newIndex;
}
cellCells.m().setSize(newIndex);
//forAll(cellCells, celli)
//{
// Pout<< "Original: Coarse cell " << celli << endl;
// forAll(mesh.cellCells()[celli], i)
// {
// Pout<< " nbr:" << mesh.cellCells()[celli][i] << endl;
// }
// Pout<< "Compacted: Coarse cell " << celli << endl;
// const labelUList cCells = cellCells[celli];
// forAll(cCells, i)
// {
// Pout<< " nbr:" << cCells[i] << endl;
// }
//}
}
void Foam::decompositionMethod::calcCellCells
(
const polyMesh& mesh,
const labelList& agglom,
const label nLocalCoarse,
const bool parallel,
CompactListList<label>& cellCells,
CompactListList<scalar>& cellCellWeights
)
{
const labelList& faceOwner = mesh.faceOwner();
const labelList& faceNeighbour = mesh.faceNeighbour();
const polyBoundaryMesh& patches = mesh.boundaryMesh();
// Create global cell numbers
// ~~~~~~~~~~~~~~~~~~~~~~~~~~
globalIndex globalAgglom
(
nLocalCoarse,
Pstream::msgType(),
Pstream::worldComm,
parallel
);
// Get agglomerate owner on other side of coupled faces
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
labelList globalNeighbour(mesh.nBoundaryFaces());
for (const polyPatch& pp : patches)
{
if (pp.coupled() && (parallel || !isA<processorPolyPatch>(pp)))
{
label facei = pp.start();
label bFacei = pp.start() - mesh.nInternalFaces();
forAll(pp, i)
{
globalNeighbour[bFacei] = globalAgglom.toGlobal
(
agglom[faceOwner[facei]]
);
++facei;
++bFacei;
}
}
}
// Get the cell on the other side of coupled patches
syncTools::swapBoundaryFaceList(mesh, globalNeighbour);
// Count number of faces (internal + coupled)
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
// Number of faces per coarse cell
labelList nFacesPerCell(nLocalCoarse, Zero);
for (label facei = 0; facei < mesh.nInternalFaces(); ++facei)
{
const label own = agglom[faceOwner[facei]];
const label nei = agglom[faceNeighbour[facei]];
nFacesPerCell[own]++;
nFacesPerCell[nei]++;
}
for (const polyPatch& pp : patches)
{
if (pp.coupled() && (parallel || !isA<processorPolyPatch>(pp)))
{
label facei = pp.start();
label bFacei = pp.start() - mesh.nInternalFaces();
forAll(pp, i)
{
const label own = agglom[faceOwner[facei]];
const label globalNei = globalNeighbour[bFacei];
if
(
!globalAgglom.isLocal(globalNei)
|| globalAgglom.toLocal(globalNei) != own
)
{
nFacesPerCell[own]++;
}
++facei;
++bFacei;
}
}
}
// Fill in offset and data
// ~~~~~~~~~~~~~~~~~~~~~~~
cellCells.setSize(nFacesPerCell);
cellCellWeights.setSize(nFacesPerCell);
nFacesPerCell = 0;
labelList& m = cellCells.m();
scalarList& w = cellCellWeights.m();
const labelList& offsets = cellCells.offsets();
// For internal faces is just offsetted owner and neighbour
for (label facei = 0; facei < mesh.nInternalFaces(); ++facei)
{
const label own = agglom[faceOwner[facei]];
const label nei = agglom[faceNeighbour[facei]];
const label ownIndex = offsets[own] + nFacesPerCell[own]++;
const label neiIndex = offsets[nei] + nFacesPerCell[nei]++;
m[ownIndex] = globalAgglom.toGlobal(nei);
w[ownIndex] = mag(mesh.faceAreas()[facei]);
m[neiIndex] = globalAgglom.toGlobal(own);
w[ownIndex] = mag(mesh.faceAreas()[facei]);
}
// For boundary faces is offsetted coupled neighbour
for (const polyPatch& pp : patches)
{
if (pp.coupled() && (parallel || !isA<processorPolyPatch>(pp)))
{
label facei = pp.start();
label bFacei = pp.start()-mesh.nInternalFaces();
forAll(pp, i)
{
const label own = agglom[faceOwner[facei]];
const label globalNei = globalNeighbour[bFacei];
if
(
!globalAgglom.isLocal(globalNei)
|| globalAgglom.toLocal(globalNei) != own
)
{
const label ownIndex = offsets[own] + nFacesPerCell[own]++;
m[ownIndex] = globalNei;
w[ownIndex] = mag(mesh.faceAreas()[facei]);
}
++facei;
++bFacei;
}
}
}
// Check for duplicates connections between cells
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
// Done as postprocessing step since we now have cellCells.
if (cellCells.size() == 0)
{
return;
}
label newIndex = 0;
labelHashSet nbrCells;
label startIndex = cellCells.offsets()[0];
forAll(cellCells, celli)
{
nbrCells.clear();
nbrCells.insert(globalAgglom.toGlobal(celli));
const label endIndex = cellCells.offsets()[celli+1];
for (label i = startIndex; i < endIndex; ++i)
{
if (nbrCells.insert(cellCells.m()[i]))
{
cellCells.m()[newIndex] = cellCells.m()[i];
cellCellWeights.m()[newIndex] = cellCellWeights.m()[i];
newIndex++;
}
}
startIndex = endIndex;
cellCells.offsets()[celli+1] = newIndex;
cellCellWeights.offsets()[celli+1] = newIndex;
}
cellCells.m().setSize(newIndex);
cellCellWeights.m().setSize(newIndex);
}
// NOTE:
// - alternative calcCellCells that handled explicitConnections was
// deactivated (2014 or earlier) and finally removed APR-2018.
Foam::labelList Foam::decompositionMethod::decompose
(
const polyMesh& mesh,
const scalarField& cellWeights,
//- Whether owner and neighbour should be on same processor
// (takes priority over explicitConnections)
const boolList& blockedFace,
//- Whether whole sets of faces (and point neighbours) need to be kept
// on single processor
const PtrList<labelList>& specifiedProcessorFaces,
const labelList& specifiedProcessor,
//- Additional connections between boundary faces
const List<labelPair>& explicitConnections
) const
{
// Any weights specified?
const bool hasWeights = returnReduce(!cellWeights.empty(), orOp<bool>());
if (hasWeights && cellWeights.size() != mesh.nCells())
{
FatalErrorInFunction
<< "Number of weights " << cellWeights.size()
<< " differs from number of cells " << mesh.nCells()
<< exit(FatalError);
}
// Any faces not blocked?
const bool hasUnblocked =
returnReduce
(
(!blockedFace.empty() && !BitOps::all(blockedFace)),
orOp<bool>()
);
// Any non-mesh connections?
const label nConnections = returnReduce
(
explicitConnections.size(),
sumOp<label>()
);
// Any processor sets?
label nProcSets = 0;
for (const labelList& procset : specifiedProcessorFaces)
{
nProcSets += procset.size();
}
reduce(nProcSets, sumOp<label>());
// Either do decomposition on cell centres or on agglomeration
if (!hasUnblocked && !nConnections && !nProcSets)
{
// No constraints, possibly weights
return
(
hasWeights
? decompose(mesh, mesh.cellCentres(), cellWeights)
: decompose(mesh, mesh.cellCentres())
);
}
// The harder work.
// When we have processor sets, connections, or blocked faces.
// Determine local regions, separated by blockedFaces
regionSplit localRegion(mesh, blockedFace, explicitConnections, false);
if (debug)
{
// Only need to count unblocked faces for debugging
const label nUnblocked =
(
hasUnblocked
? returnReduce
(
label(BitOps::count(blockedFace, false)),
sumOp<label>()
)
: 0
);
Info<< "Constrained decomposition:" << nl
<< " faces with same owner and neighbour processor : "
<< nUnblocked << nl
<< " baffle faces with same owner processor : "
<< nConnections << nl
<< " faces all on same processor : "
<< nProcSets << nl
<< " split into " << localRegion.nLocalRegions()
<< " regions."
<< endl;
}
// Gather region weights and determine region cell centres
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
// For the region centre, just take the first cell in the region.
// If we average the region centre instead, cyclics could cause
// the average domain centre to be outside of domain.
scalarField regionWeights(localRegion.nLocalRegions(), Zero);
pointField regionCentres(localRegion.nLocalRegions(), point::max);
if (hasWeights)
{
forAll(localRegion, celli)
{
const label regioni = localRegion[celli];
regionWeights[regioni] += cellWeights[celli];
if (regionCentres[regioni] == point::max)
{
regionCentres[regioni] = mesh.cellCentres()[celli];
}
}
}
else
{
forAll(localRegion, celli)
{
const label regioni = localRegion[celli];
regionWeights[regioni] += 1.0;
if (regionCentres[regioni] == point::max)
{
regionCentres[regioni] = mesh.cellCentres()[celli];
}
}
}
// Do decomposition on agglomeration
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
labelList finalDecomp =
decompose
(
mesh,
localRegion,
regionCentres,
regionWeights
);
// Apply explicitConnections since decompose did not know about them
for (const labelPair& baffle : explicitConnections)
{
const label f0 = baffle.first();
const label f1 = baffle.second();
if (!blockedFace[f0] && !blockedFace[f1])
{
// Note: what if internal faces and owner and neighbour on
// different processor?
// So for now just push owner side proc
const label proci = finalDecomp[mesh.faceOwner()[f0]];
finalDecomp[mesh.faceOwner()[f1]] = proci;
if (mesh.isInternalFace(f1))
{
finalDecomp[mesh.faceNeighbour()[f1]] = proci;
}
}
else if (blockedFace[f0] != blockedFace[f1])
{
FatalErrorInFunction
<< "On explicit connection between faces " << f0
<< " and " << f1
<< " the two blockedFace status are not equal : "
<< blockedFace[f0] << " and " << blockedFace[f1]
<< exit(FatalError);
}
}
// blockedFaces corresponding to processor faces need to be handled
// separately since not handled by local regionSplit. We need to
// walk now across coupled faces and make sure to move a whole
// global region across
// This additionally consolidates/compacts the regions numbers globally,
// since that was skipped in the previous regionSplit.
if (Pstream::parRun())
{
// Re-do regionSplit
// Field on cells and faces.
List<minData> cellData(mesh.nCells());
List<minData> faceData(mesh.nFaces());
// Take over blockedFaces by seeding a negative number
// (so is always less than the decomposition)
label nUnblocked = 0;
forAll(blockedFace, facei)
{
if (blockedFace[facei])
{
faceData[facei] = minData(-123);
}
else
{
++nUnblocked;
}
}
// Seed unblocked faces with destination processor
labelList seedFaces(nUnblocked);
List<minData> seedData(nUnblocked);
nUnblocked = 0;
forAll(blockedFace, facei)
{
if (!blockedFace[facei])
{
const label own = mesh.faceOwner()[facei];
seedFaces[nUnblocked] = facei;
seedData[nUnblocked] = minData(finalDecomp[own]);
nUnblocked++;
}
}
// Propagate information inwards
FaceCellWave<minData> deltaCalc
(
mesh,
seedFaces,
seedData,
faceData,
cellData,
mesh.globalData().nTotalCells()+1
);
// And extract
forAll(finalDecomp, celli)
{
if (cellData[celli].valid(deltaCalc.data()))
{
finalDecomp[celli] = cellData[celli].data();
}
}
}
// For specifiedProcessorFaces rework the cellToProc to enforce
// all on one processor since we can't guarantee that the input
// to regionSplit was a single region.
// E.g. faceSet 'a' with the cells split into two regions
// by a notch formed by two walls
//
// \ /
// \ /
// ---a----+-----a-----
//
//
// Note that reworking the cellToProc might make the decomposition
// unbalanced.
forAll(specifiedProcessorFaces, seti)
{
const labelList& set = specifiedProcessorFaces[seti];
label proci = specifiedProcessor[seti];
if (proci == -1)
{
// If no processor specified - use the one from the 0th element
if (set.size())
{
proci = finalDecomp[mesh.faceOwner()[set[0]]];
}
else
{
// Zero-sized processor (e.g. from redistributePar)
proci = 0;
}
}
for (const label facei : set)
{
const face& f = mesh.faces()[facei];
for (const label pointi : f)
{
const labelList& pFaces = mesh.pointFaces()[pointi];
for (const label pFacei : pFaces)
{
finalDecomp[mesh.faceOwner()[pFacei]] = proci;
if (mesh.isInternalFace(pFacei))
{
finalDecomp[mesh.faceNeighbour()[pFacei]] = proci;
}
}
}
}
}
if (debug && Pstream::parRun())
{
labelList nbrDecomp;
syncTools::swapBoundaryCellList(mesh, finalDecomp, nbrDecomp);
const polyBoundaryMesh& patches = mesh.boundaryMesh();
for (const polyPatch& pp : patches)
{
if (pp.coupled())
{
forAll(pp, i)
{
const label facei = pp.start()+i;
const label own = mesh.faceOwner()[facei];
const label bFacei = facei-mesh.nInternalFaces();
if (!blockedFace[facei])
{
const label ownProc = finalDecomp[own];
const label nbrProc = nbrDecomp[bFacei];
if (ownProc != nbrProc)
{
FatalErrorInFunction
<< "patch:" << pp.name()
<< " face:" << facei
<< " at:" << mesh.faceCentres()[facei]
<< " ownProc:" << ownProc
<< " nbrProc:" << nbrProc
<< exit(FatalError);
}
}
}
}
}
}
return finalDecomp;
}
void Foam::decompositionMethod::setConstraints
(
const polyMesh& mesh,
boolList& blockedFace,
PtrList<labelList>& specifiedProcessorFaces,
labelList& specifiedProcessor,
List<labelPair>& explicitConnections
) const
{
blockedFace.setSize(mesh.nFaces());
blockedFace = true;
specifiedProcessorFaces.clear();
explicitConnections.clear();
for (const decompositionConstraint& decompConstraint : constraints_)
{
decompConstraint.add
(
mesh,
blockedFace,
specifiedProcessorFaces,
specifiedProcessor,
explicitConnections
);
}
}
void Foam::decompositionMethod::applyConstraints
(
const polyMesh& mesh,
const boolList& blockedFace,
const PtrList<labelList>& specifiedProcessorFaces,
const labelList& specifiedProcessor,
const List<labelPair>& explicitConnections,
labelList& decomposition
) const
{
for (const decompositionConstraint& decompConstraint : constraints_)
{
decompConstraint.apply
(
mesh,
blockedFace,
specifiedProcessorFaces,
specifiedProcessor,
explicitConnections,
decomposition
);
}
}
Foam::labelList Foam::decompositionMethod::decompose
(
const polyMesh& mesh,
const scalarField& cellWeights
) const
{
// Collect all constraints
boolList blockedFace;
PtrList<labelList> specifiedProcessorFaces;
labelList specifiedProcessor;
List<labelPair> explicitConnections;
setConstraints
(
mesh,
blockedFace,
specifiedProcessorFaces,
specifiedProcessor,
explicitConnections
);
// Construct decomposition method and either do decomposition on
// cell centres or on agglomeration
labelList finalDecomp = decompose
(
mesh,
cellWeights, // optional weights
blockedFace, // any cells to be combined
specifiedProcessorFaces,// any whole cluster of cells to be kept
specifiedProcessor,
explicitConnections // baffles
);
// Give any constraint the option of modifying the decomposition
applyConstraints
(
mesh,
blockedFace,
specifiedProcessorFaces,
specifiedProcessor,
explicitConnections,
finalDecomp
);
return finalDecomp;
}
// * * * * * * * * * * * * * * * Stub Functions * * * * * * * * * * * * * * //
Foam::labelList Foam::decompositionMethod::decompose
(
const pointField& points,
const scalarField& pointWeights
) const
{
NotImplemented;
return labelList();
}
Foam::labelList Foam::decompositionMethod::decompose
(
const pointField& points
) const
{
NotImplemented;
return labelList();
}
// ************************************************************************* //