172 lines
4.6 KiB
C
172 lines
4.6 KiB
C
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
|
\\ / O peration |
|
|
\\ / A nd | Copyright held by original author
|
|
\\/ M anipulation |
|
|
-------------------------------------------------------------------------------
|
|
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 2 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, write to the Free Software Foundation,
|
|
Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "internalWriter.H"
|
|
#include "writeFuns.H"
|
|
|
|
// * * * * * * * * * * * * * * * * Constructors * * * * * * * * * * * * * * //
|
|
|
|
// Construct from components
|
|
Foam::internalWriter::internalWriter
|
|
(
|
|
const vtkMesh& vMesh,
|
|
const bool binary,
|
|
const fileName& fName
|
|
)
|
|
:
|
|
vMesh_(vMesh),
|
|
binary_(binary),
|
|
fName_(fName),
|
|
os_(fName.c_str())
|
|
{
|
|
const fvMesh& mesh = vMesh_.mesh();
|
|
const vtkTopo& topo = vMesh_.topo();
|
|
|
|
// Write header
|
|
writeFuns::writeHeader(os_, binary_, mesh.time().caseName());
|
|
os_ << "DATASET UNSTRUCTURED_GRID" << std::endl;
|
|
|
|
|
|
//------------------------------------------------------------------
|
|
//
|
|
// Write topology
|
|
//
|
|
//------------------------------------------------------------------
|
|
|
|
const labelList& addPointCellLabels = topo.addPointCellLabels();
|
|
const label nTotPoints = mesh.nPoints() + addPointCellLabels.size();
|
|
|
|
os_ << "POINTS " << nTotPoints
|
|
<< " float" << std::endl;
|
|
|
|
DynamicList<floatScalar> ptField(3*nTotPoints);
|
|
|
|
writeFuns::insert(mesh.points(), ptField);
|
|
|
|
const pointField& ctrs = mesh.cellCentres();
|
|
forAll(addPointCellLabels, api)
|
|
{
|
|
writeFuns::insert(ctrs[addPointCellLabels[api]], ptField);
|
|
}
|
|
writeFuns::write(os_, binary_, ptField);
|
|
|
|
|
|
//
|
|
// Write cells
|
|
//
|
|
|
|
const labelListList& vtkVertLabels = topo.vertLabels();
|
|
|
|
// Count total number of vertices referenced.
|
|
label nFaceVerts = 0;
|
|
|
|
forAll(vtkVertLabels, cellI)
|
|
{
|
|
nFaceVerts += vtkVertLabels[cellI].size() + 1;
|
|
}
|
|
|
|
os_ << "CELLS " << vtkVertLabels.size() << ' ' << nFaceVerts
|
|
<< std::endl;
|
|
|
|
|
|
DynamicList<label> vertLabels(nFaceVerts);
|
|
|
|
forAll(vtkVertLabels, cellI)
|
|
{
|
|
const labelList& vtkVerts = vtkVertLabels[cellI];
|
|
|
|
vertLabels.append(vtkVerts.size());
|
|
|
|
writeFuns::insert(vtkVerts, vertLabels);
|
|
}
|
|
writeFuns::write(os_, binary_, vertLabels);
|
|
|
|
|
|
|
|
const labelList& vtkCellTypes = topo.cellTypes();
|
|
|
|
os_ << "CELL_TYPES " << vtkCellTypes.size() << std::endl;
|
|
|
|
// Make copy since writing might swap stuff.
|
|
DynamicList<label> cellTypes(vtkCellTypes.size());
|
|
|
|
writeFuns::insert(vtkCellTypes, cellTypes);
|
|
|
|
writeFuns::write(os_, binary_, cellTypes);
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * Member Functions * * * * * * * * * * * * * //
|
|
|
|
void Foam::internalWriter::writeCellIDs()
|
|
{
|
|
const fvMesh& mesh = vMesh_.mesh();
|
|
const vtkTopo& topo = vMesh_.topo();
|
|
const labelList& vtkCellTypes = topo.cellTypes();
|
|
const labelList& superCells = topo.superCells();
|
|
|
|
// Cell ids first
|
|
os_ << "cellID 1 " << vtkCellTypes.size() << " int"
|
|
<< std::endl;
|
|
|
|
labelList cellId(vtkCellTypes.size());
|
|
label labelI = 0;
|
|
|
|
|
|
if (vMesh_.useSubMesh())
|
|
{
|
|
const labelList& cMap = vMesh_.subsetMesh().cellMap();
|
|
|
|
forAll(mesh.cells(), cellI)
|
|
{
|
|
cellId[labelI++] = cMap[cellI];
|
|
}
|
|
forAll(superCells, superCellI)
|
|
{
|
|
label origCellI = cMap[superCells[superCellI]];
|
|
|
|
cellId[labelI++] = origCellI;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
forAll(mesh.cells(), cellI)
|
|
{
|
|
cellId[labelI++] = cellI;
|
|
}
|
|
forAll(superCells, superCellI)
|
|
{
|
|
label origCellI = superCells[superCellI];
|
|
|
|
cellId[labelI++] = origCellI;
|
|
}
|
|
}
|
|
|
|
writeFuns::write(os_, binary_, cellId);
|
|
}
|
|
|
|
|
|
// ************************************************************************* //
|