from gui.fig_components.fig_component import FigComponent
from modules.graphs.circular_target import CircularTarget
from modules.graphs.linear_target import LinearTarget
from modules.graphs.style_settings import DEFAULT_STYLE_SETTINGS

class Position3D(FigComponent):
    def __init__(self, parent, max_dist=80, step=10, threshold=40, color_settings=DEFAULT_STYLE_SETTINGS):
        super().__init__(parent)
        self._gs = self._fig.add_gridspec(1, 2, width_ratios=[12, 1], wspace=0.3)

        self._target = CircularTarget(self._fig, self._gs[0],
                                   max_dist=max_dist,
                                   step=step,
                                   threshold=threshold,
                                   color_settings=color_settings)

        self._depth = LinearTarget(self._fig, self._gs[1],
                                   max_dist=max_dist,
                                   step=step,
                                   threshold=threshold,
                                   color_settings=color_settings)

    def plot(self, x, y, z, resize_by_data=False):
        self._target.plot(x, y, resize_by_data)
        self._depth.plot(z, resize_by_data)