1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: SketcherPrs_Angle.cpp
4 // Created: 20 August 2015
5 // Author: Vitaly SMETANNIKOV
7 #include <ModelAPI_Tools.h>
8 #include <ModelAPI_AttributeString.h>
10 #include "PartSet_OverconstraintListener.h"
11 #include <PartSet_Module.h>
12 #include <PartSet_SketcherMgr.h>
13 #include <PartSet_SketcherReentrantMgr.h>
15 #include "XGUI_CustomPrs.h"
16 #include "XGUI_Displayer.h"
17 #include "XGUI_ModuleConnector.h"
18 #include "XGUI_OperationMgr.h"
19 #include "XGUI_Tools.h"
20 #include "XGUI_Workshop.h"
22 #include "SketcherPrs_SymbolPrs.h"
23 #include "SketchPlugin_SketchEntity.h"
24 #include "SketchPlugin_MacroArcReentrantMessage.h"
25 #include "SketchPlugin_Sketch.h"
27 #include "Events_Loop.h"
29 #include <GeomAPI_IPresentable.h>
30 #include <ModelAPI_Events.h>
31 #include <ModelAPI_EventReentrantMessage.h>
32 #include <ModuleBase_Tools.h>
36 //#define DEBUG_FEATURE_OVERCONSTRAINT_LISTENER
38 PartSet_OverconstraintListener::PartSet_OverconstraintListener(ModuleBase_IWorkshop* theWorkshop)
39 : myWorkshop(theWorkshop), myIsActive(false), myIsFullyConstrained(false)
41 Events_Loop* aLoop = Events_Loop::loop();
42 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_SOLVER_FAILED));
43 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_SOLVER_REPAIRED));
45 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_SKETCH_UNDER_CONSTRAINED));
46 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_SKETCH_FULLY_CONSTRAINED));
48 aLoop->registerListener(this, ModelAPI_EventReentrantMessage::eventId());
49 aLoop->registerListener(this, SketchPlugin_MacroArcReentrantMessage::eventId());
52 void PartSet_OverconstraintListener::setActive(const bool& theActive)
54 if (myIsActive == theActive)
57 myIsActive = theActive;
58 myIsFullyConstrained = false; /// returned to default state, no custom color for it
61 PartSet_Module* aModule = module();
62 CompositeFeaturePtr aSketch = aModule->sketchMgr()->activeSketch();
64 std::string aDOFMessage = aSketch->string(SketchPlugin_Sketch::SOLVER_DOF())->value();
65 myIsFullyConstrained = QString(aDOFMessage.c_str()).contains("DoF = 0");
70 void PartSet_OverconstraintListener::getCustomColor(const ObjectPtr& theObject,
71 std::vector<int>& theColor)
76 if (myConflictingObjects.find(theObject) != myConflictingObjects.end()) {
77 theColor = Config_PropManager::color("Visualization", "sketch_overconstraint_color");
79 if (myIsFullyConstrained) {
80 FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
81 // only entity features has custom color when sketch is fully constrained
82 if (aFeature.get() && PartSet_SketcherMgr::isEntity(aFeature->getKind()) &&
83 !PartSet_SketcherMgr::isExternalFeature(aFeature)) {
84 PartSet_Module* aModule = module();
85 CompositeFeaturePtr aSketch = aModule->sketchMgr()->activeSketch();
86 // the given object is sub feature of the current sketch(created or edited)
87 if (ModelAPI_Tools::compositeOwner(aFeature) == aSketch)
88 theColor = Config_PropManager::color("Visualization", "sketch_fully_constrained_color");
93 void PartSet_OverconstraintListener::processEvent(
94 const std::shared_ptr<Events_Message>& theMessage)
99 #ifdef DEBUG_FEATURE_OVERCONSTRAINT_LISTENER
100 bool isRepaired = theMessage->eventID() == Events_Loop::eventByName(EVENT_SOLVER_REPAIRED);
103 std::shared_ptr<ModelAPI_SolverFailedMessage> anErrorMsg =
104 std::dynamic_pointer_cast<ModelAPI_SolverFailedMessage>(theMessage);
106 if (anErrorMsg.get()) {
107 const std::set<ObjectPtr>& aConflictingObjects = anErrorMsg->objects();
108 aCount = aConflictingObjects.size();
109 anInfoStr = getObjectsInfo(aConflictingObjects);
112 QString aCurrentInfoStr = getObjectsInfo(myConflictingObjects);
114 QString aMsg("PartSet_OverconstraintListener::processEvent: %1,\nobjects "
115 "count = %2:%3\ncurrent objects count = %4:%5");
116 qDebug(aMsg.arg(isRepaired ? "REPAIRED" : "FAILED")
117 .arg(aCount).arg(anInfoStr).arg(myConflictingObjects.size())
118 .arg(aCurrentInfoStr).toStdString().c_str());
121 Events_ID anEventID = theMessage->eventID();
122 if (anEventID == Events_Loop::eventByName(EVENT_SOLVER_FAILED) ||
123 anEventID == Events_Loop::eventByName(EVENT_SOLVER_REPAIRED)) {
124 std::shared_ptr<ModelAPI_SolverFailedMessage> anErrorMsg =
125 std::dynamic_pointer_cast<ModelAPI_SolverFailedMessage>(theMessage);
126 bool anUpdated = false;
127 if (anErrorMsg.get()) {
128 const std::set<ObjectPtr>& aConflictingObjects = anErrorMsg->objects();
129 if (anEventID == Events_Loop::eventByName(EVENT_SOLVER_FAILED))
130 anUpdated = appendConflictingObjects(aConflictingObjects);
132 anUpdated = repairConflictingObjects(aConflictingObjects);
135 else if (anEventID == Events_Loop::eventByName(EVENT_SKETCH_UNDER_CONSTRAINED) ||
136 anEventID == Events_Loop::eventByName(EVENT_SKETCH_FULLY_CONSTRAINED)) {
137 bool aPrevFullyConstrained = myIsFullyConstrained;
138 myIsFullyConstrained = anEventID == Events_Loop::eventByName(EVENT_SKETCH_FULLY_CONSTRAINED);
140 if (aPrevFullyConstrained != myIsFullyConstrained) {
141 std::set<ObjectPtr> aModifiedObjects;
142 PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
143 CompositeFeaturePtr aSketch = aModule->sketchMgr()->activeSketch();
145 for (int i = 0; i < aSketch->numberOfSubs(); i++) {
146 FeaturePtr aFeature = aSketch->subFeature(i);
147 aModifiedObjects.insert(aFeature); // is necessary to redisplay presentations
148 std::list<ResultPtr> aResults = aFeature->results();
149 for (std::list<ResultPtr>::const_iterator aIt = aResults.begin();
150 aIt != aResults.end(); ++aIt) {
151 aModifiedObjects.insert(*aIt);
154 redisplayObjects(aModifiedObjects);
158 else if (anEventID == ModelAPI_EventReentrantMessage::eventId() ||
159 anEventID == SketchPlugin_MacroArcReentrantMessage::eventId()) {
160 // the message is sent to sketcher reentrant manager only if the kind of feature
161 // sender is equal to kind of the current operation. E.g. Horizontal create operation
162 // is active. Sketch Line feature is changed, so execute is called, it will send message
163 // This Line's message should not be processed, as the reentrant operation is not for Line
164 std::string aCurrentFeatureKind;
165 ModuleBase_Operation* anOperation =
166 XGUI_Tools::workshop(myWorkshop)->operationMgr()->currentOperation();
168 ModuleBase_OperationFeature* aFOperation = dynamic_cast<ModuleBase_OperationFeature*>
171 FeaturePtr aFeature = aFOperation->feature();
173 aCurrentFeatureKind = aFeature->getKind();
176 if (theMessage->sender()) {
177 ModelAPI_Object* aSender = static_cast<ModelAPI_Object*>(theMessage->sender());
179 FeaturePtr aFeatureSender =
180 std::dynamic_pointer_cast<ModelAPI_Feature>(aSender->data()->owner());
181 if (aFeatureSender.get() && aFeatureSender->getKind() == aCurrentFeatureKind) {
182 PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
183 PartSet_SketcherReentrantMgr* aReentrantMgr = aModule->sketchReentranceMgr();
184 aReentrantMgr->setReentrantMessage(theMessage);
190 #ifdef DEBUG_FEATURE_OVERCONSTRAINT_LISTENER
191 aCurrentInfoStr = getObjectsInfo(myConflictingObjects);
192 qDebug(QString("RESULT: current objects count = %1:%2\n")
193 .arg(myConflictingObjects.size()).arg(aCurrentInfoStr).toStdString().c_str());
197 bool PartSet_OverconstraintListener::appendConflictingObjects(
198 const std::set<ObjectPtr>& theConflictingObjects)
200 std::set<ObjectPtr> aModifiedObjects;
202 // set error state for new objects and append them in the internal map of objects
203 std::set<ObjectPtr>::const_iterator
204 anIt = theConflictingObjects.begin(), aLast = theConflictingObjects.end();
205 for (; anIt != aLast; anIt++) {
206 ObjectPtr anObject = *anIt;
207 if (myConflictingObjects.find(anObject) == myConflictingObjects.end()) { // it is not found
208 aModifiedObjects.insert(anObject);
209 myConflictingObjects.insert(anObject);
212 bool isUpdated = !aModifiedObjects.empty();
214 redisplayObjects(aModifiedObjects);
219 bool PartSet_OverconstraintListener::repairConflictingObjects(
220 const std::set<ObjectPtr>& theConflictingObjects)
222 std::set<ObjectPtr> aModifiedObjects;
223 // erase error state of absent current objects in the parameter list
224 std::set<ObjectPtr>::const_iterator anIt, aLast;
225 for (anIt = theConflictingObjects.begin(), aLast = theConflictingObjects.end();
226 anIt != aLast; anIt++) {
227 ObjectPtr anObject = *anIt;
228 if (theConflictingObjects.find(anObject) != theConflictingObjects.end()) { // it is found
229 myConflictingObjects.erase(anObject);
231 aModifiedObjects.insert(anObject);
234 bool isUpdated = !aModifiedObjects.empty();
236 redisplayObjects(aModifiedObjects);
241 void PartSet_OverconstraintListener::redisplayObjects(
242 const std::set<ObjectPtr>& theObjects)
244 static Events_Loop* aLoop = Events_Loop::loop();
246 static Events_ID EVENT_DISP = aLoop->eventByName(EVENT_OBJECT_TO_REDISPLAY);
247 static const ModelAPI_EventCreator* aECreator = ModelAPI_EventCreator::get();
249 std::set<ObjectPtr>::const_iterator anIt = theObjects.begin(), aLast = theObjects.end();
250 for (; anIt != aLast; anIt++)
251 aECreator->sendUpdated(*anIt, EVENT_DISP);
253 aLoop->flush(EVENT_DISP);
256 PartSet_Module* PartSet_OverconstraintListener::module() const
258 return dynamic_cast<PartSet_Module*>(myWorkshop->module());
262 QString PartSet_OverconstraintListener::getObjectsInfo(const std::set<ObjectPtr>& theObjects)
264 std::set<ObjectPtr>::const_iterator anIt = theObjects.begin(),
265 aLast = theObjects.end();
267 for (; anIt != aLast; ++anIt)
268 anInfo.append(ModuleBase_Tools::objectInfo((*anIt)));
270 return anInfo.join(";\n");