python-cc 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

pcc/ast/ast.py

@@ -140,7 +140,8 @@ class PrototypeAST(ASTNode):
         return self.isoperator and len(self.argnames) == 2
 
     def get_op_name(self):
-        assert self.isoperator
+        if not self.isoperator:
+            raise ValueError(f"get_op_name called on non-operator '{self.name}'")
         return self.name[-1]
 
     def dump(self, indent=0):

pcc/ast/ast_transforms.py

@@ -61,7 +61,10 @@ def fix_switch_cases(switch_node):
 
         A fixed AST node is returned. The argument may be modified.
     """
-    assert isinstance(switch_node, c_ast.Switch)
+    if not isinstance(switch_node, c_ast.Switch):
+        raise TypeError(
+            f"fix_switch_cases expects a Switch node, got {type(switch_node).__name__}"
+        )
     if not isinstance(switch_node.stmt, c_ast.Compound):
         return switch_node
 

pcc/ast/c_ast.py

@@ -457,6 +457,37 @@ class For(Node):
 
     attr_names = ()
 
+class GenericAssociation(Node):
+    __slots__ = ('type', 'expr', 'coord', '__weakref__')
+    def __init__(self, type, expr, coord=None):
+        self.type = type
+        self.expr = expr
+        self.coord = coord
+
+    def children(self):
+        nodelist = []
+        if self.type is not None: nodelist.append(("type", self.type))
+        if self.expr is not None: nodelist.append(("expr", self.expr))
+        return tuple(nodelist)
+
+    attr_names = ()
+
+class GenericSelection(Node):
+    __slots__ = ('expr', 'associations', 'coord', '__weakref__')
+    def __init__(self, expr, associations, coord=None):
+        self.expr = expr
+        self.associations = associations
+        self.coord = coord
+
+    def children(self):
+        nodelist = []
+        if self.expr is not None: nodelist.append(("expr", self.expr))
+        for i, child in enumerate(self.associations or []):
+            nodelist.append(("associations[%d]" % i, child))
+        return tuple(nodelist)
+
+    attr_names = ()
+
 class FuncCall(Node):
     __slots__ = ('name', 'args', 'coord', '__weakref__')
     def __init__(self, name, args, coord=None):
@@ -517,6 +548,19 @@ class Goto(Node):
 
     attr_names = ('name', )
 
+class ComputedGoto(Node):
+    __slots__ = ('expr', 'coord', '__weakref__')
+    def __init__(self, expr, coord=None):
+        self.expr = expr
+        self.coord = coord
+
+    def children(self):
+        nodelist = []
+        if self.expr is not None: nodelist.append(("expr", self.expr))
+        return tuple(nodelist)
+
+    attr_names = ()
+
 class ID(Node):
     __slots__ = ('name', 'coord', '__weakref__', "ir_type")
     def __init__(self, name, coord=None):
@@ -587,6 +631,17 @@ class Label(Node):
 
     attr_names = ('name', )
 
+class LabelAddress(Node):
+    __slots__ = ('name', 'coord', '__weakref__')
+    def __init__(self, name, coord=None):
+        self.name = name
+        self.coord = coord
+
+    def children(self):
+        return tuple()
+
+    attr_names = ('name', )
+
 class NamedInitializer(Node):
     __slots__ = ('name', 'expr', 'coord', '__weakref__')
     def __init__(self, name, expr, coord=None):
@@ -603,6 +658,21 @@ class NamedInitializer(Node):
 
     attr_names = ()
 
+class RangeDesignator(Node):
+    __slots__ = ('start', 'end', 'coord', '__weakref__')
+    def __init__(self, start, end, coord=None):
+        self.start = start
+        self.end = end
+        self.coord = coord
+
+    def children(self):
+        nodelist = []
+        if self.start is not None: nodelist.append(("start", self.start))
+        if self.end is not None: nodelist.append(("end", self.end))
+        return tuple(nodelist)
+
+    attr_names = ()
+
 class ParamList(Node):
     __slots__ = ('params', 'coord', '__weakref__')
     def __init__(self, params, coord=None):
@@ -659,6 +729,19 @@ class Struct(Node):
 
     attr_names = ('name', )
 
+class StmtExpr(Node):
+    __slots__ = ('stmt', 'coord', '__weakref__')
+    def __init__(self, stmt, coord=None):
+        self.stmt = stmt
+        self.coord = coord
+
+    def children(self):
+        nodelist = []
+        if self.stmt is not None: nodelist.append(("stmt", self.stmt))
+        return tuple(nodelist)
+
+    attr_names = ()
+
 class StructRef(Node):
     __slots__ = ('name', 'type', 'field', 'coord', '__weakref__')
     def __init__(self, name, type, field, coord=None):
@@ -796,5 +879,3 @@ class While(Node):
         return tuple(nodelist)
 
     attr_names = ()
-
-