[Spice-devel,v3,14/51] Decorate writer class to make easier ifdef/endif handling

Submitted by Frediano Ziglio on July 21, 2015, 4:45 p.m.

Details

Message ID 1437497181-26929-15-git-send-email-fziglio@redhat.com
State New
Headers show

Not browsing as part of any series.

Commit Message

Frediano Ziglio July 21, 2015, 4:45 p.m.
I'm generating code for dissector from demarshaller.
Make simple to hangle ifdef/endif not having to check manually attribute.

Signed-off-by: Frediano Ziglio <fziglio@redhat.com>
---
 python_modules/dissector.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

Patch hide | download patch | download mbox

diff --git a/python_modules/dissector.py b/python_modules/dissector.py
index 52234fc..f9ad08a 100644
--- a/python_modules/dissector.py
+++ b/python_modules/dissector.py
@@ -2,6 +2,8 @@ 
 from . import codegen
 import re
 
+import types
+
 
 # generate a new tree identifier
 ett_writer = None
@@ -95,10 +97,26 @@  def write_protocol_definitions(writer):
     writer.end_block()
 
 
+def decorate_writer(writer):
+    cls = writer.__class__
+
+    def create(old):
+        def ifdef(self, member):
+            if member.has_attr("ifdef"):
+                old(self, member.attributes["ifdef"][0])
+        return types.MethodType(ifdef, None, cls)
+
+    cls.ifdef      = create(cls.ifdef)
+    cls.ifdef_else = create(cls.ifdef_else)
+    cls.endif      = create(cls.endif)
+
+
 def write_protocol_parser(writer, proto):
     global hf_writer
     global ett_writer
 
+    decorate_writer(writer)
+
     write_parser_helpers(writer)
 
     # put fields declaration first