From f4d7bf791d677cbff05c8ee12fe1d3b5a922373f Mon Sep 17 00:00:00 2001
From: Josh Blum
Date: Fri, 30 Nov 2012 23:33:57 -0800
Subject: messing w/ pop_input_msg call

---
 python/gras/GRAS_Block.i | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'python/gras')

diff --git a/python/gras/GRAS_Block.i b/python/gras/GRAS_Block.i
index 8c7ab0e..46f04e2 100644
--- a/python/gras/GRAS_Block.i
+++ b/python/gras/GRAS_Block.i
@@ -197,11 +197,12 @@ def sig_to_dtype_sig(sig):
 Tag__ = Tag
 
 class Tag(object):
-    def __init__(self, offset=0, key=None, value=None, srcid=None):
+    def __init__(self, offset=0, key=None, value=None, srcid=None, tag=None):
         self.offset = offset
         self.key = key
         self.value = value
         self.srcid = srcid
+        self.tag = tag
 
 def YieldTagIter(iter):
     for t in iter: yield Tag(
@@ -209,6 +210,7 @@ def YieldTagIter(iter):
         key=PMC2Py(t.key),
         value=PMC2Py(t.value),
         srcid=PMC2Py(t.srcid),
+        tag=t,
     )
 
 #FIXME major kludge for ref holding
@@ -310,4 +312,7 @@ class Block(BlockPython):
                 t.offset += self.get_produced(o)
                 self.post_output_tag(o, t)
 
+    def pop_input_msg(self, which_input):
+        return list(YieldTagIter([BlockPython.pop_input_msg(self, which_input)]))[0]
+
 %}
-- 
cgit