def __init__(self):
super(MainWindow, self).__init__()
self.move(100, 100)
self.setWindowTitle('Shortest path in cubical complex')
# Create main canevas that will render the OpenGL scene
self.canevas = CanevasGLWidget(self)
# Create buttons
btn_reset = QtGui.QPushButton("Reset")
btn_reset.setStatusTip('Reset camera')
self.connect(btn_reset, QtCore.SIGNAL('clicked()'), self.btn_reset_action)
btn_generate = QtGui.QPushButton("Generate")
btn_generate.setToolTip('Generate random 2D intervals on the 3 planes')
self.connect(btn_generate, QtCore.SIGNAL('clicked()'), self.btn_generate_action)
chk_degenerate = QtGui.QCheckBox("Allow degenerate")
chk_degenerate.stateChanged.connect(self.chk_degenerate_action)
btn_save = QtGui.QPushButton("Save")
self.connect(btn_save, QtCore.SIGNAL('clicked()'), self.btn_save_action)
btn_load = QtGui.QPushButton("Load")
self.connect(btn_load, QtCore.SIGNAL('clicked()'), self.btn_load_action)
chk_draw_axis = QtGui.QCheckBox("Draw axis")
chk_draw_axis.setChecked(self.canevas.draw_axis)
chk_draw_axis.stateChanged.connect(self.chk_draw_axis_action)
self.chk_draw_projs = QtGui.QCheckBox("Draw projs")
self.chk_draw_projs.stateChanged.connect(self.chk_draw_projs_action)
# Place the buttons
vbox = QtGui.QVBoxLayout()
vbox.addWidget(btn_reset)
vbox.addWidget(btn_generate)
vbox.addWidget(chk_degenerate)
vbox.addWidget(btn_save)
vbox.addWidget(btn_load)
vbox.addWidget(chk_draw_axis)
vbox.addWidget(self.chk_draw_projs)
vbox.addStretch(1)
# Place main components
self.canevas.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
hbox = QtGui.QHBoxLayout()
hbox.addLayout(vbox)
hbox.addWidget(self.canevas)
parent_widget = QtGui.QWidget()
parent_widget.setLayout(hbox)
self.setCentralWidget(parent_widget)
class MainWindow(QtGui.QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.move(100, 100)
self.setWindowTitle('Shortest path in cubical complex')
# Create main canevas that will render the OpenGL scene
self.canevas = CanevasGLWidget(self)
# Create buttons
btn_reset = QtGui.QPushButton("Reset")
btn_reset.setStatusTip('Reset camera')
self.connect(btn_reset, QtCore.SIGNAL('clicked()'), self.btn_reset_action)
btn_generate = QtGui.QPushButton("Generate")
btn_generate.setToolTip('Generate random 2D intervals on the 3 planes')
self.connect(btn_generate, QtCore.SIGNAL('clicked()'), self.btn_generate_action)
chk_degenerate = QtGui.QCheckBox("Allow degenerate")
chk_degenerate.stateChanged.connect(self.chk_degenerate_action)
btn_save = QtGui.QPushButton("Save")
self.connect(btn_save, QtCore.SIGNAL('clicked()'), self.btn_save_action)
btn_load = QtGui.QPushButton("Load")
self.connect(btn_load, QtCore.SIGNAL('clicked()'), self.btn_load_action)
chk_draw_axis = QtGui.QCheckBox("Draw axis")
chk_draw_axis.setChecked(self.canevas.draw_axis)
chk_draw_axis.stateChanged.connect(self.chk_draw_axis_action)
self.chk_draw_projs = QtGui.QCheckBox("Draw projs")
self.chk_draw_projs.stateChanged.connect(self.chk_draw_projs_action)
# Place the buttons
vbox = QtGui.QVBoxLayout()
vbox.addWidget(btn_reset)
vbox.addWidget(btn_generate)
vbox.addWidget(chk_degenerate)
vbox.addWidget(btn_save)
vbox.addWidget(btn_load)
vbox.addWidget(chk_draw_axis)
vbox.addWidget(self.chk_draw_projs)
vbox.addStretch(1)
# Place main components
self.canevas.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
hbox = QtGui.QHBoxLayout()
hbox.addLayout(vbox)
hbox.addWidget(self.canevas)
parent_widget = QtGui.QWidget()
parent_widget.setLayout(hbox)
self.setCentralWidget(parent_widget)
#self.resize(500,500)
def btn_reset_action(self):
self.canevas.camera.reset()
self.canevas.update()
def btn_generate_action(self):
self.canevas.model.init_data()
self.chk_draw_projs_action(self.chk_draw_projs.checkState())
self.canevas.update()
def chk_degenerate_action(self, state):
self.canevas.model.degenerate = (state == QtCore.Qt.Checked)
def btn_save_action(self):
file_name = QtGui.QFileDialog.getSaveFileName(self, 'Save intervals as binary file', 'data.i3db', 'Interval3D (*.i3db*)')
if file_name:
f = open(file_name, 'w')
self.canevas.save_data(f)
f.close()
def btn_load_action(self):
file_name = QtGui.QFileDialog.getOpenFileName(self, 'Open binary intervals file', '', 'Interval3D (*.i3db*)')
if file_name:
f = open(file_name, 'r')
self.canevas.load_data(f)
f.close()
def chk_draw_axis_action(self, state):
self.canevas.draw_axis = (state == QtCore.Qt.Checked)
self.canevas.update()
def chk_draw_projs_action(self, state):
if state == QtCore.Qt.Checked:
self.canevas.model.add_projections()
else:
self.canevas.model.remove_projections()
def keyPressEvent(self, keyEvent):
import interval
key = keyEvent.key()
if key == QtCore.Qt.Key_A:
interval.Ax += 0.1
elif key == QtCore.Qt.Key_Z:
interval.Ax -= 0.1
elif key == QtCore.Qt.Key_Q:
interval.Bx += 0.1
elif key == QtCore.Qt.Key_S:
interval.Bx -= 0.1
self.canevas.model.init_data()
self.canevas.update()