PartSet_SketcherMgr.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_SketcherMgr_H
#define PartSet_SketcherMgr_H
 
#include "PartSet.h"
 
#include "PartSet_PreviewSketchPlane.h"
#include "PartSet_Tools.h"
 
#include <ModelAPI_Feature.h>
#include <ModelAPI_Attribute.h>
#include <ModelAPI_CompositeFeature.h>
#include <ModelAPI_Result.h>
#include <Events_Listener.h>
#include <ModuleBase_SelectionFilterType.h>
 
#include <ModuleBase_Definitions.h>
#include <ModuleBase_ModelWidget.h>
 
#include <GeomAPI_Pln.h>
#include <GeomAPI_AISObject.h>
 
#ifdef HAVE_SALOME
  #include <OCCViewer_ViewModel.h>
#else
  #include <AppElements_Viewer.h>
#endif
 
 
#include <SelectMgr_IndexedMapOfOwner.hxx>
#include <SelectMgr_ListOfFilter.hxx>
 
#include <TopoDS_Shape.hxx>
#include <TopTools_MapOfShape.hxx>
#include <AIS_Shape.hxx>
 
#include <QObject>
#include <QList>
#include <QMap>
#include <QPoint>
 
#include <set>
 
class PartSet_Module;
class ModuleBase_IViewWindow;
class ModuleBase_ModelWidget;
class ModuleBase_Operation;
class XGUI_OperationMgr;
class XGUI_Workshop;
class XGUI_Displayer;
class XGUI_SketchConstraintsBrowser;
class PartSet_ExternalPointsMgr;
 
class AIS_InteractiveObject;
 
class QMouseEvent;
class PartSet_SketcherMgr;
 
#ifdef HAVE_SALOME
class PartSet_Fitter : public OCCViewer_Fitter
#else
class PartSet_Fitter : public AppElements_Fitter
#endif
{
public:
  PartSet_Fitter(PartSet_SketcherMgr* theSketchMgr):
    mySketchMgr(theSketchMgr) {}
 
  virtual ~PartSet_Fitter() {}
 
  virtual void fitAll(Handle(V3d_View) theView);
 
private:
  PartSet_SketcherMgr* mySketchMgr;
};
 
 
 
class PARTSET_EXPORT PartSet_SketcherMgr : public QObject, public Events_Listener
{
  Q_OBJECT
  struct Point
  {
    Point()
    {
      myIsInitialized = false;
    }
    ~Point()
    {
    }
 
    void clear()
    {
      myIsInitialized = false;
    }
    void setValue(const double theX, const double theY)
    {
      myIsInitialized = true;
      myCurX = theX;
      myCurY = theY;
    }
 
    bool myIsInitialized;  
    double myCurX, myCurY; 
  };
 
public:
  struct SelectionInfo
  {
    std::map<AttributePtr, int> myAttributes;
    std::set<ResultPtr> myResults; 
    TopoDS_Shape myFirstResultShape; 
    TopTools_MapOfShape myLocalSelectedShapes; 
  };
  typedef QMap<FeaturePtr, SelectionInfo> FeatureToSelectionMap;
 
public:
  PartSet_SketcherMgr(PartSet_Module* theModule);
 
  virtual ~PartSet_SketcherMgr();
 
  static bool isSketchOperation(ModuleBase_Operation* theOperation);
 
  bool isNestedSketchOperation(ModuleBase_Operation* theOperation) const;
 
  bool isNestedSketchFeature(const QString& theFeatureKind) const;
 
  bool isNestedCreateOperation(ModuleBase_Operation* theOperation,
                                      const CompositeFeaturePtr& /*theSketch*/) const;
 
  bool isNestedEditOperation(ModuleBase_Operation* theOperation,
                                    const CompositeFeaturePtr& /*theSketch*/) const;
 
  static bool isEntity(const std::string& theId);
 
  static bool isExternalFeature(const FeaturePtr& theFeature);
 
  static bool isDistanceOperation(ModuleBase_Operation* theOperation);
 
  static bool isDistanceKind(std::string& theKind);
 
  bool isMouseOverWindow() { return myIsMouseOverWindow; }
 
  CompositeFeaturePtr activeSketch() const { return myCurrentSketch; }
 
  PartSet_PreviewSketchPlane* previewSketchPlane() const { return mySketchPlane; }
 
  void startSketch(ModuleBase_Operation* );
 
  void stopSketch(ModuleBase_Operation* );
 
  void startNestedSketch(ModuleBase_Operation* theOperation);
 
  void stopNestedSketch(ModuleBase_Operation* theOperation);
 
  void commitNestedSketch(ModuleBase_Operation* theOperation);
 
  bool sketchSelectionFilter(const ModuleBase_SelectionFilterType theFilterType);
 
  void registerSelectionFilter(const ModuleBase_SelectionFilterType theFilterType,
                               const Handle(SelectMgr_Filter)& theFilter);
 
  bool operationActivatedByPreselection();
 
  bool canUndo() const;
 
  bool canRedo() const;
 
  //bool canCommitOperation() const;
 
  bool canEraseObject(const ObjectPtr& theObject) const;
 
  bool canDisplayObject(const ObjectPtr& theObject) const;
 
  bool canDisplayConstraint(const FeaturePtr& theObject,
                            const PartSet_Tools::ConstraintVisibleState& theState,
                            bool& isProcessed) const;
 
  //virtual void processHiddenObject(const std::list<ObjectPtr>& theObjects);
 
  bool canDisplayCurrentCreatedFeature() const;
 
  bool canChangeCursor(ModuleBase_Operation* theOperation) const;
 
  const QMap<PartSet_Tools::ConstraintVisibleState, bool>& showConstraintStates();
 
  bool isObjectOfSketch(const ObjectPtr& theObject) const;
 
  enum SelectionType {
    ST_HighlightType, 
    ST_SelectType, 
    ST_SelectAndHighlightType 
  };
 
  void storeSelection(const SelectionType theType, FeatureToSelectionMap& theCurrentSelection);
 
  void restoreSelection(FeatureToSelectionMap& theCurrentSelection);
 
  bool sketchSolverError();
 
  QString getFeatureError(const FeaturePtr& theFeature);
 
  void clearClickedFlags();
 
  static const QStringList& replicationsIdList();
 
  static const QStringList& constraintsIdList();
 
  static void sketchSelectionModes(const CompositeFeaturePtr& theSketch, QIntList& theModes);
 
  Handle(AIS_InteractiveObject) createPresentation(const ObjectPtr& theResult);
 
  void connectToPropertyPanel(ModuleBase_ModelWidget* theWidget, const bool isToConnect);
 
  void widgetStateChanged(int thePreviousState);
 
  //void customisePresentation(const ObjectPtr& theObject);
 
  void customizeSketchPresentation(const ObjectPtr& theObject, const AISObjectPtr& thePrs) const;
 
  void updateBySketchParameters(const PartSet_Tools::ConstraintVisibleState& theType,
                                bool theState);
 
  bool isShowFreePointsShown() const {
    return myPointsHighlight.size() > 0;
  }
 
  PartSet_Module* module() const { return myModule; }
 
  virtual void processEvent(const std::shared_ptr<Events_Message>& theMessage);
 
  bool isDragModeCreation() const;
 
  // Get constraints browser
  XGUI_SketchConstraintsBrowser* constraintsBrowser();
 
public slots:
  void onPlaneSelected(const std::shared_ptr<GeomAPI_Pln>& thePln);
 
  void onShowPoints(bool toShow);
 
  void onEditValues();
  void onUpdateConstraintsList();
  void onDeactivate(bool isNeedDeactivate, std::vector<FeaturePtr> theFeatures);
 
private slots:
  void onShowConstraintsToggle(int theType, bool theState);
  void onEnterViewPort();
  void onLeaveViewPort();
  //void onBeforeValuesChangedInPropertyPanel();
  //void onAfterValuesChangedInPropertyPanel();
 
  void onMousePressed(ModuleBase_IViewWindow*, QMouseEvent*);
  void onMouseReleased(ModuleBase_IViewWindow*, QMouseEvent*);
  void onMouseMoved(ModuleBase_IViewWindow*, QMouseEvent*);
  void onMouseDoubleClick(ModuleBase_IViewWindow*, QMouseEvent*);
  void onApplicationStarted();
  //void onBeforeWidgetActivated(ModuleBase_ModelWidget* theWidget);
 
  void onBeforeContextMenu();
  void onAfterContextMenu();
 
private:
  void launchEditing();
 
  void get2dPoint(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent,
                  Point& thePoint, bool theSnap = true);
 
  static bool setDistanceValueByPreselection(ModuleBase_Operation* theOperation,
                                             ModuleBase_IWorkshop* theWorkshop,
                                             bool& theCanCommitOperation);
 
  static void getSelectionOwners(const FeaturePtr& theFeature,
                                  const FeaturePtr& theSketch,
                                  ModuleBase_IWorkshop* theWorkshop,
                                  const FeatureToSelectionMap& theSelection,
                                  SelectMgr_IndexedMapOfOwner& theOwnersToSelect);
 
  bool isVisibleCreatedFeature() const;
 
  ModuleBase_Operation* getCurrentOperation() const;
 
  ModuleBase_ModelWidget* getActiveWidget() const;
 
  void visualizeFeature(const FeaturePtr& theFeature, const bool isEditOperation,
                        const bool isToDisplay, const bool isFlushRedisplay = true);
 
private:
  void updateSelectionPriority(ObjectPtr theObject, FeaturePtr theFeature);
  XGUI_Workshop* workshop() const;
  XGUI_OperationMgr* operationMgr() const;
 
  std::vector<int> colorOfObject(const ObjectPtr& theObject,
    const FeaturePtr& aFeature, bool isConstruction, bool isSuppressedConstraint) const;
 
  friend class PartSet_PreviewSketchPlane;
 
private:
  PartSet_Module* myModule;
  PartSet_PreviewSketchPlane* mySketchPlane; // display/erase sketch plane on start/stop sketch
 
  bool myPreviousDrawModeEnabled; // the previous selection enabled state in the viewer
  bool myIsEditLaunching;
  bool myIsDragging;
  bool myDragDone;
  bool myIsMouseOverWindow; 
  bool myIsMouseOverViewProcessed;
  bool myIsPopupMenuActive; 
  Point myCurrentPoint;
  //Point myClickedPoint;
 
  CompositeFeaturePtr myCurrentSketch;
 
  std::set<ModuleBase_SelectionFilterType> mySelectionFilterTypes;
 
  FeatureToSelectionMap myCurrentSelection;
  bool myPreviousUpdateViewerEnabled;
 
  QMap<PartSet_Tools::ConstraintVisibleState, bool> myIsConstraintsShown;
 
  PartSet_ExternalPointsMgr* myExternalPointsMgr;
 
  QMap<ResultPtr, Handle(AIS_Shape)> myPointsHighlight;
 
  bool myNoDragMoving;
 
  QPoint myMousePoint;
 
  QDockWidget* myConstraintsBrowser;
};
 
 
#endif