import numpy

import os

import shutil

import glob

from setuptools import setup, Extension

from setuptools.command.build_py import build_py as _build_py

include_dirs = [numpy.get_include(),"include","include/cdTime"]

library_dirs = [ os.path.join("@prefix@","lib") ,'.']

include_dirs.append(os.path.join("@prefix@","include"))

libraries = []

link_args = []

for st in ["@NCLDFLAGS@", "@NCCFLAGS@",

"@UDUNITS2FLAGS@", "@UDUNITS2LDFLAGS@",

"@JSONCFLAGS@", "@JSONCLDFLAGS@",

"@UUIDFLAGS@", "@UUIDLDFLAGS@"]:

sp = st.strip().split()

for s in sp:

if s[:2]=='-L':

library_dirs.append(s[2:])

if s[:2]=='-l':

libraries.append(s[2:])

if s[:2]=='-I':

include_dirs.append(s[2:])

if s[:4]=='-Wl,':

link_args.append(s)

srcfiles = "@LIBSOURCES@".split()

macros=[]

for m in "@MACROS@".split():

macros.append((m[2:],None))

ld =[]

for p in library_dirs:

if os.path.exists(p):

ld.append(p)

library_dirs=ld

ld =[]

for p in include_dirs:

if os.path.exists(p):

ld.append(p)

include_dirs=ld

print('Setting up python module with:')

print('libraries:',libraries)

print('libdir:',library_dirs)

print('incdir',include_dirs)

print('src:',srcfiles)

print('macros:',macros)

class build_py(_build_py):

def run(self):

self._cmor_udunits_output = []

super().run()

xml_source = os.environ.get("CMOR_UDUNITS2_XML")

if not xml_source:

return

if not os.path.isfile(xml_source):

raise RuntimeError(f"CMOR_UDUNITS2_XML does not exist: {xml_source}")

target_dir = os.path.join(self.build_lib, "cmor", "data")

os.makedirs(target_dir, exist_ok=True)

for source_path in sorted(glob.glob(os.path.join(os.path.dirname(xml_source), "udunits2*.xml"))):

target_path = os.path.join(target_dir, os.path.basename(source_path))

shutil.copy2(source_path, target_path)

self._cmor_udunits_output.append(target_path)

def get_outputs(self, include_bytecode=1):

outputs = super().get_outputs(include_bytecode=include_bytecode)

xml_outputs = getattr(self, "_cmor_udunits_output", [])

outputs.extend(xml_outputs)

return outputs

setup (name = "CMOR",

version='@CMOR_PYTHON_PACKAGE_VERSION@',

author='Chris Mauzey, LLNL',

description = "Python Interface to CMOR output library",

url = "https://cmor.llnl.gov/",

zip_safe=False,

install_requires = ['numpy'],

packages = ['cmor', 'cmor.Test'],

package_dir = {'cmor': 'Lib', 'cmor.Test':'Test'},

package_data = {'cmor': ['data/*.xml']},

cmdclass = {'build_py': build_py},

# scripts=['scripts/PrePARE.py' ],

ext_modules = [

Extension('cmor._cmor',

srcfiles,

include_dirs = include_dirs,

library_dirs = library_dirs,

libraries = libraries,

define_macros = macros,

extra_compile_args = [ "-DgFortran"],

extra_link_args = link_args

),

]

)