Protected expressions in ternary operator with paranthesis.

2 files changed, 11 insertions, 3 deletions

diff --git a/pycparser/c_generator.py b/pycparser/c_generator.py
index ad0e89d..662a088 100644
--- a/pycparser/c_generator.py
+++ b/pycparser/c_generator.py
@@ -196,9 +196,9 @@ class CGenerator(object):
         return 'continue;'
 
     def visit_TernaryOp(self, n):
-        s = self._visit_expr(n.cond) + ' ? '
-        s += self._visit_expr(n.iftrue) + ' : '
-        s += self._visit_expr(n.iffalse)
+        s  = '(' + self._visit_expr(n.cond) + ') ? '
+        s += '(' + self._visit_expr(n.iftrue) + ') : '
+        s += '(' + self._visit_expr(n.iffalse) + ')'
         return s
 
     def visit_If(self, n):
diff --git a/tests/test_c_generator.py b/tests/test_c_generator.py
index 35afe17..cb3bd08 100644
--- a/tests/test_c_generator.py
+++ b/tests/test_c_generator.py
@@ -88,6 +88,14 @@ class TestCtoC(unittest.TestCase):
         #s = 'int test(const char* const* arg);'
         #parse_to_ast(s).show()
 
+    def test_ternary(self):
+        self._assert_ctoc_correct('''
+            int main(void)
+            {
+                int a, b;
+                (a == 0) ? (b = 1) : (b = 2);
+            }''')
+
     def test_casts(self):
         self._assert_ctoc_correct(r'''
             int main() {