mirror of
https://github.com/PhasicFlow/phasicFlow.git
synced 2025-07-08 03:07:03 +00:00
pFlowToVTK enhanced
- pointField conversion is completed, tested (finalized). - geometry conversion is not complete yet.
This commit is contained in:
Hamidreza Norouzi
332
utilities/pFlowToVTK/pointFieldToVTK.cpp
Normal file
332
utilities/pFlowToVTK/pointFieldToVTK.cpp
Normal file
@ -0,0 +1,332 @@
|
||||
#include <regex>
|
||||
|
||||
#include "vtkFile.hpp"
|
||||
#include "vocabs.hpp"
|
||||
#include "pointFieldToVTK.hpp"
|
||||
|
||||
bool pFlow::PFtoVTK::convertTimeFolderPointFields(
|
||||
systemControl &control,
|
||||
const fileSystem &destPath,
|
||||
const word &bName)
|
||||
{
|
||||
|
||||
fileSystem timeFolder = control.time().path();
|
||||
// check if pointStructure exist in this folder
|
||||
IOfileHeader pStructHeader(
|
||||
objectFile(
|
||||
pointStructureFile__,
|
||||
timeFolder,
|
||||
objectFile::READ_ALWAYS,
|
||||
objectFile::WRITE_ALWAYS));
|
||||
|
||||
if (!pStructHeader.headerOk(true))
|
||||
{
|
||||
output << yellowColor << "Time folder " <<
|
||||
control.time().path() << " does not contain any pStructure data file."<<
|
||||
" Skipping this folder . . ."<< defaultColor << nl;
|
||||
return true;
|
||||
}
|
||||
|
||||
vtkFile vtk(destPath, bName, control.time().currentTime());
|
||||
|
||||
if (!vtk)
|
||||
return false;
|
||||
|
||||
auto pStruct = pointStructure(control);
|
||||
|
||||
// get a list of files in this timeFolder;
|
||||
|
||||
auto posVec = pStruct.pointPositionHost();
|
||||
auto *pos = posVec.data();
|
||||
|
||||
REPORT(1) << "Writing pointStructure to vtk file with " <<
|
||||
Yellow_Text(pStruct.numActive())<<
|
||||
" active particles" << END_REPORT;
|
||||
|
||||
addUndstrcuturedGridField(
|
||||
vtk(),
|
||||
pos,
|
||||
pStruct.numActive());
|
||||
|
||||
auto fileList = containingFiles(timeFolder);
|
||||
|
||||
for (const auto &file : fileList)
|
||||
{
|
||||
|
||||
IOfileHeader fieldHeader(
|
||||
objectFile(
|
||||
file.fileName(),
|
||||
file.dirPath(),
|
||||
objectFile::READ_ALWAYS,
|
||||
objectFile::WRITE_ALWAYS));
|
||||
|
||||
if (fieldHeader.headerOk(true))
|
||||
{
|
||||
if(
|
||||
convertRealx3TypePointField(vtk(), fieldHeader, pStruct) ||
|
||||
convertRealTypePointField(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<uint32>(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<uint64>(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<int32>(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<int64>(vtk(), fieldHeader, pStruct)||
|
||||
fieldHeader.objectName() == pointStructureFile__ )
|
||||
{
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
WARNING << " This object type, " <<
|
||||
fieldHeader.objectType() << " is not supported" <<
|
||||
END_WARNING;
|
||||
}
|
||||
}
|
||||
output << endl;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool pFlow::PFtoVTK::convertTimeFolderPointFieldsSelected(
|
||||
systemControl &control,
|
||||
const fileSystem &destPath,
|
||||
const word &bName,
|
||||
const wordVector &fieldsName,
|
||||
bool mustExist)
|
||||
{
|
||||
fileSystem timeFolder = control.time().path();
|
||||
// check if pointStructure exist in this folder
|
||||
IOfileHeader pStructHeader(
|
||||
objectFile(
|
||||
pointStructureFile__,
|
||||
timeFolder,
|
||||
objectFile::READ_ALWAYS,
|
||||
objectFile::WRITE_ALWAYS));
|
||||
|
||||
if (!pStructHeader.headerOk(true))
|
||||
{
|
||||
output << yellowColor << "Time folder " <<
|
||||
control.time().path() <<
|
||||
" does not contain any pStructure data file."<<
|
||||
" Skipping this folder . . ."<< defaultColor << nl;
|
||||
return true;
|
||||
}
|
||||
|
||||
vtkFile vtk(destPath, bName, control.time().currentTime());
|
||||
|
||||
if (!vtk)
|
||||
return false;
|
||||
|
||||
auto pStruct = pointStructure(control);
|
||||
|
||||
// get a list of files in this timeFolder;
|
||||
|
||||
auto posVec = pStruct.pointPositionHost();
|
||||
auto *pos = posVec.data();
|
||||
|
||||
REPORT(1) << "Writing pointStructure to vtk file with " <<
|
||||
Yellow_Text(pStruct.numActive())
|
||||
<< " active particles" << END_REPORT;
|
||||
|
||||
addUndstrcuturedGridField(
|
||||
vtk(),
|
||||
pos,
|
||||
pStruct.numActive());
|
||||
|
||||
auto fileList = containingFiles(timeFolder);
|
||||
|
||||
for (const auto &fname : fieldsName)
|
||||
{
|
||||
fileSystem fieldAddress = timeFolder + fname;
|
||||
|
||||
IOfileHeader fieldHeader(
|
||||
objectFile(
|
||||
fname,
|
||||
timeFolder,
|
||||
objectFile::READ_ALWAYS,
|
||||
objectFile::WRITE_ALWAYS));
|
||||
|
||||
if (fieldHeader.headerOk(true))
|
||||
{
|
||||
if (
|
||||
convertRealx3TypePointField(vtk(), fieldHeader, pStruct) ||
|
||||
convertRealTypePointField(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<uint32>(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<uint64>(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<int32>(vtk(), fieldHeader, pStruct) ||
|
||||
convertIntPointField<int64>(vtk(), fieldHeader, pStruct) ||
|
||||
fieldHeader.objectName() == pointStructureFile__ )
|
||||
{
|
||||
continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
|
||||
WARNING << " This object type, " <<
|
||||
fieldHeader.objectType() << " is not supported" <<
|
||||
END_WARNING;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (mustExist)
|
||||
{
|
||||
fatalErrorInFunction << "Field " << fieldAddress <<
|
||||
" does not exist." << endl;
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
REPORT(1) << "Could not find " << Yellow_Text(fieldAddress) <<
|
||||
". Skipping this field . . ." << END_REPORT;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool pFlow::PFtoVTK::addUndstrcuturedGridField(
|
||||
iOstream &os,
|
||||
realx3 *position,
|
||||
uint32 numPoints)
|
||||
{
|
||||
|
||||
os << "DATASET UNSTRUCTURED_GRID\n";
|
||||
os << "POINTS " << numPoints << " float\n";
|
||||
|
||||
if (numPoints == 0)
|
||||
return true;
|
||||
|
||||
for (uint32 i = 0; i < numPoints; i++)
|
||||
{
|
||||
os << position[i].x() <<
|
||||
' ' << position[i].y() <<
|
||||
' ' << position[i].z() << '\n';
|
||||
}
|
||||
|
||||
os << "CELLS " << numPoints << ' ' << 2 * numPoints << '\n';
|
||||
for (uint32 i = 0; i < numPoints; i++)
|
||||
{
|
||||
os << 1 << ' ' << i << '\n';
|
||||
}
|
||||
|
||||
os << "CELL_TYPES " << numPoints << '\n';
|
||||
|
||||
for (int32 i = 0; i < numPoints; i++)
|
||||
{
|
||||
os << 1 << '\n';
|
||||
}
|
||||
|
||||
os << "POINT_DATA " << numPoints << endl;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool pFlow::PFtoVTK::convertRealTypePointField(
|
||||
iOstream &os,
|
||||
const IOfileHeader &header,
|
||||
pointStructure &pStruct)
|
||||
{
|
||||
word objectType = header.objectType();
|
||||
|
||||
if (!checkFieldType<real>(objectType))
|
||||
return false;
|
||||
|
||||
auto field = realPointField_H(
|
||||
header,
|
||||
pStruct,
|
||||
static_cast<real>(0));
|
||||
|
||||
real const *data = field.deviceViewAll().data();
|
||||
|
||||
REPORT(1) << "writing " << Green_Text(header.objectName()) <<
|
||||
" field to vtk." << END_REPORT;
|
||||
|
||||
return addRealPointField(
|
||||
os,
|
||||
header.objectName(),
|
||||
data,
|
||||
pStruct.numActive());
|
||||
}
|
||||
|
||||
bool pFlow::PFtoVTK::convertRealx3TypePointField(
|
||||
iOstream &os,
|
||||
const IOfileHeader &header,
|
||||
pointStructure &pStruct)
|
||||
{
|
||||
word objectType = header.objectType();
|
||||
|
||||
if (!checkFieldType<realx3>(objectType))
|
||||
return false;
|
||||
|
||||
auto field = realx3PointField_H(
|
||||
header,
|
||||
pStruct,
|
||||
{0.0, 0.0, 0.0});
|
||||
|
||||
realx3 const *data = field.deviceViewAll().data();
|
||||
|
||||
REPORT(1) << "writing " << Green_Text(header.objectName()) <<
|
||||
" field to vtk." << END_REPORT;
|
||||
|
||||
return addRealx3PointField(
|
||||
os,
|
||||
header.objectName(),
|
||||
data,
|
||||
pStruct.numActive());
|
||||
}
|
||||
|
||||
bool pFlow::PFtoVTK::addRealPointField(
|
||||
iOstream &os,
|
||||
const word &fieldName,
|
||||
const real *field,
|
||||
uint32 numData)
|
||||
{
|
||||
if (numData == 0)
|
||||
return true;
|
||||
|
||||
os << "FIELD FieldData 1\n"
|
||||
<< fieldName << " 1 " << numData << " float\n";
|
||||
for (uint32 i = 0; i < numData; ++i)
|
||||
{
|
||||
os << field[i] << '\n';
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool pFlow::PFtoVTK::addRealx3PointField(
|
||||
iOstream &os,
|
||||
const word &fieldName,
|
||||
const realx3 *field,
|
||||
uint32 numData)
|
||||
{
|
||||
if (numData == 0)
|
||||
return true;
|
||||
|
||||
os << "FIELD FieldData 1\n"
|
||||
<< fieldName << " 3 " << numData << " float\n";
|
||||
for (uint32 i = 0; i < numData; ++i)
|
||||
{
|
||||
os << field[i].x() <<
|
||||
' ' << field[i].y() <<
|
||||
' ' << field[i].z() << '\n';
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool pFlow::PFtoVTK::regexCheck(const word &TYPENAME, const word &fieldType)
|
||||
{
|
||||
std::regex match("pointField\\<([A-Za-z1-9_]*)\\,([A-Za-z1-9_]*)\\>");
|
||||
std::smatch search1;
|
||||
std::smatch search2;
|
||||
if (!std::regex_match(fieldType, search1, match))
|
||||
return false;
|
||||
if (!std::regex_match(TYPENAME, search2, match))
|
||||
return false;
|
||||
if (search1.size() != 3)
|
||||
return false;
|
||||
if (search1.size() != search2.size())
|
||||
return false;
|
||||
return search1[1] == search2[1];
|
||||
}
|
||||
Reference in New Issue
Block a user