mirror of
https://gitlab.com/scemama/irpf90.git
synced 2024-11-07 22:53:45 +01:00
5565591b8f
Version:1.1.56
96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
#!/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 type(line) == Begin_provider:
|
|
inside = True
|
|
if inside:
|
|
buffer.append(line)
|
|
if type(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):
|
|
result = []
|
|
if command_line.do_openmp:
|
|
result += [ "!$OMP TASK" ]
|
|
result += [ \
|
|
" if (.not.%s_is_built) then"%(x) ]
|
|
result += [ \
|
|
" call provide_%s"%(x) ]
|
|
result += [ \
|
|
" 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
|