Merge pull request #17 from moeyensj/gauss

Initial Orbit Determination

Author: moeyensj

requirements.txt

@@ -1,6 +1,7 @@
 numpy
 scipy
 astropy
+astroquery
 scikit-learn
 matplotlib
 seaborn
@@ -10,6 +11,8 @@ plotly
 pyyaml
 openorb
 ipykernel
+numba
+spiceypy
 pytest
 pytest-cov
 coveralls

requirements_travis.txt

@@ -1,6 +1,7 @@
 numpy
 scipy
 astropy
+astroquery
 scikit-learn
 matplotlib
 seaborn
@@ -9,6 +10,8 @@ fortranformat
 plotly
 pyyaml
 openorb
+numba
+spiceypy
 pytest
 pytest-cov<2.6.0
 coveralls

thor/__init__.py

@@ -1,5 +1,7 @@
 from .config import *
+from .constants import *
 from .data_processing import *
+from .utils import *
 from .io import *
 from .orbits import *
 from .coordinates import *

thor/constants.py

@@ -0,0 +1,35 @@
+import numpy as np
+from astropy import units as u
+from astropy.constants import codata2014 as c
+
+AU_TO_KM = 6.684587122268446e-09
+
+class Constants:
+    
+    # Speed of light: AU per day (173.14463267424034)
+    C = c.c.to(u.au / u.d).value 
+    
+    # Gravitational constant: AU**3 / M_sun / d**2 (0.00029591220819207784)
+    G = c.G.to(u.AU**3 / u.M_sun / u.d**2).value 
+    
+    # Solar Mass: M_sun (1.0)
+    M_SUN = 1.0
+    
+    # Earth Mass: M_sun (3.0034893488507934e-06)
+    M_EARTH = u.M_earth.to(u.M_sun)
+    
+    # Earth Equatorial Radius (6378.1363 km (DE431/DE430))
+    R_EARTH = (6378.1363 * u.km).to(u.AU)
+
+    # Mean Obliquity at J2000: radians (0.40909280422232897)
+    OBLIQUITY = (84381.448 * u.arcsecond).to(u.radian).value
+
+    # Transformation matrix from Equatorial J2000 to Ecliptic J2000
+    TRANSFORM_EQ2EC = np.array([
+        [1, 0, 0],
+        [0, np.cos(OBLIQUITY), np.sin(OBLIQUITY)],
+        [0, -np.sin(OBLIQUITY), np.cos(OBLIQUITY)
+    ]])
+    
+    # Transformation matrix from Ecliptic J2000 to Equatorial J2000
+    TRANSFORM_EC2EQ = TRANSFORM_EQ2EC.T

thor/coordinates/coordinate_transforms.py

@@ -1,13 +1,9 @@
 import numpy as np
 
-OBLIQUITY = np.radians(23.43928)
-TRANSFORM_EQ2EC = np.matrix([[1, 0, 0],
-                             [0, np.cos(OBLIQUITY), np.sin(OBLIQUITY)],
-                             [0, -np.sin(OBLIQUITY), np.cos(OBLIQUITY)]])
-TRANSFORM_EC2EQ = np.matrix([[1, 0, 0],
-                             [0, np.cos(OBLIQUITY), -np.sin(OBLIQUITY)],
-                             [0, np.sin(OBLIQUITY), np.cos(OBLIQUITY)]])
+from ..constants import Constants as c
 
+TRANSFORM_EQ2EC = c.TRANSFORM_EQ2EC
+TRANSFORM_EC2EQ = c.TRANSFORM_EC2EQ
 
 __all__ = ["equatorialToEclipticCartesian",
            "eclipticToEquatorialCartesian",

thor/main.py

@@ -11,7 +11,7 @@
 from .config import Config
 from .cell import Cell
 from .particle import TestParticle
-from .orbits import propagateOrbits
+from .orbits.propagate import propagateOrbits
 from .data_processing import findExposureTimes
 from .data_processing import grabLinkedDetections
 from .plotting import plotOrbitsFindable

thor/observatories/__init__.py

@@ -0,0 +1 @@
+from .codes import *

thor/observatories/codes.py

@@ -0,0 +1,26 @@
+import os
+
+from ..utils import _downloadFile
+
+__all__ = [
+    "getMPCObsCodeFile",
+]
+
+def getMPCObsCodeFile():
+    """
+    Downloads the MPC Observatory Codes file. Checks if a newer version of the file exists online, if so, 
+    replaces the previously downloaded file if available. 
+    
+    Parameters
+    ----------
+    None
+    
+    Returns
+    -------
+    None
+    """
+    obscodes = os.path.join(os.path.dirname(__file__), "data/ObsCodes.html")
+    url = 'https://www.minorplanetcenter.net/iau/lists/ObsCodes.html'  
+    _downloadFile(obscodes, url, update=True)
+    return
+