- //QObjectPtrList aObjects = myWorkshop->selection()->selectedPresentations();
- QList<ModuleBase_ViewerPrs> aSelected = myWorkshop->selection()->getSelected();
- if (aSelected.size() > 0) {
- if (setSelection(aSelected.first()))
- emit focusOutWidget(this);
- }
-}
-
-//********************************************************************
-bool ModuleBase_WidgetShapeSelector::setSelection(ModuleBase_ViewerPrs theValue)
-{
- ObjectPtr aObject = theValue.object();
- if ((!mySelectedObject) && (!aObject))
- return false;
-
- // Check that the selected object is result (others can not be accepted)
- ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aObject);
- if (!aRes)
- return false;
-
- if (myFeature) {
- // We can not select a result of our feature
- const std::list<std::shared_ptr<ModelAPI_Result>>& aResList = myFeature->results();
- std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aIt;
- for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
- if ((*aIt) == aRes)
- return false;
- }
- }
- // Check that object belongs to active document or PartSet
- DocumentPtr aDoc = aRes->document();
- SessionPtr aMgr = ModelAPI_Session::get();
- if (!(aDoc == aMgr->activeDocument()) && !(aDoc == aMgr->moduleDocument()))
- return false;
-
- // Check that the result has a shape
- GeomShapePtr aShape = ModelAPI_Tools::shape(aRes);
- if (!aShape)
- return false;
-
- /// Check that object has acceptable type
- if (!acceptObjectType(aObject))
- return false;
-
- // Get sub-shapes from local selection
- if (!theValue.shape().IsNull()) {
- aShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape());
- aShape->setImpl(new TopoDS_Shape(theValue.shape()));
- }
+ QList<ModuleBase_ViewerPrs> aSelectedPrs = getSelectedEntitiesOrObjects(myWorkshop->selection());
+ if (aSelectedPrs.empty())
+ return;
+ ModuleBase_ViewerPrs aPrs = aSelectedPrs.first();
+ if (aPrs.isEmpty() || !isValidSelection(aPrs))
+ return;