Salome HOME
Debug of groups movement to the end for fillet feature unit-tests.
[modules/shaper.git] / src / Model / Model_AttributeSelection.cpp
1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "Model_AttributeSelection.h"
22 #include "Model_Application.h"
23 #include "Model_Events.h"
24 #include "Model_Data.h"
25 #include "Model_Document.h"
26 #include <Model_Objects.h>
27 #include <Model_AttributeSelectionList.h>
28 #include <Model_ResultConstruction.h>
29 #include <ModelAPI_Feature.h>
30 #include <ModelAPI_ResultBody.h>
31 #include <ModelAPI_ResultBody.h>
32 #include <ModelAPI_ResultConstruction.h>
33 #include <ModelAPI_ResultPart.h>
34 #include <ModelAPI_CompositeFeature.h>
35 #include <ModelAPI_Tools.h>
36 #include <ModelAPI_Session.h>
37 #include <ModelAPI_Validator.h>
38 #include <ModelGeomAlgo_Shape.h>
39 #include <Events_InfoMessage.h>
40 #include <GeomAPI_Edge.h>
41 #include <GeomAPI_Pnt.h>
42 #include <GeomAPI_ShapeIterator.h>
43 #include <GeomAPI_Vertex.h>
44 #include <GeomAPI_ShapeExplorer.h>
45 #include <GeomAlgoAPI_CompoundBuilder.h>
46 #include <GeomAlgoAPI_NExplode.h>
47 #include <Selector_Selector.h>
48
49 #include <TNaming_NamedShape.hxx>
50 #include <TNaming_Tool.hxx>
51 #include <TNaming_Builder.hxx>
52 #include <TNaming_SameShapeIterator.hxx>
53 #include <TNaming_NewShapeIterator.hxx>
54 #include <TNaming_Iterator.hxx>
55 #include <TDataStd_Integer.hxx>
56 #include <TDataStd_UAttribute.hxx>
57 #include <TDataStd_Name.hxx>
58 #include <TopTools_ListOfShape.hxx>
59 #include <TopTools_DataMapOfShapeShape.hxx>
60 #include <TopTools_MapOfShape.hxx>
61 #include <TopExp_Explorer.hxx>
62 #include <BRep_Tool.hxx>
63 #include <TopoDS.hxx>
64 #include <TopoDS_Edge.hxx>
65 #include <TopExp.hxx>
66 #include <TDF_ChildIterator.hxx>
67 #include <TDF_ChildIDIterator.hxx>
68 #include <TopoDS_Iterator.hxx>
69 #include <TDF_ChildIDIterator.hxx>
70 #include <Geom_Circle.hxx>
71 #include <Geom_Ellipse.hxx>
72 #include <BRep_Builder.hxx>
73
74 //#define DEB_NAMING 1
75 #ifdef DEB_NAMING
76 #include <BRepTools.hxx>
77 #endif
78 /// added to the index in the packed map to signalize that the vertex of edge is selected
79 /// (multiplied by the index of the edge)
80 static const int kSTART_VERTEX_DELTA = 1000000;
81 // identifier that there is simple reference: selection equals to context
82 Standard_GUID kSIMPLE_REF_ID("635eacb2-a1d6-4dec-8348-471fae17cb29");
83 // reference to Part sub-object
84 Standard_GUID kPART_REF_ID("635eacb2-a1d6-4dec-8348-471fae17cb27");
85 // selection is invalid after recomputation
86 Standard_GUID kINVALID_SELECTION("bce47fd7-80fa-4462-9d63-2f58acddd49d");
87
88 // identifier of the selection of the center of circle on edge
89 Standard_GUID kCIRCLE_CENTER("d0d0e0f1-217a-4b95-8fbb-0c4132f23718");
90 // identifier of the selection of the first focus point of ellipse on edge
91 Standard_GUID kELLIPSE_CENTER1("f70df04c-3168-4dc9-87a4-f1f840c1275d");
92 // identifier of the selection of the second focus point of ellipse on edge
93 Standard_GUID kELLIPSE_CENTER2("1395ae73-8e02-4cf8-b204-06ff35873a32");
94
95 // prefix for the whole feature context identification
96 const static std::string kWHOLE_FEATURE = "all-in-";
97
98 // on this label is stored:
99 // TNaming_NamedShape - selected shape
100 // TNaming_Naming - topological selection information (for the body)
101 // TDataStd_IntPackedMap - indexes of edges in composite element (for construction)
102 // TDataStd_Integer - type of the selected shape (for construction)
103 // TDF_Reference - from ReferenceAttribute, the context
104 bool Model_AttributeSelection::setValue(const ObjectPtr& theContext,
105   const std::shared_ptr<GeomAPI_Shape>& theSubShape, const bool theTemporarily)
106 {
107   if (theTemporarily &&
108       (!theContext.get() || theContext->groupName() != ModelAPI_Feature::group())) {
109     // just keep the stored without DF update
110     myTmpContext = std::dynamic_pointer_cast<ModelAPI_Result>(theContext);
111     myTmpSubShape = theSubShape;
112     owner()->data()->sendAttributeUpdated(this);
113     return true;
114   } else {
115     myTmpContext.reset();
116     myTmpSubShape.reset();
117     myTmpCenterType = NOT_CENTER;
118   }
119
120   CenterType aType;
121   const std::shared_ptr<GeomAPI_Shape>& anOldShape = internalValue(aType);
122   bool isOldContext = theContext == myRef.value();
123   bool isOldShape = isOldContext &&
124     (theSubShape == anOldShape || (theSubShape && anOldShape && theSubShape->isEqual(anOldShape)));
125   if (isOldShape) return false; // shape is the same, so context is also unchanged
126   bool aToUnblock = false;
127   // update the referenced object if needed
128   if (!isOldContext) {
129     aToUnblock = !owner()->data()->blockSendAttributeUpdated(true);
130     myRef.setValue(theContext);
131   }
132
133   // do not use naming if selected shape is result shape itself, but not sub-shape
134   TDF_Label aSelLab = selectionLabel();
135   aSelLab.ForgetAttribute(kSIMPLE_REF_ID);
136   aSelLab.ForgetAttribute(kINVALID_SELECTION);
137   aSelLab.ForgetAttribute(kCIRCLE_CENTER);
138   aSelLab.ForgetAttribute(kELLIPSE_CENTER1);
139   aSelLab.ForgetAttribute(kELLIPSE_CENTER2);
140
141   bool isDegeneratedEdge = false;
142   // do not use the degenerated edge as a shape, a null context and shape is used in the case
143   if (theSubShape.get() && !theSubShape->isNull() && theSubShape->isEdge()) {
144     const TopoDS_Shape& aSubShape = theSubShape->impl<TopoDS_Shape>();
145     if (aSubShape.ShapeType() == TopAbs_EDGE)
146       isDegeneratedEdge = BRep_Tool::Degenerated(TopoDS::Edge(aSubShape)) == Standard_True;
147   }
148   if (!theContext.get() || isDegeneratedEdge) {
149     // to keep the reference attribute label
150     TDF_Label aRefLab = myRef.myRef->Label();
151     aSelLab.ForgetAllAttributes(true);
152     myRef.myRef = TDF_Reference::Set(aSelLab.Father(), aSelLab.Father());
153     if (aToUnblock)
154       owner()->data()->blockSendAttributeUpdated(false);
155     return false;
156   }
157   bool isSelectBody = theContext->groupName() == ModelAPI_ResultBody::group();
158   if (!isSelectBody) {
159     ResultConstructionPtr aContextConstruction =
160       std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(theContext);
161     isSelectBody = aContextConstruction.get() && !aContextConstruction->isInfinite();
162   }
163   if (isSelectBody) {
164     ResultPtr aContextResult = std::dynamic_pointer_cast<ModelAPI_Result>(theContext);
165     GeomShapePtr aContextShape = aContextResult->shape();
166     // do not select the whole shape for body:it is already must be in the data framework
167     // equal and null selected objects mean the same: object is equal to context,
168     if (aContextShape.get() && (aContextShape->isEqual(theSubShape) || !theSubShape.get())) {
169       aSelLab.ForgetAllAttributes(true);
170       TDataStd_UAttribute::Set(aSelLab, kSIMPLE_REF_ID);
171     } else {
172       selectBody(aContextResult, theSubShape);
173     }
174   } else if (theContext->groupName() == ModelAPI_ResultConstruction::group()) {
175     aSelLab.ForgetAllAttributes(true); // to remove old selection data
176     std::shared_ptr<Model_ResultConstruction> aConstruction =
177       std::dynamic_pointer_cast<Model_ResultConstruction>(theContext);
178     std::shared_ptr<GeomAPI_Shape> aSubShape;
179     if (theSubShape.get() && !aConstruction->shape()->isEqual(theSubShape))
180       aSubShape = theSubShape; // the whole context
181   } else if (theContext->groupName() == ModelAPI_ResultPart::group()) {
182     aSelLab.ForgetAllAttributes(true);
183     TDataStd_UAttribute::Set(aSelLab, kPART_REF_ID);
184     selectPart(std::dynamic_pointer_cast<ModelAPI_Result>(theContext), theSubShape);
185   } else { // check the feature context: parent-Part of this feature should not be used
186     FeaturePtr aFeatureContext = std::dynamic_pointer_cast<ModelAPI_Feature>(theContext);
187     if (aFeatureContext.get()) {
188       if (owner()->document() != aFeatureContext->document()) {
189         aSelLab.ForgetAllAttributes(true);
190         myRef.setValue(ObjectPtr());
191         if (aToUnblock)
192           owner()->data()->blockSendAttributeUpdated(false);
193         return false;
194       }
195     }
196   }
197
198   owner()->data()->sendAttributeUpdated(this);
199
200   if (aToUnblock)
201     owner()->data()->blockSendAttributeUpdated(false);
202
203   return true;
204 }
205
206 void Model_AttributeSelection::setValueCenter(
207     const ObjectPtr& theContext, const std::shared_ptr<GeomAPI_Edge>& theEdge,
208     const CenterType theCenterType, const bool theTemporarily)
209 {
210   bool anUpdated = setValue(theContext, theEdge, theTemporarily);
211   if (theTemporarily) {
212     myTmpCenterType = theCenterType;
213   } else { // store in the data structure
214     TDF_Label aSelLab = selectionLabel();
215     switch(theCenterType) {
216     case CIRCLE_CENTER:
217       if (!anUpdated)
218         anUpdated = !aSelLab.IsAttribute(kCIRCLE_CENTER);
219       TDataStd_UAttribute::Set(aSelLab, kCIRCLE_CENTER);
220       break;
221     case ELLIPSE_FIRST_FOCUS:
222       if (!anUpdated)
223         anUpdated = !aSelLab.IsAttribute(kELLIPSE_CENTER1);
224       TDataStd_UAttribute::Set(aSelLab, kELLIPSE_CENTER1);
225       break;
226     case ELLIPSE_SECOND_FOCUS:
227       if (!anUpdated)
228         anUpdated = !aSelLab.IsAttribute(kELLIPSE_CENTER2);
229       TDataStd_UAttribute::Set(aSelLab, kELLIPSE_CENTER2);
230       break;
231     }
232     if (anUpdated)
233       owner()->data()->sendAttributeUpdated(this);
234   }
235 }
236
237 void Model_AttributeSelection::selectValue(
238     const std::shared_ptr<ModelAPI_AttributeSelection>& theSource)
239 {
240   CenterType aType;
241   std::shared_ptr<GeomAPI_Shape> aValue =
242     std::dynamic_pointer_cast<Model_AttributeSelection>(theSource)->internalValue(aType);
243   if (!aValue.get() || aType == NOT_CENTER) {
244     setValue(theSource->context(), aValue);
245   } else {
246     std::shared_ptr<GeomAPI_Edge> anEdge(new GeomAPI_Edge);
247     anEdge->setImpl(new TopoDS_Shape(aValue->impl<TopoDS_Shape>()));
248     setValueCenter(theSource->context(), anEdge, aType);
249   }
250 }
251
252 void Model_AttributeSelection::removeTemporaryValues()
253 {
254   if (myTmpContext.get() || myTmpSubShape.get()) {
255     myTmpContext.reset();
256     myTmpSubShape.reset();
257   }
258 }
259
260 // returns the center of the edge: circular or elliptical
261 GeomShapePtr centerByEdge(GeomShapePtr theEdge, ModelAPI_AttributeSelection::CenterType theType)
262 {
263   if (theType != ModelAPI_AttributeSelection::NOT_CENTER && theEdge.get() != NULL) {
264     TopoDS_Shape aShape = theEdge->impl<TopoDS_Shape>();
265     if (!aShape.IsNull() && aShape.ShapeType() == TopAbs_EDGE) {
266       TopoDS_Edge anEdge = TopoDS::Edge(aShape);
267       double aFirst, aLast;
268       Handle(Geom_Curve) aCurve = BRep_Tool::Curve(anEdge, aFirst, aLast);
269       if (!aCurve.IsNull()) {
270         TopoDS_Vertex aVertex;
271         BRep_Builder aBuilder;
272         if (theType == ModelAPI_AttributeSelection::CIRCLE_CENTER) {
273           Handle(Geom_Circle) aCirc = Handle(Geom_Circle)::DownCast(aCurve);
274           if (!aCirc.IsNull()) {
275             aBuilder.MakeVertex(aVertex, aCirc->Location(), Precision::Confusion());
276           }
277         } else { // ellipse
278           Handle(Geom_Ellipse) anEll = Handle(Geom_Ellipse)::DownCast(aCurve);
279           if (!anEll.IsNull()) {
280             aBuilder.MakeVertex(aVertex,
281               theType == ModelAPI_AttributeSelection::ELLIPSE_FIRST_FOCUS ?
282               anEll->Focus1() : anEll->Focus2(), Precision::Confusion());
283           }
284         }
285         if (!aVertex.IsNull()) {
286           std::shared_ptr<GeomAPI_Vertex> aResult(new GeomAPI_Vertex);
287           aResult->setImpl(new TopoDS_Vertex(aVertex));
288           return aResult;
289         }
290       }
291     }
292   }
293   return theEdge; // no vertex, so, return the initial edge
294 }
295
296 std::shared_ptr<GeomAPI_Shape> Model_AttributeSelection::value()
297 {
298   if (!ModelAPI_AttributeSelection::isInitialized() && !myTmpContext.get() && !myTmpSubShape.get())
299     return std::shared_ptr<GeomAPI_Shape>();
300   CenterType aType = NOT_CENTER;
301   std::shared_ptr<GeomAPI_Shape> aResult = internalValue(aType);
302   return centerByEdge(aResult, aType);
303 }
304
305 std::shared_ptr<GeomAPI_Shape> Model_AttributeSelection::internalValue(CenterType& theType)
306 {
307   theType = NOT_CENTER;
308   GeomShapePtr aResult;
309   if (myTmpContext.get() || myTmpSubShape.get()) {
310     theType = myTmpCenterType;
311     ResultConstructionPtr aResulConstruction =
312       std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(myTmpContext);
313     if(aResulConstruction.get()) {
314       // it is just reference to construction.
315       return myTmpSubShape;
316     }
317     return myTmpSubShape.get() ? myTmpSubShape : myTmpContext->shape();
318   }
319
320   TDF_Label aSelLab = selectionLabel();
321   if (aSelLab.IsAttribute(kINVALID_SELECTION))
322     return aResult;
323
324   if (aSelLab.IsAttribute(kCIRCLE_CENTER))
325     theType = CIRCLE_CENTER;
326   else if (aSelLab.IsAttribute(kELLIPSE_CENTER1))
327     theType = ELLIPSE_FIRST_FOCUS;
328   else if (aSelLab.IsAttribute(kELLIPSE_CENTER2))
329     theType = ELLIPSE_SECOND_FOCUS;
330
331
332   if (myRef.isInitialized()) {
333     if (aSelLab.IsAttribute(kSIMPLE_REF_ID)) { // it is just reference to shape, not sub-shape
334       ResultPtr aContext = context();
335       if (!aContext.get() || aContext->groupName() == ModelAPI_ResultConstruction::group())
336         return aResult; // empty result, for whole construction selection also
337       return aContext->shape();
338     }
339     if (aSelLab.IsAttribute(kPART_REF_ID)) {
340       ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(context());
341       if (!aPart.get() || !aPart->isActivated())
342         return std::shared_ptr<GeomAPI_Shape>(); // postponed naming needed
343       Handle(TDataStd_Integer) anIndex;
344       if (aSelLab.FindAttribute(TDataStd_Integer::GetID(), anIndex)) {
345         if (anIndex->Get()) { // special selection attribute was created, use it
346           return aPart->selectionValue(anIndex->Get());
347         } else { // face with name is already in the data model, so try to take it by name
348           Handle(TDataStd_Name) aName;
349           if (aSelLab.FindAttribute(TDataStd_Name::GetID(), aName)) {
350             std::string aSubShapeName(TCollection_AsciiString(aName->Get()).ToCString());
351             std::size_t aPartEnd = aSubShapeName.find('/');
352             if (aPartEnd != std::string::npos && aPartEnd != aSubShapeName.rfind('/')) {
353               std::string aNameInPart = aSubShapeName.substr(aPartEnd + 1);
354               int anIndex;
355               std::string aType; // to reuse already existing selection the type is not needed
356               return aPart->shapeInPart(aNameInPart, aType, anIndex);
357             }
358           }
359         }
360       }
361     }
362
363     std::shared_ptr<Model_ResultConstruction> aConstr =
364       std::dynamic_pointer_cast<Model_ResultConstruction>(context());
365     if (aConstr) {
366       if (aConstr->isInfinite())
367         return aResult; // empty result
368     }
369     if (!aConstr.get()) { // for construction context, return empty result as usual even
370       // the whole feature is selected
371       FeaturePtr aFeature = contextFeature();
372       if (aFeature.get()) {
373         std::list<GeomShapePtr> allShapes;
374         std::list<ResultPtr>::const_iterator aRes = aFeature->results().cbegin();
375         for (; aRes != aFeature->results().cend(); aRes++) {
376           if (aRes->get() && !(*aRes)->isDisabled()) {
377             GeomShapePtr aShape = (*aRes)->shape();
378             if (aShape.get() && !aShape->isNull()) {
379               allShapes.push_back(aShape);
380             }
381           }
382         }
383         return GeomAlgoAPI_CompoundBuilder::compound(allShapes);
384       }
385     } else {
386       if (contextFeature().get()) {
387         return aResult; // for the whole sketch feature selected return null => all faces
388       }
389     }
390
391     Handle(TNaming_NamedShape) aSelection;
392     if (aSelLab.FindAttribute(TNaming_NamedShape::GetID(), aSelection)) {
393       TopoDS_Shape aSelShape = aSelection->Get();
394       aResult = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape);
395       aResult->setImpl(new TopoDS_Shape(aSelShape));
396     } else if (aConstr) { // simple construction element: just shape of this construction element
397       aResult = aConstr->shape();
398     }
399   }
400   return aResult;
401 }
402
403 bool Model_AttributeSelection::isInvalid()
404 {
405   return selectionLabel().IsAttribute(kINVALID_SELECTION) == Standard_True;
406 }
407
408 bool Model_AttributeSelection::isInitialized()
409 {
410   if (ModelAPI_AttributeSelection::isInitialized()) { // additional checks if it is initialized
411     std::shared_ptr<GeomAPI_Shape> aResult;
412     if (myRef.isInitialized()) {
413       TDF_Label aSelLab = selectionLabel();
414       if (aSelLab.IsAttribute(kSIMPLE_REF_ID)) { // it is just reference to shape, not sub-shape
415         ResultPtr aContext = context();
416         return aContext.get() != NULL;
417       }
418       Handle(TNaming_NamedShape) aSelection;
419       if (selectionLabel().FindAttribute(TNaming_NamedShape::GetID(), aSelection)) {
420         return !aSelection->Get().IsNull();
421       } else { // for simple construction element: just shape of this construction element
422         if (myRef.value().get())
423           return true;
424         // check that this is on open of document, so, results are not initialized yet
425         TDF_Label aRefLab = myRef.myRef->Get();
426         if (aRefLab.IsNull() || !owner().get())
427           return false;
428         std::shared_ptr<Model_Document> aMyDoc =
429           std::dynamic_pointer_cast<Model_Document>(owner()->document());
430         if (!aMyDoc.get())
431           return false;
432         // check at least the feature exists
433         return aMyDoc->featureByLab(aRefLab).get() != NULL;
434       }
435     }
436   }
437   return false;
438 }
439
440 Model_AttributeSelection::Model_AttributeSelection(TDF_Label& theLabel)
441 : myRef(theLabel),
442   myIsGeometricalSelection(false)
443 {
444   myIsInitialized = myRef.isInitialized();
445   myParent = NULL;
446 }
447
448 void Model_AttributeSelection::setID(const std::string theID)
449 {
450   myRef.setID(theID);
451   ModelAPI_AttributeSelection::setID(theID);
452   FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(owner());
453   if (myParent) {
454     // to be able to select as geometrical selection and then - split
455     myIsGeometricalSelection = true; // myParent->isGeometricalSelection();
456   } else {
457     myIsGeometricalSelection =
458       ModelAPI_Session::get()->validators()->isGeometricalSelection(aFeature->getKind(), id());
459   }
460 }
461
462 ResultPtr Model_AttributeSelection::context()
463 {
464   if (!ModelAPI_AttributeSelection::isInitialized() && !myTmpContext.get() && !myTmpSubShape.get())
465     return ResultPtr();
466
467   if (myTmpContext.get() || myTmpSubShape.get()) {
468     return myTmpContext;
469   }
470
471   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(myRef.value());
472   // for parts there could be same-data result, so take the last enabled
473   if (aResult.get()) {
474     if(aResult->groupName() == ModelAPI_ResultPart::group()) {
475       int aSize = aResult->document()->size(ModelAPI_ResultPart::group());
476       for(int a = aSize - 1; a >= 0; a--) {
477         ObjectPtr aPart = aResult->document()->object(ModelAPI_ResultPart::group(), a);
478         if(aPart.get() && aPart->data() == aResult->data()) {
479           ResultPtr aPartResult = std::dynamic_pointer_cast<ModelAPI_Result>(aPart);
480           FeaturePtr anOwnerFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(owner());
481           // check that this result is not this-feature result (it is forbidden to select itself)
482           if(anOwnerFeature.get() && anOwnerFeature->firstResult() != aPartResult) {
483             return aPartResult;
484           }
485         }
486       }
487     }
488   } else { // if feature - construction is selected, it has only one result, return this result
489     FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(myRef.value());
490     if (aFeature.get() && aFeature->results().size() == 1 &&
491         aFeature->firstResult()->groupName() == ModelAPI_ResultConstruction::group())
492       return aFeature->firstResult();
493   }
494   return aResult;
495 }
496
497 FeaturePtr Model_AttributeSelection::contextFeature() {
498   if (myTmpContext.get() || myTmpSubShape.get()) {
499     return FeaturePtr(); // feature can not be selected temporarily
500   }
501   return std::dynamic_pointer_cast<ModelAPI_Feature>(myRef.value());
502 }
503 ObjectPtr Model_AttributeSelection::contextObject() {
504   FeaturePtr aRes = contextFeature();
505   if (aRes.get())
506     return aRes;
507   return context();
508 }
509
510
511 void Model_AttributeSelection::setObject(const std::shared_ptr<ModelAPI_Object>& theObject)
512 {
513   ModelAPI_AttributeSelection::setObject(theObject);
514   myRef.setObject(theObject);
515 }
516
517 /// Sets the invalid flag if flag is false, or removes it if "true"
518 /// Returns theFlag
519 static bool setInvalidIfFalse(TDF_Label& theLab, const bool theFlag) {
520   if (theFlag) {
521     theLab.ForgetAttribute(kINVALID_SELECTION);
522   } else {
523     TDataStd_UAttribute::Set(theLab, kINVALID_SELECTION);
524   }
525   return theFlag;
526 }
527
528 void Model_AttributeSelection::split(
529   ResultPtr theContext, TopoDS_Shape theNewShape, TopAbs_ShapeEnum theType)
530 {
531   TopTools_ListOfShape aSubs;
532   for(TopoDS_Iterator anExplorer(theNewShape); anExplorer.More(); anExplorer.Next()) {
533     if (!anExplorer.Value().IsNull() &&
534       anExplorer.Value().ShapeType() == theType) {
535         aSubs.Append(anExplorer.Value());
536     } else { // invalid case; bad result shape, so, impossible to split easily
537       aSubs.Clear();
538       break;
539     }
540   }
541   if (aSubs.Extent() > 1) { // ok to split
542     TopTools_ListIteratorOfListOfShape aSub(aSubs);
543     GeomShapePtr aSubSh(new GeomAPI_Shape);
544     aSubSh->setImpl(new TopoDS_Shape(aSub.Value()));
545     setValue(theContext, aSubSh);
546     for(aSub.Next(); aSub.More(); aSub.Next()) {
547       GeomShapePtr aSubSh(new GeomAPI_Shape);
548       aSubSh->setImpl(new TopoDS_Shape(aSub.Value()));
549       myParent->append(theContext, aSubSh);
550     }
551   }
552 }
553
554 bool Model_AttributeSelection::update()
555 {
556   FeaturePtr aContextFeature = contextFeature();
557   if (aContextFeature.get()) {
558     owner()->data()->sendAttributeUpdated(this);  // send updated if "update" called in any way
559     return true;
560   }
561   TDF_Label aSelLab = selectionLabel();
562   ResultPtr aContext = context();
563   if (!aContext.get())
564     return setInvalidIfFalse(aSelLab, false);
565   if (aSelLab.IsAttribute(kSIMPLE_REF_ID)) { // it is just reference to shape, not sub-shape
566     return setInvalidIfFalse(aSelLab, aContext->shape() && !aContext->shape()->isNull());
567   }
568
569   if (aSelLab.IsAttribute(kPART_REF_ID)) { // it is reference to the part object
570     std::shared_ptr<GeomAPI_Shape> aNoSelection;
571     bool aResult = selectPart(aContext, aNoSelection, true);
572     aResult = setInvalidIfFalse(aSelLab, aResult);
573     if (aResult) {
574       owner()->data()->sendAttributeUpdated(this);
575     }
576     return aResult;
577   }
578
579   if (aContext->groupName() == ModelAPI_ResultBody::group()) {
580     // body: just a named shape, use topological selection mechanism
581     bool aResult = false;
582     TopoDS_Shape anOldShape;
583     Handle(TNaming_NamedShape) aNS;
584     if (aSelLab.FindAttribute(TNaming_NamedShape::GetID(), aNS))
585       anOldShape = aNS->Get();
586
587     TopoDS_Shape aContextShape = aContext->shape()->impl<TopoDS_Shape>();
588     Selector_Selector aSelector(aSelLab, baseDocumentLab());
589     aResult = aSelector.restore(aContextShape);
590     setInvalidIfFalse(aSelLab, aResult);
591
592     TopoDS_Shape aNewShape;
593     if (aSelLab.FindAttribute(TNaming_NamedShape::GetID(), aNS))
594       aNewShape = aNS->Get();
595
596     if (anOldShape.IsNull() || aNewShape.IsNull() || !anOldShape.IsEqual(aNewShape)) {
597       // shape type should not be changed: if shape becomes compound of such shapes, then split
598       if (myParent && !anOldShape.IsNull() && !aNewShape.IsNull() &&
599           anOldShape.ShapeType() != aNewShape.ShapeType() &&
600           aNewShape.ShapeType() == TopAbs_COMPOUND) {
601         split(aContext, aNewShape, anOldShape.ShapeType());
602       }
603       owner()->data()->sendAttributeUpdated(this);  // send updated if shape is changed
604     }
605     return aResult;
606   }
607
608   if (aContext->groupName() == ModelAPI_ResultConstruction::group()) {
609     bool aResult = true;
610     std::shared_ptr<Model_ResultConstruction> aConstructionContext =
611       std::dynamic_pointer_cast<Model_ResultConstruction>(aContext);
612     if (!aConstructionContext->isInfinite()) {
613       TopoDS_Shape aContextShape = aContext->shape()->impl<TopoDS_Shape>();
614       Selector_Selector aSelector(aSelLab, baseDocumentLab());
615       TopoDS_Shape anOldShape = aSelector.value();
616       aResult = aSelector.restore(aContextShape);
617       setInvalidIfFalse(aSelLab, aResult);
618       if (aResult && !anOldShape.IsEqual(aSelector.value()))
619         owner()->data()->sendAttributeUpdated(this);  // send updated if shape is changed
620     } else {
621       owner()->data()->sendAttributeUpdated(this);  // send updated if "update" called in any way
622     }
623     return aResult;
624   }
625   return setInvalidIfFalse(aSelLab, false); // unknown case
626 }
627
628 void Model_AttributeSelection::selectBody(
629   const ResultPtr& theContext, const std::shared_ptr<GeomAPI_Shape>& theSubShape)
630 {
631   // perform the selection
632   TopoDS_Shape aContext;
633
634   ResultPtr aBody = std::dynamic_pointer_cast<ModelAPI_Result>(theContext);
635   if (aBody) {
636     aContext = aBody->shape()->impl<TopoDS_Shape>();
637   } else {
638     ResultPtr aResult =
639       std::dynamic_pointer_cast<ModelAPI_Result>(myRef.value());
640     if (aResult) {
641       aContext = aResult->shape()->impl<TopoDS_Shape>();
642     } else {
643       Events_InfoMessage("Model_AttributeSelection", "A result with shape is expected").send();
644       return;
645     }
646   }
647
648   if (!aContext.IsNull()) {
649     TDF_Label aSelLab = selectionLabel();
650     TopoDS_Shape aNewSub = theSubShape->impl<TopoDS_Shape>();
651
652     bool aSelectorOk = true;
653     Selector_Selector aSelector(aSelLab, baseDocumentLab());
654     try {
655       aSelectorOk = aSelector.select(aContext, aNewSub, myIsGeometricalSelection);
656       if (aSelectorOk) {
657         aSelectorOk = aSelector.store(aContext);
658       }
659     } catch(...) {
660       aSelectorOk = false;
661     }
662     if (aSelectorOk) {
663       TopoDS_Shape aShape = aSelector.value();
664       aSelectorOk = !aShape.IsNull() && aShape.ShapeType() == aNewSub.ShapeType();
665     }
666     setInvalidIfFalse(aSelLab, aSelectorOk);
667   }
668 }
669
670 bool Model_AttributeSelection::selectPart(
671   const ResultPtr& theContext, const std::shared_ptr<GeomAPI_Shape>& theSubShape,
672   const bool theUpdate)
673 {
674   ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(theContext);
675   if (!aPart.get() || !aPart->isActivated())
676     return true; // postponed naming
677   if (theUpdate) {
678     Handle(TDataStd_Integer) anIndex;
679     if (selectionLabel().FindAttribute(TDataStd_Integer::GetID(), anIndex)) {
680       // by internal selection
681       if (anIndex->Get() > 0) {
682         // update the selection by index
683         return aPart->updateInPart(anIndex->Get());
684       } else {
685         return true; // nothing to do, referencing just by name
686       }
687     }
688     return true; // nothing to do, referencing just by name
689   }
690   // store the shape (in case part is not loaded it should be useful
691   TopoDS_Shape aShape;
692   std::string aName = theContext->data()->name();
693   if (!theSubShape.get() || theSubShape->isNull()) {// the whole part shape is selected
694     aShape = theContext->shape()->impl<TopoDS_Shape>();
695   } else {
696     aShape = theSubShape->impl<TopoDS_Shape>();
697     int anIndex;
698     aName += "/" + aPart->nameInPart(theSubShape, anIndex);
699     TDataStd_Integer::Set(selectionLabel(), anIndex);
700   }
701   TNaming_Builder aBuilder(selectionLabel());
702   aBuilder.Select(aShape, aShape);
703   // identify by name in the part
704   TDataStd_Name::Set(selectionLabel(), aName.c_str());
705   return !aName.empty();
706 }
707
708 TDF_Label Model_AttributeSelection::selectionLabel()
709 {
710   return myRef.myRef->Label().FindChild(1);
711 }
712
713 /// prefixes of the shape names with centers defined
714 static std::map<ModelAPI_AttributeSelection::CenterType, std::string> kCENTERS_PREFIX;
715
716 /// returns the map that contains all possible prefixes of the center-names
717 static std::map<ModelAPI_AttributeSelection::CenterType, std::string>& centersMap()
718 {
719   if (kCENTERS_PREFIX.empty()) { // fill map by initial values
720     kCENTERS_PREFIX[ModelAPI_AttributeSelection::CIRCLE_CENTER] = "__cc";
721     kCENTERS_PREFIX[ModelAPI_AttributeSelection::ELLIPSE_FIRST_FOCUS] = "__eff";
722     kCENTERS_PREFIX[ModelAPI_AttributeSelection::ELLIPSE_SECOND_FOCUS] = "__esf";
723   }
724   return kCENTERS_PREFIX;
725 }
726
727 std::string Model_AttributeSelection::namingName(const std::string& theDefaultName)
728 {
729   std::string aName("");
730   if(!this->isInitialized())
731     return !theDefaultName.empty() ? theDefaultName : aName;
732
733   CenterType aCenterType = NOT_CENTER;
734   std::shared_ptr<GeomAPI_Shape> aSubSh = internalValue(aCenterType);
735   ResultPtr aCont = context();
736
737   if (!aCont.get() ||
738       (aCont->groupName() == ModelAPI_ResultConstruction::group() && contextFeature().get())) {
739     // selection of a full feature
740     FeaturePtr aFeatureCont = contextFeature();
741     if (aFeatureCont.get()) {
742       return kWHOLE_FEATURE + aFeatureCont->name();
743     }
744     // in case of selection of removed result
745     return "";
746   }
747
748   TDF_Label aSelLab = selectionLabel();
749   if (aSelLab.IsAttribute(kSIMPLE_REF_ID)) { // whole context, no value
750     return contextName(aCont);
751   }
752
753   // if it is in result of another part
754   if (aCont->groupName() == ModelAPI_ResultPart::group()) {
755     ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aCont);
756     int anIndex;
757     GeomShapePtr aValue = value();
758     if (aValue.get())
759       return aPart->data()->name() + "/" + aPart->nameInPart(aValue, anIndex);
760     else
761       return aPart->data()->name();
762   }
763
764
765   // whole infinitive construction
766   if (aCont->groupName() == ModelAPI_ResultConstruction::group()) {
767     ResultConstructionPtr aConstr = std::dynamic_pointer_cast<Model_ResultConstruction>(aCont);
768     if (aConstr->isInfinite()) {
769       return contextName(aCont);
770     }
771   }
772
773   Selector_Selector aSelector(aSelLab, baseDocumentLab());
774   std::string aResult;
775   if (aCont->shape().get() && aSelector.restore(aCont->shape()->impl<TopoDS_Shape>()))
776     aResult = aSelector.name(this);
777   if (aCenterType != NOT_CENTER) {
778     aResult += centersMap()[aCenterType];
779   }
780   return aResult;
781 }
782
783 // returns the center type and modifies the shape name if this name is center-name
784 static ModelAPI_AttributeSelection::CenterType centerTypeByName(std::string& theShapeName)
785 {
786   std::map<ModelAPI_AttributeSelection::CenterType, std::string>::iterator aPrefixIter =
787     centersMap().begin();
788   for(; aPrefixIter != centersMap().end(); aPrefixIter++) {
789     std::size_t aFound = theShapeName.find(aPrefixIter->second);
790     if (aFound != std::string::npos &&
791         aFound == theShapeName.size() - aPrefixIter->second.size()) {
792       theShapeName = theShapeName.substr(0, aFound);
793       return aPrefixIter->first;
794     }
795   }
796   return ModelAPI_AttributeSelection::NOT_CENTER;
797 }
798
799 // type ::= COMP | COMS | SOLD | SHEL | FACE | WIRE | EDGE | VERT
800 void Model_AttributeSelection::selectSubShape(
801   const std::string& theType, const std::string& theSubShapeName)
802 {
803   if(theSubShapeName.empty() || theType.empty()) return;
804
805   std::string aSubShapeName = theSubShapeName;
806   CenterType aCenterType = theType[0] == 'v' || theType[0] == 'V' ? // only for vertex-type
807     centerTypeByName(aSubShapeName) : NOT_CENTER;
808   std::string aType = aCenterType == NOT_CENTER ? theType : "EDGE"; // search for edge now
809   static const GeomShapePtr anEmptyShape;
810
811   // first iteration is selection by name without center prefix, second - in case of problem,
812   // try with initial name
813   for(int aUseCenter = 1; aUseCenter >= 0; aUseCenter--)  {
814     if (aUseCenter == 0 && aCenterType != NOT_CENTER) {
815       aSubShapeName = theSubShapeName;
816       aCenterType = NOT_CENTER;
817       aType = theType;
818     } else if (aUseCenter != 1) continue;
819     TopAbs_ShapeEnum aShapeType =  TopAbs_ShapeEnum(GeomAPI_Shape::shapeTypeByStr(aType));
820
821     std::shared_ptr<Model_Document> aDoc =
822       std::dynamic_pointer_cast<Model_Document>(owner()->document());
823     // check this is Part-name: 2 delimiters in the name
824     std::size_t aPartEnd = aSubShapeName.find('/');
825     if (aPartEnd != std::string::npos) {
826       std::string aPartName = aSubShapeName.substr(0, aPartEnd);
827       DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
828       if (aPartName == aRootDoc->kind()) {
829         aDoc = std::dynamic_pointer_cast<Model_Document>(aRootDoc);
830         aSubShapeName = aSubShapeName.substr(aPartEnd + 1);
831       } else {
832         ObjectPtr aFound =
833           owner()->document()->objectByName(ModelAPI_ResultPart::group(), aPartName);
834         if (aFound.get()) { // found such part, so asking it for the name
835           ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aFound);
836           std::string aNameInPart = aSubShapeName.substr(aPartEnd + 1);
837           if (aNameInPart.empty()) { // whole part
838             setValue(aPart, anEmptyShape);
839             return;
840           }
841           int anIndex;
842           std::shared_ptr<GeomAPI_Shape> aSelected =
843             aPart->shapeInPart(aNameInPart, aType, anIndex);
844           if (aSelected.get()) {
845             if (aCenterType != NOT_CENTER) {
846               if (!aSelected->isEdge())
847                 continue;
848               std::shared_ptr<GeomAPI_Edge> aSelectedEdge(new GeomAPI_Edge(aSelected));
849               setValueCenter(aPart, aSelectedEdge, aCenterType);
850             } else
851               setValue(aPart, aSelected);
852             TDataStd_Integer::Set(selectionLabel(), anIndex);
853             return;
854           }
855         }
856       }
857     }
858
859     // check this is a whole feature context
860     if (aSubShapeName.size() > kWHOLE_FEATURE.size() &&
861       aSubShapeName.substr(0, kWHOLE_FEATURE.size()) == kWHOLE_FEATURE) {
862       std::string aFeatureName = aSubShapeName.substr(kWHOLE_FEATURE.size());
863       ObjectPtr anObj = aDoc->objectByName(ModelAPI_Feature::group(), aFeatureName);
864       if (anObj.get()) {
865         setValue(anObj, anEmptyShape);
866         return;
867       }
868     }
869
870     // the whole result selection check
871     if (aSubShapeName.find('/') == std::string::npos) {
872       ObjectPtr aRes = aDoc->objectByName(ModelAPI_ResultConstruction::group(), aSubShapeName);
873       if (!aRes.get())
874         aRes = aDoc->objectByName(ModelAPI_ResultBody::group(), aSubShapeName);
875       if (aRes.get()) {
876         setValue(aRes, anEmptyShape);
877         return;
878       }
879     }
880
881     Selector_Selector aSelector(selectionLabel(), baseDocumentLab());
882     myRestoreDocument = aDoc;
883     TDF_Label aContextLabel = aSelector.restoreByName(
884       aSubShapeName, aShapeType, this, myIsGeometricalSelection);
885     myRestoreDocument.reset();
886     if (!aContextLabel.IsNull()) {
887       ResultPtr aContext = aDoc->resultByLab(aContextLabel); // any label for document access
888       if (aContext.get() && aContext->shape().get()) {
889         TopoDS_Shape aContextShape = aContext->shape()->impl<TopoDS_Shape>();
890         if (aSelector.solve(aContextShape)) {
891           TopoDS_Shape aSelectorShape = aSelector.value();
892           GeomShapePtr aShapeToBeSelected(new GeomAPI_Shape);
893           aShapeToBeSelected->setImpl<TopoDS_Shape>(new TopoDS_Shape(aSelectorShape));
894           // make the context result the latest existing
895           aContext = newestContext(aContext, aShapeToBeSelected);
896           if (myIsGeometricalSelection || aCenterType == NOT_CENTER) {
897             // store the currently generated name
898             selectionLabel().ForgetAllAttributes(true);
899             bool aToUnblock = false;
900             aToUnblock = !owner()->data()->blockSendAttributeUpdated(true);
901             myRef.setValue(aContext);
902             aSelector.store(aContextShape);
903             owner()->data()->sendAttributeUpdated(this);
904             if (aToUnblock)
905               owner()->data()->blockSendAttributeUpdated(false);
906             return;
907           } else { // re-select center of circle/arc by context and value
908             if (!aShapeToBeSelected->isEdge())
909               continue;
910             std::shared_ptr<GeomAPI_Edge> aSelectedEdge(new GeomAPI_Edge(aShapeToBeSelected));
911             setValueCenter(aContext, aSelectedEdge, aCenterType);
912           }
913           return;
914         }
915       }
916     }
917   }
918   // invalid
919   TDF_Label aSelLab = selectionLabel();
920   setInvalidIfFalse(aSelLab, false);
921   reset();
922 }
923
924 void Model_AttributeSelection::selectSubShape(const std::string& theType,
925                                               const GeomPointPtr& thePoint)
926 {
927   if (theType.empty() || !thePoint)
928     return;
929
930   // list of parent features
931   FeaturePtr anOwner = ModelAPI_Feature::feature(owner());
932   std::set<FeaturePtr> aParents = ModelAPI_Tools::getParents(anOwner);
933
934   int aSelectionIndex = 0;
935   GeomAPI_Shape::ShapeType aType = GeomAPI_Shape::shapeTypeByStr(theType);
936   if (aType == GeomAPI_Shape::SHAPE) {
937     // possibly, the string consists of the type and the index,
938     // thus, try to separate them
939     size_t aUndersporePos = theType.find_first_of('_');
940     if (aUndersporePos != std::string::npos)
941       aType = GeomAPI_Shape::shapeTypeByStr(theType.substr(0, aUndersporePos));
942
943     if (aType != GeomAPI_Shape::SHAPE) {
944       for (std::string::const_iterator aChar = theType.begin() + aUndersporePos + 1;
945            aChar != theType.end(); ++aChar) {
946         if (std::isdigit(*aChar))
947           aSelectionIndex = aSelectionIndex * 10 + (*aChar - '0');
948         else {
949           aSelectionIndex = 1;
950           break;
951         }
952       }
953       aSelectionIndex -= 1;
954     }
955   }
956
957   std::list<ModelGeomAlgo_Shape::SubshapeOfResult> anAppropriate;
958
959   // collect features from PartSet and the current part
960   SessionPtr aSession = ModelAPI_Session::get();
961   std::list<FeaturePtr> aFeatures = aSession->moduleDocument()->allFeatures();
962   if (aSession->moduleDocument() != owner()->document()) {
963     std::list<FeaturePtr> aPartFeatures = owner()->document()->allFeatures();
964     aFeatures.insert(aFeatures.end(), aPartFeatures.begin(), aPartFeatures.end());
965   }
966   // Process results of all features from the last to the first
967   // to find appropriate sub-shape
968   for (std::list<FeaturePtr>::const_reverse_iterator anIt = aFeatures.rbegin();
969        anIt != aFeatures.rend(); ++anIt) {
970     // selection cannot be linked to the parent features
971     if (aParents.find(*anIt) != aParents.end())
972       continue;
973     // check the feature is a part of composite feature (like sketch elements),
974     // then do not process it, it will be processed in scope of composite feature
975     bool isSubOfComposite = false;
976     const std::set<AttributePtr>& aRefs = (*anIt)->data()->refsToMe();
977     for (std::set<AttributePtr>::const_iterator aRefIt = aRefs.begin();
978          aRefIt != aRefs.end() && !isSubOfComposite; ++aRefIt) {
979       FeaturePtr aFeature = ModelAPI_Feature::feature((*aRefIt)->owner());
980       CompositeFeaturePtr aCompFeature =
981           std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aFeature);
982       isSubOfComposite = aCompFeature && aCompFeature->isSub(*anIt);
983     }
984     if (isSubOfComposite)
985       continue;
986
987     // process results of the current feature to find appropriate sub-shape
988     if (ModelGeomAlgo_Shape::findSubshapeByPoint(*anIt, thePoint, aType, anAppropriate)) {
989       std::list<ModelGeomAlgo_Shape::SubshapeOfResult>::iterator anApIt = anAppropriate.begin();
990       for (; aSelectionIndex > 0 && anApIt != anAppropriate.end(); --aSelectionIndex)
991         ++anApIt; // skip this shape, because one of the previous is selected
992
993       if (anApIt != anAppropriate.end()) {
994         if (anApIt->myCenterType == (int)ModelAPI_AttributeSelection::NOT_CENTER)
995           setValue(anApIt->myResult, anApIt->mySubshape);
996         else
997           setValueCenter(anApIt->myResult, anApIt->mySubshape->edge(),
998                          (ModelAPI_AttributeSelection::CenterType)anApIt->myCenterType);
999         return;
1000       }
1001     }
1002   }
1003
1004   TDF_Label aSelLab = selectionLabel();
1005   setInvalidIfFalse(aSelLab, false);
1006   reset();
1007 }
1008
1009 void Model_AttributeSelection::selectSubShape(const std::string& theType,
1010   const std::string& theContextName, const int theIndex)
1011 {
1012   // selection of context by name
1013   selectSubShape(theType, theContextName);
1014   ResultPtr aContext = context();
1015   if (aContext.get()) {
1016     GeomShapePtr aContShape = aContext->shape();
1017     if (aContShape.get()) {
1018       GeomAlgoAPI_NExplode aNExp(aContShape, GeomAPI_Shape::shapeTypeByStr(theType));
1019       GeomShapePtr aValue = aNExp.shape(theIndex);
1020       if (aValue.get())
1021         setValue(aContext, aValue);
1022     }
1023   }
1024 }
1025
1026 void Model_AttributeSelection::setId(int theID)
1027 {
1028   std::shared_ptr<GeomAPI_Shape> aSelection;
1029
1030   ResultPtr aContextRes = context();
1031   // support for compsolids:
1032   while(ModelAPI_Tools::bodyOwner(aContextRes).get()) {
1033     aContextRes = ModelAPI_Tools::bodyOwner(aContextRes);
1034   }
1035   std::shared_ptr<GeomAPI_Shape> aContext = aContextRes->shape();
1036
1037   TopoDS_Shape aMainShape = aContext->impl<TopoDS_Shape>();
1038   // searching for the latest main shape
1039   if (theID > 0 && aContext && !aContext->isNull())
1040   {
1041     std::shared_ptr<Model_Document> aDoc =
1042       std::dynamic_pointer_cast<Model_Document>(aContextRes->document());
1043     if (aDoc.get()) {
1044       Handle(TNaming_NamedShape) aNS = TNaming_Tool::NamedShape(aMainShape, aDoc->generalLabel());
1045       if (!aNS.IsNull()) {
1046         aMainShape = TNaming_Tool::CurrentShape(aNS);
1047       }
1048     }
1049
1050     TopTools_IndexedMapOfShape aSubShapesMap;
1051     TopExp::MapShapes(aMainShape, aSubShapesMap);
1052     const TopoDS_Shape& aSelShape = aSubShapesMap.FindKey(theID);
1053
1054     std::shared_ptr<GeomAPI_Shape> aResult(new GeomAPI_Shape);
1055     aResult->setImpl(new TopoDS_Shape(aSelShape));
1056
1057     aSelection = aResult;
1058   }
1059
1060   setValue(aContextRes, aSelection);
1061 }
1062
1063 std::string Model_AttributeSelection::contextName(const ResultPtr& theContext) const
1064 {
1065   std::string aResult;
1066   if (owner()->document() != theContext->document()) {
1067     if (theContext->document() == ModelAPI_Session::get()->moduleDocument()) {
1068       aResult = theContext->document()->kind() + "/";
1069     } else {
1070       ResultPtr aDocRes = ModelAPI_Tools::findPartResult(
1071         ModelAPI_Session::get()->moduleDocument(), theContext->document());
1072       if (aDocRes.get()) {
1073         aResult = aDocRes->data()->name() + "/";
1074       }
1075     }
1076   }
1077   aResult += theContext->data()->name();
1078   return aResult;
1079 }
1080
1081 void Model_AttributeSelection::computeValues(
1082   ResultPtr theOldContext, ResultPtr theNewContext, TopoDS_Shape theValShape,
1083   TopTools_ListOfShape& theShapes)
1084 {
1085   bool aWasWholeContext = theValShape.IsNull();
1086   if (aWasWholeContext) {
1087     theValShape = theOldContext->shape()->impl<TopoDS_Shape>();
1088   }
1089   TopAbs_ShapeEnum aValType = theValShape.ShapeType();
1090   TopoDS_Shape aNewContShape = theNewContext->shape()->impl<TopoDS_Shape>();
1091   // if a new value is unchanged in the new context, do nothing: value is correct
1092   TopExp_Explorer aSubExp(aNewContShape, aValType);
1093   for(; aSubExp.More(); aSubExp.Next()) {
1094     if (aSubExp.Current().IsSame(theValShape)) {
1095       theShapes.Append(theValShape);
1096       return;
1097     }
1098   }
1099   // if new context becomes compsolid, the resulting sub may be in sub-solids
1100   std::list<ResultPtr> aNewToIterate;
1101   aNewToIterate.push_back(theNewContext);
1102   ResultBodyPtr aComp = std::dynamic_pointer_cast<ModelAPI_ResultBody>(theNewContext);
1103   if (aComp.get()) {
1104     std::list<ResultPtr> allNewContextSubs;
1105     ModelAPI_Tools::allSubs(aComp, allNewContextSubs);
1106     std::list<ResultPtr>::iterator aSub = allNewContextSubs.begin();
1107     for(; aSub != allNewContextSubs.end(); aSub++) {
1108       ResultBodyPtr aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aSub);
1109       if (aBody.get() && aBody->numberOfSubs() == 0) // add only lower level subs
1110         aNewToIterate.push_back(aBody);
1111     }
1112   }
1113
1114   // first iteration: searching for the whole shape appearance (like face of the box)
1115   // second iteration: searching for sub-shapes that contain the sub (like vertex on faces)
1116   int aToFindPart = 0;
1117   TopTools_DataMapOfShapeShape aNewToOld; // map from new containers to old containers (with val)
1118   TopTools_MapOfShape anOlds; // to know how many olds produced new containers
1119   for(; aToFindPart != 2 && theShapes.IsEmpty(); aToFindPart++) {
1120     std::list<ResultPtr>::iterator aNewContIter = aNewToIterate.begin();
1121     for(; aNewContIter != aNewToIterate.end(); aNewContIter++) {
1122       std::shared_ptr<Model_Data> aNewData =
1123         std::dynamic_pointer_cast<Model_Data>((*aNewContIter)->data());
1124       TDF_Label aNewLab = aNewData->shapeLab();
1125       // searching for produced sub-shape fully on some label
1126       TDF_ChildIDIterator aNSIter(aNewLab, TNaming_NamedShape::GetID());
1127       for(; aNSIter.More(); aNSIter.Next()) {
1128         Handle(TNaming_NamedShape) aNS = Handle(TNaming_NamedShape)::DownCast(aNSIter.Value());
1129         for(TNaming_Iterator aPairIter(aNS); aPairIter.More(); aPairIter.Next()) {
1130           if (aToFindPart == 0) { // search shape is fully inside
1131             if (aPairIter.OldShape().IsSame(theValShape)) {
1132               if (aPairIter.NewShape().IsNull()) {// value was removed
1133                 theShapes.Clear();
1134                 return;
1135               }
1136               // don't add edges generated from faces
1137               if (aPairIter.NewShape().ShapeType() <= aValType)
1138                 theShapes.Append(aPairIter.NewShape());
1139             }
1140           } else if (!aPairIter.OldShape().IsNull()) { // search shape that contains this sub
1141             TopExp_Explorer anExp(aPairIter.OldShape(), aValType);
1142             for(; anExp.More(); anExp.Next()) {
1143               if (anExp.Current().IsSame(theValShape)) { // found a new container
1144                 if (aPairIter.NewShape().IsNull()) // skip removed high-level shape
1145                   continue;
1146                 aNewToOld.Bind(aPairIter.NewShape(), aPairIter.OldShape());
1147                 anOlds.Add(aPairIter.OldShape());
1148                 break;
1149               }
1150             }
1151           }
1152         }
1153       }
1154     }
1155   }
1156   if (aToFindPart == 2 && !aNewToOld.IsEmpty()) {
1157     // also iterate the whole old shape to find not-modified shapes that contain this old
1158     TopoDS_Shape anOldContShape = theOldContext->shape()->impl<TopoDS_Shape>();
1159     NCollection_Map<TopAbs_ShapeEnum> aNewTypes; // types of shapes to iterate
1160     TopTools_DataMapOfShapeShape::Iterator aNewTypeIter(aNewToOld);
1161     for(; aNewTypeIter.More(); aNewTypeIter.Next()) {
1162       if (aNewTypeIter.Key().ShapeType() != aValType)
1163         aNewTypes.Add(aNewTypeIter.Key().ShapeType());
1164     }
1165     NCollection_Map<TopAbs_ShapeEnum>::Iterator aTypeIter(aNewTypes);
1166     for(; aTypeIter.More(); aTypeIter.Next()) {
1167       for(TopExp_Explorer anExp(anOldContShape, aTypeIter.Value()); anExp.More(); anExp.Next()) {
1168         TopoDS_Shape anOld = anExp.Current();
1169         if (aNewToOld.IsBound(anOld) || anOlds.Contains(anOld)) // this was modified
1170           continue;
1171         TopExp_Explorer aValExp(anOld, aValType);
1172         for(; aValExp.More(); aValExp.Next()) {
1173           const TopoDS_Shape& anUnchanged = aValExp.Current();
1174           if (anUnchanged.IsSame(theValShape)) {
1175             aNewToOld.Bind(anOld, anOld);
1176             anOlds.Add(anOld);
1177             break;
1178           }
1179         }
1180       }
1181     }
1182
1183     // map of sub-shapes -> number of occurrences of these shapes in containers
1184     NCollection_DataMap<TopoDS_Shape, TopTools_MapOfShape, TopTools_ShapeMapHasher> aSubs;
1185     TopTools_DataMapOfShapeShape::Iterator aContIter(aNewToOld);
1186     for(; aContIter.More(); aContIter.Next()) {
1187       TopExp_Explorer aSubExp(aContIter.Key(), aValType);
1188       for(; aSubExp.More(); aSubExp.Next()) {
1189         if (!aSubs.IsBound(aSubExp.Current())) {
1190           aSubs.Bind(aSubExp.Current(), TopTools_MapOfShape());
1191         }
1192         // store old to know how many olds produced this shape
1193         aSubs.ChangeFind(aSubExp.Current()).Add(aContIter.Value());
1194       }
1195     }
1196     // if sub is appeared same times in containers as the number of old shapes that contain it
1197     int aCountInOld = anOlds.Size();
1198     NCollection_DataMap<TopoDS_Shape, TopTools_MapOfShape, TopTools_ShapeMapHasher>::Iterator
1199       aSubsIter(aSubs);
1200     for(; aSubsIter.More(); aSubsIter.Next()) {
1201       if (aSubsIter.Value().Size() == aCountInOld) {
1202         TopoDS_Shape anOld = aSubsIter.Key();
1203         // check this exists in the new shape
1204         TopExp_Explorer aNew(aNewContShape, anOld.ShapeType());
1205         for (; aNew.More(); aNew.Next()) {
1206           if (aNew.Current().IsSame(anOld))
1207             break;
1208         }
1209         if (aNew.More())
1210           theShapes.Append(anOld);
1211       }
1212     }
1213   }
1214   if (theShapes.IsEmpty()) { // nothing was changed
1215     if (aWasWholeContext)
1216       theShapes.Append(TopoDS_Shape());
1217     else { // if theValShape exists in new context, add it without changes, otherwise - nothing
1218       for (TopExp_Explorer aNew(aNewContShape, aValType); aNew.More(); aNew.Next()){
1219         if (aNew.Current().IsSame(theValShape)) {
1220           theShapes.Append(theValShape);
1221           break;
1222         }
1223       }
1224     }
1225   } else if (theShapes.Size() > 1) {
1226     // check it is possible to remove extra sub-shapes:
1227     // keep only shapes with the same number of containers if possible
1228     TopAbs_ShapeEnum anAncType = TopAbs_FACE;
1229     if (aValType == TopAbs_VERTEX)
1230       anAncType = TopAbs_EDGE;
1231     TopoDS_Shape anOldContext = theOldContext->shape()->impl<TopoDS_Shape>();
1232     TopTools_IndexedDataMapOfShapeListOfShape anOldMap;
1233     TopExp::MapShapesAndUniqueAncestors(anOldContext, aValType,  anAncType, anOldMap);
1234     if (anOldMap.Contains(theValShape)) {
1235       int aNumInOld = anOldMap.FindFromKey(theValShape).Extent();
1236       TopTools_IndexedDataMapOfShapeListOfShape aNewMap;
1237       TopExp::MapShapesAndUniqueAncestors(aNewContShape, aValType,  anAncType, aNewMap);
1238       TopTools_ListOfShape aNewResults;
1239       for(TopTools_ListOfShape::Iterator aNewSubs(theShapes); aNewSubs.More(); aNewSubs.Next()) {
1240         TopoDS_Shape aCand = aNewSubs.Value();
1241         if (aNewMap.Contains(aCand) && aNewMap.FindFromKey(aCand).Extent() == aNumInOld)
1242           aNewResults.Append(aCand);
1243       }
1244       if (!aNewResults.IsEmpty() && aNewResults.Size() < theShapes.Size())
1245         theShapes = aNewResults;
1246     }
1247   }
1248 }
1249
1250 bool Model_AttributeSelection::searchNewContext(std::shared_ptr<Model_Document> theDoc,
1251   const TopoDS_Shape theContShape, ResultPtr theContext, TopoDS_Shape theValShape,
1252   TDF_Label theAccessLabel,
1253   std::list<ResultPtr>& theResults, TopTools_ListOfShape& theValShapes)
1254 {
1255   std::set<ResultPtr> aResults; // to avoid duplicates, new context, null if deleted
1256   // iterate context and shape, but also if it is sub-shape of main shape, check also it
1257   TopTools_ListOfShape aContextList;
1258   aContextList.Append(theContShape);
1259   if (theContext.get()) {
1260     ResultPtr aComposite = ModelAPI_Tools::bodyOwner(theContext);
1261     if (aComposite.get() && aComposite->shape().get() && !aComposite->shape()->isNull())
1262       aContextList.Append(aComposite->shape()->impl<TopoDS_Shape>());
1263   }
1264   for(TopTools_ListOfShape::Iterator aContIter(aContextList); aContIter.More(); aContIter.Next()) {
1265     TNaming_SameShapeIterator aModifIter(aContIter.ChangeValue(), theAccessLabel);
1266     for(; aModifIter.More(); aModifIter.Next()) {
1267       TDF_Label anObjLab = aModifIter.Label().Father();
1268       ResultPtr aModifierObj = std::dynamic_pointer_cast<ModelAPI_Result>
1269         (theDoc->objects()->object(anObjLab));
1270       if (!aModifierObj.get()) {
1271         // #2241: shape may be sub-element of new object, not main (shell created from faces)
1272         if (!anObjLab.IsRoot())
1273           aModifierObj = std::dynamic_pointer_cast<ModelAPI_Result>
1274           (theDoc->objects()->object(anObjLab.Father()));
1275         if (!aModifierObj.get())
1276           continue;
1277       }
1278       FeaturePtr aModifierFeat = theDoc->feature(aModifierObj);
1279       if (!aModifierFeat.get())
1280         continue;
1281       FeaturePtr aThisFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(owner());
1282       if (aModifierFeat == aThisFeature || !theDoc->isLaterByDep(aThisFeature, aModifierFeat))
1283         continue; // the modifier feature is later than this, so, should not be used
1284       FeaturePtr aCurrentModifierFeat = theDoc->feature(theContext);
1285       if (aCurrentModifierFeat == aModifierFeat ||
1286         !theDoc->isLaterByDep(aModifierFeat, aCurrentModifierFeat))
1287         continue; // the current modifier is later than the found, so, useless
1288       Handle(TNaming_NamedShape) aNewNS;
1289       aModifIter.Label().FindAttribute(TNaming_NamedShape::GetID(), aNewNS);
1290       if (aNewNS->Evolution() == TNaming_MODIFY || aNewNS->Evolution() == TNaming_GENERATED) {
1291         aResults.insert(aModifierObj);
1292       } else if (aNewNS->Evolution() == TNaming_DELETE) { // a shape was deleted => result is empty
1293         aResults.insert(ResultPtr());
1294       } else { // not-processed modification => don't support it
1295         continue;
1296       }
1297     }
1298   }
1299   // if there exist context composite and sub-result(s), leave only sub(s)
1300   for(std::set<ResultPtr>::iterator aResIter = aResults.begin(); aResIter != aResults.end();) {
1301     ResultPtr aParent = ModelAPI_Tools::bodyOwner(*aResIter);
1302     for(; aParent.get(); aParent = ModelAPI_Tools::bodyOwner(aParent))
1303       if (aResults.count(aParent))
1304         break;
1305     if (aParent.get()) { // erase from set, so, restart iteration
1306       aResults.erase(aParent);
1307       aResIter = aResults.begin();
1308     } else aResIter++;
1309   }
1310
1311   if (aResults.empty()) {
1312     // check the context become concealed by operation which is earlier than this selection
1313     std::list<ResultPtr> allRes;
1314     ResultPtr aRoot = ModelAPI_Tools::bodyOwner(theContext, true);
1315     if (!aRoot.get())
1316       aRoot = theContext;
1317     ResultBodyPtr aRootBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aRoot);
1318     if (aRootBody.get()) {
1319       ModelAPI_Tools::allSubs(aRootBody, allRes);
1320       allRes.push_back(aRootBody);
1321     } else
1322       allRes.push_back(aRoot);
1323
1324     FeaturePtr aThisFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(owner());
1325     for (std::list<ResultPtr>::iterator aSub = allRes.begin(); aSub != allRes.end(); aSub++) {
1326       ResultPtr aResCont = *aSub;
1327       const std::set<AttributePtr>& aRefs = aResCont->data()->refsToMe();
1328       std::set<AttributePtr>::const_iterator aRef = aRefs.begin();
1329       for (; aRef != aRefs.end(); aRef++) {
1330         if (!aRef->get() || !(*aRef)->owner().get())
1331           continue;
1332         // concealed attribute only
1333         FeaturePtr aRefFeat = std::dynamic_pointer_cast<ModelAPI_Feature>((*aRef)->owner());
1334         if (aRefFeat == aThisFeature)
1335           continue;
1336         if (!ModelAPI_Session::get()->validators()->isConcealed(
1337           aRefFeat->getKind(), (*aRef)->id()))
1338           continue;
1339         if (theDoc->isLaterByDep(aThisFeature, aRefFeat)) {
1340           // for extrusion cut in python script the nested sketch reference may be concealed before
1341           // it is nested, so, check this composite feature is valid
1342           static ModelAPI_ValidatorsFactory* aFactory = ModelAPI_Session::get()->validators();
1343           // need to be validated to update the "Apply" state if not previewed
1344           if (aFactory->validate(aRefFeat))
1345             return true; // feature conceals result, return true, so the context will be removed
1346         }
1347       }
1348     }
1349     return false; // no modifications found, must stay the same
1350   }
1351   // iterate all results to find further modifications
1352   std::set<ResultPtr>::iterator aResIter = aResults.begin();
1353   for(; aResIter != aResults.end(); aResIter++) {
1354     if (aResIter->get() != NULL) {
1355       // compute new values by two contexts: the old and the new
1356       TopTools_ListOfShape aValShapes;
1357       computeValues(theContext, *aResIter, theValShape, aValShapes);
1358
1359       TopTools_ListIteratorOfListOfShape aNewVal(aValShapes);
1360       for(; aNewVal.More(); aNewVal.Next()) {
1361         std::list<ResultPtr> aNewRes;
1362         TopTools_ListOfShape aNewUpdatedVal;
1363         TopoDS_Shape aNewValSh = aNewVal.Value();
1364         TopoDS_Shape aNewContShape = (*aResIter)->shape()->impl<TopoDS_Shape>();
1365         if (theValShape.IsNull() && aNewContShape.IsSame(aNewValSh))
1366           aNewValSh.Nullify();
1367         if (searchNewContext(theDoc, aNewContShape, *aResIter, aNewValSh,
1368                              theAccessLabel, aNewRes, aNewUpdatedVal))
1369         {
1370           // append new results instead of the current ones
1371           std::list<ResultPtr>::iterator aNewIter = aNewRes.begin();
1372           TopTools_ListIteratorOfListOfShape aNewUpdVal(aNewUpdatedVal);
1373           for(; aNewIter != aNewRes.end(); aNewIter++, aNewUpdVal.Next()) {
1374             theResults.push_back(*aNewIter);
1375             theValShapes.Append(aNewUpdVal.Value());
1376           }
1377         } else { // the current result is good
1378           theResults.push_back(*aResIter);
1379           theValShapes.Append(aNewValSh);
1380         }
1381       }
1382     }
1383   }
1384   return true; // theResults must be empty: everything is deleted
1385 }
1386
1387 void Model_AttributeSelection::updateInHistory(bool& theRemove)
1388 {
1389   ResultPtr aContext = std::dynamic_pointer_cast<ModelAPI_Result>(myRef.value());
1390   // only bodies and parts may be modified later in the history, don't do anything otherwise
1391   if (!aContext.get() || (aContext->groupName() != ModelAPI_ResultBody::group() &&
1392       aContext->groupName() != ModelAPI_ResultPart::group()))
1393     return;
1394   std::shared_ptr<Model_Document> aDoc =
1395     std::dynamic_pointer_cast<Model_Document>(aContext->document());
1396   std::shared_ptr<Model_Data> aContData = std::dynamic_pointer_cast<Model_Data>(aContext->data());
1397   if (!aContData.get() || !aContData->isValid())
1398     return;
1399   TDF_Label aContLab = aContData->shapeLab(); // named shape where the selected context is located
1400   Handle(TNaming_NamedShape) aContNS;
1401   if (!aContLab.FindAttribute(TNaming_NamedShape::GetID(), aContNS)) {
1402     bool aFoundNewContext = true;
1403     ResultPtr aNewContext = aContext;
1404     while(aFoundNewContext) {
1405       aFoundNewContext = false;
1406       // parts have no shape in result, so, trace references using the Part info
1407       if (aNewContext->groupName() == ModelAPI_ResultPart::group()) {
1408         ResultPartPtr aPartContext = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aNewContext);
1409         if (aPartContext.get()) { // searching for the up to date references to the referenced cont
1410           const std::set<AttributePtr>& aRefs = aPartContext->data()->refsToMe();
1411           std::set<AttributePtr>::const_iterator aRef = aRefs.begin();
1412           for(; aRef != aRefs.end(); aRef++) {
1413             // to avoid detection of part changes by local selection only
1414             AttributeSelectionPtr aSel =
1415               std::dynamic_pointer_cast<ModelAPI_AttributeSelection>(*aRef);
1416             if (aSel.get() && aSel->value().get() &&
1417                 !aSel->value()->isSame(aSel->context()->shape()))
1418               continue;
1419
1420             FeaturePtr aRefFeat = std::dynamic_pointer_cast<ModelAPI_Feature>((*aRef)->owner());
1421             if (aRefFeat.get() && aRefFeat != owner()) {
1422               FeaturePtr aThisFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(owner());
1423               if (!aDoc->isLaterByDep(aRefFeat, aThisFeature)) { // found better feature
1424                 aFoundNewContext = true;
1425                 aNewContext = aRefFeat->firstResult();
1426               }
1427             }
1428           }
1429         }
1430       }
1431     }
1432     if (aNewContext != aContext) {
1433       setValue(aNewContext, value());
1434     }
1435     return;
1436   }
1437   FeaturePtr aThisFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(owner());
1438   FeaturePtr aCurrentModifierFeat = aDoc->feature(aContext);
1439   // iterate the context shape modifications in order to find a feature that is upper in history
1440   // that this one and is really modifies the referenced result to refer to it
1441   ResultPtr aModifierResFound;
1442   TNaming_Iterator aPairIter(aContNS);
1443   if (!aPairIter.More())
1444     return;
1445   TopoDS_Shape aNewCShape = aPairIter.NewShape();
1446   bool anIterate = true;
1447   // trying to update also the sub-shape selected
1448   GeomShapePtr aSubShape = value();
1449   if (aSubShape.get() && aSubShape->isEqual(aContext->shape()))
1450     aSubShape.reset();
1451   TopoDS_Shape aValShape;
1452   if (aSubShape.get()) {
1453     aValShape = aSubShape->impl<TopoDS_Shape>();
1454   }
1455
1456   std::list<ResultPtr> aNewContexts;
1457   TopTools_ListOfShape aValShapes;
1458   if (searchNewContext(aDoc, aNewCShape, aContext, aValShape, aContLab, aNewContexts, aValShapes))
1459   {
1460     std::set<ResultPtr> allContexts, aSkippedContext;
1461     std::list<ResultPtr>::iterator aNewContext = aNewContexts.begin();
1462     for(; aNewContext != aNewContexts.end(); aNewContext++)
1463       allContexts.insert(*aNewContext);
1464
1465     // if there exist context composite and sub-result(s), leave only sub(s)
1466     std::set<ResultPtr>::iterator aResIter = allContexts.begin();
1467     for(; aResIter != allContexts.end(); aResIter++) {
1468       ResultPtr aParent = ModelAPI_Tools::bodyOwner(*aResIter);
1469       for(; aParent.get(); aParent = ModelAPI_Tools::bodyOwner(aParent))
1470         if (allContexts.count(aParent))
1471           aSkippedContext.insert(aParent);
1472     }
1473
1474     GeomAPI_Shape::ShapeType aListShapeType = GeomAPI_Shape::SHAPE;
1475     if (myParent) {
1476       if (myParent->selectionType() == "VERTEX") aListShapeType = GeomAPI_Shape::VERTEX;
1477       else if (myParent->selectionType() == "EDGE") aListShapeType = GeomAPI_Shape::EDGE;
1478       else if (myParent->selectionType() == "FACE") aListShapeType = GeomAPI_Shape::FACE;
1479     }
1480
1481     std::list<ResultPtr>::iterator aNewCont = aNewContexts.begin();
1482     TopTools_ListIteratorOfListOfShape aNewValues(aValShapes);
1483     bool aFirst = true; // first is set to this, next are appended to parent
1484     for(; aNewCont != aNewContexts.end(); aNewCont++, aNewValues.Next()) {
1485       if (aSkippedContext.count(*aNewCont))
1486         continue;
1487
1488       GeomShapePtr aValueShape;
1489       if (!aNewValues.Value().IsNull()) {
1490         aValueShape = std::make_shared<GeomAPI_Shape>();
1491         aValueShape->setImpl<TopoDS_Shape>(new TopoDS_Shape(aNewValues.Value()));
1492       }
1493       // Check that list has the same type of shape selection before adding.
1494       GeomAPI_Shape::ShapeType aShapeShapeType = GeomAPI_Shape::SHAPE;
1495       if (aValueShape.get()) {
1496         aShapeShapeType = aValueShape->shapeType();
1497       } else {
1498         aShapeShapeType = (*aNewCont)->shape()->shapeType();
1499       }
1500       if (aListShapeType != GeomAPI_Shape::SHAPE && aListShapeType != aShapeShapeType) {
1501         continue;
1502       }
1503
1504       if (aFirst) {
1505         setValue(*aNewCont, aValueShape);
1506         aFirst = false;
1507       } else if (myParent) {
1508         if (!myParent->isInList(*aNewCont, aValueShape)) // avoid addition of duplicates
1509           myParent->append(*aNewCont, aValueShape);
1510       }
1511     }
1512     if (aFirst) { // nothing was added, all results were deleted
1513       if (myParent) {
1514         theRemove = true;
1515       } else {
1516         ResultPtr anEmptyContext;
1517         std::shared_ptr<GeomAPI_Shape> anEmptyShape;
1518         setValue(anEmptyContext, anEmptyShape); // nullify the selection
1519         return;
1520       }
1521     }
1522   }
1523 }
1524
1525 void Model_AttributeSelection::setParent(Model_AttributeSelectionList* theParent)
1526 {
1527   myParent = theParent;
1528 }
1529
1530 std::string Model_AttributeSelection::contextName(const TDF_Label theSelectionLab)
1531 {
1532   std::shared_ptr<Model_Document> aDoc = myRestoreDocument.get() ? myRestoreDocument :
1533     std::dynamic_pointer_cast<Model_Document>(owner()->document());
1534   FeaturePtr aFeatureOwner = aDoc->featureByLab(theSelectionLab);
1535   bool aBaseDocumnetUsed = false;
1536   if (!aFeatureOwner.get()) { // use module document
1537     aDoc = std::dynamic_pointer_cast<Model_Document>(ModelAPI_Session::get()->moduleDocument());
1538     aFeatureOwner = aDoc->featureByLab(theSelectionLab);
1539     aBaseDocumnetUsed = true;
1540   }
1541   if (aFeatureOwner.get()) {
1542     // if it is sub-element of the sketch, the context name is the name of the sketch
1543     // searching also for result - real context
1544     ResultPtr aResult;
1545     FeaturePtr aComposite = ModelAPI_Tools::compositeOwner(aFeatureOwner);
1546     if (aComposite.get() && aComposite->results().size() == 1 &&
1547         aComposite->firstResult()->groupName() == ModelAPI_ResultConstruction::group()) {
1548       aFeatureOwner = aComposite;
1549       aResult = aFeatureOwner->firstResult();
1550     } else {
1551       aResult = aDoc->resultByLab(theSelectionLab);
1552     }
1553     if (aResult.get()) {
1554       // this is to avoid duplicated names of results problem
1555       std::string aContextName = aResult->data()->name();
1556       // myLab corresponds to the current time
1557       TDF_Label aCurrentLab = selectionLabel();
1558       while(aCurrentLab.Depth() > 3)
1559         aCurrentLab = aCurrentLab.Father();
1560
1561       int aNumInHistoryNames =
1562         aDoc->numberOfNameInHistory(aResult, aCurrentLab);
1563       while(aNumInHistoryNames > 1) { // add "_" before name the needed number of times
1564         aContextName = "_" + aContextName;
1565         aNumInHistoryNames--;
1566       }
1567       if (aBaseDocumnetUsed)
1568         aContextName = aDoc->kind() + "/" + aContextName;
1569       return aContextName;
1570     }
1571   }
1572   return ""; // invalid case
1573 }
1574
1575 /// This method restores by the context and value name the context label and
1576 /// sub-label where the value is. Returns true if it is valid.
1577 bool Model_AttributeSelection::restoreContext(std::string theName,
1578   TDF_Label& theContext, TDF_Label& theValue)
1579 {
1580   static const GeomShapePtr anEmptyShape; // to store context only
1581   std::string aName = theName;
1582   std::shared_ptr<Model_Document> aDoc = myRestoreDocument.get() ? myRestoreDocument :
1583     std::dynamic_pointer_cast<Model_Document>(owner()->document());
1584
1585   // remove the sub-value part if exists
1586   std::string aSubShapeName = aName;
1587   std::string::size_type n = aName.find('/');
1588   if (n != std::string::npos) {
1589     aName = aName.substr(0, n);
1590   }
1591
1592   if (aName.empty()) return false;
1593   bool anUniqueContext = false;
1594   ResultPtr aCont = aDoc->findByName(aName, aSubShapeName, anUniqueContext);
1595   if (!aCont.get() || !aCont->shape().get() || aCont->shape()->isNull()) {
1596     // name in PartSet?
1597     aDoc = std::dynamic_pointer_cast<Model_Document>(
1598       ModelAPI_Session::get()->moduleDocument());
1599     if (theName.find(aDoc->kind()) == 0) { // remove the document identifier from name if exists
1600       aSubShapeName = theName.substr(aDoc->kind().size() + 1);
1601       aName = aSubShapeName;
1602       std::string::size_type n = aName.find('/');
1603       if (n != std::string::npos) {
1604         aName = aName.substr(0, n);
1605       }
1606     }
1607     aCont = aDoc->findByName(aName, aSubShapeName, anUniqueContext);
1608     if (!aCont.get() || !aCont->shape().get() || aCont->shape()->isNull())
1609       return false;
1610   }
1611
1612   // searching the sub-shape
1613   static const ResultPtr anEmpty;
1614   theValue = aDoc->findNamingName(aSubShapeName, anUniqueContext ? aCont : anEmpty);
1615
1616   // sketch sub-component shape and name is located in separated feature label, try the sub-name
1617   if (theValue.IsNull() && aCont->groupName() == ModelAPI_ResultConstruction::group()) {
1618     std::string::size_type aSlash = aSubShapeName.rfind('/');
1619     if (aSlash != std::string::npos) {
1620       std::string aCompName = aSubShapeName.substr(aSlash + 1);
1621       CompositeFeaturePtr aComposite =
1622         std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aDoc->feature(aCont));
1623       if (aComposite.get() && aComposite->numberOfSubs()) {
1624         const int aSubNum = aComposite->numberOfSubs();
1625         for (int a = 0; a < aSubNum && theValue.IsNull(); a++) {
1626           FeaturePtr aSub = aComposite->subFeature(a);
1627           const std::list<std::shared_ptr<ModelAPI_Result> >& aResults = aSub->results();
1628           std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aRes = aResults.cbegin();
1629           for (; aRes != aResults.cend() && theValue.IsNull(); aRes++) {
1630             if ((*aRes)->data()->name() == aCompName) {
1631               theValue = std::dynamic_pointer_cast<Model_Data>((*aRes)->data())->shapeLab();
1632               break;
1633             } else { // any sub-label because the sketch line may be renamed, but not sub-vertices
1634               TDF_Label aLab = std::dynamic_pointer_cast<Model_Data>((*aRes)->data())->shapeLab();
1635               TDF_ChildIDIterator aSubNames(aLab, TDataStd_Name::GetID());
1636               for(; aSubNames.More(); aSubNames.Next()) {
1637                 if (Handle(TDataStd_Name)::DownCast(aSubNames.Value())->Get().
1638                   IsEqual(aCompName.c_str())) {
1639                   theValue = aSubNames.Value()->Label();
1640                   break;
1641                 }
1642               }
1643             }
1644           }
1645         }
1646       }
1647     }
1648   }
1649
1650   if (aCont.get()) {
1651     theContext = std::dynamic_pointer_cast<Model_Data>(aCont->data())->label();
1652   }
1653   return true;
1654 }
1655
1656 bool Model_AttributeSelection::isLater(
1657   const TDF_Label theResult1, const TDF_Label theResult2) const
1658 {
1659   std::shared_ptr<Model_Document> aDoc = myRestoreDocument.get() ? myRestoreDocument :
1660     std::dynamic_pointer_cast<Model_Document>(owner()->document());
1661   FeaturePtr aFeat1 = aDoc->featureByLab(theResult1);
1662   if (!aFeat1.get())
1663     return false;
1664   FeaturePtr aFeat2 = aDoc->featureByLab(theResult2);
1665   if (!aFeat2.get())
1666     return false;
1667   return aDoc->isLaterByDep(aFeat1, aFeat2);
1668 }
1669
1670 ResultPtr Model_AttributeSelection::newestContext(
1671   const ResultPtr theCurrent, const GeomShapePtr theValue)
1672 {
1673   ResultPtr aResult = theCurrent;
1674   GeomShapePtr aSelectedShape = theValue.get() ? theValue : theCurrent->shape();
1675   std::shared_ptr<Model_Document> aDoc =
1676     std::dynamic_pointer_cast<Model_Document>(owner()->document());
1677   bool aFindNewContext = true;
1678   while (aFindNewContext && aResult.get()) {
1679     aFindNewContext = false;
1680     // try to find the last context to find the up to date shape
1681     TopoDS_Shape aConShape = aResult->shape()->impl<TopoDS_Shape>();
1682     if (TNaming_Tool::HasLabel(selectionLabel(), aConShape)) {
1683       Handle(TNaming_NamedShape) aNS = TNaming_Tool::NamedShape(aConShape, selectionLabel());
1684       if (!aNS.IsNull()) {
1685         aNS = TNaming_Tool::CurrentNamedShape(aNS);
1686         if (!aNS.IsNull()
1687             && isLater(selectionLabel(), aNS->Label())
1688             && isLater(aNS->Label(),
1689                        std::dynamic_pointer_cast<Model_Data>(aResult->data())->label()))
1690         {
1691           TDF_Label aLab = aNS->Label();
1692           ResultPtr aRes = aDoc->resultByLab(aLab);
1693           if (aRes.get()) {
1694             if (aRes->shape()->isSubShape(aSelectedShape)) {
1695               aResult = aRes;
1696               aFindNewContext = true;
1697               continue;
1698             }
1699           }
1700         }
1701       }
1702     }
1703
1704     // TestFillWireVertex.py - sketch constructions for wire may participate too
1705     //if (aResult->groupName() == ModelAPI_ResultBody::group()) {
1706       // try to search newer context by the concealment references
1707       // take references to all results: root one, any sub
1708     std::list<ResultPtr> allRes;
1709     ResultPtr aCompContext;
1710     ResultBodyPtr aCompBody = ModelAPI_Tools::bodyOwner(aResult, true);
1711     if (aCompBody.get()) {
1712       ModelAPI_Tools::allSubs(aCompBody, allRes);
1713       allRes.push_back(aCompBody);
1714       aCompContext = aCompBody;
1715     }
1716     if (allRes.empty())
1717       allRes.push_back(aResult);
1718
1719     for (std::list<ResultPtr>::iterator aSub = allRes.begin(); aSub != allRes.end(); aSub++) {
1720       ResultPtr aResCont = *aSub;
1721       ResultBodyPtr aResBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aResCont);
1722       if (aResBody.get() && aResBody->numberOfSubs() > 0 && aResBody != aCompContext)
1723         continue; // only lower and higher level subs are counted
1724       const std::set<AttributePtr>& aRefs = aResCont->data()->refsToMe();
1725       std::set<AttributePtr>::const_iterator aRef = aRefs.begin();
1726       for (; !aFindNewContext && aRef != aRefs.end(); aRef++) {
1727         if (!aRef->get() || !(*aRef)->owner().get())
1728           continue;
1729         // concealed attribute only
1730         FeaturePtr aRefFeat = std::dynamic_pointer_cast<ModelAPI_Feature>((*aRef)->owner());
1731         if (!ModelAPI_Session::get()->validators()->isConcealed(
1732           aRefFeat->getKind(), (*aRef)->id()))
1733           continue;
1734         // search the feature result that contains sub-shape selected
1735         std::list<std::shared_ptr<ModelAPI_Result> > aResults;
1736
1737         // take all sub-results or one result
1738         std::list<ResultPtr> aRefFeatResults;
1739         ModelAPI_Tools::allResults(aRefFeat, aRefFeatResults);
1740         std::list<ResultPtr>::iterator aRefResIter = aRefFeatResults.begin();
1741         for (; aRefResIter != aRefFeatResults.end(); aRefResIter++) {
1742           ResultBodyPtr aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aRefResIter);
1743           if (aBody.get() && aBody->numberOfSubs() == 0) // add only lower level subs
1744             aResults.push_back(aBody);
1745         }
1746         std::list<std::shared_ptr<ModelAPI_Result> >::iterator aResIter = aResults.begin();
1747
1748         // searching by sub-shape
1749         for (; aResIter != aResults.end(); aResIter++) {
1750           if (!aResIter->get() || !(*aResIter)->data()->isValid() || (*aResIter)->isDisabled())
1751             continue;
1752           GeomShapePtr aShape = (*aResIter)->shape();
1753           if (aShape.get() && aShape->isSubShape(aSelectedShape, false)) {
1754             aResult = *aResIter; // found new context (produced from this) with same subshape
1755             aFindNewContext = true; // continue searching further
1756             break;
1757           }
1758         }
1759       }
1760     }
1761   }
1762   // if compsolid is context, try to take sub-solid as context: like in GUI and scripts
1763   ResultBodyPtr aComp = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aResult);
1764   if (aComp && aComp->numberOfSubs()) {
1765     std::list<ResultPtr> allSubs;
1766     ModelAPI_Tools::allSubs(aComp, allSubs);
1767     std::list<ResultPtr>::iterator aS = allSubs.begin();
1768     for (; aS != allSubs.end(); aS++) {
1769       ResultBodyPtr aSub = std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aS);
1770       if (aSub && aSub->numberOfSubs() == 0 && aSub->shape().get() &&
1771           aSub->shape()->isSubShape(aSelectedShape)) {
1772         aResult = aSub;
1773         break;
1774       }
1775     }
1776   }
1777   // in case sketch line was selected for wire, but wire was concealed and not such line anymore,
1778   // so, actually, the sketch element was selected (which is never concealed)
1779   if (aResult != theCurrent && aResult->isConcealed())
1780     aResult = theCurrent;
1781   return aResult;
1782 }
1783
1784 void Model_AttributeSelection::combineGeometrical()
1785 {
1786   if (myTmpContext.get() || myTmpSubShape.get())
1787     return;
1788   TDF_Label aSelLab = selectionLabel();
1789   if (aSelLab.IsAttribute(kINVALID_SELECTION) || !myRef.isInitialized())
1790     return;
1791
1792   if (aSelLab.IsAttribute(kCIRCLE_CENTER) || aSelLab.IsAttribute(kELLIPSE_CENTER1) ||
1793       aSelLab.IsAttribute(kELLIPSE_CENTER2) || aSelLab.IsAttribute(kSIMPLE_REF_ID))
1794     return;
1795
1796   if (aSelLab.IsAttribute(kPART_REF_ID)) {
1797     ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(context());
1798     if (!aPart.get() || !aPart->isActivated())
1799       return; // postponed naming needed
1800     Handle(TDataStd_Integer) anIndex;
1801     if (aSelLab.FindAttribute(TDataStd_Integer::GetID(), anIndex)) {
1802       if (anIndex->Get()) { // special selection attribute was created, use it
1803         std::string aNewName;
1804         aPart->combineGeometrical(anIndex->Get(), aNewName);
1805         TDataStd_Name::Set(aSelLab, aNewName.c_str());
1806       }
1807     }
1808     return;
1809   }
1810
1811   std::shared_ptr<Model_ResultConstruction> aConstr =
1812     std::dynamic_pointer_cast<Model_ResultConstruction>(context());
1813   if (aConstr.get())
1814     return;
1815   FeaturePtr aFeature = contextFeature();
1816   if (aFeature.get())
1817     return;
1818
1819   Selector_Selector aSelector(aSelLab, baseDocumentLab());
1820   TopoDS_Shape aContextShape = context()->shape()->impl<TopoDS_Shape>();
1821   if (aSelector.restore(aContextShape)) {
1822     aSelector.combineGeometrical(aContextShape);
1823   }
1824 }
1825
1826 TDF_Label Model_AttributeSelection::baseDocumentLab()
1827 {
1828   if (ModelAPI_Session::get()->moduleDocument() != owner()->document())
1829     return std::dynamic_pointer_cast<Model_Document>
1830       (ModelAPI_Session::get()->moduleDocument())->extConstructionsLabel();
1831   static TDF_Label anEmpty;
1832   return anEmpty;
1833 }