Fix and unit test for the issue #1064

[modules/shaper.git] / src / Model / Model_ResultPart.h

diff --git a/src/Model/Model_ResultPart.h b/src/Model/Model_ResultPart.h
index f53ac5e852c6e48444789f0b1812030357206203..220e1b408f50e5a6eb7cac65884bb98a818d7b67 100644 (file)
--- a/src/Model/Model_ResultPart.h
+++ b/src/Model/Model_ResultPart.h
@@ -61,7 +61,8 @@ class Model_ResultPart : public ModelAPI_ResultPart
   /// Updates the selection inside of the part by the selection index
   MODEL_EXPORT virtual bool updateInPart(const int theIndex);
   /// Returns the shape by the name in the part
-  MODEL_EXPORT virtual std::shared_ptr<GeomAPI_Shape> shapeInPart(const std::string& theName);
+  MODEL_EXPORT virtual std::shared_ptr<GeomAPI_Shape> shapeInPart(
+    const std::string& theName, const std::string& theType, int& theIndex);
   /// Updates the shape-result of the part (called on Part feature execution)
   MODEL_EXPORT virtual void updateShape();
   /// Applies the additional transformation of the part
@@ -72,6 +73,9 @@ class Model_ResultPart : public ModelAPI_ResultPart
   MODEL_EXPORT virtual void colorConfigInfo(std::string& theSection, std::string& theName,
                                             std::string& theDefault);
 
+  /// Returns the shape selected in the selection index
+  MODEL_EXPORT virtual std::shared_ptr<GeomAPI_Shape> selectionValue(const int theIndex);
+
 protected:
   /// makes a result on a temporary feature (an action)
   Model_ResultPart();