Module pyspecan.controller.tkGUI.shared.iq3d

def define_args(parser: argparse.ArgumentParser):
    pass

class IQ3D(TimePlotController):
    def __init__(self, parent, pane: Panel, **kwargs):
        pane.master.config(text="IQ 3D")
        self.amp_max = 0
        self.x_arr = np.arange(0, dtype=np.float16)
        super().__init__(parent, pane, **kwargs)
        fig = plt.figure(figsize=(5,5), layout="constrained")

        fig.canvas = FigureCanvasTkAgg(fig, master=self.fr_canv)
        self.plotter = BlitPlot(fig) # type: ignore
        self.plotter.canvas.draw()
        self.plotter.canvas.get_tk_widget().pack(side=tk.LEFT, fill=tk.BOTH, expand=True) # type: ignore
        self.plotter.add_ax("iq", fig.add_subplot(projection="3d"))

        self.plotter.ax("iq").ax.set_autoscale_on(False)
        self.plotter.ax("iq").ax.grid(True, alpha=0.2)
        self.plotter.ax("iq").ax.set_box_aspect((3,1,1)) # type: ignore

        self.update()

    def reset(self):
        self.amp_max = 0

    def _plot(self, samps):
        prev_max = self.amp_max
        self.amp_max = np.max((self.amp_max, -np.min(samps.real), np.max(samps.real), -np.min(samps.imag), np.max(samps.imag)))
        if not self.amp_max == prev_max:
            self.plotter.ax("iq").set_ylim(-self.amp_max, self.amp_max)
            self.plotter.ax("iq").ax.set_zlim(-self.amp_max, self.amp_max) # type: ignore
        s_len = len(samps)
        if not s_len == len(self.x_arr):
            self.x_arr = np.arange(s_len) / s_len
            self.plotter.ax("iq").set_xlim(0, 1)

        r = np.repeat(self.amp_max, s_len)

        line_i = self.plotter.ax("iq").plot3d(self.x_arr, samps.real, -r, name="i", color="r", linestyle="solid", linewidth=1, alpha=0.5)
        line_q = self.plotter.ax("iq").plot3d(self.x_arr, r, samps.imag, name="q", color="b", linestyle="solid", linewidth=1, alpha=0.5)
        line_iq = self.plotter.ax("iq").plot3d(self.x_arr, samps.real, samps.imag, name="iq", color="g", linestyle="solid", marker=".", linewidth=3)

        self.update()

    def draw_settings(self, row=0):
        pass

class PlotConfig:
    pass