* Modification of setup.py to compile vtk part too. Then, we have 2

[gdcm.git] / WrapSwig.py

diff --git a/WrapSwig.py b/WrapSwig.py
new file mode 100644 (file)
index 0000000..f8e02f9
--- /dev/null
+++ b/WrapSwig.py
@@ -0,0 +1,106 @@
+from distutilsWrapping import *
+from types import ListType
+import os
+
+class SwigWrapper(Wrapper):
+       """
+       This distutils command is meant to be used with MyExtension extension,which
+       defines a swig_include attribute.
+       """
+       def WrapSources(self,distutil,extWrap,sources):
+               """Walk the list of source files in 'sources',looking for SWIG
+               interface(.i) files.  Run SWIG on all that are found,and
+               return a modified 'sources' list with SWIG source files replaced
+               by the generated C(or C++) files.
+               """
+
+               new_sources=[]
+               swig_sources=[]
+               swig_targets={}
+
+               # XXX this drops generated C/C++ files into the source tree,which
+               # is fine for developers who want to distribute the generated
+               # source -- but there should be an option to put SWIG output in
+               # the temp dir.
+
+               ## Modified lines(compared to buil_exts.wig_sources original method)
+               if extWrap.swig_cpp:
+                       target_ext='_wrap.cpp'
+               else:
+                       target_ext='_wrap.c'
+               ## End of modification
+
+               for source in sources:
+                       (base,ext)=os.path.splitext(source)
+                       if ext==".i":             # SWIG interface file
+                               new_sources.append(base + target_ext)
+                               swig_sources.append(source)
+                               swig_targets[source]=new_sources[-1]
+                       elif ext==".h":
+                               continue
+                       else:
+                               new_sources.append(source)
+
+               if not swig_sources:
+                       return new_sources
+
+               swig=distutil.find_swig()
+
+               ## Modified lines(compared to buil_exts.wig_sources original method)
+               swig_cmd=[swig,"-python"]
+               if extWrap.swig_cpp:
+                       swig_cmd.append("-c++")
+
+               if extWrap.swig_include:
+                       for pth in extWrap.swig_include:
+                               swig_cmd.append("-I%s"%pth)
+               ## End of modification
+
+               for source in swig_sources:
+                       target=swig_targets[source]
+                       distutil.announce("swigging %s to %s" %(source,target))
+                       distutil.spawn(swig_cmd + ["-o",target,source])
+                       ## Modified lines(compared to buil_exts.wig_sources original method)
+                       # When swig generated some shadow classes,place them under
+                       # self.build_lib(the build directory for Python source).
+                       if extWrap.swig_cpp:
+                               # Generate the full pathname of the shadow classes file
+                               import string
+                               swig_shadow=string.split(os.path.basename(source),".")[0]
+                               swig_shadow=swig_shadow + '.py'
+                               # On win32 swig places the shadow classes in the directory
+                               # where it was invoked. This is to be opposed to posix where
+                               # swig places the shadow classes aside the C++ wrapping code
+                               #(the target in our context).
+                               if(os.name=='posix'):
+                                       infile=os.path.join(os.path.dirname(source),swig_shadow)
+                               else:
+                                       infile=swig_shadow
+                               if os.path.isfile(infile):
+                                       outfile=[distutil.build_lib,distutil.distribution.get_name()]
+                                       outfile.append(swig_shadow)
+                                       outfile=apply(os.path.join,outfile)
+                                       distutil.copy_file(infile,outfile,preserve_mode=0)
+                               else:
+                                       distutil.announce("Warning: swig shadow classes not copied")
+                       ## End of modification
+
+               return new_sources
+
+class SwigExtension(ExtensionWrap):
+       """
+       This class extends basic distutils Extension class,adding two keyword
+       arguments :
+               * swig_cpp,which triggers -c++ mode when swigging
+               * swig_include,which specifies -I flag when swigging
+       This class is meant to be build with mybuild_ext distutils command.
+       """
+       def __init__(self,swig_include=None,swig_cpp=None,**args):
+               ExtensionWrap.__init__(self,SwigWrapper(),**args)
+
+               assert((swig_include==None or type(swig_include) is ListType),
+                                 "swig_include must be a list of strings")
+
+               self.swig_include=swig_include or []
+               self.swig_cpp=swig_cpp
+