blob: 4a8c1d38c83b504cc4040429f372f78f0d84e7d6 [file] [log] [blame]
Greg Ward3f81cf71999-07-10 02:03:53 +00001"""distutils.ccompiler
2
3Contains CCompiler, an abstract base class that defines the interface
4for the Distutils compiler abstraction model."""
5
6# created 1999/07/05, Greg Ward
7
Greg Ward3ce77fd2000-03-02 01:49:45 +00008__revision__ = "$Id$"
Greg Ward3f81cf71999-07-10 02:03:53 +00009
Greg Ward802d6b71999-09-29 12:20:55 +000010import sys, os
Greg Ward3f81cf71999-07-10 02:03:53 +000011from types import *
12from copy import copy
13from distutils.errors import *
Greg Warde1aaaa61999-08-14 23:50:50 +000014from distutils.spawn import spawn
Greg Ward32c4a8a2000-03-06 03:40:29 +000015from distutils.util import move_file, mkpath, newer_pairwise, newer_group
Greg Ward3f81cf71999-07-10 02:03:53 +000016
17
18class CCompiler:
19 """Abstract base class to define the interface that must be implemented
20 by real compiler abstraction classes. Might have some use as a
21 place for shared code, but it's not yet clear what code can be
22 shared between compiler abstraction models for different platforms.
23
24 The basic idea behind a compiler abstraction class is that each
25 instance can be used for all the compile/link steps in building
26 a single project. Thus, attributes common to all of those compile
27 and link steps -- include directories, macros to define, libraries
28 to link against, etc. -- are attributes of the compiler instance.
29 To allow for variability in how individual files are treated,
30 most (all?) of those attributes may be varied on a per-compilation
31 or per-link basis."""
32
Greg Ward802d6b71999-09-29 12:20:55 +000033 # 'compiler_type' is a class attribute that identifies this class. It
34 # keeps code that wants to know what kind of compiler it's dealing with
35 # from having to import all possible compiler classes just to do an
36 # 'isinstance'. In concrete CCompiler subclasses, 'compiler_type'
37 # should really, really be one of the keys of the 'compiler_class'
38 # dictionary (see below -- used by the 'new_compiler()' factory
39 # function) -- authors of new compiler interface classes are
40 # responsible for updating 'compiler_class'!
41 compiler_type = None
Greg Ward3f81cf71999-07-10 02:03:53 +000042
43 # XXX things not handled by this compiler abstraction model:
44 # * client can't provide additional options for a compiler,
45 # e.g. warning, optimization, debugging flags. Perhaps this
46 # should be the domain of concrete compiler abstraction classes
47 # (UnixCCompiler, MSVCCompiler, etc.) -- or perhaps the base
48 # class should have methods for the common ones.
49 # * can't put output files (object files, libraries, whatever)
50 # into a separate directory from their inputs. Should this be
51 # handled by an 'output_dir' attribute of the whole object, or a
52 # parameter to the compile/link_* methods, or both?
53 # * can't completely override the include or library searchg
54 # path, ie. no "cc -I -Idir1 -Idir2" or "cc -L -Ldir1 -Ldir2".
Greg Warde1aaaa61999-08-14 23:50:50 +000055 # I'm not sure how widely supported this is even by Unix
Greg Ward3f81cf71999-07-10 02:03:53 +000056 # compilers, much less on other platforms. And I'm even less
Greg Warde1aaaa61999-08-14 23:50:50 +000057 # sure how useful it is; maybe for cross-compiling, but
58 # support for that is a ways off. (And anyways, cross
59 # compilers probably have a dedicated binary with the
60 # right paths compiled in. I hope.)
Greg Ward3f81cf71999-07-10 02:03:53 +000061 # * can't do really freaky things with the library list/library
62 # dirs, e.g. "-Ldir1 -lfoo -Ldir2 -lfoo" to link against
63 # different versions of libfoo.a in different locations. I
64 # think this is useless without the ability to null out the
65 # library search path anyways.
Greg Ward3f81cf71999-07-10 02:03:53 +000066
67
Greg Ward32c4a8a2000-03-06 03:40:29 +000068 # Subclasses that rely on the standard filename generation methods
69 # implemented below should override these; see the comment near
70 # those methods ('object_filenames()' et. al.) for details:
71 src_extensions = None # list of strings
72 obj_extension = None # string
73 static_lib_extension = None
74 shared_lib_extension = None # string
75 static_lib_format = None # format string
76 shared_lib_format = None # prob. same as static_lib_format
77 exe_extension = None # string
78
79
Greg Warde1aaaa61999-08-14 23:50:50 +000080 def __init__ (self,
81 verbose=0,
Greg Ward3febd601999-10-03 20:41:02 +000082 dry_run=0,
83 force=0):
Greg Warde1aaaa61999-08-14 23:50:50 +000084
85 self.verbose = verbose
86 self.dry_run = dry_run
Greg Ward3febd601999-10-03 20:41:02 +000087 self.force = force
Greg Ward3f81cf71999-07-10 02:03:53 +000088
Greg Ward9b17cb51999-09-13 03:07:24 +000089 # 'output_dir': a common output directory for object, library,
90 # shared object, and shared library files
91 self.output_dir = None
92
Greg Ward3f81cf71999-07-10 02:03:53 +000093 # 'macros': a list of macro definitions (or undefinitions). A
94 # macro definition is a 2-tuple (name, value), where the value is
95 # either a string or None (no explicit value). A macro
96 # undefinition is a 1-tuple (name,).
97 self.macros = []
98
Greg Ward3f81cf71999-07-10 02:03:53 +000099 # 'include_dirs': a list of directories to search for include files
100 self.include_dirs = []
101
102 # 'libraries': a list of libraries to include in any link
103 # (library names, not filenames: eg. "foo" not "libfoo.a")
104 self.libraries = []
105
106 # 'library_dirs': a list of directories to search for libraries
107 self.library_dirs = []
108
Greg Warde1aaaa61999-08-14 23:50:50 +0000109 # 'runtime_library_dirs': a list of directories to search for
110 # shared libraries/objects at runtime
111 self.runtime_library_dirs = []
112
Greg Ward3f81cf71999-07-10 02:03:53 +0000113 # 'objects': a list of object files (or similar, such as explicitly
114 # named library files) to include on any link
115 self.objects = []
116
117 # __init__ ()
118
119
120 def _find_macro (self, name):
121 i = 0
122 for defn in self.macros:
123 if defn[0] == name:
124 return i
125 i = i + 1
126
127 return None
128
129
130 def _check_macro_definitions (self, definitions):
131 """Ensures that every element of 'definitions' is a valid macro
132 definition, ie. either (name,value) 2-tuple or a (name,)
133 tuple. Do nothing if all definitions are OK, raise
134 TypeError otherwise."""
135
136 for defn in definitions:
137 if not (type (defn) is TupleType and
138 (len (defn) == 1 or
139 (len (defn) == 2 and
140 (type (defn[1]) is StringType or defn[1] is None))) and
141 type (defn[0]) is StringType):
142 raise TypeError, \
143 ("invalid macro definition '%s': " % defn) + \
144 "must be tuple (string,), (string, string), or " + \
145 "(string, None)"
146
147
148 # -- Bookkeeping methods -------------------------------------------
149
150 def define_macro (self, name, value=None):
151 """Define a preprocessor macro for all compilations driven by
152 this compiler object. The optional parameter 'value' should be
153 a string; if it is not supplied, then the macro will be defined
154 without an explicit value and the exact outcome depends on the
155 compiler used (XXX true? does ANSI say anything about this?)"""
156
157 # Delete from the list of macro definitions/undefinitions if
158 # already there (so that this one will take precedence).
159 i = self._find_macro (name)
160 if i is not None:
161 del self.macros[i]
162
163 defn = (name, value)
164 self.macros.append (defn)
165
166
167 def undefine_macro (self, name):
168 """Undefine a preprocessor macro for all compilations driven by
169 this compiler object. If the same macro is defined by
170 'define_macro()' and undefined by 'undefine_macro()' the last
171 call takes precedence (including multiple redefinitions or
172 undefinitions). If the macro is redefined/undefined on a
173 per-compilation basis (ie. in the call to 'compile()'), then
174 that takes precedence."""
175
176 # Delete from the list of macro definitions/undefinitions if
177 # already there (so that this one will take precedence).
178 i = self._find_macro (name)
179 if i is not None:
180 del self.macros[i]
181
182 undefn = (name,)
183 self.macros.append (undefn)
184
185
186 def add_include_dir (self, dir):
187 """Add 'dir' to the list of directories that will be searched
188 for header files. The compiler is instructed to search
189 directories in the order in which they are supplied by
190 successive calls to 'add_include_dir()'."""
191 self.include_dirs.append (dir)
192
193 def set_include_dirs (self, dirs):
194 """Set the list of directories that will be searched to 'dirs'
195 (a list of strings). Overrides any preceding calls to
196 'add_include_dir()'; subsequence calls to 'add_include_dir()'
197 add to the list passed to 'set_include_dirs()'. This does
198 not affect any list of standard include directories that
199 the compiler may search by default."""
200 self.include_dirs = copy (dirs)
201
202
203 def add_library (self, libname):
204 """Add 'libname' to the list of libraries that will be included
205 in all links driven by this compiler object. Note that
206 'libname' should *not* be the name of a file containing a
207 library, but the name of the library itself: the actual filename
208 will be inferred by the linker, the compiler, or the compiler
209 abstraction class (depending on the platform).
210
211 The linker will be instructed to link against libraries in the
212 order they were supplied to 'add_library()' and/or
213 'set_libraries()'. It is perfectly valid to duplicate library
214 names; the linker will be instructed to link against libraries
215 as many times as they are mentioned."""
216 self.libraries.append (libname)
217
218 def set_libraries (self, libnames):
219 """Set the list of libraries to be included in all links driven
220 by this compiler object to 'libnames' (a list of strings).
221 This does not affect any standard system libraries that the
222 linker may include by default."""
223
224 self.libraries = copy (libnames)
225
226
227 def add_library_dir (self, dir):
228 """Add 'dir' to the list of directories that will be searched for
229 libraries specified to 'add_library()' and 'set_libraries()'.
230 The linker will be instructed to search for libraries in the
231 order they are supplied to 'add_library_dir()' and/or
232 'set_library_dirs()'."""
233 self.library_dirs.append (dir)
234
235 def set_library_dirs (self, dirs):
236 """Set the list of library search directories to 'dirs' (a list
237 of strings). This does not affect any standard library
238 search path that the linker may search by default."""
239 self.library_dirs = copy (dirs)
240
241
Greg Warde1aaaa61999-08-14 23:50:50 +0000242 def add_runtime_library_dir (self, dir):
243 """Add 'dir' to the list of directories that will be searched for
244 shared libraries at runtime."""
245 self.runtime_library_dirs.append (dir)
246
247 def set_runtime_library_dirs (self, dirs):
248 """Set the list of directories to search for shared libraries
249 at runtime to 'dirs' (a list of strings). This does not affect
250 any standard search path that the runtime linker may search by
251 default."""
252 self.runtime_library_dirs = copy (dirs)
253
254
Greg Ward3f81cf71999-07-10 02:03:53 +0000255 def add_link_object (self, object):
256 """Add 'object' to the list of object files (or analogues, such
257 as explictly named library files or the output of "resource
258 compilers") to be included in every link driven by this
259 compiler object."""
260 self.objects.append (object)
261
262 def set_link_objects (self, objects):
263 """Set the list of object files (or analogues) to be included
264 in every link to 'objects'. This does not affect any
265 standard object files that the linker may include by default
266 (such as system libraries)."""
267 self.objects = copy (objects)
268
269
Greg Ward32c4a8a2000-03-06 03:40:29 +0000270 # -- Priviate utility methods --------------------------------------
271 # (here for the convenience of subclasses)
272
273 def _fix_compile_args (self, output_dir, macros, include_dirs):
274 """Typecheck and fix-up some of the arguments to the 'compile()' method,
275 and return fixed-up values. Specifically: if 'output_dir' is
276 None, replaces it with 'self.output_dir'; ensures that 'macros'
277 is a list, and augments it with 'self.macros'; ensures that
278 'include_dirs' is a list, and augments it with
279 'self.include_dirs'. Guarantees that the returned values are of
280 the correct type, i.e. for 'output_dir' either string or None,
281 and for 'macros' and 'include_dirs' either list or None."""
282
283 if output_dir is None:
284 output_dir = self.output_dir
285 elif type (output_dir) is not StringType:
286 raise TypeError, "'output_dir' must be a string or None"
287
288 if macros is None:
289 macros = self.macros
290 elif type (macros) is ListType:
291 macros = macros + (self.macros or [])
292 else:
293 raise TypeError, \
294 "'macros' (if supplied) must be a list of tuples"
295
296 if include_dirs is None:
297 include_dirs = self.include_dirs
298 elif type (include_dirs) in (ListType, TupleType):
299 include_dirs = list (include_dirs) + (self.include_dirs or [])
300 else:
301 raise TypeError, \
302 "'include_dirs' (if supplied) must be a list of strings"
303
304 return (output_dir, macros, include_dirs)
305
306 # _fix_compile_args ()
307
308
309 def _prep_compile (self, sources, output_dir):
310 """Determine the list of object files corresponding to 'sources', and
311 figure out which ones really need to be recompiled. Return a list
312 of all object files and a dictionary telling which source files can
313 be skipped."""
314
315 # Get the list of expected output (object) files
316 objects = self.object_filenames (sources,
317 output_dir=output_dir)
318
319 if self.force:
320 skip_source = {} # rebuild everything
321 for source in sources:
322 skip_source[source] = 0
323 else:
324 # Figure out which source files we have to recompile according
325 # to a simplistic check -- we just compare the source and
326 # object file, no deep dependency checking involving header
327 # files.
328 skip_source = {} # rebuild everything
329 for source in sources: # no wait, rebuild nothing
330 skip_source[source] = 1
331
332 (n_sources, n_objects) = newer_pairwise (sources, objects)
333 for source in n_sources: # no really, only rebuild what's out-of-date
334 skip_source[source] = 0
335
336 return (objects, skip_source)
337
338 # _prep_compile ()
339
340
341 def _fix_link_args (self, objects, output_dir,
342 takes_libs=0, libraries=None, library_dirs=None):
343 """Typecheck and fix up some of the arguments supplied to the
344 'link_*' methods and return the fixed values. Specifically:
345 ensure that 'objects' is a list; if output_dir is None, use
346 self.output_dir; ensure that 'libraries' and 'library_dirs' are
347 both lists, and augment them with 'self.libraries' and
348 'self.library_dirs'. If 'takes_libs' is true, return a tuple
349 (objects, output_dir, libraries, library_dirs; else return
350 (objects, output_dir)."""
351
352 if type (objects) not in (ListType, TupleType):
353 raise TypeError, \
354 "'objects' must be a list or tuple of strings"
355 objects = list (objects)
356
357 if output_dir is None:
358 output_dir = self.output_dir
359 elif type (output_dir) is not StringType:
360 raise TypeError, "'output_dir' must be a string or None"
361
362 if takes_libs:
363 if libraries is None:
364 libraries = self.libraries
365 elif type (libraries) in (ListType, TupleType):
366 libraries = list (libraries) + (self.libraries or [])
367 else:
368 raise TypeError, \
369 "'libraries' (if supplied) must be a list of strings"
370
371 if library_dirs is None:
372 library_dirs = self.library_dirs
373 elif type (library_dirs) in (ListType, TupleType):
374 library_dirs = list (library_dirs) + (self.library_dirs or [])
375 else:
376 raise TypeError, \
377 "'library_dirs' (if supplied) must be a list of strings"
378
379 return (objects, output_dir, libraries, library_dirs)
380 else:
381 return (objects, output_dir)
382
383 # _fix_link_args ()
384
385
386 def _need_link (self, objects, output_file):
387 """Return true if we need to relink the files listed in 'objects' to
388 recreate 'output_file'."""
389
390 if self.force:
391 return 1
392 else:
393 if self.dry_run:
394 newer = newer_group (objects, output_file, missing='newer')
395 else:
396 newer = newer_group (objects, output_file)
397 return newer
398
399 # _need_link ()
400
401
Greg Ward3f81cf71999-07-10 02:03:53 +0000402 # -- Worker methods ------------------------------------------------
403 # (must be implemented by subclasses)
404
405 def compile (self,
406 sources,
Greg Ward9b17cb51999-09-13 03:07:24 +0000407 output_dir=None,
Greg Ward3f81cf71999-07-10 02:03:53 +0000408 macros=None,
Greg Ward0bdd90a1999-12-12 17:19:58 +0000409 include_dirs=None,
Greg Ward3c045a52000-02-09 02:16:14 +0000410 debug=0,
Greg Ward802d6b71999-09-29 12:20:55 +0000411 extra_preargs=None,
412 extra_postargs=None):
Greg Ward3f81cf71999-07-10 02:03:53 +0000413 """Compile one or more C/C++ source files. 'sources' must be
414 a list of strings, each one the name of a C/C++ source
Greg Ward32c4a8a2000-03-06 03:40:29 +0000415 file. Return a list of object filenames, one per source
416 filename in 'sources'. Depending on the implementation,
417 not all source files will necessarily be compiled, but
418 all corresponding object filenames will be returned.
419
420 If 'output_dir' is given, object files will be put under it,
421 while retaining their original path component. That is,
422 "foo/bar.c" normally compiles to "foo/bar.o" (for a Unix
423 implementation); if 'output_dir' is "build", then it would
424 compile to "build/foo/bar.o".
Greg Ward3f81cf71999-07-10 02:03:53 +0000425
426 'macros', if given, must be a list of macro definitions. A
427 macro definition is either a (name, value) 2-tuple or a (name,)
428 1-tuple. The former defines a macro; if the value is None, the
429 macro is defined without an explicit value. The 1-tuple case
430 undefines a macro. Later definitions/redefinitions/
431 undefinitions take precedence.
432
Greg Ward3c045a52000-02-09 02:16:14 +0000433 'include_dirs', if given, must be a list of strings, the
434 directories to add to the default include file search path for
435 this compilation only.
436
437 'debug' is a boolean; if true, the compiler will be instructed
438 to output debug symbols in (or alongside) the object file(s).
Greg Ward802d6b71999-09-29 12:20:55 +0000439
Greg Ward32c4a8a2000-03-06 03:40:29 +0000440 'extra_preargs' and 'extra_postargs' are implementation-
441 dependent. On platforms that have the notion of a command-line
442 (e.g. Unix, DOS/Windows), they are most likely lists of strings:
443 extra command-line arguments to prepand/append to the compiler
444 command line. On other platforms, consult the implementation
445 class documentation. In any event, they are intended as an
Greg Ward802d6b71999-09-29 12:20:55 +0000446 escape hatch for those occasions when the abstract compiler
447 framework doesn't cut the mustard."""
448
Greg Ward3f81cf71999-07-10 02:03:53 +0000449 pass
450
451
Greg Ward036c8052000-03-10 01:48:32 +0000452 def create_static_lib (self,
453 objects,
454 output_libname,
455 output_dir=None,
456 debug=0):
Greg Ward3f81cf71999-07-10 02:03:53 +0000457 """Link a bunch of stuff together to create a static library
458 file. The "bunch of stuff" consists of the list of object
459 files supplied as 'objects', the extra object files supplied
460 to 'add_link_object()' and/or 'set_link_objects()', the
461 libraries supplied to 'add_library()' and/or
462 'set_libraries()', and the libraries supplied as 'libraries'
463 (if any).
464
Greg Ward3c045a52000-02-09 02:16:14 +0000465 'output_libname' should be a library name, not a filename; the
466 filename will be inferred from the library name. 'output_dir'
467 is the directory where the library file will be put.
468
469 'debug' is a boolean; if true, debugging information will be
470 included in the library (note that on most platforms, it is the
471 compile step where this matters: the 'debug' flag is included
472 here just for consistency)."""
473
474 pass
475
476
477 def link_shared_lib (self,
478 objects,
479 output_libname,
480 output_dir=None,
481 libraries=None,
482 library_dirs=None,
483 debug=0,
484 extra_preargs=None,
485 extra_postargs=None):
486 """Link a bunch of stuff together to create a shared library
Greg Ward036c8052000-03-10 01:48:32 +0000487 file. Similar semantics to 'create_static_lib()', with the
488 addition of other libraries to link against and directories to
489 search for them. Also, of course, the type and name of
490 the generated file will almost certainly be different, as will
491 the program used to create it.
Greg Ward3f81cf71999-07-10 02:03:53 +0000492
Greg Ward3febd601999-10-03 20:41:02 +0000493 'libraries' is a list of libraries to link against. These are
494 library names, not filenames, since they're translated into
495 filenames in a platform-specific way (eg. "foo" becomes
496 "libfoo.a" on Unix and "foo.lib" on DOS/Windows). However, they
497 can include a directory component, which means the linker will
498 look in that specific directory rather than searching all the
499 normal locations.
500
501 'library_dirs', if supplied, should be a list of directories to
502 search for libraries that were specified as bare library names
503 (ie. no directory component). These are on top of the system
504 default and those supplied to 'add_library_dir()' and/or
505 'set_library_dirs()'.
Greg Ward802d6b71999-09-29 12:20:55 +0000506
Greg Ward036c8052000-03-10 01:48:32 +0000507 'debug' is as for 'compile()' and 'create_static_lib()', with the
Greg Ward3c045a52000-02-09 02:16:14 +0000508 slight distinction that it actually matters on most platforms
Greg Ward036c8052000-03-10 01:48:32 +0000509 (as opposed to 'create_static_lib()', which includes a 'debug'
Greg Ward3c045a52000-02-09 02:16:14 +0000510 flag mostly for form's sake).
511
Greg Ward802d6b71999-09-29 12:20:55 +0000512 'extra_preargs' and 'extra_postargs' are as for 'compile()'
513 (except of course that they supply command-line arguments
Greg Ward3c045a52000-02-09 02:16:14 +0000514 for the particular linker being used)."""
Greg Ward3f81cf71999-07-10 02:03:53 +0000515
516 pass
517
518
Greg Ward3f81cf71999-07-10 02:03:53 +0000519 def link_shared_object (self,
520 objects,
521 output_filename,
Greg Ward9b17cb51999-09-13 03:07:24 +0000522 output_dir=None,
Greg Ward3f81cf71999-07-10 02:03:53 +0000523 libraries=None,
Greg Ward26e48ea1999-08-29 18:17:36 +0000524 library_dirs=None,
Greg Ward3c045a52000-02-09 02:16:14 +0000525 debug=0,
Greg Ward802d6b71999-09-29 12:20:55 +0000526 extra_preargs=None,
527 extra_postargs=None):
Greg Ward3f81cf71999-07-10 02:03:53 +0000528 """Link a bunch of stuff together to create a shared object
Greg Ward9b17cb51999-09-13 03:07:24 +0000529 file. Much like 'link_shared_lib()', except the output filename
530 is explicitly supplied as 'output_filename'. If 'output_dir' is
531 supplied, 'output_filename' is relative to it
Greg Ward802d6b71999-09-29 12:20:55 +0000532 (i.e. 'output_filename' can provide directory components if
Greg Ward9b17cb51999-09-13 03:07:24 +0000533 needed)."""
Greg Ward3f81cf71999-07-10 02:03:53 +0000534 pass
535
Greg Warde1aaaa61999-08-14 23:50:50 +0000536
Greg Ward5baf1c22000-01-09 22:41:02 +0000537 def link_executable (self,
538 objects,
539 output_progname,
540 output_dir=None,
541 libraries=None,
542 library_dirs=None,
Greg Ward3c045a52000-02-09 02:16:14 +0000543 debug=0,
Greg Ward5baf1c22000-01-09 22:41:02 +0000544 extra_preargs=None,
545 extra_postargs=None):
546 """Link a bunch of stuff together to create a binary executable
Greg Ward036c8052000-03-10 01:48:32 +0000547 file. The "bunch of stuff" is as for 'link_shared_lib()'.
Greg Ward5baf1c22000-01-09 22:41:02 +0000548 'output_progname' should be the base name of the executable
549 program--e.g. on Unix the same as the output filename, but
550 on DOS/Windows ".exe" will be appended."""
551 pass
552
553
554
Greg Ward32c4a8a2000-03-06 03:40:29 +0000555 # -- Filename generation methods -----------------------------------
Greg Warde1aaaa61999-08-14 23:50:50 +0000556
Greg Ward32c4a8a2000-03-06 03:40:29 +0000557 # The default implementation of the filename generating methods are
558 # prejudiced towards the Unix/DOS/Windows view of the world:
559 # * object files are named by replacing the source file extension
560 # (eg. .c/.cpp -> .o/.obj)
561 # * library files (shared or static) are named by plugging the
562 # library name and extension into a format string, eg.
563 # "lib%s.%s" % (lib_name, ".a") for Unix static libraries
564 # * executables are named by appending an extension (possibly
565 # empty) to the program name: eg. progname + ".exe" for
566 # Windows
567 #
568 # To reduce redundant code, these methods expect to find
569 # several attributes in the current object (presumably defined
570 # as class attributes):
571 # * src_extensions -
572 # list of C/C++ source file extensions, eg. ['.c', '.cpp']
573 # * obj_extension -
574 # object file extension, eg. '.o' or '.obj'
575 # * static_lib_extension -
576 # extension for static library files, eg. '.a' or '.lib'
577 # * shared_lib_extension -
578 # extension for shared library/object files, eg. '.so', '.dll'
579 # * static_lib_format -
580 # format string for generating static library filenames,
581 # eg. 'lib%s.%s' or '%s.%s'
582 # * shared_lib_format
583 # format string for generating shared library filenames
584 # (probably same as static_lib_format, since the extension
585 # is one of the intended parameters to the format string)
586 # * exe_extension -
587 # extension for executable files, eg. '' or '.exe'
Greg Ward9b17cb51999-09-13 03:07:24 +0000588
Greg Ward32c4a8a2000-03-06 03:40:29 +0000589 def object_filenames (self,
590 source_filenames,
591 strip_dir=0,
592 output_dir=''):
593 if output_dir is None: output_dir = ''
594 obj_names = []
595 for src_name in source_filenames:
596 (base, ext) = os.path.splitext (src_name)
597 if ext not in self.src_extensions:
598 continue
599 if strip_dir:
600 base = os.path.basename (base)
601 obj_names.append (os.path.join (output_dir,
602 base + self.obj_extension))
603 return obj_names
Greg Warde1aaaa61999-08-14 23:50:50 +0000604
Greg Ward32c4a8a2000-03-06 03:40:29 +0000605 # object_filenames ()
Greg Warde1aaaa61999-08-14 23:50:50 +0000606
Greg Warde1aaaa61999-08-14 23:50:50 +0000607
Greg Ward32c4a8a2000-03-06 03:40:29 +0000608 def shared_object_filename (self,
609 basename,
610 strip_dir=0,
611 output_dir=''):
612 if output_dir is None: output_dir = ''
613 if strip_dir:
614 basename = os.path.basename (basename)
615 return os.path.join (output_dir, basename + self.shared_lib_extension)
Greg Warde1aaaa61999-08-14 23:50:50 +0000616
Greg Ward26e48ea1999-08-29 18:17:36 +0000617
Greg Ward32c4a8a2000-03-06 03:40:29 +0000618 def library_filename (self,
619 libname,
620 lib_type='static', # or 'shared'
621 strip_dir=0,
622 output_dir=''):
623
624 if output_dir is None: output_dir = ''
625 if lib_type not in ("static","shared"):
626 raise ValueError, "'lib_type' must be \"static\" or \"shared\""
627 fmt = getattr (self, lib_type + "_lib_format")
628 ext = getattr (self, lib_type + "_lib_extension")
629
630 (dir, base) = os.path.split (libname)
631 filename = fmt % (base, ext)
632 if strip_dir:
633 dir = ''
634
635 return os.path.join (output_dir, dir, filename)
636
Greg Warde1aaaa61999-08-14 23:50:50 +0000637
638 # -- Utility methods -----------------------------------------------
639
Greg Ward9b17cb51999-09-13 03:07:24 +0000640 def announce (self, msg, level=1):
641 if self.verbose >= level:
642 print msg
643
Greg Ward3febd601999-10-03 20:41:02 +0000644 def warn (self, msg):
645 sys.stderr.write ("warning: %s\n" % msg)
646
Greg Warde1aaaa61999-08-14 23:50:50 +0000647 def spawn (self, cmd):
648 spawn (cmd, verbose=self.verbose, dry_run=self.dry_run)
649
Greg Ward9b17cb51999-09-13 03:07:24 +0000650 def move_file (self, src, dst):
651 return move_file (src, dst, verbose=self.verbose, dry_run=self.dry_run)
652
Greg Ward013f0c82000-03-01 14:43:12 +0000653 def mkpath (self, name, mode=0777):
654 mkpath (name, mode, self.verbose, self.dry_run)
655
Greg Warde1aaaa61999-08-14 23:50:50 +0000656
Greg Ward3f81cf71999-07-10 02:03:53 +0000657# class CCompiler
658
659
Greg Ward802d6b71999-09-29 12:20:55 +0000660# Map a platform ('posix', 'nt') to the default compiler type for
661# that platform.
662default_compiler = { 'posix': 'unix',
663 'nt': 'msvc',
664 }
665
666# Map compiler types to (module_name, class_name) pairs -- ie. where to
667# find the code that implements an interface to this compiler. (The module
668# is assumed to be in the 'distutils' package.)
669compiler_class = { 'unix': ('unixccompiler', 'UnixCCompiler'),
670 'msvc': ('msvccompiler', 'MSVCCompiler'),
671 }
672
673
Greg Warde1aaaa61999-08-14 23:50:50 +0000674def new_compiler (plat=None,
Greg Ward802d6b71999-09-29 12:20:55 +0000675 compiler=None,
Greg Warde1aaaa61999-08-14 23:50:50 +0000676 verbose=0,
Greg Ward3febd601999-10-03 20:41:02 +0000677 dry_run=0,
678 force=0):
Greg Ward3f81cf71999-07-10 02:03:53 +0000679
Greg Ward802d6b71999-09-29 12:20:55 +0000680 """Generate an instance of some CCompiler subclass for the supplied
681 platform/compiler combination. 'plat' defaults to 'os.name'
682 (eg. 'posix', 'nt'), and 'compiler' defaults to the default
683 compiler for that platform. Currently only 'posix' and 'nt'
684 are supported, and the default compilers are "traditional Unix
685 interface" (UnixCCompiler class) and Visual C++ (MSVCCompiler
686 class). Note that it's perfectly possible to ask for a Unix
687 compiler object under Windows, and a Microsoft compiler object
688 under Unix -- if you supply a value for 'compiler', 'plat'
689 is ignored."""
690
691 if plat is None:
692 plat = os.name
693
694 try:
695 if compiler is None:
696 compiler = default_compiler[plat]
697
698 (module_name, class_name) = compiler_class[compiler]
699 except KeyError:
700 msg = "don't know how to compile C/C++ code on platform '%s'" % plat
701 if compiler is not None:
702 msg = msg + " with '%s' compiler" % compiler
703 raise DistutilsPlatformError, msg
704
705 try:
706 module_name = "distutils." + module_name
707 __import__ (module_name)
708 module = sys.modules[module_name]
709 klass = vars(module)[class_name]
710 except ImportError:
711 raise DistutilsModuleError, \
712 "can't compile C/C++ code: unable to load module '%s'" % \
713 module_name
714 except KeyError:
715 raise DistutilsModuleError, \
716 ("can't compile C/C++ code: unable to find class '%s' " +
717 "in module '%s'") % (class_name, module_name)
718
Greg Ward3febd601999-10-03 20:41:02 +0000719 return klass (verbose, dry_run, force)
Greg Wardf7a39ec1999-09-08 02:29:08 +0000720
721
Greg Ward0bdd90a1999-12-12 17:19:58 +0000722def gen_preprocess_options (macros, include_dirs):
Greg Wardf7a39ec1999-09-08 02:29:08 +0000723 """Generate C pre-processor options (-D, -U, -I) as used by at
724 least two types of compilers: the typical Unix compiler and Visual
725 C++. 'macros' is the usual thing, a list of 1- or 2-tuples, where
726 (name,) means undefine (-U) macro 'name', and (name,value) means
Greg Ward0bdd90a1999-12-12 17:19:58 +0000727 define (-D) macro 'name' to 'value'. 'include_dirs' is just a list of
Greg Wardf7a39ec1999-09-08 02:29:08 +0000728 directory names to be added to the header file search path (-I).
729 Returns a list of command-line options suitable for either
730 Unix compilers or Visual C++."""
731
732 # XXX it would be nice (mainly aesthetic, and so we don't generate
733 # stupid-looking command lines) to go over 'macros' and eliminate
734 # redundant definitions/undefinitions (ie. ensure that only the
735 # latest mention of a particular macro winds up on the command
736 # line). I don't think it's essential, though, since most (all?)
737 # Unix C compilers only pay attention to the latest -D or -U
738 # mention of a macro on their command line. Similar situation for
Greg Ward0bdd90a1999-12-12 17:19:58 +0000739 # 'include_dirs'. I'm punting on both for now. Anyways, weeding out
Greg Wardf7a39ec1999-09-08 02:29:08 +0000740 # redundancies like this should probably be the province of
741 # CCompiler, since the data structures used are inherited from it
742 # and therefore common to all CCompiler classes.
743
744 pp_opts = []
745 for macro in macros:
Greg Wardfbf8aff1999-09-21 18:35:09 +0000746
747 if not (type (macro) is TupleType and
748 1 <= len (macro) <= 2):
749 raise TypeError, \
750 ("bad macro definition '%s': " +
751 "each element of 'macros' list must be a 1- or 2-tuple") % \
752 macro
753
Greg Wardf7a39ec1999-09-08 02:29:08 +0000754 if len (macro) == 1: # undefine this macro
755 pp_opts.append ("-U%s" % macro[0])
756 elif len (macro) == 2:
757 if macro[1] is None: # define with no explicit value
758 pp_opts.append ("-D%s" % macro[0])
759 else:
760 # XXX *don't* need to be clever about quoting the
761 # macro value here, because we're going to avoid the
762 # shell at all costs when we spawn the command!
763 pp_opts.append ("-D%s=%s" % macro)
764
Greg Ward0bdd90a1999-12-12 17:19:58 +0000765 for dir in include_dirs:
Greg Wardf7a39ec1999-09-08 02:29:08 +0000766 pp_opts.append ("-I%s" % dir)
767
768 return pp_opts
769
770# gen_preprocess_options ()
771
772
Greg Wardd03f88a2000-03-18 15:19:51 +0000773def gen_lib_options (compiler, library_dirs, runtime_library_dirs, libraries):
Greg Wardf7a39ec1999-09-08 02:29:08 +0000774 """Generate linker options for searching library directories and
775 linking with specific libraries. 'libraries' and 'library_dirs'
776 are, respectively, lists of library names (not filenames!) and
Greg Ward3febd601999-10-03 20:41:02 +0000777 search directories. Returns a list of command-line options suitable
778 for use with some compiler (depending on the two format strings
779 passed in)."""
Greg Wardf7a39ec1999-09-08 02:29:08 +0000780
781 lib_opts = []
782
783 for dir in library_dirs:
Greg Ward3febd601999-10-03 20:41:02 +0000784 lib_opts.append (compiler.library_dir_option (dir))
Greg Wardf7a39ec1999-09-08 02:29:08 +0000785
Greg Wardd03f88a2000-03-18 15:19:51 +0000786 for dir in runtime_library_dirs:
787 lib_opts.append (compiler.runtime_library_dir_option (dir))
788
Greg Wardf7a39ec1999-09-08 02:29:08 +0000789 # XXX it's important that we *not* remove redundant library mentions!
790 # sometimes you really do have to say "-lfoo -lbar -lfoo" in order to
791 # resolve all symbols. I just hope we never have to say "-lfoo obj.o
792 # -lbar" to get things to work -- that's certainly a possibility, but a
793 # pretty nasty way to arrange your C code.
794
795 for lib in libraries:
Greg Ward3febd601999-10-03 20:41:02 +0000796 (lib_dir, lib_name) = os.path.split (lib)
797 if lib_dir:
798 lib_file = compiler.find_library_file ([lib_dir], lib_name)
799 if lib_file:
800 lib_opts.append (lib_file)
801 else:
802 compiler.warn ("no library file corresponding to "
803 "'%s' found (skipping)" % lib)
804 else:
805 lib_opts.append (compiler.library_option (lib))
Greg Wardf7a39ec1999-09-08 02:29:08 +0000806
807 return lib_opts
808
Greg Ward32c4a8a2000-03-06 03:40:29 +0000809# gen_lib_options ()