10
0
mirror of https://gitlab.com/scemama/irpf90.git synced 2024-12-07 21:13:34 +01:00
irpf90/src/subroutine.py

114 lines
3.8 KiB
Python
Raw Normal View History

2009-09-04 18:37:10 +02:00
#!/usr/bin/python
2009-09-23 12:51:27 +02:00
# 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
2009-09-23 12:51:27 +02:00
2009-09-04 18:37:10 +02:00
from irpf90_t import *
from util import *
import error
class Sub(object):
############################################################
def __init__(self,text):
2010-10-02 23:42:55 +02:00
assert type(text) == list
2009-09-04 18:37:10 +02:00
assert len(text) > 0
2009-09-05 15:37:23 +02:00
assert type(text[0]) in [Subroutine, Function]
2009-09-04 18:37:10 +02:00
self.text = text
############################################################
def name(self):
'''Name is lowercase'''
if '_name' not in self.__dict__:
self._name = find_subname(self.line)
return self._name
name = property(name)
2009-10-21 17:24:24 +02:00
############################################################
def is_function(self):
if '_is_function' not in self.__dict__:
2010-05-19 19:01:05 +02:00
self._is_function = "function" in self.line.lower
2009-10-21 17:24:24 +02:00
return self._is_function
is_function = property(is_function)
2009-09-04 18:37:10 +02:00
############################################################
def doc(self):
if '_doc' not in self.__dict__:
def f(l): return
2010-10-02 23:42:55 +02:00
buffer = filter(lambda l:type(l) == Doc, self.text)
2009-09-04 18:37:10 +02:00
self._doc = map(lambda l: l.text[1:], buffer)
if buffer == []:
error.warn(None,"Subroutine %s is not documented"%(self.name))
return self._doc
doc = property(doc)
############################################################
def line(self):
if '_line' not in self.__dict__:
self._line = self.text[0]
return self._line
line = property(line)
############################################################
def touches(self):
if '_touches' not in self.__dict__:
2009-09-04 23:48:49 +02:00
from subroutines import subroutines
self._touches = []
2010-10-02 23:42:55 +02:00
for line in filter(lambda x: type(x) in [Touch, SoftTouch],self.text):
2009-09-04 23:48:49 +02:00
self._touches += line.text.split()[1:]
for sub in self.calls:
2009-09-05 16:05:00 +02:00
if sub in subroutines:
self._touches += subroutines[sub].touches
2009-09-04 23:48:49 +02:00
self._touches = make_single(self._touches)
2009-09-04 18:37:10 +02:00
return self._touches
2009-09-04 23:48:49 +02:00
touches = property(touches)
2009-09-04 18:37:10 +02:00
############################################################
def regexp(self):
if '_regexp' not in self.__dict__:
import re
self._regexp = re.compile( \
2009-09-09 16:59:43 +02:00
r"([^a-z0-9'\"_]|^)%s([^a-z0-9_]|$)"%(self.name),re.I)
2009-09-04 18:37:10 +02:00
return self._regexp
regexp = property(regexp)
############################################################
def calls(self):
if '_calls' not in self.__dict__:
2010-10-02 23:42:55 +02:00
buffer = filter(lambda x: type(x) == Call,self.text)
2009-09-04 18:37:10 +02:00
self._calls = []
for line in buffer:
sub = line.text.split('(',1)[0].split()[1]
self._calls.append(sub)
2009-09-04 23:48:49 +02:00
self._calls = make_single(self._calls)
2009-09-04 18:37:10 +02:00
return self._calls
calls = property(calls)
######################################################################
if __name__ == '__main__':
from preprocessed_text import preprocessed_text
from subroutines import subroutines
2009-09-04 23:48:49 +02:00
print subroutines['brownian_step'].touches