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 <QTreeWidget>
25 #include <QPushButton>
26 #include <QToolButton>
27 #include <QStyledItemDelegate>
29 #include <QMessageBox>
39 const char* NoName = "<NoName>";
40 const char* NoValue = "<NoValue>";
42 class ParametersPlugin_ItemDelegate : public QStyledItemDelegate
45 ParametersPlugin_ItemDelegate(QObject* thaParent) :
46 QStyledItemDelegate(thaParent) {}
48 virtual void paint(QPainter* painter,
49 const QStyleOptionViewItem& option,
50 const QModelIndex& index ) const;
52 //virtual QWidget* createEditor(QWidget* parent,
53 // const QStyleOptionViewItem& option,
54 // const QModelIndex& index) const;
56 bool isEditable(const QModelIndex& theIndex) const;
59 bool ParametersPlugin_ItemDelegate::isEditable(const QModelIndex& theIndex) const
61 QModelIndex aParent = theIndex.parent();
62 if (aParent.isValid() && (aParent.row() == 0)) {
63 if (theIndex.column() == 2)
70 void ParametersPlugin_ItemDelegate::paint(QPainter* painter,
71 const QStyleOptionViewItem& option,
72 const QModelIndex& index ) const
74 QBrush aBrush = painter->brush();
75 QPen aPen = painter->pen();
76 if (!isEditable(index))
77 painter->setBrush(Qt::lightGray);
79 painter->setPen(Qt::darkGray);
80 painter->drawRect(option.rect);
81 painter->setPen(aPen);
83 //QString aText = index.data().toString();
84 //if ((aText == NoName) || (aText == NoValue))
85 // painter->setPen(Qt::red);
87 QStyledItemDelegate::paint(painter, option, index);
89 //painter->setPen(aPen);
90 painter->setBrush(aBrush);
94 //QWidget* ParametersPlugin_ItemDelegate::createEditor(QWidget* parent,
95 // const QStyleOptionViewItem& option,
96 // const QModelIndex& index) const
98 // QWidget* aWgt = QStyledItemDelegate::createEditor(parent, option, index);
99 // aWgt->setMinimumSize(option.rect.width() - option.decorationSize.width(),
100 // option.rect.height());
105 /////////////////////////////////////////////////////////////////////////////////////////////////
107 ParametersPlugin_WidgetParamsMgr::ParametersPlugin_WidgetParamsMgr(QWidget* theParent, const Config_WidgetAPI* theData)
108 : ModuleBase_ModelWidget(theParent, theData)
110 QVBoxLayout* aLayout = new QVBoxLayout(this);
112 myTable = new QTreeWidget(this);
113 myTable->setColumnCount(4);
114 QStringList aHeaders;
115 aHeaders << tr("Name") << tr("Equation") << tr("Result") << tr("Comment");
116 myTable->setHeaderLabels(aHeaders);
117 myTable->setColumnWidth(Col_Name, 200);
118 myTable->setColumnWidth(Col_Equation, 100);
119 myTable->setColumnWidth(Col_Result, 80);
120 myTable->setColumnWidth(Col_Comment, 200);
121 myTable->setMinimumWidth(600);
122 myTable->setEditTriggers(QAbstractItemView::NoEditTriggers);
123 myTable->setSelectionMode(QAbstractItemView::SingleSelection);
125 connect(myTable, SIGNAL(doubleClicked(const QModelIndex&)),
126 SLOT(onDoubleClick(const QModelIndex&)));
127 connect(myTable, SIGNAL(itemSelectionChanged()), SLOT(onSelectionChanged()));
129 myDelegate = new ParametersPlugin_ItemDelegate(myTable);
130 connect(myDelegate, SIGNAL(closeEditor(QWidget*, QAbstractItemDelegate::EndEditHint)),
131 SLOT(onCloseEditor(QWidget*, QAbstractItemDelegate::EndEditHint)));
133 myTable->setItemDelegate(myDelegate);
134 aLayout->addWidget(myTable);
138 aNames<<tr("Parameters");
139 myParameters = new QTreeWidgetItem(aNames);
140 myTable->addTopLevelItem(myParameters);
143 aNames<<tr("Features");
144 myFeatures = new QTreeWidgetItem(aNames);
145 myTable->addTopLevelItem(myFeatures);
147 QHBoxLayout* aBtnLayout = new QHBoxLayout(this);
149 myUpBtn = new QToolButton(this);
150 myUpBtn->setArrowType(Qt::UpArrow);
151 connect(myUpBtn, SIGNAL(clicked(bool)), SLOT(onUp()));
152 aBtnLayout->addWidget(myUpBtn);
154 myDownBtn = new QToolButton(this);
155 myDownBtn->setArrowType(Qt::DownArrow);
156 connect(myDownBtn, SIGNAL(clicked(bool)), SLOT(onDown()));
157 aBtnLayout->addWidget(myDownBtn);
159 aBtnLayout->addStretch();
161 myAddBtn = new QPushButton(tr("Add"), this);
162 connect(myAddBtn, SIGNAL(clicked(bool)), SLOT(onAdd()));
163 aBtnLayout->addWidget(myAddBtn);
165 myInsertBtn = new QPushButton(tr("Insert"), this);
166 connect(myInsertBtn, SIGNAL(clicked(bool)), SLOT(onInsert()));
167 aBtnLayout->addWidget(myInsertBtn);
169 myRemoveBtn = new QPushButton(tr("Remove"), this);
170 connect(myRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemove()));
171 aBtnLayout->addWidget(myRemoveBtn);
173 aLayout->addLayout(aBtnLayout);
175 onSelectionChanged();
178 QList<QWidget*> ParametersPlugin_WidgetParamsMgr::getControls() const
180 QList<QWidget*> aList;
185 void ParametersPlugin_WidgetParamsMgr::selectItemScroll(QTreeWidgetItem* aItem)
187 myTable->clearSelection();
188 QModelIndex aParent = myTable->model()->index(0, 0);
189 int aChildIdx = myParameters->indexOfChild(aItem);
190 QModelIndex aIndex = myTable->model()->index(aChildIdx, Col_Name, aParent);
191 myTable->selectionModel()->select(aIndex,
192 QItemSelectionModel::SelectCurrent | QItemSelectionModel::Rows);
193 myTable->scrollToItem(aItem);
197 bool ParametersPlugin_WidgetParamsMgr::storeValueCustom()
199 ParametersPlugin_ExpressionValidator aValidator;
200 std::list<std::string> aArgs;
201 std::string aAttrId = ParametersPlugin_Parameter::VARIABLE_ID();
204 foreach(FeaturePtr aFeature, myParametersList) {
205 if (!aValidator.isValid(aFeature->attribute(aAttrId), aArgs, aErr)) {
206 QMessageBox::warning(this, tr("Warning"), aErr.c_str());
207 selectItemScroll(myParameters->child(aId));
215 bool ParametersPlugin_WidgetParamsMgr::restoreValueCustom()
220 void ParametersPlugin_WidgetParamsMgr::activateCustom()
222 FeaturePtr aFeature = feature();
223 DocumentPtr aDoc = aFeature->document();
224 int aNbParam = aDoc->size(ModelAPI_ResultParameter::group());
226 QTreeWidgetItem* aItem;
227 ResultParameterPtr aParam;
228 FeaturePtr aParamFeature;
229 for (int i = 0; i < aNbParam; i++) {
230 aObj = aDoc->object(ModelAPI_ResultParameter::group(), i);
231 aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aObj);
233 // Set parameter feature
234 aParamFeature = ModelAPI_Feature::feature(aParam);
237 aValues << aParamFeature->string(ParametersPlugin_Parameter::VARIABLE_ID())->value().c_str();
238 aValues << aParamFeature->string(ParametersPlugin_Parameter::EXPRESSION_ID())->value().c_str();
240 AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
241 aValues << QString::number(aValueAttribute->value());
243 aValues << aParamFeature->string(ParametersPlugin_Parameter::COMMENT_ID())->value().c_str();
245 aItem = new QTreeWidgetItem(aValues);
246 aItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
247 myParameters->addChild(aItem);
249 myParametersList.append(aParamFeature);
251 // Set features where the parameter is used
252 const std::set<std::shared_ptr<ModelAPI_Attribute>>& aRefs = aParam->data()->refsToMe();
253 std::set<std::shared_ptr<ModelAPI_Attribute> >::const_iterator aIt;
254 for(aIt = aRefs.cbegin(); aIt != aRefs.cend(); aIt++) {
255 std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
256 FeaturePtr aReferenced = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
257 if (aReferenced.get()) {
258 QStringList aValNames;
259 aValNames << aReferenced->data()->name().c_str();
261 AttributeDoublePtr aDouble = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(aAttr);
263 aValNames << aDouble->text().c_str();
264 aValNames << QString::number(aDouble->value());
266 AttributeIntegerPtr aInt = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>(aAttr);
268 aValNames << aInt->text().c_str();
269 aValNames << QString::number(aInt->value());
273 aItem = new QTreeWidgetItem(aValNames);
274 myFeatures->addChild(aItem);
279 myFeatures->setExpanded(true);
280 myParameters->setExpanded(true);
284 void ParametersPlugin_WidgetParamsMgr::rebuildFeatures()
286 myFeatures->takeChildren(); // Clear list
288 ResultParameterPtr aParam;
289 QTreeWidgetItem* aItem;
290 foreach(FeaturePtr aParameter, myParametersList) {
291 aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aParameter->firstResult());
292 const std::set<std::shared_ptr<ModelAPI_Attribute>>& aRefs = aParam->data()->refsToMe();
293 std::set<std::shared_ptr<ModelAPI_Attribute> >::const_iterator aIt;
294 for(aIt = aRefs.cbegin(); aIt != aRefs.cend(); aIt++) {
295 std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
296 FeaturePtr aReferenced = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
297 if (aReferenced.get()) {
298 QStringList aValNames;
299 aValNames << aReferenced->data()->name().c_str();
301 AttributeDoublePtr aDouble = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(aAttr);
303 aValNames << aDouble->text().c_str();
304 aValNames << QString::number(aDouble->value());
306 AttributeIntegerPtr aInt = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>(aAttr);
308 aValNames << aInt->text().c_str();
309 aValNames << QString::number(aInt->value());
313 aItem = new QTreeWidgetItem(aValNames);
314 myFeatures->addChild(aItem);
321 void ParametersPlugin_WidgetParamsMgr::onDoubleClick(const QModelIndex& theIndex)
323 if (myDelegate->isEditable(theIndex)) {
324 myTable->edit(theIndex);
325 myEditingIndex = theIndex;
329 void ParametersPlugin_WidgetParamsMgr::onCloseEditor(QWidget* theEditor,
330 QAbstractItemDelegate::EndEditHint theHint)
332 FeaturePtr aFeature = myParametersList.at(myEditingIndex.row());
333 QTreeWidgetItem* aItem = myParameters->child(myEditingIndex.row());
334 int aColumn = myEditingIndex.column();
335 QString aText = aItem->text(aColumn);
336 bool isModified = false;
341 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::VARIABLE_ID());
342 if (!aText.isEmpty()) {
343 if (hasName(aText)) {
344 myMessage = tr("Name %1 already exists.").arg(aText);
345 aItem->setText(Col_Name, aStringAttr->value().c_str());
346 QTimer::singleShot(50, this, SLOT(sendWarning()));
349 aStringAttr->setValue(aText.toStdString());
352 aItem->setText(Col_Name, aStringAttr->value().c_str());
358 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::EXPRESSION_ID());
359 if (!aText.isEmpty()) {
360 if (aText != aStringAttr->value().c_str()) {
361 aStringAttr->setValue(aText.toStdString());
365 aItem->setText(Col_Equation, aStringAttr->value().c_str());
371 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::COMMENT_ID());
372 aStringAttr->setValue(aText.toStdString());
380 Events_Loop* aLoop = Events_Loop::loop();
381 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
383 ResultParameterPtr aResult =
384 std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aFeature->firstResult());
386 AttributeDoublePtr aValueAttribute =
387 aResult->data()->real(ModelAPI_ResultParameter::VALUE());
388 aItem->setText(Col_Result, QString::number(aValueAttribute->value()));
390 myEditingIndex = QModelIndex();
391 updateFeaturesPart();
394 void ParametersPlugin_WidgetParamsMgr::updateFeaturesPart()
396 std::shared_ptr<ModelAPI_ResultParameter> aParam;
398 foreach(FeaturePtr aFeature, myParametersList) {
399 aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aFeature->firstResult());
400 const std::set<std::shared_ptr<ModelAPI_Attribute>>& aRefs = aParam->data()->refsToMe();
401 std::set<std::shared_ptr<ModelAPI_Attribute> >::const_iterator aIt;
402 for(aIt = aRefs.cbegin(); aIt != aRefs.cend(); aIt++) {
403 std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
404 FeaturePtr aReferenced = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
405 if (aReferenced.get()) {
406 QStringList aValNames;
407 aValNames << aReferenced->data()->name().c_str();
409 AttributeDoublePtr aDouble = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(aAttr);
411 aValNames << aDouble->text().c_str();
412 aValNames << QString::number(aDouble->value());
414 AttributeIntegerPtr aInt = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>(aAttr);
416 aValNames << aInt->text().c_str();
417 aValNames << QString::number(aInt->value());
421 QTreeWidgetItem* aItem = myFeatures->child(i++);
422 for(int i = 0; i < aValNames.count(); i++)
423 aItem->setText(i, aValNames.at(i));
429 FeaturePtr ParametersPlugin_WidgetParamsMgr::createParameter() const
431 SessionPtr aMgr = ModelAPI_Session::get();
432 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
434 FeaturePtr aFeature = aDoc->addFeature(ParametersPlugin_Parameter::ID());
435 if (aFeature.get()) {
436 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
437 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
443 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::createNewItem() const
449 QTreeWidgetItem* aItem = new QTreeWidgetItem(aValues);
450 aItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
455 void ParametersPlugin_WidgetParamsMgr::onAdd()
457 FeaturePtr aFeature = createParameter();
461 QTreeWidgetItem* aItem = createNewItem();
462 myParameters->addChild(aItem);
463 myParametersList.append(aFeature);
465 myTable->scrollToItem(aItem);
468 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::selectedItem() const
470 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
471 if (aItemsList.count() == 0)
474 QTreeWidgetItem* aCurrentItem = aItemsList.first();
475 if (aCurrentItem->parent() != myParameters)
482 void ParametersPlugin_WidgetParamsMgr::onInsert()
484 QTreeWidgetItem* aCurrentItem = selectedItem();
488 SessionPtr aMgr = ModelAPI_Session::get();
489 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
491 FeaturePtr aNewFeature = createParameter();
492 if (!aNewFeature.get())
495 QTreeWidgetItem* aItem = createNewItem();
496 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
497 if (aCurrentPos == 0) {
498 aDoc->moveFeature(aNewFeature, FeaturePtr());
500 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos - 1);
501 aDoc->moveFeature(aNewFeature, aCurFeature);
503 myParametersList.insert(aCurrentPos, aNewFeature);
504 myParameters->insertChild(aCurrentPos, aItem);
507 void ParametersPlugin_WidgetParamsMgr::onRemove()
509 QTreeWidgetItem* aCurrentItem = selectedItem();
513 SessionPtr aMgr = ModelAPI_Session::get();
514 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
516 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
517 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
519 QObjectPtrList anObjects;
520 anObjects.append(aCurFeature);
522 std::set<FeaturePtr> aDirectRefFeatures, aIndirectRefFeatures;
523 ModuleBase_Tools::findReferences(anObjects, aDirectRefFeatures, aIndirectRefFeatures);
525 bool doDeleteReferences = true;
526 if (ModuleBase_Tools::isDeleteFeatureWithReferences(anObjects, aDirectRefFeatures,
527 aIndirectRefFeatures, this, doDeleteReferences)) {
528 myParametersList.removeOne(aCurFeature);
529 myParameters->removeChild(aCurrentItem);
531 std::set<FeaturePtr> aFeaturesToDelete;
532 if (doDeleteReferences) {
533 aFeaturesToDelete = aDirectRefFeatures;
534 aFeaturesToDelete.insert(aIndirectRefFeatures.begin(), aIndirectRefFeatures.end());
536 aDoc->removeFeature(aCurFeature);
537 std::set<FeaturePtr>::const_iterator anIt = aFeaturesToDelete.begin(),
538 aLast = aFeaturesToDelete.end();
539 for (; anIt != aLast; anIt++) {
540 FeaturePtr aFeature = (*anIt);
541 DocumentPtr aDoc = aFeature->document();
542 aDoc->removeFeature(aFeature);
545 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
550 void ParametersPlugin_WidgetParamsMgr::onUp()
552 QTreeWidgetItem* aCurrentItem = selectedItem();
556 QString aName = aCurrentItem->text(0);
558 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
559 if (aCurrentPos == 0)
561 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
563 std::string aNm = aCurFeature->data()->name();
565 SessionPtr aMgr = ModelAPI_Session::get();
566 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
568 FeaturePtr aa = myParametersList.at(aCurrentPos - 1);
569 std::string aN = aa->data()->name();
571 if (aCurrentPos == 1)
572 aDoc->moveFeature(aCurFeature, FeaturePtr());
574 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos - 2));
576 myParametersList.removeOne(aCurFeature);
577 myParametersList.insert(aCurrentPos - 1, aCurFeature);
579 myParameters->removeChild(aCurrentItem);
580 myParameters->insertChild(aCurrentPos - 1, aCurrentItem);
582 selectItemScroll(aCurrentItem);
585 void ParametersPlugin_WidgetParamsMgr::onDown()
587 QTreeWidgetItem* aCurrentItem = selectedItem();
591 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
592 if (aCurrentPos == (myParametersList.count() - 1))
594 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
596 SessionPtr aMgr = ModelAPI_Session::get();
597 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
598 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos + 1));
600 myParametersList.removeOne(aCurFeature);
601 myParametersList.insert(aCurrentPos + 1, aCurFeature);
603 myParameters->removeChild(aCurrentItem);
604 myParameters->insertChild(aCurrentPos + 1, aCurrentItem);
606 selectItemScroll(aCurrentItem);
610 bool ParametersPlugin_WidgetParamsMgr::hasName(const QString& theName) const
612 foreach(FeaturePtr aFeature, myParametersList) {
613 if (aFeature->data()->name() == theName.toStdString())
619 void ParametersPlugin_WidgetParamsMgr::sendWarning()
621 QMessageBox::warning(this, tr("Warning"), myMessage);
624 void ParametersPlugin_WidgetParamsMgr::onSelectionChanged()
626 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
627 bool isParameter = false;
628 foreach(QTreeWidgetItem* aItem, aItemsList) {
629 if (aItem->parent() == myParameters) {
634 myInsertBtn->setEnabled(isParameter);
635 myRemoveBtn->setEnabled(isParameter);
636 myUpBtn->setEnabled(isParameter);
637 myDownBtn->setEnabled(isParameter);