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>";
42 const char* NotValid = "<NotValid>";
44 class ParametersPlugin_ItemDelegate : public QStyledItemDelegate
47 ParametersPlugin_ItemDelegate(QObject* thaParent) :
48 QStyledItemDelegate(thaParent) {}
50 virtual void paint(QPainter* painter,
51 const QStyleOptionViewItem& option,
52 const QModelIndex& index ) const;
54 virtual QWidget* createEditor(QWidget* parent,
55 const QStyleOptionViewItem& option,
56 const QModelIndex& index) const;
58 bool isEditable(const QModelIndex& theIndex) const;
60 QModelIndex editIndex() const { return myEditingIdx; }
63 mutable QModelIndex myEditingIdx;
66 bool ParametersPlugin_ItemDelegate::isEditable(const QModelIndex& theIndex) const
68 QModelIndex aParent = theIndex.parent();
69 if (aParent.isValid() && (aParent.row() == 0)) {
70 if (theIndex.column() == 2)
77 void ParametersPlugin_ItemDelegate::paint(QPainter* painter,
78 const QStyleOptionViewItem& option,
79 const QModelIndex& index ) const
81 QBrush aBrush = painter->brush();
82 QPen aPen = painter->pen();
83 if (!isEditable(index))
84 painter->setBrush(Qt::lightGray);
86 painter->setPen(Qt::darkGray);
87 painter->drawRect(option.rect);
88 painter->setPen(aPen);
90 QStyledItemDelegate::paint(painter, option, index);
91 painter->setBrush(aBrush);
94 QWidget* ParametersPlugin_ItemDelegate::createEditor(QWidget* parent,
95 const QStyleOptionViewItem& option,
96 const QModelIndex& index) const
99 return QStyledItemDelegate::createEditor(parent, option, index);
102 /////////////////////////////////////////////////////////////////////////////////////////////////
103 void ParametersPlugin_TreeWidget::closeEditor(QWidget* theEditor, QAbstractItemDelegate::EndEditHint theHint)
105 if (theHint == QAbstractItemDelegate::EditNextItem) {
106 QModelIndex aCurrent = currentIndex();
107 QModelIndex aParent = model()->index(0, 0);
108 int aNbRows = model()->rowCount(aParent);
110 switch (aCurrent.column()) {
112 aIdx = model()->index(aCurrent.row(), 1, aParent);
115 if (aCurrent.row() < (aNbRows - 1))
116 aIdx = model()->index(aCurrent.row() + 1, 0, aParent);
118 QTreeWidget::closeEditor(theEditor, QAbstractItemDelegate::NoHint);
123 QTreeWidget::closeEditor(theEditor, theHint);
126 if (aIdx.isValid()) {
127 QTreeWidget::closeEditor(theEditor, QAbstractItemDelegate::NoHint);
128 setCurrentIndex(aIdx);
133 QTreeWidget::closeEditor(theEditor, theHint);
136 /////////////////////////////////////////////////////////////////////////////////////////////////
138 ParametersPlugin_WidgetParamsMgr::ParametersPlugin_WidgetParamsMgr(QWidget* theParent, const Config_WidgetAPI* theData)
139 : ModuleBase_ModelWidget(theParent, theData)
141 QVBoxLayout* aLayout = new QVBoxLayout(this);
143 myTable = new ParametersPlugin_TreeWidget(this);
144 myTable->setColumnCount(4);
145 QStringList aHeaders;
146 aHeaders << tr("Name") << tr("Equation") << tr("Result") << tr("Comment");
147 myTable->setHeaderLabels(aHeaders);
148 myTable->setColumnWidth(Col_Name, 200);
149 myTable->setColumnWidth(Col_Equation, 100);
150 myTable->setColumnWidth(Col_Result, 80);
151 myTable->setColumnWidth(Col_Comment, 200);
152 myTable->setMinimumWidth(600);
153 myTable->setEditTriggers(QAbstractItemView::NoEditTriggers);
154 myTable->setSelectionMode(QAbstractItemView::SingleSelection);
156 connect(myTable, SIGNAL(doubleClicked(const QModelIndex&)),
157 SLOT(onDoubleClick(const QModelIndex&)));
158 connect(myTable, SIGNAL(itemSelectionChanged()), SLOT(onSelectionChanged()));
160 myDelegate = new ParametersPlugin_ItemDelegate(myTable);
161 connect(myDelegate, SIGNAL(closeEditor(QWidget*, QAbstractItemDelegate::EndEditHint)),
162 SLOT(onCloseEditor(QWidget*, QAbstractItemDelegate::EndEditHint)));
164 myTable->setItemDelegate(myDelegate);
165 aLayout->addWidget(myTable);
169 aNames<<tr("Parameters");
170 myParameters = new QTreeWidgetItem(aNames);
171 myTable->addTopLevelItem(myParameters);
174 aNames<<tr("Features");
175 myFeatures = new QTreeWidgetItem(aNames);
176 myTable->addTopLevelItem(myFeatures);
178 QHBoxLayout* aBtnLayout = new QHBoxLayout(this);
180 myUpBtn = new QToolButton(this);
181 myUpBtn->setArrowType(Qt::UpArrow);
182 connect(myUpBtn, SIGNAL(clicked(bool)), SLOT(onUp()));
183 aBtnLayout->addWidget(myUpBtn);
185 myDownBtn = new QToolButton(this);
186 myDownBtn->setArrowType(Qt::DownArrow);
187 connect(myDownBtn, SIGNAL(clicked(bool)), SLOT(onDown()));
188 aBtnLayout->addWidget(myDownBtn);
190 aBtnLayout->addStretch();
192 myAddBtn = new QPushButton(tr("Add"), this);
193 connect(myAddBtn, SIGNAL(clicked(bool)), SLOT(onAdd()));
194 aBtnLayout->addWidget(myAddBtn);
196 myInsertBtn = new QPushButton(tr("Insert"), this);
197 connect(myInsertBtn, SIGNAL(clicked(bool)), SLOT(onInsert()));
198 aBtnLayout->addWidget(myInsertBtn);
200 myRemoveBtn = new QPushButton(tr("Remove"), this);
201 connect(myRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemove()));
202 aBtnLayout->addWidget(myRemoveBtn);
204 aLayout->addLayout(aBtnLayout);
206 onSelectionChanged();
209 QList<QWidget*> ParametersPlugin_WidgetParamsMgr::getControls() const
211 QList<QWidget*> aList;
216 void ParametersPlugin_WidgetParamsMgr::selectItemScroll(QTreeWidgetItem* aItem)
218 myTable->clearSelection();
219 QModelIndex aParent = myTable->model()->index(0, 0);
220 int aChildIdx = myParameters->indexOfChild(aItem);
221 QModelIndex aIndex = myTable->model()->index(aChildIdx, Col_Name, aParent);
222 myTable->selectionModel()->select(aIndex,
223 QItemSelectionModel::SelectCurrent | QItemSelectionModel::Rows);
224 myTable->scrollToItem(aItem);
228 bool ParametersPlugin_WidgetParamsMgr::storeValueCustom()
230 ParametersPlugin_ExpressionValidator aValidator;
231 std::list<std::string> aArgs;
232 std::string aAttrId = ParametersPlugin_Parameter::VARIABLE_ID();
235 foreach(FeaturePtr aFeature, myParametersList) {
236 if (!aValidator.isValid(aFeature->attribute(aAttrId), aArgs, aErr)) {
237 QMessageBox::warning(this, tr("Warning"), aErr.c_str());
238 selectItemScroll(myParameters->child(aId));
246 bool ParametersPlugin_WidgetParamsMgr::restoreValueCustom()
251 void ParametersPlugin_WidgetParamsMgr::activateCustom()
253 updateParametersFeatures();
254 updateParametersPart();
255 updateFeaturesPart();
257 myFeatures->setExpanded(true);
258 myParameters->setExpanded(true);
261 void ParametersPlugin_WidgetParamsMgr::updateParametersFeatures()
263 myParametersList.clear();
264 FeaturePtr aFeature = feature();
265 DocumentPtr aDoc = aFeature->document();
267 FeaturePtr aParamFeature;
268 int aNbFeatures = aDoc->numInternalFeatures();
269 for (int i = 0; i < aNbFeatures; i++) {
270 aParamFeature = aDoc->internalFeature(i);
271 if (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));
285 bool aIsValid = checkIsValid();
286 enableButtons(aIsValid);
290 QList<QStringList> ParametersPlugin_WidgetParamsMgr::
291 featuresItems(const QList<FeaturePtr>& theFeatures) const
293 QList<QStringList> aItemsList;
294 ResultParameterPtr aParam;
295 QList<FeaturePtr> aFeatures;
296 foreach(FeaturePtr aParameter, theFeatures) {
297 aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aParameter->firstResult());
298 const std::set<std::shared_ptr<ModelAPI_Attribute>>& aRefs = aParam->data()->refsToMe();
299 std::set<std::shared_ptr<ModelAPI_Attribute> >::const_iterator aIt;
300 for(aIt = aRefs.cbegin(); aIt != aRefs.cend(); aIt++) {
301 std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
302 FeaturePtr aReferenced = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
303 if (aReferenced.get()) {
304 if (aReferenced->getKind() == ParametersPlugin_Parameter::ID()) {
305 // Find referenced feature Recursive
306 QList<FeaturePtr> aList;
307 aList.append(aReferenced);
308 QList<QStringList> aItems = featuresItems(aList);
309 aItemsList.append(aItems);
311 if (!aFeatures.contains(aReferenced)) {
312 QStringList aValNames;
313 aValNames << aReferenced->data()->name().c_str();
315 AttributeDoublePtr aDouble = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(aAttr);
317 aValNames << aDouble->text().c_str();
318 aValNames << QString::number(aDouble->value());
320 AttributeIntegerPtr aInt = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>(aAttr);
322 aValNames << aInt->text().c_str();
323 aValNames << QString::number(aInt->value());
326 aItemsList.append(aValNames);
327 aFeatures.append(aReferenced);
337 QList<QStringList> ParametersPlugin_WidgetParamsMgr::
338 parametersItems(const QList<FeaturePtr>& theFeatures) const
340 std::list<std::string> aArgs;
342 QList<QStringList> aItemsList;
343 foreach(FeaturePtr aParameter, theFeatures) {
344 ResultPtr aParam = aParameter->firstResult();
347 std::string aName = aParameter->string(ParametersPlugin_Parameter::VARIABLE_ID())->value();
351 aValues << aName.c_str();
353 std::string aExpr = aParameter->string(ParametersPlugin_Parameter::EXPRESSION_ID())->value();
357 aValues << aExpr.c_str();
359 std::string aErr = aParameter->data()->string(ParametersPlugin_Parameter::EXPRESSION_ERROR_ID())->value();
361 AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
362 aValues << QString::number(aValueAttribute->value());
364 aValues << aErr.c_str();
366 aValues << aParameter->string(ParametersPlugin_Parameter::COMMENT_ID())->value().c_str();
367 aItemsList.append(aValues);
373 void ParametersPlugin_WidgetParamsMgr::onDoubleClick(const QModelIndex& theIndex)
375 if (myDelegate->isEditable(theIndex)) {
376 myTable->setCurrentIndex(theIndex);
377 myTable->edit(theIndex);
381 void ParametersPlugin_WidgetParamsMgr::onCloseEditor(QWidget* theEditor,
382 QAbstractItemDelegate::EndEditHint theHint)
384 FeaturePtr aFeature = myParametersList.at(myDelegate->editIndex().row());
385 QTreeWidgetItem* aItem = myParameters->child(myDelegate->editIndex().row());
386 int aColumn = myDelegate->editIndex().column();
387 QString aText = aItem->text(aColumn);
388 bool isModified = false;
393 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::VARIABLE_ID());
394 if (!aText.isEmpty()) {
395 if (hasName(aText)) {
396 myMessage = tr("Name %1 already exists.").arg(aText);
397 QTimer::singleShot(50, this, SLOT(sendWarning()));
400 aStringAttr->setValue(aText.toStdString());
407 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::EXPRESSION_ID());
408 if (!aText.isEmpty()) {
409 if (aText != aStringAttr->value().c_str()) {
410 aStringAttr->setValue(aText.toStdString());
419 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::COMMENT_ID());
420 aStringAttr->setValue(aText.toStdString());
428 Events_Loop* aLoop = Events_Loop::loop();
429 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
430 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
431 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
433 if (aColumn == Col_Equation)
434 updateParametersPart();
435 updateFeaturesPart();
438 void ParametersPlugin_WidgetParamsMgr::updateItem(QTreeWidgetItem* theItem,
439 const QList<QStringList>& theFeaturesList)
441 if (theFeaturesList.count() != theItem->childCount()) {
442 if (theItem->childCount() < theFeaturesList.count()) {
443 while (theItem->childCount() != theFeaturesList.count())
444 theItem->addChild(createNewItem());
446 while (theItem->childCount() != theFeaturesList.count())
447 theItem->removeChild(theItem->child(theItem->childCount() - 1));
451 foreach(QStringList aFeature, theFeaturesList) {
453 foreach(QString aText, aFeature) {
454 if (aText.length() > 0) {
455 theItem->child(i)->setText(aCol, aText);
456 theItem->child(i)->setToolTip(aCol, aText);
464 FeaturePtr ParametersPlugin_WidgetParamsMgr::createParameter() const
466 SessionPtr aMgr = ModelAPI_Session::get();
467 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
469 FeaturePtr aFeature = aDoc->addFeature(ParametersPlugin_Parameter::ID());
470 if (aFeature.get()) {
471 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
472 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
478 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::createNewItem() const
484 QTreeWidgetItem* aItem = new QTreeWidgetItem(aValues);
485 aItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
490 void ParametersPlugin_WidgetParamsMgr::onAdd()
492 FeaturePtr aFeature = createParameter();
496 myParametersList.append(aFeature);
497 updateParametersPart();
499 QTreeWidgetItem* aItem = myParameters->child(myParameters->childCount() - 1);
501 myTable->scrollToItem(aItem);
502 myTable->setCurrentItem(aItem);
503 myTable->editItem(aItem);
505 enableButtons(false);
508 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::selectedItem() const
510 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
511 if (aItemsList.count() == 0)
514 QTreeWidgetItem* aCurrentItem = aItemsList.first();
515 if (aCurrentItem->parent() != myParameters)
522 void ParametersPlugin_WidgetParamsMgr::onInsert()
524 QTreeWidgetItem* aCurrentItem = selectedItem();
528 SessionPtr aMgr = ModelAPI_Session::get();
529 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
531 FeaturePtr aNewFeature = createParameter();
532 if (!aNewFeature.get())
535 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
536 if (aCurrentPos == 0) {
537 aDoc->moveFeature(aNewFeature, FeaturePtr());
539 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos - 1);
540 aDoc->moveFeature(aNewFeature, aCurFeature);
542 updateParametersFeatures();
543 updateParametersPart();
545 myTable->scrollToItem(aCurrentItem);
546 myTable->setCurrentItem(aCurrentItem);
547 myTable->editItem(aCurrentItem);
550 void ParametersPlugin_WidgetParamsMgr::onRemove()
552 QTreeWidgetItem* aCurrentItem = selectedItem();
556 SessionPtr aMgr = ModelAPI_Session::get();
557 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
559 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
560 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
562 QObjectPtrList anObjects;
563 anObjects.append(aCurFeature);
565 std::set<FeaturePtr> aDirectRefFeatures, aIndirectRefFeatures;
566 ModuleBase_Tools::findReferences(anObjects, aDirectRefFeatures, aIndirectRefFeatures);
568 bool doDeleteReferences = true;
569 if (ModuleBase_Tools::isDeleteFeatureWithReferences(anObjects, aDirectRefFeatures,
570 aIndirectRefFeatures, this, doDeleteReferences)) {
572 std::set<FeaturePtr> aFeaturesToDelete;
573 if (doDeleteReferences) {
574 aFeaturesToDelete = aDirectRefFeatures;
575 aFeaturesToDelete.insert(aIndirectRefFeatures.begin(), aIndirectRefFeatures.end());
577 aDoc->removeFeature(aCurFeature);
578 std::set<FeaturePtr>::const_iterator anIt = aFeaturesToDelete.begin(),
579 aLast = aFeaturesToDelete.end();
580 for (; anIt != aLast; anIt++) {
581 FeaturePtr aFeature = (*anIt);
582 DocumentPtr aDoc = aFeature->document();
583 aDoc->removeFeature(aFeature);
586 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
587 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
588 updateParametersFeatures();
589 updateFeaturesPart();
590 updateParametersPart();
594 void ParametersPlugin_WidgetParamsMgr::onUp()
596 QTreeWidgetItem* aCurrentItem = selectedItem();
600 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
601 if (aCurrentPos == 0)
603 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
605 SessionPtr aMgr = ModelAPI_Session::get();
606 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
608 if (aCurrentPos == 1)
609 aDoc->moveFeature(aCurFeature, FeaturePtr());
611 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos - 2));
614 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED));
615 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
616 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
617 updateParametersFeatures();
618 updateParametersPart();
619 updateFeaturesPart();
621 if (aCurrentPos > 0) {
622 aCurrentItem = myParameters->child(aCurrentPos - 1);
623 myTable->setCurrentItem(aCurrentItem);
624 selectItemScroll(aCurrentItem);
628 void ParametersPlugin_WidgetParamsMgr::onDown()
630 QTreeWidgetItem* aCurrentItem = selectedItem();
634 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
635 if (aCurrentPos == (myParametersList.count() - 1))
637 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
639 SessionPtr aMgr = ModelAPI_Session::get();
640 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
641 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos + 1));
643 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED));
644 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
645 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
646 updateParametersFeatures();
647 updateParametersPart();
648 updateFeaturesPart();
650 if (aCurrentPos < myParameters->childCount() - 1) {
651 aCurrentItem = myParameters->child(aCurrentPos + 1);
652 myTable->setCurrentItem(aCurrentItem);
653 selectItemScroll(aCurrentItem);
658 bool ParametersPlugin_WidgetParamsMgr::hasName(const QString& theName) const
660 int aCurrent = myDelegate->editIndex().row();
662 foreach(FeaturePtr aFeature, myParametersList) {
663 if ((i != aCurrent) && (aFeature->data()->name() == theName.toStdString()))
670 void ParametersPlugin_WidgetParamsMgr::sendWarning()
672 QMessageBox::warning(this, tr("Warning"), myMessage);
675 void ParametersPlugin_WidgetParamsMgr::onSelectionChanged()
677 bool isValid = checkIsValid();
679 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
680 bool isParameter = false;
681 foreach(QTreeWidgetItem* aItem, aItemsList) {
682 if (aItem->parent() == myParameters) {
687 myInsertBtn->setEnabled(isParameter);
688 myRemoveBtn->setEnabled(isParameter);
689 myUpBtn->setEnabled(isParameter);
690 myDownBtn->setEnabled(isParameter);
692 myInsertBtn->setEnabled(false);
693 myRemoveBtn->setEnabled(false);
694 myUpBtn->setEnabled(false);
695 myDownBtn->setEnabled(false);
699 void ParametersPlugin_WidgetParamsMgr::enableButtons(bool theEnable)
701 myAddBtn->setEnabled(theEnable);
703 onSelectionChanged();
705 myInsertBtn->setEnabled(theEnable);
706 myRemoveBtn->setEnabled(theEnable);
707 myUpBtn->setEnabled(theEnable);
708 myDownBtn->setEnabled(theEnable);
712 bool ParametersPlugin_WidgetParamsMgr::checkIsValid()
714 QTreeWidgetItem* aItem;
715 bool aIsValid = true;
716 for(int i = 0; i < myParameters->childCount(); i++) {
717 aItem = myParameters->child(i);
718 if ((aItem->text(Col_Name) == NoName) || (aItem->text(Col_Equation) == NoValue)) {