PartSet_OperationPrs.h

// Copyright (C) 2014-2025  CEA, EDF
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
//
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//
 
#ifndef PartSet_OperationPrs_H
#define PartSet_OperationPrs_H
 
#include "PartSet.h"
 
#include <ModelAPI_Object.h>
#include <ModelAPI_Feature.h>
#include <ModelAPI_Result.h>
#include <ModelAPI_Attribute.h>
 
#include <ModuleBase_IWorkshop.h>
 
#include <GeomAPI_Shape.h>
 
#include <ViewerData_AISShape.hxx>
#include <Standard_DefineHandle.hxx>
 
#include <Quantity_Color.hxx>
#include <NCollection_List.hxx>
#include <NCollection_DataMap.hxx>
#include <TopoDS_Shape.hxx>
#include <AIS_InteractiveObject.hxx>
 
#include <QMap>
#include <QList>
 
#include <list>
 
DEFINE_STANDARD_HANDLE(PartSet_OperationPrs, ViewerData_AISShape)
 
class XGUI_Displayer;
class SelectMgr_Selection;
 
class PartSet_OperationPrs : public ViewerData_AISShape
{
public:
  Standard_EXPORT PartSet_OperationPrs(ModuleBase_IWorkshop* theWorkshop);
 
  bool hasShapes();
 
  void setShapeColor(const Quantity_Color& theColor);
 
  void useAISWidth();
 
  DEFINE_STANDARD_RTTIEXT(PartSet_OperationPrs, ViewerData_AISShape)
 
protected:
  Standard_EXPORT virtual void Compute(
            const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
            const Handle(Prs3d_Presentation)& thePresentation, const Standard_Integer theMode = 0);
 
  Standard_EXPORT virtual void ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
                                                const Standard_Integer aMode) ;
 
protected:
  NCollection_DataMap<TopoDS_Shape, Handle(AIS_InteractiveObject)>& shapesMap();
 
private:
  static void getFeatureShapes(const FeaturePtr& theFeature,
                               ModuleBase_IWorkshop* theWorkshop,
                               QMap<ObjectPtr, QList<GeomShapePtr> >& theObjectShapes);
 
  static void getResultShapes(const FeaturePtr& theFeature,
                              ModuleBase_IWorkshop* theWorkshop,
                              QMap<ObjectPtr, QList<GeomShapePtr> >& theObjectShapes,
                              const bool theListShouldBeCleared = true);
 
  static void getPresentationShapes(const FeaturePtr& theFeature,
                              ModuleBase_IWorkshop* theWorkshop,
                              QMap<ObjectPtr, QList<GeomShapePtr> >& theObjectShapes,
                              const bool theListShouldBeCleared = true);
 
  static void getHighlightedShapes(ModuleBase_IWorkshop* theWorkshop,
                                   QMap<ObjectPtr, QList<GeomShapePtr> >& theObjectShapes);
 
  static bool isSelectionAttribute(const AttributePtr& theAttribute);
 
  static void addValue(const ObjectPtr& theObject, const GeomShapePtr& theShape,
                const FeaturePtr& theFeature, ModuleBase_IWorkshop* theWorkshop,
                QMap<ObjectPtr, QList<GeomShapePtr> >& theObjectShapes);
 
  static void appendShapeIfVisible(ModuleBase_IWorkshop* theWorkshop,
                                   const ObjectPtr& theObject,
                                   GeomShapePtr theGeomShape,
                                   QMap<ObjectPtr, QList<GeomShapePtr> >& theObjectShapes);
 
  static void fillShapeList(const QMap<ObjectPtr, QList<GeomShapePtr> >& theFeatureShapes,
          ModuleBase_IWorkshop* theWorkshop,
          NCollection_DataMap<TopoDS_Shape, Handle(AIS_InteractiveObject)>& theShapeToPrsMap);
 
private:
  NCollection_DataMap<TopoDS_Shape, Handle(AIS_InteractiveObject)> myShapeToPrsMap;
 
  ModuleBase_IWorkshop* myWorkshop; 
  Quantity_Color myShapeColor; 
 
  bool myUseAISWidth;
 
  friend class PartSet_CustomPrs;
};
 
 
#endif