def _initWidget(self):
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
# create the options
self.intersect = QCheckBox()
self.intersect.setChecked(True)
self.timeSelection = SimpleTimeDateSelection()
self.referenceLine = QComboBox()
# create the submit button
self.btnSubmit = QPushButton('Submit\nto .csv',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Write</b> output to .csv')
self.btnSubmit.setFixedSize(105, 50)
self.btnSubmit.setEnabled(False)
# create the output file name box
self.csvNameBox = QLineEdit()
self.csvNameBox.setReadOnly(True)
self.csvNameBox.setFixedHeight(30)
# create the widget displaying message logs
self.logTextBox = QPlainTextEditLogger(self)
self.logTextBox.setFormatter(
logging.Formatter('%(asctime)s - [%(levelname)s] - \n%(message)s'))
logging.getLogger().addHandler(self.logTextBox)
logging.getLogger().setLevel(self.parent.logging_level)
def _initWidgets(self):
# create a checkbox for operation selection
self.opBox = QGroupBox('Compute')
hlayout = QHBoxLayout()
self.maxButton = QRadioButton('Max')
self.minButton = QRadioButton('Min')
meanButton = QRadioButton('Mean')
hlayout.addWidget(self.maxButton)
hlayout.addWidget(self.minButton)
hlayout.addWidget(meanButton)
self.opBox.setLayout(hlayout)
self.opBox.setMaximumHeight(80)
self.opBox.setMaximumWidth(200)
self.maxButton.setChecked(True)
# create a slider for time selection
self.timeSlider = TimeRangeSlider()
self.timeSlider.setFixedHeight(30)
self.timeSlider.setMinimumWidth(600)
self.timeSlider.setEnabled(False)
# create text boxes for displaying the time selection
self.timeSelection = TimeSelection(self)
self.timeSelection.startIndex.setEnabled(False)
self.timeSelection.endIndex.setEnabled(False)
self.timeSelection.startValue.setEnabled(False)
self.timeSelection.endValue.setEnabled(False)
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
self.secondTable.setMinimumHeight(300)
# create the widget displaying message logs
self.logTextBox = QPlainTextEditLogger(self)
self.logTextBox.setFormatter(
logging.Formatter('%(asctime)s - [%(levelname)s] - \n%(message)s'))
logging.getLogger().addHandler(self.logTextBox)
logging.getLogger().setLevel(self.parent.logging_level)
# create a check box for output file format (simple or double precision)
self.singlePrecisionBox = QCheckBox(
'Convert to SERAFIN \n(single precision)', self)
self.singlePrecisionBox.setEnabled(False)
# create the submit button
self.btnSubmit = QPushButton('Submit',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Submit</b> to write a Serafin output')
self.btnSubmit.setFixedSize(105, 50)
self.btnSubmit.setEnabled(False)
def _initWidgets(self):
# create the button open points
self.btnOpenPoints = QPushButton('Load\nPoints',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogOpenButton))
self.btnOpenPoints.setToolTip('<b>Open</b> a .shp file')
self.btnOpenPoints.setFixedSize(105, 50)
self.btnOpenPoints.setEnabled(False)
self.btnOpenAttributes = QPushButton('Attributes\nTable',
self,
icon=self.style().standardIcon(
QStyle.SP_FileDialogListView))
self.btnOpenAttributes.setToolTip('<b>Open</b> the attribute table')
self.btnOpenAttributes.setFixedSize(105, 50)
self.btnOpenAttributes.setEnabled(False)
# create some text fields displaying the IO files info
self.pointsNameBox = QPlainTextEdit()
self.pointsNameBox.setReadOnly(True)
self.pointsNameBox.setFixedHeight(50)
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
self.timeSampling = QLineEdit('1')
self.timeSampling.setFixedWidth(50)
# create the map button
self.btnMap = QPushButton('Locate points\non map',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogHelpButton))
self.btnMap.setFixedSize(135, 50)
self.btnMap.setEnabled(False)
# create the submit button
self.btnSubmit = QPushButton('Submit\nto .csv',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Write</b> output to .csv')
self.btnSubmit.setFixedSize(105, 50)
self.btnSubmit.setEnabled(False)
# create the output file name box
self.csvNameBox = QLineEdit()
self.csvNameBox.setReadOnly(True)
self.csvNameBox.setFixedHeight(30)
def _initWidgets(self):
"""!
@brief (Used in __init__) Create widgets
"""
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
self.secondTable.setMinimumHeight(300)
# create a button for interpreting W from user-defined friction law
self.btnAddUS = QPushButton('Add US from friction law', self)
self.btnAddUS.setToolTip('Compute <b>US</b> based on a friction law')
self.btnAddUS.setEnabled(False)
self.btnAddUS.setFixedWidth(200)
# create a button for adding Rouse number from user-defined settling velocity
self.btnAddWs = QPushButton('Add Rouse from settling velocity', self)
self.btnAddWs.setToolTip(
'Compute <b>Rouse</b> for specific settling velocity')
self.btnAddWs.setEnabled(False)
self.btnAddWs.setFixedWidth(200)
def _initWidgets(self):
# create a text field for mesh intersection info display
self.infoBox = QPlainTextEdit()
self.infoBox.setFixedHeight(60)
self.infoBox.setReadOnly(True)
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
# create combo box widgets for choosing the operation
self.operationBox = QComboBox()
self.operationBox.setFixedSize(400, 30)
for op in [
'Project B on A', 'B - A', 'A - B', 'max(A, B)', 'min(A, B)'
]:
self.operationBox.addItem(op)
# create the widget displaying message logs
self.logTextBox = QPlainTextEditLogger(self)
self.logTextBox.setFormatter(
logging.Formatter('%(asctime)s - [%(levelname)s] - \n%(message)s'))
logging.getLogger().addHandler(self.logTextBox)
logging.getLogger().setLevel(self.parent.logging_level)
# create a check box for output file format (simple or double precision)
self.singlePrecisionBox = QCheckBox(
'Convert to SERAFIN \n(single precision)', self)
self.singlePrecisionBox.setEnabled(False)
# create the submit button
self.btnSubmit = QPushButton('Submit',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Submit</b> to write a Serafin output')
self.btnSubmit.setFixedSize(105, 50)
class SubmitTab(QWidget):
def __init__(self, inputTab, parent):
super().__init__()
self.input = inputTab
self.parent = parent
self._initWidgets()
self._bindEvents()
self._setLayout()
def _initWidgets(self):
# create a text field for mesh intersection info display
self.infoBox = QPlainTextEdit()
self.infoBox.setFixedHeight(60)
self.infoBox.setReadOnly(True)
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
# create combo box widgets for choosing the operation
self.operationBox = QComboBox()
self.operationBox.setFixedSize(400, 30)
for op in [
'Project B on A', 'B - A', 'A - B', 'max(A, B)', 'min(A, B)'
]:
self.operationBox.addItem(op)
# create the widget displaying message logs
self.logTextBox = QPlainTextEditLogger(self)
self.logTextBox.setFormatter(
logging.Formatter('%(asctime)s - [%(levelname)s] - \n%(message)s'))
logging.getLogger().addHandler(self.logTextBox)
logging.getLogger().setLevel(self.parent.logging_level)
# create a check box for output file format (simple or double precision)
self.singlePrecisionBox = QCheckBox(
'Convert to SERAFIN \n(single precision)', self)
self.singlePrecisionBox.setEnabled(False)
# create the submit button
self.btnSubmit = QPushButton('Submit',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Submit</b> to write a Serafin output')
self.btnSubmit.setFixedSize(105, 50)
def _bindEvents(self):
self.btnSubmit.clicked.connect(self.btnSubmitEvent)
def _setLayout(self):
mainLayout = QVBoxLayout()
mainLayout.addItem(QSpacerItem(1, 10))
mainLayout.addWidget(self.infoBox)
mainLayout.addItem(QSpacerItem(1, 15))
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
vlayout = QVBoxLayout()
lb = QLabel('Available variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.firstTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(15, 1))
vlayout = QVBoxLayout()
lb = QLabel('Output variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.secondTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(30, 1))
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(50, 20))
glayout = QGridLayout()
glayout.addWidget(QLabel(' Select an operation'), 1, 1)
glayout.addWidget(self.operationBox, 1, 2)
glayout.setSpacing(10)
mainLayout.addLayout(glayout)
mainLayout.setAlignment(glayout, Qt.AlignTop | Qt.AlignLeft)
mainLayout.addItem(QSpacerItem(50, 20))
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(50, 10))
hlayout.addWidget(self.btnSubmit)
hlayout.addItem(QSpacerItem(50, 10))
hlayout.addWidget(self.singlePrecisionBox)
hlayout.addItem(QSpacerItem(50, 10))
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(30, 15))
mainLayout.addWidget(QLabel(' Message logs'))
mainLayout.addWidget(self.logTextBox.widget)
self.setLayout(mainLayout)
def _initVarTables(self, common_vars):
for i, (var_ID, var_name, var_unit) in enumerate(common_vars):
self.firstTable.insertRow(self.firstTable.rowCount())
id_item = QTableWidgetItem(var_ID.strip())
name_item = QTableWidgetItem(var_name.decode('utf-8').strip())
unit_item = QTableWidgetItem(var_unit.decode('utf-8').strip())
self.firstTable.setItem(i, 0, id_item)
self.firstTable.setItem(i, 1, name_item)
self.firstTable.setItem(i, 2, unit_item)
def _getSelectedVariables(self):
return self.secondTable.get_selected_all()
def _getOutputHeader(self):
selected_vars = self._getSelectedVariables()
output_header = self.input.first_data.header.copy()
output_header.nb_var = len(selected_vars)
output_header.var_IDs, output_header.var_names, output_header.var_units = [], [], []
for var_ID, var_name, var_unit in selected_vars:
output_header.var_IDs.append(var_ID)
output_header.var_names.append(var_name)
output_header.var_units.append(var_unit)
if self.singlePrecisionBox.isChecked():
output_header.to_single_precision()
return output_header
def _updateInfo(self):
self.infoBox.clear()
self.infoBox.appendPlainText(
'The two files has {} common variables and {} common frames.\n'
'The mesh A has {} / {} nodes inside the mesh B.'.format(
self.firstTable.rowCount() + self.secondTable.rowCount(),
len(self.input.common_frames), sum(self.input.is_inside),
self.input.first_data.header.nb_nodes))
def reset(self):
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
self.singlePrecisionBox.setChecked(False)
self.singlePrecisionBox.setEnabled(False)
def resetFirst(self):
common_vars = [
(var_ID, var_name, var_unit) for var_ID, var_name, var_unit in zip(
self.input.first_data.header.var_IDs, self.input.first_data.
header.var_names, self.input.first_data.header.var_units)
if var_ID in self.input.second_data.header.var_IDs
]
if not common_vars:
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
return False
else:
# recover, if possible, old variable selection
old_selected = self._getSelectedVariables()
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
self._initVarTables([(var_ID, var_name, var_unit)
for var_ID, var_name, var_unit in common_vars
if var_ID not in old_selected])
for var_ID, var_name, var_unit in common_vars:
if var_ID in old_selected:
row = self.secondTable.rowCount()
self.secondTable.insertRow(row)
id_item = QTableWidgetItem(var_ID.strip())
name_item = QTableWidgetItem(
var_name.decode('utf-8').strip())
unit_item = QTableWidgetItem(
var_unit.decode('utf-8').strip())
self.secondTable.setItem(row, 0, id_item)
self.secondTable.setItem(row, 1, name_item)
self.secondTable.setItem(row, 2, unit_item)
return True
def getSecond(self, old_second, common_vars):
if not old_second:
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
self._initVarTables(common_vars)
if self.input.first_data.header.is_double_precision():
self.singlePrecisionBox.setEnabled(True)
self._updateInfo()
def btnSubmitEvent(self):
if self.secondTable.rowCount() == 0:
QMessageBox.critical(
self, 'Error',
'Choose at least one output variable before submit!',
QMessageBox.Ok)
return
canceled, filename = save_dialog('Serafin',
input_names=[
self.input.first_data.filename,
self.input.second_data.filename
])
if canceled:
return
# deduce header from selected variable IDs
output_header = self._getOutputHeader()
time_indices = self.input.common_frames
operation_type = {
0: operations.PROJECT,
1: operations.DIFF,
2: operations.REV_DIFF,
3: operations.MAX_BETWEEN,
4: operations.MIN_BETWEEN
}[self.operationBox.currentIndex()]
self.parent.inDialog()
progressBar = OutputProgressDialog()
# do some calculations
with Serafin.Read(self.input.first_data.filename,
self.input.first_data.language) as first_in:
first_in.header = self.input.first_data.header
first_in.time = self.input.first_data.time
with Serafin.Read(self.input.second_data.filename,
self.input.second_data.language) as second_in:
second_in.header = self.input.second_data.header
second_in.time = self.input.second_data.time
progressBar.setValue(5)
QApplication.processEvents()
with Serafin.Write(
filename,
self.input.first_data.language) as out_stream:
out_stream.write_header(output_header)
process = ProjectMeshThread(first_in, second_in,
out_stream, output_header,
self.input.is_inside,
self.input.point_interpolators,
time_indices, operation_type)
progressBar.connectToThread(process)
process.run()
if not process.canceled:
progressBar.outputFinished()
progressBar.exec_()
self.parent.outDialog()
class SynchMaxTab(QWidget):
def __init__(self, parent):
super().__init__()
self.data = None
self.parent = parent
self._initWidgets()
self._setLayout()
self._bindEvents()
def _initWidgets(self):
# create a combox box for variable selection
self.varBox = QComboBox()
self.varBox.setFixedSize(200, 30)
# create a slider for time selection
self.timeSlider = TimeRangeSlider()
self.timeSlider.setFixedHeight(30)
self.timeSlider.setMinimumWidth(600)
self.timeSlider.setEnabled(False)
# create text boxes for displaying the time selection
self.timeSelection = TimeSelection(self)
self.timeSelection.startIndex.setEnabled(False)
self.timeSelection.endIndex.setEnabled(False)
self.timeSelection.startValue.setEnabled(False)
self.timeSelection.endValue.setEnabled(False)
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
self.secondTable.setMinimumHeight(300)
# create the widget displaying message logs
self.logTextBox = QPlainTextEditLogger(self)
self.logTextBox.setFormatter(
logging.Formatter('%(asctime)s - [%(levelname)s] - \n%(message)s'))
logging.getLogger().addHandler(self.logTextBox)
logging.getLogger().setLevel(self.parent.logging_level)
# create a check box for output file format (simple or double precision)
self.singlePrecisionBox = QCheckBox(
'Convert to SERAFIN \n(single precision)', self)
self.singlePrecisionBox.setEnabled(False)
# create the submit button
self.btnSubmit = QPushButton('Submit',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Submit</b> to write a Serafin output')
self.btnSubmit.setFixedSize(105, 50)
self.btnSubmit.setEnabled(False)
def _setLayout(self):
mainLayout = QVBoxLayout()
mainLayout.addItem(QSpacerItem(1, 10))
mainLayout.addWidget(self.timeSlider)
mainLayout.addWidget(self.timeSelection)
mainLayout.addItem(QSpacerItem(1, 10))
hlayout = QHBoxLayout()
hlayout.addWidget(QLabel(' Select the reference variable '))
hlayout.addWidget(self.varBox, Qt.AlignLeft)
hlayout.addStretch()
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(1, 10))
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
vlayout = QVBoxLayout()
vlayout.setAlignment(Qt.AlignHCenter)
lb = QLabel('Available variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.firstTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(15, 1))
vlayout = QVBoxLayout()
lb = QLabel('Output variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.secondTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(30, 1))
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(50, 20))
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(50, 10))
hlayout.addWidget(self.btnSubmit)
hlayout.addItem(QSpacerItem(50, 10))
hlayout.addWidget(self.singlePrecisionBox)
hlayout.addItem(QSpacerItem(50, 10))
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(30, 15))
mainLayout.addWidget(QLabel(' Message logs'))
mainLayout.addWidget(self.logTextBox.widget)
self.setLayout(mainLayout)
def _bindEvents(self):
self.btnSubmit.clicked.connect(self.btnSubmitEvent)
self.timeSelection.startIndex.editingFinished.connect(
self.timeSlider.enterIndexEvent)
self.timeSelection.endIndex.editingFinished.connect(
self.timeSlider.enterIndexEvent)
self.timeSelection.startValue.editingFinished.connect(
self.timeSlider.enterValueEvent)
self.timeSelection.endValue.editingFinished.connect(
self.timeSlider.enterValueEvent)
def _initVarTables(self):
self.firstTable.fill(self.data.header)
self.secondTable.insertRow(0)
self.secondTable.setItem(0, 0, QTableWidgetItem('MAX TIME'))
self.secondTable.setItem(0, 1, QTableWidgetItem('MAX TIME'))
self.secondTable.setItem(0, 2, QTableWidgetItem('S'))
for j in range(3):
self.secondTable.item(0, j).setFlags(Qt.NoItemFlags)
def _getSelectedVariables(self):
return self.secondTable.get_selected_all()
def reset(self):
self.varBox.clear()
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
self.timeSelection.disable()
self.btnSubmit.setEnabled(False)
self.singlePrecisionBox.setChecked(False)
self.singlePrecisionBox.setEnabled(False)
def getOutputHeader(self, selected_vars):
output_header = self.data.header.copy()
output_header.nb_var = len(selected_vars)
output_header.var_IDs, output_header.var_names, output_header.var_units = [], [], []
for var_ID, var_name, var_unit in selected_vars:
output_header.var_IDs.append(var_ID)
output_header.var_names.append(var_name)
output_header.var_units.append(var_unit)
if self.singlePrecisionBox.isChecked():
output_header.to_single_precision()
return output_header
def getInput(self, data):
self.data = data
self._initVarTables()
for var, var_name in zip(self.data.header.var_IDs,
self.data.header.var_names):
item = '%s (%s)' % (var, var_name.decode('utf-8').strip())
self.varBox.addItem(item)
self.btnSubmit.setEnabled(True)
# unlock convert to single precision
if self.data.header.is_double_precision():
self.singlePrecisionBox.setEnabled(True)
if self.data.header.date is not None:
year, month, day, hour, minute, second = self.data.header.date
start_time = datetime.datetime(year, month, day, hour, minute,
second)
else:
start_time = datetime.datetime(1900, 1, 1, 0, 0, 0)
time_frames = list(
map(lambda x: datetime.timedelta(seconds=x), self.data.time))
self.timeSlider.reinit(start_time, time_frames, self.timeSelection)
self.timeSlider.setEnabled(True)
self.timeSelection.enable()
def btnSubmitEvent(self):
# fetch the list of selected variables
selected_vars = self._getSelectedVariables()
if not selected_vars:
QMessageBox.critical(self, 'Error',
'Select at least one variable.',
QMessageBox.Ok)
return
canceled, filename = save_dialog('Serafin', self.data.filename)
if canceled:
return
# deduce header from selected variable IDs and write header
output_header = self.getOutputHeader(selected_vars)
start_index = int(self.timeSelection.startIndex.text()) - 1
end_index = int(self.timeSelection.endIndex.text())
time_indices = list(range(start_index, end_index))
var = self.varBox.currentText().split(' (')[0]
output_message = 'Computing SynchMax of variables %s between frame %d and %d.' \
% (str(list(map(lambda x: x[0], selected_vars[1:]))), start_index+1, end_index)
self.parent.inDialog()
logging.info(output_message)
progressBar = OutputProgressDialog()
# do some calculations
with Serafin.Read(self.data.filename,
self.data.language) as input_stream:
input_stream.header = self.data.header
input_stream.time = self.data.time
progressBar.setValue(5)
QApplication.processEvents()
with Serafin.Write(filename, self.data.language) as output_stream:
process = SynchMaxThread(input_stream, selected_vars[1:],
time_indices, var)
progressBar.connectToThread(process)
values = process.run()
if not process.canceled:
output_stream.write_header(output_header)
output_stream.write_entire_frame(output_header,
self.data.time[0], values)
progressBar.outputFinished()
progressBar.exec_()
self.parent.outDialog()
class CSVTab(QWidget):
def __init__(self, inputTab, parent):
super().__init__()
self.input = inputTab
self.parent = parent
self._initWidget()
self._setLayout()
self.btnSubmit.clicked.connect(self.btnSubmitEvent)
def _initWidget(self):
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
# create the options
self.intersect = QCheckBox()
self.intersect.setChecked(True)
self.timeSelection = SimpleTimeDateSelection()
self.referenceLine = QComboBox()
# create the submit button
self.btnSubmit = QPushButton('Submit\nto .csv',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Write</b> output to .csv')
self.btnSubmit.setFixedSize(105, 50)
self.btnSubmit.setEnabled(False)
# create the output file name box
self.csvNameBox = QLineEdit()
self.csvNameBox.setReadOnly(True)
self.csvNameBox.setFixedHeight(30)
# create the widget displaying message logs
self.logTextBox = QPlainTextEditLogger(self)
self.logTextBox.setFormatter(
logging.Formatter('%(asctime)s - [%(levelname)s] - \n%(message)s'))
logging.getLogger().addHandler(self.logTextBox)
logging.getLogger().setLevel(self.parent.logging_level)
def _setLayout(self):
mainLayout = QVBoxLayout()
mainLayout.setSpacing(15)
mainLayout.addItem(QSpacerItem(10, 10))
mainLayout.addWidget(self.timeSelection)
hlayout = QHBoxLayout()
hlayout.addWidget(QLabel(' Reference line'))
hlayout.addWidget(self.referenceLine)
hlayout.setAlignment(self.referenceLine, Qt.AlignLeft)
hlayout.addStretch()
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(10, 10))
glayout = QGridLayout()
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
vlayout = QVBoxLayout()
lb = QLabel('Available variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.firstTable)
hlayout2 = QHBoxLayout()
hlayout2.addItem(QSpacerItem(30, 1))
hlayout2.addWidget(self.intersect)
hlayout2.addWidget(QLabel('Add intersection points'))
hlayout2.setAlignment(self.intersect, Qt.AlignLeft)
hlayout2.addStretch()
vlayout.addLayout(hlayout2)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(15, 1))
vlayout = QVBoxLayout()
lb = QLabel('Output variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.secondTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(30, 1))
glayout.addLayout(hlayout, 1, 1)
glayout.setAlignment(Qt.AlignLeft)
glayout.setSpacing(10)
mainLayout.addLayout(glayout)
mainLayout.addItem(QSpacerItem(30, 10))
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
hlayout.addWidget(self.btnSubmit)
hlayout.addWidget(self.csvNameBox)
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(30, 15))
mainLayout.addWidget(QLabel(' Message logs'))
mainLayout.addWidget(self.logTextBox.widget)
self.setLayout(mainLayout)
def getSelectedVariables(self):
return self.secondTable.get_selected()
def _initVarTables(self):
self.firstTable.fill(self.input.data.header)
def getInput(self):
self._initVarTables()
self.btnSubmit.setEnabled(True)
self.csvNameBox.clear()
if self.input.data.header.date is not None:
year, month, day, hour, minute, second = self.input.data.header.date
start_time = datetime.datetime(year, month, day, hour, minute,
second)
else:
start_time = datetime.datetime(1900, 1, 1, 0, 0, 0)
frames = list(
map(lambda x: start_time + datetime.timedelta(seconds=x),
self.input.data.time))
self.timeSelection.initTime(self.input.data.time, frames)
for i in range(len(self.input.lines)):
id_line = str(i + 1)
if self.input.line_interpolators[i][0]:
self.referenceLine.addItem('Line %s' % id_line)
def reset(self):
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
self.intersect.setChecked(False)
self.btnSubmit.setEnabled(False)
self.timeSelection.clearText()
self.csvNameBox.clear()
self.referenceLine.clear()
self.intersect.setChecked(True)
def btnSubmitEvent(self):
selected_var_IDs = self.getSelectedVariables()
if not selected_var_IDs:
QMessageBox.critical(
self, 'Error',
'Choose at least one output variable before submit!',
QMessageBox.Ok)
return
canceled, filename = save_dialog('CSV')
if canceled:
return
self.csvNameBox.setText(filename)
logging.info('Writing the output to %s' % filename)
self.parent.inDialog()
indices_nonempty = [
i for i in range(len(self.input.lines))
if self.input.line_interpolators[i][0]
]
reference = self.input.lines[
int(self.referenceLine.currentText().split()[1]) - 1]
time_index = int(self.timeSelection.index.text()) - 1
# initialize the progress bar
process = WriteCSVProcess(self.parent.csv_separator,
self.parent.digits, self.input.mesh)
progressBar = OutputProgressDialog()
with Serafin.Read(self.input.data.filename,
self.input.data.language) as input_stream:
input_stream.header = self.input.data.header
input_stream.time = self.input.data.time
progressBar.setValue(1)
QApplication.processEvents()
with open(filename, 'w') as output_stream:
progressBar.connectToThread(process)
if self.intersect.isChecked():
process.write_csv(input_stream, selected_var_IDs,
output_stream,
self.input.line_interpolators,
indices_nonempty, reference, time_index)
else:
process.write_csv(input_stream, selected_var_IDs,
output_stream,
self.input.line_interpolators_internal,
indices_nonempty, reference, time_index)
if not process.canceled:
progressBar.outputFinished()
progressBar.exec_()
self.parent.outDialog()
if process.canceled:
self.csvNameBox.clear()
return
class InputTab(SerafinInputTab):
def __init__(self, parent):
super().__init__(parent)
self.old_frequency = '1'
canvas = MapCanvas()
self.map = MapViewer(canvas)
self.has_map = False
self.data = None
self.mesh = None
self.points = []
self.point_interpolators = []
self.fields = []
self.attributes = []
self.attribute_table = PointAttributeTable()
self._initWidgets() # some instance attributes will be set there
self._setLayout()
self._bindEvents()
def _initWidgets(self):
# create the button open points
self.btnOpenPoints = QPushButton('Load\nPoints',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogOpenButton))
self.btnOpenPoints.setToolTip('<b>Open</b> a .shp file')
self.btnOpenPoints.setFixedSize(105, 50)
self.btnOpenPoints.setEnabled(False)
self.btnOpenAttributes = QPushButton('Attributes\nTable',
self,
icon=self.style().standardIcon(
QStyle.SP_FileDialogListView))
self.btnOpenAttributes.setToolTip('<b>Open</b> the attribute table')
self.btnOpenAttributes.setFixedSize(105, 50)
self.btnOpenAttributes.setEnabled(False)
# create some text fields displaying the IO files info
self.pointsNameBox = QPlainTextEdit()
self.pointsNameBox.setReadOnly(True)
self.pointsNameBox.setFixedHeight(50)
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
self.timeSampling = QLineEdit('1')
self.timeSampling.setFixedWidth(50)
# create the map button
self.btnMap = QPushButton('Locate points\non map',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogHelpButton))
self.btnMap.setFixedSize(135, 50)
self.btnMap.setEnabled(False)
# create the submit button
self.btnSubmit = QPushButton('Submit\nto .csv',
self,
icon=self.style().standardIcon(
QStyle.SP_DialogSaveButton))
self.btnSubmit.setToolTip('<b>Write</b> output to .csv')
self.btnSubmit.setFixedSize(105, 50)
self.btnSubmit.setEnabled(False)
# create the output file name box
self.csvNameBox = QLineEdit()
self.csvNameBox.setReadOnly(True)
self.csvNameBox.setFixedHeight(30)
def _bindEvents(self):
self.btnOpen.clicked.connect(self.btnOpenSerafinEvent)
self.btnOpenPoints.clicked.connect(self.btnOpenPointsEvent)
self.btnOpenAttributes.clicked.connect(self.btnOpenAttributesEvent)
self.btnMap.clicked.connect(self.btnMapEvent)
self.btnSubmit.clicked.connect(self.btnSubmitEvent)
self.timeSampling.editingFinished.connect(self._checkSamplingFrequency)
def _setLayout(self):
mainLayout = QVBoxLayout()
mainLayout.addItem(QSpacerItem(10, 10))
mainLayout.setSpacing(15)
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
hlayout.setAlignment(Qt.AlignLeft)
hlayout.addWidget(self.btnOpen)
hlayout.addItem(QSpacerItem(30, 1))
hlayout.addWidget(self.langBox)
hlayout.addItem(QSpacerItem(30, 1))
hlayout.addWidget(self.btnOpenPoints)
hlayout.addWidget(self.btnOpenAttributes)
hlayout.addWidget(self.btnMap)
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(10, 10))
glayout = QGridLayout()
glayout.addWidget(QLabel(' Input file'), 1, 1)
glayout.addWidget(self.inNameBox, 1, 2)
glayout.addWidget(QLabel(' Summary'), 2, 1)
glayout.addWidget(self.summaryTextBox, 2, 2)
glayout.addWidget(QLabel(' Points file'), 3, 1)
glayout.addWidget(self.pointsNameBox, 3, 2)
glayout.setAlignment(Qt.AlignLeft)
glayout.setSpacing(10)
mainLayout.addLayout(glayout)
mainLayout.addItem(QSpacerItem(10, 10))
glayout = QGridLayout()
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
vlayout = QVBoxLayout()
lb = QLabel('Available variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.firstTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(15, 1))
vlayout = QVBoxLayout()
lb = QLabel('Output variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.secondTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(30, 1))
glayout.addLayout(hlayout, 1, 1)
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
hlayout.addWidget(QLabel('Time sampling frequency'))
hlayout.addWidget(self.timeSampling)
hlayout.setAlignment(self.timeSampling, Qt.AlignLeft)
hlayout.addStretch()
glayout.addLayout(hlayout, 2, 1)
glayout.setAlignment(Qt.AlignLeft)
glayout.setSpacing(10)
mainLayout.addLayout(glayout)
mainLayout.addItem(QSpacerItem(30, 10))
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
hlayout.addWidget(self.btnSubmit)
hlayout.addWidget(self.csvNameBox)
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(30, 15))
mainLayout.addWidget(QLabel(' Message logs'))
mainLayout.addWidget(self.logTextBox.widget)
self.setLayout(mainLayout)
def _reinitInput(self):
self.reset()
self.data = None
self.has_map = False
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
self.btnMap.setEnabled(False)
self.btnOpenAttributes.setEnabled(False)
self.mesh = None
self.btnOpenPoints.setEnabled(False)
self.old_frequency = self.timeSampling.text()
self.timeSampling.setText('1')
self.btnSubmit.setEnabled(False)
self.csvNameBox.clear()
self.parent.tab.setTabEnabled(1, False)
def _resetDefaultOptions(self):
if int(self.old_frequency) <= len(self.data.time):
self.timeSampling.setText(self.old_frequency)
is_inside, self.point_interpolators = self.mesh.get_point_interpolators(
self.points)
nb_inside = sum(map(int, is_inside))
if nb_inside == 0:
self.pointsNameBox.clear()
self.points = []
self.point_interpolators = []
else:
self.attribute_table.getData(self.points, is_inside, self.fields,
self.attributes)
old_filename = self.pointsNameBox.toPlainText().split('\n')[0]
self.pointsNameBox.clear()
self.pointsNameBox.appendPlainText(
old_filename + '\n' + 'The file contains {} point{}.'
'{} point{} inside the mesh.'.format(
len(self.points), 's' if len(self.points) > 1 else '',
nb_inside, 's are' if nb_inside > 1 else ' is'))
self.btnSubmit.setEnabled(True)
self.btnOpenAttributes.setEnabled(True)
self.btnMap.setEnabled(True)
def _initVarTables(self):
self.firstTable.fill(self.data.header)
def _checkSamplingFrequency(self):
try:
sampling_frequency = int(self.timeSampling.text())
except ValueError:
QMessageBox.critical(self, 'Error',
'The sampling frequency must be a number!',
QMessageBox.Ok)
self.timeSampling.setText('1')
return
if sampling_frequency < 1 or sampling_frequency > len(self.data.time):
QMessageBox.critical(
self, 'Error',
'The sampling frequency must be in the range [1; nbFrames]!',
QMessageBox.Ok)
self.timeSampling.setText('1')
return
def getSelectedVariables(self):
return self.secondTable.get_selected()
def btnOpenSerafinEvent(self):
canceled, filename = super().open_event()
if canceled:
return
self._reinitInput()
success, data = self.read_2d(filename)
if not success:
return
# record the mesh for future visualization and calculations
self.parent.inDialog()
meshLoader = LoadMeshDialog('interpolation', data.header)
self.mesh = meshLoader.run()
self.parent.outDialog()
if meshLoader.thread.canceled:
self.pointsNameBox.clear()
self.summaryTextBox.clear()
return
self.data = data
self.btnOpenPoints.setEnabled(True)
self._resetDefaultOptions()
self.parent.imageTab.reset()
# displaying the available variables
self._initVarTables()
def btnOpenPointsEvent(self):
success, filename, points, attributes, fields = open_points()
if not success:
return
logging.info('Finished reading the points file %s' % filename)
is_inside, point_interpolators = self.mesh.get_point_interpolators(
points)
nb_inside = sum(map(int, is_inside))
if nb_inside == 0:
QMessageBox.critical(self, 'Error', 'No point inside the mesh.',
QMessageBox.Ok)
return
self.points = points
self.attributes = attributes
self.fields = fields
self.attribute_table.getData(self.points, is_inside, self.fields,
self.attributes)
self.point_interpolators = point_interpolators
self.pointsNameBox.clear()
self.pointsNameBox.appendPlainText(
filename + '\n' + 'The file contains {} point{}.'
'{} point{} inside the mesh.'.format(
len(self.points), 's' if len(self.points) > 1 else '',
nb_inside, 's are' if nb_inside > 1 else ' is'))
self.has_map = False
self.btnMap.setEnabled(True)
self.btnOpenAttributes.setEnabled(True)
self.btnSubmit.setEnabled(True)
self.csvNameBox.clear()
self.parent.imageTab.reset()
self.parent.tab.setTabEnabled(1, False)
def btnOpenAttributesEvent(self):
self.attribute_table.show()
def btnMapEvent(self):
if not self.has_map:
self.map.canvas.initFigure(self.mesh)
self.map.canvas.axes.scatter(*zip(*self.points))
labels = ['%d' % (i + 1) for i in range(len(self.points))]
for label, (x, y) in zip(labels, self.points):
self.map.canvas.axes.annotate(
label,
xy=(x, y),
xytext=(-20, 20),
fontsize=8,
textcoords='offset points',
ha='right',
va='bottom',
bbox=dict(boxstyle='round,pad=0.5', fc='yellow',
alpha=0.5),
arrowprops=dict(arrowstyle='->',
connectionstyle='arc3,rad=0'))
self.map.canvas.draw()
self.has_map = True
self.map.show()
def btnSubmitEvent(self):
selected_var_IDs = self.getSelectedVariables()
if not selected_var_IDs:
QMessageBox.critical(
self, 'Error',
'Choose at least one output variable before submit!',
QMessageBox.Ok)
return
canceled, filename = save_dialog('CSV')
if canceled:
return
self.csvNameBox.setText(filename)
logging.info('Writing the output to %s' % filename)
self.parent.inDialog()
sampling_frequency = int(self.timeSampling.text())
selected_time = self.data.time[::sampling_frequency]
indices_inside = [
i for i in range(len(self.points))
if self.point_interpolators[i] is not None
]
# initialize the progress bar
process = WriteCSVProcess(self.parent.csv_separator,
self.parent.digits, self.mesh)
progressBar = OutputProgressDialog()
with Serafin.Read(self.data.filename,
self.data.language) as input_stream:
input_stream.header = self.data.header
input_stream.time = self.data.time
progressBar.setValue(1)
QApplication.processEvents()
with open(filename, 'w') as output_stream:
progressBar.connectToThread(process)
process.write_csv(
input_stream, selected_time, selected_var_IDs,
output_stream, indices_inside,
[self.points[i] for i in indices_inside],
[self.point_interpolators[i] for i in indices_inside])
if not process.canceled:
progressBar.outputFinished()
progressBar.exec_()
self.parent.outDialog()
if process.canceled:
self.csvNameBox.clear()
return
self.parent.imageTab.getData(selected_var_IDs, indices_inside)
self.parent.tab.setTabEnabled(1, True)
class InputTab(SerafinInputTab):
def __init__(self, parent):
super().__init__(parent)
self.data = None
self.time = []
self.available_vars = []
self.settling_velocities = []
self.YELLOW = QColor(245, 255, 207)
self.GREEN = QColor(200, 255, 180)
self.BLUE = QColor(200, 230, 250)
self._initWidgets() # some instance attributes will be set there
self._setLayout()
self._bindEvents()
def _initWidgets(self):
"""!
@brief (Used in __init__) Create widgets
"""
# create two 3-column tables for variables selection
self.firstTable = VariableTable()
self.secondTable = VariableTable()
self.secondTable.setMinimumHeight(300)
# create a button for interpreting W from user-defined friction law
self.btnAddUS = QPushButton('Add US from friction law', self)
self.btnAddUS.setToolTip('Compute <b>US</b> based on a friction law')
self.btnAddUS.setEnabled(False)
self.btnAddUS.setFixedWidth(200)
# create a button for adding Rouse number from user-defined settling velocity
self.btnAddWs = QPushButton('Add Rouse from settling velocity', self)
self.btnAddWs.setToolTip(
'Compute <b>Rouse</b> for specific settling velocity')
self.btnAddWs.setEnabled(False)
self.btnAddWs.setFixedWidth(200)
def _bindEvents(self):
"""!
@brief (Used in __init__) Bind events to widgets
"""
self.btnOpen.clicked.connect(self.btnOpenEvent)
self.btnAddUS.clicked.connect(self.btnAddUSEvent)
self.btnAddWs.clicked.connect(self.btnAddWsEvent)
def _setLayout(self):
"""
@brief: (Used in __init__) Set up layout
"""
mainLayout = QVBoxLayout()
mainLayout.addItem(QSpacerItem(10, 10))
mainLayout.setSpacing(15)
mainLayout.addLayout(self.input_layout)
mainLayout.addItem(QSpacerItem(1, 10))
hlayout = QHBoxLayout()
hlayout.addItem(QSpacerItem(30, 1))
vlayout = QVBoxLayout()
vlayout.setAlignment(Qt.AlignHCenter)
lb = QLabel('Available variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.firstTable)
vlayout.addItem(QSpacerItem(1, 5))
hlayout2 = QHBoxLayout()
hlayout2.addItem(QSpacerItem(30, 1))
hlayout2.addWidget(self.btnAddUS)
hlayout2.addItem(QSpacerItem(30, 1))
vlayout.addLayout(hlayout2)
hlayout2 = QHBoxLayout()
hlayout2.addItem(QSpacerItem(30, 1))
hlayout2.addWidget(self.btnAddWs)
hlayout2.addItem(QSpacerItem(30, 1))
vlayout.addLayout(hlayout2)
vlayout.addItem(QSpacerItem(1, 5))
vlayout.setAlignment(Qt.AlignLeft)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(15, 1))
vlayout = QVBoxLayout()
lb = QLabel('Output variables')
vlayout.addWidget(lb)
vlayout.setAlignment(lb, Qt.AlignHCenter)
vlayout.addWidget(self.secondTable)
hlayout.addLayout(vlayout)
hlayout.addItem(QSpacerItem(30, 1))
mainLayout.addLayout(hlayout)
mainLayout.addItem(QSpacerItem(10, 15))
mainLayout.addWidget(QLabel(' Message logs'))
mainLayout.addWidget(self.logTextBox.widget)
self.setLayout(mainLayout)
def _initVarTables(self):
"""!
@brief (Used in btnOpenEvent) Put available variables ID-name-unit in the table display
"""
# add original variables to the table
for var_ID, var_name, var_unit in zip(self.data.header.var_IDs,
self.data.header.var_names,
self.data.header.var_units):
if not self.data.header.is_2d and var_ID == 'Z':
self.secondTable.insertRow(self.secondTable.rowCount())
id_item = QTableWidgetItem(var_ID.strip())
name_item = QTableWidgetItem(var_name.decode('utf-8').strip())
unit_item = QTableWidgetItem(var_unit.decode('utf-8').strip())
for j, item in enumerate([id_item, name_item, unit_item]):
item.setFlags(Qt.NoItemFlags)
self.secondTable.setItem(0, j, item)
continue
row = self.firstTable.rowCount()
self.firstTable.insertRow(row)
id_item = QTableWidgetItem(var_ID.strip())
name_item = QTableWidgetItem(var_name.decode('utf-8').strip())
unit_item = QTableWidgetItem(var_unit.decode('utf-8').strip())
for j, item in enumerate([id_item, name_item, unit_item]):
self.firstTable.setItem(row, j, item)
# find computable variables (stored as slf.variables.Variable objects)
self.available_vars = get_available_variables(
self.data.header.var_IDs, is_2d=self.data.header.is_2d)
# add new variables to the table
for var in self.available_vars:
row = self.firstTable.rowCount()
self.firstTable.insertRow(row)
id_item = QTableWidgetItem(var.ID())
name_item = QTableWidgetItem(var.name(self.data.language))
unit_item = QTableWidgetItem(var.unit())
for j, item in enumerate([id_item, name_item, unit_item]):
self.firstTable.setItem(row, j, item)
self.firstTable.item(row, j).setBackground(
self.YELLOW) # set new variables colors to yellow
def _reinitInput(self):
"""!
@brief (Used in btnOpenEvent) Reinitialize input file data before reading a new file
"""
self.available_vars = []
self.data = None
self.settling_velocities = []
self.btnAddUS.setEnabled(False)
self.btnAddWs.setEnabled(False)
self.firstTable.setRowCount(0)
self.secondTable.setRowCount(0)
self.reset()
self.parent.reset()
def getSelectedVariables(self):
return self.secondTable.get_selected_all()
def btnAddUSEvent(self):
msg = FrictionLawMessage()
value = msg.exec_()
if value != QDialog.Accepted:
return
friction_law = msg.getChoice()
self.data.us_equation = get_US_equation(friction_law)
new_vars = new_variables_from_US(self.data.header.var_IDs)
# add US, TAU, DMAX and FROTP to available variables
offset = self.firstTable.rowCount()
for i in range(len(new_vars)):
var = new_vars[i]
self.firstTable.insertRow(self.firstTable.rowCount())
id_item = QTableWidgetItem(var.ID().strip())
name_item = QTableWidgetItem(var.name(self.data.language))
unit_item = QTableWidgetItem(var.unit())
self.firstTable.setItem(offset + i, 0, id_item)
self.firstTable.setItem(offset + i, 1, name_item)
self.firstTable.setItem(offset + i, 2, unit_item)
self.firstTable.item(offset + i, 0).setBackground(
self.GREEN) # set new US color to green
self.firstTable.item(offset + i, 1).setBackground(self.GREEN)
self.firstTable.item(offset + i, 2).setBackground(self.GREEN)
# lock the add US button again
self.btnAddUS.setEnabled(False)
# unlock add Ws button
self.btnAddWs.setEnabled(True)
def btnAddWsEvent(self):
msg = SettlingVelocityMessage(self.settling_velocities)
value = msg.exec_()
if value != QDialog.Accepted:
return
table = msg.get_table()
offset = self.secondTable.rowCount()
for i in range(len(table)):
self.secondTable.insertRow(offset + i)
for j in range(3):
item = QTableWidgetItem(table[i][j])
self.secondTable.setItem(offset + i, j, item)
self.secondTable.item(offset + i, j).setBackground(self.BLUE)
for i in range(len(table)):
self.settling_velocities.append(float(table[i][0][6:]))
def btnOpenEvent(self):
canceled, filename = super().open_event()
if canceled:
return
# reinitialize input file data
self._reinitInput()
success, self.data = self.read(filename)
if not success:
return
self.time = self.data.time
# displaying the available variables
self._initVarTables()
if self.data.header.is_2d:
# unlock add US button
if 'US' not in self.data.header.var_IDs and 'W' in self.data.header.var_IDs:
available_var_IDs = list(
map(lambda x: x.ID(), self.available_vars))
available_var_IDs.extend(self.data.header.var_IDs)
if 'H' in available_var_IDs and 'M' in available_var_IDs:
self.btnAddUS.setEnabled(True)
# unlock add Ws button
if 'US' in self.data.header.var_IDs:
self.btnAddWs.setEnabled(True)
self.parent.getInput(self.data)