Files
openfoam/src/fileFormats/ensight/output/ensightOutput.C
Mark Olesen a97628121c ENH: overhaul ensight handling (#1579)
- includes restructuring and simplification of low-level ensight part
  handling and refactor of backends to improve code reuse.

foamToEnsight
-------------

  * new cellZone support.
    This was previously only possible via a separate foamToEnsightParts
    utility that was not parallelized.

  * support for point fields.

  * `-nearCellValue` option (as per foamToVTK)

  * data indexing now uses values from the time index.
    This is consistent with the ensightWrite function object and
    can help with restarts.

  * existing ensight directories are removed, unless the -no-overwrite
    option is supplied

foamToEnsightParts
------------------
  * now redundant and removed.

ensightOutputSurface (new class)
--------------------------------
  * a lightweight wrapper for point/face references that is tailored
    for the ensightSurfaceWriter. It uses compact face/point information
    and is serial only, since this is the format requirements from the
    surfaceWriter class.

ensightMesh (revised class)
---------------------------
  * now only holds a polyMesh reference, which removes its dependency
    on finiteVolume and allows it to be relocated under fileFormats
    instead of conversion.

Removed classes: ensightParts, ensighPartFaces, ensightPartCells

- these were used by foamToEnsightParts, but not needed anymore.
2020-02-18 11:09:43 +00:00

491 lines
10 KiB
C

/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration |
\\ / A nd | www.openfoam.com
\\/ M anipulation |
-------------------------------------------------------------------------------
Copyright (C) 2020 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 "ensightOutput.H"
#include "cell.H"
#include "cellShape.H"
#include "face.H"
#include "polyMesh.H"
#include "ListOps.H"
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
// Sizes
Foam::labelList Foam::ensightOutput::Detail::getFaceSizes
(
const UList<face>& faces
)
{
labelList list(faces.size());
auto outIter = list.begin();
for (const face& f : faces)
{
*outIter = f.size();
++outIter;
}
return list;
}
Foam::labelList Foam::ensightOutput::Detail::getFaceSizes
(
const UIndirectList<face>& faces
)
{
labelList list(faces.size());
auto outIter = list.begin();
for (const face& f : faces)
{
*outIter = f.size();
++outIter;
}
return list;
}
Foam::labelList Foam::ensightOutput::Detail::getPolysNFaces
(
const polyMesh& mesh,
const labelUList& addr
)
{
const cellList& meshCells = mesh.cells();
labelList list(addr.size());
auto outIter = list.begin();
// The number of faces per element
for (const label cellId : addr)
{
*outIter = meshCells[cellId].size();
++outIter;
}
return list;
}
Foam::labelList Foam::ensightOutput::Detail::getPolysNPointsPerFace
(
const polyMesh& mesh,
const labelUList& addr
)
{
const cellList& meshCells = mesh.cells();
const faceList& meshFaces = mesh.faces();
// Count the number of faces per element
label nTotFaces = 0;
for (const label cellId : addr)
{
nTotFaces += meshCells[cellId].size();
}
labelList list(nTotFaces);
auto outIter = list.begin();
// The number of points per element face
for (const label cellId : addr)
{
for (const label facei : meshCells[cellId])
{
*outIter = meshFaces[facei].size();
++outIter;
}
}
return list;
}
void Foam::ensightOutput::writeFaceList
(
ensightGeoFile& os,
const UList<face>& faces
)
{
for (const face& f : faces)
{
for (const label labi : f)
{
os.write(labi + 1);
}
os.newline();
}
}
void Foam::ensightOutput::writeFaceList
(
ensightGeoFile& os,
const UIndirectList<face>& faces
)
{
for (const face& f : faces)
{
for (const label labi : f)
{
os.write(labi + 1);
}
os.newline();
}
}
void Foam::ensightOutput::writeCellShapes
(
ensightGeoFile& os,
const UList<cellShape>& shapes
)
{
for (const cellShape& cellPoints : shapes)
{
// Convert global -> local index
// (note: Ensight indices start with 1)
// In ASCII, write one cell per line
for (const label pointi : cellPoints)
{
os.write(pointi + 1);
}
os.newline();
}
}
void Foam::ensightOutput::writePolysPoints
(
ensightGeoFile& os,
const polyMesh& mesh,
const labelUList& addr,
const labelList& pointMap
)
{
const cellList& meshCells = mesh.cells();
const faceList& meshFaces = mesh.faces();
const labelList& owner = mesh.faceOwner();
for (const label cellId : addr)
{
for (const label faceId : meshCells[cellId])
{
const face& f = meshFaces[faceId];
if (faceId < owner.size() && owner[faceId] != cellId)
{
// The neighbour of an internal face
// - write as face::reverseFace()
os.write(pointMap[f[0]] + 1);
for (label pti = f.size()-1; pti > 0; --pti)
{
os.write(pointMap[f[pti]] + 1);
}
}
else
{
for (const label pointi : f)
{
os.write(pointMap[pointi] + 1);
}
}
os.newline();
}
}
}
void Foam::ensightOutput::writePolysPoints
(
ensightGeoFile& os,
const cellUList& meshCells,
const labelUList& addr,
const faceUList& meshFaces,
const labelUList& owner
)
{
for (const label cellId : addr)
{
for (const label faceId : meshCells[cellId])
{
const face& f = meshFaces[faceId];
if (faceId < owner.size() && owner[faceId] != cellId)
{
// The neighbour of an internal face
// - write as face::reverseFace()
os.write(f[0] + 1);
for (label pti = f.size()-1; pti > 0; --pti)
{
os.write(f[pti] + 1);
}
}
else
{
for (const label pointi : f)
{
os.write(pointi + 1);
}
}
os.newline();
}
}
}
void Foam::ensightOutput::writeFaceConnectivity
(
ensightGeoFile& os,
const ensightFaces::elemType etype,
const label nTotal,
const faceUList& faces,
bool parallel
)
{
if (!nTotal)
{
return;
}
parallel = parallel && Pstream::parRun();
const label nSlaves = (parallel ? Pstream::nProcs() : 0);
if (Pstream::master())
{
os.writeKeyword(ensightFaces::key(etype));
os.write(nTotal);
os.newline();
}
if (etype == ensightFaces::NSIDED)
{
// Face sizes (number of points per face)
labelList send(ensightOutput::Detail::getFaceSizes(faces));
if (Pstream::master())
{
os.writeLabels(send);
for (int slave=1; slave < nSlaves; ++slave)
{
IPstream fromSlave(Pstream::commsTypes::scheduled, slave);
labelList recv(fromSlave);
os.writeLabels(recv);
}
}
else if (nSlaves)
{
OPstream toMaster
(
Pstream::commsTypes::scheduled,
Pstream::masterNo()
);
toMaster << send;
}
}
// List of points id for each face
if (Pstream::master())
{
writeFaceList(os, faces);
for (int slave=1; slave < nSlaves; ++slave)
{
IPstream fromSlave(Pstream::commsTypes::scheduled, slave);
List<face> recv(fromSlave);
writeFaceList(os, recv);
}
}
else if (nSlaves)
{
OPstream toMaster
(
Pstream::commsTypes::scheduled,
Pstream::masterNo()
);
toMaster << faces;
}
}
void Foam::ensightOutput::writeFaceConnectivity
(
ensightGeoFile& os,
const ensightFaces::elemType etype,
const label nTotal,
const UIndirectList<face>& faces,
bool parallel
)
{
if (!nTotal)
{
return;
}
parallel = parallel && Pstream::parRun();
const label nSlaves = (parallel ? Pstream::nProcs() : 0);
if (Pstream::master())
{
os.writeKeyword(ensightFaces::key(etype));
os.write(nTotal);
os.newline();
}
if (etype == ensightFaces::NSIDED)
{
// Face sizes (number of points per face)
labelList send(ensightOutput::Detail::getFaceSizes(faces));
if (Pstream::master())
{
os.writeLabels(send);
for (int slave=1; slave < nSlaves; ++slave)
{
IPstream fromSlave(Pstream::commsTypes::scheduled, slave);
labelList recv(fromSlave);
os.writeLabels(recv);
}
}
else if (nSlaves)
{
OPstream toMaster
(
Pstream::commsTypes::scheduled,
Pstream::masterNo()
);
toMaster << send;
}
}
// List of points id per face
if (Pstream::master())
{
writeFaceList(os, faces);
for (int slave=1; slave < nSlaves; ++slave)
{
IPstream fromSlave(Pstream::commsTypes::scheduled, slave);
List<face> recv(fromSlave);
writeFaceList(os, recv);
}
}
else if (nSlaves)
{
OPstream toMaster
(
Pstream::commsTypes::scheduled,
Pstream::masterNo()
);
toMaster << faces;
}
}
void Foam::ensightOutput::writeFaceConnectivity
(
ensightGeoFile& os,
const ensightFaces& part,
const faceUList& faces,
bool parallel
)
{
for (label typei=0; typei < ensightFaces::nTypes; ++typei)
{
const auto etype = ensightFaces::elemType(typei);
writeFaceConnectivity
(
os,
etype,
part.total(etype),
UIndirectList<face>(faces, part.faceIds(etype)),
parallel
);
}
}
void Foam::ensightOutput::writeFaceConnectivityPresorted
(
ensightGeoFile& os,
const ensightFaces& part,
const faceUList& faces,
bool parallel
)
{
for (label typei=0; typei < ensightFaces::nTypes; ++typei)
{
const auto etype = ensightFaces::elemType(typei);
writeFaceConnectivity
(
os,
etype,
part.total(etype),
SubList<face>(faces, part.range(etype)),
parallel
);
}
}
// ************************************************************************* //