mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-16 21:42:19 +00:00
Backspace everything regarding HDF5
This commit is contained in:
Thomas Krijnen
@@ -64,11 +64,6 @@
|
||||
#include <IntTools_FaceFace.hxx>
|
||||
#include "clash_utils.h"
|
||||
|
||||
#ifdef WITH_HDF5
|
||||
#include "H5Cpp.h"
|
||||
#endif
|
||||
|
||||
|
||||
namespace IfcGeom {
|
||||
namespace {
|
||||
|
||||
@@ -1498,225 +1493,6 @@ namespace IfcGeom {
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef WITH_HDF5
|
||||
void write_h5() {
|
||||
H5::H5File file("filename.h5", H5F_ACC_TRUNC);
|
||||
H5::Group shapes = file.createGroup("/shapes");
|
||||
|
||||
std::set<std::string> processed_geometry_ids;
|
||||
std::vector<int> element_shape_ids;
|
||||
std::unordered_map<std::string, int> geometry_id_to_shape_id;
|
||||
int geometry_index = 0;
|
||||
|
||||
std::vector<std::vector<float>> matrices;
|
||||
std::vector<std::array<float, 4>> colours;
|
||||
std::vector<std::string> names;
|
||||
std::vector<std::string> global_ids;
|
||||
|
||||
const float tolerance = 0.01f; // Tolerance value for comparison
|
||||
|
||||
for (const auto& elem : triangulation_elements_) {
|
||||
const auto geometry_id = elem->geometry().id();
|
||||
|
||||
const auto& placement = placements_[elem->product()];
|
||||
{
|
||||
decltype(matrices)::value_type m;
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
m.push_back(static_cast<float>(placement->ccomponents().data()[i]));
|
||||
}
|
||||
matrices.push_back(m);
|
||||
}
|
||||
|
||||
names.push_back(names_[elem->product()]);
|
||||
global_ids.push_back(global_ids_[elem->product()]);
|
||||
|
||||
if (processed_geometry_ids.find(geometry_id) != processed_geometry_ids.end()) {
|
||||
element_shape_ids.push_back(geometry_id_to_shape_id[geometry_id]);
|
||||
continue;
|
||||
}
|
||||
|
||||
processed_geometry_ids.insert(geometry_id);
|
||||
H5::Group group = shapes.createGroup(std::to_string(geometry_index));
|
||||
geometry_id_to_shape_id[geometry_id] = geometry_index;
|
||||
element_shape_ids.push_back(geometry_index);
|
||||
|
||||
geometry_index++;
|
||||
|
||||
const auto& faces = local_faces_[geometry_id];
|
||||
const auto& verts = local_verts_[geometry_id];
|
||||
const auto& materials = local_materials_[geometry_id];
|
||||
const auto& material_ids = local_material_ids_[geometry_id];
|
||||
|
||||
std::vector<float> verts_float(verts.size());
|
||||
std::transform(verts.begin(), verts.end(), verts_float.begin(),
|
||||
[](double val) { return static_cast<float>(val); });
|
||||
|
||||
// Write faces
|
||||
size_t total_verts = verts.size() / 3;
|
||||
hsize_t faces_dims[1] = {faces.size()};
|
||||
H5::DataSpace faces_dataspace(1, faces_dims);
|
||||
H5::DSetCreatPropList faces_propList;
|
||||
faces_propList.setChunk(1, faces_dims);
|
||||
faces_propList.setDeflate(9);
|
||||
if (total_verts < (1 << 8)) {
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT8;
|
||||
std::vector<uint8_t> faces_dtype(faces.begin(), faces.end());
|
||||
H5::DataSet faces_dataset = group.createDataSet("faces", dtype, faces_dataspace, faces_propList);
|
||||
faces_dataset.write(faces_dtype.data(), dtype);
|
||||
} else if (total_verts < (1 << 16)) {
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT16;
|
||||
std::vector<uint16_t> faces_dtype(faces.begin(), faces.end());
|
||||
H5::DataSet faces_dataset = group.createDataSet("faces", dtype, faces_dataspace, faces_propList);
|
||||
faces_dataset.write(faces_dtype.data(), dtype);
|
||||
} else {
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT32;
|
||||
H5::DataSet faces_dataset = group.createDataSet("faces", dtype, faces_dataspace, faces_propList);
|
||||
faces_dataset.write(faces.data(), dtype);
|
||||
}
|
||||
|
||||
// Write verts
|
||||
H5::DataType dtype = H5::PredType::NATIVE_FLOAT;
|
||||
hsize_t dims[1] = {verts.size()};
|
||||
H5::DataSpace dataspace(1, dims);
|
||||
H5::DSetCreatPropList propList;
|
||||
propList.setChunk(1, dims);
|
||||
propList.setDeflate(9);
|
||||
H5::DataSet dataset = group.createDataSet("verts", dtype, dataspace, propList);
|
||||
dataset.write(verts_float.data(), H5::PredType::NATIVE_FLOAT);
|
||||
|
||||
// Write materials
|
||||
std::vector<uint8_t> material_keys;
|
||||
for (const auto& material : materials) {
|
||||
float alpha = 1.0;
|
||||
if (material->has_transparency() && material->transparency > 0) {
|
||||
alpha = 1.0 - material->transparency;
|
||||
}
|
||||
|
||||
int i = 0;
|
||||
bool is_existing_colour = false;
|
||||
for (const auto& colour : colours) {
|
||||
if (std::abs(colour[0] - static_cast<float>(material->diffuse.ccomponents()[0])) < tolerance
|
||||
&& std::abs(colour[1] - static_cast<float>(material->diffuse.ccomponents()[1])) < tolerance
|
||||
&& std::abs(colour[2] - static_cast<float>(material->diffuse.ccomponents()[2])) < tolerance
|
||||
&& std::abs(colour[3] - alpha) < tolerance) {
|
||||
is_existing_colour = true;
|
||||
break;
|
||||
}
|
||||
i++;
|
||||
}
|
||||
|
||||
if ( ! is_existing_colour) {
|
||||
colours.push_back({
|
||||
static_cast<float>(material->diffuse.ccomponents()[0]),
|
||||
static_cast<float>(material->diffuse.ccomponents()[1]),
|
||||
static_cast<float>(material->diffuse.ccomponents()[2]),
|
||||
alpha});
|
||||
}
|
||||
material_keys.push_back(static_cast<decltype(material_keys)::value_type>(i));
|
||||
}
|
||||
|
||||
size_t total_material_keys = material_keys.size();
|
||||
if (total_material_keys) {
|
||||
hsize_t dims[1] = {material_keys.size()};
|
||||
H5::DataSpace dataspace(1, dims);
|
||||
H5::DSetCreatPropList propList;
|
||||
propList.setChunk(1, dims);
|
||||
propList.setDeflate(9);
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT8;
|
||||
H5::DataSet dataset = group.createDataSet("materials", dtype, dataspace, propList);
|
||||
dataset.write(material_keys.data(), dtype);
|
||||
}
|
||||
|
||||
if (total_material_keys > 1) {
|
||||
hsize_t dims[1] = {material_ids.size()};
|
||||
H5::DataSpace dataspace(1, dims);
|
||||
H5::DSetCreatPropList propList;
|
||||
propList.setChunk(1, dims);
|
||||
propList.setDeflate(9);
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT8;
|
||||
H5::DataSet dataset = group.createDataSet("material_ids", dtype, dataspace, propList);
|
||||
std::vector<uint8_t> data_dtype(material_ids.begin(), material_ids.end());
|
||||
dataset.write(data_dtype.data(), dtype);
|
||||
}
|
||||
}
|
||||
|
||||
// Write GlobalIds
|
||||
std::vector<uint8_t> uuids_array;
|
||||
for (const auto& id_str : global_ids) {
|
||||
for (size_t i = 0; i < id_str.length(); i += 2) {
|
||||
uuids_array.push_back(std::stoi(id_str.substr(i, 2), 0, 16));
|
||||
}
|
||||
}
|
||||
|
||||
hsize_t global_ids_dims[2] = {global_ids.size(), 16}; // 16 bytes per UUID
|
||||
H5::DataSpace global_ids_dataspace(2, global_ids_dims);
|
||||
H5::DataSet global_ids_dataset = file.createDataSet("element_global_ids", H5::PredType::NATIVE_UINT8, global_ids_dataspace);
|
||||
global_ids_dataset.write(uuids_array.data(), H5::PredType::NATIVE_UINT8);
|
||||
|
||||
// Write names
|
||||
H5::StrType strType(H5::PredType::C_S1, H5T_VARIABLE);
|
||||
hsize_t names_dims[1] = {names.size()};
|
||||
H5::DataSpace names_dataspace(1, names_dims);
|
||||
H5::DataSet names_dataset = file.createDataSet("element_names", strType, names_dataspace);
|
||||
std::vector<const char*> cstr_names;
|
||||
for (const auto& name : names) {
|
||||
cstr_names.push_back(name.c_str());
|
||||
}
|
||||
names_dataset.write(&cstr_names[0], strType);
|
||||
|
||||
// Write matrices
|
||||
std::vector<float> flat_matrices;
|
||||
for (const auto& matrix : matrices) {
|
||||
flat_matrices.insert(flat_matrices.end(), matrix.begin(), matrix.end());
|
||||
}
|
||||
hsize_t dims[2] = {matrices.size(), matrices[0].size()};
|
||||
H5::DataSpace dataspace(2, dims);
|
||||
H5::DSetCreatPropList propList;
|
||||
propList.setChunk(2, dims);
|
||||
propList.setDeflate(9);
|
||||
H5::DataSet dataset = file.createDataSet("element_matrices", H5::PredType::NATIVE_FLOAT, dataspace, propList);
|
||||
dataset.write(flat_matrices.data(), H5::PredType::NATIVE_FLOAT);
|
||||
|
||||
// Write element_shape_ids
|
||||
hsize_t element_shape_ids_dims[1] = {element_shape_ids.size()};
|
||||
H5::DataSpace element_shape_ids_dataspace(1, element_shape_ids_dims);
|
||||
H5::DSetCreatPropList element_shape_ids_propList;
|
||||
element_shape_ids_propList.setChunk(1, element_shape_ids_dims);
|
||||
element_shape_ids_propList.setDeflate(9);
|
||||
if (geometry_index < (1 << 8)) {
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT8;
|
||||
std::vector<uint8_t> element_shape_ids_dtype(element_shape_ids.begin(), element_shape_ids.end());
|
||||
H5::DataSet element_shape_ids_dataset = file.createDataSet("element_shape_ids", dtype, element_shape_ids_dataspace, element_shape_ids_propList);
|
||||
element_shape_ids_dataset.write(element_shape_ids_dtype.data(), dtype);
|
||||
} else if (geometry_index < (1 << 16)) {
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT16;
|
||||
std::vector<uint16_t> element_shape_ids_dtype(element_shape_ids.begin(), element_shape_ids.end());
|
||||
H5::DataSet element_shape_ids_dataset = file.createDataSet("element_shape_ids", dtype, element_shape_ids_dataspace, element_shape_ids_propList);
|
||||
element_shape_ids_dataset.write(element_shape_ids_dtype.data(), dtype);
|
||||
} else if (geometry_index < (1UL << 32)) {
|
||||
H5::DataType dtype = H5::PredType::NATIVE_UINT32;
|
||||
std::vector<uint32_t> element_shape_ids_dtype(element_shape_ids.begin(), element_shape_ids.end());
|
||||
H5::DataSet element_shape_ids_dataset = file.createDataSet("element_shape_ids", dtype, element_shape_ids_dataspace, element_shape_ids_propList);
|
||||
element_shape_ids_dataset.write(element_shape_ids_dtype.data(), dtype);
|
||||
}
|
||||
|
||||
// Write colours
|
||||
if (colours.size()) {
|
||||
std::vector<float> flat_colours;
|
||||
for (const auto& colour : colours) {
|
||||
flat_colours.insert(flat_colours.end(), colour.begin(), colour.end());
|
||||
}
|
||||
hsize_t colours_dims[2] = {colours.size(), colours[0].size()};
|
||||
H5::DataSpace colours_dataspace(2, colours_dims);
|
||||
H5::DSetCreatPropList colours_propList;
|
||||
colours_propList.setChunk(2, colours_dims);
|
||||
colours_propList.setDeflate(9);
|
||||
H5::DataSet colours_dataset = file.createDataSet("materials", H5::PredType::NATIVE_FLOAT, colours_dataspace, colours_propList);
|
||||
colours_dataset.write(flat_colours.data(), H5::PredType::NATIVE_FLOAT);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
template <typename T>
|
||||
void apply_matrix_to_flat_verts(const std::vector<T>& flat_list, const ifcopenshell::geometry::taxonomy::matrix4::ptr& matrix, std::vector<T>& result) {
|
||||
Eigen::Vector3d vin;
|
||||
|
||||
@@ -164,12 +164,6 @@ namespace ifcopenshell {
|
||||
return tree_.clash_clearance_many(set_a, set_b, clearance, check_all);
|
||||
}
|
||||
|
||||
#ifdef WITH_HDF5
|
||||
void write_h5() override {
|
||||
tree_.write_h5();
|
||||
}
|
||||
#endif
|
||||
|
||||
private:
|
||||
IfcGeom::opencascade_tree tree_;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user