polly/utils/jscop2cloog.py

   1 #!/usr/bin/env python
   2 import argparse, os
   3 import json
   4
   5 def getDomains(scop):
   6   statements = scop['statements'];
   7   numStatements = len(statements)
   8
   9   output = "%s\n\n" % str(numStatements)
  10
  11   for statement in scop['statements']:
  12     output += "%s\n\n" % statement['domain']
  13     output += "0  0  0               # for future options\n\n"
  14
  15
  16   return output
  17
  18 def getSchedules(scop):
  19   statements = scop['statements'];
  20   numStatements = len(statements)
  21
  22   output = "%s\n\n" % str(numStatements)
  23
  24   for statement in scop['statements']:
  25     output += "%s\n\n" % statement['schedule']
  26
  27   return output
  28
  29 def writeCloog(scop):
  30   template = """
  31 # ---------------------- CONTEXT ----------------------
  32 c # language is C
  33
  34 # Context (no constraints on two parameters)
  35 %s
  36
  37 0 # We do not want to set manually the parameter names
  38
  39 # --------------------- STATEMENTS --------------------
  40 %s
  41
  42 0 # We do not want to set manually the iterator names
  43
  44 # --------------------- SCATTERING --------------------
  45 %s
  46
  47 0 # We do not want to set manually the schedule dimension names
  48 """
  49
  50   context = scop['context']
  51   domains = getDomains(scop)
  52   schedules = getSchedules(scop)
  53   print template % (context, domains, schedules)
  54
  55 def __main__():
  56   description = 'Translate JSCoP into iscc input'
  57   parser = argparse.ArgumentParser(description)
  58   parser.add_argument('inputFile', metavar='N', type=file,
  59                       help='The JSCoP file')
  60
  61   args = parser.parse_args()
  62   inputFile = args.inputFile
  63   scop = json.load(inputFile)
  64
  65   writeCloog(scop)
  66
  67 __main__()
  68