from distutils.core import setup
import glob
import subprocess
import os
import sys

if "--enable-devel" in sys.argv:
    install_dev = True
    sys.argv.remove("--enable-devel")
else:
    install_dev = False

Version = "0.6.0"
p = subprocess.Popen(
    ("git",
     "describe",
     "--tags"),
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE)
try:
    descr = p.stdout.readlines()[0].strip()
    Version = "-".join(descr.split("-")[:-2])
    if Version == "":
        Version = descr
except:
    Version = "0.9.pre-release"
    descr = Version

p = subprocess.Popen(
    ("git",
     "log",
     "-n1",
     "--pretty=short"),
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE)
try:
    commit = p.stdout.readlines()[0].split()[1]
except:
    commit = ""
f = open("src/python/version.py", "w")
print >>f, "__version__ = '%s'" % Version
print >>f, "__git_tag_describe__ = '%s'" % descr
print >>f, "__git_sha1__ = '%s'" % commit
f.close()

portrait_files = ["src/python/graphics/share/portraits.scr", ]

packages = {'pcmdi_metrics': 'src/python',
            'pcmdi_metrics.io': 'src/python/io',
            'pcmdi_metrics.pcmdi': 'src/python/pcmdi',
            'pcmdi_metrics.graphics': 'src/python/graphics',
            'pcmdi_metrics.driver': 'src/python/pcmdi/scripts/driver',
            }
scripts = ['src/python/pcmdi/scripts/pcmdi_metrics_driver.py',
           'src/python/pcmdi/scripts/pcmdi_metrics_driver_legacy.py',
           'src/python/pcmdi/scripts/pcmdi_compute_climatologies.py',
           'src/python/misc/scripts/install_metrics_from_branches.py',
           'demo/pmp_demo_1.py',
           'demo/pmp_demo.py',
           ]

demo_files = glob.glob("demo/*/*")
print "demo files"

data_files = [
              ('share/pmp/graphics/vcs', portrait_files),
              ('share/pmp/graphics/png', ['share/pcmdi/171101_doutriaux1_UVCDATLogo_446x119px_72dpi.png',
                                          'share/pcmdi/CDATLogo_1866x651px_300dpi.png',
                                          'share/pcmdi/CDATLogo_200x70px_72dpi.png',
                                          'share/pcmdi/CDATLogoText_1898x863px_300dpi.png',
                                          'share/pcmdi/CDATLogoText_200x91px_72dpi.png',
                                          'share/pcmdi/PCMDILogo_1588x520px_300dpi.png',
                                          'share/pcmdi/PCMDILogo_200x65px_72dpi.png',
                                          'share/pcmdi/PCMDILogo_300x98px_72dpi.png',
                                          'share/pcmdi/PCMDILogo_400x131px_72dpi.png',
                                          'share/pcmdi/PCMDILogo_500x164px_72dpi.png',
                                          'share/pcmdi/PCMDILogoText_1365x520px_300dpi.png',
                                          'share/pcmdi/PCMDILogo-old-oblong_377x300px_72dpi.png',
                                          'share/pcmdi/PCMDILogo-old_348x300px_72dpi.png',
                                          'share/pcmdi/PMPLogoText_1359x1146px_300dpi.png',
                                          'share/pcmdi/PMPLogo_1359x1146px_300dpi.png',
                                          'share/pcmdi/PMPLogo_500x421px_72dpi.png'
                                         ]),
              ('share/pmp', ('doc/obs_info_dictionary.json',
                               'share/pcmdi_metrics_table',
                               'share/disclaimer.txt',
                               'share/default_regions.py'
                            )),
              ('share/pmp/demo', demo_files),
             ]

if install_dev:
    print "Adding experimental packages"
    dev_packages = glob.glob("src/python/devel/*")
    dev_packages.remove("src/python/devel/example_dev")
    for p in dev_packages:
        if not os.path.isdir(p):
            dev_packages.pop(p)
    dev_scripts = []
    for p in dev_packages:
        scripts = glob.glob(os.path.join(p, "scripts", "*"))
        dev_scripts += scripts
    dev_pkg = {}
    dev_data = []
    for p in dev_packages:
        nm = p.replace("/", ".")
        nm = nm.replace("src.python.devel", "pcmdi_metrics")
        pnm = nm.split(".")[-1]
        pkg_dir = os.path.join(p, "lib")
        dev_pkg[nm] = pkg_dir
        data = glob.glob(os.path.join(p, "data", "*"))
        for d in data:
            dir_nm = os.path.split(d)[-1]
            dev_data.append([os.path.join(dir_nm, pnm),
                             glob.glob(os.path.join(d, "*"))])
    packages.update(dev_pkg)
    data_files += dev_data
    scripts += dev_scripts

setup(name='pcmdi_metrics',
      version=descr,
      author='PCMDI',
      description='model metrics tools',
      url='http://github.com/PCMDI/pcmdi_metrics',
      packages=packages.keys(),
      package_dir=packages,
      scripts=scripts,
      data_files=data_files
      # include_dirs = [numpy.lib.utils.get_include()],
      #  ext_modules = [
      #               Extension('pcmdi_metrics.exts',
      #               ['src/C/add.c',],
      #               library_dirs = [],
      #               libraries = [],
      #               define_macros = [],
      #               extra_compile_args = [],
      #               extra_link_args = [],
      #               ]
      )