test/Fortran/FORTRANPPFILESUFFIXES.py

   1 #!/usr/bin/env python
   2 #
   3 # __COPYRIGHT__
   4 #
   5 # Permission is hereby granted, free of charge, to any person obtaining
   6 # a copy of this software and associated documentation files (the
   7 # "Software"), to deal in the Software without restriction, including
   8 # without limitation the rights to use, copy, modify, merge, publish,
   9 # distribute, sublicense, and/or sell copies of the Software, and to
  10 # permit persons to whom the Software is furnished to do so, subject to
  11 # the following conditions:
  12 #
  13 # The above copyright notice and this permission notice shall be included
  14 # in all copies or substantial portions of the Software.
  15 #
  16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
  17 # KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  18 # WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  19 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  20 # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  21 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  22 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23 #
  24 """ Test manipulating FORTRANPPFILESUFFIXES. """
  25
  26 __revision__ = "__FILE__ __REVISION__ __DATE__ __DEVELOPER__"
  27
  28 import TestSCons
  29
  30
  31
  32 _python_ = TestSCons._python_
  33 _exe   = TestSCons._exe
  34
  35 test = TestSCons.TestSCons()
  36
  37 test.file_fixture('mylink.py')
  38
  39 test.write('myfortran.py', r"""
  40 import getopt
  41 import sys
  42
  43 comment = ('#' + sys.argv[1]).encode()
  44 length = len(comment)
  45 args = sys.argv[2:]
  46 # First parse defines, since getopt won't have it
  47 defines = []
  48 for a in args[:]:
  49     if a.startswith("-D") or a.startswith("/D"):
  50         defines.append(a[2:])
  51         args.remove(a)
  52
  53 opts, args = getopt.getopt(args, 'co:')
  54 for opt, arg in opts:
  55     if opt == '-o': out = arg
  56 infile = open(args[0], 'rb')
  57 outfile = open(out, 'wb')
  58 for d in defines:
  59     outfile.write(("#define %s\n" % (d,)).encode())
  60 for l in infile.readlines():
  61     if l[:length] != comment:
  62         outfile.write(l)
  63 sys.exit(0)
  64 """)
  65
  66 # Test non default file suffix: .f, .f90 and .f95 for FORTRAN
  67 test.write('SConstruct', """
  68 env = Environment(LINK=r'%(_python_)s mylink.py',
  69                   LINKFLAGS=[],
  70                   CPPDEFINES=["mosdef"],
  71                   F77=r'%(_python_)s myfortran.py g77',
  72                   FORTRAN=r'%(_python_)s myfortran.py fortran',
  73                   FORTRANPPFILESUFFIXES=['.f', '.fpp'],
  74                   tools=['default', 'fortran'])
  75 env.Program(target = 'test01', source = 'test01.f')
  76 env.Program(target = 'test02', source = 'test02.fpp')
  77 """ % locals())
  78
  79 test.write('test01.f',   "This is a .f file.\n#link\n#fortran\n")
  80 test.write('test02.fpp',   "This is a .fpp file.\n#link\n#fortran\n")
  81
  82 test.run(arguments='.', stderr=None)
  83
  84 test.must_match('test01' + _exe, "#define mosdef\nThis is a .f file.\n")
  85 test.must_match('test02' + _exe, "#define mosdef\nThis is a .fpp file.\n")
  86
  87 test.pass_test()