created get_input_tags pythonic wrapper

1 files changed, 16 insertions, 0 deletions

diff --git a/python/gras/GRAS_Block.i b/python/gras/GRAS_Block.i
index a7e27fc..1c04cb6 100644
--- a/python/gras/GRAS_Block.i
+++ b/python/gras/GRAS_Block.i
@@ -171,6 +171,13 @@ struct BlockPython : Block
         const std::vector<void *> &,
         const std::vector<size_t> &
     ) = 0;
+
+    std::vector<Tag> get_input_tags(const size_t which_input)
+    {
+        Block::TagIter it = Block::get_input_tags(which_input);
+        std::vector<Tag> tags(it.begin(), it.end());
+        return tags;
+    }
 };
 
 }
@@ -290,4 +297,13 @@ class Block(BlockPython):
         )
         BlockPython.post_output_tag(self, which_output, t)
 
+    def get_input_tags(self, which_input):
+        for t in BlockPython.get_input_tags(self, which_input):
+            yield Tag(
+                offset=t.offset,
+                key=PMC2Py(t.key),
+                value=PMC2Py(t.value),
+                srcid=PMC2Py(t.srcid),
+            )
+
 %}