src/PartSet/PartSet_WidgetMultiSelector.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        PartSet_WidgetMultiSelector.cpp
   4 // Created:     15 Apr 2015
   5 // Author:      Natalia Ermolaeva
   6
   7 #include "PartSet_WidgetMultiSelector.h"
   8
   9 #include <ModelAPI_AttributeRefAttr.h>
  10 #include <ModelAPI_Session.h>
  11 #include <ModelAPI_Validator.h>
  12
  13 #include <ModuleBase_Definitions.h>
  14 #include <ModuleBase_Tools.h>
  15 #include <ModuleBase_IWorkshop.h>
  16 #include <ModuleBase_ISelection.h>
  17
  18 #include <Config_WidgetAPI.h>
  19
  20 #include <PartSet_Tools.h>
  21 #include <PartSet_ExternalObjectsMgr.h>
  22 #include <SketchPlugin_Feature.h>
  23
  24 #include <SketchPlugin_ConstraintRigid.h>
  25
  26 #include <XGUI_Workshop.h>
  27
  28 #include <QComboBox>
  29
  30 PartSet_WidgetMultiSelector::PartSet_WidgetMultiSelector(QWidget* theParent,
  31                                                          ModuleBase_IWorkshop* theWorkshop,
  32                                                          const Config_WidgetAPI* theData,
  33                                                          const std::string& theParentId)
  34 : ModuleBase_WidgetMultiSelector(theParent, theWorkshop, theData, theParentId)
  35 {
  36   myExternalObjectMgr = new PartSet_ExternalObjectsMgr(theData->getProperty("use_external"), false);
  37 }
  38
  39 PartSet_WidgetMultiSelector::~PartSet_WidgetMultiSelector()
  40 {
  41   delete myExternalObjectMgr;
  42 }
  43
  44 //********************************************************************
  45 void PartSet_WidgetMultiSelector::onSelectionChanged()
  46 {
  47   ModuleBase_WidgetMultiSelector::onSelectionChanged();
  48   // TODO(nds): unite with externalObject(), remove parameters
  49   //myFeature->execute();
  50
  51   DataPtr aData = myFeature->data();
  52   AttributeSelectionListPtr aSelectionListAttr =
  53     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
  54
  55   QObjectPtrList aListOfAttributeObjects;
  56   for (int i = 0; i < aSelectionListAttr->size(); i++) {
  57     AttributeSelectionPtr anAttr = aSelectionListAttr->value(i);
  58     aListOfAttributeObjects.append(anAttr->context());
  59   }
  60
  61   myExternalObjectMgr->removeUnusedExternalObjects(aListOfAttributeObjects, sketch(), myFeature);
  62 }
  63
  64 //********************************************************************
  65 void PartSet_WidgetMultiSelector::storeAttributeValue()
  66 {
  67   myIsInVaildate = true;
  68   ModuleBase_WidgetMultiSelector::storeAttributeValue();
  69
  70 }
  71
  72 //********************************************************************
  73 void PartSet_WidgetMultiSelector::restoreAttributeValue(const bool theValid)
  74 {
  75   myIsInVaildate = false;
  76   ModuleBase_WidgetMultiSelector::restoreAttributeValue(theValid);
  77
  78   myExternalObjectMgr->removeExternalValidated(sketch(), myFeature);
  79 }
  80
  81 //********************************************************************
  82 bool PartSet_WidgetMultiSelector::setSelectionCustom(const ModuleBase_ViewerPrs& thePrs)
  83 {
  84   TopoDS_Shape aShape = thePrs.shape();
  85   if ((myTypeCombo->count() > 1) && (!aShape.IsNull())) {
  86     TopAbs_ShapeEnum aType = ModuleBase_Tools::shapeType(myTypeCombo->currentText());
  87     if (aShape.ShapeType() != aType)
  88       return false;
  89   }
  90   ResultPtr aResult;
  91   if (!thePrs.owner().IsNull()) {
  92     ObjectPtr anObject = myWorkshop->selection()->getSelectableObject(thePrs.owner());
  93     aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
  94   }
  95   else {
  96     aResult = std::dynamic_pointer_cast<ModelAPI_Result>(thePrs.object());
  97   }
  98
  99
 100   if (myFeature) {
 101     // We can not select a result of our feature
 102     const std::list<ResultPtr>& aResList = myFeature->results();
 103     std::list<ResultPtr>::const_iterator aIt;
 104     bool isSkipSelf = false;
 105     for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
 106       if ((*aIt) == aResult) {
 107         isSkipSelf = true;
 108         break;
 109       }
 110     }
 111     if(isSkipSelf)
 112       return false;
 113   }
 114
 115   GeomShapePtr aGShape = GeomShapePtr();
 116   const TopoDS_Shape& aTDSShape = thePrs.shape();
 117   // if only result is selected, an empty shape is set to the model
 118   if (aTDSShape.IsNull()) {
 119     //aSelectionListAttr->append(aResult, GeomShapePtr());
 120   }
 121   else {
 122     aGShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape());
 123     //GeomShapePtr aShape(new GeomAPI_Shape());
 124     aGShape->setImpl(new TopoDS_Shape(aTDSShape));
 125     // We can not select a result of our feature
 126     if (aGShape->isEqual(aResult->shape())) {
 127       //aSelectionListAttr->append(aResult, GeomShapePtr());
 128       aGShape = GeomShapePtr();
 129     }
 130     else {
 131       //aSelectionListAttr->append(aResult, aShape);
 132     }
 133   }
 134
 135   setObject(aResult, aGShape);
 136   return true;
 137 }
 138
 139 //********************************************************************
 140 bool PartSet_WidgetMultiSelector::setObject(const ObjectPtr& theSelectedObject,
 141                                             const GeomShapePtr& theShape)
 142 {
 143   ObjectPtr aSelectedObject = theSelectedObject;
 144   GeomShapePtr aShape = theShape;
 145
 146   FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(aSelectedObject);
 147   //if (aSelectedFeature == myFeature)  // In order to avoid selection of the same object
 148   //  return false;
 149
 150   // Do check using of external feature
 151   std::shared_ptr<SketchPlugin_Feature> aSPFeature =
 152           std::dynamic_pointer_cast<SketchPlugin_Feature>(aSelectedFeature);
 153
 154   // Do check that we can use external feature
 155   if ((aSPFeature.get() != NULL) && aSPFeature->isExternal() && (!myExternalObjectMgr->useExternal()))
 156     return false;
 157
 158   if (aSPFeature.get() == NULL && aShape.get() != NULL && !aShape->isNull() && myExternalObjectMgr->useExternal()) {
 159     if (myIsInVaildate)
 160       aSelectedObject = myExternalObjectMgr->externalObjectValidated(theSelectedObject, theShape, sketch());
 161     else
 162       aSelectedObject = myExternalObjectMgr->externalObject(theSelectedObject, theShape, sketch());
 163   }
 164
 165
 166   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aSelectedObject);
 167
 168   DataPtr aData = myFeature->data();
 169   AttributeSelectionListPtr aSelectionListAttr =
 170     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
 171
 172   aSelectionListAttr->append(aResult, aShape);
 173
 174   return true;
 175 }