def test_packet_label():
expected_result = """{
"id": "0",
"label": {
"show": true,
"font": "20px sans-serif",
"style": "FILL",
"fillColor": {
"rgbaf": [
0.2,
0.3,
0.4,
1.0
]
},
"outlineColor": {
"rgba": [
0,
233,
255,
2
]
},
"outlineWidth": 2.0
}
}"""
packet = Packet(
id="0",
label=Label(
font="20px sans-serif",
fillColor=Color.from_list([0.2, 0.3, 0.4]),
outlineColor=Color.from_list([0, 233, 255, 2]),
outlineWidth=2.0,
),
)
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
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 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
),
Packet(
id="Facility/AGI-to-Satellite/Geoeye1/Sensor/Sensor",
name="AGI to Sensor",
parent=accesses_id,
description="<h2>No accesses</h2>",
),
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,