From 7f8767ee5ecb0f2636db232d768d053dde1e5cc1 Mon Sep 17 00:00:00 2001
From: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
Date: Fri, 4 Sep 2009 15:11:42 +0200
Subject: [PATCH] Introduced variables Version:1.1.2

---
 src/command_line.py      |  12 +--
 src/error.py             |  11 ++-
 src/irpf90_t.py          |  13 ++-
 src/preprocessed_text.py |  72 +++++++++--------
 src/util.py              |  16 +++-
 src/variable.py          | 168 +++++++++++++++++++++++++++++++++++++++
 src/variables.py         |  30 +++++++
 7 files changed, 272 insertions(+), 50 deletions(-)
 create mode 100644 src/variable.py
 create mode 100644 src/variables.py

diff --git a/src/command_line.py b/src/command_line.py
index 94a667d..ccea683 100644
--- a/src/command_line.py
+++ b/src/command_line.py
@@ -23,9 +23,7 @@ class CommandLine(object):
     self.executable_name = self.argv[0]
 
   def defined(self):
-    try:
-      d = self._defined
-    except AttributeError:
+    if '_defined' not in self.__dict__:
       self._defined = []
       for o,a in self.opts:
         if o in [ "-D", options['D'][0] ]:
@@ -79,9 +77,7 @@ Options:
   
   t = """
 def do_$LONG(self):
-    try:
-      x = self._do_$LONG
-    except AttributeError:
+    if '_do_$LONG' not in self.__dict__:
       self._do_$LONG = False
       for o,a in self.opts:
         if o in ("-$SHORT", "--$LONG"):
@@ -95,9 +91,7 @@ do_$LONG = property(fget=do_$LONG)
     exec t.replace("$LONG",long).replace("$SHORT",short)
 
   def do_run(self):
-   try:
-     x = self._do_run
-   except AttributeError:
+   if '_do_run' not in self.__dict__:
      self._do_run = not (self.do_version or self.do_init)
    return self._do_run
   do_run = property(fget=do_run)
diff --git a/src/error.py b/src/error.py
index 58b5ec9..a0b6112 100644
--- a/src/error.py
+++ b/src/error.py
@@ -11,20 +11,23 @@ Error:
 -----
 """
   print message, '\n'
-  print "file %s ; line %d :\n %s"%(line.filename,line.i,line.text)
+  if line is not None:
+    print "file %s ; line %d :\n %s"%(line.filename,line.i,line.text)
   sys.exit(1)
 
 
 ######################################################################
 def warn(line,message):
   assert isinstance(line,Line)
-  print """
+  if line is not None:
+    print """
 Warning:
 -------
 """
-  if line.i > 0:
+    print message, '\n'
     print "file %s, line %d:\n %s"%(line.filename,line.i,line.text)
-  print message, '\n'
+  else:
+    print "Warning: %s"%(message)
 
 
 ######################################################################
diff --git a/src/irpf90_t.py b/src/irpf90_t.py
index 9954d16..aaae847 100644
--- a/src/irpf90_t.py
+++ b/src/irpf90_t.py
@@ -257,7 +257,18 @@ class Provide_all (Line):
     return "%20s:%5d : %s"%("Provide_all",self.i,self.text)
 
 
-
+######################################################################
+
+def create_irpf90_files():
+  result = []
+  from command_line import command_line
+  import os
+  if command_line.do_run:
+    def is_irpf90_file(filename):
+      return filename.endswith(".irp.f")
+    result = filter ( is_irpf90_file, os.listdir(os.getcwd()) )
+  return result
+irpf90_files = create_irpf90_files()
 
 
 
diff --git a/src/preprocessed_text.py b/src/preprocessed_text.py
index 0bc0eea..98a3bf4 100644
--- a/src/preprocessed_text.py
+++ b/src/preprocessed_text.py
@@ -323,7 +323,7 @@ def irp_simple_statements(text):
     result = [
        Simple_line(i,"!",f),
        Simple_line(i,"! >>> %s"%(txt,),f ),
-       Provide_all(i,"      call %ser_%s('%s')"%(rw,variable,num),f),
+       Provide_all(i,"   call %ser_%s('%s')"%(rw,variable,num),f),
        Simple_line(i,"! >>> END %s "%(txt,),f ),
        Simple_line(line.i,"!",f),
     ]
@@ -340,7 +340,7 @@ def irp_simple_statements(text):
   def process_return(line):
     assert isinstance(line,Return)
     if command_line.do_assert or command_line.do_debug:
-      newline = Simple_line(line.i,"      call irp_leave(irp_here)",line.filename)
+      newline = Simple_line(line.i," call irp_leave(irp_here)",line.filename)
       result = [newline, line]
     else:
       result = [ line ]
@@ -357,8 +357,8 @@ def irp_simple_statements(text):
       matches = [ match.group(1).strip(), match.group(3).strip() ]
       for m in matches:
         if not(m.isdigit() or ("'" in m) or (m == "")):
-          result.append ( Simple_line (line.i, "       print *, '%s = ', %s"%(m,m), line.filename) )
-      result.append ( Simple_line (line.i, "       print *, ''", line.filename) )
+          result.append ( Simple_line (line.i, " print *, '%s = ', %s"%(m,m), line.filename) )
+      result.append ( Simple_line (line.i, " print *, ''", line.filename) )
     return result
    
   def process_assert(line):
@@ -374,14 +374,14 @@ def irp_simple_statements(text):
       result = [
        Simple_line(i, "!", f),
        Simple_line(i, "! >>> %s"%(txt,), f),
-       If         (i, "      if (.not.%s) then"%(condition,), f),
-       Simple_line(i, "       call irp_trace", f),
-       Simple_line(i, "       print *, irp_here//': Assert failed:'", f),
-       Simple_line(i, "       print *, ' file: %s, line: %d'"%(f,i), f),
-       Simple_line(i, "       print *, '%s'"%(condition_str,), f),
+       If         (i, "  if (.not.%s) then"%(condition,), f),
+       Simple_line(i, "   call irp_trace", f),
+       Simple_line(i, "   print *, irp_here//': Assert failed:'", f),
+       Simple_line(i, "   print *, ' file: %s, line: %d'"%(f,i), f),
+       Simple_line(i, "   print *, '%s'"%(condition_str,), f),
        ] + debug_conditions(line) + [
-       Simple_line(i, "       stop 1", f),
-       Endif      (i, "      endif", f),
+       Simple_line(i, "   stop 1", f),
+       Endif      (i, "  endif", f),
        Simple_line(i, "! <<< END %s"%(txt,), f),
        Simple_line(i, "!", f)
       ]
@@ -395,7 +395,7 @@ def irp_simple_statements(text):
       i = line.i
       f = line.filename
       result = [
-          Simple_line(i,"      call irp_leave(irp_here)", f),
+          Simple_line(i," call irp_leave(irp_here)", f),
           line
         ]
     else:
@@ -415,10 +415,10 @@ def irp_simple_statements(text):
     i = line.i
     f = line.filename
     result = [ line,
-      Declaration(i,"      character*(%d), parameter :: irp_here = '%s'"%(length,varname), f) ]
+      Declaration(i,"  character*(%d), parameter :: irp_here = '%s'"%(length,varname), f) ]
     if command_line.do_assert or command_line.do_debug:
       result += [
-        Simple_line(i,"      call irp_enter(irp_here)", f),
+        Simple_line(i,"  call irp_enter(irp_here)", f),
       ]
     return result
 
@@ -438,10 +438,10 @@ def irp_simple_statements(text):
     i = line.i
     f = line.filename
     result = [ line,
-      Declaration(i,"      character*(%d), parameter :: irp_here = '%s'"%(length,subname), f) ]
+      Declaration(i,"  character*(%d), parameter :: irp_here = '%s'"%(length,subname), f) ]
     if command_line.do_assert or command_line.do_debug:
       result += [
-        Simple_line(i,"      call irp_enter(irp_here)", f),
+        Simple_line(i,"  call irp_enter(irp_here)", f),
       ]
     return result
 
@@ -459,10 +459,10 @@ def irp_simple_statements(text):
     i = line.i
     f = line.filename
     result = [ line,
-      Declaration(i,"      character*(%d), parameter :: irp_here = '%s'"%(length,subname), f) ]
+      Declaration(i,"  character*(%d), parameter :: irp_here = '%s'"%(length,subname), f) ]
     if command_line.do_assert or command_line.do_debug:
       result += [
-        Simple_line(i,"      call irp_enter(irp_here)", f),
+        Simple_line(i,"  call irp_enter(irp_here)", f),
       ]
     return result
 
@@ -511,7 +511,7 @@ def change_includes(text):
      try:
        file = open(filename,'r')
        file.close()
-       result += preprocessed_text(filename)
+       result += create_preprocessed_text(filename)
       #result += get_text(file.readlines(), filename)
      except IOError:
        result.append(line)
@@ -531,7 +531,7 @@ def process_old_style_do(text):
          isinstance(line,Enddo):
         buffer = line.text.split()
         if buffer[0] == number:
-          text[i] = Enddo(line.i,"      enddo",line.filename)
+          text[i] = Enddo(line.i,"  enddo",line.filename)
           return
     error.fail(text[begin],"Old-style do loops should end with 'continue' or 'end do'")
       
@@ -543,7 +543,7 @@ def process_old_style_do(text):
       if buffer[1].isdigit():
         number = buffer.pop(1)
         change_matching_enddo(i,number)
-        line.text = "      "+" ".join(buffer)
+        line.text = " ".join(buffer)
     result.append(line)
   return result
 
@@ -591,7 +591,7 @@ endif'''
         f = line.filename
         result.append( If(i,"%s then"%(test,),f) )
         result += get_type(i,f,code,False)[0]
-        result.append( Endif(i,"      endif",f) )
+        result.append( Endif(i,"  endif",f) )
     else:
       result.append(line)
   return result
@@ -705,7 +705,7 @@ def move_to_top(text,t):
   return text
 
 ######################################################################
-def preprocessed_text(filename):
+def create_preprocessed_text(filename):
   file = open(filename,"r")
   lines = file.readlines()
   file.close()
@@ -724,13 +724,21 @@ def preprocessed_text(filename):
   result = move_to_top(result,Use)
   return result
 
-if __name__ == '__main__': 
-  txt = preprocessed_text('testfile.irp.f')
-  check_begin_end(txt)
-  for line in txt:
-    print line
-  txt = preprocessed_text('testfile_fixed.irp.f')
-  check_begin_end(txt)
-  for line in txt:
-    print line
+######################################################################
+preprocessed_text = []
+for filename in irpf90_files:
+   result = create_preprocessed_text(filename)
+   check_begin_end(result)
+   preprocessed_text.append( (filename, result) )
+
+######################################################################
+def debug():
+  for filename, txt in preprocessed_text:
+    print "=== "+filename+" ==="
+    for line in txt:
+      print line
+  print irpf90_files
+
+if __name__ == '__main__': 
+  debug()
 
diff --git a/src/util.py b/src/util.py
index 9d9ea91..3ae9534 100644
--- a/src/util.py
+++ b/src/util.py
@@ -1,5 +1,8 @@
 #!/usr/bin/python
 
+def strip(x):
+  return x.strip()
+
 def same_file(filename,txt):
   assert isinstance(filename,str)
   assert isinstance(txt,list)
@@ -17,8 +20,13 @@ def same_file(filename,txt):
       return False
   return True
 
+def build_dim(dim):
+  if len(dim) == 0:
+    return ""
+  else:
+    return "(%s)"%( ",".join(dim) )
+
 if __name__ == '__main__':
-  txt = open('/etc/passwd','r').readlines()
-  print same_file('/etc/passwd',txt)
-  print same_file('/etc/group',txt)
-  print same_file('/etc/passwd-',txt)
+  print build_dim([])
+  print build_dim(['a'])
+  print build_dim(['a','b'])
diff --git a/src/variable.py b/src/variable.py
new file mode 100644
index 0000000..db4aa7b
--- /dev/null
+++ b/src/variable.py
@@ -0,0 +1,168 @@
+#!/usr/bin/python
+
+from irpf90_t import *
+from util import *
+import error
+
+class Variable(object):
+
+  ############################################################
+  def __init__(self,text,name = None):
+    assert isinstance(text,list)
+    assert len(text) > 0
+    assert isinstance(text[0],Line)
+    self.text = text
+    if name is not None:
+      self._name = name.lower()
+
+  ############################################################
+  def name(self):
+    '''Name is lowercase'''
+    if '_name' not in self.__dict__:
+      buffer = None
+      for line in self.text:
+        if isinstance(line,Begin_provider):
+          buffer = line.text.replace(']',',').split(',')
+          break
+      assert buffer is not None
+      if len(buffer) < 3:
+        error.fail(line, "Error in Begin_provider line")
+      self._name = buffer[1].strip().lower()
+    return self._name
+  name = property(name)
+
+  ############################################################
+  def doc(self):
+    if '_doc' not in self.__dict__:
+      def f(l): return 
+      buffer = filter(lambda l:isinstance(l,Doc), self.text)
+      self._doc = map(lambda l: l.text[1:], buffer)
+      if buffer == []:
+        error.warn(None,"Variable %s is not documented"%(self.name))
+    return self._doc
+  doc = property(doc)
+
+  ############################################################
+  def others(self):
+    if '_others' not in self.__dict__:
+      result = []
+      def f(l):
+        return isinstance(l,Begin_provider) or isinstance(l,Cont_provider)
+      lines = filter(f, self.text)
+      for line in lines:
+        buffer = line.text.replace(']',',').split(',')
+        if len(buffer) < 3:
+          error.fail(line,"Syntax Error") 
+        buffer = buffer[1].strip().lower()
+        result.append(buffer)
+      result.remove(self.name)
+      self._others = result
+    return self._others
+  others = property(others)
+
+  ############################################################
+  def same_as(self):
+    if '_same_as' not in self.__dict__:
+      if isinstance(self.line,Begin_provider):
+        result = None
+      else:
+        buffer = self.text[0].text.replace(']',',').split(',')
+        if len(buffer) < 3:
+          error.fail(line,"Syntax Error")
+        result = buffer[1].strip().lower()
+      self._same_as = result
+    return self._same_as
+  same_as = property(same_as)
+
+  ############################################################
+  def allocate(self):
+    if '_allocate' not in self.__dict__:
+      from variables import variables
+      def f(var):
+        return variables[var].dim != []
+      self._allocate = filter ( f, self.others + [self.name] )
+    return self._allocate
+  allocate = property(allocate)
+
+  ############################################################
+  def dim(self):
+    if '_dim' not in self.__dict__:
+      line = self.line.text
+      buffer = line.replace(']','').split(',',2)
+      if len(buffer) == 2:
+        self._dim = []
+      else:
+        buffer = buffer[2].strip()[1:-1].split(',')
+        self._dim = map(strip,buffer)
+    return self._dim
+  dim = property(dim)
+
+  ############################################################
+  def type(self):
+    if '_type' not in self.__dict__:
+      line = self.line.text
+      buffer = line.split(',')[0]
+      buffer = buffer.split('[')[1].strip()
+      if self.dim != '':
+        buffer = "%s, allocatable"%(buffer)
+      self._type = buffer
+    return self._type
+  type = property(type)
+
+  ############################################################
+  def fmodule(self):
+    if '_fmodule' not in self.__dict__:
+      self._fmodule = self.line.filename.replace('.irp.f','_mod')
+    return self._fmodule
+  fmodule = property(fmodule)
+
+  ############################################################
+  def regexp(self):
+    if '_regexp' not in self.__dict__:
+      import re
+      self._regexp = re.compile( \
+        r"^.*[^a-z0-9'\"_]+%s([^a-z0-9_]|$)"%(self.name),re.I)
+    return self._regexp
+  regexp = property(regexp)
+
+  ############################################################
+  def line(self):
+    if '_line' not in self.__dict__:
+      def f(l):
+        return isinstance(l,Begin_provider) or isinstance(l,Cont_provider)
+      lines = filter(f, self.text)
+      for line in lines:
+        buffer = line.text.replace(']',',').split(',')
+        if len(buffer) < 3:
+          error.fail(line,"Syntax Error") 
+        buffer = buffer[1].strip().lower()
+        if self.name == buffer:
+          self._line = line
+          break
+    assert '_line' in self.__dict__
+    return self._line
+  line = property(line)
+
+  ############################################################
+  def needs(self):
+    if '_needs' not in self.__dict__:
+      self._needs = None
+    return self._needs
+  needs = property(needs)
+
+  ############################################################
+  def header(self):
+    if '_header' not in self.__dict__:
+      name = self.name
+      self._header = [
+        "  %s :: %s %s"%(self.type, name, build_dim(self.dim) ),
+        "  logical :: %s_is_built = .False."%(name),
+      ]
+    return self._header
+  header = property(header)
+
+######################################################################
+if __name__ == '__main__':
+  from preprocessed_text import preprocessed_text
+  from variables import variables
+  print variables['elec_fitcusp_lapl'].doc
diff --git a/src/variables.py b/src/variables.py
new file mode 100644
index 0000000..654207f
--- /dev/null
+++ b/src/variables.py
@@ -0,0 +1,30 @@
+#!/usr/bin/python
+
+from variable import *
+from irpf90_t 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()
+
+if __name__ == '__main__':
+  for v in variables.keys():
+    print v