def test_czml_add_orbit():
start_epoch = iss.epoch
end_epoch = iss.epoch + molniya.period
sample_points = 10
extractor = CZMLExtractor(start_epoch, end_epoch, sample_points)
extractor.add_orbit(molniya,
label_text="Molniya",
label_fill_color=[125, 80, 120, 255])
extractor.add_orbit(iss, label_text="ISS", path_show=False)
cords_iss = extractor.czml[1]["position"]["cartesian"]
h_iss = (end_epoch - iss.epoch).to(u.second) / sample_points
for i in range(sample_points):
position_iss_test = propagate(iss, TimeDelta(i * h_iss))
cords_iss_test = (
position_iss_test.represent_as(CartesianRepresentation).xyz.to(
u.meter).value)
cords_iss_test = np_insert(cords_iss_test, 0, h_iss.value * i, axis=0)
for j in range(4):
assert_allclose(cords_iss[4 * i + j], cords_iss_test[j], rtol=1e-5)
# Test label params and that the values between the two objects are not overwritten
assert extractor.czml[0]["label"]["text"] == "Molniya"
assert extractor.czml[1]["label"]["text"] == "ISS"
assert extractor.czml[0]["label"]["fillColor"]["rgba"] == [
125, 80, 120, 255
]
assert extractor.czml[1]["path"]["show"]["boolean"] is False
def test_czml_invalid_orbit_epoch_error():
start_epoch = molniya.epoch
end_epoch = molniya.epoch + molniya.period
extractor = CZMLExtractor(start_epoch, end_epoch, 10)
with pytest.raises(ValueError) as excinfo:
extractor.add_orbit(iss, label_text="ISS", path_show=False)
assert (
"ValueError: The orbit's epoch cannot exceed the constructor's ending epoch"
in excinfo.exconly())
def test_czml_add_orbit_negative_rtol_raises_error_if_beyond_range():
start_epoch = iss.epoch
end_epoch = iss.epoch + molniya.period
sample_points = 10
extractor = CZMLExtractor(start_epoch, end_epoch, sample_points)
with pytest.raises(ValueError) as excinfo:
extractor.add_orbit(
molniya,
rtol=1e2, # rtol > 1.
label_text="Molniya",
label_fill_color=[125, 80, 120, 255],
)
assert ("The relative tolerance must be a value in the range (0, 1)"
in excinfo.exconly())
def test_czml_add_orbit():
start_epoch = iss.epoch
end_epoch = iss.epoch + molniya.period
sample_points = 10
expected_doc = """[{
"id": "document",
"version": "1.0",
"name": "document_packet",
"clock": {
"interval": "2013-03-18T12:00:00Z/2013-03-18T23:59:35Z",
"currentTime": "2013-03-18T12:00:00Z",
"multiplier": 60,
"range": "LOOP_STOP",
"step": "SYSTEM_CLOCK_MULTIPLIER"
}
}, {
"id": "custom_properties",
"properties": {
"custom_attractor": true,
"ellipsoid": [
{
"array": [
6378136.6,
6378136.6,
6356751.9
]
}
],
"map_url": [
"https://upload.wikimedia.org/wikipedia/commons/c/c4/Earthmap1000x500compac.jpg"
],
"scene3D": true
}
}, {
"id": 0,
"availability": "2013-03-18T12:00:00Z/2013-03-18T23:59:35Z",
"position": {
"epoch": "2013-03-18T12:00:00Z",
"interpolationAlgorithm": "LAGRANGE",
"interpolationDegree": 5,
"referenceFrame": "INERTIAL",
"cartesian": [
0.0,
-5874061.7732,
20159787.8726,
40258166.1204,
4317.5108,
-11640044.4185,
17864068.2608,
35673719.9901,
8635.0217,
-16129541.0197,
13690511.4239,
27339319.5732,
12952.5325,
-17354822.8091,
6974596.3025,
13927946.9776,
17270.0433,
-2602428.8865,
-2846586.9792,
-5684502.8433,
21587.5541,
17029383.6011,
5939060.5897,
11860029.9433,
25905.065,
16513858.107,
13042152.6754,
26044577.0704,
30222.5758,
12264870.9451,
17472285.3549,
34891347.6056,
34540.0866,
6599717.4362,
19973508.8879,
39886175.5836,
38857.5975,
377235.7086,
20840447.5591,
41617412.1057,
43175.1083,
-5874061.7732,
20159787.8726,
40258166.1204,
47492.6191,
-11640044.4185,
17864068.2608,
35673719.9901
]
},
"billboard": {
"image": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsMAAA7DAcdvqGQAAADJSURBVDhPnZHRDcMgEEMZjVEYpaNklIzSEfLfD4qNnXAJSFWfhO7w2Zc0Tf9QG2rXrEzSUeZLOGm47WoH95x3Hl3jEgilvDgsOQUTqsNl68ezEwn1vae6lceSEEYvvWNT/Rxc4CXQNGadho1NXoJ+9iaqc2xi2xbt23PJCDIB6TQjOC6Bho/sDy3fBQT8PrVhibU7yBFcEPaRxOoeTwbwByCOYf9VGp1BYI1BA+EeHhmfzKbBoJEQwn1yzUZtyspIQUha85MpkNIXB7GizqDEECsAAAAASUVORK5CYII=",
"show": true
},
"label": {
"text": "Molniya",
"font": "11pt Lucida Console",
"style": "FILL",
"fillColor": {
"rgba": [
125,
80,
120,
255
]
},
"outlineColor": {
"rgba": [
255,
255,
0,
255
]
},
"outlineWidth": 1.0
},
"path": {
"resolution": 120,
"material": {
"solidColor": {
"color": {
"rgba": [
255,
255,
0,
255
]
}
}
}
}
}, {
"id": 1,
"availability": "2013-03-18T12:00:00Z/2013-03-18T23:59:35Z",
"position": {
"epoch": "2013-03-18T12:00:00Z",
"interpolationAlgorithm": "LAGRANGE",
"interpolationDegree": 5,
"referenceFrame": "INERTIAL",
"cartesian": [
0.0,
859072.56,
-4137203.68,
5295568.71,
4317.5108,
-6275612.3001,
-2500051.0065,
496823.4938,
8635.0217,
-2951952.7192,
3308039.2745,
-5135374.5943,
12952.5325,
5284474.5583,
3626657.2697,
-2240110.0348,
17270.0433,
4755661.276,
-2067765.0104,
4367882.9214,
21587.5541,
-3674993.328,
-4315653.7297,
3705413.3249,
25905.065,
-5976376.971,
633013.0667,
-3135590.7655,
30222.5758,
1672241.9629,
4536136.0948,
-4766186.3254,
34540.0866,
6551964.1874,
915641.8629,
1510256.5228,
38857.5975,
554555.7813,
-4218442.3606,
5271693.5079,
43175.1083,
-6358691.2001,
-2321273.8442,
249707.6563,
47492.6191,
-2675736.8087,
3448931.4082,
-5194186.3038
]
},
"billboard": {
"image": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsMAAA7DAcdvqGQAAADJSURBVDhPnZHRDcMgEEMZjVEYpaNklIzSEfLfD4qNnXAJSFWfhO7w2Zc0Tf9QG2rXrEzSUeZLOGm47WoH95x3Hl3jEgilvDgsOQUTqsNl68ezEwn1vae6lceSEEYvvWNT/Rxc4CXQNGadho1NXoJ+9iaqc2xi2xbt23PJCDIB6TQjOC6Bho/sDy3fBQT8PrVhibU7yBFcEPaRxOoeTwbwByCOYf9VGp1BYI1BA+EeHhmfzKbBoJEQwn1yzUZtyspIQUha85MpkNIXB7GizqDEECsAAAAASUVORK5CYII=",
"show": true
},
"label": {
"text": "ISS",
"font": "11pt Lucida Console",
"style": "FILL",
"fillColor": {
"rgba": [
255,
255,
0,
255
]
},
"outlineColor": {
"rgba": [
255,
255,
0,
255
]
},
"outlineWidth": 1.0
},
"path": {
"show": false,
"resolution": 120,
"material": {
"solidColor": {
"color": {
"rgba": [
255,
255,
0,
255
]
}
}
}
}
}]"""
extractor = CZMLExtractor(start_epoch, end_epoch, sample_points)
extractor.add_orbit(molniya,
rtol=1e-4,
label_text="Molniya",
label_fill_color=[125, 80, 120, 255])
extractor.add_orbit(iss, rtol=1e-4, label_text="ISS", path_show=False)
assert repr(extractor.packets) == expected_doc
def test_czml_groundtrack():
start_epoch = molniya.epoch
end_epoch = molniya.epoch + molniya.period
sample_points = 10
expected_doc = """[{
"id": "document",
"version": "1.0",
"name": "document_packet",
"clock": {
"interval": "2000-01-01T12:00:00Z/2000-01-01T23:59:35Z",
"currentTime": "2000-01-01T12:00:00Z",
"multiplier": 60,
"range": "LOOP_STOP",
"step": "SYSTEM_CLOCK_MULTIPLIER"
}
}, {
"id": "custom_properties",
"properties": {
"custom_attractor": true,
"ellipsoid": [
{
"array": [
6378136.6,
6378136.6,
6356751.9
]
}
],
"map_url": [
"https://upload.wikimedia.org/wikipedia/commons/c/c4/Earthmap1000x500compac.jpg"
],
"scene3D": true
}
}, {
"id": 0,
"availability": "2000-01-01T12:00:00Z/2000-01-01T23:59:35Z",
"position": {
"epoch": "2000-01-01T12:00:00Z",
"interpolationAlgorithm": "LAGRANGE",
"interpolationDegree": 5,
"referenceFrame": "INERTIAL",
"cartesian": [
0.0,
10140093.6393,
-800580.7546,
-1598722.8246,
4317.5108,
17593269.6479,
8807305.9945,
17587783.6632,
8635.0217,
15276972.8693,
14840949.3322,
29636690.9896,
12952.5325,
10390291.9946,
18544487.5609,
37032485.9349,
17270.0433,
4459288.7909,
20454288.4376,
40846270.159,
21587.5541,
-1827745.3661,
20778744.495,
41494193.9337,
25905.065,
-7974181.1167,
19549661.224,
39039771.3569,
30222.5758,
-13412369.429,
16641263.6853,
33231835.6785,
34540.0866,
-17116767.3657,
11688710.5829,
23341815.6655,
38857.5975,
-15852598.671,
3766095.5927,
7520719.1144,
43175.1083,
10140093.6393,
-800580.7546,
-1598722.8246,
47492.6191,
17593269.6479,
8807305.9945,
17587783.6632
]
},
"billboard": {
"image": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsMAAA7DAcdvqGQAAADJSURBVDhPnZHRDcMgEEMZjVEYpaNklIzSEfLfD4qNnXAJSFWfhO7w2Zc0Tf9QG2rXrEzSUeZLOGm47WoH95x3Hl3jEgilvDgsOQUTqsNl68ezEwn1vae6lceSEEYvvWNT/Rxc4CXQNGadho1NXoJ+9iaqc2xi2xbt23PJCDIB6TQjOC6Bho/sDy3fBQT8PrVhibU7yBFcEPaRxOoeTwbwByCOYf9VGp1BYI1BA+EeHhmfzKbBoJEQwn1yzUZtyspIQUha85MpkNIXB7GizqDEECsAAAAASUVORK5CYII=",
"show": true
},
"label": {
"text": "Molniya",
"font": "11pt Lucida Console",
"style": "FILL",
"fillColor": {
"rgba": [
125,
80,
120,
255
]
},
"outlineColor": {
"rgba": [
255,
255,
0,
255
]
},
"outlineWidth": 1.0
},
"path": {
"resolution": 120,
"material": {
"solidColor": {
"color": {
"rgba": [
255,
255,
0,
255
]
}
}
}
}
}, {
"id": "groundtrack0",
"availability": "2000-01-01T12:00:00Z/2000-01-01T23:59:35Z",
"position": {
"epoch": "2000-01-01T12:00:00Z",
"interpolationAlgorithm": "LAGRANGE",
"interpolationDegree": 5,
"referenceFrame": "INERTIAL",
"cartesian": [
0.0,
6280728.255788034,
-495875.99989224557,
-990241.5895687921,
4317.5108,
4245756.819479091,
2125453.716924488,
4244432.895224581,
8635.0217,
2663915.8656318397,
2587884.4433988784,
5167885.717379646,
12952.5325,
1548100.5195240811,
2763033.9962378335,
5517651.292250773,
17270.0433,
618022.3343837302,
2834803.0552899935,
5660970.791429868,
21587.5541,
-250341.43583128517,
2846010.7675710567,
5683352.0756686535,
25905.065,
-1142958.8118256037,
2802100.92918071,
5595666.1546065565,
30222.5758,
-2159933.044540537,
2679915.6206089617,
5351667.73157586,
34540.0866,
-3490752.3604656975,
2383767.686202804,
4760273.972119002,
38857.5975,
-5630876.894771479,
1337725.3192625502,
2671375.634339552,
43175.1083,
6280728.255788034,
-495875.99989224557,
-990241.5895687921,
47492.6191,
4245756.819479091,
2125453.716924488,
4244432.895224581
]
},
"path": {
"show": true,
"leadTime": 100,
"trailTime": 100,
"resolution": 60,
"material": {
"solidColor": {
"color": {
"rgba": [
255,
255,
0,
255
]
}
}
}
}
}]"""
extractor = CZMLExtractor(start_epoch, end_epoch, sample_points)
extractor.add_orbit(
molniya,
rtol=1e-4,
label_text="Molniya",
groundtrack_show=True,
label_fill_color=[125, 80, 120, 255],
)
assert repr(extractor.packets) == expected_doc
#/
from poliastro.czml.extract_czml import CZMLExtractor
from poliastro.examples import iss, molniya # noqa: E402
start_epoch = iss.epoch
end_epoch = iss.epoch + molniya.period
sample_points = 100
extractor = CZMLExtractor(start_epoch, end_epoch, sample_points)
extractor.add_orbit(molniya,
rtol=1e-4,
label_text="Molniya",
label_fill_color=[125, 80, 120, 255])
extractor.add_orbit(iss, rtol=1e-4, label_text="ISS", path_show=False)
f = open("molniya100.czml", "w")
f.write(repr(extractor.packets))
f.close()
#
# ``groundtrack_lead_time``: The time the animation is ahead of the real-time groundtrack
#
# ``groundtrack_trail_time``: The time the animation is behind the real-time groundtrack
#
# ``groundtrack_width``: The groundtrack width
#
# ``groundtrack_color``: The groundtrack color. By default, it is set to the trajectory's color
#
# In[5]:
extractor.add_orbit(molniya,
id_name="MolniyaOrbit",
path_width=2,
label_text="Molniya",
label_fill_color=[125, 80, 120, 255]
)
# You can now export the extractor packets by simply calling ``extractor.packets`` and load it to the Cesium app as described [here](https://github.com/poliastro/cesium-app)
# In[6]:
extractor.packets
# ## Landing on Mars
#
# You can customize the attractor of your orbit by defining any valid ellipsoid with the help of poliastro's ``Body`` class. For your convenience, poliastro offers a pre-defined list of all the major planetary bodies of the solar system so you can simply import them.
# from poliastro.twobody import Orbit
#
# Orbit.from_horizons('Ceres')
#
# Orbit.from_sbdb('apophis')
from poliastro.examples import molniya, iss,Orbit
from poliastro.czml.extract_czml import CZMLExtractor
from astropy.time import Time
from astropy import units as u
start_epoch = iss.epoch
print(start_epoch)
end_epoch = iss.epoch + molniya.period
print(end_epoch)
N = 100
EPOCH = Time(start_epoch, scale="tdb")
extractor = CZMLExtractor(start_epoch, end_epoch, N)
#roadster = Orbit.from_horizons(name="SpaceX Roadster", epoch=EPOCH, id_type="majorbody")
#extractor.add_orbit(roadster, label_text="SpaceX Roadster")
extractor.add_orbit(iss, label_text="ISS")
print(extractor.packets)
