1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 // File: ParametersPlugin_WidgetParamsMgr.cpp
4 // Created: 11 Apr 2016
5 // Author: Vitaly SMETANNIKOV
7 #include "ParametersPlugin_WidgetParamsMgr.h"
8 #include "ParametersPlugin_Parameter.h"
9 #include "ParametersPlugin_Validators.h"
11 #include <ModelAPI_ResultParameter.h>
12 #include <ModelAPI_AttributeString.h>
13 #include <ModelAPI_AttributeRefList.h>
14 #include <ModelAPI_AttributeDouble.h>
15 #include <ModelAPI_AttributeInteger.h>
16 #include <ModelAPI_Events.h>
17 #include <ModelAPI_Session.h>
19 #include <ModuleBase_Tools.h>
21 #include <Events_Loop.h>
24 #include <QPushButton>
25 #include <QToolButton>
26 #include <QStyledItemDelegate>
28 #include <QMessageBox>
40 const char* NoName = "<NoName>";
41 const char* NoValue = "<NoValue>";
43 class ParametersPlugin_ItemDelegate : public QStyledItemDelegate
46 ParametersPlugin_ItemDelegate(QObject* thaParent) :
47 QStyledItemDelegate(thaParent) {}
49 virtual void paint(QPainter* painter,
50 const QStyleOptionViewItem& option,
51 const QModelIndex& index ) const;
53 virtual QWidget* createEditor(QWidget* parent,
54 const QStyleOptionViewItem& option,
55 const QModelIndex& index) const;
57 bool isEditable(const QModelIndex& theIndex) const;
59 QModelIndex editIndex() const { return myEditingIdx; }
62 mutable QModelIndex myEditingIdx;
65 bool ParametersPlugin_ItemDelegate::isEditable(const QModelIndex& theIndex) const
67 QModelIndex aParent = theIndex.parent();
68 if (aParent.isValid() && (aParent.row() == 0)) {
69 if (theIndex.column() == 2)
76 void ParametersPlugin_ItemDelegate::paint(QPainter* painter,
77 const QStyleOptionViewItem& option,
78 const QModelIndex& index ) const
80 QBrush aBrush = painter->brush();
81 QPen aPen = painter->pen();
82 if (!isEditable(index))
83 painter->setBrush(Qt::lightGray);
85 painter->setPen(Qt::darkGray);
86 painter->drawRect(option.rect);
87 painter->setPen(aPen);
89 QStyledItemDelegate::paint(painter, option, index);
90 painter->setBrush(aBrush);
93 QWidget* ParametersPlugin_ItemDelegate::createEditor(QWidget* parent,
94 const QStyleOptionViewItem& option,
95 const QModelIndex& index) const
98 return QStyledItemDelegate::createEditor(parent, option, index);
101 /////////////////////////////////////////////////////////////////////////////////////////////////
102 void ParametersPlugin_TreeWidget::closeEditor(QWidget* theEditor, QAbstractItemDelegate::EndEditHint theHint)
104 if (theHint == QAbstractItemDelegate::EditNextItem) {
105 QModelIndex aCurrent = currentIndex();
106 QModelIndex aParent = model()->index(0, 0);
107 int aNbRows = model()->rowCount(aParent);
109 switch (aCurrent.column()) {
111 aIdx = model()->index(aCurrent.row(), 1, aParent);
114 if (aCurrent.row() < (aNbRows - 1))
115 aIdx = model()->index(aCurrent.row() + 1, 0, aParent);
117 QTreeWidget::closeEditor(theEditor, QAbstractItemDelegate::NoHint);
122 QTreeWidget::closeEditor(theEditor, theHint);
125 if (aIdx.isValid()) {
126 QTreeWidget::closeEditor(theEditor, QAbstractItemDelegate::NoHint);
127 setCurrentIndex(aIdx);
132 QTreeWidget::closeEditor(theEditor, theHint);
135 /////////////////////////////////////////////////////////////////////////////////////////////////
137 ParametersPlugin_WidgetParamsMgr::ParametersPlugin_WidgetParamsMgr(QWidget* theParent, const Config_WidgetAPI* theData)
138 : ModuleBase_ModelWidget(theParent, theData)
140 QVBoxLayout* aLayout = new QVBoxLayout(this);
142 myTable = new ParametersPlugin_TreeWidget(this);
143 myTable->setColumnCount(4);
144 QStringList aHeaders;
145 aHeaders << tr("Name") << tr("Equation") << tr("Result") << tr("Comment");
146 myTable->setHeaderLabels(aHeaders);
147 myTable->setColumnWidth(Col_Name, 200);
148 myTable->setColumnWidth(Col_Equation, 100);
149 myTable->setColumnWidth(Col_Result, 80);
150 myTable->setColumnWidth(Col_Comment, 200);
151 myTable->setMinimumWidth(600);
152 myTable->setEditTriggers(QAbstractItemView::NoEditTriggers);
153 myTable->setSelectionMode(QAbstractItemView::SingleSelection);
155 connect(myTable, SIGNAL(doubleClicked(const QModelIndex&)),
156 SLOT(onDoubleClick(const QModelIndex&)));
157 connect(myTable, SIGNAL(itemSelectionChanged()), SLOT(onSelectionChanged()));
159 myDelegate = new ParametersPlugin_ItemDelegate(myTable);
160 connect(myDelegate, SIGNAL(closeEditor(QWidget*, QAbstractItemDelegate::EndEditHint)),
161 SLOT(onCloseEditor(QWidget*, QAbstractItemDelegate::EndEditHint)));
163 myTable->setItemDelegate(myDelegate);
164 aLayout->addWidget(myTable);
168 aNames<<tr("Parameters");
169 myParameters = new QTreeWidgetItem(aNames);
170 myTable->addTopLevelItem(myParameters);
173 aNames<<tr("Features");
174 myFeatures = new QTreeWidgetItem(aNames);
175 myTable->addTopLevelItem(myFeatures);
177 QHBoxLayout* aBtnLayout = new QHBoxLayout(this);
179 myUpBtn = new QToolButton(this);
180 myUpBtn->setArrowType(Qt::UpArrow);
181 connect(myUpBtn, SIGNAL(clicked(bool)), SLOT(onUp()));
182 aBtnLayout->addWidget(myUpBtn);
184 myDownBtn = new QToolButton(this);
185 myDownBtn->setArrowType(Qt::DownArrow);
186 connect(myDownBtn, SIGNAL(clicked(bool)), SLOT(onDown()));
187 aBtnLayout->addWidget(myDownBtn);
189 aBtnLayout->addStretch();
191 myAddBtn = new QPushButton(tr("Add"), this);
192 connect(myAddBtn, SIGNAL(clicked(bool)), SLOT(onAdd()));
193 aBtnLayout->addWidget(myAddBtn);
195 myInsertBtn = new QPushButton(tr("Insert"), this);
196 connect(myInsertBtn, SIGNAL(clicked(bool)), SLOT(onInsert()));
197 aBtnLayout->addWidget(myInsertBtn);
199 myRemoveBtn = new QPushButton(tr("Remove"), this);
200 connect(myRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemove()));
201 aBtnLayout->addWidget(myRemoveBtn);
203 aLayout->addLayout(aBtnLayout);
205 onSelectionChanged();
208 QList<QWidget*> ParametersPlugin_WidgetParamsMgr::getControls() const
210 QList<QWidget*> aList;
215 void ParametersPlugin_WidgetParamsMgr::selectItemScroll(QTreeWidgetItem* aItem)
217 myTable->clearSelection();
218 QModelIndex aParent = myTable->model()->index(0, 0);
219 int aChildIdx = myParameters->indexOfChild(aItem);
220 QModelIndex aIndex = myTable->model()->index(aChildIdx, Col_Name, aParent);
221 myTable->selectionModel()->select(aIndex,
222 QItemSelectionModel::SelectCurrent | QItemSelectionModel::Rows);
223 myTable->scrollToItem(aItem);
227 bool ParametersPlugin_WidgetParamsMgr::storeValueCustom()
229 ParametersPlugin_ExpressionValidator aValidator;
230 std::list<std::string> aArgs;
231 std::string aAttrId = ParametersPlugin_Parameter::VARIABLE_ID();
234 foreach(FeaturePtr aFeature, myParametersList) {
235 if (!aValidator.isValid(aFeature->attribute(aAttrId), aArgs, aErr)) {
236 QMessageBox::warning(this, tr("Warning"), aErr.c_str());
237 selectItemScroll(myParameters->child(aId));
245 bool ParametersPlugin_WidgetParamsMgr::restoreValueCustom()
250 void ParametersPlugin_WidgetParamsMgr::activateCustom()
252 updateParametersFeatures();
253 updateParametersPart();
254 updateFeaturesPart();
256 myFeatures->setExpanded(true);
257 myParameters->setExpanded(true);
260 void ParametersPlugin_WidgetParamsMgr::updateParametersFeatures()
262 myParametersList.clear();
263 FeaturePtr aFeature = feature();
264 DocumentPtr aDoc = aFeature->document();
265 int aNbParam = aDoc->size(ModelAPI_ResultParameter::group());
267 FeaturePtr aParamFeature;
268 for (int i = 0; i < aNbParam; i++) {
269 aObj = aDoc->object(ModelAPI_ResultParameter::group(), i);
270 aParamFeature = ModelAPI_Feature::feature(aObj);
271 if (aParamFeature.get() && (aParamFeature->getKind() == ParametersPlugin_Parameter::ID())) {
272 myParametersList.append(aParamFeature);
277 void ParametersPlugin_WidgetParamsMgr::updateFeaturesPart()
279 updateItem(myFeatures, featuresItems(myParametersList));
282 void ParametersPlugin_WidgetParamsMgr::updateParametersPart()
284 updateItem(myParameters, parametersItems(myParametersList));
288 QList<QStringList> ParametersPlugin_WidgetParamsMgr::
289 featuresItems(const QList<FeaturePtr>& theFeatures) const
291 QList<QStringList> aItemsList;
292 ResultParameterPtr aParam;
293 foreach(FeaturePtr aParameter, theFeatures) {
294 aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aParameter->firstResult());
295 const std::set<std::shared_ptr<ModelAPI_Attribute>>& aRefs = aParam->data()->refsToMe();
296 std::set<std::shared_ptr<ModelAPI_Attribute> >::const_iterator aIt;
297 for(aIt = aRefs.cbegin(); aIt != aRefs.cend(); aIt++) {
298 std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
299 FeaturePtr aReferenced = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
300 if (aReferenced.get()) {
301 if (aReferenced->getKind() == ParametersPlugin_Parameter::ID()) {
302 // Find referenced feature Recursive
303 QList<FeaturePtr> aList;
304 aList.append(aReferenced);
305 QList<QStringList> aItems = featuresItems(aList);
306 aItemsList.append(aItems);
308 QStringList aValNames;
309 aValNames << aReferenced->data()->name().c_str();
311 AttributeDoublePtr aDouble = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(aAttr);
313 aValNames << aDouble->text().c_str();
314 aValNames << QString::number(aDouble->value());
316 AttributeIntegerPtr aInt = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>(aAttr);
318 aValNames << aInt->text().c_str();
319 aValNames << QString::number(aInt->value());
322 aItemsList.append(aValNames);
331 QList<QStringList> ParametersPlugin_WidgetParamsMgr::
332 parametersItems(const QList<FeaturePtr>& theFeatures) const
334 QList<QStringList> aItemsList;
335 foreach(FeaturePtr aParameter, theFeatures) {
336 ResultPtr aParam = aParameter->firstResult();
338 aValues << aParameter->string(ParametersPlugin_Parameter::VARIABLE_ID())->value().c_str();
339 aValues << aParameter->string(ParametersPlugin_Parameter::EXPRESSION_ID())->value().c_str();
341 AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
342 aValues << QString::number(aValueAttribute->value());
344 aValues << aParameter->string(ParametersPlugin_Parameter::COMMENT_ID())->value().c_str();
345 aItemsList.append(aValues);
351 void ParametersPlugin_WidgetParamsMgr::onDoubleClick(const QModelIndex& theIndex)
353 if (myDelegate->isEditable(theIndex)) {
354 myTable->setCurrentIndex(theIndex);
355 myTable->edit(theIndex);
359 void ParametersPlugin_WidgetParamsMgr::onCloseEditor(QWidget* theEditor,
360 QAbstractItemDelegate::EndEditHint theHint)
362 FeaturePtr aFeature = myParametersList.at(myDelegate->editIndex().row());
363 QTreeWidgetItem* aItem = myParameters->child(myDelegate->editIndex().row());
364 int aColumn = myDelegate->editIndex().column();
365 QString aText = aItem->text(aColumn);
366 bool isModified = false;
371 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::VARIABLE_ID());
372 if (!aText.isEmpty()) {
373 if (hasName(aText)) {
374 myMessage = tr("Name %1 already exists.").arg(aText);
375 if (aStringAttr->value().length() > 0)
376 aItem->setText(Col_Name, aStringAttr->value().c_str());
378 aItem->setText(Col_Name, NoName);
379 QTimer::singleShot(50, this, SLOT(sendWarning()));
382 aStringAttr->setValue(aText.toStdString());
385 aItem->setText(Col_Name, aStringAttr->value().c_str());
391 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::EXPRESSION_ID());
392 if (!aText.isEmpty()) {
393 if (aText != aStringAttr->value().c_str()) {
394 aStringAttr->setValue(aText.toStdString());
398 aItem->setText(Col_Equation, aStringAttr->value().c_str());
404 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::COMMENT_ID());
405 aStringAttr->setValue(aText.toStdString());
413 Events_Loop* aLoop = Events_Loop::loop();
414 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
415 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
416 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
418 ResultParameterPtr aResult =
419 std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aFeature->firstResult());
421 AttributeDoublePtr aValueAttribute =
422 aResult->data()->real(ModelAPI_ResultParameter::VALUE());
423 aItem->setText(Col_Result, QString::number(aValueAttribute->value()));
425 if (aColumn == Col_Equation)
426 updateParametersPart();
427 updateFeaturesPart();
430 void ParametersPlugin_WidgetParamsMgr::updateItem(QTreeWidgetItem* theItem,
431 const QList<QStringList>& theFeaturesList)
433 if (theFeaturesList.count() != theItem->childCount()) {
434 if (theItem->childCount() < theFeaturesList.count()) {
435 while (theItem->childCount() != theFeaturesList.count())
436 theItem->addChild(createNewItem());
438 while (theItem->childCount() != theFeaturesList.count())
439 theItem->removeChild(theItem->child(theItem->childCount() - 1));
443 foreach(QStringList aFeature, theFeaturesList) {
445 foreach(QString aText, aFeature) {
446 if (aText.length() > 0)
447 theItem->child(i)->setText(aCol, aText);
454 FeaturePtr ParametersPlugin_WidgetParamsMgr::createParameter() const
456 SessionPtr aMgr = ModelAPI_Session::get();
457 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
459 FeaturePtr aFeature = aDoc->addFeature(ParametersPlugin_Parameter::ID());
460 if (aFeature.get()) {
461 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
462 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
468 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::createNewItem() const
474 QTreeWidgetItem* aItem = new QTreeWidgetItem(aValues);
475 aItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
480 void ParametersPlugin_WidgetParamsMgr::onAdd()
482 FeaturePtr aFeature = createParameter();
486 QTreeWidgetItem* aItem = createNewItem();
487 if (aFeature->name().length() > 0)
488 aItem->setText(Col_Name, aFeature->name().c_str());
489 myParameters->addChild(aItem);
490 myParametersList.append(aFeature);
492 myTable->scrollToItem(aItem);
493 myTable->setCurrentItem(aItem);
494 myTable->editItem(aItem);
497 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::selectedItem() const
499 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
500 if (aItemsList.count() == 0)
503 QTreeWidgetItem* aCurrentItem = aItemsList.first();
504 if (aCurrentItem->parent() != myParameters)
511 void ParametersPlugin_WidgetParamsMgr::onInsert()
513 QTreeWidgetItem* aCurrentItem = selectedItem();
517 SessionPtr aMgr = ModelAPI_Session::get();
518 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
520 FeaturePtr aNewFeature = createParameter();
521 if (!aNewFeature.get())
524 QTreeWidgetItem* aItem = createNewItem();
525 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
526 if (aCurrentPos == 0) {
527 aDoc->moveFeature(aNewFeature, FeaturePtr());
529 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos - 1);
530 aDoc->moveFeature(aNewFeature, aCurFeature);
532 myParametersList.insert(aCurrentPos, aNewFeature);
533 myParameters->insertChild(aCurrentPos, aItem);
536 void ParametersPlugin_WidgetParamsMgr::onRemove()
538 QTreeWidgetItem* aCurrentItem = selectedItem();
542 SessionPtr aMgr = ModelAPI_Session::get();
543 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
545 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
546 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
548 QObjectPtrList anObjects;
549 anObjects.append(aCurFeature);
551 std::set<FeaturePtr> aDirectRefFeatures, aIndirectRefFeatures;
552 ModuleBase_Tools::findReferences(anObjects, aDirectRefFeatures, aIndirectRefFeatures);
554 bool doDeleteReferences = true;
555 if (ModuleBase_Tools::isDeleteFeatureWithReferences(anObjects, aDirectRefFeatures,
556 aIndirectRefFeatures, this, doDeleteReferences)) {
557 myParametersList.removeOne(aCurFeature);
558 myParameters->removeChild(aCurrentItem);
560 std::set<FeaturePtr> aFeaturesToDelete;
561 if (doDeleteReferences) {
562 aFeaturesToDelete = aDirectRefFeatures;
563 aFeaturesToDelete.insert(aIndirectRefFeatures.begin(), aIndirectRefFeatures.end());
565 aDoc->removeFeature(aCurFeature);
566 std::set<FeaturePtr>::const_iterator anIt = aFeaturesToDelete.begin(),
567 aLast = aFeaturesToDelete.end();
568 for (; anIt != aLast; anIt++) {
569 FeaturePtr aFeature = (*anIt);
570 DocumentPtr aDoc = aFeature->document();
571 aDoc->removeFeature(aFeature);
574 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
575 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
576 updateParametersFeatures();
577 updateFeaturesPart();
578 updateParametersPart();
582 void ParametersPlugin_WidgetParamsMgr::onUp()
584 QTreeWidgetItem* aCurrentItem = selectedItem();
588 QString aName = aCurrentItem->text(0);
590 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
591 if (aCurrentPos == 0)
593 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
595 std::string aNm = aCurFeature->data()->name();
597 SessionPtr aMgr = ModelAPI_Session::get();
598 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
600 FeaturePtr aa = myParametersList.at(aCurrentPos - 1);
601 std::string aN = aa->data()->name();
603 if (aCurrentPos == 1)
604 aDoc->moveFeature(aCurFeature, FeaturePtr());
606 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos - 2));
608 myParametersList.removeOne(aCurFeature);
609 myParametersList.insert(aCurrentPos - 1, aCurFeature);
611 myParameters->removeChild(aCurrentItem);
612 myParameters->insertChild(aCurrentPos - 1, aCurrentItem);
614 selectItemScroll(aCurrentItem);
616 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED));
617 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
618 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
619 //updateParametersFeatures();
620 //updateParametersPart();
621 //updateFeaturesPart();
624 void ParametersPlugin_WidgetParamsMgr::onDown()
626 QTreeWidgetItem* aCurrentItem = selectedItem();
630 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
631 if (aCurrentPos == (myParametersList.count() - 1))
633 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
635 SessionPtr aMgr = ModelAPI_Session::get();
636 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
637 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos + 1));
639 myParametersList.removeOne(aCurFeature);
640 myParametersList.insert(aCurrentPos + 1, aCurFeature);
642 myParameters->removeChild(aCurrentItem);
643 myParameters->insertChild(aCurrentPos + 1, aCurrentItem);
645 selectItemScroll(aCurrentItem);
647 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED));
648 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
649 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
650 //updateParametersFeatures();
651 //updateParametersPart();
652 //updateFeaturesPart();
656 bool ParametersPlugin_WidgetParamsMgr::hasName(const QString& theName) const
658 int aCurrent = myDelegate->editIndex().row();
660 foreach(FeaturePtr aFeature, myParametersList) {
661 if ((i != aCurrent) && (aFeature->data()->name() == theName.toStdString()))
668 void ParametersPlugin_WidgetParamsMgr::sendWarning()
670 QMessageBox::warning(this, tr("Warning"), myMessage);
673 void ParametersPlugin_WidgetParamsMgr::onSelectionChanged()
675 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
676 bool isParameter = false;
677 foreach(QTreeWidgetItem* aItem, aItemsList) {
678 if (aItem->parent() == myParameters) {
683 myInsertBtn->setEnabled(isParameter);
684 myRemoveBtn->setEnabled(isParameter);
685 myUpBtn->setEnabled(isParameter);
686 myDownBtn->setEnabled(isParameter);