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