1
0
Fork 0
mirror of https://github.com/Findus23/rebound-stubs.git synced 2024-09-19 15:13:48 +02:00
rebound-stubs/rebound/plotting.pyi
2021-01-14 19:09:42 +01:00

32 lines
1.5 KiB
Python

from typing import Any, Optional, Tuple, Union, List, Iterable
from matplotlib.axes import Axes
from matplotlib.collections import LineCollection
from matplotlib.figure import Figure
from rebound import Simulation, Particle
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: ...
def fading_line(x: Iterable[float], y: Iterable[float], color: Union[str, ColorTuple] = ..., alpha_initial: float = ...,
alpha_final: float = ..., glow: bool = ..., **kwargs: Any) -> LineCollection: ...
def OrbitPlotOneSlice(sim: Simulation, ax: Axes, Narc: int = ..., color: bool = ..., periastron: bool = ...,
orbit_type: str = ..., lw: float = ..., axes: str = ..., plotparticles: List[int] = ...,
primary: Optional[Particle] = ..., fancy: bool = ...) -> None: ...
def OrbitPlotAddFancyStars(ax: Axes, lw: float, slices: float = ...) -> None: ...