Provide presentation for a step

[modules/shaper.git] / src / ModuleBase / ModuleBase_ResultPrs.h

diff --git a/src/ModuleBase/ModuleBase_ResultPrs.h b/src/ModuleBase/ModuleBase_ResultPrs.h
index f1ad5e1a0c2c588b59fe1d14e3543c1b486361a9..df01f95c0a241afba04c0194dc1e6f2e6058b3ee 100644 (file)
--- a/src/ModuleBase/ModuleBase_ResultPrs.h
+++ b/src/ModuleBase/ModuleBase_ResultPrs.h
@@ -23,6 +23,7 @@
 #include "ModuleBase.h"
 
 #include <ModelAPI_Result.h>
+#include <ModelAPI_ResultField.h>
 
 #include <BRep_Builder.hxx>
 #include <NCollection_List.hxx>
@@ -56,6 +57,10 @@ public:
   /// \param theResult a result object
   Standard_EXPORT ModuleBase_ResultPrs(ResultPtr theResult);
 
+  /// Constructor
+  /// \param theResult a result object
+  Standard_EXPORT ModuleBase_ResultPrs(FieldStepPtr theStep);
+
   //! Method which draws selected owners ( for fast presentation draw )
   Standard_EXPORT virtual void HilightSelected(const Handle(PrsMgr_PresentationManager3d)& thePM,
                                                const SelectMgr_SequenceOfOwner& theOwners);
@@ -147,10 +152,17 @@ private:
 
   void setEdgesDefaultColor();
 
+  void init();
+
+  GeomShapePtr getOriginalShape() const;
+
 private:
   /// Reference to result object
   ResultPtr myResult;
 
+  /// Reference to step object (initialised is only or step or result)
+  FieldStepPtr myStep;
+
   /// Original shape of the result object
   TopoDS_Shape myOriginalShape;