373 lines
8.4 KiB
C++
373 lines
8.4 KiB
C++
#include "vocabs.hpp"
|
|
#include "vtkFile.hpp"
|
|
#include "triSurfaceFieldToVTK.hpp"
|
|
|
|
bool pFlow::TSFtoVTK::convertTimeFolderTriSurfaceFields(
|
|
systemControl &control,
|
|
const fileSystem &destPath,
|
|
const word &bName,
|
|
bool separate,
|
|
wordList& surfNames,
|
|
wordList& fileNames)
|
|
{
|
|
|
|
auto timeFolder = control.geometry().path();
|
|
|
|
surfNames.clear();
|
|
fileNames.clear();
|
|
|
|
// check if pointStructure exist in this folder
|
|
IOfileHeader triSurfaeHeader(
|
|
objectFile(
|
|
triSurfaceFile__,
|
|
timeFolder,
|
|
objectFile::READ_ALWAYS,
|
|
objectFile::WRITE_ALWAYS));
|
|
|
|
if (!triSurfaeHeader.headerOk(true))
|
|
{
|
|
WARNING << "Time folder " << timeFolder << " does not contain any triSurface data file."
|
|
<< " Skipping this folder . . ." << END_WARNING;
|
|
return true;
|
|
}
|
|
|
|
auto triSurfaceObj = multiTriSurface(
|
|
objectFile(
|
|
triSurfaceFile__,
|
|
"",
|
|
objectFile::READ_ALWAYS,
|
|
objectFile::WRITE_ALWAYS),
|
|
&control.geometry());
|
|
|
|
if(separate)
|
|
{
|
|
return convertTimeFolderTriSurfaceFieldsSeparate(
|
|
triSurfaceObj,
|
|
destPath,
|
|
control.time().currentTime(),
|
|
bName,
|
|
surfNames,
|
|
fileNames);
|
|
}
|
|
else
|
|
{
|
|
return convertTimeFolderTriSurfaceFieldsSingle(
|
|
triSurfaceObj,
|
|
destPath,
|
|
control.time().currentTime(),
|
|
bName,
|
|
surfNames,
|
|
fileNames );
|
|
}
|
|
}
|
|
|
|
bool pFlow::TSFtoVTK::triSurfaceToVTK(
|
|
Ostream &os,
|
|
const realx3 *points,
|
|
const uint32x3 *vertices,
|
|
const subSurface &subSurf)
|
|
{
|
|
auto nP = subSurf.numPoints();
|
|
|
|
os << "DATASET UNSTRUCTURED_GRID" << endl;
|
|
os << "POINTS " << nP << " float" << endl;
|
|
|
|
for ( auto i=subSurf.pointStart(); i<subSurf.pointEnd(); i++ )
|
|
{
|
|
os << points[i].x() <<
|
|
" " << points[i].y() <<
|
|
" " << points[i].z() << endl;
|
|
if (!os) return false;
|
|
}
|
|
|
|
auto nTri = subSurf.size();
|
|
|
|
os << "CELLS " << nTri << " " << 4*nTri << endl;
|
|
|
|
for(auto i=subSurf.start(); i<subSurf.end(); i++)
|
|
{
|
|
os<< 3 <<
|
|
" "<< vertices[i].x()-subSurf.pointStart() <<
|
|
" "<< vertices[i].y()-subSurf.pointStart() <<
|
|
" "<< vertices[i].z()-subSurf.pointStart()<<endl;
|
|
if (!os) return false;
|
|
}
|
|
|
|
os<<"CELL_TYPES "<< nTri<<'\n';
|
|
|
|
for(uint32 i=0; i<nTri; i++)
|
|
{
|
|
os<< 5 <<'\n';
|
|
}
|
|
|
|
os << "CELL_DATA " << nTri << endl;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool pFlow::TSFtoVTK::triSurfaceToVTK(
|
|
Ostream &os,
|
|
const realx3 *points,
|
|
const uint32x3 *vertices,
|
|
uint32 numPoints,
|
|
uint32 numTris)
|
|
{
|
|
|
|
os << "DATASET UNSTRUCTURED_GRID" << endl;
|
|
os << "POINTS " << numPoints << " float" << endl;
|
|
|
|
for ( auto i=0; i<numPoints; i++ )
|
|
{
|
|
os << points[i].x() <<
|
|
" " << points[i].y() <<
|
|
" " << points[i].z() << endl;
|
|
if (!os) return false;
|
|
}
|
|
|
|
os << "CELLS " << numTris << " " << 4*numTris << endl;
|
|
|
|
for(auto i=0; i<numTris; i++)
|
|
{
|
|
os<< 3 <<
|
|
" "<< vertices[i].x()<<
|
|
" "<< vertices[i].y()<<
|
|
" "<< vertices[i].z()<<endl;
|
|
if (!os) return false;
|
|
}
|
|
|
|
os<<"CELL_TYPES "<< numTris<<'\n';
|
|
|
|
for(uint32 i=0; i<numTris; i++)
|
|
{
|
|
os<< 5 <<'\n';
|
|
}
|
|
|
|
os << "CELL_DATA " << numTris << endl;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool pFlow::TSFtoVTK::convertTimeFolderTriSurfaceFieldsSingle
|
|
(
|
|
multiTriSurface &surface,
|
|
const fileSystem &destPath,
|
|
real time,
|
|
const word &bName,
|
|
wordList& surfNames,
|
|
wordList& fileNames
|
|
)
|
|
{
|
|
vtkFile vtk(destPath, bName, time, false);
|
|
|
|
if (!vtk)
|
|
return false;
|
|
|
|
auto hPoints = surface.points().hostView();
|
|
auto hVertices = surface.vertices().hostView();
|
|
|
|
realx3 const* pData = hPoints.data();
|
|
uint32x3 const* vData = hVertices.data();
|
|
|
|
REPORT(2) << "Wrting surface to "<<
|
|
Green_Text(vtk.fileName()) << END_REPORT;
|
|
|
|
if (! triSurfaceToVTK(
|
|
vtk(),
|
|
pData,
|
|
vData,
|
|
surface.numPoints(),
|
|
surface.size()))
|
|
{
|
|
fatalErrorInFunction <<
|
|
"error in writing triSurface data to vtk file " <<
|
|
vtk.fileName() << endl;
|
|
return false;
|
|
}
|
|
|
|
auto fileList = containingFiles(surface.path().dirPath());
|
|
|
|
for(const auto& file:fileList)
|
|
{
|
|
IOfileHeader fieldHeader(
|
|
objectFile(
|
|
file.fileName(),
|
|
file.dirPath(),
|
|
objectFile::READ_ALWAYS,
|
|
objectFile::WRITE_ALWAYS) );
|
|
|
|
if( fieldHeader.headerOk(true) )
|
|
{
|
|
convertRealx3TypetriSurfaceField(vtk(), fieldHeader, surface);
|
|
}
|
|
}
|
|
|
|
surfNames.push_back(bName);
|
|
fileNames.push_back(vtk.fileName().wordPath());
|
|
|
|
output<<endl;
|
|
return true;
|
|
}
|
|
|
|
bool pFlow::TSFtoVTK::convertTimeFolderTriSurfaceFieldsSeparate(
|
|
multiTriSurface &surface,
|
|
const fileSystem &destPath,
|
|
real time,
|
|
const word &bName,
|
|
wordList& surfNames,
|
|
wordList& fileNames)
|
|
{
|
|
|
|
auto hPoints = surface.points().hostView();
|
|
auto hVertices = surface.vertices().hostView();
|
|
|
|
realx3 const* pData = hPoints.data();
|
|
uint32x3 const* vData = hVertices.data();
|
|
|
|
REPORT(1) << "Wrting triSurface geometry to vtk file . . ."<<
|
|
END_REPORT;
|
|
|
|
wordList sNames, fNames;
|
|
|
|
auto nSurf = surface.numSurfaces();
|
|
for(auto nS=0; nS<nSurf; nS++)
|
|
{
|
|
auto sName = surface.subSurfaceName(nS);
|
|
word fName = groupNames(bName,sName,'_');
|
|
vtkFile vtk(destPath, fName, time, false);
|
|
REPORT(2) << "Wrting sub-surface to "<<
|
|
Green_Text(vtk.fileName())<<END_REPORT;
|
|
|
|
if (!vtk)
|
|
return false;
|
|
|
|
fNames.push_back(vtk.fileName().wordPath());
|
|
sNames.push_back(fName);
|
|
|
|
if (! triSurfaceToVTK(
|
|
vtk(),
|
|
pData,
|
|
vData,
|
|
surface.subSurfaces()[nS]) )
|
|
{
|
|
fatalErrorInFunction <<
|
|
"error in writing triSurface data to vtk file " <<
|
|
vtk.fileName() << endl;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
auto fileList = containingFiles(surface.path().dirPath());
|
|
|
|
for(const auto& file:fileList)
|
|
{
|
|
IOfileHeader fieldHeader(
|
|
objectFile(
|
|
file.fileName(),
|
|
file.dirPath(),
|
|
objectFile::READ_ALWAYS,
|
|
objectFile::WRITE_ALWAYS) );
|
|
|
|
if( fieldHeader.headerOk(true) )
|
|
{
|
|
convertRealx3TypetriSurfaceFieldSeparate(
|
|
destPath,
|
|
fieldHeader,
|
|
surface,
|
|
bName,
|
|
time);
|
|
}
|
|
|
|
}
|
|
|
|
fileNames = fNames;
|
|
surfNames = sNames;
|
|
output<<endl;
|
|
return true;
|
|
}
|
|
|
|
|
|
bool pFlow::TSFtoVTK::convertRealx3TypetriSurfaceField(
|
|
Ostream& os,
|
|
const IOfileHeader& header,
|
|
multiTriSurface& tSurface)
|
|
{
|
|
word objectType = header.objectType();
|
|
|
|
if(!checkTriFieldType<realx3>(objectType))return false;
|
|
|
|
auto field = realx3TriSurfaceField_H
|
|
(
|
|
header,
|
|
tSurface,
|
|
static_cast<real>(0)
|
|
);
|
|
|
|
const realx3* data = field.deviceViewAll().data();
|
|
|
|
REPORT(1)<<"writing "<< greenColor <<header.objectName()<<
|
|
defaultColor<<" field to vtk."<<END_REPORT;
|
|
|
|
return pFlow::PFtoVTK::addRealx3PointField
|
|
(
|
|
os,
|
|
header.objectName(),
|
|
data,
|
|
field.size()
|
|
);
|
|
}
|
|
|
|
|
|
bool pFlow::TSFtoVTK::convertRealx3TypetriSurfaceFieldSeparate
|
|
(
|
|
const fileSystem& destPath,
|
|
const IOfileHeader& header,
|
|
multiTriSurface& tSurface,
|
|
const word& bName,
|
|
real time
|
|
)
|
|
{
|
|
word objectType = header.objectType();
|
|
|
|
if(!checkTriFieldType<realx3>(objectType))return false;
|
|
|
|
auto field = realx3TriSurfaceField_H
|
|
(
|
|
header,
|
|
tSurface,
|
|
static_cast<real>(0)
|
|
);
|
|
|
|
const realx3* data = field.deviceViewAll().data();
|
|
|
|
/*REPORT(1)<<"writing "<< greenColor <<header.objectName()<<
|
|
defaultColor<<" field to vtk."<<END_REPORT;*/
|
|
|
|
auto nSurf = tSurface.numSurfaces();
|
|
|
|
for(auto nS=0; nS<nSurf; nS++)
|
|
{
|
|
auto& subSurf = tSurface.subSurfaces()[nS];
|
|
|
|
auto sName = subSurf.name();
|
|
vtkFile vtk(destPath, groupNames(bName,sName,'_'), time, false, true);
|
|
REPORT(2) << "Wrting sub-surface to "<<
|
|
Green_Text(vtk.fileName())<<END_REPORT;
|
|
|
|
pFlow::PFtoVTK::addRealx3PointField
|
|
(
|
|
vtk(),
|
|
header.objectName(),
|
|
data+subSurf.start(),
|
|
subSurf.size()
|
|
);
|
|
}
|
|
|
|
return true;
|
|
/*return pFlow::PFtoVTK::addRealx3PointField
|
|
(
|
|
os,
|
|
header.objectName(),
|
|
data,
|
|
field.size()
|
|
);*/
|
|
} |