]> SALOME platform Git repositories - modules/shaper.git/blob - src/ParametersPlugin/ParametersPlugin_WidgetParamsMgr.cpp
Salome HOME
Optimization of the widget code
[modules/shaper.git] / src / ParametersPlugin / ParametersPlugin_WidgetParamsMgr.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
2
3 // File:        ParametersPlugin_WidgetParamsMgr.cpp
4 // Created:     11 Apr 2016
5 // Author:      Vitaly SMETANNIKOV
6
7 #include "ParametersPlugin_WidgetParamsMgr.h"
8 #include "ParametersPlugin_Parameter.h"
9 #include "ParametersPlugin_Validators.h"
10
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>
18
19 #include <ModuleBase_Tools.h>
20
21 #include <Events_Loop.h>
22
23 #include <QLayout>
24 #include <QTreeWidget>
25 #include <QPushButton>
26 #include <QToolButton>
27 #include <QStyledItemDelegate>
28 #include <QPainter>
29 #include <QMessageBox>
30 #include <QTimer>
31
32 enum ColumnType {
33   Col_Name,
34   Col_Equation,
35   Col_Result,
36   Col_Comment
37 };
38
39 const char* NoName = "<NoName>";
40 const char* NoValue = "<NoValue>";
41
42 class ParametersPlugin_ItemDelegate : public QStyledItemDelegate
43 {
44 public:
45   ParametersPlugin_ItemDelegate(QObject* thaParent) : 
46       QStyledItemDelegate(thaParent) {}
47
48   virtual void paint(QPainter* painter, 
49     const QStyleOptionViewItem& option, 
50     const QModelIndex& index ) const;
51   
52   //virtual QWidget* createEditor(QWidget* parent, 
53   //                              const QStyleOptionViewItem& option, 
54   //                              const QModelIndex& index) const;
55
56   bool isEditable(const QModelIndex& theIndex) const;
57 };
58
59 bool ParametersPlugin_ItemDelegate::isEditable(const QModelIndex& theIndex) const
60 {
61   QModelIndex aParent = theIndex.parent();
62   if (aParent.isValid() && (aParent.row() == 0)) {
63     if (theIndex.column() == 2)
64       return false;
65   } else
66     return false;
67   return true;
68 }
69
70 void ParametersPlugin_ItemDelegate::paint(QPainter* painter, 
71                                           const QStyleOptionViewItem& option, 
72                                           const QModelIndex& index ) const
73 {
74   QBrush aBrush = painter->brush();
75   QPen aPen = painter->pen();
76   if (!isEditable(index))
77     painter->setBrush(Qt::lightGray);
78
79   painter->setPen(Qt::darkGray);
80   painter->drawRect(option.rect);
81   painter->setPen(aPen);
82   
83   //QString aText = index.data().toString();
84   //if ((aText == NoName) || (aText == NoValue))
85   //  painter->setPen(Qt::red);
86
87   QStyledItemDelegate::paint(painter, option, index);
88
89   //painter->setPen(aPen);
90   painter->setBrush(aBrush);
91 }
92
93
94 //QWidget* ParametersPlugin_ItemDelegate::createEditor(QWidget* parent, 
95 //                                                    const QStyleOptionViewItem& option, 
96 //                                                    const QModelIndex& index) const
97 //{
98 //  QWidget* aWgt = QStyledItemDelegate::createEditor(parent, option, index); 
99 //  aWgt->setMinimumSize(option.rect.width() - option.decorationSize.width(), 
100 //                       option.rect.height());
101 //  return aWgt;
102 //}
103
104
105 /////////////////////////////////////////////////////////////////////////////////////////////////
106
107 ParametersPlugin_WidgetParamsMgr::ParametersPlugin_WidgetParamsMgr(QWidget* theParent, const Config_WidgetAPI* theData)
108   : ModuleBase_ModelWidget(theParent, theData)
109 {
110   QVBoxLayout* aLayout = new QVBoxLayout(this);
111
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);
124
125   connect(myTable, SIGNAL(doubleClicked(const QModelIndex&)),
126           SLOT(onDoubleClick(const QModelIndex&)));
127   connect(myTable, SIGNAL(itemSelectionChanged()), SLOT(onSelectionChanged()));
128
129   myDelegate = new ParametersPlugin_ItemDelegate(myTable);
130   connect(myDelegate, SIGNAL(closeEditor(QWidget*, QAbstractItemDelegate::EndEditHint)), 
131           SLOT(onCloseEditor(QWidget*, QAbstractItemDelegate::EndEditHint)));
132
133   myTable->setItemDelegate(myDelegate);
134   aLayout->addWidget(myTable);
135
136   // Define root nodes
137   QStringList aNames;
138   aNames<<tr("Parameters");
139   myParameters = new QTreeWidgetItem(aNames);
140   myTable->addTopLevelItem(myParameters);
141
142   aNames.clear();
143   aNames<<tr("Features");
144   myFeatures = new QTreeWidgetItem(aNames);
145   myTable->addTopLevelItem(myFeatures);
146
147   QHBoxLayout* aBtnLayout = new QHBoxLayout(this);
148
149   myUpBtn = new QToolButton(this);
150   myUpBtn->setArrowType(Qt::UpArrow);
151   connect(myUpBtn, SIGNAL(clicked(bool)), SLOT(onUp()));
152   aBtnLayout->addWidget(myUpBtn);
153
154   myDownBtn = new QToolButton(this);
155   myDownBtn->setArrowType(Qt::DownArrow);
156   connect(myDownBtn, SIGNAL(clicked(bool)), SLOT(onDown()));
157   aBtnLayout->addWidget(myDownBtn);
158
159   aBtnLayout->addStretch();
160
161   myAddBtn = new QPushButton(tr("Add"), this);
162   connect(myAddBtn, SIGNAL(clicked(bool)), SLOT(onAdd()));
163   aBtnLayout->addWidget(myAddBtn);
164
165   myInsertBtn = new QPushButton(tr("Insert"), this);
166   connect(myInsertBtn, SIGNAL(clicked(bool)), SLOT(onInsert()));
167   aBtnLayout->addWidget(myInsertBtn);
168
169   myRemoveBtn = new QPushButton(tr("Remove"), this);
170   connect(myRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemove()));
171   aBtnLayout->addWidget(myRemoveBtn);
172
173   aLayout->addLayout(aBtnLayout);
174
175   onSelectionChanged();
176 }
177
178 QList<QWidget*> ParametersPlugin_WidgetParamsMgr::getControls() const
179 {
180   QList<QWidget*> aList;
181
182   return aList;
183 }
184
185 void ParametersPlugin_WidgetParamsMgr::selectItemScroll(QTreeWidgetItem* aItem)
186 {
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);
194 }
195
196
197 bool ParametersPlugin_WidgetParamsMgr::storeValueCustom()
198 {
199   ParametersPlugin_ExpressionValidator aValidator;
200   std::list<std::string> aArgs;
201   std::string aAttrId = ParametersPlugin_Parameter::VARIABLE_ID();
202   std::string aErr;
203   int aId = 0;
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));
208       return false;
209     }
210     aId++;
211   }
212   return true;
213 }
214
215 bool ParametersPlugin_WidgetParamsMgr::restoreValueCustom()
216 {
217   return true;
218 }
219
220 void ParametersPlugin_WidgetParamsMgr::activateCustom()
221 {
222   FeaturePtr aFeature = feature();
223   DocumentPtr aDoc = aFeature->document();
224   int aNbParam = aDoc->size(ModelAPI_ResultParameter::group());
225   ObjectPtr aObj;
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);
232     if (aParam.get()) {
233       // Set parameter feature
234       aParamFeature = ModelAPI_Feature::feature(aParam);
235
236       QStringList aValues;
237       aValues << aParamFeature->string(ParametersPlugin_Parameter::VARIABLE_ID())->value().c_str();
238       aValues << aParamFeature->string(ParametersPlugin_Parameter::EXPRESSION_ID())->value().c_str();
239
240       AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
241       aValues << QString::number(aValueAttribute->value());
242
243       aValues << aParamFeature->string(ParametersPlugin_Parameter::COMMENT_ID())->value().c_str();
244
245       aItem = new QTreeWidgetItem(aValues);
246       aItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
247       myParameters->addChild(aItem);
248
249       myParametersList.append(aParamFeature);
250     }
251   }
252   updateFeaturesPart();
253
254   myFeatures->setExpanded(true);
255   myParameters->setExpanded(true);
256 }
257
258
259 QList<QStringList> ParametersPlugin_WidgetParamsMgr::
260   featuresItems(const QList<FeaturePtr>& theFeatures) const
261 {
262   QList<QStringList> aItemsList;
263   ResultParameterPtr aParam;
264   foreach(FeaturePtr aParameter, theFeatures) {
265     aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aParameter->firstResult());
266     const std::set<std::shared_ptr<ModelAPI_Attribute>>& aRefs = aParam->data()->refsToMe();
267     std::set<std::shared_ptr<ModelAPI_Attribute> >::const_iterator aIt;
268     for(aIt = aRefs.cbegin(); aIt != aRefs.cend(); aIt++) {
269       std::shared_ptr<ModelAPI_Attribute> aAttr = (*aIt);
270       FeaturePtr aReferenced = std::dynamic_pointer_cast<ModelAPI_Feature>(aAttr->owner());
271       if (aReferenced.get()) {
272         QStringList aValNames;
273         aValNames << aReferenced->data()->name().c_str();
274
275         AttributeDoublePtr aDouble = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(aAttr);
276         if (aDouble.get()) {
277           aValNames << aDouble->text().c_str();
278           aValNames << QString::number(aDouble->value());
279         } else {
280           AttributeIntegerPtr aInt = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>(aAttr);
281           if (aInt.get()) {
282             aValNames << aInt->text().c_str();
283             aValNames << QString::number(aInt->value());
284           }
285         }
286         aItemsList.append(aValNames);
287       }
288     }
289   }
290   return aItemsList;
291 }
292
293
294 void ParametersPlugin_WidgetParamsMgr::onDoubleClick(const QModelIndex& theIndex)
295 {
296   if (myDelegate->isEditable(theIndex)) {
297     myTable->edit(theIndex);
298     myEditingIndex = theIndex;
299   }
300 }
301
302 void ParametersPlugin_WidgetParamsMgr::onCloseEditor(QWidget* theEditor, 
303                                                      QAbstractItemDelegate::EndEditHint theHint)
304 {
305   FeaturePtr aFeature = myParametersList.at(myEditingIndex.row());
306   QTreeWidgetItem* aItem = myParameters->child(myEditingIndex.row());
307   int aColumn = myEditingIndex.column();
308   QString aText = aItem->text(aColumn);
309   bool isModified = false;
310
311   switch (aColumn) {
312   case Col_Name:
313     {
314       AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::VARIABLE_ID());
315       if (!aText.isEmpty()) {
316         if (hasName(aText)) {
317           myMessage = tr("Name %1 already exists.").arg(aText);
318           aItem->setText(Col_Name, aStringAttr->value().c_str());
319           QTimer::singleShot(50, this, SLOT(sendWarning()));
320           return;
321         }
322         aStringAttr->setValue(aText.toStdString());
323         isModified = true;
324       } else {
325         aItem->setText(Col_Name, aStringAttr->value().c_str());
326       }
327     }
328     break;
329   case Col_Equation:
330     {
331       AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::EXPRESSION_ID());
332       if (!aText.isEmpty()) {
333         if (aText != aStringAttr->value().c_str()) {
334           aStringAttr->setValue(aText.toStdString());
335           isModified = true;
336         }
337       } else {
338         aItem->setText(Col_Equation, aStringAttr->value().c_str());
339       }
340     }
341     break;
342   case Col_Comment:
343     {
344       AttributeStringPtr aStringAttr = aFeature->string(ParametersPlugin_Parameter::COMMENT_ID());
345       aStringAttr->setValue(aText.toStdString());
346       isModified = true;
347     }
348     break;
349   }
350
351   if (!isModified)
352     return;
353   Events_Loop* aLoop = Events_Loop::loop();
354   aLoop->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
355
356   ResultParameterPtr aResult = 
357     std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aFeature->firstResult());
358   if (aResult.get()) {
359     AttributeDoublePtr aValueAttribute = 
360       aResult->data()->real(ModelAPI_ResultParameter::VALUE());
361     aItem->setText(Col_Result, QString::number(aValueAttribute->value()));
362   }
363   myEditingIndex = QModelIndex();
364   updateFeaturesPart();
365 }
366
367 void ParametersPlugin_WidgetParamsMgr::updateFeaturesPart()
368 {
369   QList<QStringList> aFeaturesList = featuresItems(myParametersList);
370   if (aFeaturesList.count() != myFeatures->childCount()) {
371     if (myFeatures->childCount()  < aFeaturesList.count()) {
372       while (myFeatures->childCount() != aFeaturesList.count()) 
373         myFeatures->addChild(new QTreeWidgetItem());
374     } else {
375       while (myFeatures->childCount() != aFeaturesList.count()) 
376         myFeatures->removeChild(myFeatures->child(myFeatures->childCount() - 1));
377     }
378   }
379   int i = 0;
380   foreach(QStringList aFeature, aFeaturesList) {
381     int aCol = 0;
382     foreach(QString aText, aFeature) {
383       myFeatures->child(i)->setText(aCol, aText);
384       aCol++;
385     }
386     i++;
387   }
388 }
389
390 FeaturePtr ParametersPlugin_WidgetParamsMgr::createParameter() const
391 {
392   SessionPtr aMgr = ModelAPI_Session::get();
393   std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
394
395   FeaturePtr aFeature = aDoc->addFeature(ParametersPlugin_Parameter::ID());
396   if (aFeature.get()) {
397     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
398     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
399   }
400   return aFeature;
401 }
402
403
404 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::createNewItem() const
405 {
406   QStringList aValues;
407   aValues << NoName;
408   aValues << NoValue;
409
410   QTreeWidgetItem* aItem = new QTreeWidgetItem(aValues);
411   aItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
412   return aItem;
413 }
414
415
416 void ParametersPlugin_WidgetParamsMgr::onAdd()
417 {
418   FeaturePtr aFeature = createParameter();
419   if (!aFeature.get())
420     return;
421
422   QTreeWidgetItem* aItem = createNewItem();
423   if (aFeature->name().length() > 0)
424     aItem->setText(Col_Name, aFeature->name().c_str());
425   myParameters->addChild(aItem);
426   myParametersList.append(aFeature);
427       
428   myTable->scrollToItem(aItem);
429 }
430
431 QTreeWidgetItem* ParametersPlugin_WidgetParamsMgr::selectedItem() const
432 {
433   QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
434   if (aItemsList.count() == 0)
435     return 0;
436
437   QTreeWidgetItem* aCurrentItem = aItemsList.first();
438   if (aCurrentItem->parent() != myParameters)
439     return 0;
440
441   return aCurrentItem;
442 }
443
444
445 void ParametersPlugin_WidgetParamsMgr::onInsert()
446 {
447   QTreeWidgetItem* aCurrentItem = selectedItem();
448   if (!aCurrentItem)
449     return;
450
451   SessionPtr aMgr = ModelAPI_Session::get();
452   std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
453
454   FeaturePtr aNewFeature = createParameter();
455   if (!aNewFeature.get())
456     return;
457
458   QTreeWidgetItem* aItem = createNewItem();
459   int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
460   if (aCurrentPos == 0) {
461     aDoc->moveFeature(aNewFeature, FeaturePtr());
462   } else {
463     FeaturePtr aCurFeature = myParametersList.at(aCurrentPos - 1);
464     aDoc->moveFeature(aNewFeature, aCurFeature);
465   }
466   myParametersList.insert(aCurrentPos, aNewFeature);
467   myParameters->insertChild(aCurrentPos, aItem);
468 }
469
470 void ParametersPlugin_WidgetParamsMgr::onRemove()
471 {
472   QTreeWidgetItem* aCurrentItem = selectedItem();
473   if (!aCurrentItem)
474     return;
475
476   SessionPtr aMgr = ModelAPI_Session::get();
477   std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
478
479   int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
480   FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
481
482   QObjectPtrList anObjects;
483   anObjects.append(aCurFeature);
484
485   std::set<FeaturePtr> aDirectRefFeatures, aIndirectRefFeatures;
486   ModuleBase_Tools::findReferences(anObjects, aDirectRefFeatures, aIndirectRefFeatures);
487
488   bool doDeleteReferences = true;
489   if (ModuleBase_Tools::isDeleteFeatureWithReferences(anObjects, aDirectRefFeatures, 
490       aIndirectRefFeatures, this, doDeleteReferences)) {
491     myParametersList.removeOne(aCurFeature);
492     myParameters->removeChild(aCurrentItem);
493
494     std::set<FeaturePtr> aFeaturesToDelete;
495     if (doDeleteReferences) {
496       aFeaturesToDelete = aDirectRefFeatures;
497       aFeaturesToDelete.insert(aIndirectRefFeatures.begin(), aIndirectRefFeatures.end());
498     }
499     aDoc->removeFeature(aCurFeature);
500     std::set<FeaturePtr>::const_iterator anIt = aFeaturesToDelete.begin(),
501                                          aLast = aFeaturesToDelete.end();
502     for (; anIt != aLast; anIt++) {
503       FeaturePtr aFeature = (*anIt);
504       DocumentPtr aDoc = aFeature->document();
505       aDoc->removeFeature(aFeature);
506     }
507
508     Events_Loop::loop()->flush(Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY));
509     updateFeaturesPart();
510   }
511 }
512
513 void ParametersPlugin_WidgetParamsMgr::onUp()
514 {
515   QTreeWidgetItem* aCurrentItem = selectedItem();
516   if (!aCurrentItem)
517     return;
518
519   QString aName = aCurrentItem->text(0);
520
521   int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
522   if (aCurrentPos == 0)
523     return;
524   FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
525
526   std::string aNm = aCurFeature->data()->name();
527
528   SessionPtr aMgr = ModelAPI_Session::get();
529   std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
530
531   FeaturePtr aa = myParametersList.at(aCurrentPos - 1);
532   std::string aN = aa->data()->name();
533
534   if (aCurrentPos == 1)
535     aDoc->moveFeature(aCurFeature, FeaturePtr());
536   else
537     aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos - 2));
538
539   myParametersList.removeOne(aCurFeature);
540   myParametersList.insert(aCurrentPos - 1, aCurFeature);
541
542   myParameters->removeChild(aCurrentItem);
543   myParameters->insertChild(aCurrentPos - 1, aCurrentItem);
544
545   selectItemScroll(aCurrentItem);
546 }
547
548 void ParametersPlugin_WidgetParamsMgr::onDown()
549 {
550   QTreeWidgetItem* aCurrentItem = selectedItem();
551   if (!aCurrentItem)
552     return;
553
554   int aCurrentPos = myParameters->indexOfChild(aCurrentItem);
555   if (aCurrentPos == (myParametersList.count() - 1))
556     return;
557   FeaturePtr aCurFeature = myParametersList.at(aCurrentPos);
558
559   SessionPtr aMgr = ModelAPI_Session::get();
560   std::shared_ptr<ModelAPI_Document> aDoc = aMgr->activeDocument();
561   aDoc->moveFeature(aCurFeature, myParametersList.at(aCurrentPos + 1));
562
563   myParametersList.removeOne(aCurFeature);
564   myParametersList.insert(aCurrentPos + 1, aCurFeature);
565
566   myParameters->removeChild(aCurrentItem);
567   myParameters->insertChild(aCurrentPos + 1, aCurrentItem);
568
569   selectItemScroll(aCurrentItem);
570 }
571
572
573 bool ParametersPlugin_WidgetParamsMgr::hasName(const QString& theName) const
574 {
575   foreach(FeaturePtr aFeature, myParametersList) {
576     if (aFeature->data()->name() == theName.toStdString())
577       return true;
578   }
579   return false;
580 }
581
582 void ParametersPlugin_WidgetParamsMgr::sendWarning()
583 {
584   QMessageBox::warning(this, tr("Warning"), myMessage);
585 }
586
587 void ParametersPlugin_WidgetParamsMgr::onSelectionChanged()
588 {
589   QList<QTreeWidgetItem*> aItemsList = myTable->selectedItems();
590   bool isParameter = false;
591   foreach(QTreeWidgetItem* aItem, aItemsList) {
592     if (aItem->parent() == myParameters) {
593       isParameter = true;
594       break;
595     }
596   }
597   myInsertBtn->setEnabled(isParameter);
598   myRemoveBtn->setEnabled(isParameter);
599   myUpBtn->setEnabled(isParameter);
600   myDownBtn->setEnabled(isParameter);
601 }