From 95af979753aa48744acfcba0a3707fed31391027 Mon Sep 17 00:00:00 2001
From: Alan Mishchenko <alanmi@berkeley.edu>
Date: Mon, 6 Jul 2015 22:47:47 -0700
Subject: Adding new Python API 'co_supp'.

---
 src/python/pyabc.i | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'src/python')

diff --git a/src/python/pyabc.i b/src/python/pyabc.i
index 91c52872..208235a4 100644
--- a/src/python/pyabc.i
+++ b/src/python/pyabc.i
@@ -359,6 +359,31 @@ PyObject* eq_classes()
     return eq_classes;    
 }
 
+PyObject* co_supp( int iCo )
+{
+    PyObject* co_supp;
+    Vec_Int_t * vSupp;
+    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
+    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);
+
+    if ( !pNtk )
+    {
+        Py_RETURN_NONE;
+    }
+
+    vSupp = Abc_NtkNodeSupportInt( pNtk, iCo );
+
+    if( !vSupp )
+    {
+        Py_RETURN_NONE;
+    }
+    
+    co_supp = VecInt_To_PyList( vSupp );
+    Vec_IntFree( vSupp );
+
+    return co_supp;    
+}
+
 void _pyabc_array_clear()
 {
     Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
@@ -715,6 +740,7 @@ int _cex_get_po(Abc_Cex_t* pCex);
 int _cex_get_frame(Abc_Cex_t* pCex);
 
 PyObject* eq_classes();
+PyObject* co_supp(int iCo);
 
 void _pyabc_array_clear();
 void _pyabc_array_push(int i);
-- 
cgit v1.2.3