This repository has been archived on 2023-11-20. You can view files and clone it, but cannot push or open issues or pull requests.
foam-extend4.1-coherent-io/applications/utilities/mesh/manipulation/mergeMeshes/mergePolyMesh.C
2015-05-17 15:58:16 +02:00

579 lines
15 KiB
C

/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | foam-extend: Open Source CFD
\\ / O peration | Version: 3.2
\\ / 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 "Time.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<DynamicList<label> > 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<DynamicList<label> > 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<DynamicList<label> > 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();
}
// ************************************************************************* //