songsenand
/
cgal
mirror of https://github.com/CGAL/cgal
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use boost variant for attributes to be able to send `double`s, `uint_8`s and `std::size_t`s

This commit is contained in:
Maxime Gimeno 2019-03-18 13:14:00 +01:00
parent 713c672326
commit c787e40ca7
2 changed files with 39 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
Mesh_3/include/CGAL/IO/output_to_vtu.h
View File

@ -258,12 +258,20 @@ write_attributes(std::ostream& os,
write_vector(os,att); write_vector(os,att);
} }
enum VTU_ATTRIBUTE_TYPE{
DOUBLE=0,
UNIT_8,
SIZE_TYPE
};
template <class C3T3> template <class C3T3>
void output_to_vtu_with_attributes(std::ostream& os, void output_to_vtu_with_attributes(std::ostream& os,
const C3T3& c3t3, const C3T3& c3t3,
std::vector<std::pair<const char*, const std::vector<double>*> >& attributes, std::vector<std::pair<const char*, const boost::variant<std::vector<double>, std::vector<uint8_t>, std::vector<std::size_t> >*> >& attributes,
// std::vector<CGAL::VTU_ATTRIBUTE_TYPE>& attribute_types,
IO::Mode mode = IO::BINARY) IO::Mode mode = IO::BINARY)
{ {
//CGAL_assertion(attributes.size() == attribute_types.size());
typedef typename C3T3::Triangulation Tr; typedef typename C3T3::Triangulation Tr;
typedef typename Tr::Vertex_handle Vertex_handle; typedef typename Tr::Vertex_handle Vertex_handle;
const Tr& tr = c3t3.triangulation(); const Tr& tr = c3t3.triangulation();
@ -296,7 +304,17 @@ void output_to_vtu_with_attributes(std::ostream& os,
os << " <CellData Scalars=\""<<attributes.front().first<<"\">\n"; os << " <CellData Scalars=\""<<attributes.front().first<<"\">\n";
for(std::size_t i = 0; i< attributes.size(); ++i) for(std::size_t i = 0; i< attributes.size(); ++i)
{ {
write_attribute_tag(os,attributes[i].first, *attributes[i].second, binary,offset); switch(attributes[i].second->which()){
case 0:
write_attribute_tag(os,attributes[i].first, *boost::get<std::vector<double> >(attributes[i].second), binary,offset);
break;
case 1:
write_attribute_tag(os,attributes[i].first, *boost::get<std::vector<uint8_t> >(attributes[i].second), binary,offset);
break;
default:
write_attribute_tag(os,attributes[i].first, *boost::get<std::vector<std::size_t> >(attributes[i].second), binary,offset);
break;
}
} }
os << " </CellData>\n"; os << " </CellData>\n";
os << " </Piece>\n" os << " </Piece>\n"
@ -306,7 +324,17 @@ void output_to_vtu_with_attributes(std::ostream& os,
write_c3t3_points(os,tr,V); // fills V if the mode is BINARY write_c3t3_points(os,tr,V); // fills V if the mode is BINARY
write_cells(os,c3t3,V); write_cells(os,c3t3,V);
for(std::size_t i = 0; i< attributes.size(); ++i) for(std::size_t i = 0; i< attributes.size(); ++i)
write_attributes(os, *attributes[i].second); switch(attributes[i].second->which()){
case 0:
write_attributes(os, *boost::get<std::vector<double> >(attributes[i].second));
break;
case 1:
write_attributes(os, *boost::get<std::vector<uint8_t> >(attributes[i].second));
break;
default:
write_attributes(os, *boost::get<std::vector<std::size_t> >(attributes[i].second));
break;
}
} }
os << "</VTKFile>\n"; os << "</VTKFile>\n";
} }
@ -325,10 +353,12 @@ void output_to_vtu(std::ostream& os,
cit != c3t3.cells_in_complex_end() ; cit != c3t3.cells_in_complex_end() ;
++cit ) ++cit )
{ {
mids.push_back(cit->subdomain_index()); double v = cit->subdomain_index();
mids.push_back(v);
} }
std::vector<std::pair<const char*, const std::vector<double>* > > atts; std::vector<std::pair<const char*, const boost::variant<std::vector<double>, std::vector<uint8_t>, std::vector<std::size_t> >* > > atts;
atts.push_back(std::make_pair("MeshDomain", &mids)); boost::variant<std::vector<double>, std::vector<uint8_t>, std::vector<std::size_t> > v = mids;
atts.push_back(std::make_pair("MeshDomain", &v));
output_to_vtu_with_attributes(os, c3t3, atts, mode); output_to_vtu_with_attributes(os, c3t3, atts, mode);
} }