from scipy_distutils.core import setup,Extension
import os

version_no='1.0.80'

include_dirs=['./src']

HAVE_NAG=os.getenv('HAVE_NAG','0')
HAVE_NAG=int(HAVE_NAG)

HAVE_QUAD=os.getenv('HAVE_QUAD','0')
HAVE_QUAD=int(HAVE_QUAD)

if HAVE_QUAD:
    tmatrix_sig_file='./src/tmatrixQUAD.pyf'
    tmd_sig_file='./src/tmdQUAD.pyf'
else:
    tmatrix_sig_file='./src/tmatrix.pyf'
    tmd_sig_file='./src/tmd.pyf'

if HAVE_NAG:
    tmdsources=[tmd_sig_file,'./src/tmd_NAG.f','./src/ampld2.f',
                './src/main_dummy.c']
    tmatrixsources=[tmatrix_sig_file,'./src/tmatrix_NAG.f',
                    './src/avg_tmatrix.f','./src/ampld2.f','./src/main_dummy.c']
    if HAVE_QUAD:
        tmatrixsources.extend(['./src/tmatrixq_NAG.f','./src/ampld2q.f'])
        tmdsources.extend(['./src/tmq_NAG.f','./src/ampld2q.f'])
else:
    tmdsources=[tmd_sig_file,'./src/tmd.f','./src/main_dummy.c']
    tmatrixsources=[tmatrix_sig_file,'./src/tmatrix.f',
                    './src/avg_tmatrix.f','./src/main_dummy.c']
    if HAVE_QUAD:
        tmatrixsources.append('./src/tmatrixq.f')
        tmdsources.append('./src/tmq.f')


scatsubs=Extension(name = 'scatsubs',
		   sources = ['./src/scatsubs.pyf','./src/scatsubs.f','./src/main_dummy.c'],
		   include_dirs=include_dirs
		   )

tmd=Extension(name = 'tmd',
		   sources = tmdsources,
		   include_dirs=include_dirs
	      )

tmatrix=Extension(name = 'tmatrix',
		   sources = tmatrixsources,
		   include_dirs=include_dirs
		  )

refr_ind=Extension(name = 'refr_ind',
		   sources = ['./src/refr_ind.f','./src/main_dummy.c'],
		   include_dirs=include_dirs
		  )

REFICE=Extension(
    name = 'REFICE',
    sources = ['./src/REFICE.pyf','./src/REFICE.f','./src/main_dummy.c'],
    include_dirs=include_dirs
    )

erfc=Extension(
    name='erfc',
    sources=['./src/erfc_wrap.c','./src/erfc.c',
	     './src/error.c','./src/stream.c'],
    include_dirs=['./src']
    )

setup(name = 'PyARTS',
      version = version_no,
      description  = 'ARTS related python modules',
      author       = 'Cory Davis',
      author_email = 'cory@met.ed.ac.uk',
      packages  = ['PyARTS'],
      ext_package = 'PyARTS',
      ext_modules  = [scatsubs,tmd,tmatrix,REFICE,erfc,refr_ind],#refr_ind
      scripts = ['scat_file_batch']
      )