PartSet_CustomPrs.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_CustomPrs_H
#define PartSet_CustomPrs_H
 
#include "PartSet.h"
 
#include "PartSet_OperationPrs.h"
 
#include <ModuleBase_IModule.h>
#include <ModelAPI_Object.h>
#include <ModelAPI_Result.h>
#include <ModelAPI_Feature.h>
 
#include <Events_Listener.h>
 
#include <GeomAPI_ICustomPrs.h>
#include <GeomAPI_AISObject.h>
#include <GeomAPI_Shape.h>
 
class ModuleBase_IWorkshop;
class XGUI_Workshop;
 
class PartSet_CustomPrs : public Events_Listener
{
public:
  static const std::string OPERATION_PARAMETER_COLOR() { return "255, 255, 0"; }
  static const std::string OPERATION_RESULT_COLOR() { return "0,225, 0"; }
  static const std::string OPERATION_HIGHLIGHT_COLOR() { return "128, 0, 0"; }
 
  static const std::string OPERATION_REMOVE_FEATURE_COLOR() { return "255, 174, 201"; }
 
  static const std::string OPERATION_SKETCH_PLANE() { return "170,255,0"; }
public:
  PARTSET_EXPORT PartSet_CustomPrs(ModuleBase_IWorkshop* theWorkshop);
  PARTSET_EXPORT virtual ~PartSet_CustomPrs() {};
 
  bool isActive(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag);
 
  bool activate(const FeaturePtr& theObject,
                const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
                const bool theUpdateViewer);
 
  bool deactivate(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
                  const bool theUpdateViewer);
 
  bool redisplay(const ObjectPtr& theObject,
                 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
                 const bool theUpdateViewer);
 
  void clearPrs();
 
  void disableCustomMode(ModuleBase_IModule::ModuleBase_CustomizeFlag theMode) {
    myDisabledMode = theMode;
    erasePresentation(theMode, false);
  }
 
  void enableCustomModes() { myDisabledMode = -1; }
 
  virtual void processEvent(const std::shared_ptr<Events_Message>& theMessage);
 
private:
  void initPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag);
 
  AISObjectPtr getPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
                               const bool theToCreate);
 
  XGUI_Workshop* workshop() const;
 
  bool displayPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
                           const bool theUpdateViewer);
 
  bool erasePresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
                         const bool theUpdateViewer);
 
  void clearPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag);
 
  Quantity_Color getShapeColor(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag);
 
  void clearErrorShape();
 
private:
  ModuleBase_IWorkshop* myWorkshop; 
  FeaturePtr myFeature; 
  bool myPresentationIsEmpty; 
  QMap<ModuleBase_IModule::ModuleBase_CustomizeFlag, AISObjectPtr> myPresentations;
  QMap<ModuleBase_IModule::ModuleBase_CustomizeFlag, bool> myIsActive;
 
  int myDisabledMode;
 
  Handle(AIS_Shape) myErrorShapes;
};
 
#endif