"""
 Installation script for SAS models

  - To compile and install:
      python setup.py install
  - To create distribution:
      python setup.py bdist_wininst

"""
import sys
import os
from distutils.core import setup
from distutils.core import Extension
from distutils import util
from distutils import sysconfig

def get_c_files(path):
    """
        Utility function to return all files 
        to be compiled in a directory
    """
    clist = []
    for file in os.listdir(path):
        if file.endswith(".cc") or file.endswith(".c"):
            clist.append("%s/%s" % (path, file))
    return clist

# Top package name
#pck_top = "sasModeling"
pck_top = "sas.sascalc.simulation"
pck_dir = os.path.join("src","sas", "simulation")
# The temp directory that the compiled files will be put in
tempdir = "build/temp."+util.get_platform()+'-'+sysconfig.get_python_version()
if os.name=='nt':
    tempdir += "/Release"

# List of sub modules
mod_list = ["iqPy", "geoshapespy", "pointsmodelpy", "analmodelpy"]

# Define package directors, will be extended below
pck_dirs = {pck_top:pck_dir}

# List of modules to install, will be extende below
pck_list = [pck_top]

# List of python extensions, will be extended below
exts = []

# List of include dirs, will be extended below
incl_dirs = ["iqPy/libiqPy/tnt"]

# List of files to compile, per module, will be extended below
file_list = {}

# List of dependencies
deps = {}
deps["iqPy"] = []
deps["geoshapespy"] = ["iqPy"]
deps["analmodelpy"] = ["iqPy", "geoshapespy"]
deps["pointsmodelpy"] = ["iqPy", "geoshapespy"]

# List of extra objects to link with, per module, will be extended below 
libs = {}

for module in mod_list:
    temp_path = os.path.join(pck_dir, module, module)
    pck_dirs["%s.%s" % (pck_top, module)] = temp_path
    pck_list.append("%s.%s" % (pck_top, module))

    src_m = os.path.join(pck_dir, module, "%smodule" %(module))
    src_l = os.path.join(pck_dir, module, "lib%s" % (module))

    files = get_c_files(src_m)
    list_lib = get_c_files(src_l)

    libs[module] = []    
    for dep in deps[module]:
        
        for f in get_c_files(os.path.join(pck_dir, dep, "lib%s" % dep)):
            index = f.rindex('.')
            if os.name=='nt':
                fo = f[:index]+'.obj'
            else:
                fo = f[:index]+'.o'
            libs[module].append("%s/%s" %(tempdir, fo))
        
    files.extend(list_lib)
    file_list[module] = files
    
    incl_dirs.append(src_m)
    incl_dirs.append(src_l)
    
for module in mod_list:
    exts.append( Extension("%s.%s.%s" % (pck_top, module, module),
        sources = file_list[module],
        extra_objects=libs[module],
        include_dirs=incl_dirs) )

setup(
    name="sasrealspace_modeling",
    version = "0.2",
    description = "Python module for SAS simulation",
    author = "University of Tennessee",
    url = "http://danse.us/trac/sas",
    package_dir = pck_dirs,
    packages    = pck_list,
    ext_modules = exts)