polly/utils/pyscop/pyscop.py

import json
from isl import *

class Scop:
  def __init__(self, filename):
    f = open(filename, 'r')
    self.json = json.load(f)
    return

  def __str__(self):
    return json.dumps(self.json, indent=2)

  def __repr__(self):
    return str(self)

  @property
  def statements(self):
    return self.json['statements']

class Transforms:
  """
  Create a map that interchanges two dimensions 'A' and 'B'

  numberDimensions: The overall number of dimensions
  dimensionA: The dimension of dimension 'A'
  dimensionB: The dimension of dimension 'B'

  getInterchange(2, 0, 1):
  {[d0, d1] -> [d1, d0]}
  """
  @staticmethod
  def getInterchange(numberDimensions, dimensionA, dimensionB):

    dims = ['d' + str(i) for i in range(numberDimensions)]
    dimString = ",".join(dims)

    changedDims = dims
    first = dims[dimensionA]
    second = dims[dimensionB]
    changedDims[dimensionA] = second
    changedDims[dimensionB] = first
    changedDimString = ",".join(changedDims)

    return Map("{[%s] -> [%s]}" % (dimString, changedDimString))

  """
  Create a map that strip mines one dimension

  numberDimensions: The overall number of dimensions
  stripMineDim: The dimension to strip mine
  factor: The strip mining factor

  getStripMine(2, 1, 64):
  {[d0, d1] -> [d0, o, d1] : o % 64 = 0 and o <= d1 <= d1 + 63}
  """
  @staticmethod
  def getStripMine(numberDimensions, stripMineDim, factor):

    dims = ['d' + str(i) for i in range(numberDimensions)]
    dimString = ",".join(dims)

    changedDims = dims
    smd = dims[stripMineDim]
    changedDims[stripMineDim] = "o,%s" % smd
    changedDimString = ",".join(changedDims)
    string = "{[%s] -> [%s]: o %% %i = 0 and o <= %s <= o + %i}" % \
          (dimString, changedDimString, factor, smd, factor - 1)
    return Map(string)