forked from JanBojko/SEN2COR
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
231 lines (211 loc) · 9.02 KB
/
setup.py
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
#!/usr/bin/env python
#
# sen2cor setup script
#
# usage:
# to build distribution for windows:
# 'python setup.py sdist --formats=zip'
# else:
# 'python setup.py sdist --formats=gztar'
# to install at target system:
# 'python setup.py install' and follow the instructions ...
#
from setuptools import setup, find_packages
from distutils.dir_util import mkpath
from distutils.file_util import copy_file
import os, sys, platform
name = 'sen2cor'
#
# This needs to be changed with each new version:
# ------------------------------------------------
version = '2.2.1'
longVersion = '02.02.01'
#
# Do not change anything below:
# ----------------------------------------------------------------------------
consoleScript1 = 'L2A_Process = sen2cor.L2A_Process:main'
consoleScript2 = 'L2A_Process-' + version + ' = sen2cor.L2A_Process:main'
consoleScript3 = 'L2A_Process-' + longVersion + ' = sen2cor.L2A_Process:main'
L2A_Process = 'L2A_Process'
L2A_Process_Version = L2A_Process + '-' + version
def copyConfiguration():
sys.stdout.write('\nSEN2COR ' + version + ' setup script:\n')
sys.stdout.write('This will finish the configuration of the environment settings.\n')
user_input = raw_input('\nOK to continue? [y/n]: ')
if user_input == 'n':
return False
cfgf = 'L2A_GIPP.xml'
srcf = os.path.join(modulefolder, os.path.join('cfg', cfgf))
SEN2COR_HOME = os.path.join(cfghome, 'sen2cor')
break_condition = True
while True:
sys.stdout.write('\nPlease input a path for the sen2cor home directory:\n')
sys.stdout.write('default is: ' + SEN2COR_HOME + '\n')
user_input = raw_input('Is this OK? [y/n]: ')
if user_input == 'n':
SEN2COR_HOME = raw_input('New path: ')
sys.stdout.write('New path is: ' + SEN2COR_HOME + '\n')
user_input = raw_input('Is this OK? [y/n]: ')
if user_input == 'y':
break_condition = True
else:
break_condition = False
else:
break_condition = True
if break_condition:
os.environ['SEN2COR_HOME'] = SEN2COR_HOME
s2l2appcfg = os.path.join(SEN2COR_HOME, 'cfg')
mkpath(s2l2appcfg)
copy_file(srcf, os.path.join(s2l2appcfg, cfgf))
break
sys.stdout.write('Setting environment for sen2cor ...\n')
if system == 'Windows':
try:
setGDAL_DATA = 'setx GDAL_DATA ' + gdal_data
sys.stdout.write('Setting environment variable GDAL_DATA:')
os.system(setGDAL_DATA)
setSEN2COR_HOME = 'setx SEN2COR_HOME ' + SEN2COR_HOME
sys.stdout.write('Setting environment variable SEN2COR_HOME:')
os.system(setSEN2COR_HOME)
setSEN2COR_BIN = 'setx SEN2COR_BIN ' + modulefolder
sys.stdout.write('Setting environment variable SEN2COR_BIN:')
os.system(setSEN2COR_BIN)
except:
sys.stderr.write('Error in environment settings!\n')
return False
sys.stdout.write('Congratulations, you are done!\n')
else:
# setting the environments for the application into L2A_Bashrc (Linux and MacOSX):
L2A_Bashrc = '#!/usr/bin/env bash\n'
L2A_Bashrc += '# BEGIN\n'
L2A_Bashrc += '# Sen2Cor environmental setup, version ' + version + '\n'
L2A_Bashrc += '# settings automatically generated by setup.py\n'
L2A_Bashrc += '# source this script either manually via: "source L2A_Bashrc"\n'
L2A_Bashrc += '# or call it from your .bashrc or .profile script\n#\n'
if system == 'Darwin':
L2A_Bashrc += 'export LC_ALL=en_US.UTF-8\n'
L2A_Bashrc += 'export LANG=en_US.UTF-8\n'
L2A_Bashrc += 'export SEN2COR_HOME=' + SEN2COR_HOME + '\n'
L2A_Bashrc += 'export SEN2COR_BIN=' + modulefolder + '\n'
L2A_Bashrc += 'export GDAL_DATA=' + gdal_data + '\n'
L2A_Bashrc += '# END\n'
sys.stdout.write('Creating L2A_Bashrc script under:\n' + SEN2COR_HOME + '\n')
try:
textFile = open(SEN2COR_HOME + '/L2A_Bashrc', 'w')
textFile.write(L2A_Bashrc)
textFile.close()
except:
sys.stderr.write('Cannot create the L2A_Bashrc script under:\n' + SEN2COR_HOME + '\n')
return False
# creating the L2A_Process.bash script:
CONDA_BIN = sys.prefix + '/bin'
L2A_Process_bash = '#!/usr/bin/env bash\n'
L2A_Process_bash += '# BEGIN\n'
L2A_Process_bash += '# Sen2Cor L2A_Process.bash, version ' + version + '\n'
L2A_Process_bash += '# settings automatically generated by setup.py\n#\n'
L2A_Process_bash += 'export PATH=' + CONDA_BIN + ':$PATH\n'
L2A_Process_bash += 'source ./L2A_Bashrc\n'
L2A_Process_bash += 'L2A_Process-$1 $2 $3 $4\n'
L2A_Process_bash += '# END\n'
sys.stdout.write('Creating L2A_Process.bash script under:\n' + SEN2COR_HOME + '\n')
try:
textFile = open(SEN2COR_HOME + '/L2A_Process.bash', 'w')
textFile.write(L2A_Process_bash)
textFile.close()
except:
sys.stderr.write('Cannot create the L2A_Process.bash script under:\n' + SEN2COR_HOME + '\n')
return False
# create the glymur configuration file for OpenJPEG2:
glymurrc = '[library]\n'
glymurrc += 'openjp2: ' + libOpj2Target + '\n'
glymurrcPath = os.path.join(cfghome, '.config/glymur')
glymurrcFile = os.path.join(glymurrcPath, 'glymurrc')
sys.stdout.write('Creating the configuration file for OpenJPEG2 under:\n' + glymurrcPath + '\n\n')
try:
mkpath(glymurrcPath)
except:
sys.stdout.write('Path already exists ...')
try:
textFile = open(glymurrcFile, 'w')
textFile.write(glymurrc)
textFile.close()
except:
sys.stderr.write('Cannot create the configuration file for OpenJPEG2 under:\n ' + glymurrcPath + '\n\n')
return False
sys.stdout.write('Congratulations, you are nearly done ...\n')
sys.stdout.write('Last step: cd to ' + SEN2COR_HOME + ',\n')
sys.stdout.write('source the <L2A_Bashrc> script either manually via: "source L2A_Bashrc"\n')
sys.stdout.write('or integrate this call in your .bashrc or .profile script. Afterwards,\n')
sys.stdout.write('- you can call the processor from everywhere via: "L2A_Process"\n')
sys.stdout.write('- you will find the default configuration called "L2A_GIPP.xml" under:\n' + s2l2appcfg + '\n\n')
return True
setup(
name=name,
version=version,
description='sen2cor: Sentinel 2 Level 2A Prototype Processor',
long_description=open('README.md').read(),
packages=find_packages(),
include_package_data=True,
platforms=['linux-x86_64', 'macosx-10.5-x86_64', 'win-amd64'],
entry_points={
'console_scripts': [consoleScript1, consoleScript2, consoleScript3, ]
},
zip_safe=False,
)
try:
sys.argv[1] == True
except:
sys.stderr.write('argument must either be "sdist" or "install ..."\n')
sys.exit(0)
if (sys.argv[1] != 'install'):
sys.exit(0)
system = platform.system()
if system == 'Windows':
prefix = os.path.join(sys.prefix, 'Lib', 'site-packages')
else:
prefix = os.path.join(sys.prefix, 'lib', 'python2.7', 'site-packages')
modulefolder = os.path.join(prefix, name + '-' + version + '-py2.7.egg/sen2cor')
buildfolder = os.path.join(modulefolder, 'build')
gdal_data = os.path.join(modulefolder, 'cfg', 'gdal_data')
if system == 'Darwin':
platform = 'lib.macosx-10.5-x86_64-2.7'
libopj2 = 'libopenjp2.dylib'
libAtmCor = 'L2A_AtmCorr.so'
targetfolder = os.path.join(sys.prefix, 'lib')
try:
cfghome = os.environ['XDG_CONFIG_HOME']
except:
cfghome = os.environ['HOME']
elif system == 'Linux':
platform = 'lib.linux-x86_64-2.7'
libopj2 = 'libopenjp2.so'
libAtmCor = 'L2A_AtmCorr.so'
targetfolder = os.path.join(sys.prefix, 'lib')
try:
cfghome = os.environ['XDG_CONFIG_HOME']
except:
cfghome = os.environ['HOME']
elif system == 'Windows':
platform = 'lib.win-amd64-2.7'
libopj2 = 'openjp2.dll'
libAtmCor = 'L2A_AtmCorr.pyd'
targetfolder = os.path.join(sys.prefix, 'Scripts')
cfghome = os.path.join(os.environ['USERPROFILE'], 'documents')
os.system('conda install --yes -c sunpy glymur')
os.system('conda install --yes libgdal')
os.system('conda install --yes gdal')
# better to do this manually, see SUM section uninstall:
# os.system('conda clean --yes --tarballs --index-cache --packages --source-cache')
libOpj2Source = os.path.join(buildfolder, platform, libopj2)
libAtmSource = os.path.join(buildfolder, platform, libAtmCor)
libOpj2Target = os.path.join(targetfolder, libopj2)
libAtmTarget = os.path.join(modulefolder, libAtmCor)
copy_file(libOpj2Source, libOpj2Target)
copy_file(libAtmSource, libAtmTarget)
if copyConfiguration() == True:
sys.stdout.write('Installation sucessfully performed.\n')
else:
sys.stdout.write('Errors during installation occurred.\n')
sys.exit(0)
if __name__ == '__main__':
pass