Files
IfcOpenShell/src/ifcparse/hierarchy_helper.h
T
2026-07-09 22:01:21 +02:00

299 lines
17 KiB
C++

/********************************************************************************
* *
* This file is part of IfcOpenShell. *
* *
* IfcOpenShell is free software: you can redistribute it and/or modify *
* it under the terms of the Lesser GNU General Public License as published by *
* the Free Software Foundation, either version 3.0 of the License, or *
* (at your option) any later version. *
* *
* IfcOpenShell is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* Lesser GNU General Public License for more details. *
* *
* You should have received a copy of the Lesser GNU General Public License *
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
* *
********************************************************************************/
/********************************************************************************
* *
* This class is a subclass of the regular file class that implements *
* several convenience functions for creating geometrical representations and *
* spatial containers. *
* *
********************************************************************************/
#ifndef IFCHIERARCHYHELPER_H
#define IFCHIERARCHYHELPER_H
#include "ifc_parse_api.h"
#include "logger.h"
#include <map>
#include <type_traits>
#include <vector>
#include "file.h"
#include "global_id.h"
namespace ifcopenshell::hierarchy_detail {
inline const char* parent_attribute_name(const ifcopenshell::declaration& declaration) {
return declaration.name() == "IfcRelContainedInSpatialStructure" ? "RelatingStructure" : "RelatingObject";
}
inline const char* children_attribute_name(const ifcopenshell::declaration& declaration) {
return declaration.name() == "IfcRelContainedInSpatialStructure" ? "RelatedElements" : "RelatedObjects";
}
template <typename Relation>
express::Base get_parent_of_relation(const Relation& relation) {
return relation.template as<express::Entity>().get(parent_attribute_name(relation.declaration()));
}
template <typename Relation>
std::vector<express::Base> get_children_of_relation(const Relation& relation) {
return relation.template as<express::Entity>().get(children_attribute_name(relation.declaration()));
}
template <typename Relation>
void set_children_of_relation(Relation& relation, std::vector<express::Base>& children) {
relation.set_attribute_value(children_attribute_name(relation.declaration()), children);
}
template <typename Schema, typename = void>
struct styled_item_accepts_presentation_style : std::false_type {};
template <typename Schema>
struct styled_item_accepts_presentation_style<Schema,
std::void_t<decltype(std::declval<typename Schema::IfcStyledItem&>().setStyles(
std::declval<std::vector<typename Schema::IfcPresentationStyle>>()))>> : std::true_type {};
template <typename Schema, bool DirectStyle = styled_item_accepts_presentation_style<Schema>::value>
struct surface_style_type_selector {
using type = typename Schema::IfcPresentationStyleAssignment;
};
template <typename Schema>
struct surface_style_type_selector<Schema, true> {
using type = typename Schema::IfcPresentationStyle;
};
template <typename Schema>
using surface_style_type = typename surface_style_type_selector<Schema>::type;
} // namespace ifcopenshell::hierarchy_detail
template <typename Schema>
class IFC_SCHEMA_API hierarchy_helper : public ifcopenshell::file {
public:
hierarchy_helper() : ifcopenshell::file(&Schema::get_schema()) {}
template <class T>
T addTriplet(double x, double y, double z) {
auto t = create<T>();
t.set_attribute_value(0, std::vector<double>{x, y, z});
return t;
}
template <class T>
T addDoublet(double x, double y) {
auto t = create<T>();
t.set_attribute_value(0, std::vector<double>{x, y});
return t;
}
template <typename T, typename U>
T addValue(U value) {
auto measure = create<T>();
measure.set_attribute_value(0, value);
return measure;
}
template <class T>
T getSingle() {
auto ts = instances_by_type<T>();
if (ts.size() != 1) {
return T{};
}
return ts.front();
}
typename Schema::IfcAxis2Placement3D addPlacement3d(double origin_x = 0.0, double origin_y = 0.0, double origin_z = 0.0, double z_axis_x = 0.0, double z_axis_y = 0.0, double z_axis_z = 1.0, double x_axis_x = 1.0, double x_axis_y = 0.0, double x_axis_z = 0.0);
typename Schema::IfcAxis2Placement2D addPlacement2d(double origin_x = 0.0, double origin_y = 0.0, double x_axis_x = 1.0, double x_axis_y = 0.0);
typename Schema::IfcLocalPlacement addLocalPlacement(typename Schema::IfcObjectPlacement parent_placement = typename Schema::IfcObjectPlacement{},
double origin_x = 0.0,
double origin_y = 0.0,
double origin_z = 0.0,
double z_axis_x = 0.0,
double z_axis_y = 0.0,
double z_axis_z = 1.0,
double x_axis_x = 1.0,
double x_axis_y = 0.0,
double x_axis_z = 0.0);
template <class T>
void addRelatedObject(const typename Schema::IfcObjectDefinition& relating_object,
const typename Schema::IfcObjectDefinition& related_object,
typename Schema::IfcOwnerHistory owner_history = typename Schema::IfcOwnerHistory{})
{
if constexpr (std::is_same_v<T, typename Schema::IfcRelDefinesByType>) {
auto li = instances_by_type<typename Schema::IfcRelDefinesByType>();
bool found = false;
for (auto & rel : li) {
if (rel.RelatingType() == relating_object) {
auto objects = rel.RelatedObjects();
objects.push_back(add_entity(related_object).template as<typename Schema::IfcObject>());
rel.setRelatedObjects(objects);
found = true;
break;
}
}
if (!found) {
if (!owner_history) {
owner_history = getSingle<typename Schema::IfcOwnerHistory>();
}
if (!owner_history) {
owner_history = addOwnerHistory();
}
std::vector<typename Schema::IfcObject> related_objects = {related_object.template as<typename Schema::IfcObject>()};
auto t = create<typename Schema::IfcRelDefinesByType>();
t.setGlobalId(ifcopenshell::global_id());
t.setOwnerHistory(owner_history);
t.setRelatedObjects(related_objects);
t.setRelatingType(relating_object.template as<typename Schema::IfcTypeObject>());
}
} else {
auto li = instances_by_type<T>();
bool found = false;
for (auto& rel : li) {
try {
if (ifcopenshell::hierarchy_detail::get_parent_of_relation(rel) == relating_object) {
auto products = ifcopenshell::hierarchy_detail::get_children_of_relation(rel);
products.push_back(add_entity(related_object));
ifcopenshell::hierarchy_detail::set_children_of_relation(rel, products);
found = true;
break;
}
} catch (std::exception& e) {
::logger::root().error(e);
} catch (...) {
::logger::root().error("Unknown error in addRelatedObject()");
}
}
if (!found) {
if (!owner_history) {
owner_history = getSingle<typename Schema::IfcOwnerHistory>();
}
if (!owner_history) {
owner_history = addOwnerHistory();
}
std::vector<express::Base> related_objects;
related_objects.push_back(related_object);
T t = create<T>();
t.set_attribute_value(0, (std::string)ifcopenshell::global_id());
t.set_attribute_value(1, owner_history);
int relating_index = 4;
int related_index = 5;
if (T::Class().name() == "IfcRelContainedInSpatialStructure" || T::Class().name() == "IfcRelReferencedInSpatialStructure" || std::is_base_of<typename Schema::IfcRelDefines, T>::value) {
// some classes have attributes reversed.
std::swap(relating_index, related_index);
}
t.set_attribute_value(relating_index, relating_object);
t.set_attribute_value(related_index, related_objects);
}
}
}
typename Schema::IfcOwnerHistory addOwnerHistory();
typename Schema::IfcProject addProject(typename Schema::IfcOwnerHistory owner_history = typename Schema::IfcOwnerHistory{});
void relatePlacements(typename Schema::IfcProduct parent, typename Schema::IfcProduct product);
typename Schema::IfcSite addSite(typename Schema::IfcProject project = typename Schema::IfcProject{}, typename Schema::IfcOwnerHistory owner_history = typename Schema::IfcOwnerHistory{});
typename Schema::IfcBuilding addBuilding(typename Schema::IfcSite site = typename Schema::IfcSite{}, typename Schema::IfcOwnerHistory owner_history = typename Schema::IfcOwnerHistory{});
typename Schema::IfcBuildingStorey addBuildingStorey(typename Schema::IfcBuilding building = typename Schema::IfcBuilding{},
typename Schema::IfcOwnerHistory owner_history = typename Schema::IfcOwnerHistory{});
typename Schema::IfcBuildingStorey addBuildingProduct(typename Schema::IfcProduct product,
typename Schema::IfcBuildingStorey storey = typename Schema::IfcBuildingStorey{},
typename Schema::IfcOwnerHistory owner_history = typename Schema::IfcOwnerHistory{});
void addExtrudedPolyline(typename Schema::IfcShapeRepresentation representation, const std::vector<std::pair<double, double>>& points, double height, typename Schema::IfcAxis2Placement2D profile_placement = typename Schema::IfcAxis2Placement2D{}, typename Schema::IfcAxis2Placement3D extrusion_placement = typename Schema::IfcAxis2Placement3D{}, typename Schema::IfcDirection extrusion_direction = typename Schema::IfcDirection{}, typename Schema::IfcRepresentationContext context = typename Schema::IfcRepresentationContext{});
typename Schema::IfcProductDefinitionShape addExtrudedPolyline(const std::vector<std::pair<double, double>>& points, double height, typename Schema::IfcAxis2Placement2D profile_placement = typename Schema::IfcAxis2Placement2D{}, typename Schema::IfcAxis2Placement3D extrusion_placement = typename Schema::IfcAxis2Placement3D{}, typename Schema::IfcDirection extrusion_direction = typename Schema::IfcDirection{}, typename Schema::IfcRepresentationContext context = typename Schema::IfcRepresentationContext{});
void addBox(typename Schema::IfcShapeRepresentation representation, double width, double depth, double height, typename Schema::IfcAxis2Placement2D profile_placement = typename Schema::IfcAxis2Placement2D{}, typename Schema::IfcAxis2Placement3D extrusion_placement = typename Schema::IfcAxis2Placement3D{}, typename Schema::IfcDirection extrusion_direction = typename Schema::IfcDirection{}, typename Schema::IfcRepresentationContext context = typename Schema::IfcRepresentationContext{});
typename Schema::IfcProductDefinitionShape addBox(double width, double depth, double height, typename Schema::IfcAxis2Placement2D profile_placement = typename Schema::IfcAxis2Placement2D{}, typename Schema::IfcAxis2Placement3D extrusion_placement = typename Schema::IfcAxis2Placement3D{}, typename Schema::IfcDirection extrusion_direction = typename Schema::IfcDirection{}, typename Schema::IfcRepresentationContext context = typename Schema::IfcRepresentationContext{});
void addAxis(typename Schema::IfcShapeRepresentation representation, double length, typename Schema::IfcRepresentationContext context = typename Schema::IfcRepresentationContext{});
typename Schema::IfcProductDefinitionShape addAxisBox(double width, double depth, double height, typename Schema::IfcRepresentationContext context = typename Schema::IfcRepresentationContext{});
void clipRepresentation(typename Schema::IfcProductRepresentation representation,
typename Schema::IfcAxis2Placement3D placement,
bool sense_agreement);
void clipRepresentation(typename Schema::IfcRepresentation representation,
typename Schema::IfcAxis2Placement3D placement,
bool sense_agreement);
typename Schema::IfcProductDefinitionShape addMappedItem(typename Schema::IfcShapeRepresentation source_representation,
typename Schema::IfcCartesianTransformationOperator3D transform = typename Schema::IfcCartesianTransformationOperator3D{},
typename Schema::IfcProductDefinitionShape definition_shape = typename Schema::IfcProductDefinitionShape{});
typename Schema::IfcProductDefinitionShape addMappedItem(std::vector<typename Schema::IfcShapeRepresentation>& source_representations,
typename Schema::IfcCartesianTransformationOperator3D transform = typename Schema::IfcCartesianTransformationOperator3D{});
typename Schema::IfcShapeRepresentation addEmptyRepresentation(const std::string& representation_identifier = "Body", const std::string& representation_type = "SweptSolid");
typename Schema::IfcGeometricRepresentationContext getRepresentationContext(const std::string& context_identifier);
typename Schema::IfcGeometricRepresentationSubContext getRepresentationSubContext(const std::string& context_identifier, const std::string& context_type);
private:
std::map<std::string, typename Schema::IfcGeometricRepresentationContext> contexts_;
};
template <typename Schema>
IFC_SCHEMA_API ifcopenshell::hierarchy_detail::surface_style_type<Schema>
addStyleAssignment(hierarchy_helper<Schema>& model, double red, double green, double blue, double alpha = 1.0);
template <typename Schema>
IFC_SCHEMA_API ifcopenshell::hierarchy_detail::surface_style_type<Schema>
setSurfaceColour(hierarchy_helper<Schema>& model,
const typename Schema::IfcProductRepresentation& shape,
double red,
double green,
double blue,
double alpha = 1.0);
template <typename Schema>
IFC_SCHEMA_API ifcopenshell::hierarchy_detail::surface_style_type<Schema>
setSurfaceColour(hierarchy_helper<Schema>& model,
const typename Schema::IfcRepresentation& shape,
double red,
double green,
double blue,
double alpha = 1.0);
template <typename Schema>
IFC_SCHEMA_API void setSurfaceColour(
hierarchy_helper<Schema>& model,
const typename Schema::IfcProductRepresentation& shape,
const ifcopenshell::hierarchy_detail::surface_style_type<Schema>& style);
template <typename Schema>
IFC_SCHEMA_API void setSurfaceColour(
hierarchy_helper<Schema>& model,
const typename Schema::IfcRepresentation& shape,
const ifcopenshell::hierarchy_detail::surface_style_type<Schema>& style);
#endif