src/PartSet/PartSet_OperationPrs.h

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        PartSet_OperationPrs.h
   4 // Created:     01 Jul 2015
   5 // Author:      Natalia ERMOLAEVA
   6
   7 #ifndef PartSet_OperationPrs_H
   8 #define PartSet_OperationPrs_H
   9
  10 #include "PartSet.h"
  11
  12 #include <ModelAPI_Object.h>
  13 #include <ModelAPI_Feature.h>
  14 #include <ModelAPI_Attribute.h>
  15
  16 #include <ModuleBase_IWorkshop.h>
  17
  18 #include <GeomAPI_Shape.h>
  19
  20 #include <ViewerData_AISShape.hxx>
  21 #include <Standard_DefineHandle.hxx>
  22
  23 #include <QMap>
  24 #include <QList>
  25
  26 DEFINE_STANDARD_HANDLE(PartSet_OperationPrs, ViewerData_AISShape)
  27
  28 /**
  29 * \ingroup GUI
  30 * A redefinition of standard AIS Interactive Object in order to provide colored presentation of
  31 * a list of shapes. It contains a shapes where the parameter feature refers. It processes the following
  32 * types of attributes: ModelAPI_AttributeSelectionList, ModelAPI_AttributeRefAttr,
  33 * ModelAPI_AttributeSelection and ModelAPI_AttributeReference.
  34 * The selection for this presentation is absent.
  35 */
  36 class PartSet_OperationPrs : public ViewerData_AISShape
  37 {
  38 public:
  39   /// Constructor
  40   Standard_EXPORT PartSet_OperationPrs(ModuleBase_IWorkshop* theWorkshop);
  41
  42   /// Returns true if the feature contains attributes, which has references to other features
  43   /// \param theFeature a feature
  44   /// \return boolean result
  45   bool canActivate(const FeaturePtr& theFeature);
  46
  47   /// Sets the operation feature. It is used in Compute method to group the feature parameter shapes
  48   /// theFeature a feature
  49   void setFeature(const FeaturePtr& theFeature);
  50
  51   /// Returns true if the presentation
  52   bool dependOn(const ObjectPtr& theObject);
  53
  54   // Recompute internal list of shaped dependent on the current feature
  55   void updateShapes();
  56
  57   DEFINE_STANDARD_RTTI(PartSet_OperationPrs)
  58
  59 protected:
  60   /// Redefinition of virtual function
  61   Standard_EXPORT virtual void Compute(
  62             const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
  63             const Handle(Prs3d_Presentation)& thePresentation, const Standard_Integer theMode = 0);
  64
  65   /// Redefinition of virtual function
  66   Standard_EXPORT virtual void ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
  67                                                 const Standard_Integer aMode) ;
  68 private:
  69   /// Fills the map by the feature object and shapes, which should be visuaziled
  70   /// Gets the feature attribute, collect objects to whom the attribute refers
  71   /// \param theObjectShape an output map of objects
  72   void getFeatureShapes(QMap<ObjectPtr, QList<GeomShapePtr> >& theObjectShapes);
  73
  74   /// Returns true if the attribute type belong to reference attribute
  75   /// \param theAttribute an attribute
  76   /// \return a boolean value
  77   static bool isSelectionAttribute(const AttributePtr& theAttribute);
  78
  79 private:
  80   ModuleBase_IWorkshop* myWorkshop;
  81   FeaturePtr myFeature; /// Reference to a feature object
  82   QMap<ObjectPtr, QList<GeomShapePtr> > myFeatureShapes;
  83 };
  84
  85
  86 #endif