| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- import os
- from distutils.msvc9compiler import MSVCCompiler as _MSVCCompiler
- from .system_info import platform_bits
- def _merge(old, new):
- """Concatenate two environment paths avoiding repeats.
- Here `old` is the environment string before the base class initialize
- function is called and `new` is the string after the call. The new string
- will be a fixed string if it is not obtained from the current environment,
- or the same as the old string if obtained from the same environment. The aim
- here is not to append the new string if it is already contained in the old
- string so as to limit the growth of the environment string.
- Parameters
- ----------
- old : string
- Previous environment string.
- new : string
- New environment string.
- Returns
- -------
- ret : string
- Updated environment string.
- """
- if not old:
- return new
- if new in old:
- return old
- # Neither new nor old is empty. Give old priority.
- return ';'.join([old, new])
- class MSVCCompiler(_MSVCCompiler):
- def __init__(self, verbose=0, dry_run=0, force=0):
- _MSVCCompiler.__init__(self, verbose, dry_run, force)
- def initialize(self, plat_name=None):
- # The 'lib' and 'include' variables may be overwritten
- # by MSVCCompiler.initialize, so save them for later merge.
- environ_lib = os.getenv('lib')
- environ_include = os.getenv('include')
- _MSVCCompiler.initialize(self, plat_name)
- # Merge current and previous values of 'lib' and 'include'
- os.environ['lib'] = _merge(environ_lib, os.environ['lib'])
- os.environ['include'] = _merge(environ_include, os.environ['include'])
- # msvc9 building for 32 bits requires SSE2 to work around a
- # compiler bug.
- if platform_bits == 32:
- self.compile_options += ['/arch:SSE2']
- self.compile_options_debug += ['/arch:SSE2']
- def manifest_setup_ldargs(self, output_filename, build_temp, ld_args):
- ld_args.append('/MANIFEST')
- _MSVCCompiler.manifest_setup_ldargs(self, output_filename,
- build_temp, ld_args)
|
