src/ModuleBase/ModuleBase_WidgetShapeSelector.cpp

   1 // File:        ModuleBase_WidgetShapeSelector.h
   2 // Created:     2 June 2014
   3 // Author:      Vitaly Smetannikov
   4
   5 #include "ModuleBase_WidgetShapeSelector.h"
   6 #include "ModuleBase_IWorkshop.h"
   7
   8 #include <Events_Loop.h>
   9 #include <ModelAPI_Events.h>
  10 #include <ModelAPI_Tools.h>
  11
  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>
  17
  18 #include <GeomAPI_Shape.h>
  19
  20 #include <TopoDS_Shape.hxx>
  21 #include <TopExp_Explorer.hxx>
  22
  23 #include <QWidget>
  24 #include <QLayout>
  25 #include <QLabel>
  26 #include <QLineEdit>
  27 #include <QToolButton>
  28 #include <QString>
  29 #include <QEvent>
  30 #include <QDockWidget>
  31
  32 #include <stdexcept>
  33
  34 typedef QMap<QString, TopAbs_ShapeEnum> ShapeTypes;
  35 static ShapeTypes MyShapeTypes;
  36
  37 TopAbs_ShapeEnum ModuleBase_WidgetShapeSelector::shapeType(const QString& theType)
  38 {
  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;
  46   }
  47   if (MyShapeTypes.contains(theType))
  48     return MyShapeTypes[theType];
  49   throw std::invalid_argument("Shape type defined in XML is not implemented!");
  50 }
  51
  52 ModuleBase_WidgetShapeSelector::ModuleBase_WidgetShapeSelector(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), myIsActive(false)
  58 {
  59   myContainer = new QWidget(theParent);
  60   QHBoxLayout* aLayout = new QHBoxLayout(myContainer);
  61
  62   aLayout->setContentsMargins(0, 0, 0, 0);
  63   QString aLabelText = QString::fromStdString(theData->widgetLabel());
  64   QString aLabelIcon = QString::fromStdString(theData->widgetIcon());
  65   myLabel = new QLabel(aLabelText, myContainer);
  66   if (!aLabelIcon.isEmpty())
  67     myLabel->setPixmap(QPixmap(aLabelIcon));
  68
  69   aLayout->addWidget(myLabel);
  70
  71   QString aToolTip = QString::fromStdString(theData->widgetTooltip());
  72   myTextLine = new QLineEdit(myContainer);
  73   myTextLine->setReadOnly(true);
  74   myTextLine->setToolTip(aToolTip);
  75   processFocus(myTextLine);
  76   myTextLine->installEventFilter(this);
  77
  78   myBasePalet = myTextLine->palette();
  79   myInactivePalet = myBasePalet;
  80   myInactivePalet.setBrush(QPalette::Base, QBrush(Qt::gray, Qt::Dense6Pattern));
  81   myTextLine->setPalette(myInactivePalet);
  82
  83   aLayout->addWidget(myTextLine, 1);
  84
  85   std::string aTypes = theData->getProperty("shape_types");
  86   myShapeTypes = QString(aTypes.c_str()).split(' ');
  87 }
  88
  89 //********************************************************************
  90 ModuleBase_WidgetShapeSelector::~ModuleBase_WidgetShapeSelector()
  91 {
  92 }
  93
  94 //********************************************************************
  95 bool ModuleBase_WidgetShapeSelector::storeValue() const
  96 {
  97   FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(mySelectedObject);
  98   if (aSelectedFeature == myFeature)  // In order to avoid selection of the same object
  99     return false;
 100
 101   DataPtr aData = myFeature->data();
 102   boost::shared_ptr<ModelAPI_AttributeReference> aRef = boost::dynamic_pointer_cast<
 103       ModelAPI_AttributeReference>(aData->attribute(attributeID()));
 104
 105   ObjectPtr aObject = aRef->value();
 106   if (!(aObject && aObject->isSame(mySelectedObject))) {
 107     aRef->setValue(mySelectedObject);
 108     updateObject(myFeature);
 109   }
 110   return true;
 111 }
 112
 113 //********************************************************************
 114 bool ModuleBase_WidgetShapeSelector::restoreValue()
 115 {
 116   DataPtr aData = myFeature->data();
 117   boost::shared_ptr<ModelAPI_AttributeReference> aRef = aData->reference(attributeID());
 118
 119   bool isBlocked = this->blockSignals(true);
 120   mySelectedObject = aRef->value();
 121   updateSelectionName();
 122
 123   this->blockSignals(isBlocked);
 124   return true;
 125 }
 126
 127 //********************************************************************
 128 QList<QWidget*> ModuleBase_WidgetShapeSelector::getControls() const
 129 {
 130   QList<QWidget*> aControls;
 131   aControls.append(myLabel);
 132   aControls.append(myTextLine);
 133   return aControls;
 134 }
 135
 136 //********************************************************************
 137 void ModuleBase_WidgetShapeSelector::onSelectionChanged()
 138 {
 139   QList<ObjectPtr> aObjects = myWorkshop->selectedObjects();
 140   if (aObjects.size() > 0) {
 141     ObjectPtr aObject = aObjects.first();
 142     if ((!mySelectedObject) && (!aObject))
 143       return;
 144     if (mySelectedObject && aObject && mySelectedObject->isSame(aObject))
 145       return;
 146
 147     // Check that the selection corresponds to selection type
 148     if (!isAccepted(aObject))
 149       return;
 150
 151     mySelectedObject = aObject;
 152     if (mySelectedObject) {
 153       updateSelectionName();
 154       raisePanel();
 155       static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_TOHIDE);
 156       ModelAPI_EventCreator::get()->sendUpdated(mySelectedObject, anEvent);
 157       Events_Loop::loop()->flush(anEvent);
 158     } else {
 159       myTextLine->setText("");
 160     }
 161     activateSelection(false);
 162     emit valuesChanged();
 163     emit focusOutWidget(this);
 164   }
 165 }
 166
 167 //********************************************************************
 168 bool ModuleBase_WidgetShapeSelector::isAccepted(const ObjectPtr theResult) const
 169 {
 170   ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(theResult);
 171   boost::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
 172   if (!aShapePtr)
 173     return false;
 174   TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
 175   if (aShape.IsNull())
 176     return false;
 177
 178   TopAbs_ShapeEnum aShapeType = aShape.ShapeType();
 179   if (aShapeType == TopAbs_COMPOUND) {
 180     foreach (QString aType, myShapeTypes) {
 181       TopExp_Explorer aEx(aShape, shapeType(aType));
 182       if (aEx.More())
 183         return true;
 184     }
 185   } else {
 186     foreach (QString aType, myShapeTypes) {
 187       if (shapeType(aType) == aShapeType)
 188         return true;
 189     }
 190   }
 191   return false;
 192 }
 193
 194 //********************************************************************
 195 void ModuleBase_WidgetShapeSelector::updateSelectionName()
 196 {
 197   if (mySelectedObject) {
 198     std::string aName = mySelectedObject->data()->name();
 199     myTextLine->setText(QString::fromStdString(aName));
 200   } else {
 201     if (myIsActive)
 202       myTextLine->setText(tr("Select an object"));
 203     else
 204       myTextLine->setText(tr("No object selected"));
 205   }
 206 }
 207
 208
 209 //********************************************************************
 210 void ModuleBase_WidgetShapeSelector::activateSelection(bool toActivate)
 211 {
 212   myIsActive = toActivate;
 213   if (myIsActive)
 214     myTextLine->setPalette(myBasePalet);
 215   else
 216     myTextLine->setPalette(myInactivePalet);
 217   updateSelectionName();
 218
 219   if (myIsActive)
 220     connect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
 221   else
 222     disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
 223 }
 224
 225 //********************************************************************
 226 void ModuleBase_WidgetShapeSelector::raisePanel() const
 227 {
 228   QWidget* aParent = myContainer->parentWidget();
 229   QWidget* aLastPanel = 0;
 230   while (!aParent->inherits("QDockWidget")) {
 231     aLastPanel = aParent;
 232     aParent = aParent->parentWidget();
 233     if (!aParent)
 234       return;
 235   }
 236   if (aParent->inherits("QDockWidget")) {
 237     QDockWidget* aTabWgt = (QDockWidget*) aParent;
 238     aTabWgt->raise();
 239   }
 240 }
 241
 242 //********************************************************************
 243 bool ModuleBase_WidgetShapeSelector::focusTo()
 244 {
 245   activateSelection(true);
 246   return true;
 247 }
 248
 249 //********************************************************************
 250 bool ModuleBase_WidgetShapeSelector::eventFilter(QObject* theObj, QEvent* theEvent)
 251 {
 252   if (theObj == myTextLine) {
 253     if (theEvent->type() == QEvent::FocusIn)
 254       activateSelection(true);
 255   }
 256   return ModuleBase_ModelWidget::eventFilter(theObj, theEvent);
 257 }