Compare commits

...

3 Commits

Author SHA1 Message Date
Thomas Krijnen d017dd1575 Help swig not to trip over old occt handle definitions 2023-06-29 13:18:41 +02:00
Thomas Krijnen 6253553b18 Add --prefilter flag to draw.py 2023-06-29 13:10:17 +02:00
Thomas Krijnen e3a08856d5 SVG prefiltering elements based on large faces 2023-06-29 13:08:22 +02:00
5 changed files with 383 additions and 122 deletions
+3
View File
@@ -414,6 +414,8 @@ int main(int argc, char** argv) {
"Stores name and guid in a separate namespace as opposed to data-name, data-guid")
("svg-poly",
"Uses the polygonal algorithm for hidden line rendering")
("svg-prefilter",
"Prefilter faces and shapes before feeding to HLR algorithm")
("svg-write-poly",
"Approximate every curve as polygonal in SVG output")
("svg-project",
@@ -1082,6 +1084,7 @@ int main(int argc, char** argv) {
}
static_cast<SvgSerializer*>(serializer.get())->setUseNamespace(vmap.count("svg-xmlns") > 0);
static_cast<SvgSerializer*>(serializer.get())->setUseHlrPoly(vmap.count("svg-poly") > 0);
static_cast<SvgSerializer*>(serializer.get())->setUsePrefiltering(vmap.count("svg-prefilter") > 0);
static_cast<SvgSerializer*>(serializer.get())->setPolygonal(vmap.count("svg-write-poly") > 0);
static_cast<SvgSerializer*>(serializer.get())->setAlwaysProject(vmap.count("svg-project") > 0);
static_cast<SvgSerializer*>(serializer.get())->setWithoutStoreys(vmap.count("svg-without-storeys") > 0);
@@ -59,6 +59,7 @@ class draw_settings:
cells: bool = True
merge_cells: bool = False
include_projection: bool = True
prefilter: bool = True
def main(settings, files, iterators=None, merge_projection=True, progress_function=DO_NOTHING):
@@ -131,6 +132,8 @@ def main(settings, files, iterators=None, merge_projection=True, progress_functi
if settings.subtract_before_hlr:
sr.setSubtractionSettings(W.ALWAYS)
sr.setUsePrefiltering(settings.prefilter)
try:
sh = ["none", "full", "left"].index(settings.storey_heights)
sr.setDrawStoreyHeights(sh)
+4
View File
@@ -636,6 +636,10 @@ struct ShapeRTTI : public boost::static_visitor<PyObject*>
}
%}
%ignore hlr_writer;
%ignore hlr_calc;
%ignore occt_join;
%ignore prefiltered_hlr;
%ignore svgfill::svg_to_line_segments;
%ignore svgfill::line_segments_to_polygons;
+11 -119
View File
@@ -713,31 +713,6 @@ void SvgSerializer::write(const IfcGeom::BRepElement* brep_obj) {
}
}
namespace {
class hlr_writer {
const TopoDS_Shape& shape_;
public:
typedef void result_type;
hlr_writer(const TopoDS_Shape& shape) : shape_(shape)
{}
void operator()(boost::blank&) const {
throw std::runtime_error("");
}
void operator()(Handle(HLRBRep_Algo)& algo) const {
algo->Add(shape_);
}
void operator()(Handle(HLRBRep_PolyAlgo)& algo) const {
BRepMesh_IncrementalMesh(shape_, 0.10);
algo->Load(shape_);
}
};
}
namespace {
int infront_or_behind(const gp_Pln& pln, const gp_Pnt& p) {
auto d = (p.XYZ() - pln.Location().XYZ()).Dot(pln.Axis().Direction().XYZ());
@@ -1159,22 +1134,16 @@ void SvgSerializer::write(const geometry_data& data) {
if (is_floor_plan_) {
if (storey) {
if (storey_hlr.find(storey) == storey_hlr.end()) {
if (use_hlr_poly_) {
storey_hlr[storey] = new HLRBRep_PolyAlgo;
} else {
storey_hlr[storey] = new HLRBRep_Algo;
}
auto it = storey_hlr.find(storey);
if (it == storey_hlr.end()) {
it = storey_hlr.insert({ storey, hlr_t(use_prefiltering_, use_hlr_poly_, projection_plane) }).first;
}
hlr_writer vis(*compound_to_hlr);
boost::apply_visitor(vis, storey_hlr[storey]);
it->second.add(*compound_to_hlr);
} else {
Logger::Warning("Unable to invoke HLR due to absence of storey containment", data.product);
}
}
else {
hlr_writer vis(*compound_to_hlr);
boost::apply_visitor(vis, hlr);
} else if (hlr) {
hlr->add(*compound_to_hlr);
}
}
}
@@ -1709,81 +1678,8 @@ std::array<std::array<double, 3>, 3> SvgSerializer::resize() {
return m;
}
namespace {
template <typename T>
TopoDS_Compound occt_join(T t) {
BRep_Builder B;
TopoDS_Compound C;
B.MakeCompound(C);
if (!t.IsNull()) {
TopoDS_Iterator it(t);
for (; it.More(); it.Next()) {
B.Add(C, it.Value());
}
}
return C;
}
template <typename T, typename... Ts>
TopoDS_Compound occt_join(T t, Ts... tss) {
BRep_Builder B;
TopoDS_Compound C;
B.MakeCompound(C);
if (!t.IsNull()) {
TopoDS_Iterator it(t);
for (; it.More(); it.Next()) {
B.Add(C, it.Value());
}
}
auto rest = occt_join(tss...);
if (!rest.IsNull()) {
TopoDS_Iterator it(rest);
for (; it.More(); it.Next()) {
B.Add(C, it.Value());
}
}
return C;
}
class hlr_calc {
private:
const HLRAlgo_Projector& projector_;
public:
typedef TopoDS_Shape result_type;
hlr_calc(const HLRAlgo_Projector& projector) : projector_(projector)
{}
TopoDS_Shape operator()(boost::blank&) const {
throw std::runtime_error("");
}
TopoDS_Shape operator()(Handle(HLRBRep_Algo)& algo) {
algo->Projector(projector_);
algo->Update();
algo->Hide();
HLRBRep_HLRToShape hlr_shapes(algo);
return occt_join(hlr_shapes.OutLineVCompound(), hlr_shapes.VCompound());
}
TopoDS_Shape operator()(Handle(HLRBRep_PolyAlgo)& algo) {
algo->Projector(projector_);
algo->Update();
HLRBRep_PolyHLRToShape hlr_shapes;
hlr_shapes.Update(algo);
return occt_join(hlr_shapes.OutLineVCompound(), hlr_shapes.VCompound());
}
};
}
void SvgSerializer::draw_hlr(const gp_Pln& pln, const drawing_key& drawing_name) {
gp_Trsf trsf;
trsf.SetTransformation(pln.Position());
HLRAlgo_Projector projector(trsf, false, 1.);
hlr_calc vis(projector);
TopoDS_Shape hlr_compound_unmirrored = boost::apply_visitor(vis, drawing_name.first ? this->storey_hlr[drawing_name.first] : hlr);
TopoDS_Shape hlr_compound_unmirrored = (drawing_name.first ? this->storey_hlr.find(drawing_name.first)->second : *hlr).build();
if (!hlr_compound_unmirrored.IsNull()) {
// Compound 3D curves for mirroring to work
@@ -2062,12 +1958,9 @@ void SvgSerializer::finalize() {
pln = &section.plane;
}
if (use_hlr) {
if (use_hlr_poly_) {
hlr = new HLRBRep_PolyAlgo;
} else {
hlr = new HLRBRep_Algo;
}
// @todo do we have always have pln here?
if (use_hlr && pln) {
hlr = new hlr_t(use_prefiltering_, use_hlr_poly_, *pln);
}
section_data_ = std::vector<section_data>{ sd };
@@ -2141,8 +2034,7 @@ void SvgSerializer::finalize() {
resetScale();
// @todo does this probably call Nullify()
hlr = boost::blank();
delete hlr;
}
}
+362 -3
View File
@@ -23,6 +23,7 @@
#define SVGSERIALIZER_H
#include "../ifcgeom_schema_agnostic/GeometrySerializer.h"
#include "../ifcgeom_schema_agnostic/base_utils.h"
#include "../serializers/serializers_api.h"
#include "../serializers/util.h"
@@ -35,6 +36,11 @@
#include <gp_Pln.hxx>
#include <Bnd_Box.hxx>
#include <Standard_Version.hxx>
#include <BRep_Builder.hxx>
#include <HLRBRep_PolyHLRToShape.hxx>
#include <BRepTopAdaptor_FClass2d.hxx>
#include <Geom_Plane.hxx>
#include <BRepBndLib.hxx>
#if OCC_VERSION_HEX >= 0x70300
#include <Bnd_OBB.hxx>
@@ -43,6 +49,7 @@
#include <sstream>
#include <string>
#include <limits>
#include <array>
typedef std::pair<IfcUtil::IfcBaseEntity*, std::string> drawing_key;
@@ -134,7 +141,349 @@ typedef boost::variant<
boost::blank,
Handle(HLRBRep_Algo),
Handle(HLRBRep_PolyAlgo)
> hlr_t;
> hlr_brep_or_poly_t;
namespace {
class hlr_writer {
const TopoDS_Shape& shape_;
public:
typedef void result_type;
hlr_writer(const TopoDS_Shape& shape) : shape_(shape)
{}
void operator()(boost::blank&) const {
throw std::runtime_error("");
}
void operator()(opencascade::handle<HLRBRep_Algo>& algo) const {
algo->Add(shape_);
}
void operator()(opencascade::handle<HLRBRep_PolyAlgo>& algo) const {
BRepMesh_IncrementalMesh(shape_, 0.10);
algo->Load(shape_);
}
};
template <typename T>
TopoDS_Compound occt_join(T t) {
BRep_Builder B;
TopoDS_Compound C;
B.MakeCompound(C);
if (!t.IsNull()) {
TopoDS_Iterator it(t);
for (; it.More(); it.Next()) {
B.Add(C, it.Value());
}
}
return C;
}
template <typename T, typename... Ts>
TopoDS_Compound occt_join(T t, Ts... tss) {
BRep_Builder B;
TopoDS_Compound C;
B.MakeCompound(C);
if (!t.IsNull()) {
TopoDS_Iterator it(t);
for (; it.More(); it.Next()) {
B.Add(C, it.Value());
}
}
auto rest = occt_join(tss...);
if (!rest.IsNull()) {
TopoDS_Iterator it(rest);
for (; it.More(); it.Next()) {
B.Add(C, it.Value());
}
}
return C;
}
class hlr_calc {
private:
const HLRAlgo_Projector& projector_;
public:
typedef TopoDS_Shape result_type;
hlr_calc(const HLRAlgo_Projector& projector) : projector_(projector)
{}
TopoDS_Shape operator()(boost::blank&) const {
throw std::runtime_error("");
}
TopoDS_Shape operator()(opencascade::handle<HLRBRep_Algo>& algo) {
algo->Projector(projector_);
algo->Update();
algo->Hide();
HLRBRep_HLRToShape hlr_shapes(algo);
return occt_join(hlr_shapes.OutLineVCompound(), hlr_shapes.VCompound());
}
TopoDS_Shape operator()(opencascade::handle<HLRBRep_PolyAlgo>& algo) {
algo->Projector(projector_);
algo->Update();
HLRBRep_PolyHLRToShape hlr_shapes;
hlr_shapes.Update(algo);
return occt_join(hlr_shapes.OutLineVCompound(), hlr_shapes.VCompound());
}
};
class prefiltered_hlr {
class face_info {
private:
gp_XYZ dxyz, xdir, ydir;
public:
std::list<TopoDS_Shape>::const_iterator item;
TopoDS_Face face;
bool is_convex;
// @note copying the BRepTopAdaptor_FClass2d didn't work so it's a pointer
BRepTopAdaptor_FClass2d* fclass;
face_info(std::list<TopoDS_Shape>::const_iterator it, const TopoDS_Face& fa)
: item(it)
, face(fa)
, fclass(nullptr)
{
TopExp_Explorer exp(face, TopAbs_WIRE);
is_convex = exp.More() && IfcGeom::util::is_convex(TopoDS::Wire(exp.Current()), 1.e-5) && ([&exp]() {exp.Next(); return true; })() && !exp.More();
auto surf = BRep_Tool::Surface(fa);
if (surf->DynamicType() != STANDARD_TYPE(Geom_Plane)) {
throw std::runtime_error("Not implemented");
}
auto pln = Handle(Geom_Plane)::DownCast(surf);
dxyz = pln->Position().Location().XYZ();
xdir = pln->Position().XDirection().XYZ();
ydir = pln->Position().YDirection().XYZ();
}
~face_info() {
delete fclass;
}
void project(const gp_Pnt& xyz, gp_Pnt2d& uv) {
const gp_Vec d = xyz.XYZ() - dxyz;
uv.SetX(d.Dot(xdir));
uv.SetY(d.Dot(ydir));
}
void interp(const gp_Pnt2d& a, const gp_Pnt2d& b, double d, gp_Pnt2d& out) {
out.SetCoord(a.X() + (b.X() - a.X()) * d, a.Y() + (b.Y() - a.Y()) * d);
}
bool contains(const gp_Pnt& bottomleft, const gp_Pnt& topright) {
gp_Pnt2d a, b;
project(bottomleft, a);
project(topright, b);
return contains(a, b);
}
bool contains(const gp_Pnt2d& bottomleft, const gp_Pnt2d& topright) {
if (!fclass) {
fclass = new BRepTopAdaptor_FClass2d(face, 1.e-5);
}
// @todo unify with the 2d boolean algo
gp_Pnt2d bottomright(topright.X(), bottomleft.Y());
gp_Pnt2d topleft(bottomleft.X(), topright.Y());
std::array<gp_Pnt2d const*, 4> loop{ {
&bottomleft,
&bottomright,
&topright,
&topleft
} };
if (is_convex) {
for (int i = 0; i < 4; ++i) {
if (fclass->Perform(*loop[i]) == TopAbs_OUT) {
return false;
}
}
} else {
gp_Pnt2d tmp;
for (int i = 0; i < 4; ++i) {
// @todo proper edge intersection
for (int j = 0; j < 16; ++j) {
const gp_Pnt2d& a = *loop[i];
const gp_Pnt2d& b = *loop[(i + 1) % 4];
interp(a, b, j / 16.0, tmp);
if (fclass->Perform(tmp) == TopAbs_OUT) {
return false;
}
}
}
}
return true;
}
};
hlr_brep_or_poly_t engine_;
bool use_prefiltering_;
bool use_hlr_poly_;
gp_Ax1 view_direction_;
HLRAlgo_Projector projector_;
std::multimap<double, face_info> large_ortho_faces_;
std::list<TopoDS_Shape> items_;
public:
prefiltered_hlr(bool use_prefiltering, bool use_hlr_poly, const gp_Pln& view_direction)
: use_prefiltering_(use_prefiltering)
, use_hlr_poly_(use_hlr_poly)
// @nb negative z in accordance with occt projector convention (and opengl)
, view_direction_(view_direction.Axis())
{
if (use_hlr_poly_) {
engine_ = new HLRBRep_PolyAlgo;
} else {
engine_ = new HLRBRep_Algo;
}
gp_Trsf trsf;
trsf.SetTransformation(view_direction.Position());
projector_ = HLRAlgo_Projector(trsf, false, 1.);
}
bool is_obscured_(std::list<TopoDS_Shape>::const_iterator sit) {
const TopoDS_Shape& s = *sit;
double min_d = std::numeric_limits<double>::infinity();
TopExp_Explorer exp(s, TopAbs_VERTEX);
for (; exp.More(); exp.Next()) {
const auto& v = TopoDS::Vertex(exp.Current());
auto pnt = BRep_Tool::Pnt(v);
auto d = -(pnt.XYZ() - view_direction_.Location().XYZ()).Dot(view_direction_.Direction().XYZ());
if (d < min_d) {
min_d = d;
}
}
Bnd_Box box;
BRepBndLib::AddClose(s, box);
auto lower = large_ortho_faces_.lower_bound(0.);
auto upper = large_ortho_faces_.upper_bound(min_d);
for (auto it = lower; it != upper; ++it) {
if (it->second.item == sit) {
continue;
}
if (it->second.contains(box.CornerMin(), box.CornerMax())) {
return true;
}
}
return false;
}
void add(const TopoDS_Shape& s) {
if (!use_prefiltering_) {
items_.insert(items_.end(), s);
}
TopoDS_Compound C;
BRep_Builder BB;
BB.MakeCompound(C);
gp_Pnt P;
gp_Vec V;
gp_Dir D;
if (IfcGeom::util::is_manifold(s)) {
size_t n_faces_included = 0, n_total = 0;
{
TopExp_Explorer exp(s, TopAbs_FACE);
for (; exp.More(); exp.Next(), n_total++) {
const auto& face = TopoDS::Face(exp.Current());
if (BRep_Tool::Surface(face)->DynamicType() == STANDARD_TYPE(Geom_Plane)) {
BRepGProp_Face prop(face);
prop.Normal(0., 0., P, V);
if (V.SquareMagnitude() > 1.e-9) {
D = V;
// keep only front-facing
if (D.Dot(view_direction_.Direction()) > 1.e-3) {
BB.Add(C, face);
n_faces_included++;
}
}
} else {
BB.Add(C, face);
n_faces_included++;
}
}
}
Logger::Notice("Included " + std::to_string(n_faces_included) + " faces out of " + std::to_string(n_total) + " after prefiltering");
auto it = items_.insert(items_.end(), C);
{
TopExp_Explorer exp(C, TopAbs_FACE);
for (; exp.More(); exp.Next()) {
const auto& face = TopoDS::Face(exp.Current());
if (BRep_Tool::Surface(face)->DynamicType() == STANDARD_TYPE(Geom_Plane)) {
// find large faces orthogonal to view dir
BRepGProp_Face prop(face);
prop.Normal(0., 0., P, V);
D = V;
if (D.Dot(view_direction_.Direction()) > (1. - 1.e-3)) {
if (IfcGeom::util::face_area(face) > 2.) {
// arbitrary vertex, is ok because orthogonal to view dir
TopExp_Explorer expv(face, TopAbs_VERTEX);
if (expv.More()) {
const auto& v = TopoDS::Vertex(expv.Current());
auto pnt = BRep_Tool::Pnt(v);
auto d = -(pnt.XYZ() - view_direction_.Location().XYZ()).Dot(view_direction_.Direction().XYZ());
if (d > 1.e-5) {
large_ortho_faces_.insert({ d, face_info(it, face) });
}
}
}
}
}
}
}
} else {
items_.insert(items_.end(), s);
}
}
TopoDS_Shape build() {
size_t n_included = 0;
for (auto it = items_.begin(); it != items_.end(); ++it) {
if (!use_prefiltering_ || !is_obscured_(it)) {
hlr_writer vis(*it);
boost::apply_visitor(vis, engine_);
n_included++;
}
}
if (use_prefiltering_) {
Logger::Notice("Included " + std::to_string(n_included) + " elements out of " + std::to_string(items_.size()) + " after prefiltering");
}
hlr_calc vis(projector_);
return boost::apply_visitor(vis, engine_);
}
};
}
typedef prefiltered_hlr hlr_t;
class SERIALIZERS_API SvgSerializer : public WriteOnlyGeometrySerializer {
public:
@@ -162,7 +511,7 @@ protected:
storey_height_display_types storey_height_display_;
bool draw_door_arcs_, is_floor_plan_;
bool auto_section_, auto_elevation_;
bool use_namespace_, use_hlr_poly_, always_project_, polygonal_;
bool use_namespace_, use_hlr_poly_, use_prefiltering_, always_project_, polygonal_;
bool emit_building_storeys_;
bool no_css_;
@@ -181,7 +530,7 @@ protected:
std::list<geometry_data> element_buffer_;
hlr_t hlr;
hlr_t* hlr;
std::string namespace_prefix_;
@@ -211,6 +560,7 @@ public:
, auto_elevation_(false)
, use_namespace_(false)
, use_hlr_poly_(false)
, use_prefiltering_(false)
, always_project_(false)
, polygonal_(false)
, emit_building_storeys_(true)
@@ -221,6 +571,7 @@ public:
, xcoords_begin(0)
, ycoords_begin(0)
, radii_begin(0)
, hlr(nullptr)
, namespace_prefix_("data-")
, subtraction_settings_(ON_SLABS_AT_FLOORPLANS)
{}
@@ -286,6 +637,14 @@ public:
use_hlr_poly_ = b;
}
void setUsePrefiltering(bool b) {
use_prefiltering_ = b;
}
bool getUsePrefiltering() const {
return use_prefiltering_;
}
void setPolygonal(bool b) {
polygonal_ = b;
}