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 (!acceptSubShape(aShape))
  86     return false;
  87
  88   ResultPtr aResult;
  89   if (!thePrs.owner().IsNull()) {
  90     ObjectPtr anObject = myWorkshop->selection()->getSelectableObject(thePrs.owner());
  91     aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
  92   }
  93   else {
  94     aResult = std::dynamic_pointer_cast<ModelAPI_Result>(thePrs.object());
  95   }
  96
  97
  98   if (myFeature) {
  99     // We can not select a result of our feature
 100     const std::list<ResultPtr>& aResList = myFeature->results();
 101     std::list<ResultPtr>::const_iterator aIt;
 102     bool isSkipSelf = false;
 103     for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
 104       if ((*aIt) == aResult) {
 105         isSkipSelf = true;
 106         break;
 107       }
 108     }
 109     if(isSkipSelf)
 110       return false;
 111   }
 112
 113   GeomShapePtr aGShape = GeomShapePtr();
 114   const TopoDS_Shape& aTDSShape = thePrs.shape();
 115   // if only result is selected, an empty shape is set to the model
 116   if (aTDSShape.IsNull()) {
 117     //aSelectionListAttr->append(aResult, GeomShapePtr());
 118   }
 119   else {
 120     aGShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape());
 121     //GeomShapePtr aShape(new GeomAPI_Shape());
 122     aGShape->setImpl(new TopoDS_Shape(aTDSShape));
 123     // We can not select a result of our feature
 124     if (aGShape->isEqual(aResult->shape())) {
 125       //aSelectionListAttr->append(aResult, GeomShapePtr());
 126       aGShape = GeomShapePtr();
 127     }
 128     else {
 129       //aSelectionListAttr->append(aResult, aShape);
 130     }
 131   }
 132
 133   setObject(aResult, aGShape);
 134   return true;
 135 }
 136
 137 //********************************************************************
 138 bool PartSet_WidgetMultiSelector::setObject(const ObjectPtr& theSelectedObject,
 139                                             const GeomShapePtr& theShape)
 140 {
 141   ObjectPtr aSelectedObject = theSelectedObject;
 142   GeomShapePtr aShape = theShape;
 143
 144   FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(aSelectedObject);
 145   //if (aSelectedFeature == myFeature)  // In order to avoid selection of the same object
 146   //  return false;
 147
 148   // Do check using of external feature
 149   std::shared_ptr<SketchPlugin_Feature> aSPFeature =
 150           std::dynamic_pointer_cast<SketchPlugin_Feature>(aSelectedFeature);
 151
 152   // Do check that we can use external feature
 153   if ((aSPFeature.get() != NULL) && aSPFeature->isExternal() && (!myExternalObjectMgr->useExternal()))
 154     return false;
 155
 156   if (aSPFeature.get() == NULL && aShape.get() != NULL && !aShape->isNull() && myExternalObjectMgr->useExternal()) {
 157     if (myIsInVaildate)
 158       aSelectedObject = myExternalObjectMgr->externalObjectValidated(theSelectedObject, theShape, sketch());
 159     else
 160       aSelectedObject = myExternalObjectMgr->externalObject(theSelectedObject, theShape, sketch());
 161   }
 162
 163
 164   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aSelectedObject);
 165
 166   DataPtr aData = myFeature->data();
 167   AttributeSelectionListPtr aSelectionListAttr =
 168     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aData->attribute(attributeID()));
 169
 170   aSelectionListAttr->append(aResult, aShape);
 171
 172   return true;
 173 }