1 // File: ModuleBase_WidgetSelector.h
2 // Created: 2 June 2014
3 // Author: Vitaly Smetannikov
5 #include "ModuleBase_WidgetSelector.h"
6 #include "ModuleBase_IWorkshop.h"
8 #include <Events_Loop.h>
9 #include <ModelAPI_Events.h>
10 #include <ModelAPI_Tools.h>
12 #include <ModelAPI_Data.h>
13 #include <ModelAPI_Object.h>
14 #include <ModelAPI_Result.h>
15 #include <ModelAPI_AttributeReference.h>
16 #include <Config_WidgetAPI.h>
18 #include <GeomAPI_Shape.h>
20 #include <TopoDS_Shape.hxx>
21 #include <TopExp_Explorer.hxx>
27 #include <QToolButton>
30 #include <QDockWidget>
34 typedef QMap<QString, TopAbs_ShapeEnum> ShapeTypes;
35 static ShapeTypes MyShapeTypes;
37 TopAbs_ShapeEnum ModuleBase_WidgetSelector::shapeType(const QString& theType)
39 if (MyShapeTypes.count() == 0) {
40 MyShapeTypes["face"] = TopAbs_FACE;
41 MyShapeTypes["vertex"] = TopAbs_VERTEX;
42 MyShapeTypes["wire"] = TopAbs_WIRE;
43 MyShapeTypes["edge"] = TopAbs_EDGE;
44 MyShapeTypes["shell"] = TopAbs_SHELL;
45 MyShapeTypes["solid"] = TopAbs_SOLID;
47 if (MyShapeTypes.contains(theType))
48 return MyShapeTypes[theType];
49 throw std::invalid_argument("Shape type defined in XML is not implemented!");
52 ModuleBase_WidgetSelector::ModuleBase_WidgetSelector(QWidget* theParent,
53 ModuleBase_IWorkshop* theWorkshop,
54 const Config_WidgetAPI* theData,
55 const std::string& theParentId)
56 : ModuleBase_ModelWidget(theParent, theData, theParentId),
57 myWorkshop(theWorkshop),
58 myActivateOnStart(false)
60 myContainer = new QWidget(theParent);
61 QHBoxLayout* aLayout = new QHBoxLayout(myContainer);
63 aLayout->setContentsMargins(0, 0, 0, 0);
64 QString aLabelText = QString::fromStdString(theData->widgetLabel());
65 QString aLabelIcon = QString::fromStdString(theData->widgetIcon());
66 myLabel = new QLabel(aLabelText, myContainer);
67 myLabel->setPixmap(QPixmap(aLabelIcon));
69 aLayout->addWidget(myLabel);
71 QString aToolTip = QString::fromStdString(theData->widgetTooltip());
72 myTextLine = new QLineEdit(myContainer);
73 myTextLine->setReadOnly(true);
74 myTextLine->setToolTip(aToolTip);
75 myTextLine->installEventFilter(this);
77 aLayout->addWidget(myTextLine);
79 myActivateBtn = new QToolButton(myContainer);
80 myActivateBtn->setIcon(QIcon(":icons/hand_point.png"));
81 myActivateBtn->setCheckable(true);
82 myActivateBtn->setToolTip(tr("Activate/Deactivate selection"));
83 connect(myActivateBtn, SIGNAL(toggled(bool)), this, SLOT(activateSelection(bool)));
85 aLayout->addWidget(myActivateBtn);
87 QString aActivateTxt = QString::fromStdString(theData->getProperty("activate"));
88 if (!aActivateTxt.isNull()) {
89 myActivateOnStart = (aActivateTxt == "true");
92 std::string aTypes = theData->getProperty("shape_types");
93 myShapeTypes = QString(aTypes.c_str()).split(',');
96 //********************************************************************
97 ModuleBase_WidgetSelector::~ModuleBase_WidgetSelector()
101 //********************************************************************
102 bool ModuleBase_WidgetSelector::storeValue() const
104 FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(mySelectedObject);
105 if (aSelectedFeature == myFeature) // In order to avoid selection of the same object
108 DataPtr aData = myFeature->data();
109 boost::shared_ptr<ModelAPI_AttributeReference> aRef = boost::dynamic_pointer_cast<
110 ModelAPI_AttributeReference>(aData->attribute(attributeID()));
112 ObjectPtr aObject = aRef->value();
113 if (!(aObject && aObject->isSame(mySelectedObject))) {
114 aRef->setValue(mySelectedObject);
115 updateObject(myFeature);
120 //********************************************************************
121 bool ModuleBase_WidgetSelector::restoreValue()
123 DataPtr aData = myFeature->data();
124 boost::shared_ptr<ModelAPI_AttributeReference> aRef = aData->reference(attributeID());
126 bool isBlocked = this->blockSignals(true);
127 mySelectedObject = aRef->value();
128 updateSelectionName();
130 this->blockSignals(isBlocked);
134 //********************************************************************
135 QList<QWidget*> ModuleBase_WidgetSelector::getControls() const
137 QList<QWidget*> aControls;
138 aControls.append(myLabel);
139 aControls.append(myTextLine);
140 aControls.append(myActivateBtn);
144 //********************************************************************
145 void ModuleBase_WidgetSelector::onSelectionChanged()
147 QList<ObjectPtr> aObjects = myWorkshop->selectedObjects();
148 if (aObjects.size() > 0) {
149 ObjectPtr aObject = aObjects.first();
150 if ((!mySelectedObject) && (!aObject))
152 if (mySelectedObject && aObject && mySelectedObject->isSame(aObject))
155 // Check that the selection corresponds to selection type
156 if (!isAccepted(aObject))
159 mySelectedObject = aObject;
160 if (mySelectedObject) {
161 updateSelectionName();
162 myActivateBtn->setChecked(false);
165 myTextLine->setText("");
167 emit valuesChanged();
171 //********************************************************************
172 bool ModuleBase_WidgetSelector::isAccepted(const ObjectPtr theResult) const
174 ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(theResult);
175 boost::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
178 TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
182 TopAbs_ShapeEnum aShapeType = aShape.ShapeType();
183 if (aShapeType == TopAbs_COMPOUND) {
184 foreach (QString aType, myShapeTypes)
186 TopExp_Explorer aEx(aShape, shapeType(aType));
191 foreach (QString aType, myShapeTypes)
193 if (shapeType(aType) == aShapeType)
200 //********************************************************************
201 void ModuleBase_WidgetSelector::updateSelectionName()
203 if (mySelectedObject) {
204 std::string aName = mySelectedObject->data()->name();
206 myTextLine->setText(QString::fromStdString(aName));
208 myTextLine->setText("");
211 //********************************************************************
212 bool ModuleBase_WidgetSelector::eventFilter(QObject* theObj, QEvent* theEvent)
214 if (theObj == myTextLine) {
215 if (theEvent->type() == QEvent::Polish) {
216 myActivateBtn->setChecked(myActivateOnStart);
217 onSelectionChanged();
220 return ModuleBase_ModelWidget::eventFilter(theObj, theEvent);
223 //********************************************************************
224 void ModuleBase_WidgetSelector::enableOthersControls(bool toEnable) const
226 QWidget* aParent = myContainer->parentWidget();
227 QList<QWidget*> aChldList = aParent->findChildren<QWidget*>();
228 foreach(QWidget* aWgt, aChldList)
230 if ((aWgt != myLabel) && (aWgt != myActivateBtn) && (aWgt != myTextLine)
231 && (aWgt != myContainer))
232 aWgt->setEnabled(toEnable);
236 //********************************************************************
237 void ModuleBase_WidgetSelector::activateSelection(bool toActivate)
239 enableOthersControls(!toActivate);
240 myTextLine->setEnabled(toActivate);
243 connect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
245 disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
247 myActivateBtn->setDown(toActivate);
250 //********************************************************************
251 void ModuleBase_WidgetSelector::raisePanel() const
253 QWidget* aParent = myContainer->parentWidget();
254 QWidget* aLastPanel = 0;
255 while (!aParent->inherits("QDockWidget")) {
256 aLastPanel = aParent;
257 aParent = aParent->parentWidget();
261 if (aParent->inherits("QDockWidget")) {
262 QDockWidget* aTabWgt = (QDockWidget*) aParent;