def data(self, index, role=_Qt.DisplayRole):
if not index.isValid() or not 0 <= index.row() < self.rowCount():
return _QVariant()
row = index.row()
col = index.column()
if role == _Qt.DisplayRole:
if col == TableModelImages._COL_CMAP:
return str(self._model_data[row]['cmap_name'])
elif col == TableModelImages._COL_ALPHA:
return str(self._model_data[row]['alpha'])
elif col == TableModelImages._COL_CLIM_LOW:
return str(self._model_data[row]['clim_low'])
elif col == TableModelImages._COL_CLIM_HIGH:
return str(self._model_data[row]['clim_high'])
elif col == TableModelImages._COL_CBAR:
# print('1')
return str(self._model_data[row]['colorbar'])
elif col == TableModelImages._COL_LABEL:
return str(self._model_data[row]['label'])
elif col == TableModelImages._COL_DELETE:
return ''
elif role == _Qt.DecorationRole:
if col == TableModelImages._COL_DELETE:
color = [1, 0, 0]
color_256 = [color[0]*255, color[1]*255, color[2]*255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
else:
return _QVariant()
def data(self, index, role=_Qt.DisplayRole):
if not index.isValid() or not 0 <= index.row() < self.rowCount():
return _QVariant()
row = index.row()
col = index.column()
if role == _Qt.DisplayRole:
if col == TableModelImages._COL_CMAP:
return str(self._model_data[row]['cmap_name'])
elif col == TableModelImages._COL_ALPHA:
return str(self._model_data[row]['alpha'])
elif col == TableModelImages._COL_CLIM_LOW:
return str(self._model_data[row]['clim_low'])
elif col == TableModelImages._COL_CLIM_HIGH:
return str(self._model_data[row]['clim_high'])
elif col == TableModelImages._COL_CBAR:
# print('1')
return str(self._model_data[row]['colorbar'])
elif col == TableModelImages._COL_LABEL:
return str(self._model_data[row]['label'])
elif col == TableModelImages._COL_DELETE:
return ''
elif role == _Qt.DecorationRole:
if col == TableModelImages._COL_DELETE:
color = [1, 0, 0]
color_256 = [color[0] * 255, color[1] * 255, color[2] * 255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
else:
return _QVariant()
def data(self, index, role=_Qt.DisplayRole):
if not index.isValid() or not 0 <= index.row() < self.rowCount():
return _QVariant()
row = index.row()
col = index.column()
if role == _Qt.DisplayRole:
if col == TableModelLines._COL_COLOR:
return str(round_list(self._model_data[row]['color']))
elif col == TableModelLines._COL_ALPHA:
return str(self._model_data[row]['alpha'])
elif col == TableModelLines._COL_LINEWIDTH:
return str(self._model_data[row]['linewidth'])
elif col == TableModelLines._COL_LINESTYLE:
ls = self._model_data[row]['linestyle']
return str(MplLines.LINESTYLE_DICT[ls])
elif col == TableModelLines._COL_MARKER:
mk = self._model_data[row]['marker']
return str(MplMarkers.MARKER_DICT[mk])
elif col == TableModelLines._COL_MARKERSIZE:
return str(self._model_data[row]['markersize'])
elif col == TableModelLines._COL_LABEL:
return str(self._model_data[row]['label'])
elif col == TableModelLines._COL_DELETE:
return ''
elif role == _Qt.DecorationRole:
if col == TableModelLines._COL_COLOR:
color = self._model_data[row]['color']
color_256 = [color[0] * 255, color[1] * 255, color[2] * 255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
elif col == TableModelLines._COL_DELETE:
color = [1, 0, 0]
color_256 = [color[0] * 255, color[1] * 255, color[2] * 255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
else:
pass
else:
return _QVariant()
def data(self, index, role=_Qt.DisplayRole):
if not index.isValid() or not 0 <= index.row() < self.rowCount():
return _QVariant()
row = index.row()
col = index.column()
if role == _Qt.DisplayRole:
if col == TableModelLines._COL_COLOR:
return str(self._model_data[row]['color'])
elif col == TableModelLines._COL_ALPHA:
return str(self._model_data[row]['alpha'])
elif col == TableModelLines._COL_LINEWIDTH:
return str(self._model_data[row]['linewidth'])
elif col == TableModelLines._COL_LINESTYLE:
ls = self._model_data[row]['linestyle']
return str(MplLines.LINESTYLE_DICT[ls])
elif col == TableModelLines._COL_MARKER:
mk = self._model_data[row]['marker']
return str(MplMarkers.MARKER_DICT[mk])
elif col == TableModelLines._COL_MARKERSIZE:
return str(self._model_data[row]['markersize'])
elif col == TableModelLines._COL_LABEL:
return str(self._model_data[row]['label'])
elif col == TableModelLines._COL_DELETE:
return ''
elif role == _Qt.DecorationRole:
if col == TableModelLines._COL_COLOR:
color = self._model_data[row]['color']
color_256 = [color[0]*255, color[1]*255, color[2]*255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
elif col == TableModelLines._COL_DELETE:
color = [1, 0, 0]
color_256 = [color[0]*255, color[1]*255, color[2]*255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
else:
pass
else:
return _QVariant()
def createEditor(self, parent, option, index):
col = index.column()
if col == TableModelLines._COL_COLOR:
# Color handled by doubleClicked SIGNAL
pass
# LineWidth or Marker size or Alpha
elif (col == TableModelLines._COL_LINEWIDTH
or col == TableModelLines._COL_ALPHA
or col == TableModelLines._COL_MARKERSIZE):
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(20)
spinBoxSize.setSingleStep(.5)
return spinBoxSize
elif col == TableModelLines._COL_LINESTYLE: # LineStyle
comboBoxLineStyle = _QComboBox(parent)
for line_style in MplLines.LINESTYLE_DESC:
comboBoxLineStyle.addItem(line_style)
return comboBoxLineStyle
elif col == TableModelLines._COL_MARKER: # Marker
comboBoxMarker = _QComboBox(parent)
for marker_style in MplMarkers.MARKER_DESC:
comboBoxMarker.addItem(marker_style)
return comboBoxMarker
elif col == TableModelLines._COL_LABEL: # Label
lineEditLabel = _QLineEdit(parent)
return lineEditLabel
elif col == TableModelLines._COL_DELETE: # Delete?
pass
else:
return _QVariant()
def headerData(self, col, orientation, role):
"""
Basic horizontal header with no special role
"""
if orientation == _Qt.Horizontal and role == _Qt.DisplayRole:
return self.headers[col]
return _QVariant()
def createEditor(self, parent, option, index):
col = index.column()
if col == TableModelLines._COL_COLOR:
# Color handled by doubleClicked SIGNAL
pass
# LineWidth or Marker size or Alpha
elif (col == TableModelLines._COL_LINEWIDTH or
col == TableModelLines._COL_ALPHA or
col == TableModelLines._COL_MARKERSIZE):
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(20)
spinBoxSize.setSingleStep(.5)
return spinBoxSize
elif col == TableModelLines._COL_LINESTYLE: # LineStyle
comboBoxLineStyle = _QComboBox(parent)
for line_style in MplLines.LINESTYLE_DESC:
comboBoxLineStyle.addItem(line_style)
return comboBoxLineStyle
elif col == TableModelLines._COL_MARKER: # Marker
comboBoxMarker = _QComboBox(parent)
for marker_style in MplMarkers.MARKER_DESC:
comboBoxMarker.addItem(marker_style)
return comboBoxMarker
elif col == TableModelLines._COL_LABEL: # Label
lineEditLabel = _QLineEdit(parent)
return lineEditLabel
elif col == TableModelLines._COL_DELETE: # Delete?
pass
else:
return _QVariant()
def data(self, index, role=_Qt.DisplayRole):
if not index.isValid() or not 0 <= index.row() < self.rowCount():
return _QVariant()
row = index.row()
col = index.column()
if role == _Qt.DisplayRole:
if col == TableModelBars._COL_FACECOLOR:
return str(self._model_data[row]['facecolor'])
elif col == TableModelBars._COL_ALPHA:
return str(self._model_data[row]['alpha'])
elif col == TableModelBars._COL_EDGECOLOR:
return str(self._model_data[row]['edgecolor'])
elif col == TableModelBars._COL_LINEWIDTH:
return str(self._model_data[row]['linewidth'])
elif col == TableModelBars._COL_WIDTH_FACTOR:
return str(self._model_data[row]['width_factor'])
elif col == TableModelBars._COL_LABEL:
return str(self._model_data[row]['label'])
elif col == TableModelBars._COL_DELETE:
return ''
elif role == _Qt.DecorationRole:
if col == TableModelBars._COL_FACECOLOR:
color = self._model_data[row]['facecolor']
elif col == TableModelBars._COL_EDGECOLOR:
color = self._model_data[row]['edgecolor']
if (col == TableModelBars._COL_FACECOLOR or
col == TableModelBars._COL_EDGECOLOR):
color_256 = [color[0]*255, color[1]*255, color[2]*255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
elif col == TableModelBars._COL_DELETE:
color = [1, 0, 0]
color_256 = [color[0]*255, color[1]*255, color[2]*255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
else:
pass
else:
return _QVariant()
def data(self, index, role=_Qt.DisplayRole):
if not index.isValid() or not 0 <= index.row() < self.rowCount():
return _QVariant()
row = index.row()
col = index.column()
if role == _Qt.DisplayRole:
if col == TableModelBars._COL_FACECOLOR:
return str(round_list(self._model_data[row]['facecolor']))
elif col == TableModelBars._COL_ALPHA:
return str(self._model_data[row]['alpha'])
elif col == TableModelBars._COL_EDGECOLOR:
return str(round_list(self._model_data[row]['edgecolor']))
elif col == TableModelBars._COL_LINEWIDTH:
return str(self._model_data[row]['linewidth'])
elif col == TableModelBars._COL_WIDTH_FACTOR:
return str(self._model_data[row]['width_factor'])
elif col == TableModelBars._COL_LABEL:
return str(self._model_data[row]['label'])
elif col == TableModelBars._COL_DELETE:
return ''
elif role == _Qt.DecorationRole:
if col == TableModelBars._COL_FACECOLOR:
color = self._model_data[row]['facecolor']
elif col == TableModelBars._COL_EDGECOLOR:
color = self._model_data[row]['edgecolor']
if (col == TableModelBars._COL_FACECOLOR
or col == TableModelBars._COL_EDGECOLOR):
color_256 = [color[0] * 255, color[1] * 255, color[2] * 255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
elif col == TableModelBars._COL_DELETE:
color = [1, 0, 0]
color_256 = [color[0] * 255, color[1] * 255, color[2] * 255]
qcolor = _QColor(color_256[0], color_256[1], color_256[2])
pm = _QPixmap(20, 20)
pm.fill(qcolor)
icon = _QIcon(pm)
return icon
else:
pass
else:
return _QVariant()
def createEditor(self, parent, option, index):
col = index.column()
if (col == TableModelFillBetween._COL_FACECOLOR or
col == TableModelFillBetween._COL_EDGECOLOR):
# Color handled by doubleClicked SIGNAL
pass
# LineWidth or Marker size or Alpha
elif (col == TableModelFillBetween._COL_LINEWIDTH or
col == TableModelFillBetween._COL_ALPHA):
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(20)
spinBoxSize.setSingleStep(.5)
return spinBoxSize
elif col == TableModelFillBetween._COL_LABEL: # Label
lineEditLabel = _QLineEdit(parent)
return lineEditLabel
elif col == TableModelFillBetween._COL_DELETE: # Delete?
pass
else:
return _QVariant()
def createEditor(self, parent, option, index):
col = index.column()
if col == TableModelBars._COL_FACECOLOR:
# Color handled by doubleClicked SIGNAL
pass
elif col == TableModelBars._COL_EDGECOLOR:
# Color handled by doubleClicked SIGNAL
pass
# Alpha
elif col == TableModelBars._COL_ALPHA:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(1)
spinBoxSize.setSingleStep(.1)
return spinBoxSize
# Width Fraction
elif col == TableModelBars._COL_WIDTH_FACTOR:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(1000000000)
spinBoxSize.setSingleStep(.1)
return spinBoxSize
# LineWidth
elif col == TableModelBars._COL_LINEWIDTH:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(20)
spinBoxSize.setSingleStep(.5)
return spinBoxSize
elif col == TableModelBars._COL_LABEL: # Label
lineEditLabel = _QLineEdit(parent)
return lineEditLabel
elif col == TableModelBars._COL_DELETE: # Delete?
pass
else:
return _QVariant()
def createEditor(self, parent, option, index):
col = index.column()
if col == TableModelBars._COL_FACECOLOR:
# Color handled by doubleClicked SIGNAL
pass
elif col == TableModelBars._COL_EDGECOLOR:
# Color handled by doubleClicked SIGNAL
pass
# Alpha
elif col == TableModelBars._COL_ALPHA:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(1)
spinBoxSize.setSingleStep(.1)
return spinBoxSize
# Width Fraction
elif col == TableModelBars._COL_WIDTH_FACTOR:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(1000000000)
spinBoxSize.setSingleStep(.1)
return spinBoxSize
# LineWidth
elif col == TableModelBars._COL_LINEWIDTH:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(20)
spinBoxSize.setSingleStep(.5)
return spinBoxSize
elif col == TableModelBars._COL_LABEL: # Label
lineEditLabel = _QLineEdit(parent)
return lineEditLabel
elif col == TableModelBars._COL_DELETE: # Delete?
pass
else:
return _QVariant()
def createEditor(self, parent, option, index):
col = index.column()
if col == TableModelImages._COL_ALPHA:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(1)
spinBoxSize.setSingleStep(.1)
return spinBoxSize
# clim_low, clim_high
elif (col == TableModelImages._COL_CLIM_LOW
or col == TableModelImages._COL_CLIM_HIGH):
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(-1e10)
spinBoxSize.setMaximum(1e10)
spinBoxSize.setSingleStep(.5)
return spinBoxSize
elif col == TableModelImages._COL_CBAR: # colorbar
# print('3')
comboBoxTrueFalse = _QComboBox(parent)
comboBoxTrueFalse.addItems(['True', 'False'])
return comboBoxTrueFalse
elif col == TableModelImages._COL_CMAP: # cmaps
comboBoxCmapNames = _QComboBox(parent)
list_cmaps = list(_mpl.cm.cmap_d.keys())
list_cmaps.sort()
for cmap_name in list_cmaps:
comboBoxCmapNames.addItem(cmap_name)
return comboBoxCmapNames
elif col == TableModelImages._COL_LABEL: # Label
lineEditLabel = _QLineEdit(parent)
return lineEditLabel
elif col == TableModelImages._COL_DELETE: # Delete?
pass
else:
return _QVariant()
def createEditor(self, parent, option, index):
col = index.column()
if col == TableModelImages._COL_ALPHA:
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(0)
spinBoxSize.setMaximum(1)
spinBoxSize.setSingleStep(.1)
return spinBoxSize
# clim_low, clim_high
elif (col == TableModelImages._COL_CLIM_LOW or
col == TableModelImages._COL_CLIM_HIGH):
spinBoxSize = _QDoubleSpinBox(parent)
spinBoxSize.setMinimum(-1e10)
spinBoxSize.setMaximum(1e10)
spinBoxSize.setSingleStep(.5)
return spinBoxSize
elif col == TableModelImages._COL_CBAR: # colorbar
# print('3')
comboBoxTrueFalse = _QComboBox(parent)
comboBoxTrueFalse.addItems(['True','False'])
return comboBoxTrueFalse
elif col == TableModelImages._COL_CMAP: # cmaps
comboBoxCmapNames = _QComboBox(parent)
list_cmaps = list(_mpl.cm.cmap_d.keys())
list_cmaps.sort()
for cmap_name in list_cmaps:
comboBoxCmapNames.addItem(cmap_name)
return comboBoxCmapNames
elif col == TableModelImages._COL_LABEL: # Label
lineEditLabel = _QLineEdit(parent)
return lineEditLabel
elif col == TableModelImages._COL_DELETE: # Delete?
pass
else:
return _QVariant()
def headerData(self, col, orientation, role):
if orientation == _Qt.Horizontal and role == _Qt.DisplayRole:
return self.headers[col]
return _QVariant()
def headerData(self, col, orientation, role):
if orientation == _Qt.Horizontal and role == _Qt.DisplayRole:
return self.headers[col]
return _QVariant()
