Salome HOME
Improve Model package coverage: remove unused methods, open/save tests improvements
[modules/shaper.git] / src / Model / Model_AttributeSelection.h
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 #ifndef Model_AttributeSelection_H_
22 #define Model_AttributeSelection_H_
23
24 #include "Model.h"
25 #include "Model_AttributeReference.h"
26 #include <ModelAPI_AttributeSelection.h>
27 #include <Selector_NameGenerator.h>
28 #include <TDF_LabelMap.hxx>
29 #include <TopoDS_Shape.hxx>
30 #include <TopTools_ListOfShape.hxx>
31
32 class Model_AttributeSelectionList;
33 class Model_Document;
34
35 /**\class Model_AttributeSelection
36  * \ingroup DataModel
37  * \brief Attribute that contains reference to the sub-shape of some result, the selected shape.
38  */
39 class Model_AttributeSelection : public ModelAPI_AttributeSelection,
40                                  public Selector_NameGenerator
41 {
42   Model_AttributeReference myRef;  ///< The reference functionality re-usage
43   /// temporarily storages to avoid keeping in the data structure if not needed
44   ResultPtr myTmpContext;
45   /// temporarily storages to avoid keeping in the data structure if not needed
46   std::shared_ptr<GeomAPI_Shape> myTmpSubShape;
47   /// temporarily storages to avoid keeping in the data structure if not needed
48   CenterType myTmpCenterType;
49   /// Reference to the parent attribute, if any (to split selection compounds in issue 1799)
50   Model_AttributeSelectionList* myParent;
51
52   std::shared_ptr<Model_Document> myRestoreDocument; // current document to restore by name
53   /// If true attribute selects geometry instead of shape.
54   bool myIsGeometricalSelection;
55
56 public:
57   /// Defines the result and its selected sub-shape
58   /// \param theContext object where the sub-shape was selected
59   /// \param theSubShape selected sub-shape (if null, the whole context is selected)
60   /// \param theTemporarily if it is true, do not store and name the added in the data framework
61   ///           (used to remove immediately, without the following updates)
62   /// \returns true if attribute was updated
63   MODEL_EXPORT virtual bool setValue(
64     const ObjectPtr& theContext, const std::shared_ptr<GeomAPI_Shape>& theSubShape,
65     const bool theTemporarily = false);
66
67   /// Same as SetValue, but it takes an edge (on circular or elliptical curve)
68   /// and stores the vertex of the central point (for ellipse the first or the second focus point)
69   MODEL_EXPORT virtual void setValueCenter(
70     const ObjectPtr& theContext, const std::shared_ptr<GeomAPI_Edge>& theEdge,
71     const CenterType theCenterType,
72     const bool theTemporarily = false);
73
74   /// Makes this selection attribute selects the same as in theSource selection
75   MODEL_EXPORT virtual void selectValue(
76     const std::shared_ptr<ModelAPI_AttributeSelection>& theSource);
77
78   /// Reset temporary stored values
79   virtual void removeTemporaryValues();
80
81   /// Returns the selected subshape
82   MODEL_EXPORT virtual std::shared_ptr<GeomAPI_Shape> value();
83
84   /// Returns the context of the selection (the whole shape owner)
85   MODEL_EXPORT virtual ResultPtr context();
86
87   /// Returns the context of the selection if the whole feature was selected
88   MODEL_EXPORT virtual FeaturePtr contextFeature();
89   /// Returns the context of the selection : result or feature
90   MODEL_EXPORT virtual std::shared_ptr<ModelAPI_Object> contextObject();
91
92   /// Sets the feature object
93   MODEL_EXPORT virtual void setObject(const std::shared_ptr<ModelAPI_Object>& theObject);
94
95   /// Updates the selection due to the changes in the referenced objects
96   /// \returns false if update is failed
97   MODEL_EXPORT virtual bool update();
98
99   /// Returns a textual string of the selection
100   /// \param theDefaultValue a name, which is returned if the naming name can not be obtained
101   MODEL_EXPORT virtual std::string namingName(const std::string& theDefaultValue = "");
102
103   /// Selects (i.e. creates Naming data structure) of sub-shape specified by textual name
104   MODEL_EXPORT virtual void selectSubShape(const std::string& theType,
105                                            const std::string& theSubShapeName);
106
107   /// Selects sub-shape by its inner point
108   MODEL_EXPORT virtual void selectSubShape(const std::string& theType,
109                                            const std::shared_ptr<GeomAPI_Pnt>& thePoint);
110   /// Selects sub-shape by weak naming index
111   MODEL_EXPORT virtual void selectSubShape(const std::string& theType,
112     const std::string& theContextName, const int theIndex);
113
114   /// Returns true if attribute was  initialized by some value
115   MODEL_EXPORT virtual bool isInitialized();
116
117   /// Returns true if recompute of selection become impossible
118   MODEL_EXPORT virtual bool isInvalid();
119
120   /// Updates the arguments of selection if something was affected by creation
121   /// or reorder of features upper in the history line (issue #1757)
122   MODEL_EXPORT virtual void updateInHistory();
123
124   // Implementation of the name generator method from the Selector package
125   // This method returns the context name by the label of the sub-selected shape
126   MODEL_EXPORT virtual std::string contextName(const TDF_Label theSelectionLab) override;
127
128   /// This method restores by the context and value name the context label and
129   /// sub-label where the value is. Returns true if it is valid.
130   MODEL_EXPORT virtual bool restoreContext(std::string theName,
131     TDF_Label& theContext, TDF_Label& theValue) override;
132
133   /// Returns the label of the newest context presented by the current one
134   MODEL_EXPORT virtual TDF_Label newestContext(const TDF_Label theCurrentContext) override;
135
136   /// Returns true if the first result is newer than the second one in the tree of features
137   MODEL_EXPORT virtual bool isLater(const TDF_Label theResult1, const TDF_Label theResult2)
138     const override;
139
140   /// Returns the name by context. Adds the part name if the context is located in other document
141   MODEL_EXPORT virtual std::string contextName(const ResultPtr& theContext) const;
142
143   /// Makes the current local selection becomes all sub-shapes with same base geometry.
144   MODEL_EXPORT virtual void combineGeometrical();
145
146 protected:
147   /// Objects are created for features automatically
148   MODEL_EXPORT Model_AttributeSelection(TDF_Label& theLabel);
149
150   /// Returns the selected subshape, internal method that works without knowledge
151   /// about special selection of circle and ellipse focuses (for that the public value
152   /// method calls this and makes additional processing).
153   /// Returns theType type of the center, or NOT_CENTER if it is not.
154   std::shared_ptr<GeomAPI_Shape> internalValue(CenterType& theType);
155
156   /// Performs the selection for the body result (TNaming selection)
157   virtual void selectBody(
158     const ResultPtr& theContext, const std::shared_ptr<GeomAPI_Shape>& theSubShape);
159
160   /// Performs the selection for the part result (selection by name of body result inside of part)
161   /// \param theContext the result - owner of the selection
162   /// \param theSubShape selected shape
163   /// \param theUpdate flag that shows that it must be just update, theShape is null
164   /// \returns true if everything is selected correctly
165   virtual bool selectPart(
166     const ResultPtr& theContext, const std::shared_ptr<GeomAPI_Shape>& theSubShape,
167     const bool theUpdate = false);
168
169   /// Returns the label where TNaming_Selection results are stored
170   /// Note: there must be no attributes stored at the same label because Selector clears this lab
171   TDF_Label selectionLabel();
172
173   /// Sets the ID of the attribute in Data (called from Data): here it is used for myRef ID setting
174   MODEL_EXPORT virtual void setID(const std::string theID);
175
176   /// Sets the parent attribute
177   void setParent(Model_AttributeSelectionList* theParent);
178
179   /// Splits theNewShape into sub-shapes of theType type (for the list parent of this attribute)
180   void split(ResultPtr theContext, TopoDS_Shape theNewShape, TopAbs_ShapeEnum theType);
181
182   /// When group position is updated, searches the new context and new values
183   bool searchNewContext(std::shared_ptr<Model_Document> theDoc, const TopoDS_Shape theContShape,
184                         ResultPtr theContext, TopoDS_Shape theValShape, TDF_Label theAccessLabel,
185                         std::list<ResultPtr>& theResults, TopTools_ListOfShape& theValShapes);
186
187   /// Searches for the newest context, modification of the current, that contains theValue
188   ResultPtr newestContext(const ResultPtr theCurrent,
189     const std::shared_ptr<GeomAPI_Shape> theValue, const bool theAnyValue = false);
190
191   /// computes theShapes list - shapes that were generated/modified/deleted the theValShape
192   /// during creation from new to old context
193   void computeValues(ResultPtr theOldContext, ResultPtr theNewContext, TopoDS_Shape theValShape,
194     TopTools_ListOfShape& theShapes);
195
196   /// Returns true if is geometrical selection.
197   virtual bool isGeometricalSelection() const {
198     return myIsGeometricalSelection;
199   };
200
201   /// Returns the module document label if this selection attribute is not in this document.
202   /// Returns null label otherwise.
203   TDF_Label baseDocumentLab();
204
205   friend class Model_Data;
206   friend class Model_AttributeSelectionList;
207 };
208
209 #endif