commit ae36e9629694dd6dc12b2f03f4856f9cd90896f6
author eli.bendersky <devnull@localhost> Sun Feb 27 08:25:05 2011 +0200
committer eli.bendersky <devnull@localhost> Sun Feb 27 08:25:05 2011 +0200
tree addf4f0395ed8b70d76b1f9a81c0ae33f235a6e4
parent 3ae9f7af34cad7499262b45140bc4cca887036d5

a couple more statements implemented

examples/c-to-c.py

diff --git a/examples/c-to-c.py b/examples/c-to-c.py
index a329be6..7201909 100644
--- a/examples/c-to-c.py
+++ b/examples/c-to-c.py
@@ -54,6 +54,12 @@
     def visit_ArrayRef(self, n):
         return self.visit(n.name) + '[' + self.visit(n.subscript) + ']'
 
+    def visit_UnaryOp(self, n):
+        if n.op == 'p++':
+            return '%s++' % self.visit(n.expr)
+        else:
+            return '%s%s' % (n.op, self.visit(n.expr))
+
     def visit_BinaryOp(self, n):
         return '%s %s %s' % (self.visit(n.left), n.op, self.visit(n.right))
     
@@ -135,15 +141,29 @@
         if n.expr: s += ' ' + self.visit(n.expr)
         return s + ';'
 
-    def _generate_stmt(self, n):
+    def visit_For(self, n):
+        s = 'for ('
+        if n.init: s += self.visit(n.init)
+        s += ';'
+        if n.cond: s += ' ' + self.visit(n.cond)
+        s += ';'
+        if n.next: s += ' ' + self.visit(n.next)
+        s += ')\n'
+        s += self._generate_stmt(n.stmt, add_indent=True)
+        return s
+
+    def _generate_stmt(self, n, add_indent=False):
         """ Generation from a statement node. This method exists as a wrapper
             for individual visit_* methods to handle different treatment of 
             some statements in this context.
         """
         typ = type(n)
+        if add_indent: self.indent_level += 2
         s = self._make_indent()
+        if add_indent: self.indent_level -= 2
         
-        if typ in (c_ast.Decl, c_ast.Assignment, c_ast.Cast):
+        if typ in ( c_ast.Decl, c_ast.Assignment, c_ast.Cast, c_ast.UnaryOp,
+                    c_ast.BinaryOp):
             # These can also appear in an expression context so no semicolon
             # is added to them automatically
             #
@@ -153,7 +173,7 @@
             # compound - because it consists of multiple lines it has to 
             # compute its own indentation.
             #
-            return self.visit(n) + '\n'
+            return self.visit(n)
         else:
             return s + self.visit(n) + '\n'
 
@@ -217,24 +237,19 @@
         translate_to_c(sys.argv[1])
     else:
         src = r'''
-        
-typedef enum tagReturnCode {SUCCESS=99, FAIL} ReturnCode;
-
-
-typedef struct tagEntry
+static unsigned int hash_func(const char* str, unsigned int table_size)
 {
-    char* key;
-    char* value;
-} Entry;
+    unsigned int hash_value;
+    unsigned int a = 127;
+    a++;
+    ++a;
 
+    for (hash_value = 0; *str != 0; ++str)
+        {hash_value = (a*hash_value + *str) % table_size;}
 
+    return hash_value;
+}
 
-typedef struct tagNode
-{
-    Entry* entry;
-
-    struct tagNode* next;
-} Node;
         '''
         parser = c_parser.CParser()
         ast = parser.parse(src)