from gui.fig_components.fig_component import FigComponent

class ImuSeries(FigComponent):
    def __init__(self, parent):
        super().__init__(parent)

        self._gs = self._fig.add_gridspec(2, 1, hspace=0.4)
        self._acc = self._fig.add_subplot(self._gs[0])
        self._gyr = self._fig.add_subplot(self._gs[1])

    def plot_acc(self, t, mag):
        self._acc.set_title("Akcelerometer")
        self._acc.set_xlabel("Čas (s)")
        self._acc.set_ylabel("$m*s^{-2}$")
        self._acc.plot(t, mag)

    def plot_gyr(self, t, mag):
        self._gyr.set_title("Gyroskop")
        self._gyr.set_xlabel("Čas (s)")
        self._gyr.set_ylabel("$deg*s^{-1}$")
        self._gyr.plot(t, mag)