-
Notifications
You must be signed in to change notification settings - Fork 8
/
setup.py
42 lines (38 loc) · 1.74 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
import os
from setuptools import setup
from subprocess import check_output
# Download the mwa_full_embedded_element_pattern.h5 file if it doesn't exist
datadir = os.path.join(os.path.dirname(__file__), 'mwa_pb', 'data')
h5file = os.path.join(datadir, 'mwa_full_embedded_element_pattern.h5')
print(h5file)
if not os.path.exists(h5file):
# Importing download functions here to avoid unnessiary imports when the file is available
import urllib.request
print("The mwa_full_embedded_element_pattern.h5 file does not exist. Downloading it from http://ws.mwatelescope.org")
response = urllib.request.urlopen("http://ws.mwatelescope.org/static/mwa_full_embedded_element_pattern.h5", timeout = 5)
content = response.read()
f = open(h5file, 'wb' )
f.write( content )
f.close()
print("Download complete")
setup(
name='mwa_pb',
version='1.4',
packages=['mwa_pb'],
package_data={'mwa_pb':['data/*.fits', 'data/*.txt', 'data/*.h5', 'data/*.fab', 'data/*.dat']},
url='https://github.com/MWATelescope/mwa_pb',
license='GPLv3',
author='MWA Team members, repo managed by Andrew Williams',
author_email='Andrew.Williams@curtin.edu.au',
description='MWA Primary beam code',
scripts=['scripts/beam_correct_image.py',
'scripts/beamtest.py',
'scripts/calc_jones.py',
'scripts/make_beam_test.py',
'scripts/mwa_sensitivity.py',
'scripts/plot_skymap.py',
'scripts/primarybeammap_tant_test.py',
'scripts/track_and_suppress.py'],
install_requires=["numpy", "astropy", "skyfield>=1.16", "matplotlib", "scipy>=0.15.1", "h5py"],
extras_require={'skymap':["ephem", "Pillow"]} # Needed only to generate sky maps in mwa_pb/skymap.py
)