third_party/waf/waflib/extras/pgicc.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3 # Antoine Dechaume 2011
   4
   5 """
   6 Detect the PGI C compiler
   7 """
   8
   9 import sys, re
  10 from waflib import Errors
  11 from waflib.Configure import conf
  12 from waflib.Tools.compiler_c import c_compiler
  13 c_compiler['linux'].append('pgicc')
  14
  15 @conf
  16 def find_pgi_compiler(conf, var, name):
  17         """
  18         Find the program name, and execute it to ensure it really is itself.
  19         """
  20         if sys.platform == 'cygwin':
  21                 conf.fatal('The PGI compiler does not work on Cygwin')
  22
  23         v = conf.env
  24         cc = None
  25         if v[var]:
  26                 cc = v[var]
  27         elif var in conf.environ:
  28                 cc = conf.environ[var]
  29         if not cc:
  30                 cc = conf.find_program(name, var=var)
  31         if not cc:
  32                 conf.fatal('PGI Compiler (%s) was not found' % name)
  33
  34         v[var + '_VERSION'] = conf.get_pgi_version(cc)
  35         v[var] = cc
  36         v[var + '_NAME'] = 'pgi'
  37
  38 @conf
  39 def get_pgi_version(conf, cc):
  40         """Find the version of a pgi compiler."""
  41         version_re = re.compile(r"The Portland Group", re.I).search
  42         cmd = cc + ['-V', '-E'] # Issue 1078, prevent wrappers from linking
  43
  44         try:
  45                 out, err = conf.cmd_and_log(cmd, output=0)
  46         except Errors.WafError:
  47                 conf.fatal('Could not find pgi compiler %r' % cmd)
  48
  49         if out:
  50                 match = version_re(out)
  51         else:
  52                 match = version_re(err)
  53
  54         if not match:
  55                 conf.fatal('Could not verify PGI signature')
  56
  57         cmd = cc + ['-help=variable']
  58         try:
  59                 out, err = conf.cmd_and_log(cmd, output=0)
  60         except Errors.WafError:
  61                 conf.fatal('Could not find pgi compiler %r' % cmd)
  62
  63         version = re.findall(r'^COMPVER\s*=(.*)', out, re.M)
  64         if len(version) != 1:
  65                 conf.fatal('Could not determine the compiler version')
  66         return version[0]
  67
  68 def configure(conf):
  69         conf.find_pgi_compiler('CC', 'pgcc')
  70         conf.find_ar()
  71         conf.gcc_common_flags()
  72         conf.cc_load_tools()
  73         conf.cc_add_flags()
  74         conf.link_add_flags()
  75