Add support for case ranges. (#62)

* Add support for case ranges. * Tweak line wrapping to content the linter. Co-authored-by: Andreas Klöckner <[email protected]>
inducer · Sep 22, 2021 · 17c03ac · 17c03ac
1 parent 502dbb9
commit 17c03ac
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/pycparserext/ext_c_parser.py b/pycparserext/ext_c_parser.py
@@ -558,6 +558,15 @@ def p_range_designator(self, p):
         """
         p[0] = RangeExpression(p[2], p[4], coord=self._coord(p.lineno(1)))
 
+    def p_labeled_statement_4(self, p):
+        """ labeled_statement : CASE constant_expression ELLIPSIS constant_expression \
+                COLON pragmacomp_or_statement
+        """
+        p[0] = c_ast.Case(
+                RangeExpression(p[2], p[4], coord=self._coord(p.lineno(1))),
+                [p[6]],
+                self._coord(p.lineno(1)))
+
     def p_unified_volatile_gnu(self, p):
         """ unified_volatile : VOLATILE
                              | __VOLATILE

diff --git a/test/test_pycparserext.py b/test/test_pycparserext.py
@@ -469,6 +469,22 @@ def test_designated_initializers():
     assert _round_trip_matches(src)
 
 
+def test_case_ranges():
+    src = """
+    void f() {
+        switch (1) {
+            case 3:
+                break;
+            case 0 ... 5:
+                break;
+            case 'A' ... 'Z':
+                break;
+        }
+    }
+    """
+    assert _round_trip_matches(src)
+
+
 @pytest.mark.parametrize("restrict_kw", ["restrict", "__restrict__", "__restrict"])
 def test_restrict(restrict_kw):
     src = """