tiling.py

Go to the documentation of this file.

#   This file is part of the Ansel project.
#   Copyright (C) 2022 Aurélien PIERRE.
#   
#   Ansel is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#   
#   Ansel is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.
#   
#   You should have received a copy of the GNU General Public License
#   along with Ansel.  If not, see <http://www.gnu.org/licenses/>.
#   
#   
#   
#   
#   
#   
#   
#   
#   
#   
#   
 Read an IOP file and guess how many buffers it needs from reading allocs
 This can be used to predict the RAM footprint of an IOP and setup tiling properly
 Example use :
  `python tiling.py filmicrgb.c OpenCL` for OpenCL buffers
  `python tiling.py filmicrgb.c C` for C buffers
#   
import os
import re
import sys
directory = "../src/iop/"
 
path = os.path.join(directory, sys.argv[1])
 
if(not os.path.isfile(path)):
  print("%s is not a file" % path)
  exit(1)
 
lang = ""
if(len(sys.argv) > 2):
  lang = sys.argv[2]
 
f = open(path, "r")
content = f.read()
print(path)
 
excluded_keywords = [ "float", "size_t", "int", "uint", "uint8_t", "uint16_t", "cl_mem", "sizeof" ]
 
def parse_allocs(regex, content):
  matches = re.finditer(regex, content, re.MULTILINE)
 
  for matchNum, match in enumerate(matches):
    # Get the name of the pointer to the allocation
    variable_name = match.group(1)
 
    # Get the type of alloc function
    alloc_type = match.group(2)
 
    # Get the arguments of the function
    args = match.group(3)
    print("\t", match.group(0))
 
    # Extract all arguments
    args = args.split(",")
 
    for arg in args:
      # Try to find variables in arguments
      variables_mask = r"([a-zA-Z\_]+\[{0,1}[0-9A-Z]*\]{0,1}[\->]*[a-zA-Z0-9\_]*\[{0,1}[0-9A-Z]*\]{0,1})"
      variables = set(re.findall(variables_mask, arg))
 
      for var in variables:
        # Try to find the variables assignations/declarations
        if var not in excluded_keywords:
          print("\t\t-", var, ":")
 
          # Remove arrays if any
          var = re.sub(r"\[.*\]", "", var)
          declaration_mask = r"%s\[{0,1}[0-9A-Z]*\]{0,1} = .+;" % var
          declarations = set(re.findall(declaration_mask, content))
 
          # Print the declaration line of the variables if found
          if(len(declarations) > 0):
            for declaration in declarations:
              print("\t\t\t", declaration)
          else:
            print("\t\t\t no assignation found")
 
if(lang == "C"):
  print("C buffers allocated:")
  alloc_regex = r"([a-zA-Z0-9_\->\.\[\]]+) = (dt_|c|m)alloc.*\‍((.+)\‍)"
  parse_allocs(alloc_regex, content)
 
elif(lang == "OpenCL"):
  print("\nOpenCL buffers allocated:")
  alloc_regex = r"([a-zA-Z0-9_\->\.\[\]]+) = dt_opencl(.*)alloc[a-zA-Z_-]*\‍((.+)\‍)"
  parse_allocs(alloc_regex, content)
 
else:
  print("Option 2 `%s` not recognized" % lang)
  print("valid options are `C` or `OpenCL`")
  print("Example:")
  print("\t`python tiling.py filmicrgb.c OpenCL` for OpenCL buffers")
  print("\t`python tiling.py filmicrgb.c C` for C buffers")
 
  f.close()
  exit(1)
 
f.close()