openfoam/applications/utilities/postProcessing/dataConversion/foamToEnsight/writeVolFields.H
Mark Olesen db16d80840 ENH: use objectRegistry/IOobjectList sorted instead of lookupClass
- in most cases a parallel-consistent order is required.
  Even when the order is not important, it will generally require
  fewer allocations to create a UPtrList of entries instead of a
  HashTable or even a wordList.
2023-07-31 20:11:32 +02:00

153 lines
3.6 KiB
C++

/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration |
\\ / A nd | www.openfoam.com
\\/ M anipulation |
-------------------------------------------------------------------------------
Copyright (C) 2018-2023 OpenCFD Ltd.
-------------------------------------------------------------------------------
License
This file is part of OpenFOAM, distributed under GPL-3.0-or-later.
InNamespace
Foam
Description
Read volume fields from disk and write with ensightMesh
\*---------------------------------------------------------------------------*/
#ifndef ensight_writeVolFields_H
#define ensight_writeVolFields_H
#include "readFields.H"
#include "fvMesh.H"
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
namespace Foam
{
template<class Type>
bool writeVolField
(
ensightCase& ensCase,
const ensightMesh& ensMesh,
const tmp<VolumeField<Type>>& tfield,
const bool nearCellValue = false
)
{
if (!tfield)
{
return false;
}
else if (nearCellValue)
{
auto tzgrad = makeZeroGradientField<Type>(tfield);
// Recursive call
return writeVolField
(
ensCase,
ensMesh,
tzgrad,
false // No nearCellValue, we already have zero-gradient
);
}
const auto& field = tfield();
// Forced use of node values?
const bool nodeValues = ensCase.nodeValues();
autoPtr<ensightFile> os =
ensCase.newData<Type>(field.name(), nodeValues);
bool wrote = ensightOutput::writeVolField<Type>
(
os.ref(),
field,
ensMesh,
nodeValues
);
tfield.clear();
return wrote;
}
template<class Type>
label writeVolFields
(
ensightCase& ensCase,
const ensightMesh& ensMesh,
const IOobjectList& objects,
const bool nearCellValue = false
)
{
typedef VolumeField<Type> FieldType;
const auto& mesh = refCast<const fvMesh>(ensMesh.mesh());
label count = 0;
for (const IOobject& io : objects.csorted<FieldType>())
{
if
(
writeVolField<Type>
(
ensCase,
ensMesh,
getField<FieldType>(io, mesh),
nearCellValue
)
)
{
Info<< ' ' << io.name();
++count;
}
}
return count;
}
label writeAllVolFields
(
ensightCase& ensCase,
const ensightMesh& ensMesh,
const IOobjectList& objects,
const bool nearCellValue = false
)
{
#undef ensight_WRITE_FIELD
#define ensight_WRITE_FIELD(PrimitiveType) \
writeVolFields<PrimitiveType> \
( \
ensCase, \
ensMesh, \
objects, \
nearCellValue \
)
label count = 0;
count += ensight_WRITE_FIELD(scalar);
count += ensight_WRITE_FIELD(vector);
count += ensight_WRITE_FIELD(sphericalTensor);
count += ensight_WRITE_FIELD(symmTensor);
count += ensight_WRITE_FIELD(tensor);
#undef ensight_WRITE_FIELD
return count;
}
} // End namespace Foam
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
#endif
// ************************************************************************* //