[英]Write vtk file in binary format
I am trying to write a vtk file from my C++ code. 我试图从我的C ++代码编写一个vtk文件。 I have two versions, one ASCII and the other binary. 我有两个版本,一个是ASCII,另一个是二进制版本。 The ASCII version works well. ASCII版本运行良好。
I want to do something similar to this post : Error writing binary VTK files 我想做一些类似于这篇文章: 写二进制VTK文件时出错
Here is my code : 这是我的代码:
std::ofstream file;
if(is_binary)
file.open("test.vtk", std::ios::out | std::ios::binary);
else
file.open("test.vtk");
file << "# vtk DataFile Version 2.0" << std::endl
<< "Comment if needed" << std::endl;
if(is_binary)
file << "BINARY"<< std::endl << std::endl;
else
file << "ASCII"<< std::endl << std::endl;
file << "DATASET POLYDATA" << std::endl << "POINTS " << nb_particles << " float" << std::endl;
for(size_t cell_i=0;cell_i<n_cells;cell_i++)
for(size_t i =0; i<cells[cell_i].size();++i)
{
if(is_binary)
{
double rx = cells[cell_i][field::rx][i];
double ry = cells[cell_i][field::ry][i];
double rz = cells[cell_i][field::rz][i];
SwapEnd(rx);
SwapEnd(ry);
SwapEnd(rz);
file.write(reinterpret_cast<char*>(&rx), sizeof(double));
file.write(reinterpret_cast<char*>(&ry), sizeof(double));
file.write(reinterpret_cast<char*>(&rz), sizeof(double));
}
else
{
Vec3d tmp = grid.particle_position(cell_i,i);
file << tmp.x << " " << tmp.y << " " << tmp.z << std::endl;
}
}
if(is_binary)
file << std::endl;
file << "POINT_DATA " << nb_particles << std::endl;
if(has_id_field)
{
file<< "SCALARS index int 1" << std::endl<< "LOOKUP_TABLE default" << std::endl;
for(size_t cell_i=0;cell_i<n_cells;cell_i++)
for(size_t i =0; i<cells[cell_i].size();++i)
{
if(is_binary)
{
uint64_t id = cells[cell_i][field::id][i];
SwapEnd(id);
file.write(reinterpret_cast<char*>(&id), sizeof(uint64_t));
}
else
file << cells[cell_i][field::id][i] << std::endl;
}
if(is_binary)
file << std::endl;
}
if(has_type_field)
{
file<< "SCALARS type int 1" << std::endl<< "LOOKUP_TABLE default" << std::endl;
for(size_t cell_i=0;cell_i<n_cells;cell_i++)
for(size_t i =0; i<cells[cell_i].size();++i)
{
if(is_binary)
{
uint8_t type;
SwapEnd(type);
file.write(reinterpret_cast<char*>(&type), sizeof(uint8_t));
}
else
file << static_cast<int>(cells[cell_i][field::type][i]) << std::endl;
}
if(is_binary)
file << std::endl;
}
file.close();
with the function : 功能:
void SwapEnd(T& var)
{
char* varArray = reinterpret_cast<char*>(&var);
for(long i = 0; i < static_cast<long>(sizeof(var)/2); i++)
std::swap(varArray[sizeof(var) - 1 - i],varArray[i]);
}
My system is little endian : 我的系统是小端:
lscpu | grep "Byte Order"
Byte Order: Little Endian
paraview make an error on the reading the output file : paraview在读取输出文件时出错:
vtkPolyDataReader (0x3863800): Unrecognized keyword: @aic�9h��8
What am I not doing well ? 我做得不好?
ok, the problem was the size of datas when I wrote the binary. 好吧,问题是我写二进制文件时数据的大小。 This following peace of code works. 以下代码的和平工作。 Hope that can help someone. 希望可以帮助别人。
std::ofstream file;
if(is_binary)
file.open("test.vtk", std::ios::out | std::ios::binary);
else
file.open("test.vtk");
file << "# vtk DataFile Version 2.0" << std::endl
<< "Comment if needed" << std::endl;
if(is_binary)
file << "BINARY"<< std::endl << std::endl;
else
file << "ASCII"<< std::endl << std::endl;
file << "DATASET POLYDATA" << std::endl << "POINTS " << nb_particles << " double" << std::endl;
for(size_t cell_i=0;cell_i<n_cells;cell_i++)
for(size_t i =0; i<cells[cell_i].size();++i)
{
if(is_binary)
{
double rx = cells[cell_i][field::rx][i];
double ry = cells[cell_i][field::ry][i];
double rz = cells[cell_i][field::rz][i];
SwapEnd(rx);
SwapEnd(ry);
SwapEnd(rz);
file.write(reinterpret_cast<char*>(&rx), sizeof(double));
file.write(reinterpret_cast<char*>(&ry), sizeof(double));
file.write(reinterpret_cast<char*>(&rz), sizeof(double));
}
else
{
Vec3d tmp = grid.particle_position(cell_i,i);
file << tmp.x << " " << tmp.y << " " << tmp.z << std::endl;
}
}
if(is_binary)
file << std::endl;
file << "POINT_DATA " << nb_particles << std::endl;
if(has_id_field)
{
file<< "SCALARS index int 1" << std::endl<< "LOOKUP_TABLE default" << std::endl;
for(size_t cell_i=0;cell_i<n_cells;cell_i++)
for(size_t i =0; i<cells[cell_i].size();++i)
{
if(is_binary)
{
uint64_t id = cells[cell_i][field::id][i];
int id_i = static_cast<int>(id);
SwapEnd(id_i);
file.write(reinterpret_cast<char*>(&id_i), sizeof(int));
}
else
file << cells[cell_i][field::id][i] << std::endl;
}
if(is_binary)
file << std::endl;
}
if(has_type_field)
{
file<< "SCALARS type int 1" << std::endl<< "LOOKUP_TABLE default" << std::endl;
for(size_t cell_i=0;cell_i<n_cells;cell_i++)
for(size_t i =0; i<cells[cell_i].size();++i)
{
if(is_binary)
{
uint8_t type= cells[cell_i][field::type][i];
int type_i = static_cast<int>(type);
SwapEnd(type_i);
file.write(reinterpret_cast<char*>(&type_i), sizeof(int));
}
else
file << static_cast<int>(cells[cell_i][field::type][i]) << std::endl;
}
if(is_binary)
file << std::endl;
}
file.close();
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.