1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
4 * ModuleBase_WidgetMultiSelector.cpp
6 * Created on: Aug 28, 2014
10 #include <ModuleBase_WidgetMultiSelector.h>
11 #include <ModuleBase_WidgetShapeSelector.h>
12 #include <ModuleBase_ISelection.h>
13 #include <ModuleBase_IWorkshop.h>
14 #include <ModuleBase_IViewer.h>
15 #include <ModuleBase_Tools.h>
16 #include <ModuleBase_Definitions.h>
18 #include <GeomValidators_ShapeType.h>
20 #include <ModelAPI_Data.h>
21 #include <ModelAPI_Object.h>
23 #include <Config_WidgetAPI.h>
25 #include <QGridLayout>
27 #include <QListWidget>
33 #include <QApplication>
39 //#define DEBUG_SHAPE_VALIDATION_PREVIOUS
41 ModuleBase_WidgetMultiSelector::ModuleBase_WidgetMultiSelector(QWidget* theParent,
42 ModuleBase_IWorkshop* theWorkshop,
43 const Config_WidgetAPI* theData,
44 const std::string& theParentId)
45 : ModuleBase_WidgetValidated(theParent, theData, theParentId),
46 myWorkshop(theWorkshop)
48 QGridLayout* aMainLay = new QGridLayout(this);
49 ModuleBase_Tools::adjustMargins(aMainLay);
51 QLabel* aTypeLabel = new QLabel(tr("Type"), this);
52 aMainLay->addWidget(aTypeLabel, 0, 0);
54 myTypeCombo = new QComboBox(this);
55 // There is no sence to paramerize list of types while we can not parametrize selection mode
57 myShapeValidator = new GeomValidators_ShapeType();
59 std::string aPropertyTypes = theData->getProperty("type_choice");
60 QString aTypesStr = aPropertyTypes.c_str();
61 QStringList aShapeTypes = aTypesStr.split(' ');
63 //myIsUseChoice = theData->getBooleanAttribute("use_choice", true);
65 myTypeCombo->addItems(aShapeTypes);
66 aMainLay->addWidget(myTypeCombo, 0, 1);
67 // if the xml definition contains one type, the controls to select a type should not be shown
68 if (aShapeTypes.size() == 1/* || !myIsUseChoice*/) {
69 aTypeLabel->setVisible(false);
70 myTypeCombo->setVisible(false);
73 std::string aLabelText = theData->getProperty("label");
74 QLabel* aListLabel = new QLabel(!aLabelText.empty() ? aLabelText.c_str()
75 : tr("Selected objects:"), this);
76 aMainLay->addWidget(aListLabel, 1, 0);
77 // if the xml definition contains one type, an information label should be shown near to the latest
78 if (aShapeTypes.size() == 1) {
79 QString aLabelText = QString::fromStdString(theData->widgetLabel());
80 QString aLabelIcon = QString::fromStdString(theData->widgetIcon());
81 QLabel* aSelectedLabel = new QLabel(aLabelText, this);
82 if (!aLabelIcon.isEmpty())
83 aSelectedLabel->setPixmap(QPixmap(aLabelIcon));
84 aMainLay->addWidget(aSelectedLabel, 1, 1);
85 aMainLay->setColumnStretch(2, 1);
88 myListControl = new QListWidget(this);
89 aMainLay->addWidget(myListControl, 2, 0, 2, -1);
90 aMainLay->setRowStretch(2, 1);
91 aMainLay->addWidget(new QLabel(this)); //FIXME(sbh)???
92 aMainLay->setRowMinimumHeight(3, 20);
93 this->setLayout(aMainLay);
94 connect(myTypeCombo, SIGNAL(currentIndexChanged(int)), this, SLOT(onSelectionTypeChanged()));
96 myCopyAction = new QAction(QIcon(":pictures/copy.png"), tr("Copy"), this);
97 myCopyAction->setShortcut(QKeySequence::Copy);
98 myCopyAction->setEnabled(false);
99 connect(myCopyAction, SIGNAL(triggered(bool)), SLOT(onCopyItem()));
100 myListControl->addAction(myCopyAction);
101 myListControl->setContextMenuPolicy(Qt::ActionsContextMenu);
102 connect(myListControl, SIGNAL(itemSelectionChanged()), SLOT(onListSelection()));
105 ModuleBase_WidgetMultiSelector::~ModuleBase_WidgetMultiSelector()
107 delete myShapeValidator;
110 //TODO: nds stabilization hotfix
111 void ModuleBase_WidgetMultiSelector::disconnectSignals()
113 disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
116 //********************************************************************
117 void ModuleBase_WidgetMultiSelector::activateCustom()
119 ModuleBase_IViewer* aViewer = myWorkshop->viewer();
120 connect(myWorkshop, SIGNAL(selectionChanged()),
121 this, SLOT(onSelectionChanged()),
122 Qt::UniqueConnection);
124 activateShapeSelection(true);
126 // Restore selection in the viewer by the attribute selection list
127 myWorkshop->setSelected(getAttributeSelection());
129 activateFilters(myWorkshop, true);
132 //********************************************************************
133 void ModuleBase_WidgetMultiSelector::deactivate()
135 disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
136 activateShapeSelection(false);
137 activateFilters(myWorkshop, false);
140 //********************************************************************
141 bool ModuleBase_WidgetMultiSelector::storeValueCustom() const
143 // the value is stored on the selection changed signal processing
144 // A rare case when plugin was not loaded.
147 DataPtr aData = myFeature->data();
148 AttributeSelectionListPtr aSelectionListAttr =
149 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
151 if (aSelectionListAttr) {
153 TopAbs_ShapeEnum aCurrentType =
154 ModuleBase_Tools::shapeType(myTypeCombo->currentText());
155 aSelectionListAttr->setSelectionType(myTypeCombo->currentText().toStdString());
160 //********************************************************************
161 bool ModuleBase_WidgetMultiSelector::restoreValue()
163 // A rare case when plugin was not loaded.
166 DataPtr aData = myFeature->data();
167 AttributeSelectionListPtr aSelectionListAttr =
168 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
170 if (aSelectionListAttr) {
171 // Restore shape type
172 if (!aSelectionListAttr->selectionType().empty())
173 setCurrentShapeType(ModuleBase_Tools::shapeType(aSelectionListAttr->selectionType().c_str()));
174 updateSelectionList(aSelectionListAttr);
180 //********************************************************************
181 void ModuleBase_WidgetMultiSelector::storeAttributeValue()
183 DataPtr aData = myFeature->data();
184 AttributeSelectionListPtr aSelectionListAttr =
185 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
186 if (aSelectionListAttr.get() == NULL)
189 mySelectionType = aSelectionListAttr->selectionType();
191 int aSize = aSelectionListAttr->size();
192 for (int i = 0; i < aSelectionListAttr->size(); i++) {
193 AttributeSelectionPtr aSelectAttr = aSelectionListAttr->value(i);
194 mySelection.append(GeomSelection(aSelectAttr->context(), aSelectAttr->value()));
198 //********************************************************************
199 void ModuleBase_WidgetMultiSelector::restoreAttributeValue(bool/* theValid*/)
201 DataPtr aData = myFeature->data();
202 AttributeSelectionListPtr aSelectionListAttr =
203 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
204 if (aSelectionListAttr.get() == NULL)
206 aSelectionListAttr->clear();
209 aSelectionListAttr->setSelectionType(mySelectionType);
211 // Store selection in the attribute
212 int aSize = mySelection.size();
213 foreach (GeomSelection aSelec, mySelection) {
214 aSelectionListAttr->append(aSelec.first, aSelec.second);
218 //********************************************************************
219 void ModuleBase_WidgetMultiSelector::customValidators(
220 std::list<ModelAPI_Validator*>& theValidators,
221 std::list<std::list<std::string> >& theArguments) const
224 std::list<std::string> anArguments;
226 theValidators.push_back(myShapeValidator);
227 if (true/*myIsUseChoice*/) {
228 QString aType = myTypeCombo->currentText();
229 anArguments.push_back(validatorType(aType));
232 for(int i = 0, aCount = myTypeCombo->count(); i < aCount; i++) {
233 anArguments.push_back(validatorType(myTypeCombo->itemText(i)));
236 theArguments.push_back(anArguments);
239 //********************************************************************
240 bool ModuleBase_WidgetMultiSelector::acceptSubShape(const TopoDS_Shape& theShape) const
243 if (theShape.IsNull()) {
244 aValid = true; // do not check the shape type if the shape is empty
245 // extrusion uses a sketch object selectected in Object browser
249 TopAbs_ShapeEnum aShapeType = theShape.ShapeType();
250 if (myTypeCombo->count() > 1) {
251 TopAbs_ShapeEnum aType = ModuleBase_Tools::shapeType(myTypeCombo->currentText());
252 aValid = aShapeType == aType;
255 for(int i = 0, aCount = myTypeCombo->count(); i < aCount && !aValid; i++) {
256 TopAbs_ShapeEnum aType = ModuleBase_Tools::shapeType(myTypeCombo->itemText(i));
257 aValid = aShapeType == aType;
264 //********************************************************************
265 bool ModuleBase_WidgetMultiSelector::setSelection(QList<ModuleBase_ViewerPrs>& theValues)
267 QList<ModuleBase_ViewerPrs> aSkippedValues;
269 QList<ModuleBase_ViewerPrs>::const_iterator anIt = theValues.begin(), aLast = theValues.end();
271 for (; anIt != aLast; anIt++) {
272 ModuleBase_ViewerPrs aValue = *anIt;
273 bool aProcessed = false;
274 if (isValidSelection(aValue)) {
275 aProcessed = setSelectionCustom(aValue);
278 aSkippedValues.append(aValue);
279 // if there is at least one set, the result is true
280 isDone = isDone || aProcessed;
282 // updateObject - to update/redisplay feature
283 // it is commented in order to perfom it outside the method
285 //updateObject(myFeature);
286 // this emit is necessary to call store/restore method an restore type of selection
287 //emit valuesChanged();
290 if (!aSkippedValues.empty())
291 theValues.append(aSkippedValues);
296 //********************************************************************
297 bool ModuleBase_WidgetMultiSelector::isValidSelectionCustom(const ModuleBase_ViewerPrs& thePrs)
299 #ifdef DEBUG_SHAPE_VALIDATION_PREVIOUS
302 GeomShapePtr aShape = myWorkshop->selection()->getShape(thePrs);
303 // if there is no selected shape, the method returns true
308 // Check that the selection corresponds to selection type
309 TopoDS_Shape aTopoShape = aShape->impl<TopoDS_Shape>();
310 aValid = acceptSubShape(aTopoShape);
314 ResultPtr aResult = myWorkshop->selection()->getResult(thePrs);
317 // We can not select a result of our feature
318 const std::list<ResultPtr>& aResList = myFeature->results();
319 std::list<ResultPtr>::const_iterator aIt;
320 bool isSkipSelf = false;
321 for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
322 if ((*aIt) == aResult) {
336 //********************************************************************
337 bool ModuleBase_WidgetMultiSelector::setSelectionCustom(const ModuleBase_ViewerPrs& thePrs)
339 //TopoDS_Shape aShape = thePrs.shape();
340 //if (!acceptSubShape(aShape))
343 ResultPtr aResult = myWorkshop->selection()->getResult(thePrs);
346 // We can not select a result of our feature
347 const std::list<ResultPtr>& aResList = myFeature->results();
348 std::list<ResultPtr>::const_iterator aIt;
349 bool isSkipSelf = false;
350 for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
351 if ((*aIt) == aResult) {
360 // if the result has the similar shap as the parameter shape, just the context is set to the
361 // selection list attribute.
362 DataPtr aData = myFeature->data();
363 AttributeSelectionListPtr aSelectionListAttr =
364 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
366 /*const TopoDS_Shape& aTDSShape = thePrs.shape();
367 // if only result is selected, an empty shape is set to the model
368 if (aTDSShape.IsNull()) {
369 aSelectionListAttr->append(aResult, GeomShapePtr());
372 GeomShapePtr aShape(new GeomAPI_Shape());
373 aShape->setImpl(new TopoDS_Shape(aTDSShape));
374 // We can not select a result of our feature
375 if (aShape->isEqual(aResult->shape()))
376 aSelectionListAttr->append(aResult, GeomShapePtr());
378 aSelectionListAttr->append(aResult, aShape);
380 GeomShapePtr aShape = myWorkshop->selection()->getShape(thePrs);
381 aSelectionListAttr->append(aResult, aShape);
386 //********************************************************************
387 QList<QWidget*> ModuleBase_WidgetMultiSelector::getControls() const
389 QList<QWidget*> result;
390 //result << myTypeCombo;
391 result << myListControl;
395 //********************************************************************
396 void ModuleBase_WidgetMultiSelector::onSelectionTypeChanged()
398 activateShapeSelection(true);
399 activateFilters(myWorkshop, true);
400 QList<ModuleBase_ViewerPrs> anEmptyList;
401 // This method will call Selection changed event which will call onSelectionChanged
402 // To clear mySelection, myListControl and storeValue()
403 // So, we don't need to call it
404 myWorkshop->setSelected(anEmptyList);
407 //********************************************************************
408 void ModuleBase_WidgetMultiSelector::onSelectionChanged()
410 QList<ModuleBase_ViewerPrs> aSelected = myWorkshop->selection()->getSelected(ModuleBase_ISelection::AllControls);
412 DataPtr aData = myFeature->data();
413 AttributeSelectionListPtr aSelectionListAttr =
414 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
416 aSelectionListAttr->clear();
417 if (aSelected.size() > 0) {
418 foreach (ModuleBase_ViewerPrs aPrs, aSelected) {
419 if (isValidSelection(aPrs)) {
420 setSelectionCustom(aPrs);
424 emit valuesChanged();
425 // the updateObject method should be called to flush the updated sigal. The workshop listens it,
426 // calls validators for the feature and, as a result, updates the Apply button state.
427 updateObject(myFeature);
429 // Set focus to List control in order to make possible
430 // to use Tab key for transfer the focus to next widgets
431 myListControl->setCurrentRow(myListControl->model()->rowCount() - 1);
432 myListControl->setFocus();
435 //********************************************************************
436 void ModuleBase_WidgetMultiSelector::setCurrentShapeType(const TopAbs_ShapeEnum theShapeType)
438 QString aShapeTypeName;
440 for (int idx = 0; idx < myTypeCombo->count(); ++idx) {
441 aShapeTypeName = myTypeCombo->itemText(idx);
442 TopAbs_ShapeEnum aRefType = ModuleBase_Tools::shapeType(aShapeTypeName);
443 if(aRefType == theShapeType && idx != myTypeCombo->currentIndex()) {
444 activateShapeSelection(false);
445 activateFilters(myWorkshop, false);
446 bool isBlocked = myTypeCombo->blockSignals(true);
447 myTypeCombo->setCurrentIndex(idx);
448 myTypeCombo->blockSignals(isBlocked);
450 activateShapeSelection(true);
451 activateFilters(myWorkshop, true);
457 void ModuleBase_WidgetMultiSelector::activateShapeSelection(const bool isActivated)
459 ModuleBase_IViewer* aViewer = myWorkshop->viewer();
462 QString aNewType = myTypeCombo->currentText();
464 if (true /*myIsUseChoice*/) {
465 aList.append(ModuleBase_Tools::shapeType(aNewType));
468 for(int i = 0, aCount = myTypeCombo->count(); i < aCount; i++)
469 aList.append(ModuleBase_Tools::shapeType(myTypeCombo->itemText(i)));
471 myWorkshop->activateSubShapesSelection(aList);
473 myWorkshop->deactivateSubShapesSelection();
477 QList<ModuleBase_ViewerPrs> ModuleBase_WidgetMultiSelector::getAttributeSelection() const
479 QList<ModuleBase_ViewerPrs> aSelected;
480 // Restore selection in the viewer by the attribute selection list
482 DataPtr aData = myFeature->data();
483 AttributeSelectionListPtr aListAttr =
484 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
486 for (int i = 0; i < aListAttr->size(); i++) {
487 AttributeSelectionPtr anAttr = aListAttr->value(i);
488 ResultPtr anObject = anAttr->context();
489 if (anObject.get()) {
491 std::shared_ptr<GeomAPI_Shape> aShapePtr = anAttr->value();
492 if (aShapePtr.get()) {
493 aShape = aShapePtr->impl<TopoDS_Shape>();
495 aSelected.append(ModuleBase_ViewerPrs(anObject, aShape, NULL));
503 //********************************************************************
504 void ModuleBase_WidgetMultiSelector::updateSelectionList(AttributeSelectionListPtr theList)
506 myListControl->clear();
507 for (int i = 0; i < theList->size(); i++) {
508 AttributeSelectionPtr aAttr = theList->value(i);
509 myListControl->addItem(aAttr->namingName().c_str());
511 // We have to call repaint because sometimes the List control is not updated
512 myListControl->repaint();
515 //********************************************************************
516 std::string ModuleBase_WidgetMultiSelector::validatorType(const QString& theType) const
520 if (theType == "Vertices")
522 else if (theType == "Edges")
524 else if (theType == "Faces")
526 else if (theType == "Solids")
532 //********************************************************************
533 void ModuleBase_WidgetMultiSelector::onCopyItem()
535 QList<QListWidgetItem*> aItems = myListControl->selectedItems();
537 foreach(QListWidgetItem* aItem, aItems) {
540 aRes += aItem->text();
542 if (!aRes.isEmpty()) {
543 QClipboard *clipboard = QApplication::clipboard();
544 clipboard->setText(aRes);
548 //********************************************************************
549 void ModuleBase_WidgetMultiSelector::onListSelection()
551 QList<QListWidgetItem*> aItems = myListControl->selectedItems();
552 myCopyAction->setEnabled(!aItems.isEmpty());