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_Tools.h>
16 #include <ModelAPI_Data.h>
17 #include <ModelAPI_Object.h>
18 #include <ModelAPI_AttributeSelectionList.h>
20 #include <Config_WidgetAPI.h>
22 #include <QGridLayout>
24 #include <QListWidget>
33 ModuleBase_WidgetMultiSelector::ModuleBase_WidgetMultiSelector(QWidget* theParent,
34 ModuleBase_IWorkshop* theWorkshop,
35 const Config_WidgetAPI* theData,
36 const std::string& theParentId)
37 : ModuleBase_ModelWidget(theParent, theData, theParentId),
38 myWorkshop(theWorkshop), myIsActive(false), myUseSubShapes(false)
40 myMainWidget = new QWidget(theParent);
41 QGridLayout* aMainLay = new QGridLayout(myMainWidget);
42 ModuleBase_Tools::adjustMargins(aMainLay);
44 QLabel* aTypeLabel = new QLabel(tr("Type"), myMainWidget);
45 aMainLay->addWidget(aTypeLabel, 0, 0);
47 myTypeCombo = new QComboBox(myMainWidget);
48 // There is no sence to paramerize list of types while we can not parametrize selection mode
49 QString aTypesStr("Vertices Edges Faces Solids");
50 QStringList aShapeTypes = aTypesStr.split(' ');
51 myTypeCombo->addItems(aShapeTypes);
52 aMainLay->addWidget(myTypeCombo, 0, 1);
54 QLabel* aListLabel = new QLabel(tr("Selected objects:"), myMainWidget);
55 aMainLay->addWidget(aListLabel, 1, 0, 1, -1);
57 myListControl = new QListWidget(myMainWidget);
58 aMainLay->addWidget(myListControl, 2, 0, 2, -1);
59 aMainLay->setRowStretch(2, 1);
60 aMainLay->addWidget(new QLabel(myMainWidget));
61 aMainLay->setRowMinimumHeight(3, 20);
62 myMainWidget->setLayout(aMainLay);
63 //TODO: Move into the base class
64 myUseSubShapes = theData->getBooleanAttribute("use_subshapes", false);
66 connect(myTypeCombo, SIGNAL(currentIndexChanged(int)), this, SLOT(onSelectionTypeChanged()));
68 activateSelection(true);
71 ModuleBase_WidgetMultiSelector::~ModuleBase_WidgetMultiSelector()
73 activateSelection(false);
76 //********************************************************************
77 bool ModuleBase_WidgetMultiSelector::storeValue() const
79 // A rare case when plugin was not loaded.
82 DataPtr aData = myFeature->data();
83 AttributeSelectionListPtr aSelectionListAttr =
84 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
86 if (aSelectionListAttr) {
87 aSelectionListAttr->clear();
89 TopAbs_ShapeEnum aCurrentType =
90 ModuleBase_WidgetShapeSelector::shapeType(myTypeCombo->currentText());
91 aSelectionListAttr->setSelectionType((int) aCurrentType);
92 // Store selection in the attribute
93 foreach (GeomSelection aSelec, mySelection) {
94 aSelectionListAttr->append(aSelec.first, aSelec.second);
96 updateObject(myFeature);
102 //********************************************************************
103 bool ModuleBase_WidgetMultiSelector::restoreValue()
105 // A rare case when plugin was not loaded.
108 DataPtr aData = myFeature->data();
109 AttributeSelectionListPtr aSelectionListAttr =
110 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
112 if (aSelectionListAttr) {
114 // Restore shape type
115 TopAbs_ShapeEnum aShapeType = (TopAbs_ShapeEnum) aSelectionListAttr->selectionType();
116 setCurrentShapeType(aShapeType);
117 // Restore selection in the list
118 for (int i = 0; i < aSelectionListAttr->size(); i++) {
119 AttributeSelectionPtr aSelectAttr = aSelectionListAttr->value(i);
120 mySelection.append(GeomSelection(aSelectAttr->context(), aSelectAttr->value()));
122 updateSelectionList();
128 //********************************************************************
129 QWidget* ModuleBase_WidgetMultiSelector::getControl() const
134 //********************************************************************
135 QList<QWidget*> ModuleBase_WidgetMultiSelector::getControls() const
137 QList<QWidget*> result;
138 //result << myTypeCombo;
139 result << myListControl;
143 //********************************************************************
144 bool ModuleBase_WidgetMultiSelector::eventFilter(QObject* theObj, QEvent* theEvent)
146 //TODO: Remove maybe?
147 return ModuleBase_ModelWidget::eventFilter(theObj, theEvent);
150 //********************************************************************
151 void ModuleBase_WidgetMultiSelector::activateSelection(bool toActivate)
153 myIsActive = toActivate;
155 connect(myWorkshop, SIGNAL(selectionChanged()),
156 this, SLOT(onSelectionChanged()),
157 Qt::UniqueConnection);
158 activateShapeSelection();
160 disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
161 myWorkshop->deactivateSubShapesSelection();
165 //********************************************************************
166 void ModuleBase_WidgetMultiSelector::onSelectionTypeChanged()
168 activateShapeSelection();
169 QObjectPtrList anEmptyList;
170 myWorkshop->setSelected(anEmptyList);
171 // Clear mySelection, myListControl and storeValue()
172 onSelectionChanged();
175 //********************************************************************
176 void ModuleBase_WidgetMultiSelector::onSelectionChanged()
178 ModuleBase_ISelection* aSelection = myWorkshop->selection();
179 NCollection_List<TopoDS_Shape> aSelectedShapes; //, aFilteredShapes;
180 std::list<ObjectPtr> aOwnersList;
181 aSelection->selectedShapes(aSelectedShapes, aOwnersList);
184 std::list<ObjectPtr>::const_iterator aIt;
185 NCollection_List<TopoDS_Shape>::Iterator aShpIt(aSelectedShapes);
187 for (aIt = aOwnersList.cbegin(); aIt != aOwnersList.cend(); aShpIt.Next(), aIt++) {
188 ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(*aIt);
190 // We can not select a result of our feature
191 const std::list<ResultPtr>& aResList = myFeature->results();
192 std::list<ResultPtr>::const_iterator aIt;
193 bool isSkipSelf = false;
194 for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
195 if ((*aIt) == aResult) {
203 aShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape());
204 aShape->setImpl(new TopoDS_Shape(aShpIt.Value()));
205 mySelection.append(GeomSelection(aResult, aShape));
207 updateSelectionList();
208 emit valuesChanged();
211 //********************************************************************
212 void ModuleBase_WidgetMultiSelector::filterShapes(const NCollection_List<TopoDS_Shape>& theShapesToFilter,
213 NCollection_List<TopoDS_Shape>& theResult)
215 if(myTypeCombo->count() == 0 || theShapesToFilter.IsEmpty())
217 TopAbs_ShapeEnum aReferenceType =
218 ModuleBase_WidgetShapeSelector::shapeType(myTypeCombo->currentText());
219 NCollection_List<TopoDS_Shape>::Iterator anIter(theShapesToFilter);
220 for (; anIter.More(); anIter.Next()) {
221 TopoDS_Shape aShape = anIter.Value();
222 if (aShape.IsNull() || aShape.ShapeType() != aReferenceType)
224 theResult.Append(aShape);
228 //********************************************************************
229 void ModuleBase_WidgetMultiSelector::setCurrentShapeType(const TopAbs_ShapeEnum theShapeType)
231 QString aShapeTypeName;
233 for (int idx = 0; idx < myTypeCombo->count(); ++idx) {
234 aShapeTypeName = myTypeCombo->itemText(idx);
235 TopAbs_ShapeEnum aRefType = ModuleBase_WidgetShapeSelector::shapeType(aShapeTypeName);
236 if(aRefType == theShapeType && idx != myTypeCombo->currentIndex()) {
237 activateSelection(false);
238 bool isBlocked = myTypeCombo->blockSignals(true);
239 myTypeCombo->setCurrentIndex(idx);
240 myTypeCombo->blockSignals(isBlocked);
241 activateSelection(true);
247 void ModuleBase_WidgetMultiSelector::activateShapeSelection()
249 QString aNewType = myTypeCombo->currentText();
251 aList.append(ModuleBase_WidgetShapeSelector::shapeType(aNewType));
252 myWorkshop->activateSubShapesSelection(aList);
255 //********************************************************************
256 void ModuleBase_WidgetMultiSelector::updateSelectionList()
259 if (myTypeCombo->currentText().toLower() == "vertices")
261 else if (myTypeCombo->currentText().toLower() == "edges")
263 else if (myTypeCombo->currentText().toLower() == "faces")
265 else if (myTypeCombo->currentText().toLower() == "solids")
268 myListControl->clear();
270 foreach (GeomSelection aSel, mySelection) {
271 QString aName(aSel.first->data()->name().c_str());
272 aName += ":" + aType + QString("_%1").arg(i);
273 myListControl->addItem(aName);
276 myListControl->repaint();