PEP8

m-labs · Sep 5, 2014 · 4915b4b · 4915b4b
1 parent 7e9df82
commit 4915b4b
Showing 43 changed files with 3,332 additions and 3,068 deletions.
diff --git a/artiq/compiler/fold_constants.py b/artiq/compiler/fold_constants.py
@@ -1,81 +1,86 @@
-import ast, operator
+import ast
+import operator
 
 from artiq.compiler.tools import *
 from artiq.language.core import int64, round64
 
+
 _ast_unops = {
-	ast.Invert: operator.inv,
-	ast.Not: operator.not_,
-	ast.UAdd: operator.pos,
-	ast.USub: operator.neg
+    ast.Invert: operator.inv,
+    ast.Not: operator.not_,
+    ast.UAdd: operator.pos,
+    ast.USub: operator.neg
 }
 
+
 _ast_binops = {
-	ast.Add: operator.add,
-	ast.Sub: operator.sub,
-	ast.Mult: operator.mul,
-	ast.Div: operator.truediv,
-	ast.FloorDiv: operator.floordiv,
-	ast.Mod: operator.mod,
-	ast.Pow: operator.pow,
-	ast.LShift: operator.lshift,
-	ast.RShift: operator.rshift,
-	ast.BitOr: operator.or_,
-	ast.BitXor: operator.xor,
-	ast.BitAnd: operator.and_
+    ast.Add: operator.add,
+    ast.Sub: operator.sub,
+    ast.Mult: operator.mul,
+    ast.Div: operator.truediv,
+    ast.FloorDiv: operator.floordiv,
+    ast.Mod: operator.mod,
+    ast.Pow: operator.pow,
+    ast.LShift: operator.lshift,
+    ast.RShift: operator.rshift,
+    ast.BitOr: operator.or_,
+    ast.BitXor: operator.xor,
+    ast.BitAnd: operator.and_
 }
 
+
 class _ConstantFolder(ast.NodeTransformer):
-	def visit_UnaryOp(self, node):
-		self.generic_visit(node)
-		try:
-			operand = eval_constant(node.operand)
-		except NotConstant:
-			return node
-		try:
-			op = _ast_unops[type(node.op)]
-		except KeyError:
-			return node
-		try:
-			result = value_to_ast(op(operand))
-		except:
-			return node
-		return ast.copy_location(result, node)
+    def visit_UnaryOp(self, node):
+        self.generic_visit(node)
+        try:
+            operand = eval_constant(node.operand)
+        except NotConstant:
+            return node
+        try:
+            op = _ast_unops[type(node.op)]
+        except KeyError:
+            return node
+        try:
+            result = value_to_ast(op(operand))
+        except:
+            return node
+        return ast.copy_location(result, node)
+
+    def visit_BinOp(self, node):
+        self.generic_visit(node)
+        try:
+            left, right = eval_constant(node.left), eval_constant(node.right)
+        except NotConstant:
+            return node
+        try:
+            op = _ast_binops[type(node.op)]
+        except KeyError:
+            return node
+        try:
+            result = value_to_ast(op(left, right))
+        except:
+            return node
+        return ast.copy_location(result, node)
 
-	def visit_BinOp(self, node):
-		self.generic_visit(node)
-		try:
-			left, right = eval_constant(node.left), eval_constant(node.right)
-		except NotConstant:
-			return node
-		try:
-			op = _ast_binops[type(node.op)]
-		except KeyError:
-			return node
-		try:
-			result = value_to_ast(op(left, right))
-		except:
-			return node
-		return ast.copy_location(result, node)
+    def visit_Call(self, node):
+        self.generic_visit(node)
+        fn = node.func.id
+        constant_ops = {
+            "int":    int,
+            "int64":  int64,
+            "round":  round,
+            "round64": round64
+        }
+        if fn in constant_ops:
+            try:
+                arg = eval_constant(node.args[0])
+            except NotConstant:
+                return node
+            result = value_to_ast(constant_ops[fn](arg))
+            return ast.copy_location(result, node)
+        else:
+            return node
 
-	def visit_Call(self, node):
-		self.generic_visit(node)
-		fn = node.func.id
-		constant_ops = {
-			"int":    int,
-			"int64":  int64,
-			"round":  round,
-			"round64": round64
-		}
-		if fn in constant_ops:
-			try:
-				arg = eval_constant(node.args[0])
-			except NotConstant:
-				return node
-			result = value_to_ast(constant_ops[fn](arg))
-			return ast.copy_location(result, node)
-		else:
-			return node
 
 def fold_constants(node):
-	_ConstantFolder().visit(node)
+    _ConstantFolder().visit(node)