exoclock 1.0.1__py3-none-any.whl

exoclock/_0_local_bjd_hjd.py

@@ -0,0 +1,184 @@
+from my_packages import *
+
+# https://ssd.jpl.nasa.gov/horizons.cgi#top
+# Astrometric RA & DEC
+# One-way (down-leg) light-time
+# Delta-T (TDB - UT)
+# Start=2000-01-01 12:00, Stop=2040-01-01 12:00, Step=6 h
+
+
+
+data_t = np.arange(2451545.0, 2451545.0+14610.000000001, 0.25)
+
+data_ra = []
+data_ra_noleaps = []
+data_dec = []
+data_d = []
+data_dt = []
+leaps = 0
+for xx in open('/Users/angelos/Desktop/BJD.txt').readlines():
+    test_ra = float(xx.split()[2])+float(xx.split()[3])/60.0+float(xx.split()[4])/3600.0
+    test_dec = (abs(float(xx.split()[5]))+float(xx.split()[6])/60.0+float(xx.split()[7])/3600.0)*float('{0}1'.format(xx.split()[5][0]))
+    if len(data_ra) > 0:
+        if test_ra < data_ra_noleaps[-1]:
+            leaps += 1
+    data_ra.append((test_ra + 24 * leaps) * 15 * np.pi / 180)
+    data_ra_noleaps.append(test_ra)
+    data_dec.append(test_dec * np.pi / 180)
+    data_d.append(float(xx.split()[-2]))
+    data_dt.append(float(xx.split()[-1]))
+
+plt.plot(data_t, data_ra, 'o')
+plt.show()
+
+plt.plot(data_t, data_dec, 'o')
+plt.show()
+
+plt.plot(data_t, data_d, 'o')
+plt.show()
+
+plt.plot(data_t, data_dt, 'o')
+plt.show()
+
+
+data_dict = {int(ff):{'t':np.array([]), 'ra':np.array([]), 'dec':np.array([]), 'd':np.array([]), 'dt':np.array([])} for ff in np.arange(2451545.0, 2451545.0+14610)}
+for xx in range(len(data_t)):
+    indx = int(data_t[xx])
+    if indx in data_dict:
+        data_dict[indx]['t'] = np.append(data_dict[indx]['t'], data_t[xx])
+        data_dict[indx]['ra'] = np.append(data_dict[indx]['ra'], data_ra[xx])
+        data_dict[indx]['dec'] = np.append(data_dict[indx]['dec'], data_dec[xx])
+        data_dict[indx]['d'] = np.append(data_dict[indx]['d'], data_d[xx])
+        data_dict[indx]['dt'] = np.append(data_dict[indx]['dt'], data_dt[xx])
+    if indx == data_t[xx]:
+        indx = int(data_t[xx] - 1)
+        if indx in data_dict:
+            data_dict[indx]['t'] = np.append(data_dict[indx]['t'], data_t[xx])
+            data_dict[indx]['ra'] = np.append(data_dict[indx]['ra'], data_ra[xx])
+            data_dict[indx]['dec'] = np.append(data_dict[indx]['dec'], data_dec[xx])
+            data_dict[indx]['d'] = np.append(data_dict[indx]['d'], data_d[xx])
+            data_dict[indx]['dt'] = np.append(data_dict[indx]['dt'], data_dt[xx])
+
+plc.save_dict(data_dict, '/Users/angelos/Desktop/bjd_dict.pickle')
+
+
+
+
+
+
+
+
+
+data_t = np.arange(2451545.0, 2451545.0+14610.000000001, 0.25)
+
+data_ra = []
+data_ra_noleaps = []
+data_dec = []
+data_d = []
+data_dt = []
+leaps = 0
+for xx in open('/Users/angelos/Desktop/HJD.txt').readlines():
+    test_ra = float(xx.split()[2])+float(xx.split()[3])/60.0+float(xx.split()[4])/3600.0
+    test_dec = (abs(float(xx.split()[5]))+float(xx.split()[6])/60.0+float(xx.split()[7])/3600.0)*float('{0}1'.format(xx.split()[5][0]))
+    if len(data_ra) > 0:
+        if test_ra < data_ra_noleaps[-1]:
+            leaps += 1
+    data_ra.append((test_ra + 24.0 * leaps) * 15.0 * np.pi / 180.0)
+    data_ra_noleaps.append(test_ra)
+    data_dec.append(test_dec * np.pi / 180)
+    data_d.append(float(xx.split()[-2]))
+    data_dt.append(float(xx.split()[-1]))
+
+plt.plot(data_t, data_ra, 'o')
+plt.show()
+
+plt.plot(data_t, data_dec, 'o')
+plt.show()
+
+plt.plot(data_t, data_d, 'o')
+plt.show()
+
+plt.plot(data_t, data_dt, 'o')
+plt.show()
+
+
+data_dict = {int(ff): {'t':np.array([]), 'ra':np.array([]), 'dec':np.array([]), 'd':np.array([]), 'dt':np.array([])} for ff in np.arange(2451545.0, 2451545.0+14610)}
+for xx in range(len(data_t)):
+    indx = int(data_t[xx])
+    if indx in data_dict:
+        data_dict[indx]['t'] = np.append(data_dict[indx]['t'], data_t[xx])
+        data_dict[indx]['ra'] = np.append(data_dict[indx]['ra'], data_ra[xx])
+        data_dict[indx]['dec'] = np.append(data_dict[indx]['dec'], data_dec[xx])
+        data_dict[indx]['d'] = np.append(data_dict[indx]['d'], data_d[xx])
+        data_dict[indx]['dt'] = np.append(data_dict[indx]['dt'], data_dt[xx])
+    if indx == data_t[xx]:
+        indx = int(data_t[xx] - 1)
+        if indx in data_dict:
+            data_dict[indx]['t'] = np.append(data_dict[indx]['t'], data_t[xx])
+            data_dict[indx]['ra'] = np.append(data_dict[indx]['ra'], data_ra[xx])
+            data_dict[indx]['dec'] = np.append(data_dict[indx]['dec'], data_dec[xx])
+            data_dict[indx]['d'] = np.append(data_dict[indx]['d'], data_d[xx])
+            data_dict[indx]['dt'] = np.append(data_dict[indx]['dt'], data_dt[xx])
+
+plc.save_dict(data_dict, '/Users/angelos/Desktop/hjd_dict.pickle')
+
+
+
+
+
+
+
+import numpy as np
+import matplotlib.pyplot as plt
+data_t = np.arange(2451545.0, 2451545.0+14610.000000001, 0.25)
+
+data_ra = []
+data_ra_noleaps = []
+data_dec = []
+data_d = []
+data_dt = []
+leaps = 0
+for xx in open('/Users/angelos/Downloads/MOON.txt').readlines():
+    test_ra = float(xx.split()[2])+float(xx.split()[3])/60.0+float(xx.split()[4])/3600.0
+    test_dec = (abs(float(xx.split()[5]))+float(xx.split()[6])/60.0+float(xx.split()[7])/3600.0)*float('{0}1'.format(xx.split()[5][0]))
+    if len(data_ra) > 0:
+        if test_ra < data_ra_noleaps[-1]:
+            leaps += 1
+    data_ra.append((test_ra + 24.0 * leaps) * 15.0 * np.pi / 180.0)
+    data_ra_noleaps.append(test_ra)
+    data_dec.append(test_dec * np.pi / 180)
+    data_d.append(float(xx.split()[-2]))
+    data_dt.append(float(xx.split()[-1]))
+
+plt.plot(data_t, data_ra, 'o')
+plt.show()
+
+plt.plot(data_t, data_dec, 'o')
+plt.show()
+
+plt.plot(data_t, data_d, 'o')
+plt.show()
+
+plt.plot(data_t, data_dt, 'o')
+plt.show()
+
+
+data_dict = {int(ff): {'t':np.array([]), 'ra':np.array([]), 'dec':np.array([]), 'd':np.array([]), 'dt':np.array([])} for ff in np.arange(2451545.0, 2451545.0+14610)}
+for xx in range(len(data_t)):
+    indx = int(data_t[xx])
+    if indx in data_dict:
+        data_dict[indx]['t'] = np.append(data_dict[indx]['t'], data_t[xx])
+        data_dict[indx]['ra'] = np.append(data_dict[indx]['ra'], data_ra[xx])
+        data_dict[indx]['dec'] = np.append(data_dict[indx]['dec'], data_dec[xx])
+        data_dict[indx]['d'] = np.append(data_dict[indx]['d'], data_d[xx])
+        data_dict[indx]['dt'] = np.append(data_dict[indx]['dt'], data_dt[xx])
+    if indx == data_t[xx]:
+        indx = int(data_t[xx] - 1)
+        if indx in data_dict:
+            data_dict[indx]['t'] = np.append(data_dict[indx]['t'], data_t[xx])
+            data_dict[indx]['ra'] = np.append(data_dict[indx]['ra'], data_ra[xx])
+            data_dict[indx]['dec'] = np.append(data_dict[indx]['dec'], data_dec[xx])
+            data_dict[indx]['d'] = np.append(data_dict[indx]['d'], data_d[xx])
+            data_dict[indx]['dt'] = np.append(data_dict[indx]['dt'], data_dt[xx])
+
+plc.save_dict(data_dict, '/Users/angelos/Desktop/moon_dict.pickle')

exoclock/__author__.txt

@@ -0,0 +1 @@
+Angelos Tsiaras

exoclock/__author_email__.txt

@@ -0,0 +1 @@
+aggelostsiaras@gmail.com

exoclock/__databases_setup__.py

@@ -0,0 +1,182 @@
+
+import os
+import time
+import shutil
+
+from .processes.files import open_dict, download
+
+try:
+    import zipfile
+    download_zip = True
+except:
+    download_zip = False
+
+
+def _setup_database(package_data, database_name):
+
+        print('exoclock: Checking {0} database...'.format(database_name))
+
+        # define paths
+
+        database_directory_path = os.path.join(package_data.databases_directory_path, database_name)
+        database_file_path = os.path.join(package_data.databases_directory_path, database_name + '.pickle')
+        database_link_file_path = os.path.join(package_data.databases_directory_path, database_name + '_link.txt')
+        database_file_path_new = os.path.join(package_data.databases_directory_path, database_name + '_new.pickle')
+        database_file_path_old = os.path.join(package_data.databases_directory_path, database_name + '_old.pickle')
+        last_update_file_path = os.path.join(package_data.databases_directory_path, '{0}_last_update.txt'.format(database_name))
+
+        # define paths
+
+        # check if everything exists, if not reset database
+
+        if not os.path.isdir(database_directory_path) or not os.path.isfile(database_file_path) or not os.path.isfile(database_link_file_path):
+
+            try:
+                shutil.rmtree(database_directory_path)
+            except:
+                pass
+
+            try:
+                os.remove(database_file_path)
+            except:
+                pass
+
+            try:
+                os.remove(database_file_path_old)
+            except:
+                pass
+
+            try:
+                os.remove(database_file_path_new)
+            except:
+                pass
+
+            try:
+                os.remove(database_link_file_path)
+            except:
+                pass
+
+            try:
+                os.remove(last_update_file_path)
+            except:
+                pass
+
+            os.mkdir(database_directory_path)
+
+
+            print(package_data.databases[package_data.version][database_name], database_file_path)
+
+            if not download(package_data.databases[package_data.version][database_name], database_file_path):
+                print('\n{0} features cannot be used.'.format(database_name))
+                return False
+            else:
+                shutil.copy(database_file_path, database_file_path_old)
+                w = open(database_link_file_path, 'w')
+                w.write(package_data.databases[package_data.version][database_name])
+                w.close()
+
+                try:
+                    new_database = open_dict(database_file_path)
+                    download(new_database['zipfile'], database_directory_path + '.zip')
+                    new_database = zipfile.ZipFile(database_directory_path + '.zip', 'r')
+                    here = os.path.abspath('.')
+                    os.chdir(package_data.databases_directory_path)
+                    new_database.extractall()
+                    os.chdir(here)
+                    os.remove(database_directory_path + '.zip')
+                except:
+                    pass
+
+        # check if everything exists, if not reset database
+
+        # download database if there is an update
+
+        if package_data.databases[package_data.version][database_name] != open(database_link_file_path).read():
+
+            if not download(package_data.databases[package_data.version][database_name], database_file_path_new):
+                pass
+            else:
+                shutil.move(database_file_path, database_file_path_old)
+                shutil.move(database_file_path_new, database_file_path)
+
+                w = open(database_link_file_path, 'w')
+                w.write(package_data.databases[package_data.version][database_name])
+                w.close()
+
+        # download database if there is an update
+
+        # check all files in database, remove files that need to be updated
+
+        current_database = open_dict(database_file_path_old)
+        new_database = open_dict(database_file_path)
+
+        for dbx_file in current_database['files']:
+
+            if dbx_file not in new_database['files']:
+                try:
+                    os.remove(os.path.join(package_data.databases_directory_path,
+                                           new_database['files'][dbx_file]['local_path']))
+                except:
+                    pass
+            elif new_database['files'][dbx_file]['link'] != current_database['files'][dbx_file]['link']:
+                try:
+                    os.remove(os.path.join(package_data.databases_directory_path,
+                                           new_database['files'][dbx_file]['local_path']))
+                except:
+                    pass
+
+        # check for updates, remove files that need to be updated
+
+        # download missing files
+
+        final_check = True
+
+        for dbx_file in new_database['files']:
+            if not os.path.isfile(os.path.join(package_data.databases_directory_path,
+                                               new_database['files'][dbx_file]['local_path'])):
+                try:
+                    os.remove(last_update_file_path)
+                except:
+                    pass
+                if not download(new_database['files'][dbx_file]['link'],
+                                os.path.join(package_data.databases_directory_path,
+                                             new_database['files'][dbx_file]['local_path'])):
+                    final_check = False
+
+        # download missing files
+
+        # update files from external links
+
+        frequency = new_database['frequency']
+        if frequency:
+
+            try:
+                last_update_date = int(open(last_update_file_path).read())
+            except:
+                last_update_date = 0
+
+            today = int(time.strftime('%y%m%d'))
+
+            if today >= last_update_date + frequency:
+
+                for dbx_file in new_database['files']:
+                    if 'external_link' in new_database['files'][dbx_file]:
+                        print('\tUpdating: ', dbx_file)
+                        if not download(new_database['files'][dbx_file]['external_link'],
+                                        os.path.join(package_data.databases_directory_path,
+                                                     new_database['files'][dbx_file]['local_path'])):
+                            final_check = False
+
+                w = open(last_update_file_path, 'w')
+                w.write(time.strftime('%y%m%d'))
+                w.close()
+
+        # update files from external links
+
+        if not final_check:
+            print('\n{0} features cannot be used.'.format(database_name))
+            return False
+        else:
+            if current_database != new_database:
+                shutil.copy(database_file_path, database_file_path_old)
+            return database_directory_path

exoclock/__description__.txt

@@ -0,0 +1 @@
+A python package for programmatic interaction with the ExoClock project archives.

exoclock/__init__.py

@@ -0,0 +1,20 @@
+import os
+
+__version__ = open(os.path.join(os.path.abspath(os.path.dirname(__file__)), '__version__.txt')).read()
+__author__ = open(os.path.join(os.path.abspath(os.path.dirname(__file__)), '__author__.txt')).read()
+__author_email__ = open(os.path.join(os.path.abspath(os.path.dirname(__file__)), '__author_email__.txt')).read()
+__description__ = open(os.path.join(os.path.abspath(os.path.dirname(__file__)), '__description__.txt')).read()
+__url__ = open(os.path.join(os.path.abspath(os.path.dirname(__file__)), '__url__.txt')).read()
+
+from .catalogues.catalogues import *
+from .catalogues.simbad import *
+
+from .spacetime.angles import *
+from .spacetime.moments import *
+from .spacetime.observatories import *
+from .spacetime.targets import *
+
+from .processes.files import *
+
+from .errors import *
+from .databases import *

exoclock/__url__.txt

@@ -0,0 +1 @@
+https://github.com/atsiaras/exoclock

exoclock/__version__.txt

@@ -0,0 +1 @@
+1.0.1

exoclock/catalogues/catalogues.py

@@ -0,0 +1,121 @@
+
+__all__ = ['get_planet', 'get_system', 'get_all_planets', 'locate_planet', 'locate_system', 'get_all_planets']
+
+import numpy as np
+
+
+from .simbad import simbad_search_by_name
+
+from ..spacetime.angles import *
+from ..spacetime.angles import _request_angle, _reformat_or_request_angle
+from ..spacetime.targets import *
+from ..databases import exoclock_data
+from ..errors import *
+
+
+def _flat_name(name):
+
+    flat_name_list = [
+        [' ', ''],
+        ['-', ''],
+        ['cancri', 'cnc'],
+        ['hatp10', 'wasp11'],
+        ['wasp40', 'hatp27'],
+        ['wasp51', 'hatp30'],
+        ['wasp86', 'kelt12'],
+        ['kelt22', 'wasp173'],
+    ]
+
+    name = name.lower()
+
+    for char in flat_name_list:
+        name = name.replace(char[0], char[1])
+
+    return name
+
+
+def _search_by_planet(name):
+
+    planets = exoclock_data.ecc()['planets']
+
+    name_or = name
+
+    if name in planets:
+        return name
+
+    else:
+        for i in planets:
+            if _flat_name(i) == _flat_name(name):
+                return str(i)
+            elif (_flat_name(i)[-1] == _flat_name(name)[-1] and _flat_name(i)[:-2] == _flat_name(name)[:-1] and
+                  _flat_name(i)[-2] in ['a', 'b', 'n']):
+                return str(i)
+
+    raise ExoClockInputError('No planet {0} found in the catalogue.'.format(name_or))
+
+
+def get_all_planets():
+
+    return list(exoclock_data.ecc()['planets'].keys())
+
+
+def get_planet(name):
+
+    name = _search_by_planet(name)
+
+    planet_data = exoclock_data.ecc()['planets'][name]
+    star_data = exoclock_data.ecc()['stars'][planet_data['star']]
+    star_data['ra_deg'] = Hours(star_data['ra']).deg()
+    star_data['dec_deg'] = Degrees(star_data['dec']).deg_coord()
+
+    return {'name': name, 'planet': planet_data, 'star': star_data}
+
+
+def get_system(star_name):
+
+    target = simbad_search_by_name(star_name)
+    if not target:
+        return []
+
+    test = list(set(target.all_names).intersection(list(exoclock_data.ecc()['hosts'])))
+
+    if len(test) == 0:
+        return []
+
+    planets = exoclock_data.ecc()['hosts'][test[0]]
+
+    return [get_planet(planet) for planet in planets]
+
+
+def locate_planet(ra, dec, radius=Degrees(0.02), observation_time=None, observation_type=None):
+
+    _request_angle(ra)
+    _request_angle(dec)
+    _request_angle(radius)
+
+    pointing = FixedTarget(ra, dec)
+
+    test_planets = []
+
+    for test_planet_name in get_all_planets():
+        test_planet = get_planet(test_planet_name)
+        test_planet = FixedTarget(Degrees(test_planet['star']['ra_deg']), Degrees(test_planet['star']['dec_deg']))
+        test_planets.append([pointing.distance_on_sphere(test_planet).deg(), test_planet_name])
+
+    test_planets.sort()
+
+    if test_planets[0][0] < radius.deg():
+        return get_planet(test_planets[0][1])
+    else:
+        raise ExoClockLibraryError('Planet could not be located')
+
+
+def locate_system(ra, dec, radius=0.02):
+
+    return get_system(locate_planet(ra, dec, radius)['star']['simbad_id'])
+
+
+def get_all_planets():
+    return list(exoclock_data.ecc()['planets'].keys())
+
+