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;
58 QModelIndex editIndex() const { return myEditingIdx; }
61 mutable QModelIndex myEditingIdx;
64 bool ParametersPlugin_ItemDelegate::isEditable(const QModelIndex& theIndex) const
66 QModelIndex aParent = theIndex.parent();
67 if (aParent.isValid() && (aParent.row() == 0)) {
68 if (theIndex.column() == 2)
75 void ParametersPlugin_ItemDelegate::paint(QPainter* painter,
76 const QStyleOptionViewItem& option,
77 const QModelIndex& index ) const
79 QBrush aBrush = painter->brush();
80 QPen aPen = painter->pen();
81 if (!isEditable(index))
82 painter->setBrush(Qt::lightGray);
84 painter->setPen(Qt::darkGray);
85 painter->drawRect(option.rect);
86 painter->setPen(aPen);
88 QStyledItemDelegate::paint(painter, option, index);
89 painter->setBrush(aBrush);
92 QWidget* ParametersPlugin_ItemDelegate::createEditor(QWidget* parent,
93 const QStyleOptionViewItem& option,
94 const QModelIndex& index) const
97 return QStyledItemDelegate::createEditor(parent, option, index);
101 /////////////////////////////////////////////////////////////////////////////////////////////////
103 ParametersPlugin_WidgetParamsMgr::ParametersPlugin_WidgetParamsMgr(QWidget* theParent, const Config_WidgetAPI* theData)
104 : ModuleBase_ModelWidget(theParent, theData)
106 QVBoxLayout* aLayout = new QVBoxLayout(this);
108 myTable = new QTreeWidget(this);
109 myTable->setColumnCount(4);
110 QStringList aHeaders;
111 aHeaders << tr("Name") << tr("Equation") << tr("Result") << tr("Comment");
112 myTable->setHeaderLabels(aHeaders);
113 myTable->setColumnWidth(Col_Name, 200);
114 myTable->setColumnWidth(Col_Equation, 100);
115 myTable->setColumnWidth(Col_Result, 80);
116 myTable->setColumnWidth(Col_Comment, 200);
117 myTable->setMinimumWidth(600);
118 myTable->setEditTriggers(QAbstractItemView::NoEditTriggers);
119 myTable->setSelectionMode(QAbstractItemView::SingleSelection);
121 connect(myTable, SIGNAL(doubleClicked(const QModelIndex&)),
122 SLOT(onDoubleClick(const QModelIndex&)));
123 connect(myTable, SIGNAL(itemSelectionChanged()), SLOT(onSelectionChanged()));
125 myDelegate = new ParametersPlugin_ItemDelegate(myTable);
126 connect(myDelegate, SIGNAL(closeEditor(QWidget*, QAbstractItemDelegate::EndEditHint)),
127 SLOT(onCloseEditor(QWidget*, QAbstractItemDelegate::EndEditHint)));
129 myTable->setItemDelegate(myDelegate);
130 aLayout->addWidget(myTable);
134 aNames<<tr("Parameters");
135 myParameters = new QTreeWidgetItem(aNames);
136 myTable->addTopLevelItem(myParameters);
139 aNames<<tr("Features");
140 myFeatures = new QTreeWidgetItem(aNames);
141 myTable->addTopLevelItem(myFeatures);
143 QHBoxLayout* aBtnLayout = new QHBoxLayout(this);
145 myUpBtn = new QToolButton(this);
146 myUpBtn->setArrowType(Qt::UpArrow);
147 connect(myUpBtn, SIGNAL(clicked(bool)), SLOT(onUp()));
148 aBtnLayout->addWidget(myUpBtn);
150 myDownBtn = new QToolButton(this);
151 myDownBtn->setArrowType(Qt::DownArrow);
152 connect(myDownBtn, SIGNAL(clicked(bool)), SLOT(onDown()));
153 aBtnLayout->addWidget(myDownBtn);
155 aBtnLayout->addStretch();
157 myAddBtn = new QPushButton(tr("Add"), this);
158 connect(myAddBtn, SIGNAL(clicked(bool)), SLOT(onAdd()));
159 aBtnLayout->addWidget(myAddBtn);
161 myInsertBtn = new QPushButton(tr("Insert"), this);
162 connect(myInsertBtn, SIGNAL(clicked(bool)), SLOT(onInsert()));
163 aBtnLayout->addWidget(myInsertBtn);
165 myRemoveBtn = new QPushButton(tr("Remove"), this);
166 connect(myRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemove()));
167 aBtnLayout->addWidget(myRemoveBtn);
169 aLayout->addLayout(aBtnLayout);
171 onSelectionChanged();
174 QList<QWidget*> ParametersPlugin_WidgetParamsMgr::getControls() const
176 QList<QWidget*> aList;
181 void ParametersPlugin_WidgetParamsMgr::selectItemScroll(QTreeWidgetItem* aItem)
183 myTable->clearSelection();
184 QModelIndex aParent = myTable->model()->index(0, 0);
185 int aChildIdx = myParameters->indexOfChild(aItem);
186 QModelIndex aIndex = myTable->model()->index(aChildIdx, Col_Name, aParent);
187 myTable->selectionModel()->select(aIndex,
188 QItemSelectionModel::SelectCurrent | QItemSelectionModel::Rows);
189 myTable->scrollToItem(aItem);
193 bool ParametersPlugin_WidgetParamsMgr::storeValueCustom()
195 ParametersPlugin_ExpressionValidator aValidator;
196 std::list<std::string> aArgs;
197 std::string aAttrId = ParametersPlugin_Parameter::VARIABLE_ID();
200 foreach(FeaturePtr aFeature, myParametersList) {
201 if (!aValidator.isValid(aFeature->attribute(aAttrId), aArgs, aErr)) {
202 QMessageBox::warning(this, tr("Warning"), aErr.c_str());
203 selectItemScroll(myParameters->child(aId));
211 bool ParametersPlugin_WidgetParamsMgr::restoreValueCustom()
216 void ParametersPlugin_WidgetParamsMgr::activateCustom()
218 updateParametersFeatures();
219 updateParametersPart();
220 updateFeaturesPart();
222 myFeatures->setExpanded(true);
223 myParameters->setExpanded(true);
226 void ParametersPlugin_WidgetParamsMgr::updateParametersFeatures()
228 myParametersList.clear();
229 FeaturePtr aFeature = feature();
230 DocumentPtr aDoc = aFeature->document();
231 int aNbParam = aDoc->size(ModelAPI_ResultParameter::group());
233 FeaturePtr aParamFeature;
234 for (int i = 0; i < aNbParam; i++) {
235 aObj = aDoc->object(ModelAPI_ResultParameter::group(), i);
236 aParamFeature = ModelAPI_Feature::feature(aObj);
237 if (aParamFeature.get() && (aParamFeature->getKind() == ParametersPlugin_Parameter::ID())) {
238 myParametersList.append(aParamFeature);
243 void ParametersPlugin_WidgetParamsMgr::updateFeaturesPart()
245 updateItem(myFeatures, featuresItems(myParametersList));
248 void ParametersPlugin_WidgetParamsMgr::updateParametersPart()
250 updateItem(myParameters, parametersItems(myParametersList));
254 QList<QStringList> ParametersPlugin_WidgetParamsMgr::
255 featuresItems(const QList<FeaturePtr>& theFeatures) const
257 QList<QStringList> aItemsList;
258 ResultParameterPtr aParam;
259 foreach(FeaturePtr aParameter, theFeatures) {
260 aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aParameter->firstResult());
261 const std::set<std::shared_ptr<ModelAPI_Attribute>>& aRefs = aParam->data()->refsToMe();
262 std::set<std::shared_ptr<ModelAPI_Attribute> >::const_iterator aIt;
263 for(aIt = aRefs.cbegin(); aIt != aRefs.cend(); aIt++) {
264 std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
265 FeaturePtr aReferenced = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
266 if (aReferenced.get()) {
267 if (aReferenced->getKind() == ParametersPlugin_Parameter::ID()) {
268 // Find referenced feature Recursive
269 QList<FeaturePtr> aList;
270 aList.append(aReferenced);
271 QList<QStringList> aItems = featuresItems(aList);
272 aItemsList.append(aItems);
274 QStringList aValNames;
275 aValNames << aReferenced->data()->name().c_str();
277 AttributeDoublePtr aDouble = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(aAttr);
279 aValNames << aDouble->text().c_str();
280 aValNames << QString::number(aDouble->value());
282 AttributeIntegerPtr aInt = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>(aAttr);
284 aValNames << aInt->text().c_str();
285 aValNames << QString::number(aInt->value());
288 aItemsList.append(aValNames);
297 QList<QStringList> ParametersPlugin_WidgetParamsMgr::
298 parametersItems(const QList<FeaturePtr>& theFeatures) const
300 QList<QStringList> aItemsList;
301 foreach(FeaturePtr aParameter, theFeatures) {
302 ResultPtr aParam = aParameter->firstResult();
304 aValues << aParameter->string(ParametersPlugin_Parameter::VARIABLE_ID())->value().c_str();
305 aValues << aParameter->string(ParametersPlugin_Parameter::EXPRESSION_ID())->value().c_str();
307 AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
308 aValues << QString::number(aValueAttribute->value());
310 aValues << aParameter->string(ParametersPlugin_Parameter::COMMENT_ID())->value().c_str();
311 aItemsList.append(aValues);
317 void ParametersPlugin_WidgetParamsMgr::onDoubleClick(const QModelIndex& theIndex)
319 if (myDelegate->isEditable(theIndex)) {
320 myTable->edit(theIndex);
324 void ParametersPlugin_WidgetParamsMgr::onCloseEditor(QWidget* theEditor,
325 QAbstractItemDelegate::EndEditHint theHint)
327 FeaturePtr aFeature = myParametersList.at(myDelegate->editIndex().row());
328 QTreeWidgetItem* aItem = myParameters->child(myDelegate->editIndex().row());
329 int aColumn = myDelegate->editIndex().column();
330 QString aText = aItem->text(aColumn);
331 bool isModified = false;
336 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::VARIABLE_ID());
337 if (!aText.isEmpty()) {
338 if (hasName(aText)) {
339 myMessage = tr("Name %1 already exists.").arg(aText);
340 if (aStringAttr->value().length() > 0)
341 aItem->setText(Col_Name, aStringAttr->value().c_str());
343 aItem->setText(Col_Name, NoName);
344 QTimer::singleShot(50, this, SLOT(sendWarning()));
347 aStringAttr->setValue(aText.toStdString());
350 aItem->setText(Col_Name, aStringAttr->value().c_str());
356 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::EXPRESSION_ID());
357 if (!aText.isEmpty()) {
358 if (aText != aStringAttr->value().c_str()) {
359 aStringAttr->setValue(aText.toStdString());
363 aItem->setText(Col_Equation, aStringAttr->value().c_str());
369 AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::COMMENT_ID());
370 aStringAttr->setValue(aText.toStdString());
378 Events_Loop* aLoop = Events_Loop::loop();
379 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
380 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
381 aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
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 if (aColumn == Col_Equation)
391 updateParametersPart();
392 updateFeaturesPart();
395 void ParametersPlugin_WidgetParamsMgr::updateItem(QTreeWidgetItem* theItem,
396 const QList<QStringList>& theFeaturesList)
398 if (theFeaturesList.count() != theItem->childCount()) {
399 if (theItem->childCount() < theFeaturesList.count()) {
400 while (theItem->childCount() != theFeaturesList.count())
401 theItem->addChild(createNewItem());
403 while (theItem->childCount() != theFeaturesList.count())
404 theItem->removeChild(theItem->child(theItem->childCount() - 1));
408 foreach(QStringList aFeature, theFeaturesList) {
410 foreach(QString aText, aFeature) {
411 if (aText.length() > 0)
412 theItem->child(i)->setText(aCol, aText);
419 FeaturePtr ParametersPlugin_WidgetParamsMgr::createParameter() const
421 SessionPtr aMgr = ModelAPI_Session::get();
422 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
424 FeaturePtr aFeature = aDoc->addFeature(ParametersPlugin_Parameter::ID());
425 if (aFeature.get()) {
426 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
427 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
433 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::createNewItem() const
439 QTreeWidgetItem* aItem = new QTreeWidgetItem(aValues);
440 aItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
445 void ParametersPlugin_WidgetParamsMgr::onAdd()
447 FeaturePtr aFeature = createParameter();
451 QTreeWidgetItem* aItem = createNewItem();
452 if (aFeature->name().length() > 0)
453 aItem->setText(Col_Name, aFeature->name().c_str());
454 myParameters->addChild(aItem);
455 myParametersList.append(aFeature);
457 myTable->scrollToItem(aItem);
460 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::selectedItem() const
462 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
463 if (aItemsList.count() == 0)
466 QTreeWidgetItem* aCurrentItem = aItemsList.first();
467 if (aCurrentItem->parent() != myParameters)
474 void ParametersPlugin_WidgetParamsMgr::onInsert()
476 QTreeWidgetItem* aCurrentItem = selectedItem();
480 SessionPtr aMgr = ModelAPI_Session::get();
481 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
483 FeaturePtr aNewFeature = createParameter();
484 if (!aNewFeature.get())
487 QTreeWidgetItem* aItem = createNewItem();
488 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
489 if (aCurrentPos == 0) {
490 aDoc->moveFeature(aNewFeature, FeaturePtr());
492 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos - 1);
493 aDoc->moveFeature(aNewFeature, aCurFeature);
495 myParametersList.insert(aCurrentPos, aNewFeature);
496 myParameters->insertChild(aCurrentPos, aItem);
499 void ParametersPlugin_WidgetParamsMgr::onRemove()
501 QTreeWidgetItem* aCurrentItem = selectedItem();
505 SessionPtr aMgr = ModelAPI_Session::get();
506 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
508 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
509 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
511 QObjectPtrList anObjects;
512 anObjects.append(aCurFeature);
514 std::set<FeaturePtr> aDirectRefFeatures, aIndirectRefFeatures;
515 ModuleBase_Tools::findReferences(anObjects, aDirectRefFeatures, aIndirectRefFeatures);
517 bool doDeleteReferences = true;
518 if (ModuleBase_Tools::isDeleteFeatureWithReferences(anObjects, aDirectRefFeatures,
519 aIndirectRefFeatures, this, doDeleteReferences)) {
520 myParametersList.removeOne(aCurFeature);
521 myParameters->removeChild(aCurrentItem);
523 std::set<FeaturePtr> aFeaturesToDelete;
524 if (doDeleteReferences) {
525 aFeaturesToDelete = aDirectRefFeatures;
526 aFeaturesToDelete.insert(aIndirectRefFeatures.begin(), aIndirectRefFeatures.end());
528 aDoc->removeFeature(aCurFeature);
529 std::set<FeaturePtr>::const_iterator anIt = aFeaturesToDelete.begin(),
530 aLast = aFeaturesToDelete.end();
531 for (; anIt != aLast; anIt++) {
532 FeaturePtr aFeature = (*anIt);
533 DocumentPtr aDoc = aFeature->document();
534 aDoc->removeFeature(aFeature);
537 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
538 Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
539 updateParametersFeatures();
540 updateFeaturesPart();
541 updateParametersPart();
545 void ParametersPlugin_WidgetParamsMgr::onUp()
547 QTreeWidgetItem* aCurrentItem = selectedItem();
551 QString aName = aCurrentItem->text(0);
553 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
554 if (aCurrentPos == 0)
556 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
558 std::string aNm = aCurFeature->data()->name();
560 SessionPtr aMgr = ModelAPI_Session::get();
561 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
563 FeaturePtr aa = myParametersList.at(aCurrentPos - 1);
564 std::string aN = aa->data()->name();
566 if (aCurrentPos == 1)
567 aDoc->moveFeature(aCurFeature, FeaturePtr());
569 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos - 2));
571 myParametersList.removeOne(aCurFeature);
572 myParametersList.insert(aCurrentPos - 1, aCurFeature);
574 myParameters->removeChild(aCurrentItem);
575 myParameters->insertChild(aCurrentPos - 1, aCurrentItem);
577 selectItemScroll(aCurrentItem);
579 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED));
580 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
581 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
582 //updateParametersFeatures();
583 //updateParametersPart();
584 //updateFeaturesPart();
587 void ParametersPlugin_WidgetParamsMgr::onDown()
589 QTreeWidgetItem* aCurrentItem = selectedItem();
593 int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
594 if (aCurrentPos == (myParametersList.count() - 1))
596 FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
598 SessionPtr aMgr = ModelAPI_Session::get();
599 std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
600 aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos + 1));
602 myParametersList.removeOne(aCurFeature);
603 myParametersList.insert(aCurrentPos + 1, aCurFeature);
605 myParameters->removeChild(aCurrentItem);
606 myParameters->insertChild(aCurrentPos + 1, aCurrentItem);
608 selectItemScroll(aCurrentItem);
610 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED));
611 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED));
612 //Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
613 //updateParametersFeatures();
614 //updateParametersPart();
615 //updateFeaturesPart();
619 bool ParametersPlugin_WidgetParamsMgr::hasName(const QString& theName) const
621 int aCurrent = myDelegate->editIndex().row();
623 foreach(FeaturePtr aFeature, myParametersList) {
624 if ((i != aCurrent) && (aFeature->data()->name() == theName.toStdString()))
631 void ParametersPlugin_WidgetParamsMgr::sendWarning()
633 QMessageBox::warning(this, tr("Warning"), myMessage);
636 void ParametersPlugin_WidgetParamsMgr::onSelectionChanged()
638 QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
639 bool isParameter = false;
640 foreach(QTreeWidgetItem* aItem, aItemsList) {
641 if (aItem->parent() == myParameters) {
646 myInsertBtn->setEnabled(isParameter);
647 myRemoveBtn->setEnabled(isParameter);
648 myUpBtn->setEnabled(isParameter);
649 myDownBtn->setEnabled(isParameter);