home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / CCOMPILER.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  41.8 KB  |  1,008 lines
  1. """distutils.ccompiler
  2.  
  3. Contains CCompiler, an abstract base class that defines the interface
  4. for the Distutils compiler abstraction model."""
  5.  
  6. # created 1999/07/05, Greg Ward
  7.  
  8. __revision__ = "$Id: ccompiler.py,v 1.35 2000/09/27 02:24:21 gward Exp $"
  9.  
  10. import sys, os
  11. from types import *
  12. from copy import copy
  13. from distutils.errors import *
  14. from distutils.spawn import spawn
  15. from distutils.file_util import move_file
  16. from distutils.dir_util import mkpath
  17. from distutils.dep_util import newer_pairwise, newer_group
  18. from distutils.util import split_quoted, execute
  19.  
  20.  
  21. class CCompiler:
  22.     """Abstract base class to define the interface that must be implemented
  23.     by real compiler classes.  Also has some utility methods used by
  24.     several compiler classes.
  25.  
  26.     The basic idea behind a compiler abstraction class is that each
  27.     instance can be used for all the compile/link steps in building a
  28.     single project.  Thus, attributes common to all of those compile and
  29.     link steps -- include directories, macros to define, libraries to link
  30.     against, etc. -- are attributes of the compiler instance.  To allow for
  31.     variability in how individual files are treated, most of those
  32.     attributes may be varied on a per-compilation or per-link basis.
  33.     """
  34.  
  35.     # 'compiler_type' is a class attribute that identifies this class.  It
  36.     # keeps code that wants to know what kind of compiler it's dealing with
  37.     # from having to import all possible compiler classes just to do an
  38.     # 'isinstance'.  In concrete CCompiler subclasses, 'compiler_type'
  39.     # should really, really be one of the keys of the 'compiler_class'
  40.     # dictionary (see below -- used by the 'new_compiler()' factory
  41.     # function) -- authors of new compiler interface classes are
  42.     # responsible for updating 'compiler_class'!
  43.     compiler_type = None
  44.  
  45.     # XXX things not handled by this compiler abstraction model:
  46.     #   * client can't provide additional options for a compiler,
  47.     #     e.g. warning, optimization, debugging flags.  Perhaps this
  48.     #     should be the domain of concrete compiler abstraction classes
  49.     #     (UnixCCompiler, MSVCCompiler, etc.) -- or perhaps the base
  50.     #     class should have methods for the common ones.
  51.     #   * can't completely override the include or library searchg
  52.     #     path, ie. no "cc -I -Idir1 -Idir2" or "cc -L -Ldir1 -Ldir2".
  53.     #     I'm not sure how widely supported this is even by Unix
  54.     #     compilers, much less on other platforms.  And I'm even less
  55.     #     sure how useful it is; maybe for cross-compiling, but
  56.     #     support for that is a ways off.  (And anyways, cross
  57.     #     compilers probably have a dedicated binary with the
  58.     #     right paths compiled in.  I hope.)
  59.     #   * can't do really freaky things with the library list/library
  60.     #     dirs, e.g. "-Ldir1 -lfoo -Ldir2 -lfoo" to link against
  61.     #     different versions of libfoo.a in different locations.  I
  62.     #     think this is useless without the ability to null out the
  63.     #     library search path anyways.
  64.     
  65.  
  66.     # Subclasses that rely on the standard filename generation methods
  67.     # implemented below should override these; see the comment near
  68.     # those methods ('object_filenames()' et. al.) for details:
  69.     src_extensions = None               # list of strings
  70.     obj_extension = None                # string
  71.     static_lib_extension = None
  72.     shared_lib_extension = None         # string
  73.     static_lib_format = None            # format string
  74.     shared_lib_format = None            # prob. same as static_lib_format
  75.     exe_extension = None                # string
  76.  
  77.  
  78.     def __init__ (self,
  79.                   verbose=0,
  80.                   dry_run=0,
  81.                   force=0):
  82.  
  83.         self.verbose = verbose
  84.         self.dry_run = dry_run
  85.         self.force = force
  86.  
  87.         # 'output_dir': a common output directory for object, library,
  88.         # shared object, and shared library files
  89.         self.output_dir = None
  90.  
  91.         # 'macros': a list of macro definitions (or undefinitions).  A
  92.         # macro definition is a 2-tuple (name, value), where the value is
  93.         # either a string or None (no explicit value).  A macro
  94.         # undefinition is a 1-tuple (name,).
  95.         self.macros = []
  96.  
  97.         # 'include_dirs': a list of directories to search for include files
  98.         self.include_dirs = []
  99.  
  100.         # 'libraries': a list of libraries to include in any link
  101.         # (library names, not filenames: eg. "foo" not "libfoo.a")
  102.         self.libraries = []
  103.  
  104.         # 'library_dirs': a list of directories to search for libraries
  105.         self.library_dirs = []
  106.  
  107.         # 'runtime_library_dirs': a list of directories to search for
  108.         # shared libraries/objects at runtime
  109.         self.runtime_library_dirs = []
  110.  
  111.         # 'objects': a list of object files (or similar, such as explicitly
  112.         # named library files) to include on any link
  113.         self.objects = []
  114.  
  115.         for key in self.executables.keys():
  116.             self.set_executable(key, self.executables[key])
  117.  
  118.     # __init__ ()
  119.  
  120.  
  121.     def set_executables (self, **args):
  122.  
  123.         """Define the executables (and options for them) that will be run
  124.         to perform the various stages of compilation.  The exact set of
  125.         executables that may be specified here depends on the compiler
  126.         class (via the 'executables' class attribute), but most will have:
  127.           compiler      the C/C++ compiler
  128.           linker_so     linker used to create shared objects and libraries
  129.           linker_exe    linker used to create binary executables
  130.           archiver      static library creator
  131.  
  132.         On platforms with a command-line (Unix, DOS/Windows), each of these
  133.         is a string that will be split into executable name and (optional)
  134.         list of arguments.  (Splitting the string is done similarly to how
  135.         Unix shells operate: words are delimited by spaces, but quotes and
  136.         backslashes can override this.  See
  137.         'distutils.util.split_quoted()'.)
  138.         """
  139.  
  140.         # Note that some CCompiler implementation classes will define class
  141.         # attributes 'cpp', 'cc', etc. with hard-coded executable names;
  142.         # this is appropriate when a compiler class is for exactly one
  143.         # compiler/OS combination (eg. MSVCCompiler).  Other compiler
  144.         # classes (UnixCCompiler, in particular) are driven by information
  145.         # discovered at run-time, since there are many different ways to do
  146.         # basically the same things with Unix C compilers.
  147.  
  148.         for key in args.keys():
  149.             if not self.executables.has_key(key):
  150.                 raise ValueError, \
  151.                       "unknown executable '%s' for class %s" % \
  152.                       (key, self.__class__.__name__)
  153.             self.set_executable(key, args[key])
  154.  
  155.     # set_executables ()
  156.  
  157.     def set_executable(self, key, value):
  158.         if type(value) is StringType:
  159.             setattr(self, key, split_quoted(value))
  160.         else:
  161.             setattr(self, key, value)
  162.         
  163.  
  164.  
  165.     def _find_macro (self, name):
  166.         i = 0
  167.         for defn in self.macros:
  168.             if defn[0] == name:
  169.                 return i
  170.             i = i + 1
  171.  
  172.         return None
  173.  
  174.  
  175.     def _check_macro_definitions (self, definitions):
  176.         """Ensures that every element of 'definitions' is a valid macro
  177.         definition, ie. either (name,value) 2-tuple or a (name,) tuple.  Do
  178.         nothing if all definitions are OK, raise TypeError otherwise.
  179.         """
  180.         for defn in definitions:
  181.             if not (type (defn) is TupleType and
  182.                     (len (defn) == 1 or
  183.                      (len (defn) == 2 and
  184.                       (type (defn[1]) is StringType or defn[1] is None))) and
  185.                     type (defn[0]) is StringType):
  186.                 raise TypeError, \
  187.                       ("invalid macro definition '%s': " % defn) + \
  188.                       "must be tuple (string,), (string, string), or " + \
  189.                       "(string, None)"
  190.  
  191.  
  192.     # -- Bookkeeping methods -------------------------------------------
  193.  
  194.     def define_macro (self, name, value=None):
  195.         """Define a preprocessor macro for all compilations driven by this
  196.         compiler object.  The optional parameter 'value' should be a
  197.         string; if it is not supplied, then the macro will be defined
  198.         without an explicit value and the exact outcome depends on the
  199.         compiler used (XXX true? does ANSI say anything about this?)
  200.         """
  201.         # Delete from the list of macro definitions/undefinitions if
  202.         # already there (so that this one will take precedence).
  203.         i = self._find_macro (name)
  204.         if i is not None:
  205.             del self.macros[i]
  206.  
  207.         defn = (name, value)
  208.         self.macros.append (defn)
  209.  
  210.  
  211.     def undefine_macro (self, name):
  212.         """Undefine a preprocessor macro for all compilations driven by
  213.         this compiler object.  If the same macro is defined by
  214.         'define_macro()' and undefined by 'undefine_macro()' the last call
  215.         takes precedence (including multiple redefinitions or
  216.         undefinitions).  If the macro is redefined/undefined on a
  217.         per-compilation basis (ie. in the call to 'compile()'), then that
  218.         takes precedence.
  219.         """
  220.         # Delete from the list of macro definitions/undefinitions if
  221.         # already there (so that this one will take precedence).
  222.         i = self._find_macro (name)
  223.         if i is not None:
  224.             del self.macros[i]
  225.  
  226.         undefn = (name,)
  227.         self.macros.append (undefn)
  228.  
  229.  
  230.     def add_include_dir (self, dir):
  231.         """Add 'dir' to the list of directories that will be searched for
  232.         header files.  The compiler is instructed to search directories in
  233.         the order in which they are supplied by successive calls to
  234.         'add_include_dir()'.
  235.         """
  236.         self.include_dirs.append (dir)
  237.  
  238.     def set_include_dirs (self, dirs):
  239.         """Set the list of directories that will be searched to 'dirs' (a
  240.         list of strings).  Overrides any preceding calls to
  241.         'add_include_dir()'; subsequence calls to 'add_include_dir()' add
  242.         to the list passed to 'set_include_dirs()'.  This does not affect
  243.         any list of standard include directories that the compiler may
  244.         search by default.
  245.         """
  246.         self.include_dirs = copy (dirs)
  247.  
  248.  
  249.     def add_library (self, libname):
  250.         """Add 'libname' to the list of libraries that will be included in
  251.         all links driven by this compiler object.  Note that 'libname'
  252.         should *not* be the name of a file containing a library, but the
  253.         name of the library itself: the actual filename will be inferred by
  254.         the linker, the compiler, or the compiler class (depending on the
  255.         platform).
  256.  
  257.         The linker will be instructed to link against libraries in the
  258.         order they were supplied to 'add_library()' and/or
  259.         'set_libraries()'.  It is perfectly valid to duplicate library
  260.         names; the linker will be instructed to link against libraries as
  261.         many times as they are mentioned.
  262.         """
  263.         self.libraries.append (libname)
  264.  
  265.     def set_libraries (self, libnames):
  266.         """Set the list of libraries to be included in all links driven by
  267.         this compiler object to 'libnames' (a list of strings).  This does
  268.         not affect any standard system libraries that the linker may
  269.         include by default.
  270.         """
  271.         self.libraries = copy (libnames)
  272.  
  273.  
  274.     def add_library_dir (self, dir):
  275.         """Add 'dir' to the list of directories that will be searched for
  276.         libraries specified to 'add_library()' and 'set_libraries()'.  The
  277.         linker will be instructed to search for libraries in the order they
  278.         are supplied to 'add_library_dir()' and/or 'set_library_dirs()'.
  279.         """
  280.         self.library_dirs.append (dir)
  281.  
  282.     def set_library_dirs (self, dirs):
  283.         """Set the list of library search directories to 'dirs' (a list of
  284.         strings).  This does not affect any standard library search path
  285.         that the linker may search by default.
  286.         """
  287.         self.library_dirs = copy (dirs)
  288.  
  289.  
  290.     def add_runtime_library_dir (self, dir):
  291.         """Add 'dir' to the list of directories that will be searched for
  292.         shared libraries at runtime.
  293.         """
  294.         self.runtime_library_dirs.append (dir)
  295.  
  296.     def set_runtime_library_dirs (self, dirs):
  297.         """Set the list of directories to search for shared libraries at
  298.         runtime to 'dirs' (a list of strings).  This does not affect any
  299.         standard search path that the runtime linker may search by
  300.         default.
  301.         """
  302.         self.runtime_library_dirs = copy (dirs)
  303.  
  304.  
  305.     def add_link_object (self, object):
  306.         """Add 'object' to the list of object files (or analogues, such as
  307.         explicitly named library files or the output of "resource
  308.         compilers") to be included in every link driven by this compiler
  309.         object.
  310.         """
  311.         self.objects.append (object)
  312.  
  313.     def set_link_objects (self, objects):
  314.         """Set the list of object files (or analogues) to be included in
  315.         every link to 'objects'.  This does not affect any standard object
  316.         files that the linker may include by default (such as system
  317.         libraries).
  318.         """
  319.         self.objects = copy (objects)
  320.  
  321.  
  322.     # -- Priviate utility methods --------------------------------------
  323.     # (here for the convenience of subclasses)
  324.  
  325.     def _fix_compile_args (self, output_dir, macros, include_dirs):
  326.         """Typecheck and fix-up some of the arguments to the 'compile()'
  327.         method, and return fixed-up values.  Specifically: if 'output_dir'
  328.         is None, replaces it with 'self.output_dir'; ensures that 'macros'
  329.         is a list, and augments it with 'self.macros'; ensures that
  330.         'include_dirs' is a list, and augments it with 'self.include_dirs'.
  331.         Guarantees that the returned values are of the correct type,
  332.         i.e. for 'output_dir' either string or None, and for 'macros' and
  333.         'include_dirs' either list or None.
  334.         """
  335.         if output_dir is None:
  336.             output_dir = self.output_dir
  337.         elif type (output_dir) is not StringType:
  338.             raise TypeError, "'output_dir' must be a string or None"
  339.  
  340.         if macros is None:
  341.             macros = self.macros
  342.         elif type (macros) is ListType:
  343.             macros = macros + (self.macros or [])
  344.         else:
  345.             raise TypeError, \
  346.                   "'macros' (if supplied) must be a list of tuples"
  347.  
  348.         if include_dirs is None:
  349.             include_dirs = self.include_dirs
  350.         elif type (include_dirs) in (ListType, TupleType):
  351.             include_dirs = list (include_dirs) + (self.include_dirs or [])
  352.         else:
  353.             raise TypeError, \
  354.                   "'include_dirs' (if supplied) must be a list of strings"
  355.                     
  356.         return (output_dir, macros, include_dirs)
  357.  
  358.     # _fix_compile_args ()
  359.  
  360.  
  361.     def _prep_compile (self, sources, output_dir):
  362.         """Determine the list of object files corresponding to 'sources',
  363.         and figure out which ones really need to be recompiled.  Return a
  364.         list of all object files and a dictionary telling which source
  365.         files can be skipped.
  366.         """
  367.         # Get the list of expected output (object) files 
  368.         objects = self.object_filenames (sources,
  369.                                          output_dir=output_dir)
  370.  
  371.         if self.force:
  372.             skip_source = {}            # rebuild everything
  373.             for source in sources:
  374.                 skip_source[source] = 0
  375.         else:
  376.             # Figure out which source files we have to recompile according
  377.             # to a simplistic check -- we just compare the source and
  378.             # object file, no deep dependency checking involving header
  379.             # files.
  380.             skip_source = {}            # rebuild everything
  381.             for source in sources:      # no wait, rebuild nothing
  382.                 skip_source[source] = 1
  383.  
  384.             (n_sources, n_objects) = newer_pairwise (sources, objects)
  385.             for source in n_sources:    # no really, only rebuild what's
  386.                 skip_source[source] = 0 # out-of-date
  387.  
  388.         return (objects, skip_source)
  389.  
  390.     # _prep_compile ()
  391.  
  392.  
  393.     def _fix_object_args (self, objects, output_dir):
  394.         """Typecheck and fix up some arguments supplied to various methods.
  395.         Specifically: ensure that 'objects' is a list; if output_dir is
  396.         None, replace with self.output_dir.  Return fixed versions of
  397.         'objects' and 'output_dir'.
  398.         """
  399.         if type (objects) not in (ListType, TupleType):
  400.             raise TypeError, \
  401.                   "'objects' must be a list or tuple of strings"
  402.         objects = list (objects)
  403.             
  404.         if output_dir is None:
  405.             output_dir = self.output_dir
  406.         elif type (output_dir) is not StringType:
  407.             raise TypeError, "'output_dir' must be a string or None"
  408.  
  409.         return (objects, output_dir)
  410.  
  411.  
  412.     def _fix_lib_args (self, libraries, library_dirs, runtime_library_dirs):
  413.         """Typecheck and fix up some of the arguments supplied to the
  414.         'link_*' methods.  Specifically: ensure that all arguments are
  415.         lists, and augment them with their permanent versions
  416.         (eg. 'self.libraries' augments 'libraries').  Return a tuple with
  417.         fixed versions of all arguments.
  418.         """
  419.         if libraries is None:
  420.             libraries = self.libraries
  421.         elif type (libraries) in (ListType, TupleType):
  422.             libraries = list (libraries) + (self.libraries or [])
  423.         else:
  424.             raise TypeError, \
  425.                   "'libraries' (if supplied) must be a list of strings"
  426.  
  427.         if library_dirs is None:
  428.             library_dirs = self.library_dirs
  429.         elif type (library_dirs) in (ListType, TupleType):
  430.             library_dirs = list (library_dirs) + (self.library_dirs or [])
  431.         else:
  432.             raise TypeError, \
  433.                   "'library_dirs' (if supplied) must be a list of strings"
  434.  
  435.         if runtime_library_dirs is None:
  436.             runtime_library_dirs = self.runtime_library_dirs
  437.         elif type (runtime_library_dirs) in (ListType, TupleType):
  438.             runtime_library_dirs = (list (runtime_library_dirs) +
  439.                                     (self.runtime_library_dirs or []))
  440.         else:
  441.             raise TypeError, \
  442.                   "'runtime_library_dirs' (if supplied) " + \
  443.                   "must be a list of strings"
  444.  
  445.         return (libraries, library_dirs, runtime_library_dirs)
  446.  
  447.     # _fix_lib_args ()
  448.  
  449.  
  450.     def _need_link (self, objects, output_file):
  451.         """Return true if we need to relink the files listed in 'objects'
  452.         to recreate 'output_file'.
  453.         """
  454.         if self.force:
  455.             return 1
  456.         else:
  457.             if self.dry_run:
  458.                 newer = newer_group (objects, output_file, missing='newer')
  459.             else:
  460.                 newer = newer_group (objects, output_file)
  461.             return newer
  462.  
  463.     # _need_link ()
  464.  
  465.  
  466.     # -- Worker methods ------------------------------------------------
  467.     # (must be implemented by subclasses)
  468.  
  469.     def preprocess (self,
  470.                     source,
  471.                     output_file=None,
  472.                     macros=None,
  473.                     include_dirs=None,
  474.                     extra_preargs=None,
  475.                     extra_postargs=None):
  476.         """Preprocess a single C/C++ source file, named in 'source'.
  477.         Output will be written to file named 'output_file', or stdout if
  478.         'output_file' not supplied.  'macros' is a list of macro
  479.         definitions as for 'compile()', which will augment the macros set
  480.         with 'define_macro()' and 'undefine_macro()'.  'include_dirs' is a
  481.         list of directory names that will be added to the default list.
  482.  
  483.         Raises PreprocessError on failure.
  484.         """
  485.         pass
  486.  
  487.     def compile (self,
  488.                  sources,
  489.                  output_dir=None,
  490.                  macros=None,
  491.                  include_dirs=None,
  492.                  debug=0,
  493.                  extra_preargs=None,
  494.                  extra_postargs=None):
  495.         """Compile one or more source files.  'sources' must be a list of
  496.         filenames, most likely C/C++ files, but in reality anything that
  497.         can be handled by a particular compiler and compiler class
  498.         (eg. MSVCCompiler can handle resource files in 'sources').  Return
  499.         a list of object filenames, one per source filename in 'sources'.
  500.         Depending on the implementation, not all source files will
  501.         necessarily be compiled, but all corresponding object filenames
  502.         will be returned.
  503.  
  504.         If 'output_dir' is given, object files will be put under it, while
  505.         retaining their original path component.  That is, "foo/bar.c"
  506.         normally compiles to "foo/bar.o" (for a Unix implementation); if
  507.         'output_dir' is "build", then it would compile to
  508.         "build/foo/bar.o".
  509.  
  510.         'macros', if given, must be a list of macro definitions.  A macro
  511.         definition is either a (name, value) 2-tuple or a (name,) 1-tuple.
  512.         The former defines a macro; if the value is None, the macro is
  513.         defined without an explicit value.  The 1-tuple case undefines a
  514.         macro.  Later definitions/redefinitions/ undefinitions take
  515.         precedence.
  516.  
  517.         'include_dirs', if given, must be a list of strings, the
  518.         directories to add to the default include file search path for this
  519.         compilation only.
  520.  
  521.         'debug' is a boolean; if true, the compiler will be instructed to
  522.         output debug symbols in (or alongside) the object file(s).
  523.  
  524.         'extra_preargs' and 'extra_postargs' are implementation- dependent.
  525.         On platforms that have the notion of a command-line (e.g. Unix,
  526.         DOS/Windows), they are most likely lists of strings: extra
  527.         command-line arguments to prepand/append to the compiler command
  528.         line.  On other platforms, consult the implementation class
  529.         documentation.  In any event, they are intended as an escape hatch
  530.         for those occasions when the abstract compiler framework doesn't
  531.         cut the mustard.
  532.  
  533.         Raises CompileError on failure.
  534.         """
  535.         pass
  536.  
  537.  
  538.     def create_static_lib (self,
  539.                            objects,
  540.                            output_libname,
  541.                            output_dir=None,
  542.                            debug=0):
  543.         """Link a bunch of stuff together to create a static library file.
  544.         The "bunch of stuff" consists of the list of object files supplied
  545.         as 'objects', the extra object files supplied to
  546.         'add_link_object()' and/or 'set_link_objects()', the libraries
  547.         supplied to 'add_library()' and/or 'set_libraries()', and the
  548.         libraries supplied as 'libraries' (if any).
  549.  
  550.         'output_libname' should be a library name, not a filename; the
  551.         filename will be inferred from the library name.  'output_dir' is
  552.         the directory where the library file will be put.
  553.  
  554.         'debug' is a boolean; if true, debugging information will be
  555.         included in the library (note that on most platforms, it is the
  556.         compile step where this matters: the 'debug' flag is included here
  557.         just for consistency).
  558.  
  559.         Raises LibError on failure.
  560.         """
  561.         pass
  562.     
  563.  
  564.     # values for target_desc parameter in link()
  565.     SHARED_OBJECT = "shared_object"
  566.     SHARED_LIBRARY = "shared_library"
  567.     EXECUTABLE = "executable"
  568.  
  569.     def link (self,
  570.               target_desc,
  571.               objects,
  572.               output_filename,
  573.               output_dir=None,
  574.               libraries=None,
  575.               library_dirs=None,
  576.               runtime_library_dirs=None,
  577.               export_symbols=None,
  578.               debug=0,
  579.               extra_preargs=None,
  580.               extra_postargs=None,
  581.               build_temp=None):
  582.         """Link a bunch of stuff together to create an executable or
  583.         shared library file.
  584.  
  585.         The "bunch of stuff" consists of the list of object files supplied
  586.         as 'objects'.  'output_filename' should be a filename.  If
  587.         'output_dir' is supplied, 'output_filename' is relative to it
  588.         (i.e. 'output_filename' can provide directory components if
  589.         needed).
  590.  
  591.         'libraries' is a list of libraries to link against.  These are
  592.         library names, not filenames, since they're translated into
  593.         filenames in a platform-specific way (eg. "foo" becomes "libfoo.a"
  594.         on Unix and "foo.lib" on DOS/Windows).  However, they can include a
  595.         directory component, which means the linker will look in that
  596.         specific directory rather than searching all the normal locations.
  597.  
  598.         'library_dirs', if supplied, should be a list of directories to
  599.         search for libraries that were specified as bare library names
  600.         (ie. no directory component).  These are on top of the system
  601.         default and those supplied to 'add_library_dir()' and/or
  602.         'set_library_dirs()'.  'runtime_library_dirs' is a list of
  603.         directories that will be embedded into the shared library and used
  604.         to search for other shared libraries that *it* depends on at
  605.         run-time.  (This may only be relevant on Unix.)
  606.  
  607.         'export_symbols' is a list of symbols that the shared library will
  608.         export.  (This appears to be relevant only on Windows.)
  609.  
  610.         'debug' is as for 'compile()' and 'create_static_lib()', with the
  611.         slight distinction that it actually matters on most platforms (as
  612.         opposed to 'create_static_lib()', which includes a 'debug' flag
  613.         mostly for form's sake).
  614.  
  615.         'extra_preargs' and 'extra_postargs' are as for 'compile()' (except
  616.         of course that they supply command-line arguments for the
  617.         particular linker being used).
  618.  
  619.         Raises LinkError on failure.
  620.         """
  621.         raise NotImplementedError
  622.  
  623.     
  624.     # Old 'link_*()' methods, rewritten to use the new 'link()' method.
  625.  
  626.     def link_shared_lib (self,
  627.                          objects,
  628.                          output_libname,
  629.                          output_dir=None,
  630.                          libraries=None,
  631.                          library_dirs=None,
  632.                          runtime_library_dirs=None,
  633.                          export_symbols=None,
  634.                          debug=0,
  635.                          extra_preargs=None,
  636.                          extra_postargs=None,
  637.                          build_temp=None):
  638.         self.link(CCompiler.SHARED_LIBRARY, objects, 
  639.                   self.library_filename(output_libname, lib_type='shared'),
  640.                   output_dir,
  641.                   libraries, library_dirs, runtime_library_dirs,
  642.                   export_symbols, debug,
  643.                   extra_preargs, extra_postargs, build_temp)
  644.     
  645.  
  646.     def link_shared_object (self,
  647.                             objects,
  648.                             output_filename,
  649.                             output_dir=None,
  650.                             libraries=None,
  651.                             library_dirs=None,
  652.                             runtime_library_dirs=None,
  653.                             export_symbols=None,
  654.                             debug=0,
  655.                             extra_preargs=None,
  656.                             extra_postargs=None,
  657.                             build_temp=None):
  658.         self.link(CCompiler.SHARED_OBJECT, objects,
  659.                   output_filename, output_dir,
  660.                   libraries, library_dirs, runtime_library_dirs,
  661.                   export_symbols, debug,
  662.                   extra_preargs, extra_postargs, build_temp)
  663.  
  664.  
  665.     def link_executable (self,
  666.                          objects,
  667.                          output_progname,
  668.                          output_dir=None,
  669.                          libraries=None,
  670.                          library_dirs=None,
  671.                          runtime_library_dirs=None,
  672.                          debug=0,
  673.                          extra_preargs=None,
  674.                          extra_postargs=None):
  675.         self.link(CCompiler.EXECUTABLE, objects, 
  676.                   self.executable_filename(output_progname), output_dir,
  677.                   libraries, library_dirs, runtime_library_dirs, None, 
  678.                   debug, extra_preargs, extra_postargs, None)
  679.  
  680.  
  681.     # -- Miscellaneous methods -----------------------------------------
  682.     # These are all used by the 'gen_lib_options() function; there is
  683.     # no appropriate default implementation so subclasses should
  684.     # implement all of these.
  685.  
  686.     def library_dir_option (self, dir):
  687.         """Return the compiler option to add 'dir' to the list of
  688.         directories searched for libraries.
  689.         """
  690.         raise NotImplementedError
  691.  
  692.     def runtime_library_dir_option (self, dir):
  693.         """Return the compiler option to add 'dir' to the list of
  694.         directories searched for runtime libraries.
  695.         """
  696.         raise NotImplementedError
  697.  
  698.     def library_option (self, lib):
  699.         """Return the compiler option to add 'dir' to the list of libraries
  700.         linked into the shared library or executable.
  701.         """
  702.         raise NotImplementedError
  703.  
  704.     def find_library_file (self, dirs, lib, debug=0):
  705.         """Search the specified list of directories for a static or shared
  706.         library file 'lib' and return the full path to that file.  If
  707.         'debug' true, look for a debugging version (if that makes sense on
  708.         the current platform).  Return None if 'lib' wasn't found in any of
  709.         the specified directories.
  710.         """
  711.         raise NotImplementedError
  712.  
  713.  
  714.     # -- Filename generation methods -----------------------------------
  715.  
  716.     # The default implementation of the filename generating methods are
  717.     # prejudiced towards the Unix/DOS/Windows view of the world:
  718.     #   * object files are named by replacing the source file extension
  719.     #     (eg. .c/.cpp -> .o/.obj)
  720.     #   * library files (shared or static) are named by plugging the
  721.     #     library name and extension into a format string, eg.
  722.     #     "lib%s.%s" % (lib_name, ".a") for Unix static libraries
  723.     #   * executables are named by appending an extension (possibly
  724.     #     empty) to the program name: eg. progname + ".exe" for
  725.     #     Windows
  726.     #
  727.     # To reduce redundant code, these methods expect to find
  728.     # several attributes in the current object (presumably defined
  729.     # as class attributes):
  730.     #   * src_extensions -
  731.     #     list of C/C++ source file extensions, eg. ['.c', '.cpp']
  732.     #   * obj_extension -
  733.     #     object file extension, eg. '.o' or '.obj'
  734.     #   * static_lib_extension -
  735.     #     extension for static library files, eg. '.a' or '.lib'
  736.     #   * shared_lib_extension -
  737.     #     extension for shared library/object files, eg. '.so', '.dll'
  738.     #   * static_lib_format -
  739.     #     format string for generating static library filenames,
  740.     #     eg. 'lib%s.%s' or '%s.%s'
  741.     #   * shared_lib_format
  742.     #     format string for generating shared library filenames
  743.     #     (probably same as static_lib_format, since the extension
  744.     #     is one of the intended parameters to the format string)
  745.     #   * exe_extension -
  746.     #     extension for executable files, eg. '' or '.exe'
  747.  
  748.     def object_filenames (self,
  749.                           source_filenames,
  750.                           strip_dir=0,
  751.                           output_dir=''):
  752.         if output_dir is None: output_dir = ''
  753.         obj_names = []
  754.         for src_name in source_filenames:
  755.             (base, ext) = os.path.splitext (src_name)
  756.             if ext not in self.src_extensions:
  757.                 raise UnknownFileError, \
  758.                       "unknown file type '%s' (from '%s')" % \
  759.                       (ext, src_name)
  760.             if strip_dir:
  761.                 base = os.path.basename (base)
  762.             obj_names.append (os.path.join (output_dir,
  763.                                             base + self.obj_extension))
  764.         return obj_names
  765.  
  766.     # object_filenames ()
  767.  
  768.  
  769.     def shared_object_filename (self,
  770.                                 basename,
  771.                                 strip_dir=0,
  772.                                 output_dir=''):
  773.         if output_dir is None: output_dir = ''
  774.         if strip_dir:
  775.             basename = os.path.basename (basename)
  776.         return os.path.join (output_dir, basename + self.shared_lib_extension)
  777.  
  778.     def executable_filename (self,
  779.                                 basename,
  780.                                 strip_dir=0,
  781.                                 output_dir=''):
  782.         if output_dir is None: output_dir = ''
  783.         if strip_dir:
  784.             basename = os.path.basename (basename)
  785.         return os.path.join(output_dir, basename + (self.exe_extension or ''))
  786.  
  787.     def library_filename (self,
  788.                           libname,
  789.                           lib_type='static',     # or 'shared'
  790.                           strip_dir=0,
  791.                           output_dir=''):
  792.  
  793.         if output_dir is None: output_dir = ''
  794.         if lib_type not in ("static","shared"):
  795.             raise ValueError, "'lib_type' must be \"static\" or \"shared\""
  796.         fmt = getattr (self, lib_type + "_lib_format")
  797.         ext = getattr (self, lib_type + "_lib_extension")
  798.  
  799.         (dir, base) = os.path.split (libname)
  800.         filename = fmt % (base, ext)
  801.         if strip_dir:
  802.             dir = ''
  803.  
  804.         return os.path.join (output_dir, dir, filename)
  805.  
  806.  
  807.     # -- Utility methods -----------------------------------------------
  808.  
  809.     def announce (self, msg, level=1):
  810.         if self.verbose >= level:
  811.             print msg
  812.  
  813.     def debug_print (self, msg):
  814.         from distutils.core import DEBUG
  815.         if DEBUG:
  816.             print msg
  817.  
  818.     def warn (self, msg):
  819.         sys.stderr.write ("warning: %s\n" % msg)
  820.  
  821.     def execute (self, func, args, msg=None, level=1):
  822.         execute(func, args, msg, self.verbose >= level, self.dry_run)
  823.  
  824.     def spawn (self, cmd):
  825.         spawn (cmd, verbose=self.verbose, dry_run=self.dry_run)
  826.  
  827.     def move_file (self, src, dst):
  828.         return move_file (src, dst, verbose=self.verbose, dry_run=self.dry_run)
  829.  
  830.     def mkpath (self, name, mode=0777):
  831.         mkpath (name, mode, self.verbose, self.dry_run)
  832.  
  833.  
  834. # class CCompiler
  835.  
  836.  
  837. # Map a platform ('posix', 'nt') to the default compiler type for
  838. # that platform.
  839. default_compiler = { 'posix': 'unix',
  840.                      'nt': 'msvc',
  841.                    }
  842.  
  843. # Map compiler types to (module_name, class_name) pairs -- ie. where to
  844. # find the code that implements an interface to this compiler.  (The module
  845. # is assumed to be in the 'distutils' package.)
  846. compiler_class = { 'unix':    ('unixccompiler', 'UnixCCompiler',
  847.                                "standard UNIX-style compiler"),
  848.                    'msvc':    ('msvccompiler', 'MSVCCompiler',
  849.                                "Microsoft Visual C++"),
  850.                    'cygwin':  ('cygwinccompiler', 'CygwinCCompiler',
  851.                                "Cygwin port of GNU C Compiler for Win32"),
  852.                    'mingw32': ('cygwinccompiler', 'Mingw32CCompiler',
  853.                                "Mingw32 port of GNU C Compiler for Win32"),
  854.                    'bcpp':    ('bcppcompiler', 'BCPPCompiler',
  855.                                "Borland C++ Compiler"),
  856.                  }
  857.  
  858. def show_compilers():
  859.     """Print list of available compilers (used by the "--help-compiler"
  860.     options to "build", "build_ext", "build_clib").
  861.     """
  862.     # XXX this "knows" that the compiler option it's describing is
  863.     # "--compiler", which just happens to be the case for the three
  864.     # commands that use it.
  865.     from distutils.fancy_getopt import FancyGetopt 
  866.     compilers = []
  867.     for compiler in compiler_class.keys():
  868.         compilers.append(("compiler="+compiler, None,
  869.                           compiler_class[compiler][2]))
  870.     compilers.sort()
  871.     pretty_printer = FancyGetopt(compilers)
  872.     pretty_printer.print_help("List of available compilers:")
  873.     
  874.  
  875. def new_compiler (plat=None,
  876.                   compiler=None,
  877.                   verbose=0,
  878.                   dry_run=0,
  879.                   force=0):
  880.     """Generate an instance of some CCompiler subclass for the supplied
  881.     platform/compiler combination.  'plat' defaults to 'os.name'
  882.     (eg. 'posix', 'nt'), and 'compiler' defaults to the default compiler
  883.     for that platform.  Currently only 'posix' and 'nt' are supported, and
  884.     the default compilers are "traditional Unix interface" (UnixCCompiler
  885.     class) and Visual C++ (MSVCCompiler class).  Note that it's perfectly
  886.     possible to ask for a Unix compiler object under Windows, and a
  887.     Microsoft compiler object under Unix -- if you supply a value for
  888.     'compiler', 'plat' is ignored.
  889.     """
  890.     if plat is None:
  891.         plat = os.name
  892.  
  893.     try:
  894.         if compiler is None:
  895.             compiler = default_compiler[plat]
  896.         
  897.         (module_name, class_name, long_description) = compiler_class[compiler]
  898.     except KeyError:
  899.         msg = "don't know how to compile C/C++ code on platform '%s'" % plat
  900.         if compiler is not None:
  901.             msg = msg + " with '%s' compiler" % compiler
  902.         raise DistutilsPlatformError, msg
  903.               
  904.     try:
  905.         module_name = "distutils." + module_name
  906.         __import__ (module_name)
  907.         module = sys.modules[module_name]
  908.         klass = vars(module)[class_name]
  909.     except ImportError:
  910.         raise DistutilsModuleError, \
  911.               "can't compile C/C++ code: unable to load module '%s'" % \
  912.               module_name
  913.     except KeyError:
  914.         raise DistutilsModuleError, \
  915.               ("can't compile C/C++ code: unable to find class '%s' " +
  916.                "in module '%s'") % (class_name, module_name)
  917.  
  918.     return klass (verbose, dry_run, force)
  919.  
  920.  
  921. def gen_preprocess_options (macros, include_dirs):
  922.     """Generate C pre-processor options (-D, -U, -I) as used by at least
  923.     two types of compilers: the typical Unix compiler and Visual C++.
  924.     'macros' is the usual thing, a list of 1- or 2-tuples, where (name,)
  925.     means undefine (-U) macro 'name', and (name,value) means define (-D)
  926.     macro 'name' to 'value'.  'include_dirs' is just a list of directory
  927.     names to be added to the header file search path (-I).  Returns a list
  928.     of command-line options suitable for either Unix compilers or Visual
  929.     C++.
  930.     """
  931.     # XXX it would be nice (mainly aesthetic, and so we don't generate
  932.     # stupid-looking command lines) to go over 'macros' and eliminate
  933.     # redundant definitions/undefinitions (ie. ensure that only the
  934.     # latest mention of a particular macro winds up on the command
  935.     # line).  I don't think it's essential, though, since most (all?)
  936.     # Unix C compilers only pay attention to the latest -D or -U
  937.     # mention of a macro on their command line.  Similar situation for
  938.     # 'include_dirs'.  I'm punting on both for now.  Anyways, weeding out
  939.     # redundancies like this should probably be the province of
  940.     # CCompiler, since the data structures used are inherited from it
  941.     # and therefore common to all CCompiler classes.
  942.  
  943.     pp_opts = []
  944.     for macro in macros:
  945.  
  946.         if not (type (macro) is TupleType and
  947.                 1 <= len (macro) <= 2):
  948.             raise TypeError, \
  949.                   ("bad macro definition '%s': " +
  950.                    "each element of 'macros' list must be a 1- or 2-tuple") % \
  951.                   macro
  952.  
  953.         if len (macro) == 1:        # undefine this macro
  954.             pp_opts.append ("-U%s" % macro[0])
  955.         elif len (macro) == 2:
  956.             if macro[1] is None:    # define with no explicit value
  957.                 pp_opts.append ("-D%s" % macro[0])
  958.             else:
  959.                 # XXX *don't* need to be clever about quoting the
  960.                 # macro value here, because we're going to avoid the
  961.                 # shell at all costs when we spawn the command!
  962.                 pp_opts.append ("-D%s=%s" % macro)
  963.  
  964.     for dir in include_dirs:
  965.         pp_opts.append ("-I%s" % dir)
  966.  
  967.     return pp_opts
  968.  
  969. # gen_preprocess_options ()
  970.  
  971.  
  972. def gen_lib_options (compiler, library_dirs, runtime_library_dirs, libraries):
  973.     """Generate linker options for searching library directories and
  974.     linking with specific libraries.  'libraries' and 'library_dirs' are,
  975.     respectively, lists of library names (not filenames!) and search
  976.     directories.  Returns a list of command-line options suitable for use
  977.     with some compiler (depending on the two format strings passed in).
  978.     """
  979.     lib_opts = []
  980.  
  981.     for dir in library_dirs:
  982.         lib_opts.append (compiler.library_dir_option (dir))
  983.  
  984.     for dir in runtime_library_dirs:
  985.         lib_opts.append (compiler.runtime_library_dir_option (dir))
  986.  
  987.     # XXX it's important that we *not* remove redundant library mentions!
  988.     # sometimes you really do have to say "-lfoo -lbar -lfoo" in order to
  989.     # resolve all symbols.  I just hope we never have to say "-lfoo obj.o
  990.     # -lbar" to get things to work -- that's certainly a possibility, but a
  991.     # pretty nasty way to arrange your C code.
  992.  
  993.     for lib in libraries:
  994.         (lib_dir, lib_name) = os.path.split (lib)
  995.         if lib_dir:
  996.             lib_file = compiler.find_library_file ([lib_dir], lib_name)
  997.             if lib_file:
  998.                 lib_opts.append (lib_file)
  999.             else:
  1000.                 compiler.warn ("no library file corresponding to "
  1001.                                "'%s' found (skipping)" % lib)
  1002.         else:
  1003.             lib_opts.append (compiler.library_option (lib))
  1004.  
  1005.     return lib_opts
  1006.  
  1007. # gen_lib_options ()
  1008.