src/SketchSolver/SketchSolver_Storage.h

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:    SketchSolver_Storage.h
   4 // Created: 30 Nov 2015
   5 // Author:  Artem ZHIDKOV
   6
   7 #ifndef SketchSolver_Storage_H_
   8 #define SketchSolver_Storage_H_
   9
  10 #include <SketchSolver.h>
  11 #include <SketchSolver_IConstraintWrapper.h>
  12 #include <SketchSolver_IEntityWrapper.h>
  13 #include <SketchSolver_IParameterWrapper.h>
  14 #include <SketchSolver_ISolver.h>
  15
  16 #include <ModelAPI_Attribute.h>
  17 #include <ModelAPI_AttributeDouble.h>
  18 #include <ModelAPI_Feature.h>
  19 #include <SketchPlugin_Constraint.h>
  20
  21 class SketchSolver_ConstraintDistance;
  22 class SketchSolver_ConstraintFixedArcRadius;
  23 typedef std::map<EntityWrapperPtr, std::set<EntityWrapperPtr> > CoincidentPointsMap;
  24
  25
  26 /** \class   SketchSolver_Storage
  27  *  \ingroup Plugins
  28  *  \brief   Interface to map SketchPlugin features to the entities of corresponding solver.
  29  */
  30 class SketchSolver_Storage
  31 {
  32 private:
  33   SketchSolver_Storage();
  34   SketchSolver_Storage(const SketchSolver_Storage&);
  35   SketchSolver_Storage& operator=(const SketchSolver_Storage&);
  36
  37 public:
  38   SketchSolver_Storage(const GroupID& theGroup)
  39     : myGroupID(theGroup),
  40       mySketchID(EID_UNKNOWN),
  41       myNeedToResolve(false),
  42       myEventsBlocked(false),
  43       myExistArc(false)
  44   {}
  45
  46   /// \brief Change mapping between constraint from SketchPlugin and
  47   ///        a constraint applicable for corresponding solver.
  48   /// \param theConstraint       [in]   original SketchPlugin constraint
  49   /// \param theSolverConstraint [in]   solver's constraints
  50   SKETCHSOLVER_EXPORT void addConstraint(ConstraintPtr        theConstraint,
  51                                          ConstraintWrapperPtr theSolverConstraints);
  52   /// \brief Change mapping between constraint from SketchPlugin and
  53   ///        the list of constraints applicable for corresponding solver.
  54   /// \param theConstraint        [in]   original SketchPlugin constraint
  55   /// \param theSolverConstraints [in]   list of solver's constraints
  56   SKETCHSOLVER_EXPORT virtual
  57     void addConstraint(ConstraintPtr                   theConstraint,
  58                        std::list<ConstraintWrapperPtr> theSolverConstraints);
  59
  60   /// \brief Convert feature to the form applicable for specific solver and map it
  61   /// \param theFeature [in]  feature to convert
  62   /// \param theGroup   [in]  id of the group where the feature should be placed
  63   /// \return \c true if the feature has been created or updated
  64   SKETCHSOLVER_EXPORT bool update(FeaturePtr theFeature, const GroupID& theGroup = GID_UNKNOWN);
  65   /// \brief Convert attribute to the form applicable for specific solver and map it
  66   /// \param theFeature [in]  feature to convert
  67   /// \return \c true if the attribute has been created or updated
  68   SKETCHSOLVER_EXPORT bool update(AttributePtr theAttribute, const GroupID& theGroup = GID_UNKNOWN);
  69
  70   /// \brief Returns constraint related to corresponding constraint
  71   SKETCHSOLVER_EXPORT
  72     const std::list<ConstraintWrapperPtr>& constraint(const ConstraintPtr& theConstraint) const;
  73
  74   /// \brief Returns entity related to corresponding feature
  75   SKETCHSOLVER_EXPORT const EntityWrapperPtr& entity(const FeaturePtr& theFeature) const;
  76   /// \brief Returns entity related to corresponding attribute
  77   SKETCHSOLVER_EXPORT const EntityWrapperPtr& entity(const AttributePtr& theAttribute) const;
  78
  79   /// \brief Return parsed sketch entity
  80   const EntityWrapperPtr& sketch() const;
  81   /// \brief Set parsed sketch entity.
  82   /// Be careful, this method does not update fields of the storage specific for the solver.
  83   /// Does not update if the sketch already exists.
  84   void setSketch(const EntityWrapperPtr& theSketch);
  85
  86   /// \brief Mark two points as coincident
  87   virtual void addCoincidentPoints(EntityWrapperPtr theMaster, EntityWrapperPtr theSlave) = 0;
  88
  89   /// \brief Shows the storage has any constraint twice
  90   virtual bool hasDuplicatedConstraint() const = 0;
  91
  92   /// \brief Removes constraint from the storage
  93   /// \return \c true if the constraint and all its parameters are removed successfully
  94   SKETCHSOLVER_EXPORT bool removeConstraint(ConstraintPtr theConstraint);
  95   /// \brief Removes feature from the storage
  96   /// \return \c true if the feature and its attributes are removed successfully;
  97   ///         \c false if the feature or any it attribute is used by remaining constraints.
  98   SKETCHSOLVER_EXPORT bool removeEntity(FeaturePtr theFeature);
  99   /// \brief Removes attribute from the storage
 100   /// \return \c true if the attribute is not used by remaining features and constraints
 101   SKETCHSOLVER_EXPORT bool removeEntity(AttributePtr theAttribute);
 102
 103   /// \brief Remove all features became invalid
 104   SKETCHSOLVER_EXPORT void removeInvalidEntities();
 105
 106   /// \brief Check whether the feature or its attributes are used by this storage
 107   /// \param theFeature [in]  feature to be checked
 108   /// \return \c true if the feature interacts with the storage
 109   bool isInteract(const FeaturePtr& theFeature) const;
 110   /// \brief Check whether the attribute is used by this storage
 111   /// \param theAttribute [in]  attribute to be checked
 112   /// \return \c true if the attribute interacts with the storage
 113   bool isInteract(const AttributePtr& theAttribute) const;
 114
 115   /// \brief Check the features is not removed
 116   bool isConsistent() const;
 117
 118   /// \brief Check the storage has constraints
 119   bool isEmpty() const
 120   { return myConstraintMap.empty(); }
 121
 122   /// \brief Check the entity is fixed.
 123   ///        If the point is under verification, all coincident points are checked too.
 124   SKETCHSOLVER_EXPORT bool isFixed(EntityWrapperPtr theEntity) const;
 125
 126   /// \brief Shows the sketch should be resolved
 127   virtual bool isNeedToResolve()
 128   { return myNeedToResolve; }
 129   /// \brief Changes the flag of group to be resolved
 130   void setNeedToResolve(bool theFlag)
 131   { myNeedToResolve = theFlag; }
 132
 133   /// \brief Initialize solver by constraints, entities and parameters
 134   virtual void initializeSolver(SolverPtr theSolver) = 0;
 135   /// \brief Return list of conflicting constraints
 136   std::set<ObjectPtr> getConflictingConstraints(SolverPtr theSolver) const;
 137
 138   /// \brief Update SketchPlugin features after resolving constraints
 139   /// \param theFixedOnly [in]  if \c true the fixed points will be updated only
 140   virtual void refresh(bool theFixedOnly = false) const = 0;
 141
 142   /// \brief Check if some parameters or entities are returned
 143   ///        to the current group after removing temporary constraints
 144   virtual void verifyFixed() = 0;
 145
 146   /// \brief Calculate point on theBase entity. Value theCoeff is in [0.0 .. 1.0] and
 147   ///        shows the distance from the start point.
 148   virtual EntityWrapperPtr calculateMiddlePoint(EntityWrapperPtr theBase,
 149                                                 double theCoeff) = 0;
 150
 151   /// \brief Block or unblock events when refreshing features
 152   SKETCHSOLVER_EXPORT void blockEvents(bool isBlocked);
 153   /// \brief Shows the events are blocked for the features in the storage
 154   bool isEventsBlocked() const
 155   { return myEventsBlocked; }
 156
 157 protected:
 158   /// \brief Change mapping feature from SketchPlugin and
 159   ///        the entity applicable for corresponding solver.
 160   /// \param theFeature      [in]  original SketchPlugin feature
 161   /// \param theSolverEntity [in]  solver's entity, created outside
 162   SKETCHSOLVER_EXPORT
 163     void addEntity(FeaturePtr       theFeature,
 164                    EntityWrapperPtr theSolverEntity);
 165
 166   /// \brief Change mapping attribute of a feature and the entity applicable for corresponding solver.
 167   /// \param theAttribute    [in]  original attribute
 168   /// \param theSolverEntity [in]  solver's entity, created outside
 169   SKETCHSOLVER_EXPORT
 170     void addEntity(AttributePtr     theAttribute,
 171                    EntityWrapperPtr theSolverEntity);
 172
 173   /// \brief Update constraint's data
 174   /// \return \c true if any value is updated
 175   virtual bool update(ConstraintWrapperPtr theConstraint) = 0;
 176   /// \brief Update entity's data
 177   /// \return \c true if any value is updated
 178   virtual bool update(EntityWrapperPtr theEntity) = 0;
 179   /// \brief Update parameter's data
 180   /// \return \c true if the value of parameter is updated
 181   virtual bool update(ParameterWrapperPtr theParameter) = 0;
 182
 183   /// \brief Remove constraint
 184   /// \return \c true if the constraint and all its parameters are removed successfully
 185   SKETCHSOLVER_EXPORT virtual bool remove(ConstraintWrapperPtr theConstraint);
 186   /// \brief Remove entity
 187   /// \return \c true if the entity and all its parameters are removed successfully
 188   SKETCHSOLVER_EXPORT virtual bool remove(EntityWrapperPtr theEntity);
 189   /// \brief Remove parameter
 190   /// \return \c true if the parameter has been removed
 191   virtual bool remove(ParameterWrapperPtr theParameter) = 0;
 192
 193   /// \brief Remove point-point coincidence
 194   SKETCHSOLVER_EXPORT bool removeCoincidence(ConstraintWrapperPtr theConstraint);
 195
 196   /// \brief Update the group for the given entity, its sub-entities and parameters
 197   virtual void changeGroup(EntityWrapperPtr theEntity, const GroupID& theGroup) = 0;
 198   /// \brief Update the group for the given parameter
 199   virtual void changeGroup(ParameterWrapperPtr theParam, const GroupID& theGroup) = 0;
 200
 201   /// \brief Verify the feature or any its attribute is used by constraint
 202   SKETCHSOLVER_EXPORT bool isUsed(FeaturePtr theFeature) const;
 203   /// \brief Verify the attribute is used by constraint
 204   SKETCHSOLVER_EXPORT bool isUsed(AttributePtr theAttirubute) const;
 205
 206   /// \brief Find arcs without corresponding entity applicable for the solver and build them
 207   SKETCHSOLVER_EXPORT void processArcs();
 208
 209 private:
 210   /// \brief Find the normal of the sketch
 211   EntityWrapperPtr getNormal() const;
 212
 213 protected:
 214   EntityID mySketchID;     ///< identifier of the sketch
 215   GroupID myGroupID;       ///< identifier of the group, this storage belongs to
 216   bool    myNeedToResolve; ///< parameters are changed and group needs to be resolved
 217   bool    myEventsBlocked; ///< indicates that features do not send events
 218   bool    myExistArc;      ///< the storage has any point of arc but not full arc, need to add it
 219
 220   /// map SketchPlugin constraint to a list of solver's constraints
 221   std::map<ConstraintPtr, std::list<ConstraintWrapperPtr> > myConstraintMap;
 222   /// map SketchPlugin feature to solver's entity
 223   std::map<FeaturePtr, EntityWrapperPtr>                    myFeatureMap;
 224   /// map attribute to solver's entity
 225   std::map<AttributePtr, EntityWrapperPtr>                  myAttributeMap;
 226
 227   CoincidentPointsMap myCoincidentPoints; ///< lists of coincident points (first is a master point, second is a set of slaves)
 228
 229   // to be able to update entities from constraints
 230   friend class SketchSolver_ConstraintDistance;
 231   friend class SketchSolver_ConstraintFixedArcRadius;
 232 };
 233
 234 typedef std::shared_ptr<SketchSolver_Storage> StoragePtr;
 235
 236 #endif