def test_position_with_delete_has_nothing_else():
expected_result = """{
"delete": true
}"""
pos_list = Position(delete=True, cartesian=[1, 2, 3])
pos_val = Position(delete=True, cartesian=Cartesian3Value(values=[1, 2, 3]))
assert repr(pos_list) == repr(pos_val) == expected_result
def test_position_reference():
expected_result = """{
"reference": "satellite"
}"""
pos = Position(reference="satellite")
assert repr(pos) == expected_result
def test_packet_dynamic_cartesian_position_perfect():
# Trying to group the cartesian value by sample
# is much more difficult than expected.
# Pull requests welcome
expected_result = """{
"id": "InternationalSpaceStation",
"position": {
"interpolationAlgorithm": "LAGRANGE",
"referenceFrame": "INERTIAL",
"cartesian": [
0.0, -6668447.2211117, 1201886.45913705, 146789.427467256,
60.0, -6711432.84684144, 919677.673492462, -214047.552431458
]
}
}"""
packet = Packet(
id="InternationalSpaceStation",
position=Position(
interpolationAlgorithm=InterpolationAlgorithms.LAGRANGE,
referenceFrame=ReferenceFrames.INERTIAL,
cartesian=[
0.0,
-6668447.2211117,
1201886.45913705,
146789.427467256,
60.0,
-6711432.84684144,
919677.673492462,
-214047.552431458,
],
),
)
assert repr(packet) == expected_result
def test_position_no_values_raises_error():
with pytest.raises(ValueError) as exc:
Position()
assert (
"One of cartesian, cartographicDegrees or cartographicRadians must be given"
in exc.exconly())
def test_position_renders_epoch():
expected_result = """{
"epoch": "2019-03-20T12:00:00Z",
"cartesian": []
}"""
pos = Position(epoch=dt.datetime(2019, 3, 20, 12, tzinfo=dt.timezone.utc),
cartesian=[])
assert repr(pos) == expected_result
def test_position_cartographic_degrees():
expected_result = """{
"cartographicDegrees": [
10.0,
20.0,
0.0
]
}"""
pos = Position(cartographicDegrees=[10.0, 20.0, 0.0])
assert repr(pos) == expected_result
def test_position_has_given_epoch():
expected_epoch = dt.datetime(2019,
6,
11,
12,
26,
58,
tzinfo=dt.timezone.utc)
pos = Position(epoch=expected_epoch, cartesian=[])
assert pos.epoch == expected_epoch
def test_packet_constant_cartesian_position():
expected_result = """{
"id": "MyObject",
"position": {
"cartesian": [
0.0,
0.0,
0.0
]
}
}"""
packet = Packet(id="MyObject", position=Position(cartesian=[0.0, 0.0, 0.0]))
assert repr(packet) == expected_result
def write_rx_czml():
height = 50
receiver_point_packets = []
lob_packets = []
top = Preamble(name="Receivers")
rx_properties = {
"verticalOrigin": "BOTTOM",
"scale": 0.75,
"heightReference":"CLAMP_TO_GROUND",
"height": 48,
"width": 48,
}
for index, x in enumerate(receivers):
if x.isActive and ms.receiving:
lob_start_lat = x.latitude
lob_start_lon = x.longitude
lob_stop_lat, lob_stop_lon = v.direct(lob_start_lat, lob_start_lon, x.doa, d)
lob_packets.append(Packet(id=f"LOB-{x.station_id}-{index}",
polyline=Polyline(
material= Material( polylineOutline =
PolylineOutlineMaterial(
color= Color(rgba=[255, 140, 0, 255]),
outlineColor= Color(rgba=[0, 0, 0, 255]),
outlineWidth= 2
)),
clampToGround=True,
width=5,
positions=Position(cartographicDegrees=[lob_start_lon, lob_start_lat, height, lob_stop_lon, lob_stop_lat, height])
)))
else:
lob_packets = []
if x.isMobile == True:
rx_icon = {"image":{"uri":"/static/flipped_car.svg"}}
# if x.heading > 0 or x.heading < 180:
# rx_icon = {"image":{"uri":"/static/flipped_car.svg"}, "rotation":math.radians(360 - x.heading + 90)}
# elif x.heading < 0 or x.heading > 180:
# rx_icon = {"image":{"uri":"/static/car.svg"}, "rotation":math.radians(360 - x.heading - 90)}
else:
rx_icon = {"image":{"uri":"/static/tower.svg"}}
receiver_point_packets.append(Packet(id=f"{x.station_id}-{index}",
billboard={**rx_properties, **rx_icon},
position={"cartographicDegrees": [ x.longitude, x.latitude, 15 ]}))
output = Document([top] + receiver_point_packets + lob_packets)
return output
def packet_from_point(feat, id_=None):
name = feat.get("name",
"point_{}".format(id_) if id_ is not None else "point_0")
description = feat.get("description", name)
color = get_color(feat.get("color", 0xff0000))
point = feat.get_shape(WGS84_CRS)
return Packet(
id=id_,
name=name,
description=description,
point=Point(color=color, pixelSize=5),
position=Position(
# this assumes 0 elevation for all points
cartographicDegrees=[point.x, point.y, 0]),
)
def test_packet_constant_cartesian_position_perfect():
# Trying to group the cartesian value by sample
# is much more difficult than expected.
# Pull requests welcome
expected_result = """{
"id": "MyObject",
"position": {
"interpolationAlgorithm": "LINEAR",
"referenceFrame": "FIXED",
"cartesian": [
0.0, 0.0, 0.0
]
}
}"""
packet = Packet(id="MyObject", position=Position(cartesian=[0.0, 0.0, 0.0]))
assert repr(packet) == expected_result
def test_packet_dynamic_cartesian_position():
expected_result = """{
"id": "InternationalSpaceStation",
"position": {
"interpolationAlgorithm": "LAGRANGE",
"referenceFrame": "INERTIAL",
"cartesian": [
0.0,
-6668447.2211117,
1201886.45913705,
146789.427467256,
60.0,
-6711432.84684144,
919677.673492462,
-214047.552431458
]
}
}"""
packet = Packet(
id="InternationalSpaceStation",
position=Position(
interpolationAlgorithm=InterpolationAlgorithms.LAGRANGE,
referenceFrame=ReferenceFrames.INERTIAL,
cartesian=[
0.0,
-6668447.2211117,
1201886.45913705,
146789.427467256,
60.0,
-6711432.84684144,
919677.673492462,
-214047.552431458,
],
),
)
assert repr(packet) == expected_result
def add_trajectory(
self,
positions,
epochs,
groundtrack_show=False,
groundtrack_lead_time=None,
groundtrack_trail_time=None,
groundtrack_width=None,
groundtrack_color=None,
id_name=None,
id_description=None,
path_width=None,
path_show=None,
path_color=None,
label_fill_color=None,
label_outline_color=None,
label_font=None,
label_text=None,
label_show=None,
):
"""
Adds trajectory.
Parameters
----------
positions: ~astropy.coordinates.CartesianRepresentation
Trajectory to plot.
epochs: ~astropy.time.Time
Epochs for positions.
groundtrack_show: bool
If set to true, the groundtrack is
displayed.
groundtrack_lead_time: float
The time the animation is ahead of the real-time groundtrack
groundtrack_trail_time: float
The time the animation is behind the real-time groundtrack
groundtrack_width: int
Groundtrack width
groundtrack_color: list (int)
Rgba groundtrack color. By default, it is set to the path color
id_name: str
Set orbit name
id_description: str
Set orbit description
path_width: int
Path width
path_show: bool
Indicates whether the path is visible
path_color: list (int)
Rgba path color
label_fill_color: list (int)
Fill Color in rgba format
label_outline_color: list (int)
Outline Color in rgba format
label_font: str
Set label font style and size (CSS syntax)
label_text: str
Set label text
label_show: bool
Indicates whether the label is visible
"""
positions = (positions.represent_as(CartesianRepresentation).get_xyz(
1).to_value(u.m))
epochs = Time(epochs, format="isot")
if len(epochs) != len(positions):
raise ValueError("Number of Points and Epochs must be equal.")
epochs = np.fromiter(
map(lambda epoch: (epoch - epochs[0]).to_value(u.s), epochs),
dtype=float,
)
positions = np.around(
np.concatenate([epochs[..., None], positions], axis=1).ravel(),
1).tolist()
self.trajectories.append([positions, None, label_text, path_color])
start_epoch = Time(self.start_epoch, format="isot")
pckt = Packet(
id=self.i,
name=id_name,
description=id_description,
availability=TimeInterval(start=self.start_epoch,
end=self.end_epoch),
position=Position(
interpolationDegree=5,
interpolationAlgorithm=InterpolationAlgorithms.LAGRANGE,
referenceFrame=ReferenceFrames.INERTIAL,
cartesian=positions,
# Use explicit UTC timezone, rather than the default, which is a local timezone.
epoch=start_epoch.datetime.replace(tzinfo=timezone.utc),
),
path=Path(
show=path_show,
width=path_width,
material=Material(solidColor=SolidColorMaterial(
color=Color.from_list(path_color))) if path_color
is not None else Material(solidColor=SolidColorMaterial(
color=Color.from_list([255, 255, 0]))),
resolution=120,
),
label=Label(
text=label_text,
font=label_font
if label_font is not None else "11pt Lucida Console",
show=label_show,
fillColor=Color(rgba=label_fill_color) if label_fill_color
is not None else Color(rgba=[255, 255, 0, 255]),
outlineColor=Color(rgba=label_outline_color)
if label_outline_color is not None else Color(
rgba=[255, 255, 0, 255]),
),
billboard=Billboard(image=PIC_SATELLITE, show=True),
)
self.packets.append(pckt)
if groundtrack_show:
raise NotImplementedError(
"Ground tracking for trajectory not implemented yet")
self.i += 1
def add_orbit(
self,
orbit,
rtol=1e-10,
N=None,
groundtrack_show=False,
groundtrack_lead_time=None,
groundtrack_trail_time=None,
groundtrack_width=None,
groundtrack_color=None,
id_name=None,
id_description=None,
path_width=None,
path_show=None,
path_color=None,
label_fill_color=None,
label_outline_color=None,
label_font=None,
label_text=None,
label_show=None,
):
"""
Adds an orbit
Parameters
----------
orbit: poliastro.twobody.orbit.Orbit
Orbit to be added
rtol: float
Maximum relative error permitted
N: int
Number of sample points
groundtrack_show: bool
If set to true, the groundtrack is
displayed.
groundtrack_lead_time: float
The time the animation is ahead of the real-time groundtrack
groundtrack_trail_time: float
The time the animation is behind the real-time groundtrack
groundtrack_width: int
Groundtrack width
groundtrack_color: list (int)
Rgba groundtrack color. By default, it is set to the path color
id_name: str
Set orbit name
id_description: str
Set orbit description
path_width: int
Path width
path_show: bool
Indicates whether the path is visible
path_color: list (int)
Rgba path color
label_fill_color: list (int)
Fill Color in rgba format
label_outline_color: list (int)
Outline Color in rgba format
label_font: str
Set label font style and size (CSS syntax)
label_text: str
Set label text
label_show: bool
Indicates whether the label is visible
"""
if N is None:
N = self.N
if orbit.epoch < self.start_epoch:
orbit = orbit.propagate(self.start_epoch - orbit.epoch)
elif orbit.epoch > self.end_epoch:
raise ValueError(
"The orbit's epoch cannot exceed the constructor's ending epoch"
)
if rtol <= 0 or rtol >= 1:
raise ValueError(
"The relative tolerance must be a value in the range (0, 1)")
self.orbits.append([orbit, N, orbit.epoch])
cartesian_cords = self._init_orbit_packet_cords_(self.i, rtol=rtol)
start_epoch = Time(min(self.orbits[self.i][2], self.start_epoch),
format="isot")
pckt = Packet(
id=self.i,
name=id_name,
description=id_description,
availability=TimeInterval(start=self.start_epoch,
end=self.end_epoch),
position=Position(
interpolationDegree=5,
interpolationAlgorithm=InterpolationAlgorithms.LAGRANGE,
referenceFrame=ReferenceFrames.INERTIAL,
cartesian=cartesian_cords,
# Use explicit UTC timezone, rather than the default, which is a local timezone.
epoch=start_epoch.datetime.replace(tzinfo=timezone.utc),
),
path=Path(
show=path_show,
width=path_width,
material=Material(solidColor=SolidColorMaterial(
color=Color.from_list(path_color))) if path_color
is not None else Material(solidColor=SolidColorMaterial(
color=Color.from_list([255, 255, 0]))),
resolution=120,
),
label=Label(
text=label_text,
font=label_font
if label_font is not None else "11pt Lucida Console",
show=label_show,
fillColor=Color(rgba=label_fill_color) if label_fill_color
is not None else Color(rgba=[255, 255, 0, 255]),
outlineColor=Color(rgba=label_outline_color)
if label_outline_color is not None else Color(
rgba=[255, 255, 0, 255]),
),
billboard=Billboard(image=PIC_SATELLITE, show=True),
)
self.packets.append(pckt)
if groundtrack_show:
groundtrack_color = path_color
groundtrack_cords = self._init_groundtrack_packet_cords_(self.i,
rtol=rtol)
pckt = Packet(
id="groundtrack" + str(self.i),
availability=TimeInterval(start=self.start_epoch,
end=self.end_epoch),
position=Position(
interpolationDegree=5,
interpolationAlgorithm=InterpolationAlgorithms.LAGRANGE,
referenceFrame=ReferenceFrames.INERTIAL,
cartesian=groundtrack_cords,
# Use explicit UTC timezone, rather than the default, which is a local timezone.
epoch=start_epoch.datetime.replace(tzinfo=timezone.utc),
),
path=Path(
show=True,
material=Material(solidColor=SolidColorMaterial(
color=Color(rgba=groundtrack_color)))
if groundtrack_color is not None else Material(
solidColor=SolidColorMaterial(color=Color(
rgba=[255, 255, 0, 255]))),
resolution=60,
width=groundtrack_width,
leadTime=groundtrack_lead_time
if groundtrack_lead_time else 100,
trailTime=groundtrack_trail_time
if groundtrack_trail_time else 100,
),
)
self.packets.append(pckt)
self.i += 1
def add_ground_station(
self,
pos,
id_description=None,
label_fill_color=None,
label_font=None,
label_outline_color=None,
label_text=None,
label_show=True,
):
"""
Adds a ground station
Parameters
----------
pos: list [~astropy.units.Quantity]
Coordinates of ground station,
list of geodetic latitude and longitude [lon, lat] (0 elevation)
id_description: str
Set ground station description
label_fill_color: list (int)
Fill Color in rgba format
label_outline_color: list (int)
Outline Color in rgba format
label_font: str
Set label font style and size (CSS syntax)
label_text: str
Set label text
label_show: bool
Indicates whether the label is visible
"""
if (len(pos) == 2 and isinstance(pos[0], u.quantity.Quantity)
and isinstance(pos[0], u.quantity.Quantity)):
u0, v0 = pos
if self.cust_prop[0]:
a, b = (
self.cust_prop[0][0],
self.cust_prop[0][2],
) # Get semi-major and semi-minor axises
else:
a, b = Earth.R.to_value(u.m), Earth.R_polar.to_value(u.m)
f = 1 - (b / a) # Flattenning
pos = list(gd2gce(a, f, u0.to_value(u.rad), v0.to_value(u.rad), 0))
else:
raise TypeError(
"Invalid coordinates. Coordinates must be of the form [u, v] where u, v are astropy units"
)
pckt = Packet(
id="GS" + str(self.gs_n),
description=id_description,
availability=TimeInterval(start=self.start_epoch,
end=self.end_epoch),
position=Position(cartesian=pos),
label=Label(
show=label_show,
text=label_text,
font=label_font
if label_font is not None else "11pt Lucida Console",
fillColor=Color(rgba=label_fill_color)
if label_fill_color is not None else None,
outlineColor=Color(rgba=label_outline_color)
if label_outline_color is not None else None,
),
billboard=Billboard(image=PIC_GROUNDSTATION, show=True),
)
self.packets.append(pckt)
self.gs_n += 1
Packet(
id="AreaTarget/Pennsylvania",
name="Pennsylvania",
label=Label(
horizontalOrigin=HorizontalOrigins.LEFT,
show=True,
font="11pt Lucida Console",
style=LabelStyles.FILL_AND_OUTLINE,
outlineWidth=2,
text="Pennsylvania",
verticalOrigin=VerticalOrigins.CENTER,
fillColor=Color.from_list([255, 0, 0]),
outlineColor=Color.from_list([0, 0, 0]),
),
position=Position(
cartesian=[1152255.80150063, -4694317.951340558, 4147335.9067563135]
),
),
Packet(
id="Facility/AGI",
name="AGI",
availability=TimeInterval(start=start, end=end),
billboard=Billboard(
horizontalOrigin=HorizontalOrigins.CENTER,
image=(
"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/"
"9hAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsMAAA7DAc"
"dvqGQAAACvSURBVDhPrZDRDcMgDAU9GqN0lIzijw6SUbJJygUeNQgSqepJTyHG91"
"LVVpwDdfxM3T9TSl1EXZvDwii471fivK73cBFFQNTT/d2KoGpfGOpSIkhUpgUMxq"
"9DFEsWv4IXhlyCnhBFnZcFEEuYqbiUlNwWgMTdrZ3JbQFoEVG53rd8ztG9aPJMnB"
"UQf/VFraBJeWnLS0RfjbKyLJA8FkT5seDYS1Qwyv8t0B/5C2ZmH2/eTGNNBgMmAA"
def add_orbit(
self,
orbit,
rtol=1e-10,
N=None,
id_name=None,
id_description=None,
path_width=None,
path_show=None,
path_color=None,
label_fill_color=None,
label_outline_color=None,
label_font=None,
label_text=None,
label_show=None,
):
"""
Adds an orbit
Parameters
----------
orbit: poliastro.Orbit
Orbit to be added
rtol: float
Maximum relative error permitted
N: int
Number of sample points
Id parameters:
-------------
id_name: str
Set orbit name
id_description: str
Set orbit description
Path parameters
---------------
path_width: int
Path width
path_show: bool
Indicates whether the path is visible
path_color: list (int)
Rgba path color
Label parameters
----------
label_fill_color: list (int)
Fill Color in rgba format
label_outline_color: list (int)
Outline Color in rgba format
label_font: str
Set label font style and size (CSS syntax)
label_text: str
Set label text
label_show: bool
Indicates whether the label is visible
"""
if N is None:
N = self.N
if orbit.epoch < Time(self.start_epoch):
orbit = orbit.propagate(self.start_epoch - orbit.epoch)
elif orbit.epoch > Time(self.end_epoch):
raise ValueError(
"The orbit's epoch cannot exceed the constructor's ending epoch"
)
if rtol <= 0 or rtol >= 1:
raise ValueError(
"The relative tolerance must be a value in the range (0, 1)")
self.orbits.append([orbit, N, orbit.epoch])
cartesian_cords = self._init_orbit_packet_cords_(self.i, rtol=rtol)
start_epoch = Time(min(self.orbits[self.i][2], self.start_epoch),
format="isot")
pckt = Packet(
id=self.i,
name=id_name,
description=id_description,
availability=TimeInterval(start=self.start_epoch.datetime,
end=self.end_epoch.datetime),
position=Position(
interpolationDegree=5,
interpolationAlgorithm=InterpolationAlgorithms.LAGRANGE,
referenceFrame=ReferenceFrames.INERTIAL,
cartesian=cartesian_cords,
epoch=start_epoch.value,
),
path=Path(
show=path_show,
width=path_width,
material=Material(solidColor=SolidColorMaterial(color=Color(
rgba=path_color))) if path_color is not None else Material(
solidColor=SolidColorMaterial(color=Color(
rgba=[255, 255, 0, 255]))),
resolution=120,
),
label=Label(
text=label_text,
font=label_font
if label_font is not None else "11pt Lucida Console",
show=label_show,
fillColor=Color(rgba=label_fill_color) if label_fill_color
is not None else Color(rgba=[255, 255, 0, 255]),
outlineColor=Color(rgba=label_outline_color)
if label_outline_color is not None else Color(
rgba=[255, 255, 0, 255]),
),
billboard=Billboard(image=PIC_SATELLITE, show=True),
)
self.packets.append(pckt)
self.i += 1
def add_ground_station(
self,
pos,
id_description=None,
label_fill_color=None,
label_font=None,
label_outline_color=None,
label_text=None,
label_show=True,
):
"""
Adds a ground station
Parameters
----------
orbit: poliastro.Orbit
Orbit to be added
pos: list [~astropy.units]
coordinates of ground station
[u v] ellipsoidal coordinates (0 elevation)
Id parameters:
-------------
id_description: str
Set ground station description
Label parameters
----------
label_fill_color: list (int)
Fill Color in rgba format
label_outline_color: list (int)
Outline Color in rgba format
label_font: str
Set label font style and size (CSS syntax)
label_text: str
Set label text
label_show: bool
Indicates whether the label is visible
"""
if (len(pos) == 2 and isinstance(pos[0], u.quantity.Quantity)
and isinstance(pos[0], u.quantity.Quantity)):
u0, v0 = pos
if self.cust_prop[0]:
a, b = (
self.cust_prop[0][0],
self.cust_prop[0][2],
) # get semi-major and semi-minor axises
else:
a, b = Earth.R.to(u.m).value, Earth.R_polar.to(u.m).value
pos = list(
map(lambda x: x.value, ellipsoidal_to_cartesian(a, b, u0, v0)))
else:
raise TypeError(
"Invalid coordinates. Coordinates must be of the form [u, v] where u, v are astropy units"
)
pckt = Packet(
id="GS" + str(self.gs_n),
description=id_description,
availability=TimeInterval(start=self.start_epoch,
end=self.end_epoch),
position=Position(cartesian=pos),
label=Label(
show=label_show,
text=label_text,
font=label_font
if label_font is not None else "11pt Lucida Console",
fillColor=Color(rgba=label_fill_color)
if label_fill_color is not None else None,
outlineColor=Color(rgba=label_outline_color)
if label_outline_color is not None else None,
),
billboard=Billboard(image=PIC_GROUNDSTATION, show=True),
)
self.packets.append(pckt)
self.gs_n += 1
def test_position_has_delete():
pos = Position(delete=True, cartesian=[])
assert pos.delete