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 <ModelAPI_Data.h>
19 #include <ModelAPI_Object.h>
21 #include <Config_WidgetAPI.h>
23 #include <QGridLayout>
25 #include <QListWidget>
31 #include <QApplication>
37 ModuleBase_WidgetMultiSelector::ModuleBase_WidgetMultiSelector(QWidget* theParent,
38 ModuleBase_IWorkshop* theWorkshop,
39 const Config_WidgetAPI* theData,
40 const std::string& theParentId)
41 : ModuleBase_WidgetValidated(theParent, theData, theParentId),
42 myWorkshop(theWorkshop), myIsActive(false)
44 QGridLayout* aMainLay = new QGridLayout(this);
45 ModuleBase_Tools::adjustMargins(aMainLay);
47 QLabel* aTypeLabel = new QLabel(tr("Type"), this);
48 aMainLay->addWidget(aTypeLabel, 0, 0);
50 myTypeCombo = new QComboBox(this);
51 // There is no sence to paramerize list of types while we can not parametrize selection mode
53 std::string aPropertyTypes = theData->getProperty("type_choice");
54 QString aTypesStr = aPropertyTypes.c_str();
55 QStringList aShapeTypes = aTypesStr.split(' ');
57 myTypeCombo->addItems(aShapeTypes);
58 aMainLay->addWidget(myTypeCombo, 0, 1);
59 // if the xml definition contains one type, the controls to select a type should not be shown
60 if (aShapeTypes.size() == 1) {
61 aTypeLabel->setVisible(false);
62 myTypeCombo->setVisible(false);
65 QLabel* aListLabel = new QLabel(tr("Selected objects:"), this);
66 aMainLay->addWidget(aListLabel, 1, 0);
67 // if the xml definition contains one type, an information label should be shown near to the latest
68 if (aShapeTypes.size() == 1) {
69 QString aLabelText = QString::fromStdString(theData->widgetLabel());
70 QString aLabelIcon = QString::fromStdString(theData->widgetIcon());
71 QLabel* aSelectedLabel = new QLabel(aLabelText, this);
72 if (!aLabelIcon.isEmpty())
73 aSelectedLabel->setPixmap(QPixmap(aLabelIcon));
74 aMainLay->addWidget(aSelectedLabel, 1, 1);
75 aMainLay->setColumnStretch(2, 1);
78 myListControl = new QListWidget(this);
79 aMainLay->addWidget(myListControl, 2, 0, 2, -1);
80 aMainLay->setRowStretch(2, 1);
81 aMainLay->addWidget(new QLabel(this)); //FIXME(sbh)???
82 aMainLay->setRowMinimumHeight(3, 20);
83 this->setLayout(aMainLay);
84 connect(myTypeCombo, SIGNAL(currentIndexChanged(int)), this, SLOT(onSelectionTypeChanged()));
86 myCopyAction = new QAction(QIcon(":pictures/copy.png"), tr("Copy"), this);
87 myCopyAction->setShortcut(QKeySequence::Copy);
88 myCopyAction->setEnabled(false);
89 connect(myCopyAction, SIGNAL(triggered(bool)), SLOT(onCopyItem()));
90 myListControl->addAction(myCopyAction);
91 myListControl->setContextMenuPolicy(Qt::ActionsContextMenu);
92 connect(myListControl, SIGNAL(itemSelectionChanged()), SLOT(onListSelection()));
95 ModuleBase_WidgetMultiSelector::~ModuleBase_WidgetMultiSelector()
98 activateShapeSelection();
101 //********************************************************************
102 void ModuleBase_WidgetMultiSelector::activateCustom()
104 ModuleBase_IViewer* aViewer = myWorkshop->viewer();
105 connect(myWorkshop, SIGNAL(selectionChanged()),
106 this, SLOT(onSelectionChanged()),
107 Qt::UniqueConnection);
110 activateShapeSelection();
112 QObjectPtrList anObjects;
113 // Restore selection in the viewer by the attribute selection list
115 DataPtr aData = myFeature->data();
116 AttributeSelectionListPtr aListAttr =
117 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
119 for (int i = 0; i < aListAttr->size(); i++) {
120 AttributeSelectionPtr anAttr = aListAttr->value(i);
121 ResultPtr anObject = anAttr->context();
123 anObjects.append(anObject);
127 myWorkshop->setSelected(anObjects);
130 //********************************************************************
131 void ModuleBase_WidgetMultiSelector::deactivate()
133 disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
135 activateShapeSelection();
138 //********************************************************************
139 bool ModuleBase_WidgetMultiSelector::storeValueCustom() const
141 // the value is stored on the selection changed signal processing
142 // A rare case when plugin was not loaded.
145 DataPtr aData = myFeature->data();
146 AttributeSelectionListPtr aSelectionListAttr =
147 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
149 if (aSelectionListAttr) {
151 TopAbs_ShapeEnum aCurrentType =
152 ModuleBase_Tools::shapeType(myTypeCombo->currentText());
153 aSelectionListAttr->setSelectionType(myTypeCombo->currentText().toStdString());
158 //********************************************************************
159 bool ModuleBase_WidgetMultiSelector::restoreValue()
161 // A rare case when plugin was not loaded.
164 DataPtr aData = myFeature->data();
165 AttributeSelectionListPtr aSelectionListAttr =
166 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
168 if (aSelectionListAttr) {
169 // Restore shape type
170 if (!aSelectionListAttr->selectionType().empty())
171 setCurrentShapeType(ModuleBase_Tools::shapeType(aSelectionListAttr->selectionType().c_str()));
172 updateSelectionList(aSelectionListAttr);
178 //********************************************************************
179 void ModuleBase_WidgetMultiSelector::storeAttributeValue()
181 DataPtr aData = myFeature->data();
182 AttributeSelectionListPtr aSelectionListAttr =
183 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
184 if (aSelectionListAttr.get() == NULL)
187 mySelectionType = aSelectionListAttr->selectionType();
189 int aSize = aSelectionListAttr->size();
190 for (int i = 0; i < aSelectionListAttr->size(); i++) {
191 AttributeSelectionPtr aSelectAttr = aSelectionListAttr->value(i);
192 mySelection.append(GeomSelection(aSelectAttr->context(), aSelectAttr->value()));
196 //********************************************************************
197 void ModuleBase_WidgetMultiSelector::restoreAttributeValue(bool/* theValid*/)
199 DataPtr aData = myFeature->data();
200 AttributeSelectionListPtr aSelectionListAttr =
201 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
202 if (aSelectionListAttr.get() == NULL)
204 aSelectionListAttr->clear();
207 aSelectionListAttr->setSelectionType(mySelectionType);
209 // Store selection in the attribute
210 int aSize = mySelection.size();
211 foreach (GeomSelection aSelec, mySelection) {
212 aSelectionListAttr->append(aSelec.first, aSelec.second);
216 //********************************************************************
217 bool ModuleBase_WidgetMultiSelector::setSelection(const QList<ModuleBase_ViewerPrs>& theValues,
223 QList<ModuleBase_ViewerPrs>::const_iterator anIt = theValues.begin(), aLast = theValues.end();
225 for (int i = thePosition; i < theValues.size(); i++) {
226 ModuleBase_ViewerPrs aValue = theValues[i];
228 bool aProcessed = false;
229 if (isValidSelection(aValue)) {
230 aProcessed = setSelectionCustom(aValue);
232 // if there is at least one set, the result is true
233 isDone = isDone || aProcessed;
234 // when an object, which do not satisfy the validating process, stop set selection
239 updateObject(myFeature);
240 emit valuesChanged();
245 //********************************************************************
246 bool ModuleBase_WidgetMultiSelector::setSelectionCustom(const ModuleBase_ViewerPrs& thePrs)
248 TopoDS_Shape aShape = thePrs.shape();
249 if ((myTypeCombo->count() > 1) && (!aShape.IsNull())) {
250 TopAbs_ShapeEnum aType = ModuleBase_Tools::shapeType(myTypeCombo->currentText());
251 if (aShape.ShapeType() != aType)
255 if (!thePrs.owner().IsNull()) {
256 ObjectPtr anObject = myWorkshop->selection()->getSelectableObject(thePrs.owner());
257 aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
260 aResult = std::dynamic_pointer_cast<ModelAPI_Result>(thePrs.object());
265 // We can not select a result of our feature
266 const std::list<ResultPtr>& aResList = myFeature->results();
267 std::list<ResultPtr>::const_iterator aIt;
268 bool isSkipSelf = false;
269 for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
270 if ((*aIt) == aResult) {
279 // if the result has the similar shap as the parameter shape, just the context is set to the
280 // selection list attribute.
281 DataPtr aData = myFeature->data();
282 AttributeSelectionListPtr aSelectionListAttr =
283 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
285 const TopoDS_Shape& aTDSShape = thePrs.shape();
286 // if only result is selected, an empty shape is set to the model
287 if (aTDSShape.IsNull()) {
288 aSelectionListAttr->append(aResult, GeomShapePtr());
291 GeomShapePtr aShape(new GeomAPI_Shape());
292 aShape->setImpl(new TopoDS_Shape(aTDSShape));
293 // We can not select a result of our feature
294 if (aShape->isEqual(aResult->shape()))
295 aSelectionListAttr->append(aResult, GeomShapePtr());
297 aSelectionListAttr->append(aResult, aShape);
302 //********************************************************************
303 QList<QWidget*> ModuleBase_WidgetMultiSelector::getControls() const
305 QList<QWidget*> result;
306 //result << myTypeCombo;
307 result << myListControl;
311 //********************************************************************
312 bool ModuleBase_WidgetMultiSelector::eventFilter(QObject* theObj, QEvent* theEvent)
314 //TODO: Remove maybe?
315 return ModuleBase_ModelWidget::eventFilter(theObj, theEvent);
318 //********************************************************************
319 void ModuleBase_WidgetMultiSelector::onSelectionTypeChanged()
321 activateShapeSelection();
322 QObjectPtrList anEmptyList;
323 // This method will call Selection changed event which will call onSelectionChanged
324 // To clear mySelection, myListControl and storeValue()
325 // So, we don't need to call it
326 myWorkshop->setSelected(anEmptyList);
329 //********************************************************************
330 void ModuleBase_WidgetMultiSelector::onSelectionChanged()
332 QList<ModuleBase_ViewerPrs> aSelected = getSelectedEntitiesOrObjects(myWorkshop->selection());
334 DataPtr aData = myFeature->data();
335 AttributeSelectionListPtr aSelectionListAttr =
336 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
338 aSelectionListAttr->clear();
339 if (aSelected.size() > 0) {
340 foreach (ModuleBase_ViewerPrs aPrs, aSelected) {
341 if (isValidSelection(aPrs)) {
342 setSelectionCustom(aPrs);
346 emit valuesChanged();
347 // the updateObject method should be called to flush the updated sigal. The workshop listens it,
348 // calls validators for the feature and, as a result, updates the Apply button state.
349 updateObject(myFeature);
352 //********************************************************************
353 void ModuleBase_WidgetMultiSelector::setCurrentShapeType(const TopAbs_ShapeEnum theShapeType)
355 QString aShapeTypeName;
357 for (int idx = 0; idx < myTypeCombo->count(); ++idx) {
358 aShapeTypeName = myTypeCombo->itemText(idx);
359 TopAbs_ShapeEnum aRefType = ModuleBase_Tools::shapeType(aShapeTypeName);
360 if(aRefType == theShapeType && idx != myTypeCombo->currentIndex()) {
362 activateShapeSelection();
363 bool isBlocked = myTypeCombo->blockSignals(true);
364 myTypeCombo->setCurrentIndex(idx);
366 myTypeCombo->blockSignals(isBlocked);
367 activateShapeSelection();
373 void ModuleBase_WidgetMultiSelector::activateShapeSelection()
375 ModuleBase_IViewer* aViewer = myWorkshop->viewer();
378 QString aNewType = myTypeCombo->currentText();
380 aList.append(ModuleBase_Tools::shapeType(aNewType));
381 myWorkshop->activateSubShapesSelection(aList);
383 myWorkshop->deactivateSubShapesSelection();
386 activateFilters(myWorkshop, myIsActive);
389 //********************************************************************
390 void ModuleBase_WidgetMultiSelector::updateSelectionList(AttributeSelectionListPtr theList)
392 myListControl->clear();
393 for (int i = 0; i < theList->size(); i++) {
394 AttributeSelectionPtr aAttr = theList->value(i);
395 myListControl->addItem(aAttr->namingName().c_str());
397 // We have to call repaint because sometimes the List control is not updated
398 myListControl->repaint();
401 //********************************************************************
402 void ModuleBase_WidgetMultiSelector::onCopyItem()
404 QList<QListWidgetItem*> aItems = myListControl->selectedItems();
406 foreach(QListWidgetItem* aItem, aItems) {
409 aRes += aItem->text();
411 if (!aRes.isEmpty()) {
412 QClipboard *clipboard = QApplication::clipboard();
413 clipboard->setText(aRes);
417 //********************************************************************
418 void ModuleBase_WidgetMultiSelector::onListSelection()
420 QList<QListWidgetItem*> aItems = myListControl->selectedItems();
421 myCopyAction->setEnabled(!aItems.isEmpty());