def __init__(self, qapp, options):
QtGui.QWidget.__init__(self, None)
self.gui = Ui_MainWindow()
self.gui.setupUi(self)
self.connect(self.gui.filterTypeComboBox,
Qt.SIGNAL("currentIndexChanged(const QString&)"),
self.changed_filter_type)
self.connect(self.gui.filterDesignTypeComboBox,
Qt.SIGNAL("currentIndexChanged(const QString&)"),
self.changed_filter_design_type)
self.connect(self.gui.designButton,
Qt.SIGNAL("released()"),
self.design)
self.connect(self.gui.tabGroup,
Qt.SIGNAL("currentChanged(int)"),
self.tab_changed)
self.connect(self.gui.nfftEdit,
Qt.SIGNAL("textEdited(QString)"),
self.nfft_edit_changed)
self.gui.designButton.setShortcut(QtCore.Qt.Key_Return)
self.taps = []
self.fftdB = []
self.fftDeg = []
self.groupDelay = []
self.nfftpts = int(10000)
self.gui.nfftEdit.setText(Qt.QString("%1").arg(self.nfftpts))
self.firFilters = ("Low Pass", "Band Pass", "Complex Band Pass", "Band Notch",
"High Pass", "Root Raised Cosine", "Gaussian")
self.optFilters = ("Low Pass", "Band Pass", "Complex Band Pass",
"Band Notch", "High Pass")
self.set_windowed()
# Initialize to LPF
self.gui.filterTypeWidget.setCurrentWidget(self.gui.firlpfPage)
# Set Axis labels
self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.xBottom,
"Frequency (Hz)")
self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.yLeft,
"Magnitude (dB)")
self.gui.timePlot.setAxisTitle(self.gui.timePlot.xBottom,
"Tap number")
self.gui.timePlot.setAxisTitle(self.gui.timePlot.yLeft,
"Amplitude")
self.gui.phasePlot.setAxisTitle(self.gui.phasePlot.xBottom,
"Frequency (Hz)")
self.gui.phasePlot.setAxisTitle(self.gui.phasePlot.yLeft,
"Phase (Radians)")
self.gui.groupPlot.setAxisTitle(self.gui.groupPlot.xBottom,
"Frequency (Hz)")
self.gui.groupPlot.setAxisTitle(self.gui.groupPlot.yLeft,
"Delay (sec)")
# Set up plot curves
self.rcurve = Qwt.QwtPlotCurve("Real")
self.rcurve.attach(self.gui.timePlot)
self.icurve = Qwt.QwtPlotCurve("Imag")
self.icurve.attach(self.gui.timePlot)
self.freqcurve = Qwt.QwtPlotCurve("PSD")
self.freqcurve.attach(self.gui.freqPlot)
self.phasecurve = Qwt.QwtPlotCurve("Phase")
self.phasecurve.attach(self.gui.phasePlot)
self.groupcurve = Qwt.QwtPlotCurve("Group Delay")
self.groupcurve.attach(self.gui.groupPlot)
# Create zoom functionality for the plots
self.timeZoomer = Qwt.QwtPlotZoomer(self.gui.timePlot.xBottom,
self.gui.timePlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.timePlot.canvas())
self.freqZoomer = Qwt.QwtPlotZoomer(self.gui.freqPlot.xBottom,
self.gui.freqPlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.freqPlot.canvas())
self.phaseZoomer = Qwt.QwtPlotZoomer(self.gui.phasePlot.xBottom,
self.gui.phasePlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.phasePlot.canvas())
self.groupZoomer = Qwt.QwtPlotZoomer(self.gui.groupPlot.xBottom,
self.gui.groupPlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.groupPlot.canvas())
# Set up pen for colors and line width
blue = QtGui.qRgb(0x00, 0x00, 0xFF)
blueBrush = Qt.QBrush(Qt.QColor(blue))
red = QtGui.qRgb(0xFF, 0x00, 0x00)
redBrush = Qt.QBrush(Qt.QColor(red))
self.freqcurve.setPen(Qt.QPen(blueBrush, 2))
self.rcurve.setPen(Qt.QPen(blueBrush, 2))
self.icurve.setPen(Qt.QPen(redBrush, 2))
self.phasecurve.setPen(Qt.QPen(blueBrush, 2))
self.groupcurve.setPen(Qt.QPen(blueBrush, 2))
# Set up validators for edit boxes
self.intVal = Qt.QIntValidator(None)
self.dblVal = Qt.QDoubleValidator(None)
self.gui.nfftEdit.setValidator(self.intVal)
self.gui.sampleRateEdit.setValidator(self.dblVal)
self.gui.filterGainEdit.setValidator(self.dblVal)
self.gui.endofLpfPassBandEdit.setValidator(self.dblVal)
self.gui.startofLpfStopBandEdit.setValidator(self.dblVal)
self.gui.lpfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.lpfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.startofBpfPassBandEdit.setValidator(self.dblVal)
self.gui.endofBpfPassBandEdit.setValidator(self.dblVal)
self.gui.bpfTransitionEdit.setValidator(self.dblVal)
self.gui.bpfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.bpfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.startofBnfStopBandEdit.setValidator(self.dblVal)
self.gui.endofBnfStopBandEdit.setValidator(self.dblVal)
self.gui.bnfTransitionEdit.setValidator(self.dblVal)
self.gui.bnfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.bnfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.endofHpfStopBandEdit.setValidator(self.dblVal)
self.gui.startofHpfPassBandEdit.setValidator(self.dblVal)
self.gui.hpfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.hpfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.rrcSymbolRateEdit.setValidator(self.dblVal)
self.gui.rrcAlphaEdit.setValidator(self.dblVal)
self.gui.rrcNumTapsEdit.setValidator(self.dblVal)
self.gui.gausSymbolRateEdit.setValidator(self.dblVal)
self.gui.gausBTEdit.setValidator(self.dblVal)
self.gui.gausNumTapsEdit.setValidator(self.dblVal)
self.gui.nTapsEdit.setText("0")
self.filterWindows = {"Hamming Window" : gr.firdes.WIN_HAMMING,
"Hann Window" : gr.firdes.WIN_HANN,
"Blackman Window" : gr.firdes.WIN_BLACKMAN,
"Rectangular Window" : gr.firdes.WIN_RECTANGULAR,
"Kaiser Window" : gr.firdes.WIN_KAISER,
"Blackman-harris Window" : gr.firdes.WIN_BLACKMAN_hARRIS}
self.show()
class gr_plot_filter(QtGui.QMainWindow):
def __init__(self, qapp, options):
QtGui.QWidget.__init__(self, None)
self.gui = Ui_MainWindow()
self.gui.setupUi(self)
self.connect(self.gui.filterTypeComboBox,
Qt.SIGNAL("currentIndexChanged(const QString&)"),
self.changed_filter_type)
self.connect(self.gui.filterDesignTypeComboBox,
Qt.SIGNAL("currentIndexChanged(const QString&)"),
self.changed_filter_design_type)
self.connect(self.gui.designButton,
Qt.SIGNAL("released()"),
self.design)
self.connect(self.gui.tabGroup,
Qt.SIGNAL("currentChanged(int)"),
self.tab_changed)
self.connect(self.gui.nfftEdit,
Qt.SIGNAL("textEdited(QString)"),
self.nfft_edit_changed)
self.gui.designButton.setShortcut(QtCore.Qt.Key_Return)
self.taps = []
self.fftdB = []
self.fftDeg = []
self.groupDelay = []
self.nfftpts = int(10000)
self.gui.nfftEdit.setText(Qt.QString("%1").arg(self.nfftpts))
self.firFilters = ("Low Pass", "Band Pass", "Complex Band Pass", "Band Notch",
"High Pass", "Root Raised Cosine", "Gaussian")
self.optFilters = ("Low Pass", "Band Pass", "Complex Band Pass",
"Band Notch", "High Pass")
self.set_windowed()
# Initialize to LPF
self.gui.filterTypeWidget.setCurrentWidget(self.gui.firlpfPage)
# Set Axis labels
self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.xBottom,
"Frequency (Hz)")
self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.yLeft,
"Magnitude (dB)")
self.gui.timePlot.setAxisTitle(self.gui.timePlot.xBottom,
"Tap number")
self.gui.timePlot.setAxisTitle(self.gui.timePlot.yLeft,
"Amplitude")
self.gui.phasePlot.setAxisTitle(self.gui.phasePlot.xBottom,
"Frequency (Hz)")
self.gui.phasePlot.setAxisTitle(self.gui.phasePlot.yLeft,
"Phase (Radians)")
self.gui.groupPlot.setAxisTitle(self.gui.groupPlot.xBottom,
"Frequency (Hz)")
self.gui.groupPlot.setAxisTitle(self.gui.groupPlot.yLeft,
"Delay (sec)")
# Set up plot curves
self.rcurve = Qwt.QwtPlotCurve("Real")
self.rcurve.attach(self.gui.timePlot)
self.icurve = Qwt.QwtPlotCurve("Imag")
self.icurve.attach(self.gui.timePlot)
self.freqcurve = Qwt.QwtPlotCurve("PSD")
self.freqcurve.attach(self.gui.freqPlot)
self.phasecurve = Qwt.QwtPlotCurve("Phase")
self.phasecurve.attach(self.gui.phasePlot)
self.groupcurve = Qwt.QwtPlotCurve("Group Delay")
self.groupcurve.attach(self.gui.groupPlot)
# Create zoom functionality for the plots
self.timeZoomer = Qwt.QwtPlotZoomer(self.gui.timePlot.xBottom,
self.gui.timePlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.timePlot.canvas())
self.freqZoomer = Qwt.QwtPlotZoomer(self.gui.freqPlot.xBottom,
self.gui.freqPlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.freqPlot.canvas())
self.phaseZoomer = Qwt.QwtPlotZoomer(self.gui.phasePlot.xBottom,
self.gui.phasePlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.phasePlot.canvas())
self.groupZoomer = Qwt.QwtPlotZoomer(self.gui.groupPlot.xBottom,
self.gui.groupPlot.yLeft,
Qwt.QwtPicker.PointSelection,
Qwt.QwtPicker.AlwaysOn,
self.gui.groupPlot.canvas())
# Set up pen for colors and line width
blue = QtGui.qRgb(0x00, 0x00, 0xFF)
blueBrush = Qt.QBrush(Qt.QColor(blue))
red = QtGui.qRgb(0xFF, 0x00, 0x00)
redBrush = Qt.QBrush(Qt.QColor(red))
self.freqcurve.setPen(Qt.QPen(blueBrush, 2))
self.rcurve.setPen(Qt.QPen(blueBrush, 2))
self.icurve.setPen(Qt.QPen(redBrush, 2))
self.phasecurve.setPen(Qt.QPen(blueBrush, 2))
self.groupcurve.setPen(Qt.QPen(blueBrush, 2))
# Set up validators for edit boxes
self.intVal = Qt.QIntValidator(None)
self.dblVal = Qt.QDoubleValidator(None)
self.gui.nfftEdit.setValidator(self.intVal)
self.gui.sampleRateEdit.setValidator(self.dblVal)
self.gui.filterGainEdit.setValidator(self.dblVal)
self.gui.endofLpfPassBandEdit.setValidator(self.dblVal)
self.gui.startofLpfStopBandEdit.setValidator(self.dblVal)
self.gui.lpfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.lpfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.startofBpfPassBandEdit.setValidator(self.dblVal)
self.gui.endofBpfPassBandEdit.setValidator(self.dblVal)
self.gui.bpfTransitionEdit.setValidator(self.dblVal)
self.gui.bpfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.bpfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.startofBnfStopBandEdit.setValidator(self.dblVal)
self.gui.endofBnfStopBandEdit.setValidator(self.dblVal)
self.gui.bnfTransitionEdit.setValidator(self.dblVal)
self.gui.bnfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.bnfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.endofHpfStopBandEdit.setValidator(self.dblVal)
self.gui.startofHpfPassBandEdit.setValidator(self.dblVal)
self.gui.hpfStopBandAttenEdit.setValidator(self.dblVal)
self.gui.hpfPassBandRippleEdit.setValidator(self.dblVal)
self.gui.rrcSymbolRateEdit.setValidator(self.dblVal)
self.gui.rrcAlphaEdit.setValidator(self.dblVal)
self.gui.rrcNumTapsEdit.setValidator(self.dblVal)
self.gui.gausSymbolRateEdit.setValidator(self.dblVal)
self.gui.gausBTEdit.setValidator(self.dblVal)
self.gui.gausNumTapsEdit.setValidator(self.dblVal)
self.gui.nTapsEdit.setText("0")
self.filterWindows = {"Hamming Window" : gr.firdes.WIN_HAMMING,
"Hann Window" : gr.firdes.WIN_HANN,
"Blackman Window" : gr.firdes.WIN_BLACKMAN,
"Rectangular Window" : gr.firdes.WIN_RECTANGULAR,
"Kaiser Window" : gr.firdes.WIN_KAISER,
"Blackman-harris Window" : gr.firdes.WIN_BLACKMAN_hARRIS}
self.show()
def changed_filter_type(self, ftype):
strftype = str(ftype.toAscii())
if(ftype == "Low Pass"):
self.gui.filterTypeWidget.setCurrentWidget(self.gui.firlpfPage)
elif(ftype == "Band Pass"):
self.gui.filterTypeWidget.setCurrentWidget(self.gui.firbpfPage)
elif(ftype == "Complex Band Pass"):
self.gui.filterTypeWidget.setCurrentWidget(self.gui.firbpfPage)
elif(ftype == "Band Notch"):
self.gui.filterTypeWidget.setCurrentWidget(self.gui.firbnfPage)
elif(ftype == "High Pass"):
self.gui.filterTypeWidget.setCurrentWidget(self.gui.firhpfPage)
elif(ftype == "Root Raised Cosine"):
self.gui.filterTypeWidget.setCurrentWidget(self.gui.rrcPage)
elif(ftype == "Gaussian"):
self.gui.filterTypeWidget.setCurrentWidget(self.gui.gausPage)
self.design()
def changed_filter_design_type(self, design):
if(design == "Equiripple"):
self.set_equiripple()
else:
self.set_windowed()
self.design()
def set_equiripple(self):
# Stop sending the signal for this function
self.gui.filterTypeComboBox.blockSignals(True)
self.equiripple = True
self.gui.lpfPassBandRippleLabel.setVisible(True)
self.gui.lpfPassBandRippleEdit.setVisible(True)
self.gui.bpfPassBandRippleLabel.setVisible(True)
self.gui.bpfPassBandRippleEdit.setVisible(True)
self.gui.bnfPassBandRippleLabel.setVisible(True)
self.gui.bnfPassBandRippleEdit.setVisible(True)
self.gui.hpfPassBandRippleLabel.setVisible(True)
self.gui.hpfPassBandRippleEdit.setVisible(True)
# Save current type and repopulate the combo box for
# filters this window type can handle
currenttype = self.gui.filterTypeComboBox.currentText()
items = self.gui.filterTypeComboBox.count()
for i in xrange(items):
self.gui.filterTypeComboBox.removeItem(0)
self.gui.filterTypeComboBox.addItems(self.optFilters)
# If the last filter type was valid for this window type,
# go back to it; otherwise, reset
try:
index = self.optFilters.index(currenttype)
self.gui.filterTypeComboBox.setCurrentIndex(index)
except ValueError:
pass
# Tell gui its ok to start sending this signal again
self.gui.filterTypeComboBox.blockSignals(False)
def set_windowed(self):
# Stop sending the signal for this function
self.gui.filterTypeComboBox.blockSignals(True)
self.equiripple = False
self.gui.lpfPassBandRippleLabel.setVisible(False)
self.gui.lpfPassBandRippleEdit.setVisible(False)
self.gui.bpfPassBandRippleLabel.setVisible(False)
self.gui.bpfPassBandRippleEdit.setVisible(False)
self.gui.bnfPassBandRippleLabel.setVisible(False)
self.gui.bnfPassBandRippleEdit.setVisible(False)
self.gui.hpfPassBandRippleLabel.setVisible(False)
self.gui.hpfPassBandRippleEdit.setVisible(False)
# Save current type and repopulate the combo box for
# filters this window type can handle
currenttype = self.gui.filterTypeComboBox.currentText()
items = self.gui.filterTypeComboBox.count()
for i in xrange(items):
self.gui.filterTypeComboBox.removeItem(0)
self.gui.filterTypeComboBox.addItems(self.firFilters)
# If the last filter type was valid for this window type,
# go back to it; otherwise, reset
try:
index = self.optFilters.index(currenttype)
self.gui.filterTypeComboBox.setCurrentIndex(index)
except ValueError:
pass
# Tell gui its ok to start sending this signal again
self.gui.filterTypeComboBox.blockSignals(False)
def design(self):
ret = True
fs,r = self.gui.sampleRateEdit.text().toDouble()
ret = r and ret
gain,r = self.gui.filterGainEdit.text().toDouble()
ret = r and ret
if(ret):
winstr = str(self.gui.filterDesignTypeComboBox.currentText().toAscii())
ftype = str(self.gui.filterTypeComboBox.currentText().toAscii())
if(winstr == "Equiripple"):
designer = {"Low Pass" : self.design_opt_lpf,
"Band Pass" : self.design_opt_bpf,
"Complex Band Pass" : self.design_opt_cbpf,
"Band Notch" : self.design_opt_bnf,
"High Pass" : self.design_opt_hpf}
taps,r = designer[ftype](fs, gain)
else:
designer = {"Low Pass" : self.design_win_lpf,
"Band Pass" : self.design_win_bpf,
"Complex Band Pass" : self.design_win_cbpf,
"Band Notch" : self.design_win_bnf,
"High Pass" : self.design_win_hpf,
"Root Raised Cosine" : self.design_win_rrc,
"Gaussian" : self.design_win_gaus}
wintype = self.filterWindows[winstr]
taps,r = designer[ftype](fs, gain, wintype)
if(r):
self.taps = scipy.array(taps)
self.get_fft(fs, self.taps, self.nfftpts)
self.update_time_curves()
self.update_freq_curves()
self.update_phase_curves()
self.update_group_curves()
self.gui.nTapsEdit.setText(Qt.QString("%1").arg(self.taps.size))
# Filter design functions using a window
def design_win_lpf(self, fs, gain, wintype):
ret = True
pb,r = self.gui.endofLpfPassBandEdit.text().toDouble()
ret = r and ret
sb,r = self.gui.startofLpfStopBandEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.lpfStopBandAttenEdit.text().toDouble()
ret = r and ret
if(ret):
tb = sb - pb
taps = gr.firdes.low_pass_2(gain, fs, pb, tb,
atten, wintype)
return (taps, ret)
else:
return ([], ret)
def design_win_bpf(self, fs, gain, wintype):
ret = True
pb1,r = self.gui.startofBpfPassBandEdit.text().toDouble()
ret = r and ret
pb2,r = self.gui.endofBpfPassBandEdit.text().toDouble()
ret = r and ret
tb,r = self.gui.bpfTransitionEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.bpfStopBandAttenEdit.text().toDouble()
ret = r and ret
if(r):
taps = gr.firdes.band_pass_2(gain, fs, pb1, pb2, tb,
atten, wintype)
return (taps,r)
else:
return ([],r)
def design_win_cbpf(self, fs, gain, wintype):
ret = True
pb1,r = self.gui.startofBpfPassBandEdit.text().toDouble()
ret = r and ret
pb2,r = self.gui.endofBpfPassBandEdit.text().toDouble()
ret = r and ret
tb,r = self.gui.bpfTransitionEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.bpfStopBandAttenEdit.text().toDouble()
ret = r and ret
if(r):
taps = gr.firdes.complex_band_pass_2(gain, fs, pb1, pb2, tb,
atten, wintype)
return (taps,r)
else:
return ([],r)
def design_win_bnf(self, fs, gain, wintype):
ret = True
pb1,r = self.gui.startofBnfStopBandEdit.text().toDouble()
ret = r and ret
pb2,r = self.gui.endofBnfStopBandEdit.text().toDouble()
ret = r and ret
tb,r = self.gui.bnfTransitionEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.bnfStopBandAttenEdit.text().toDouble()
ret = r and ret
if(r):
taps = gr.firdes.band_reject_2(gain, fs, pb1, pb2, tb,
atten, wintype)
return (taps,r)
else:
return ([],r)
def design_win_hpf(self, fs, gain, wintype):
ret = True
sb,r = self.gui.endofHpfStopBandEdit.text().toDouble()
ret = r and ret
pb,r = self.gui.startofHpfPassBandEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.hpfStopBandAttenEdit.text().toDouble()
ret = r and ret
if(r):
tb = pb - sb
taps = gr.firdes.high_pass_2(gain, fs, pb, tb,
atten, wintype)
return (taps,r)
else:
return ([],r)
def design_win_rrc(self, fs, gain, wintype):
ret = True
sr,r = self.gui.rrcSymbolRateEdit.text().toDouble()
ret = r and ret
alpha,r = self.gui.rrcAlphaEdit.text().toDouble()
ret = r and ret
ntaps,r = self.gui.rrcNumTapsEdit.text().toInt()
ret = r and ret
if(r):
taps = gr.firdes.root_raised_cosine(gain, fs, sr,
alpha, ntaps)
return (taps,r)
else:
return ([],r)
def design_win_gaus(self, fs, gain, wintype):
ret = True
sr,r = self.gui.gausSymbolRateEdit.text().toDouble()
ret = r and ret
bt,r = self.gui.gausBTEdit.text().toDouble()
ret = r and ret
ntaps,r = self.gui.gausNumTapsEdit.text().toInt()
ret = r and ret
if(r):
spb = fs / sr
taps = gr.firdes.gaussian(gain, spb, bt, ntaps)
return (taps,r)
else:
return ([],r)
# Design Functions for Equiripple Filters
def design_opt_lpf(self, fs, gain):
ret = True
pb,r = self.gui.endofLpfPassBandEdit.text().toDouble()
ret = r and ret
sb,r = self.gui.startofLpfStopBandEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.lpfStopBandAttenEdit.text().toDouble()
ret = r and ret
ripple,r = self.gui.lpfPassBandRippleEdit.text().toDouble()
ret = r and ret
if(ret):
taps = blks2.optfir.low_pass(gain, fs, pb, sb,
ripple, atten)
return (taps, ret)
else:
return ([], ret)
def design_opt_bpf(self, fs, gain):
ret = True
pb1,r = self.gui.startofBpfPassBandEdit.text().toDouble()
ret = r and ret
pb2,r = self.gui.endofBpfPassBandEdit.text().toDouble()
ret = r and ret
tb,r = self.gui.bpfTransitionEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.bpfStopBandAttenEdit.text().toDouble()
ret = r and ret
ripple,r = self.gui.bpfPassBandRippleEdit.text().toDouble()
ret = r and ret
if(r):
sb1 = pb1 - tb
sb2 = pb2 + tb
taps = blks2.optfir.band_pass(gain, fs, sb1, pb1, pb2, sb2,
ripple, atten)
return (taps,r)
else:
return ([],r)
def design_opt_cbpf(self, fs, gain):
ret = True
pb1,r = self.gui.startofBpfPassBandEdit.text().toDouble()
ret = r and ret
pb2,r = self.gui.endofBpfPassBandEdit.text().toDouble()
ret = r and ret
tb,r = self.gui.bpfTransitionEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.bpfStopBandAttenEdit.text().toDouble()
ret = r and ret
ripple,r = self.gui.bpfPassBandRippleEdit.text().toDouble()
ret = r and ret
if(r):
sb1 = pb1 - tb
sb2 = pb2 + tb
taps = blks2.optfir.complex_band_pass(gain, fs, sb1, pb1, pb2, sb2,
ripple, atten)
return (taps,r)
else:
return ([],r)
def design_opt_bnf(self, fs, gain):
ret = True
sb1,r = self.gui.startofBnfStopBandEdit.text().toDouble()
ret = r and ret
sb2,r = self.gui.endofBnfStopBandEdit.text().toDouble()
ret = r and ret
tb,r = self.gui.bnfTransitionEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.bnfStopBandAttenEdit.text().toDouble()
ret = r and ret
ripple,r = self.gui.bnfPassBandRippleEdit.text().toDouble()
ret = r and ret
if(r):
pb1 = sb1 - tb
pb2 = sb2 + tb
taps = blks2.optfir.band_reject(gain, fs, pb1, sb1, sb2, pb2,
ripple, atten)
return (taps,r)
else:
return ([],r)
def design_opt_hpf(self, fs, gain):
ret = True
sb,r = self.gui.endofHpfStopBandEdit.text().toDouble()
ret = r and ret
pb,r = self.gui.startofHpfPassBandEdit.text().toDouble()
ret = r and ret
atten,r = self.gui.hpfStopBandAttenEdit.text().toDouble()
ret = r and ret
ripple,r = self.gui.hpfPassBandRippleEdit.text().toDouble()
ret = r and ret
if(r):
taps = blks2.optfir.high_pass(gain, fs, sb, pb,
atten, ripple)
return (taps,r)
else:
return ([],r)
def nfft_edit_changed(self, nfft):
infft,r = nfft.toInt()
if(r and (infft != self.nfftpts)):
self.nfftpts = infft
self.update_freq_curves()
def tab_changed(self, tab):
if(tab == 0):
self.update_freq_curves()
if(tab == 1):
self.update_time_curves()
if(tab == 2):
self.update_phase_curves()
if(tab == 3):
self.update_group_curves()
def get_fft(self, fs, taps, Npts):
Ts = 1.0/fs
fftpts = fftpack.fft(taps, Npts)
self.freq = scipy.arange(0, fs, 1.0/(Npts*Ts))
self.fftdB = 20.0*scipy.log10(abs(fftpts))
self.fftDeg = scipy.unwrap(scipy.angle(fftpts))
self.groupDelay = -scipy.diff(self.fftDeg)
def update_time_curves(self):
ntaps = len(self.taps)
if(ntaps > 0):
if(type(self.taps[0]) == scipy.complex128):
self.rcurve.setData(scipy.arange(ntaps), self.taps.real)
self.icurve.setData(scipy.arange(ntaps), self.taps.imag)
else:
self.rcurve.setData(scipy.arange(ntaps), self.taps)
# Reset the x-axis to the new time scale
ymax = 1.5 * max(self.taps)
ymin = 1.5 * min(self.taps)
self.gui.timePlot.setAxisScale(self.gui.timePlot.xBottom,
0, ntaps)
self.gui.timePlot.setAxisScale(self.gui.timePlot.yLeft,
ymin, ymax)
# Set the zoomer base to unzoom to the new axis
self.timeZoomer.setZoomBase()
self.gui.timePlot.replot()
def update_freq_curves(self):
npts = len(self.fftdB)
if(npts > 0):
self.freqcurve.setData(self.freq, self.fftdB)
# Reset the x-axis to the new time scale
ymax = 1.5 * max(self.fftdB[0:npts/2])
ymin = 1.1 * min(self.fftdB[0:npts/2])
xmax = self.freq[npts/2]
xmin = self.freq[0]
self.gui.freqPlot.setAxisScale(self.gui.freqPlot.xBottom,
xmin, xmax)
self.gui.freqPlot.setAxisScale(self.gui.freqPlot.yLeft,
ymin, ymax)
# Set the zoomer base to unzoom to the new axis
self.freqZoomer.setZoomBase()
self.gui.freqPlot.replot()
def update_phase_curves(self):
npts = len(self.fftDeg)
if(npts > 0):
self.phasecurve.setData(self.freq, self.fftDeg)
# Reset the x-axis to the new time scale
ymax = 1.5 * max(self.fftDeg[0:npts/2])
ymin = 1.1 * min(self.fftDeg[0:npts/2])
xmax = self.freq[npts/2]
xmin = self.freq[0]
self.gui.phasePlot.setAxisScale(self.gui.phasePlot.xBottom,
xmin, xmax)
self.gui.phasePlot.setAxisScale(self.gui.phasePlot.yLeft,
ymin, ymax)
# Set the zoomer base to unzoom to the new axis
self.phaseZoomer.setZoomBase()
self.gui.phasePlot.replot()
def update_group_curves(self):
npts = len(self.groupDelay)
if(npts > 0):
self.groupcurve.setData(self.freq, self.groupDelay)
# Reset the x-axis to the new time scale
ymax = 1.5 * max(self.groupDelay[0:npts/2])
ymin = 1.1 * min(self.groupDelay[0:npts/2])
xmax = self.freq[npts/2]
xmin = self.freq[0]
self.gui.groupPlot.setAxisScale(self.gui.groupPlot.xBottom,
xmin, xmax)
self.gui.groupPlot.setAxisScale(self.gui.groupPlot.yLeft,
ymin, ymax)
# Set the zoomer base to unzoom to the new axis
self.groupZoomer.setZoomBase()
self.gui.groupPlot.replot()
