pycrtm/setup.py at master · JCSDA/pycrtm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import os, sys, configparser, shutil, time, glob
import setuptools
from skbuild import setup
def main():
    #Completely remove previous _skbuild, because cache will remember previous interation and ignore you if you change something.
    try:shutil.rmtree('_skbuild')
    except:pass

    #path of this file.
    scriptDir = os.path.split(os.path.abspath(__file__))[0]

    #read configuration
    coef_path, coef_dest, crtm_install, link_coef = readSetup('setup.cfg',scriptDir)
    if(link_coef):
        linkCoef(coef_path, coef_dest)
    os.environ['CRTM_INSTALL'] = crtm_install

    shutil.copy(os.path.join(scriptDir,'setup.cfg'),os.path.join(scriptDir,'pycrtm_setup.txt'))


    requires=['numpy']
    setup(
        name="pycrtm_jcsda",
        version='2.0.1',
        description='Python wrapper for the CRTM.',
        author='Bryan Karpowicz',
        requires=requires,
        include_package_data=True,
        packages=['pycrtm_'],
        py_modules=['crtm_io', 'pyCRTM'],
        zip_safe = False,)

    #os.remove('pycrtm.F90')
def readSetup(setup_file, scriptDir):
    cfg = configparser.ConfigParser()
    cfg.read( os.path.join(scriptDir,'setup.cfg') )
    crtm_install = cfg['Setup']['crtm_install']
    coef_path = cfg['Coefficients']['source_path']
    coef_dest = cfg['Coefficients']['path_used']
    link_coef = cfg.getboolean('Setup','link_from_source_to_path_used')
    return coef_path, coef_dest, crtm_install, link_coef

def linkCoef(coefDir,coefDest):
    print("Linking Coefficients.")
    cwd = os.getcwd()

    if( not os.path.isdir( coefDest )  ):
        print("Creating directory:{}".format(coefDest))
        os.makedirs( coefDest )
    else:
        print("Cleaning directory:{}".format(coefDest))
        shutil.rmtree( coefDest )
        print("Creating directory:{}".format(coefDest))
        os.makedirs( coefDest )
    print("Linking Coefficients in {} to {}".format(coefDir,coefDest))
    topdir = os.listdir(coefDir)
    # use os.walk to get files regardless if they're in an emc "fix" style directory, or crtm coeff tarball.
    searchPath = coefDir
    toLink = []
    filesPresent = []
    for root,dirs,filez in os.walk(searchPath):
        for name in filez:
            srcPath = os.path.join(root,name)
            if(not ('ODAS' in srcPath or 'Big_Endian' in srcPath)):
                curF = os.path.split(srcPath)[1]
                if (curF not in filesPresent):
                    toLink.append(srcPath)
                    filesPresent.append(curF)
    coefCnt = 0
    for l in toLink:
        os.symlink(l, os.path.join(coefDest,os.path.split(l)[1]))
        coefCnt+=1
    if coefCnt == 0:
        print("Warning! Linked zero coefficients!")
        time.sleep(30)
    else:
        print("Linked {:d} Coefficients.".format(coefCnt))
    print ("Checking for missing ODAS only coefficients.")
    SpcCoeffNc = glob.glob(os.path.join(coefDest,'*SpcCoeff*.nc'))
    SpcCoeffBin = glob.glob(os.path.join(coefDest,'*SpcCoeff*.bin'))
    missingTau = []
    for c in SpcCoeffNc:
        if(not os.path.exists(c.replace('SpcCoeff','TauCoeff'))):
            cc = os.path.split(c.replace('SpcCoeff','TauCoeff'))[-1]
            missingTau.append(cc)
    for c in SpcCoeffBin:
        if(not os.path.exists(c.replace('SpcCoeff','TauCoeff'))):
            cc = os.path.split(c.replace('SpcCoeff','TauCoeff'))[-1]
            missingTau.append(cc)
    toLink = []
    filesPresent = []
    for root,dirs,filez in os.walk(searchPath):
        for name in filez:
            srcPath = os.path.join(root,name)
            if(not ('ODPS' in srcPath or 'Big_Endian' in srcPath)):
                curF = os.path.split(srcPath)[1]
                if (curF in missingTau and curF not in filesPresent):
                    toLink.append(srcPath)
                    filesPresent.append(curF)
    for l in toLink:
        os.symlink(l, os.path.join(coefDest,os.path.split(l)[1]))
        coefCnt+=1

    print("Linked {:d} Coefficients.".format(coefCnt))
    time.sleep(10)
if __name__ == "__main__":
    main()