def test_move_down_1(self):
sm = SampleModel()
sm.addItem(Layer(name='0'))
sm.addItem(Layer(name='1'))
sm.addItem(Layer(name='2'))
sm.addItem(Layer(name='3'))
sm.move_down_1([3]) #cannot move down more
expected_names = ['incoming media', '0', '1', '2', '3', 'substrate']
self.assertEqual(
[s.name for s in [sm.incoming_media] + sm.layers + [sm.substrate]],
expected_names)
sm.move_down_1([1])
expected_names = ['incoming media', '0', '2', '1', '3', 'substrate']
self.assertEqual(
[s.name for s in [sm.incoming_media] + sm.layers + [sm.substrate]],
expected_names)
sm.move_down_1([0, 2])
expected_names = ['incoming media', '2', '0', '3', '1', 'substrate']
self.assertEqual(
[s.name for s in [sm.incoming_media] + sm.layers + [sm.substrate]],
expected_names)
sm.move_up_1([1, 2, 4]) #no effect
expected_names = ['incoming media', '2', '0', '3', '1', 'substrate']
self.assertEqual(
[s.name for s in [sm.incoming_media] + sm.layers + [sm.substrate]],
expected_names)
def test_iterate(self):
sm = SampleModel()
sm.addItem(Layer(name='0'))
sm.addItem(Layer(name='1'))
sm.addItem(Layer(name='2', thickness=3.14))
sm.addItem(Layer(name='3'))
expected_names = ['0', '1', '2', '3']
self.assertEqual([s.name for s in sm], expected_names)
expected_thickness = [0, 0, 3.14, 0]
self.assertEqual([s.thickness.value for s in sm], expected_thickness)
for i, layer in enumerate(sm):
layer.nsld = 3 * i + 4j
for i in range(4):
self.assertEqual(sm.layers[i].nsld_real.value, 3 * i)
self.assertEqual(sm.layers[i].nsld_imaginary.value, 4)
def test_creation_addition_deletion(self):
sm = SampleModel()
sm.addItem(Layer(name='0'))
sm.addItem(Layer(name='1'))
sm.addItem(Layer(name='2'))
sm.addItem(Layer(name='3'))
sm.addItem('4') #should have no effect
expected_names = ['incoming media', '0', '1', '2', '3', 'substrate']
self.assertEqual(
[s.name for s in [sm.incoming_media] + sm.layers + [sm.substrate]],
expected_names)
sm.delItem(0)
expected_names = ['incoming media', '1', '2', '3', 'substrate']
self.assertEqual(
[s.name for s in [sm.incoming_media] + sm.layers + [sm.substrate]],
expected_names)
sm.delItem(3)
expected_names = ['incoming media', '1', '2', '3', 'substrate']
self.assertEqual(
[s.name for s in [sm.incoming_media] + sm.layers + [sm.substrate]],
expected_names)
class layerselector(QtWidgets.QWidget, Ui_layerselector):
invalidSelection = QtCore.pyqtSignal(str)
sampleModelChanged = QtCore.pyqtSignal(SampleModel)
def __init__(self, sample_model=SampleModel(), *args):
QtWidgets.QWidget.__init__(self, *args)
Ui_layerselector.__init__(self)
self.setupUi(self)
self.sample_model = SampleModel()
self.set_sample_model(sample_model)
self.listView.selectionModel().selectionChanged.connect(
self.selectionChanged)
self.invalidSelection[str].connect(self.module_logger)
def set_sample_model(self, sample_model):
self.sample_model.set_model(sample_model)
self.sample_model.setParent(self)
self.listView.setModel(self.sample_model)
def module_logger(self, message):
print('[layerselector]: ' + message)
def addClicked(self):
selected = self.listView.selectionModel().selectedRows()
inds = sorted([s.row() for s in selected])
selected = []
if inds:
for i, j in enumerate(inds):
if j in [0, self.sample_model.rowCount() - 1]:
self.invalidSelection.emit(
'Cannot add another substrate or incoming media')
else:
self.sample_model.addItem(
copy.deepcopy(self.sample_model.layers[i - 1 + j]),
i + j)
selected.append(i + j)
else:
self.sample_model.addItem(Layer())
self.sampleModelChanged.emit(self.sample_model)
self.listView.selectionModel().clear()
for selection in selected:
self.listView.selectionModel().select(
self.sample_model.index(selection),
QtCore.QItemSelectionModel.Select)
def delClicked(self):
inds = sorted(
[s.row() for s in self.listView.selectionModel().selectedRows()],
reverse=True)
if inds:
if inds in [[0], [self.sample_model.rowCount() - 1]]:
self.invalidSelection.emit(
'Cannot delete substrate or incoming media')
else:
for i in inds:
self.sample_model.delItem(i - 1)
else:
self.invalidSelection.emit("nothing selected")
self.sampleModelChanged.emit(self.sample_model)
def selectionEntered(self):
selection_string = str(self.select_lineEdit.text())
slice_parts = selection_string.split(':')
if len(slice_parts) > 3:
self.invalidSelection.emit(selection_string +
" has more than three arguments")
return
if len(slice_parts) == 1:
try:
index = int(slice_parts[0])
selection_slice = slice(index, index + 1)
except ValueError:
self.invalidSelection.emit(selection_string +
' cannot be converted to a slice')
return
else:
try:
s = [int(x) if x.strip() != '' else None for x in slice_parts]
try: # first layer is 1
s[0] -= 1
except TypeError:
pass
selection_slice = slice(*s)
except ValueError:
self.invalidSelection.emit(selection_string +
' cannot be converted to a slice')
return
self.listView.selectionModel().clear()
all_layers = range(1, self.sample_model.rowCount() - 1)
new_selection = all_layers[selection_slice]
for i in new_selection:
layer_index = self.sample_model.index(i)
self.listView.selectionModel().select(
layer_index, QtCore.QItemSelectionModel.Select)
def selectionChanged(self, selected, deselected):
incoming_media_index = self.sample_model.index(0)
substrate_index = self.sample_model.index(
self.sample_model.rowCount() - 1)
all_rows = self.listView.selectionModel().selectedRows()
if (substrate_index in all_rows) and len(all_rows) > 1:
if substrate_index in selected:
if len(selected) > 1:
self.listView.selectionModel().select(
substrate_index, QtCore.QItemSelectionModel.Deselect)
else:
self.listView.selectionModel().clear()
self.listView.selectionModel().select(
substrate_index, QtCore.QItemSelectionModel.Select)
else:
self.listView.selectionModel().select(
substrate_index, QtCore.QItemSelectionModel.Deselect)
if (incoming_media_index in all_rows) and len(all_rows) > 1:
if incoming_media_index in selected:
if len(selected) > 1:
self.listView.selectionModel().select(
incoming_media_index,
QtCore.QItemSelectionModel.Deselect)
else:
self.listView.selectionModel().clear()
self.listView.selectionModel().select(
incoming_media_index,
QtCore.QItemSelectionModel.Select)
else:
self.listView.selectionModel().select(
incoming_media_index, QtCore.QItemSelectionModel.Deselect)
substrate_index, QtCore.QItemSelectionModel.Select)
else:
self.listView.selectionModel().select(
substrate_index, QtCore.QItemSelectionModel.Deselect)
if (incoming_media_index in all_rows) and len(all_rows) > 1:
if incoming_media_index in selected:
if len(selected) > 1:
self.listView.selectionModel().select(
incoming_media_index,
QtCore.QItemSelectionModel.Deselect)
else:
self.listView.selectionModel().clear()
self.listView.selectionModel().select(
incoming_media_index,
QtCore.QItemSelectionModel.Select)
else:
self.listView.selectionModel().select(
incoming_media_index, QtCore.QItemSelectionModel.Deselect)
if __name__ == '__main__':
app = QtWidgets.QApplication(sys.argv)
sm = SampleModel()
sm.addItem(Layer(name='L0'))
sm.addItem(Layer(name='L1', thickness=2))
sm.addItem(Layer(name='L2'))
sm.addItem(Layer(name='L3'))
window = layerselector(sm)
window.show()
sys.exit(app.exec_())