From d942f3cb59c8849cab73a3fb67cd1d953665f21c Mon Sep 17 00:00:00 2001
From: Jean-Paul Chaput <Jean-Paul.Chaput@lip6.fr>
Date: Wed, 17 Feb 2021 23:22:31 +0100
Subject: [PATCH] Add support for Rectilinear & Polygon in PyEntity_NEW().

---
 hurricane/src/isobar/PyEntity.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

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<Pin*>( entity );
     if (pin) return PyPin_Link( pin );
     
+    Polygon* polygon = dynamic_cast<Polygon*>( entity );
+    if (polygon) return PyPolygon_Link( polygon );
+    
+    Rectilinear* rectilinear = dynamic_cast<Rectilinear*>( entity );
+    if (rectilinear) return PyRectilinear_Link( rectilinear );
+    
     Net* net = dynamic_cast<Net*>( entity );
     if (net) return PyNet_Link( net );