/tags/v0_6_1b/toolkits/basemap/setup.py

https://github.com/ericliang/matplotlib
Python | 64 lines | 57 code | 4 blank | 3 comment | 2 complexity | dcbfff8ddae4cf03ffe54dd65da9a618 MD5 | raw file
    

  1. from distutils.core import setup, Extension
    2. 

  2. from distutils.util import convert_path
    3. 

  3. import sys, glob
    4. 

  4. 

  5. def dbf_macros():
    6. 

  6.     """Return the macros to define when compiling the dbflib wrapper.
    7. 

    8. 

  8.     The returned list specifies one macro, HAVE_UPDATE_HEADER, which is
    9. 

  9.     '1' if the dbflib version we will be compiling with has the
    10. 

  10.     DBFUpdateHeader function and '0' otherwise.  To check whether
    11. 

  11.     DBFUpdateHeader is available, we scan shapefil.h for the string
    12. 

  12.     'DBFUpdateHeader'.
    13. 

  13.     """
    14. 

  14.     f = open(convert_path("pyshapelib/shapelib/shapefil.h"))
    15. 

  15.     contents = f.read()
    16. 

  16.     f.close()
    17. 

  17.     if contents.find("DBFUpdateHeader") >= 0:
    18. 

  18.         return [("HAVE_UPDATE_HEADER", "1")]
    19. 

  19.     else:
    20. 

  20.         return [("HAVE_UPDATE_HEADER", "0")]
    21. 

  21. 

  22. deps = glob.glob('src/*.c')
    23. 

  23. 

  24. extensions = [Extension("proj4",deps,include_dirs = ['src'],)]
    25. 

  25. packages          = ['matplotlib/toolkits','matplotlib/toolkits/basemap']
    26. 

  26. package_dirs       = {'':'lib'}
    27. 

  27. 

  28. # don't build pyshapelib if it is already installed.
    29. 

    30. 

  30. try:
    31. 

  31.     import shapelib
    32. 

  32.     import dbflib
    33. 

  33. except:
    34. 

  34.     packages = packages + ['shapelib','dbflib']
    35. 

  35.     package_dirs['shapelib'] ='pyshapelib/lib/shapelib'
    36. 

  36.     package_dirs['dbflib'] ='pyshapelib/lib/dbflib'
    37. 

  37.     extensions = extensions + \
    38. 

  38.              [Extension("shapelibc",
    39. 

  39.                         ["pyshapelib/shapelib_wrap.c",
    40. 

  40.                          "pyshapelib/shapelib/shpopen.c",
    41. 

  41.                          "pyshapelib/shapelib/shptree.c"],
    42. 

  42.                         include_dirs = ["pyshapelib/shapelib"]),
    43. 

  43.               Extension("shptree",
    44. 

  44.                         ["pyshapelib/shptreemodule.c"],
    45. 

  45.                         include_dirs = ["pyshapelib/shapelib"]),
    46. 

  46.               Extension("dbflibc",
    47. 

  47.                         ["pyshapelib/dbflib_wrap.c",
    48. 

  48.                          "pyshapelib/shapelib/dbfopen.c"],
    49. 

  49.                         include_dirs = ["pyshapelib/shapelib"],
    50. 

  50.                         define_macros = dbf_macros()) ]
    51. 

  51. 

  52. datadir ='share/basemap-py'+repr(sys.version_info[0])+repr(sys.version_info[1])
    53. 

  53. 

  54. setup(
    55. 

  55.   name              = "basemap",
    56. 

  56.   version           = "0.6.1",
    57. 

  57.   description       = "Plot data on map projections with matplotlib",
    58. 

  58.   url               = "http://matplotlib.sourceforge.net/toolkits.html",
    59. 

  59.   author            = "Jeff Whitaker",
    60. 

  60.   author_email      = "jeffrey.s.whitaker@noaa.gov",
    61. 

  61.   data_files        = [(datadir,['data/countries_c.txt','data/states_c.txt','data/countries_l.txt','data/states_l.txt','data/gshhs_c.txt','data/gshhs_l.txt','data/countries_i.txt','data/states_i.txt','data/gshhs_i.txt'])],
    62. 

  62.   packages          = packages,
    63. 

  63.   package_dir       = package_dirs,
    64. 

  64.   ext_modules       = extensions)
    65.