Skip to content

FFT Plotter

rfsoc_rfdc.plotter.fft_plotter

Classes

FFTPlotter

Plot complex ADC samples in the frequency domain

Source code in rfsoc_rfdc/plotter/fft_plotter.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class FFTPlotter:
    """Plot complex ADC samples in the frequency domain
    """

    def __init__(self, sample_rate, title):
        self.sample_rate = sample_rate
        self.fig = go.FigureWidget(layout={
            'title': title,
            'xaxis': {'title': 'Frequency (Hz)'},
            'yaxis': {'title': 'Magnitude',
                      'type': 'log',
                      'range': [0, 8]
                      }
        })

        self.fig.add_scattergl(x=[], y=[], name='Magnitude')
        display(self.fig)

    def update_plot(self, iq_data):
        iq_fft = fft(iq_data)
        freq = fftfreq(len(iq_data), d=1/self.sample_rate)
        magnitude = np.abs(iq_fft)
        self.fig.data[0].x = freq
        self.fig.data[0].y = magnitude

    def close(self):
        """Close the figure widget."""
        self.fig.close()

    def __del__(self):
        self.close()
Functions
close()

Close the figure widget.

Source code in rfsoc_rfdc/plotter/fft_plotter.py 
32
33
34
def close(self):
    """Close the figure widget."""
    self.fig.close()