#!/usr/bin/python2

import string
import os

properties = []
dims = {}

files = filter(lambda x: x.startswith("PROPERTIES") and \
                         x.endswith("irp.f"), os.listdir(os.getcwd()))

files = map(lambda x: 'PROPERTIES/'+x, filter(lambda x: x.endswith("irp.f"), os.listdir(os.getcwd()+'/PROPERTIES')))
                         

#files = filter(lambda x: x.endswith("irp.f"), os.listdir(os.getcwd()))

for filename in files:
  lines = []
  check_dims = False
  file = open(filename,'r')
  lines += file.readlines()
  file.close()
  for i,line in enumerate(lines):
    if line.startswith("! PROPERTIES"):
      lines = lines[i:]
      break

  for line in map(lambda x: x.lower(),lines):
    if line.lstrip().startswith('begin_provider'):
      check_dims = False
      buffer = line
      buffer = buffer.split('[')[1]
      buffer = buffer.split(']')[0]
      buffer = buffer.split(',')
      if (len(buffer) == 2):
        buffer.append("")
      else:
        buffer = [ buffer[0], buffer[1], ','.join(buffer[2:]) ]
        check_dims = True
      buffer = map(string.strip,buffer)
      properties.append(buffer)
      current_prop = buffer[1]
    elif check_dims:
      if 'dimensions :' in line:
        dims[current_prop] = line.split(':')[1].strip()



def sq(item):
  return [item[0], item[1]+"_2", item[2]]
properties_with_square = properties + map(sq,properties)

for p in [ properties, properties_with_square ]:
  def c(x,y):
    if x[1] >  y[1]: return 1
    if x[1] == y[1]: return 0
    if x[1] <  y[1]: return -1
  p.sort(c)
  
def namelist():
  buffer = ""
  for p in properties:
    buffer += "calc_"+p[1]+", &\n"
  buffer = buffer[:-4]
  result = "  namelist /properties/"+buffer
  return result

def touch_all():
  print "TOUCH", 
  for p in properties:
    print "calc_"+p[1],
  print ""

#file = open('../scripts/properties.py','w')
#print >>file,'properties = ',properties
#file.close()

def make_dims():
  template = """
BEGIN_PROVIDER [ integer, size_%(p)s ]
 implicit none
 BEGIN_DOC  
! Size of %(p)s
 END_DOC
 if (calc_%(p)s) then
  size_%(p)s = %(d)s
 else
  size_%(p)s = 1
 endif
END_PROVIDER
"""
  for p in dims:
    print template%{'p': p, 'd': dims[p]}