1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: PartSet_WidgetShapeSelector.cpp
4 // Created: 27 Nov 2014
5 // Author: Vitaly Smetannikov
7 #include "PartSet_WidgetShapeSelector.h"
9 #include <ModelAPI_AttributeRefAttr.h>
10 #include <ModelAPI_Session.h>
11 #include <ModelAPI_Validator.h>
12 #include <ModelAPI_RefAttrValidator.h>
13 #include <ModelAPI_ResultValidator.h>
15 #include <PartSet_Tools.h>
16 #include <SketchPlugin_Feature.h>
18 bool PartSet_WidgetShapeSelector::storeAttributeValues(ObjectPtr theSelectedObject, GeomShapePtr theShape) const
20 ObjectPtr aSelectedObject = theSelectedObject;
21 GeomShapePtr aShape = theShape;
23 FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(aSelectedObject);
24 if (aSelectedFeature == myFeature) // In order to avoid selection of the same object
26 std::shared_ptr<SketchPlugin_Feature> aSPFeature =
27 std::dynamic_pointer_cast<SketchPlugin_Feature>(aSelectedFeature);
28 if (aSPFeature.get() != NULL && aShape.get() != NULL && !aShape->isNull()) {
29 // Processing of external (non-sketch) object
30 ObjectPtr aObj = PartSet_Tools::createFixedObjectByExternal(aShape->impl<TopoDS_Shape>(),
31 aSelectedObject, mySketch);
33 PartSet_WidgetShapeSelector* that = (PartSet_WidgetShapeSelector*) this;
34 aSelectedObject = aObj;
38 // Processing of sketch object
39 DataPtr aData = myFeature->data();
41 AttributePtr aAttr = aData->attribute(attributeID());
42 AttributeRefAttrPtr aRefAttr =
43 std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(aAttr);
45 TopoDS_Shape aTDSShape = aShape->impl<TopoDS_Shape>();
46 AttributePtr aPntAttr = PartSet_Tools::findAttributeBy2dPoint(aSelectedObject, aTDSShape, mySketch);
48 // this is an alternative, whether the attribute should be set or object in the attribute
49 // the first check is the attribute because the object already exist
50 // the object is set only if there is no selected attribute
51 // test case is - preselection for distance operation, which contains two points selected on lines
53 aRefAttr->setAttr(aPntAttr);
54 else if (aSelectedObject)
55 aRefAttr->setObject(aSelectedObject);
60 return ModuleBase_WidgetShapeSelector::storeAttributeValues(aSelectedObject, aShape);