summaryrefslogtreecommitdiff
path: root/iss/predictions.py
blob: eb02ef9b96ba2e9b6e7cf8beedef98fa448cea6c (plain)
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
from skyfield.api import Topos, load

from .utils import chunks


ISS = "ISS (ZARYA)"
STATIONS_URL = "http://celestrak.com/NORAD/elements/stations.txt"


class Predictions(object):
    def __init__(
        self,
        lat,
        lng,
        altitude=30.0,
        tz="UTC",
        satellite=ISS,
        start=None,
        days=10,
        tle_file=None,
    ):
        self.lat = lat
        self.lng = lng
        self.altitude = altitude
        self.tz = tz
        self.satellite = satellite
        self.start = start
        self.days = days
        self.tle_file = tle_file.resolve().as_posix() if tle_file else STATIONS_URL

    def init_stations(self):
        return load.tle_file(self.tle_file)

    def get_next_days(self):
        ts = load.timescale()
        t0 = ts.now() if not self.start else ts.ut1_jd(self.start)
        t1 = ts.ut1_jd(t0.ut1 + self.days)
        return t0, t1

    def get_location(self):
        return Topos(latitude_degrees=self.lat, longitude_degrees=self.lng)

    def get_satellite(self):
        satellites = self.init_stations()
        by_name = {sat.name: sat for sat in satellites}
        return by_name[self.satellite]

    def get_prediction_events(self):
        satellite = self.get_satellite()
        t0, t1 = self.get_next_days()
        location = self.get_location()

        ts, _events = satellite.find_events(
            location, t0, t1, altitude_degrees=self.altitude
        )

        # events are returned as 3-tuples of (rise, culminate, set)
        # where rise/set are relative to given altitude
        # docs mention the possibilibity of several culminations
        # https://rhodesmill.org/skyfield/earth-satellites.html#finding-when-a-satellite-rises-and-sets
        # but this doesn't seem to happen in our case
        preds = chunks(ts, 3)
        return [[t.utc_iso() for t in p] for p in preds]