Table widget : led and type is editable

But not applied yet to the terminal strip
2025-12-17 12:40:35 +01:00 · 2021-07-13 12:27:08 +02:00
parent efed7ab5c0
commit 30dc34ebe1
3 changed files with 213 additions and 24 deletions
--- a/sources/TerminalStrip/ui/terminalstripmodel.cpp
+++ b/sources/TerminalStrip/ui/terminalstripmodel.cpp
@@ -1,4 +1,4 @@
-/*
+/*
        Copyright 2006-2021 The QElectroTech Team
        This file is part of QElectroTech.

@@ -18,6 +18,23 @@
 #include "terminalstripmodel.h"
 #include "../terminalstrip.h"
 #include <QDebug>
+#include <QBrush>
+#include <QVector>
+#include <QComboBox>
+#include <QSpinBox>
+
+/**
+ * Some const int who describe what a column contain
+ */
+const int POS_CELL = 0;
+const int LEVEL_CELL = 1;
+const int LABEL_CELL = 2;
+const int XREF_CELL = 3;
+const int CABLE_CELL = 4;
+const int CABLE_WIRE_CELL = 5;
+const int TYPE_CELL = 6;
+const int LED_CELL = 7;
+const int CONDUCTOR_CELL = 8;

 /**
 * @brief TerminalStripModel::TerminalStripModel
@@ -50,23 +67,103 @@ int TerminalStripModel::columnCount(const QModelIndex &parent) const

 QVariant TerminalStripModel::data(const QModelIndex &index, int role) const
 {
-	if (role != Qt::DisplayRole ||
-		index.row() >= m_real_terminal_data.size()) {
+	if (index.row() >= m_real_terminal_data.size()) {
 		return QVariant();
 	}

 	auto rtd = m_real_terminal_data.at(index.row());
-	switch (index.column()) {
-		case 0 : return rtd.level_;
-		case 1 : return rtd.label_;
-		case 2 : return rtd.Xref_;
-		case 3 : return rtd.cable_;
-		case 4 : return rtd.cable_wire_;
-		case 5 : return ElementData::translatedTerminalType(rtd.type_);
-		case 6 : return rtd.led_;
-		case 7 : return rtd.conductor_;
-		default : return QVariant();
+
+
+	if (role == Qt::DisplayRole)
+	{
+		switch (index.column()) {
+			case POS_CELL : return rtd.pos_;
+			case LEVEL_CELL : return rtd.level_;
+			case LABEL_CELL : return rtd.label_;
+			case XREF_CELL : return rtd.Xref_;
+			case CABLE_CELL : return rtd.cable_;
+			case CABLE_WIRE_CELL : return rtd.cable_wire_;
+			case TYPE_CELL : return ElementData::translatedTerminalType(rtd.type_);
+			case CONDUCTOR_CELL : return rtd.conductor_;
+			default : return QVariant();
+		}
 	}
+	else if (role == Qt::EditRole)
+	{
+		switch (index.column()) {
+//			case LEVEL_CELL : return rtd.level_;
+			case LABEL_CELL : return rtd.label_;
+			default: return QVariant();
+
+		}
+	}
+	else if (role == Qt::CheckStateRole &&
+			   index.column() == LED_CELL)
+	{
+		return rtd.led_ ? Qt::Checked : Qt::Unchecked;
+	}
+	else if (role == Qt::BackgroundRole && index.column() <= CONDUCTOR_CELL )
+	{
+		if (m_modified_cell.contains(rtd.m_real_terminal) &&
+			m_modified_cell.value(rtd.m_real_terminal).at(index.column()))
+		{
+			return QBrush(Qt::yellow);
+		}
+	}
+
+	return QVariant();
+}
+
+bool TerminalStripModel::setData(const QModelIndex &index, const QVariant &value, int role)
+{
+	auto rtd = m_real_terminal_data.at(index.row());
+	bool modified_ = false;
+	int modified_cell = -1;
+	auto column_ = index.column();
+
+	if (column_ == LED_CELL &&
+		role == Qt::CheckStateRole)
+	{
+			rtd.led_ = value.toBool();
+			modified_ = true;
+			modified_cell = LED_CELL;
+	}
+	else if (column_ == TYPE_CELL &&
+			 role == Qt::EditRole)
+	{
+		rtd.type_ = value.value<ElementData::TerminalType>();
+		modified_ = true;
+		modified_cell = TYPE_CELL;
+	}
+//	else if (column_ == LEVEL_CELL &&
+//			 role == Qt::EditRole)
+//	{
+//		rtd.level_ = value.toInt();
+//		modified_ = true;
+//		modified_cell = LEVEL_CELL;
+//	}
+
+		//Set the modification to the terminal data
+	if (modified_)
+	{
+		m_real_terminal_data.replace(index.row(), rtd);
+
+		if (rtd.m_real_terminal)
+		{
+			QVector<bool> vector_;
+			if (m_modified_cell.contains(rtd.m_real_terminal)) {
+				vector_ = m_modified_cell.value(rtd.m_real_terminal);
+			} else {
+				vector_ = QVector<bool>({false, false, false, false, false, false, false, false, false});
+			}
+
+			vector_.replace(modified_cell, true);
+			m_modified_cell.insert(rtd.m_real_terminal, vector_);
+		}
+		return true;
+	}
+
+	return false;
 }

 QVariant TerminalStripModel::headerData(int section, Qt::Orientation orientation, int role) const
@@ -76,24 +173,38 @@ QVariant TerminalStripModel::headerData(int section, Qt::Orientation orientation
 		if (orientation == Qt::Horizontal)
 		{
 			switch (section) {
-				case 0: return tr("Étage");
-				case 1: return tr("Label");
-				case 2: return tr("Référence croisé");
-				case 3: return tr("Câble");
-				case 4: return tr("Couleur / numéro de fil câble");
-				case 5: return tr("Type");
-				case 6: return tr("led");
-				case 7: return tr("Numéro de conducteur");
+				case POS_CELL: return tr("Position");
+				case LEVEL_CELL: return tr("Étage");
+				case LABEL_CELL: return tr("Label");
+				case XREF_CELL: return tr("Référence croisé");
+				case CABLE_CELL: return tr("Câble");
+				case CABLE_WIRE_CELL: return tr("Couleur / numéro de fil câble");
+				case TYPE_CELL: return tr("Type");
+				case LED_CELL: return tr("led");
+				case CONDUCTOR_CELL: return tr("Numéro de conducteur");
 				default : return QVariant();
 			}
-		} else {
+		} /*else {
 			return QString::number(++section);
-		}
+		}*/
 	}

 	return QVariant();
 }

+Qt::ItemFlags TerminalStripModel::flags(const QModelIndex &index) const
+{
+	Qt::ItemFlags flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+
+	auto c = index.column();
+	if (/*c == LEVEL_CELL || */c == LABEL_CELL || c == TYPE_CELL)
+		flags = flags | Qt::ItemIsEditable;
+	if (c == LED_CELL) {
+		flags = flags | Qt::ItemIsUserCheckable;
+	}
+	return flags;
+}
+
 void TerminalStripModel::fillRealTerminalData()
 {
 	if (m_terminal_strip) {
@@ -102,3 +213,55 @@ void TerminalStripModel::fillRealTerminalData()
 		}
 	}
 }
+
+/***********************************************************
+ * Alittle delegate for add a combobox to edit type
+ * and a spinbox to edit the level of a terminal
+ **********************************************************/
+
+TerminalStripModelDelegate::TerminalStripModelDelegate(QObject *parent) :
+	QStyledItemDelegate(parent)
+{}
+
+QWidget *TerminalStripModelDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index) const
+{
+	if (index.column() == TYPE_CELL) {
+		auto qcb = new QComboBox(parent);
+		qcb->setObjectName("terminal_type");
+		qcb->addItem(ElementData::translatedTerminalType(ElementData::TTGeneric),   ElementData::TTGeneric);
+		qcb->addItem(ElementData::translatedTerminalType(ElementData::TTFuse),      ElementData::TTFuse);
+		qcb->addItem(ElementData::translatedTerminalType(ElementData::TTSectional), ElementData::TTSectional);
+		qcb->addItem(ElementData::translatedTerminalType(ElementData::TTDiode),     ElementData::TTDiode);
+		qcb->addItem(ElementData::translatedTerminalType(ElementData::TTGround),    ElementData::TTGround);
+
+		return qcb;
+	}
+//	if (index.column() == LEVEL_CELL) {
+//		auto qsb = new QSpinBox(parent);
+//		qsb->setObjectName("terminal_level");
+//		qsb->setRange(0, 3);
+//		qsb->setValue(index.data(Qt::EditRole).toInt());
+//		return qsb;
+//	}
+
+	return QStyledItemDelegate::createEditor(parent, option, index);
+}
+
+void TerminalStripModelDelegate::setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const
+{
+	if (index.isValid())
+	{
+		if (editor->objectName() == QLatin1String("terminal_type"))
+		{
+			if (auto qcb = dynamic_cast<QComboBox *>(editor)) {
+				model->setData(index, qcb->currentData(), Qt::EditRole);
+			}
+		}
+//		else if (editor->objectName() == QLatin1String("terminal_level"))
+//		{
+//			if (auto qsb = dynamic_cast<QSpinBox *>(editor)) {
+//				model->setData(index, qsb->value(), Qt::EditRole);
+//			}
+//		}
+	}
+}