579 lines
15 KiB
C++
579 lines
15 KiB
C++
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | foam-extend: Open Source CFD
|
|
\\ / O peration | Version: 4.1
|
|
\\ / A nd | Web: http://www.foam-extend.org
|
|
\\/ M anipulation | For copyright notice see file Copyright
|
|
-------------------------------------------------------------------------------
|
|
License
|
|
This file is part of foam-extend.
|
|
|
|
foam-extend 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.
|
|
|
|
foam-extend 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 foam-extend. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "mergePolyMesh.H"
|
|
#include "foamTime.H"
|
|
#include "directTopoChange.H"
|
|
#include "mapPolyMesh.H"
|
|
#include "polyAddPoint.H"
|
|
#include "polyAddCell.H"
|
|
#include "polyAddFace.H"
|
|
|
|
// * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
|
|
|
|
defineTypeNameAndDebug(Foam::mergePolyMesh, 1);
|
|
|
|
|
|
// * * * * * * * * * * * * * Private Member Functions * * * * * * * * * * * //
|
|
|
|
Foam::label Foam::mergePolyMesh::patchIndex(const polyPatch& p)
|
|
{
|
|
// Find the patch name on the list. If the patch is already there
|
|
// and patch types match, return index
|
|
const word& pType = p.type();
|
|
const word& pName = p.name();
|
|
|
|
bool nameFound = false;
|
|
|
|
forAll (patchNames_, patchI)
|
|
{
|
|
if (patchNames_[patchI] == pName)
|
|
{
|
|
if (patchTypes_[patchI] == pType)
|
|
{
|
|
// Found name and types match
|
|
return patchI;
|
|
}
|
|
else
|
|
{
|
|
// Found the name, but type is different
|
|
nameFound = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Patch not found. Append to the list
|
|
patchTypes_.append(pType);
|
|
|
|
if (nameFound)
|
|
{
|
|
// Duplicate name is not allowed. Create a composite name from the
|
|
// patch name and case name
|
|
const word& caseName = p.boundaryMesh().mesh().time().caseName();
|
|
|
|
patchNames_.append(pName + "_" + caseName);
|
|
|
|
Info<< "label patchIndex(const polyPatch& p) : "
|
|
<< "Patch " << p.index() << " named "
|
|
<< pName << " in mesh " << caseName
|
|
<< " already exists, but patch types "
|
|
<< " do not match.\nCreating a composite name as "
|
|
<< patchNames_[patchNames_.size() - 1] << endl;
|
|
}
|
|
else
|
|
{
|
|
patchNames_.append(pName);
|
|
}
|
|
|
|
return patchNames_.size() - 1;
|
|
}
|
|
|
|
|
|
Foam::label Foam::mergePolyMesh::zoneIndex
|
|
(
|
|
DynamicList<word>& names,
|
|
const word& curName
|
|
)
|
|
{
|
|
forAll (names, zoneI)
|
|
{
|
|
if (names[zoneI] == curName)
|
|
{
|
|
return zoneI;
|
|
}
|
|
}
|
|
|
|
// Not found. Add new name to the list
|
|
names.append(curName);
|
|
|
|
return names.size() - 1;
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * * Constructors * * * * * * * * * * * * * * //
|
|
|
|
Foam::mergePolyMesh::mergePolyMesh(const IOobject& io)
|
|
:
|
|
polyMesh(io),
|
|
meshMod_(*this),
|
|
patchTypes_(2*boundaryMesh().size()),
|
|
patchNames_(2*boundaryMesh().size()),
|
|
pointZoneNames_(),
|
|
faceZoneNames_(),
|
|
cellZoneNames_(),
|
|
pointZones_(),
|
|
faceZones_(),
|
|
faceZoneFlips_(),
|
|
cellZones_()
|
|
{
|
|
// Insert the original patches into the list
|
|
wordList curPatchTypes = boundaryMesh().types();
|
|
wordList curPatchNames = boundaryMesh().names();
|
|
|
|
forAll (curPatchTypes, patchI)
|
|
{
|
|
patchTypes_.append(curPatchTypes[patchI]);
|
|
patchNames_.append(curPatchNames[patchI]);
|
|
}
|
|
|
|
// Insert point, face and cell zones into the list
|
|
|
|
// Point zones
|
|
wordList curPointZoneNames = pointZones().names();
|
|
if (curPointZoneNames.size())
|
|
{
|
|
pointZoneNames_.setCapacity(2*curPointZoneNames.size());
|
|
}
|
|
|
|
forAll (curPointZoneNames, zoneI)
|
|
{
|
|
pointZoneNames_.append(curPointZoneNames[zoneI]);
|
|
}
|
|
const pointField& p = points();
|
|
const pointZoneMesh& pz = pointZones();
|
|
forAll(p, pointI)
|
|
{
|
|
pointZones_.append(pz.whichZone(pointI));
|
|
}
|
|
|
|
// Face zones
|
|
wordList curFaceZoneNames = faceZones().names();
|
|
|
|
if (curFaceZoneNames.size())
|
|
{
|
|
faceZoneNames_.setCapacity(2*curFaceZoneNames.size());
|
|
}
|
|
forAll (curFaceZoneNames, zoneI)
|
|
{
|
|
faceZoneNames_.append(curFaceZoneNames[zoneI]);
|
|
}
|
|
const faceList& f = faces();
|
|
const faceZoneMesh& fz = faceZones();
|
|
forAll(f, faceI)
|
|
{
|
|
label zoneID = fz.whichZone(faceI);
|
|
faceZones_.append(zoneID);
|
|
|
|
if(zoneID > -1)
|
|
faceZoneFlips_.append(fz[zoneID].flipMap()[faceI]);
|
|
else
|
|
faceZoneFlips_.append(false);
|
|
}
|
|
faceZones_.shrink();
|
|
faceZoneFlips_.shrink();
|
|
|
|
// Cell zones
|
|
wordList curCellZoneNames = cellZones().names();
|
|
|
|
if (curCellZoneNames.size())
|
|
{
|
|
cellZoneNames_.setCapacity(2*curCellZoneNames.size());
|
|
}
|
|
forAll (curCellZoneNames, zoneI)
|
|
{
|
|
cellZoneNames_.append(curCellZoneNames[zoneI]);
|
|
}
|
|
const cellList& c = cells();
|
|
const cellZoneMesh& cz = cellZones();
|
|
forAll(c, cellI)
|
|
{
|
|
cellZones_.append(cz.whichZone(cellI));
|
|
}
|
|
cellZones_.shrink();
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * * Destructor * * * * * * * * * * * * * * * //
|
|
|
|
|
|
// * * * * * * * * * * * * * * * Member Functions * * * * * * * * * * * * * //
|
|
|
|
void Foam::mergePolyMesh::addMesh(const polyMesh& m)
|
|
{
|
|
// Add all the points, faces and cells of the new mesh
|
|
|
|
// Add points
|
|
|
|
label zoneID = -1;
|
|
|
|
const pointField& p = m.points();
|
|
labelList renumberPoints(p.size());
|
|
|
|
const pointZoneMesh& pz = m.pointZones();
|
|
labelList pointZoneIndices(pz.size());
|
|
|
|
forAll (pz, zoneI)
|
|
{
|
|
pointZoneIndices[zoneI] = zoneIndex(pointZoneNames_, pz[zoneI].name());
|
|
}
|
|
|
|
forAll (p, pointI)
|
|
{
|
|
// Grab zone ID. If a point is not in a zone, it will return -1
|
|
zoneID = pz.whichZone(pointI);
|
|
|
|
if (zoneID > -1)
|
|
{
|
|
// Translate zone ID into the new index
|
|
zoneID = pointZoneIndices[zoneID];
|
|
}
|
|
|
|
renumberPoints[pointI] =
|
|
meshMod_.setAction
|
|
(
|
|
polyAddPoint
|
|
(
|
|
p[pointI], // Point to add
|
|
-1, // Master point (straight addition)
|
|
zoneID, // Zone for point
|
|
pointI < m.nPoints() // Is in cell?
|
|
)
|
|
);
|
|
|
|
pointZones_.append(zoneID);
|
|
|
|
}
|
|
|
|
// Add cells
|
|
|
|
const cellList& c = m.cells();
|
|
labelList renumberCells(c.size());
|
|
|
|
const cellZoneMesh& cz = m.cellZones();
|
|
labelList cellZoneIndices(cz.size());
|
|
|
|
forAll (cz, zoneI)
|
|
{
|
|
cellZoneIndices[zoneI] = zoneIndex(cellZoneNames_, cz[zoneI].name());
|
|
}
|
|
|
|
forAll (c, cellI)
|
|
{
|
|
// Grab zone ID. If a cell is not in a zone, it will return -1
|
|
zoneID = cz.whichZone(cellI);
|
|
|
|
if (zoneID > -1)
|
|
{
|
|
// Translate zone ID into the new index
|
|
zoneID = cellZoneIndices[zoneID];
|
|
}
|
|
|
|
renumberCells[cellI] =
|
|
meshMod_.setAction
|
|
(
|
|
polyAddCell
|
|
(
|
|
-1, // Master point
|
|
-1, // Master edge
|
|
-1, // Master face
|
|
-1, // Master cell
|
|
zoneID // Zone for cell
|
|
)
|
|
);
|
|
|
|
cellZones_.append(zoneID);
|
|
}
|
|
|
|
// Add faces
|
|
const polyBoundaryMesh& bm = m.boundaryMesh();
|
|
|
|
// Gather the patch indices
|
|
labelList patchIndices(bm.size());
|
|
|
|
forAll (patchIndices, patchI)
|
|
{
|
|
patchIndices[patchI] = patchIndex(bm[patchI]);
|
|
}
|
|
|
|
// Temporary: update number of allowable patches. This should be
|
|
// determined at the top - before adding anything.
|
|
meshMod_.setNumPatches(patchNames_.size());
|
|
|
|
|
|
|
|
const faceZoneMesh& fz = m.faceZones();
|
|
labelList faceZoneIndices(fz.size());
|
|
|
|
forAll (fz, zoneI)
|
|
{
|
|
faceZoneIndices[zoneI] = zoneIndex(faceZoneNames_, fz[zoneI].name());
|
|
}
|
|
|
|
const faceList& f = m.faces();
|
|
labelList renumberFaces(f.size());
|
|
|
|
const labelList& own = m.faceOwner();
|
|
const labelList& nei = m.faceNeighbour();
|
|
|
|
label newOwn, newNei, newPatch, newZone;
|
|
bool newZoneFlip;
|
|
|
|
forAll (f, faceI)
|
|
{
|
|
const face& curFace = f[faceI];
|
|
|
|
face newFace(curFace.size());
|
|
|
|
forAll (curFace, pointI)
|
|
{
|
|
newFace[pointI] = renumberPoints[curFace[pointI]];
|
|
}
|
|
|
|
if (debug)
|
|
{
|
|
// Check that the face is valid
|
|
if (min(newFace) < 0)
|
|
{
|
|
FatalErrorIn("void mergePolyMesh::addMesh(const polyMesh&)")
|
|
<< "Error in point mapping for face " << faceI
|
|
<< ". Old face: " << curFace << " New face: " << newFace
|
|
<< abort(FatalError);
|
|
}
|
|
}
|
|
|
|
if (faceI < m.nInternalFaces() || faceI >= m.nFaces())
|
|
{
|
|
newPatch = -1;
|
|
}
|
|
else
|
|
{
|
|
newPatch = patchIndices[bm.whichPatch(faceI)];
|
|
}
|
|
|
|
newOwn = own[faceI];
|
|
if (newOwn > -1) newOwn = renumberCells[newOwn];
|
|
|
|
if (newPatch > -1)
|
|
{
|
|
newNei = -1;
|
|
}
|
|
else
|
|
{
|
|
newNei = nei[faceI];
|
|
newNei = renumberCells[newNei];
|
|
}
|
|
|
|
|
|
newZone = fz.whichZone(faceI);
|
|
newZoneFlip = false;
|
|
|
|
if (newZone > -1)
|
|
{
|
|
newZoneFlip = fz[newZone].flipMap()[fz[newZone].whichFace(faceI)];
|
|
|
|
// Grab the new zone
|
|
newZone = faceZoneIndices[newZone];
|
|
}
|
|
|
|
renumberFaces[faceI] =
|
|
meshMod_.setAction
|
|
(
|
|
polyAddFace
|
|
(
|
|
newFace,
|
|
newOwn,
|
|
newNei,
|
|
-1,
|
|
-1,
|
|
-1,
|
|
false,
|
|
newPatch,
|
|
newZone,
|
|
newZoneFlip
|
|
)
|
|
);
|
|
faceZones_.append(newZone);
|
|
faceZoneFlips_.append(newZoneFlip);
|
|
}
|
|
|
|
faceZones_.shrink();
|
|
faceZoneFlips_.shrink();
|
|
}
|
|
|
|
|
|
void Foam::mergePolyMesh::merge()
|
|
{
|
|
Info<< "patch names: " << patchNames_ << nl
|
|
<< "patch types: " << patchTypes_ << nl
|
|
<< "point zone names: " << pointZoneNames_ << nl
|
|
<< "face zone names: " << faceZoneNames_ << nl
|
|
<< "cell zone names: " << cellZoneNames_ << endl;
|
|
|
|
// Add the patches if necessary
|
|
if (patchNames_.size() != boundaryMesh().size())
|
|
{
|
|
Info << "Copying old patches" << endl;
|
|
|
|
List<polyPatch*> newPatches(patchNames_.size());
|
|
|
|
const polyBoundaryMesh& oldPatches = boundaryMesh();
|
|
|
|
// Note. Re-using counter in two for loops
|
|
label patchI = 0;
|
|
|
|
for (patchI = 0; patchI < oldPatches.size(); patchI++)
|
|
{
|
|
newPatches[patchI] = oldPatches[patchI].clone(oldPatches).ptr();
|
|
}
|
|
|
|
Info << "Adding new patches. " << endl;
|
|
|
|
label endOfLastPatch =
|
|
oldPatches[patchI - 1].start() + oldPatches[patchI - 1].size();
|
|
|
|
for (; patchI < patchNames_.size(); patchI++)
|
|
{
|
|
// Add a patch
|
|
newPatches[patchI] =
|
|
(
|
|
polyPatch::New
|
|
(
|
|
patchTypes_[patchI],
|
|
patchNames_[patchI],
|
|
0,
|
|
endOfLastPatch,
|
|
patchI,
|
|
oldPatches
|
|
).ptr()
|
|
);
|
|
}
|
|
|
|
removeBoundary();
|
|
addPatches(newPatches);
|
|
}
|
|
|
|
// Add the zones if necessary
|
|
if
|
|
(
|
|
pointZoneNames_.size() != pointZones().size()
|
|
|| faceZoneNames_.size() != faceZones().size()
|
|
|| cellZoneNames_.size() != cellZones().size()
|
|
)
|
|
{
|
|
|
|
List<pointZone*> pZones(pointZoneNames_.size());
|
|
List<faceZone*> fZones(faceZoneNames_.size());
|
|
List<cellZone*> cZones(cellZoneNames_.size());
|
|
|
|
if(pointZoneNames_.size() > 0)
|
|
{
|
|
Info << "Updating point zones" << endl;
|
|
|
|
List<dynamicLabelList > pzPoints(pointZoneNames_.size());
|
|
forAll(pointZones_, pointI)
|
|
{
|
|
label zoneID = pointZones_[pointI];
|
|
|
|
if(zoneID > -1)
|
|
pzPoints[zoneID].append(pointI);
|
|
}
|
|
|
|
forAll(pZones, i)
|
|
{
|
|
pZones[i] = new pointZone
|
|
(
|
|
pointZoneNames_[i],
|
|
pzPoints[i],
|
|
i,
|
|
pointZones()
|
|
);
|
|
}
|
|
}
|
|
|
|
if(faceZoneNames_.size() > 0)
|
|
{
|
|
Info << "Updating face zones" << endl;
|
|
|
|
List<dynamicLabelList > fzFaces(faceZoneNames_.size());
|
|
List<DynamicList<bool> > fzFlips(faceZoneNames_.size());
|
|
|
|
forAll(faceZones_, faceI)
|
|
{
|
|
label zoneID = faceZones_[faceI];
|
|
|
|
if(zoneID > -1)
|
|
{
|
|
fzFaces[zoneID].append(faceI);
|
|
fzFlips[zoneID].append(faceZoneFlips_[faceI]);
|
|
}
|
|
}
|
|
|
|
forAll(fZones, i)
|
|
{
|
|
fzFaces[i].shrink();
|
|
fzFlips[i].shrink();
|
|
|
|
fZones[i] = new faceZone
|
|
(
|
|
faceZoneNames_[i],
|
|
fzFaces[i],
|
|
fzFlips[i],
|
|
i,
|
|
faceZones()
|
|
);
|
|
}
|
|
}
|
|
|
|
if(cellZoneNames_.size() > 0)
|
|
{
|
|
Info << "Updating cell zones" << endl;
|
|
|
|
List<dynamicLabelList > czCells(cellZoneNames_.size());
|
|
forAll(cellZones_, cellI)
|
|
{
|
|
label zoneID = cellZones_[cellI];
|
|
|
|
if(zoneID > -1)
|
|
czCells[zoneID].append(cellI);
|
|
}
|
|
|
|
forAll(cZones, i)
|
|
{
|
|
czCells[i].shrink();
|
|
|
|
cZones[i] = new cellZone
|
|
(
|
|
cellZoneNames_[i],
|
|
czCells[i],
|
|
i,
|
|
cellZones()
|
|
);
|
|
}
|
|
}
|
|
|
|
removeZones();
|
|
addZones(pZones, fZones, cZones);
|
|
write();
|
|
}
|
|
|
|
// Change mesh. No inflation
|
|
meshMod_.changeMesh(*this, false);
|
|
|
|
// Clear topo change for the next operation
|
|
meshMod_.clear();
|
|
}
|
|
|
|
|
|
// ************************************************************************* //
|