2021-01-14 19:09:42 +01:00
|
|
|
from typing import Any, Optional, Tuple, Union, List, Iterable
|
|
|
|
|
|
|
|
from matplotlib.axes import Axes
|
|
|
|
from matplotlib.collections import LineCollection
|
|
|
|
from matplotlib.figure import Figure
|
|
|
|
|
2021-02-17 14:12:42 +01:00
|
|
|
from . import Simulation, Particle
|
2021-01-14 19:09:42 +01:00
|
|
|
|
|
|
|
Limits = Tuple[float, float]
|
|
|
|
ColorTuple = Tuple[int, int, int]
|
|
|
|
|
|
|
|
|
|
|
|
def OrbitPlot(sim: Simulation, figsize: Optional[Tuple[float, float]] = ..., fancy: bool = ...,
|
|
|
|
slices: Optional[int] = ..., xlim: Optional[Limits] = ..., ylim: Optional[Limits] = ...,
|
|
|
|
unitlabel: Optional[str] = ..., color: Union[str, bool, List, None] = ..., periastron: bool = ...,
|
|
|
|
orbit_type: str = ..., lw: float = ..., plotparticles: List[int] = ..., primary: Optional[Particle] = ...,
|
|
|
|
Narc: int = ...) -> Union[Tuple[Figure, Axes], Tuple[Figure, Axes, Axes, Axes]]: ...
|
|
|
|
|
|
|
|
|
|
|
|
def get_color(color: str) -> ColorTuple: ...
|
|
|
|
|
|
|
|
|
2021-02-26 21:33:00 +01:00
|
|
|
def fading_line(x: Iterable[float], y: Iterable[float], color: Union[str, ColorTuple] = ..., alpha: float = ...,
|
|
|
|
fading: bool = ..., fancy: bool = ..., **kwargs: Any) -> LineCollection: ...
|
2021-01-14 19:09:42 +01:00
|
|
|
|
|
|
|
|
|
|
|
def OrbitPlotOneSlice(sim: Simulation, ax: Axes, Narc: int = ..., color: bool = ..., periastron: bool = ...,
|
|
|
|
orbit_type: str = ..., lw: float = ..., axes: str = ..., plotparticles: List[int] = ...,
|
2021-02-26 21:33:00 +01:00
|
|
|
primary: Optional[Particle] = ..., fancy: bool = ..., xlim: Optional[Tuple[float, float]] = ...,
|
|
|
|
ylim: Optional[Tuple[float, float]] = ...) -> None: ...
|
2021-01-14 19:09:42 +01:00
|
|
|
|
|
|
|
|
|
|
|
def OrbitPlotAddFancyStars(ax: Axes, lw: float, slices: float = ...) -> None: ...
|