diff --git a/hurricane/src/isobar/PyEntity.cpp b/hurricane/src/isobar/PyEntity.cpp index bf752116..f2bad553 100644 --- a/hurricane/src/isobar/PyEntity.cpp +++ b/hurricane/src/isobar/PyEntity.cpp @@ -27,6 +27,8 @@ #include "hurricane/isobar/PyVertical.h" #include "hurricane/isobar/PyContact.h" #include "hurricane/isobar/PyPad.h" +#include "hurricane/isobar/PyRectilinear.h" +#include "hurricane/isobar/PyPolygon.h" #include "hurricane/isobar/PyPin.h" #include "hurricane/isobar/PyRoutingPad.h" #include "hurricane/Cell.h" @@ -127,6 +129,12 @@ extern "C" { Pin* pin = dynamic_cast( entity ); if (pin) return PyPin_Link( pin ); + Polygon* polygon = dynamic_cast( entity ); + if (polygon) return PyPolygon_Link( polygon ); + + Rectilinear* rectilinear = dynamic_cast( entity ); + if (rectilinear) return PyRectilinear_Link( rectilinear ); + Net* net = dynamic_cast( entity ); if (net) return PyNet_Link( net );