Chris@87: from __future__ import division, absolute_import, print_function
Chris@87: 
Chris@87: import os
Chris@87: from distutils.core import Command
Chris@87: from distutils.ccompiler import new_compiler
Chris@87: from numpy.distutils.misc_util import get_cmd
Chris@87: 
Chris@87: class install_clib(Command):
Chris@87:     description = "Command to install installable C libraries"
Chris@87: 
Chris@87:     user_options = []
Chris@87: 
Chris@87:     def initialize_options(self):
Chris@87:         self.install_dir = None
Chris@87:         self.outfiles = []
Chris@87: 
Chris@87:     def finalize_options(self):
Chris@87:         self.set_undefined_options('install', ('install_lib', 'install_dir'))
Chris@87: 
Chris@87:     def run (self):
Chris@87:         build_clib_cmd = get_cmd("build_clib")
Chris@87:         build_dir = build_clib_cmd.build_clib
Chris@87: 
Chris@87:         # We need the compiler to get the library name -> filename association
Chris@87:         if not build_clib_cmd.compiler:
Chris@87:             compiler = new_compiler(compiler=None)
Chris@87:             compiler.customize(self.distribution)
Chris@87:         else:
Chris@87:             compiler = build_clib_cmd.compiler
Chris@87: 
Chris@87:         for l in self.distribution.installed_libraries:
Chris@87:             target_dir = os.path.join(self.install_dir, l.target_dir)
Chris@87:             name = compiler.library_filename(l.name)
Chris@87:             source = os.path.join(build_dir, name)
Chris@87:             self.mkpath(target_dir)
Chris@87:             self.outfiles.append(self.copy_file(source, target_dir)[0])
Chris@87: 
Chris@87:     def get_outputs(self):
Chris@87:         return self.outfiles