gras: simplify PMC registration with new macro

1 files changed, 4 insertions, 23 deletions

diff --git a/python/gras/GRAS_Tags.i b/python/gras/GRAS_Tags.i
index c2c640e..84cbbe3 100644
--- a/python/gras/GRAS_Tags.i
+++ b/python/gras/GRAS_Tags.i
@@ -10,27 +10,8 @@ DECL_PMC_SWIG_TYPE(gras::StreamTag, swig_stream_tag)
 DECL_PMC_SWIG_TYPE(gras::PacketMsg, swig_packet_msg)
 
 %pythoncode %{
-
-from GRAS_SBuffer import SBuffer
-
-RegisterPy2PMC(
-    is_py = lambda x: isinstance(x, StreamTag),
-    py2pmc = swig_stream_tag_to_pmc,
-)
-
-RegisterPMC2Py(
-    is_pmc = pmc_is_swig_stream_tag,
-    pmc2py = pmc_to_swig_stream_tag,
-)
-
-RegisterPy2PMC(
-    is_py = lambda x: isinstance(x, PacketMsg),
-    py2pmc = swig_packet_msg_to_pmc,
-)
-
-RegisterPMC2Py(
-    is_pmc = pmc_is_swig_packet_msg,
-    pmc2py = pmc_to_swig_packet_msg,
-)
-
+from GRAS_Tags import StreamTag, PacketMsg
 %}
+
+REG_PMC_SWIG_TYPE(swig_stream_tag, StreamTag)
+REG_PMC_SWIG_TYPE(swig_packet_msg, PacketMsg)