Issue #555: Infinite boolean state is stored in Result construction.

[modules/shaper.git] / src / PartSet / PartSet_FilterInfinite.h

diff --git a/src/PartSet/PartSet_FilterInfinite.h b/src/PartSet/PartSet_FilterInfinite.h
index 8fe53055b8f423b45b4423d4cb60fe2ffd5ea686..f6caa6666459f6196c8dbc6bd9d8e752660d0e11 100755 (executable)
--- a/src/PartSet/PartSet_FilterInfinite.h
+++ b/src/PartSet/PartSet_FilterInfinite.h
@@ -11,6 +11,8 @@
 
 #include <SelectMgr_Filter.hxx>
 
+class ModuleBase_IWorkshop;
+
 /**
 * \class PartSet_FilterInfinite
 * \ingroup Modules
@@ -22,14 +24,17 @@ class PartSet_FilterInfinite : public SelectMgr_Filter
 public:
   /// Constructor
   /// \param theWorkshop a pointer to workshop
-  PartSet_FilterInfinite()
-    : SelectMgr_Filter() {}
+  PartSet_FilterInfinite(ModuleBase_IWorkshop* theWorkshop);
 
   /// Returns True if selected presentation can be selected
   /// \param theOwner an owner of the persentation
   Standard_EXPORT virtual Standard_Boolean IsOk(const Handle(SelectMgr_EntityOwner)& theOwner) const;
 
   DEFINE_STANDARD_RTTI(PartSet_FilterInfinite)
+
+protected:
+  /// Reference to workshop
+  ModuleBase_IWorkshop* myWorkshop;
 };
 
 #endif
\ No newline at end of file