irpf90/src/variables.py

#!/usr/bin/python
#   IRPF90 is a Fortran90 preprocessor written in Python for programming using
#   the Implicit Reference to Parameters (IRP) method.
#   Copyright (C) 2009 Anthony SCEMAMA 
#
#   This program 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 2 of the License, or
#   (at your option) any later version.
#
#   This program 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 this program; if not, write to the Free Software Foundation, Inc.,
#   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
#   Anthony Scemama
#   LCPQ - IRSAMC - CNRS
#   Universite Paul Sabatier
#   118, route de Narbonne      
#   31062 Toulouse Cedex 4      
#   scemama@irsamc.ups-tlse.fr


from variable import *
from irpf90_t import *
from command_line import command_line
from util import *

######################################################################
def create_variables():
  from preprocessed_text import preprocessed_text
  result = {}
  for filename, text in preprocessed_text:
    buffer = []
    inside = False
    for line in text:
      if isinstance(line,Begin_provider):
        inside = True
      if inside:
        buffer.append(line)
      if isinstance(line,End_provider):
        inside = False
        v = Variable(buffer)
        result[v.name] = v
        for other in v.others:
          result[other] = Variable(buffer,other)
        buffer = []
  return result

variables = create_variables()

######################################################################
def build_use(vars):
  result = map(lambda x: "  use %s"%(variables[x].fmodule), vars)
  result = make_single(result)
  return result

######################################################################
def call_provides(vars,opt=False):
  vars = make_single( map(lambda x: variables[x].same_as, vars) )
  if opt:
    all_children = flatten( map(lambda x: variables[x].children, vars ))
    vars = filter(lambda x: x not in all_children,vars)
  def fun(x):
    if command_line.do_openmp:
      result  = [ "!$OMP TASK" ]
    else:
      result = []
    result += [ \
    "  if (.not.%s_is_built) then"%(x),
    "    call provide_%s"%(x),
    "  endif" ]
    if command_line.do_openmp:
      result += [ "!$OMP END TASK" ]
    return result

  result = flatten ( map (fun, vars) )
  if command_line.do_openmp and result != []:
   result.reverse()
   result.remove("!$OMP TASK")
   result.remove("!$OMP END TASK")
   result.reverse()
   if "!$OMP TASK" in result:
     result += [ "!$OMP TASKWAIT" ]
  return result

######################################################################
if __name__ == '__main__':
  for v in variables.keys():
    print v