1 // Copyright (C) 2014-2019 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include "PartSet_IconFactory.h"
21 #include "PartSet_Tools.h"
22 #include "PartSet_Module.h"
23 #include "XGUI_DataModel.h"
25 #include <XGUI_Workshop.h>
26 #include <XGUI_ObjectsBrowser.h>
28 #include <ModuleBase_ActionInfo.h>
29 #include <ModuleBase_Tools.h>
31 #include <ModelAPI_ResultPart.h>
32 #include <ModelAPI_ResultConstruction.h>
33 #include <ModelAPI_ResultBody.h>
35 #include <Config_FeatureMessage.h>
36 #include <Events_Loop.h>
38 #include <CollectionPlugin_Group.h>
40 QMap<QString, QString> PartSet_IconFactory::myIcons;
42 PartSet_IconFactory::PartSet_IconFactory(PartSet_Module* theModule)
43 : ModuleBase_IconFactory(), myModule(theModule)
45 Events_Loop::loop()->registerListener(this,
46 Events_Loop::eventByName(Config_FeatureMessage::GUI_EVENT()));
50 QIcon PartSet_IconFactory::getIcon(ObjectPtr theObj)
55 std::string aGroup = theObj->groupName();
56 FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(theObj);
58 std::string aKind = aFeature->getKind();
59 QString aId(aKind.c_str());
60 if (!myIcons.contains(aId))
63 QString anIconString = myIcons[aId];
65 ModelAPI_ExecState aState = aFeature->data()->execState();
67 case ModelAPI_StateMustBeUpdated:
68 return ModuleBase_Tools::composite(":icons/toWork.png", anIconString);
69 //anIcon = ModuleBase_Tools::lighter(anIconString);
71 case ModelAPI_StateExecFailed:
72 return ModuleBase_Tools::composite(":icons/isFailed.png", anIconString);
74 case ModelAPI_StateInvalidArgument:
75 return ModuleBase_Tools::composite(":icons/exec_state_invalid_parameters.png",
79 return loadIcon(anIconString);
84 //if (theObj->data() && theObj->data()->execState() == ModelAPI_StateMustBeUpdated)
85 // return QIcon(":pictures/constr_object_modified.png");
87 if (aGroup == ModelAPI_ResultPart::group())
88 return QIcon(":pictures/part_ico.png");
90 if (aGroup == ModelAPI_ResultConstruction::group())
91 return QIcon(":pictures/constr_object.png");
93 if (aGroup == ModelAPI_Folder::group()) {
94 static QString anIconString(":pictures/features_folder.png");
95 int aFirst = -1, aLast = -1;
96 PartSet_Tools::getFirstAndLastIndexInFolder(theObj, aFirst, aLast);
97 if ((aFirst != -1) && (aLast != -1)) {
98 int aNbItems = aLast - aFirst + 1;
100 XGUI_ObjectsBrowser* aObBrowser = myModule->getWorkshop()->objectBrowser();
101 XGUI_DataTree* aTree = aObBrowser->treeView();
102 QModelIndex aIndex = aTree->dataModel()->objectIndex(theObj, 0);
103 if (!aTree->isExpanded(aIndex)) {
104 DocumentPtr aDoc = theObj->document();
106 ModelAPI_ExecState aState;
107 bool aHasWarning = false;
108 for (int i = aFirst; i < aLast + 1; i++) {
109 aSubObj = aDoc->object(ModelAPI_Feature::group(), i);
110 aState = aSubObj->data()->execState();
111 if ((aState == ModelAPI_StateExecFailed) || (aState == ModelAPI_StateMustBeUpdated)) {
117 return QIcon(ModuleBase_Tools::composite(":icons/hasWarning.png", anIconString));
122 return loadIcon(anIconString);
125 ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObj);
127 GeomShapePtr aShape = aResult->shape();
129 switch(aShape->shapeType()) {
130 case GeomAPI_Shape::COMPOUND: {
131 FeaturePtr aFeature = ModelAPI_Feature::feature(theObj);
132 if (aFeature.get() && aFeature->getKind() == CollectionPlugin_Group::ID()) {
133 switch (aShape->typeOfCompoundShapes()) {
134 case GeomAPI_Shape::VERTEX:
135 return QIcon(":icons/group_vertex.png");
136 case GeomAPI_Shape::EDGE:
137 return QIcon(":icons/group_edge.png");
138 case GeomAPI_Shape::FACE:
139 return QIcon(":icons/group_face.png");
140 case GeomAPI_Shape::SOLID:
141 return QIcon(":icons/group_solid.png");
144 ResultBodyPtr aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aResult);
145 if (aBody.get() && aBody->isConnectedTopology())
146 return QIcon(":pictures/compoundofsolids.png");
147 return QIcon(":pictures/compound.png");
149 case GeomAPI_Shape::COMPSOLID: return QIcon(":pictures/compsolid.png");
150 case GeomAPI_Shape::SOLID: return QIcon(":pictures/solid.png");
151 case GeomAPI_Shape::SHELL: return QIcon(":pictures/shell.png");
152 case GeomAPI_Shape::FACE: return QIcon(":pictures/face.png");
153 case GeomAPI_Shape::WIRE: return QIcon(":pictures/wire.png");
154 case GeomAPI_Shape::EDGE: return QIcon(":pictures/edge.png");
155 case GeomAPI_Shape::VERTEX: return QIcon(":pictures/vertex.png");
162 void PartSet_IconFactory::processEvent(const std::shared_ptr<Events_Message>& theMessage)
164 if (theMessage->eventID() ==
165 Events_Loop::loop()->eventByName(Config_FeatureMessage::GUI_EVENT())) {
166 std::shared_ptr<Config_FeatureMessage> aFeatureMsg =
167 std::dynamic_pointer_cast<Config_FeatureMessage>(theMessage);
168 if (!aFeatureMsg->isInternal()) {
169 ActionInfo aFeatureInfo;
170 aFeatureInfo.initFrom(aFeatureMsg);
171 // Remember features icons
172 myIcons[QString::fromStdString(aFeatureMsg->id())] = aFeatureInfo.iconFile;