def _change_custom_params(self, ellipsoid, pr_map, scene3D):
"""
Change the custom properties package.
Parameters
----------
ellipsoid: list(int)
Defines the attractor ellipsoid. The list must have three numbers
representing the radii in the x, y and z axis
pr_map: str
A URL to the projection of the defined ellipsoid (UV map)
"""
if pr_map is None:
pr_map = (
"https://upload.wikimedia.org/wikipedia/commons/c/c4/Earthmap1000x500compac.jpg",
)
self.cust_prop[1] = pr_map
custom_props = {
"custom_attractor": True,
"ellipsoid": [{
"array": ellipsoid
}],
"map_url": pr_map,
"scene3D": scene3D,
}
pckt = Packet(id="custom_properties", properties=custom_props)
self.packets.append(pckt)
def add_link(self,
sat1,
sat2,
start_time,
end_time,
color=[255, 255, 0, 255],
width=1.0,
id_name=None,
description=None):
start_time = Time(start_time, format="isot")
end_time = Time(end_time, format="isot")
pckt = Packet(id="s" + str(self.j),
name=id_name,
description=description,
availability=TimeInterval(start=start_time.value,
end=end_time.value),
polyline=Polyline(
positions=PositionList(references=[
str(sat1) + "#position",
str(sat2) + "#position"
]),
material=Material(solidColor=SolidColorMaterial(
color=Color(rgba=color))),
width=width,
))
self.packets.append(pckt)
self.j += 1
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