Add support for case ranges. (#62)

* Add support for case ranges.

* Tweak line wrapping to content the linter.

Co-authored-by: Andreas Klöckner <inform@tiker.net>

Author: qwenger

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

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 = """