def __init__(self):
QMainWindow.__init__(self)
self.setMinimumSize(QSize(640,480))
self.setWindowTitle("Test")
c = QWidget(self)
self.setCentralWidget(c)
#TODO: Add a Menu that allows user to select from certain preset filters (commonly used)
mainMenu = self.menuBar()
fileMenu = mainMenu.addMenu("&File")
importAction = QAction("&Import Settings",self)
importAction.setStatusTip("Import HSV Tolerances")
importAction.triggered.connect(self.importRanges)
exportAction = QAction("&Export Settings",self)
exportAction.setStatusTip("Export Current Tolerances")
exportAction.triggered.connect(self.exportRanges)
fileMenu.addAction(importAction)
fileMenu.addAction(exportAction)
gridLayout = QGridLayout(self)
c.setLayout(gridLayout)
self.frame = QLabel(self)
self.frame.resize(640, 480)
gridLayout.addWidget(self.frame,0,0,4,4)
self.th = vidThread()
self.th.changePixmap.connect(self.setImage)
self.th.start()
#Initialize Range Sliders
self.slider1 = QRangeSlider()
self.slider1.setFixedHeight(15)
self.slider2 = QRangeSlider(None,1)
self.slider2.setFixedHeight(15)
self.slider3 = QRangeSlider(None,2)
self.slider3.setFixedHeight(15)
gridLayout.addLayout(self.setupSlider(self.slider1, QLabel("Hue"),self.th.updateRange),5,0)
gridLayout.addLayout(self.setupSlider(self.slider2, QLabel("Saturation"),self.th.updateRange),5,1)
gridLayout.addLayout(self.setupSlider(self.slider3, QLabel("Value"),self.th.updateRange),5,2)
self.slider1.drawValues()
def make_slider_box(self, name, min, max, start, end):
label = QLabel(name)
qse = QRangeSlider()
qse.setMax(max)
qse.setMin(min)
qse.setRange(start,end)
layout = QHBoxLayout()
layout.addWidget(label)
layout.addWidget(qse)
return (layout, qse)
def add_widgets(self, layout, owner, controller):
suffix_label = QLabel("Column Suffix", owner)
self.suffix_box = QLineEdit(owner)
transform_cols_label = QLabel("Columns to transform", owner)
self.transformation_cols_box = ui_objects.ColumnList(owner, controller)
perc_label = QLabel("Select quantile range to flag (inclusive)", owner)
self.perc_slider = QRangeSlider(owner)
layout.addWidget(suffix_label)
layout.addWidget(self.suffix_box)
layout.addWidget(transform_cols_label)
layout.addWidget(self.transformation_cols_box)
layout.addWidget(perc_label)
layout.addWidget(self.perc_slider)
def make_slider_box(self, name,min,max):
label = QLabel(name)
qse = QRangeSlider()
qse.setMax(max)
qse.setMin(min)
qle = QLabel("")
#qle2 = QLineEdit()
layout = QHBoxLayout()
#qle.setEnabled(False)
layout.addWidget(qse)
layout.addWidget(qle)
layout.addWidget(label)
#layout.addWidget(qle2)
#layout.addStretch(1)
return (label,qse, qle,layout)
time.sleep(1) # wait a bit to be sure the roscore is really launched
app = QApplication(sys.argv)
window = QMainWindow()
ui = Ui_RhagGUI()
ui.setupUi(window)
# # Example 1
# rs1 = QRangeSlider()
# rs1.show()
# rs1.setWindowTitle('example 1')
# rs1.setRange(15, 35)
# rs1.setBackgroundStyle('background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #222, stop:1 #333);')
# rs1.setSpanStyle('background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #282, stop:1 #393);')
# rs1 = QRangeSlider()
ui.trajRange=QRangeSlider(ui.tab_2)
# ui.trackFly = QtGui.QCheckBox(ui.tab_2)
# ui.trajRange.setGeometry(QtCore.QRect(540, 47, 111, 22)) #functions for several buttons
ui.trajRange.show()
# rs1.setWindowTitle('example 1')
ui.trajRange.setRange(15, 35)
ui.trajRange.setBackgroundStyle('background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #222, stop:1 #333);')
ui.trajRange.setSpanStyle('background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #282, stop:1 #393);')
#applying and loading settings, closing etc.
okBtn = ui.buttonBox.button(QtGui.QDialogButtonBox.Ok)#todo: Ok needs a function
okBtn.clicked.connect(lambda : saveSettings(window, jsonVR))
#okBtn.clicked.connect(ui.statusbar.showMessage('Ok has no function yet'))
cancelBtn = ui.buttonBox.button(QtGui.QDialogButtonBox.Cancel)
def setPlotTabLayout(self):
# Create Plot Tab --------------------###
# Create Layouts
self.VLayout = QVBoxLayout()
self.H1layout = QHBoxLayout()
self.H1layout.setSpacing(15)
self.H11layout = QHBoxLayout()
self.H2layout = QHBoxLayout()
self.V11layout = QVBoxLayout()
self.V12layout = QVBoxLayout()
self.V21layout = QVBoxLayout()
self.V23layout = QVBoxLayout()
self.V22layout = QVBoxLayout()
#------------------------------------
# Create Widgets for H1layout
# Create Browser
self.plotter = SniffPlot(self.snifferConfig)
self.rangeslider = QRangeSlider(self)
self.filterGroup = QButtonGroup()
self.localFilterRadio = QRadioButton('Local',self)
self.globalFilterRadio = QRadioButton('Global', self)
self.configureFilterButt = QPushButton('Configure Filter')
self.createPlotButt = QPushButton('Create Plot')
self.exportPlotButt = QPushButton('Export Plot')
# We need to sync the UI before connecting any slots in order to prevent accidental stateChanges.
self.syncUiToConfig()
self.configureFilterButt.clicked.connect(self.configureFilter)
self.createPlotButt.clicked.connect(self.createPlot)
self.exportPlotButt.clicked.connect(self.exportPlot)
self.localFilterRadio.clicked.connect(self.localRadioSelected)
self.globalFilterRadio.clicked.connect(self.globalRadioSelected)
self.H11layout.addWidget(self.localFilterRadio)
self.H11layout.addWidget(self.globalFilterRadio)
self.H11layout.addWidget(self.createPlotButt)
self.H11layout.addWidget(self.exportPlotButt)
self.V12layout.addLayout(self.H11layout)
self.V12layout.addWidget(self.configureFilterButt)
self.H1layout.addLayout(self.V11layout)
self.H1layout.addStretch()
self.H1layout.addLayout(self.V12layout)
self.plotWindowStart=QDoubleSpinBox(self)
self.plotWindowEnd=QDoubleSpinBox(self)
self.plotWindowResetStart=QPushButton(self)
self.plotWindowResetEnd=QPushButton(self)
self.plotWindowResetStart.pressed.connect(self.plotter.resetStart)
self.plotWindowResetEnd.pressed.connect(self.plotter.resetEnd)
self.plotWindowResetStart.setText("R")
self.plotWindowResetEnd.setText("R")
self.plotWindowStart.setSuffix("ms")
self.plotWindowEnd.setSuffix("ms")
self.plotWindowStartLabel=QLabel(self)
self.plotWindowDurationLabel=QLabel(self)
self.plotWindowEndLabel=QLabel(self)
self.applyRange=QPushButton(None)
self.applyRange.setText("Apply Range")
self.H2layout.addLayout(self.V21layout)
self.H2layout.addWidget(self.plotWindowResetStart)
self.H2layout.addStretch()
self.H2layout.addLayout(self.V22layout)
self.H2layout.addStretch()
self.H2layout.addWidget(self.plotWindowResetEnd)
self.H2layout.addLayout(self.V23layout)
self.V21layout.addWidget(self.plotWindowStart)
self.V21layout.addWidget(self.plotWindowStartLabel)
self.V22layout.addWidget(self.applyRange)
self.V22layout.addWidget(self.plotWindowDurationLabel)
self.V23layout.addWidget(self.plotWindowEnd)
self.V23layout.addWidget(self.plotWindowEndLabel)
#------------------------------------
self.VLayout.addLayout(self.H1layout)
self.VLayout.addWidget(self.plotter)
self.VLayout.addWidget(self.rangeslider)
self.VLayout.addLayout(self.H2layout)
self.applyRange.pressed.connect(self.plotter.applyRange)
self.plotter.setScrollbar(self.rangeslider)
self.plotter.setWindowSpinner(self.plotWindowStart,self.plotWindowEnd)
self.plotter.setRangeLabels(self.plotWindowStartLabel,self.plotWindowEndLabel,self.plotWindowDurationLabel)
self.dockContents.setLayout(self.VLayout)
def __init__(self, *args, **kwargs):
super(VideoFrame, self).__init__(*args, **kwargs)
self.layout = QVBoxLayout()
self.img_list = []
# STREAM 1 LABEL AREA
self.label = QLabel('Nothing to show right now.', self)
self.label.setAlignment(Qt.AlignCenter)
self.label.setGeometry(30, 20, 750, 360)
self.layout.addWidget(self.label)
# PLAY BUTTON
self.play = QPushButton(self)
icon = QPixmap(os.path.join(os.path.dirname(__file__), 'icons', 'play.png'))
self.play.setIcon(QIcon(icon))
self.play.setGeometry(320, 370, 50, 30)
self.play.clicked.connect(self.timerEvent)
self.layout.addWidget(self.play)
# PAUSE BUTTON
self.pause = QPushButton(self)
icon = QPixmap(os.path.join(os.path.dirname(__file__), 'icons', 'pause.png'))
self.pause.setIcon(QIcon(icon))
self.pause.setGeometry(400, 370, 50, 30)
self.pause.clicked.connect(self.pauseTimer)
self.layout.addWidget(self.pause)
# FAST-FORWARD
self.ff = QPushButton(self)
icon = QPixmap(os.path.join(os.path.dirname(__file__), 'icons', 'ff.png'))
self.ff.setIcon(QIcon(icon))
self.ff.setGeometry(560, 370, 50, 30)
self.ff.clicked.connect(self.fastForward)
self.layout.addWidget(self.ff)
# SLOWDOWN
self.fr = QPushButton(self)
icon = QPixmap(os.path.join(os.path.dirname(__file__), 'icons', 'fr.png'))
self.fr.setIcon(QIcon(icon))
self.fr.setGeometry(240, 370, 50, 30)
self.fr.clicked.connect(self.slowdown)
self.layout.addWidget(self.fr)
self.stop = QPushButton(self)
icon = QPixmap(os.path.join(os.path.dirname(__file__), 'icons', 'stop.png'))
self.stop.setIcon(QIcon(icon))
self.stop.setGeometry(480, 370, 50, 30)
self.stop.clicked.connect(self.stopTimer)
self.layout.addWidget(self.stop)
# SLIDER 1
self.slider = QSlider(Qt.Horizontal, self)
self.layout.addWidget(self.slider)
self.slider.setGeometry(10, 400, 825, 20)
self.slider.valueChanged.connect(self.showImage)
# RANGE SLIDER
self.range_slider = QRangeSlider(self)
self.range_slider.setFixedHeight(30)
self.range_slider.setFixedWidth(825)
self.range_slider.move(10, 420)
self.range_slider.endValueChanged.connect(self.boundEnd)
self.range_slider.startValueChanged.connect(self.boundStart)
self.timer = QBasicTimer() # FIXME consider something with less lag
self.startIdx = 0
self.idx = 0
self.delay = 1000
self.skip = 1
self.calibrate_mode = False
self.alpha = None
self.beta = None
self.t = None
self.z = None
def initUI(self):
# ==================================== Set up window and wrapping layout
self.setWindowTitle("Visualization")
wrapper = QVBoxLayout()
# ========================================== Set up FPS and error labels
labels_layout = QHBoxLayout()
self.label_error = QLabel("")
self.label_fps = QLabel("")
self.label_fps.setAlignment(Qt.AlignRight | Qt.AlignVCenter)
labels_layout.addWidget(self.label_error)
labels_layout.addStretch()
labels_layout.addWidget(self.label_fps)
# ================================================== Set up graph layout
graph_view = pg.GraphicsView()
graph_layout = pg.GraphicsLayout(border=(100,100,100))
graph_view.setCentralItem(graph_layout)
# Mel filterbank plot
fft_plot = graph_layout.addPlot(title='Filterbank Output', colspan=3)
fft_plot.setRange(yRange=[-0.1, 1.2])
fft_plot.disableAutoRange(axis=pg.ViewBox.YAxis)
x_data = np.array(range(1, config.N_FFT_BINS + 1))
self.mel_curve = pg.PlotCurveItem()
self.mel_curve.setData(x=x_data, y=x_data*0)
fft_plot.addItem(self.mel_curve)
# Visualization plot
graph_layout.nextRow()
led_plot = graph_layout.addPlot(title='Visualization Output', colspan=3)
led_plot.setRange(yRange=[-5, 260])
led_plot.disableAutoRange(axis=pg.ViewBox.YAxis)
# Pen for each of the color channel curves
r_pen = pg.mkPen((255, 30, 30, 200), width=4)
g_pen = pg.mkPen((30, 255, 30, 200), width=4)
b_pen = pg.mkPen((30, 30, 255, 200), width=4)
# Color channel curves
self.r_curve = pg.PlotCurveItem(pen=r_pen)
self.g_curve = pg.PlotCurveItem(pen=g_pen)
self.b_curve = pg.PlotCurveItem(pen=b_pen)
# Define x data
x_data = np.array(range(1, config.N_PIXELS + 1))
self.r_curve.setData(x=x_data, y=x_data*0)
self.g_curve.setData(x=x_data, y=x_data*0)
self.b_curve.setData(x=x_data, y=x_data*0)
# Add curves to plot
led_plot.addItem(self.r_curve)
led_plot.addItem(self.g_curve)
led_plot.addItem(self.b_curve)
# ================================================= Set up button layout
label_active = QLabel("Active Effect")
button_grid = QGridLayout()
buttons = {}
connecting_funcs = {}
grid_width = 4
i = 0
j = 0
# Dynamically layout buttons and connect them to the visualisation effects
def connect_generator(effect):
def func():
visualizer.current_effect = effect
func.__name__ = effect
return func
# Where the magic happens
for effect in visualizer.effects:
connecting_funcs[effect] = connect_generator(effect)
buttons[effect] = QPushButton(effect)
buttons[effect].clicked.connect(connecting_funcs[effect])
button_grid.addWidget(buttons[effect], j, i)
i += 1
if i % grid_width == 0:
i = 0
j += 1
# ============================================== Set up frequency slider
# Frequency range label
label_slider = QLabel("Frequency Range")
# Frequency slider
def freq_slider_change(tick):
minf = freq_slider.tickValue(0)**2.0 * (config.MIC_RATE / 2.0)
maxf = freq_slider.tickValue(1)**2.0 * (config.MIC_RATE / 2.0)
t = 'Frequency range: {:.0f} - {:.0f} Hz'.format(minf, maxf)
freq_label.setText(t)
config.MIN_FREQUENCY = minf
config.MAX_FREQUENCY = maxf
dsp.create_mel_bank()
def set_freq_min():
config.MIN_FREQUENCY = freq_slider.start()
dsp.create_mel_bank()
def set_freq_max():
config.MAX_FREQUENCY = freq_slider.end()
dsp.create_mel_bank()
freq_slider = QRangeSlider()
freq_slider.show()
freq_slider.setMin(0)
freq_slider.setMax(20000)
freq_slider.setRange(config.MIN_FREQUENCY, config.MAX_FREQUENCY)
freq_slider.setBackgroundStyle('background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #222, stop:1 #333);')
freq_slider.setSpanStyle('background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #282, stop:1 #393);')
freq_slider.setDrawValues(True)
freq_slider.endValueChanged.connect(set_freq_max)
freq_slider.startValueChanged.connect(set_freq_min)
freq_slider.setStyleSheet("""
QRangeSlider * {
border: 0px;
padding: 0px;
}
QRangeSlider > QSplitter::handle {
background: #fff;
}
QRangeSlider > QSplitter::handle:vertical {
height: 3px;
}
QRangeSlider > QSplitter::handle:pressed {
background: #ca5;
}
""")
# ============================================ Set up option tabs layout
label_options = QLabel("Effect Options")
opts_tabs = QTabWidget()
# Dynamically set up tabs
tabs = {}
grid_layouts = {}
self.grid_layout_widgets = {}
options = visualizer.effect_opts.keys()
for effect in visualizer.effects:
# Make the tab
self.grid_layout_widgets[effect] = {}
tabs[effect] = QWidget()
grid_layouts[effect] = QGridLayout()
tabs[effect].setLayout(grid_layouts[effect])
opts_tabs.addTab(tabs[effect],effect)
# These functions make functions for the dynamic ui generation
# YOU WANT-A DYNAMIC I GIVE-A YOU DYNAMIC!
def gen_slider_valuechanger(effect, key):
def func():
visualizer.effect_opts[effect][key] = self.grid_layout_widgets[effect][key].value()
return func
def gen_float_slider_valuechanger(effect, key):
def func():
visualizer.effect_opts[effect][key] = self.grid_layout_widgets[effect][key].slider_value
return func
def gen_combobox_valuechanger(effect, key):
def func():
visualizer.effect_opts[effect][key] = self.grid_layout_widgets[effect][key].currentText()
visualizer._wavelength_set_color_mode(visualizer.effect_opts[effect][key])
return func
def gen_checkbox_valuechanger(effect, key):
def func():
visualizer.effect_opts[effect][key] = self.grid_layout_widgets[effect][key].isChecked()
return func
# Dynamically generate ui for settings
if effect in visualizer.dynamic_effects_config:
i = 0
connecting_funcs[effect] = {}
for key, label, ui_element, opts in visualizer.dynamic_effects_config[effect][:]:
if ui_element == "slider":
connecting_funcs[effect][key] = gen_slider_valuechanger(effect, key)
self.grid_layout_widgets[effect][key] = QSlider(Qt.Horizontal)
self.grid_layout_widgets[effect][key].setMinimum(opts[0])
self.grid_layout_widgets[effect][key].setMaximum(opts[1])
self.grid_layout_widgets[effect][key].setValue(opts[2])
self.grid_layout_widgets[effect][key].valueChanged.connect(
connecting_funcs[effect][key])
grid_layouts[effect].addWidget(QLabel(label),i,0)
grid_layouts[effect].addWidget(self.grid_layout_widgets[effect][key],i,1)
elif ui_element == "float_slider":
connecting_funcs[effect][key] = gen_float_slider_valuechanger(effect, key)
self.grid_layout_widgets[effect][key] = QFloatSlider(*opts)
self.grid_layout_widgets[effect][key].valueChanged.connect(
connecting_funcs[effect][key])
grid_layouts[effect].addWidget(QLabel(label),i,0)
grid_layouts[effect].addWidget(self.grid_layout_widgets[effect][key],i,1)
elif ui_element == "dropdown":
connecting_funcs[effect][key] = gen_combobox_valuechanger(effect, key)
self.grid_layout_widgets[effect][key] = QComboBox()
self.grid_layout_widgets[effect][key].addItems(opts.keys())
self.grid_layout_widgets[effect][key].currentIndexChanged.connect(
connecting_funcs[effect][key])
grid_layouts[effect].addWidget(QLabel(label),i,0)
grid_layouts[effect].addWidget(self.grid_layout_widgets[effect][key],i,1)
elif ui_element == "checkbox":
connecting_funcs[effect][key] = gen_checkbox_valuechanger(effect, key)
self.grid_layout_widgets[effect][key] = QCheckBox()
#self.grid_layout_widgets[effect][key].addItems(opts.keys())
self.grid_layout_widgets[effect][key].stateChanged.connect(
connecting_funcs[effect][key])
grid_layouts[effect].addWidget(QLabel(label),i,0)
grid_layouts[effect].addWidget(self.grid_layout_widgets[effect][key],i,1)
i += 1
#visualizer.effect_settings[effect]
else:
grid_layouts[effect].addWidget(QLabel("No customisable options for this effect :("),0,0)
# ============================================= Add layouts into wrapper
self.setLayout(wrapper)
wrapper.addLayout(labels_layout)
wrapper.addWidget(graph_view)
wrapper.addWidget(label_active)
wrapper.addLayout(button_grid)
wrapper.addWidget(label_slider)
wrapper.addWidget(freq_slider)
wrapper.addWidget(label_options)
wrapper.addWidget(opts_tabs)
self.show()
#!/usr/bin/env python
import os
import sys
from PyQt4 import QtCore
from PyQt4 import QtGui
from PyQt4 import uic
from qrangeslider import QRangeSlider
app = QtGui.QApplication(sys.argv)
# Example 1
rs1 = QRangeSlider()
rs1.show()
rs1.setWindowTitle('example 1')
rs1.setRange(15, 35)
rs1.setBackgroundStyle(
'background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #222, stop:1 #333);'
)
rs1.setSpanStyle(
'background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #282, stop:1 #393);'
)
# Example 2
rs2 = QRangeSlider()
rs2.show()
rs2.setWindowTitle('example 2')
rs2.setFixedWidth(400)
rs2.setFixedHeight(36)
def get_widget_region(self):
canvas_box = QGroupBox("Selected Region")
layout = QVBoxLayout()
# ================================================================================
# ==================== Horizontal Layout (Matches Button + Add Entire Image Button)
match_layout = QHBoxLayout()
# *** Button (Find Best Match)
self.btn_find_match = QPushButton("Find best matches")
self.btn_find_match.clicked.connect(self.on_click_btn_find_match)
self.btn_find_match.setEnabled(False)
match_layout.addWidget(self.btn_find_match)
self.btn_add_image = QPushButton("Add entire image")
self.btn_add_image.clicked.connect(self.on_click_btn_add_image)
match_layout.addWidget(self.btn_add_image)
layout.addLayout(match_layout)
# ==================== End Horizontal Layout (Match Layout)
# ================================================================================
progress_layout = QHBoxLayout()
# *** ProgressBar (Matching Completion)
self.progressbar_match = QProgressBar()
self.progressbar_match.setValue(0)
progress_layout.addWidget(self.progressbar_match)
layout.addLayout(progress_layout)
# *** Label (Matching Status)
self.label_match_status = QLabel(
"Select a region you want to find a match for")
layout.addWidget(self.label_match_status)
# *** Canvas (Region Crops)
self.canvas_region = Graph(self, width=5, height=5, dpi=100)
self.canvas_region.is_interactive = False
layout.addWidget(self.canvas_region)
# *** Label (Ratio Test Distance)
self.label_slider_ratio_test = QLabel("")
layout.addWidget(self.label_slider_ratio_test)
# *** Slider (Ratio Test Distance)
self.slider_ratio_test = QSlider(Qt.Horizontal)
self.slider_ratio_test.valueChanged.connect(
self.on_slider_change_ratio_test)
self.slider_ratio_test.setMinimum(0)
self.slider_ratio_test.setMaximum(100)
self.slider_ratio_test.setValue(int(config.DISTANCE_RATIO * 100))
self.slider_ratio_test.setTickPosition(QSlider.NoTicks)
self.slider_ratio_test.setTickInterval(1)
self.slider_ratio_test.setEnabled(False)
layout.addWidget(self.slider_ratio_test)
if (config.UI_ANGLE):
# *** Label (Angle)
self.label_angle = QLabel("Angle: 0")
layout.addWidget(self.label_angle)
# *** Slider (Angle)
self.slider_angle = QSlider(Qt.Horizontal)
self.slider_angle.valueChanged.connect(self.on_slider_change_angle)
self.slider_angle.setMinimum(0)
self.slider_angle.setMaximum(360)
self.slider_angle.setTickPosition(QSlider.NoTicks)
self.slider_angle.setTickInterval(1)
self.slider_angle.setEnabled(False)
layout.addWidget(self.slider_angle)
if (config.UI_WARP):
# *** Label (Warp Points)
self.label_warp_points = QLabel("")
layout.addWidget(self.label_warp_points)
# *** Slider (Warp Points)
self.slider_warp_points = QSlider(Qt.Horizontal)
self.slider_warp_points.valueChanged.connect(
self.on_slider_change_warp_points)
self.slider_warp_points.setMinimum(3)
self.slider_warp_points.setMaximum(50)
self.slider_warp_points.setTickPosition(QSlider.NoTicks)
self.slider_warp_points.setTickInterval(1)
self.slider_warp_points.setValue(5)
self.slider_warp_points.setEnabled(False)
layout.addWidget(self.slider_warp_points)
# *** Label (Warp Disp Min)
self.label_warp_disp_min = QLabel("Min Displacement: ")
layout.addWidget(self.label_warp_disp_min)
# *** Label (Warp Disp Max)
self.label_warp_disp_max = QLabel("Max Displacement: ")
layout.addWidget(self.label_warp_disp_max)
from qrangeslider import QRangeSlider
self.slider_warp_disp = QRangeSlider()
self.slider_warp_disp.setMin(0)
self.slider_warp_disp.setMax(50)
self.slider_warp_disp.setRange(1, 5)
self.slider_warp_disp.setEnabled(False)
layout.addWidget(self.slider_warp_disp)
self.btn_warp = QPushButton("Warp")
self.btn_warp.clicked.connect(self.on_click_btn_warp)
self.btn_warp.setEnabled(False)
layout.addWidget(self.btn_warp)
self.btn_reset = QPushButton("Reset")
self.btn_reset.clicked.connect(self.on_click_btn_reset)
self.btn_reset.setEnabled(False)
layout.addWidget(self.btn_reset)
canvas_box.setLayout(layout)
return canvas_box
def initUI(self):
port = self.port
self.create_socket(port)
range_label = QtGui.QLabel('events')
self.range_events = QRangeSlider()
self.range_events.show()
self.range_events.setFixedWidth(300)
self.range_events.setFixedHeight(36)
self.range_events.setMin(0)
self.range_events.setMax(4)
self.range_events.setRange(0, 1)
self.range_events.startValueChanged.connect(
lambda: self.keep_slider_min(self.range_events))
hbox_events = QtGui.QHBoxLayout()
hbox_events.addWidget(range_label)
hbox_events.addWidget(self.range_events)
self.textbox = QtGui.QLineEdit()
self.update_btn = QtGui.QPushButton("update")
self.update_btn.clicked.connect(lambda: self.button_click(port))
self.update_btn.setFixedWidth(100)
hbox = QtGui.QHBoxLayout()
hbox.addWidget(self.update_btn)
magnitude_label = QtGui.QLabel('magnitude in g/10')
self.range_magnitude = QRangeSlider()
self.range_magnitude.show()
self.range_magnitude.setFixedWidth(300)
self.range_magnitude.setFixedHeight(36)
self.range_magnitude.setMin(20)
self.range_magnitude.setMax(80)
self.range_magnitude.setRange(20, 30)
hbox_magnitude = QtGui.QHBoxLayout()
hbox_magnitude.addWidget(magnitude_label)
hbox_magnitude.addWidget(self.range_magnitude)
self.filter_length = QRangeSlider()
self.filter_length.show()
self.filter_length.setFixedWidth(300)
self.filter_length.setFixedHeight(36)
self.filter_length.setMin(0)
self.filter_length.setMax(250)
self.filter_length.setRange(0, 100)
self.filter_length.startValueChanged.connect(
lambda: self.keep_slider_min(self.filter_length))
filter_length_label = QtGui.QLabel('filter length in samples')
hbox_length = QtGui.QHBoxLayout()
hbox_length.addWidget(filter_length_label)
hbox_length.addWidget(self.filter_length)
self.message_label = QtGui.QLabel("messages will be here")
self.exit_btn = QtGui.QPushButton('exit')
self.exit_btn.clicked.connect(
lambda: self.exit_click(self.socket, self.context, port))
vbox = QtGui.QVBoxLayout()
vbox.addStretch(1)
vbox.addLayout(hbox_events)
vbox.addLayout(hbox_magnitude)
vbox.addLayout(hbox_length)
vbox.addWidget(self.message_label)
vbox.addWidget(self.update_btn)
vbox.addLayout(hbox)
vbox.addWidget(self.exit_btn)
self.setLayout(vbox)
self.setGeometry(300, 300, 300, 150)
self.setWindowTitle('rangesliders')
self.show()
def main():
config = {
'inputFolder': None,
'outputFolder': None,
'observationsFile': None,
'populationSize': 0,
'paramCount': 0,
'minparam': np.array([0,0]),
'maxparam':np.array([0,0]),
}
# init ui
app = QApplication([])
window = QWidget()
hlayout = QHBoxLayout()
vlayout = QVBoxLayout()
# buttons
# -- set work folder.
setWorkFolderBtn = QPushButton('Select Work folder')
setWorkFolderBtn.clicked.connect(lambda: setFolderHandler(config,'inputFolder', window))
# -- sets output folder.
setOutputFolderBtn = QPushButton("Select Output folder")
setOutputFolderBtn.clicked.connect(lambda: setFolderHandler(config, 'outputFolder', window))
# -- set observations file path
setObservationsFileBtn = QPushButton("Select Observations")
setObservationsFileBtn.clicked.connect(lambda: setFolderHandler(config, 'observationsFile', window, isFolder=False))
# -- set simulate button
simulateBtn = QPushButton('Simulate')
simulateBtn.clicked.connect(lambda: simulate(config))
# textboxes
# -- set population count textbox
populationCountTxt = QLineEdit()
populationCountTxt.textChanged.connect(lambda text: setValueChangeHandler(text, config, 'populationSize', window))
# arrange components in ui
vlayout.addWidget(setWorkFolderBtn)
vlayout.addWidget(setOutputFolderBtn)
vlayout.addWidget(setObservationsFileBtn)
vlayout.addWidget(QLabel('Calibration Parameters'))
populationFormSet = QHBoxLayout()
populationFormSet.addWidget(QLabel('Population Size'))
populationFormSet.addWidget(populationCountTxt)
paramFormSet = QHBoxLayout()
paramFormSet.addWidget(QLabel('Number of Parameters'))
paramValue = QSlider(Qt.Horizontal)
paramValue.valueChanged.connect(lambda value: setValueChangeHandler(value, config, 'paramCount', window))
paramFormSet.addWidget(paramValue)
maxCalibFormSet = QHBoxLayout()
maxCalibFormSet.addWidget(QLabel('Max Calibration'))
maxRange = QRangeSlider()
maxRange.setMin(0)
maxRange.setMax(1)
maxCalibFormSet.addWidget(maxRange)
minCalibFormSet = QHBoxLayout()
minCalibFormSet.addWidget(QLabel('Min Calibration'))
minRange = QRangeSlider()
minRange.setMin(0)
minRange.setMax(1)
minCalibFormSet.addWidget(minRange)
vlayout.addLayout(populationFormSet)
vlayout.addLayout(paramFormSet)
vlayout.addLayout(maxCalibFormSet)
vlayout.addLayout(minCalibFormSet)
vlayout.addWidget(simulateBtn)
hlayout.addLayout(vlayout)
window.setLayout(hlayout);
window.show()
app.exec_() # add event handlers
def __init__(self, *args, **kwargs):
super(MainWindow, self).__init__(*args, **kwargs)
self.setWindowIcon(
QIcon(os.path.join(os.path.dirname(__file__), 'icons',
'plant.png')))
self.setWindowTitle("Video Comparison GUI")
self.statusBar()
self.img_list = None
self.cur_video_id = 0
self.layout = QGridLayout()
# Main Widget Init
self.widget = QWidget()
self.widget.setLayout(self.layout)
self.setCentralWidget(self.widget)
self.setGeometry(250, 50, 1920, 1080)
# Open File Dialog
openFile = QAction('Open Image Directory', self)
openFile.setShortcut('Ctrl+O')
openFile.setStatusTip('Open Image Directory')
openFile.triggered.connect(self.fileDialog)
change_framerate = QAction('Change Framerate...', self)
change_framerate.setStatusTip(
'Change Playback Framerate of Video (default: 1 frame/sec)')
change_framerate.triggered.connect(self.setFrameRate)
change_skiprate = QAction('Change Skip Rate...', self)
change_skiprate.setStatusTip(
'Change Frame Skip Rate in Selected Photos (default: 1)')
change_skiprate.triggered.connect(self.setSkipRate)
set_v_1 = QAction('Video 1...', self)
set_v_1.setStatusTip('Set Video 1 (Top Left Corner) Settings')
set_v_1.triggered.connect(self.setv1)
set_v_2 = QAction('Video 2...', self)
set_v_2.setStatusTip('Set Video 2 (Top Right Corner) Settings')
set_v_2.triggered.connect(self.setv2)
set_v_3 = QAction('Video 3...', self)
set_v_3.setStatusTip('Set Video 3 (Bottom Left Corner) Settings')
set_v_3.triggered.connect(self.setv3)
set_v_4 = QAction('Video 4...', self)
set_v_4.setStatusTip('Set Video 4 (Bottom Right Corner) Settings')
set_v_4.triggered.connect(self.setv4)
change_calibrate = QAction('Turn Calibration On/Off...', self)
change_calibrate.setStatusTip(
'Calibrate auto height detection to plant pot')
change_calibrate.triggered.connect(self.calibrate)
export_video = QAction('Export Video...', self)
export_video.setStatusTip(
'Export the current video frame with current settings')
export_video.triggered.connect(self.export)
# Menu Bar
menubar = self.menuBar()
file_menu = menubar.addMenu('&File')
file_menu.addAction(openFile)
file_menu.addAction(export_video)
edit_menu = menubar.addMenu('&Edit')
edit_menu.addAction(change_framerate)
edit_menu.addAction(change_skiprate)
video_menu = menubar.addMenu('&Video')
video_menu.addAction(set_v_1)
video_menu.addAction(set_v_2)
video_menu.addAction(set_v_3)
video_menu.addAction(set_v_4)
calibrate_menu = menubar.addMenu('&Calibrate')
calibrate_menu.addAction(change_calibrate)
# Video Frame Objects
self.v1 = VideoFrame()
self.layout.addWidget(self.v1, 0, 0, 1, 2)
self.v2 = VideoFrame()
self.layout.addWidget(self.v2, 0, 2, 1, 2)
self.v3 = VideoFrame()
self.layout.addWidget(self.v3, 1, 0, 1, 2)
self.v4 = VideoFrame()
self.layout.addWidget(self.v4, 1, 2, 1, 2)
self.video_list = [self.v1, self.v2, self.v3, self.v4]
## STREAM 1 LABEL AREA
#self.label = QLabel('Nothing to show right now.')
#self.label.setAlignment(Qt.AlignCenter)
#self.layout.addWidget(self.label)
self.bottomLayout = QVBoxLayout()
# GLOBAL PLAY BUTTON
#self.play = QPushButton("Play")
#self.play.clicked.connect(self.timerEvent)
#self.bottomLayout.addWidget(self.play)
# SLOWDOWN
#self.fr = QPushButton(self)
#icon = QPixmap(os.path.join(os.path.dirname(__file__), 'icons', 'fr.png'))
#self.fr.setIcon(QIcon(icon))
#self.fr.clicked.connect(self.slowdown)
#self.layout.addWidget(self.fr, 2, 0, 1, 1)
# GLOBAL PLAY BUTTON
self.play = QPushButton(self)
icon = QPixmap(
os.path.join(os.path.dirname(__file__), 'icons', 'play.png'))
self.play.setIcon(QIcon(icon))
self.play.clicked.connect(self.timerEvent)
self.layout.addWidget(self.play, 2, 0, 1, 1)
# GLOBAL PAUSE BUTTON
self.pause = QPushButton(self)
icon = QPixmap(
os.path.join(os.path.dirname(__file__), 'icons', 'pause.png'))
self.pause.setIcon(QIcon(icon))
self.pause.clicked.connect(self.pauseTimer)
self.layout.addWidget(self.pause, 2, 1, 1, 1)
# GLOBAL FAST-FORWARD
self.ff = QPushButton(self)
icon = QPixmap(
os.path.join(os.path.dirname(__file__), 'icons', 'ff.png'))
self.ff.setIcon(QIcon(icon))
self.ff.clicked.connect(self.fastForward)
self.layout.addWidget(self.ff, 2, 3, 1, 1)
# GLOBAL STOP
self.stop = QPushButton(self)
icon = QPixmap(
os.path.join(os.path.dirname(__file__), 'icons', 'stop.png'))
self.stop.setIcon(QIcon(icon))
self.stop.clicked.connect(self.stopTimer)
self.layout.addWidget(self.stop, 2, 2, 1, 1)
# GLOBAL SLIDER
self.slider = QSlider(Qt.Horizontal)
self.bottomLayout.addWidget(self.slider)
self.slider.valueChanged.connect(self.showImage)
# GLOBAL RANGE SLIDER
self.range_slider = QRangeSlider()
self.range_slider.setFixedHeight(20)
self.bottomLayout.addWidget(self.range_slider)
self.range_slider.endValueChanged.connect(self.boundEnd)
self.range_slider.startValueChanged.connect(self.boundStart)
self.layout.addLayout(self.bottomLayout, 3, 0, 1, 4)
self.timer = QBasicTimer() # FIXME consider something with less lag
self.startIdx = 0
self.idx = 0
self.delay = 1000
self.skip = 1
