ruff format

Author: cclauss

polly/utils/jscop2cloog.py

@@ -2,32 +2,34 @@
 import argparse, os
 import json
 
+
 def getDomains(scop):
-  statements = scop['statements'];
-  numStatements = len(statements)
+    statements = scop["statements"]
+    numStatements = len(statements)
 
-  output = "%s\n\n" % str(numStatements)
+    output = "%s\n\n" % str(numStatements)
 
-  for statement in scop['statements']:
-    output += "%s\n\n" % statement['domain']
-    output += "0  0  0               # for future options\n\n"
+    for statement in scop["statements"]:
+        output += "%s\n\n" % statement["domain"]
+        output += "0  0  0               # for future options\n\n"
 
+    return output
 
-  return output
 
 def getSchedules(scop):
-  statements = scop['statements'];
-  numStatements = len(statements)
+    statements = scop["statements"]
+    numStatements = len(statements)
+
+    output = "%s\n\n" % str(numStatements)
 
-  output = "%s\n\n" % str(numStatements)
+    for statement in scop["statements"]:
+        output += "%s\n\n" % statement["schedule"]
 
-  for statement in scop['statements']:
-    output += "%s\n\n" % statement['schedule']
+    return output
 
-  return output
 
 def writeCloog(scop):
-  template = """
+    template = """
 # ---------------------- CONTEXT ----------------------
 c # language is C
 
@@ -47,22 +49,22 @@ def writeCloog(scop):
 0 # We do not want to set manually the schedule dimension names
 """
 
-  context = scop['context']
-  domains = getDomains(scop)
-  schedules = getSchedules(scop)
-  print(template % (context, domains, schedules))
+    context = scop["context"]
+    domains = getDomains(scop)
+    schedules = getSchedules(scop)
+    print(template % (context, domains, schedules))
+
 
 def __main__():
-  description = 'Translate JSCoP into iscc input'
-  parser = argparse.ArgumentParser(description)
-  parser.add_argument('inputFile', metavar='N', type=file,
-                      help='The JSCoP file')
+    description = "Translate JSCoP into iscc input"
+    parser = argparse.ArgumentParser(description)
+    parser.add_argument("inputFile", metavar="N", type=file, help="The JSCoP file")
 
-  args = parser.parse_args()
-  inputFile = args.inputFile
-  scop = json.load(inputFile)
+    args = parser.parse_args()
+    inputFile = args.inputFile
+    scop = json.load(inputFile)
 
-  writeCloog(scop)
+    writeCloog(scop)
 
-__main__()
 
+__main__()

polly/utils/pyscop/jscop2iscc.py

@@ -4,67 +4,66 @@
 import argparse, isl, os
 import json
 
+
 def printDomain(scop):
+    domain = isl.USet("{}")
 
-  domain = isl.USet('{}')
+    for statement in scop["statements"]:
+        domain = domain.union(isl.USet(statement["domain"]))
 
-  for statement in scop['statements']:
-    domain = domain.union(isl.USet(statement['domain']))
+    print("D :=", end=" ")
+    print(str(domain) + ";")
 
-  print("D :=", end=" ")
-  print(str(domain) + ";")
 
 def printAccesses(scop):
+    read = isl.UMap("{}")
 
-  read = isl.UMap('{}')
+    for statement in scop["statements"]:
+        for access in statement["accesses"]:
+            if access["kind"] == "read":
+                read = read.union(isl.UMap(access["relation"]))
 
-  for statement in scop['statements']:
-    for access in statement['accesses']:
-      if access['kind'] == 'read':
-        read = read.union(isl.UMap(access['relation']))
+    print("R :=", end=" ")
+    print(str(read) + ";")
 
-  print("R :=", end=" ")
-  print(str(read) + ";")
+    write = isl.UMap("{}")
 
-  write = isl.UMap('{}')
+    for statement in scop["statements"]:
+        for access in statement["accesses"]:
+            if access["kind"] == "write":
+                write = write.union(isl.UMap(access["relation"]))
 
-  for statement in scop['statements']:
-    for access in statement['accesses']:
-      if access['kind'] == 'write':
-        write = write.union(isl.UMap(access['relation']))
+    print("W :=", end=" ")
+    print(str(write) + ";")
 
-  print("W :=", end=" ")
-  print(str(write) + ";")
 
 def printSchedule(scop):
+    schedule = isl.UMap("{}")
 
-  schedule = isl.UMap('{}')
+    for statement in scop["statements"]:
+        schedule = schedule.union(isl.UMap(statement["schedule"]))
 
-  for statement in scop['statements']:
-    schedule = schedule.union(isl.UMap(statement['schedule']))
+    print("S :=", end=" ")
+    print(str(schedule) + ";")
 
-  print("S :=", end=" ")
-  print(str(schedule) + ";")
 
 def __main__():
-  description = 'Translate JSCoP into iscc input'
-  parser = argparse.ArgumentParser(description)
-  parser.add_argument('inputFile', metavar='N', type=file,
-                      help='The JSCoP file')
+    description = "Translate JSCoP into iscc input"
+    parser = argparse.ArgumentParser(description)
+    parser.add_argument("inputFile", metavar="N", type=file, help="The JSCoP file")
 
-  args = parser.parse_args()
-  inputFile = args.inputFile
-  scop = json.load(inputFile)
+    args = parser.parse_args()
+    inputFile = args.inputFile
+    scop = json.load(inputFile)
 
-  printDomain(scop)
-  printAccesses(scop)
-  printSchedule(scop)
+    printDomain(scop)
+    printAccesses(scop)
+    printSchedule(scop)
 
-  print('R := R * D;')
-  print('W := W * D;')
-  print('Dep := (last W before R under S)[0];')
-  print('schedule D respecting Dep minimizing Dep;')
+    print("R := R * D;")
+    print("W := W * D;")
+    print("Dep := (last W before R under S)[0];")
+    print("schedule D respecting Dep minimizing Dep;")
 
 
 __main__()
-