+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
// File: ModuleBase_WidgetShapeSelector.h
// Created: 2 June 2014
// Author: Vitaly Smetannikov
-#include "ModuleBase_WidgetShapeSelector.h"
-#include "ModuleBase_IWorkshop.h"
+#include <ModuleBase_WidgetShapeSelector.h>
+#include <ModuleBase_Definitions.h>
+#include <ModuleBase_ISelection.h>
+#include <ModuleBase_IWorkshop.h>
+#include <ModuleBase_IViewer.h>
+#include <ModuleBase_Tools.h>
+#include <ModuleBase_FilterFactory.h>
+#include <ModuleBase_Filter.h>
+#include <Config_WidgetAPI.h>
#include <Events_Loop.h>
-#include <ModelAPI_Events.h>
-#include <ModelAPI_Tools.h>
+#include <Events_Message.h>
+#include <GeomAPI_Interface.h>
+#include <GeomAPI_Shape.h>
+#include <GeomValidators_Tools.h>
+#include <ModelAPI_AttributeReference.h>
#include <ModelAPI_Data.h>
-#include <ModelAPI_Object.h>
+#include <ModelAPI_Document.h>
+#include <ModelAPI_Events.h>
+#include <ModelAPI_Feature.h>
#include <ModelAPI_Result.h>
+#include <ModelAPI_ResultConstruction.h>
#include <ModelAPI_AttributeReference.h>
+#include <ModelAPI_AttributeSelection.h>
+#include <ModelAPI_Session.h>
+#include <ModelAPI_Tools.h>
+#include <ModelAPI_ResultBody.h>
+#include <ModelAPI_AttributeRefAttr.h>
+#include <ModelAPI_Validator.h>
+#include <ModelAPI_AttributeValidator.h>
+#include <ModelAPI_ShapeValidator.h>
+
#include <Config_WidgetAPI.h>
+#include <Events_Error.h>
#include <GeomAPI_Shape.h>
+#include <SelectMgr_ListIteratorOfListOfFilter.hxx>
#include <TopoDS_Shape.hxx>
#include <TopExp_Explorer.hxx>
#include <QString>
#include <QEvent>
#include <QDockWidget>
+#include <QApplication>
+#include <QFormLayout>
-#include <stdexcept>
+#include <TopExp_Explorer.hxx>
+#include <TopoDS_Shape.hxx>
-typedef QMap<QString, TopAbs_ShapeEnum> ShapeTypes;
-static ShapeTypes MyShapeTypes;
+#include <memory>
+
+#include <list>
+#include <string>
-TopAbs_ShapeEnum ModuleBase_WidgetShapeSelector::shapeType(const QString& theType)
-{
- if (MyShapeTypes.count() == 0) {
- MyShapeTypes["face"] = TopAbs_FACE;
- MyShapeTypes["vertex"] = TopAbs_VERTEX;
- MyShapeTypes["wire"] = TopAbs_WIRE;
- MyShapeTypes["edge"] = TopAbs_EDGE;
- MyShapeTypes["shell"] = TopAbs_SHELL;
- MyShapeTypes["solid"] = TopAbs_SOLID;
- }
- if (MyShapeTypes.contains(theType))
- return MyShapeTypes[theType];
- throw std::invalid_argument("Shape type defined in XML is not implemented!");
-}
ModuleBase_WidgetShapeSelector::ModuleBase_WidgetShapeSelector(QWidget* theParent,
ModuleBase_IWorkshop* theWorkshop,
const Config_WidgetAPI* theData,
const std::string& theParentId)
- : ModuleBase_ModelWidget(theParent, theData, theParentId),
+ : ModuleBase_WidgetValidated(theParent, theData, theParentId),
myWorkshop(theWorkshop)
{
- myContainer = new QWidget(theParent);
- QHBoxLayout* aLayout = new QHBoxLayout(myContainer);
+ QFormLayout* aLayout = new QFormLayout(this);
+ ModuleBase_Tools::adjustMargins(aLayout);
- aLayout->setContentsMargins(0, 0, 0, 0);
QString aLabelText = QString::fromStdString(theData->widgetLabel());
QString aLabelIcon = QString::fromStdString(theData->widgetIcon());
- myLabel = new QLabel(aLabelText, myContainer);
+ myLabel = new QLabel(aLabelText, this);
if (!aLabelIcon.isEmpty())
myLabel->setPixmap(QPixmap(aLabelIcon));
- aLayout->addWidget(myLabel);
QString aToolTip = QString::fromStdString(theData->widgetTooltip());
- myTextLine = new QLineEdit(myContainer);
+ myTextLine = new QLineEdit(this);
myTextLine->setReadOnly(true);
myTextLine->setToolTip(aToolTip);
myTextLine->installEventFilter(this);
- myBasePalet = myTextLine->palette();
- myInactivePalet = myBasePalet;
- myInactivePalet.setBrush(QPalette::Base, QBrush(Qt::gray, Qt::Dense6Pattern));
- myTextLine->setPalette(myInactivePalet);
-
- aLayout->addWidget(myTextLine, 1);
-
- myActivateBtn = new QToolButton(myContainer);
- myActivateBtn->setIcon(QIcon(":icons/hand_point.png"));
- myActivateBtn->setCheckable(true);
- myActivateBtn->setToolTip(tr("Activate/Deactivate selection"));
- connect(myActivateBtn, SIGNAL(toggled(bool)), this, SLOT(activateSelection(bool)));
-
- aLayout->addWidget(myActivateBtn);
+ aLayout->addRow(myLabel, myTextLine);
std::string aTypes = theData->getProperty("shape_types");
- myShapeTypes = QString(aTypes.c_str()).split(' ');
+ myShapeTypes = QString(aTypes.c_str()).split(' ', QString::SkipEmptyParts);
}
//********************************************************************
ModuleBase_WidgetShapeSelector::~ModuleBase_WidgetShapeSelector()
{
+ activateSelection(false);
+ activateFilters(myWorkshop, false);
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::storeValue() const
+bool ModuleBase_WidgetShapeSelector::storeValueCustom() const
{
- FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(mySelectedObject);
+ // the value is stored on the selection changed signal processing
+ return true;
+}
+
+//********************************************************************
+bool ModuleBase_WidgetShapeSelector::setObject(ObjectPtr theSelectedObject,
+ GeomShapePtr theShape)
+{
+ bool isChanged = false;
+ FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(theSelectedObject);
if (aSelectedFeature == myFeature) // In order to avoid selection of the same object
- return false;
+ return isChanged;
DataPtr aData = myFeature->data();
- boost::shared_ptr<ModelAPI_AttributeReference> aRef = boost::dynamic_pointer_cast<
- ModelAPI_AttributeReference>(aData->attribute(attributeID()));
+ AttributeReferencePtr aRef = aData->reference(attributeID());
+ if (aRef) {
+ ObjectPtr aObject = aRef->value();
+ if (!(aObject && aObject->isSame(theSelectedObject))) {
+ aRef->setValue(theSelectedObject);
+ isChanged = true;
+ }
+ } else {
+ AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
+ if (aRefAttr) {
+ ObjectPtr aObject = aRefAttr->object();
+ if (!(aObject && aObject->isSame(theSelectedObject))) {
+ aRefAttr->setObject(theSelectedObject);
+ isChanged = true;
+ }
+ } else {
+ AttributeSelectionPtr aSelectAttr = aData->selection(attributeID());
+ ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theSelectedObject);
+ if (aSelectAttr.get() != NULL) {
+ aSelectAttr->setValue(aResult, theShape);
+ isChanged = true;
+ }
+ }
+ }
+ return isChanged;
+}
- ObjectPtr aObject = aRef->value();
- if (!(aObject && aObject->isSame(mySelectedObject))) {
- aRef->setValue(mySelectedObject);
- updateObject(myFeature);
+//********************************************************************
+QList<ModuleBase_ViewerPrs> ModuleBase_WidgetShapeSelector::getCurrentSelection() const
+{
+ QList<ModuleBase_ViewerPrs> aSelected;
+ if(myFeature) {
+ DataPtr aData = myFeature->data();
+ AttributePtr anAttribute = myFeature->attribute(attributeID());
+
+ ObjectPtr anObject = GeomValidators_Tools::getObject(anAttribute);
+ TopoDS_Shape aShape;
+ std::shared_ptr<GeomAPI_Shape> aShapePtr = getShape();
+ if (aShapePtr.get()) {
+ aShape = aShapePtr->impl<TopoDS_Shape>();
+ }
+ ModuleBase_ViewerPrs aPrs(anObject, aShape, NULL);
+ aSelected.append(aPrs);
}
- return true;
+ return aSelected;
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::restoreValue()
+void ModuleBase_WidgetShapeSelector::clearAttribute()
{
DataPtr aData = myFeature->data();
- boost::shared_ptr<ModelAPI_AttributeReference> aRef = aData->reference(attributeID());
+ AttributeSelectionPtr aSelect = aData->selection(attributeID());
+ if (aSelect) {
+ aSelect->setValue(ResultPtr(), std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape()));
+ return;
+ }
+ AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
+ if (aRefAttr) {
+ aRefAttr->setObject(ObjectPtr());
+ return;
+ }
+ AttributeReferencePtr aRef = aData->reference(attributeID());
+ if (aRef) {
+ aRef->setObject(ObjectPtr());
+ }
+}
+//********************************************************************
+bool ModuleBase_WidgetShapeSelector::restoreValue()
+{
bool isBlocked = this->blockSignals(true);
- mySelectedObject = aRef->value();
updateSelectionName();
-
this->blockSignals(isBlocked);
+
return true;
}
QList<QWidget*> ModuleBase_WidgetShapeSelector::getControls() const
{
QList<QWidget*> aControls;
- aControls.append(myLabel);
aControls.append(myTextLine);
- aControls.append(myActivateBtn);
return aControls;
}
//********************************************************************
void ModuleBase_WidgetShapeSelector::onSelectionChanged()
{
- QList<ObjectPtr> aObjects = myWorkshop->selectedObjects();
- if (aObjects.size() > 0) {
- ObjectPtr aObject = aObjects.first();
- if ((!mySelectedObject) && (!aObject))
- return;
- if (mySelectedObject && aObject && mySelectedObject->isSame(aObject))
- return;
-
- // Check that the selection corresponds to selection type
- if (!isAccepted(aObject))
- return;
-
- mySelectedObject = aObject;
- if (mySelectedObject) {
- updateSelectionName();
- myActivateBtn->setChecked(false);
- raisePanel();
- } else {
- myTextLine->setText("");
+ // In order to make reselection possible, set empty object and shape should be done
+ setObject(ObjectPtr(), std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape()));
+
+ bool aHasObject = false;
+ QList<ModuleBase_ViewerPrs> aSelectedPrs = getSelectedEntitiesOrObjects(myWorkshop->selection());
+ if (!aSelectedPrs.empty()) {
+ ModuleBase_ViewerPrs aPrs = aSelectedPrs.first();
+ if (!aPrs.isEmpty() && isValidSelection(aPrs)) {
+ setSelectionCustom(aPrs);
+ aHasObject = true;
}
- emit valuesChanged();
- emit focusOutWidget(this);
}
+ // the updateObject method should be called to flush the updated sigal. The workshop listens it,
+ // calls validators for the feature and, as a result, updates the Apply button state.
+ updateObject(myFeature);
+ // the widget loses the focus only if the selected object is set
+ if (aHasObject)
+ emit focusOutWidget(this);
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::isAccepted(const ObjectPtr theResult) const
+bool ModuleBase_WidgetShapeSelector::acceptSubShape(std::shared_ptr<GeomAPI_Shape> theShape) const
{
- ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(theResult);
- boost::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
- if (!aShapePtr)
- return false;
- TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
- if (aShape.IsNull())
- return false;
-
- TopAbs_ShapeEnum aShapeType = aShape.ShapeType();
- if (aShapeType == TopAbs_COMPOUND) {
- foreach (QString aType, myShapeTypes) {
- TopExp_Explorer aEx(aShape, shapeType(aType));
- if (aEx.More())
- return true;
- }
- } else {
- foreach (QString aType, myShapeTypes) {
- if (shapeType(aType) == aShapeType)
- return true;
- }
+ TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
+ foreach (QString aType, myShapeTypes) {
+ if (aShape.ShapeType() == ModuleBase_Tools::shapeType(aType))
+ return true;
}
return false;
}
//********************************************************************
-void ModuleBase_WidgetShapeSelector::updateSelectionName()
+GeomShapePtr ModuleBase_WidgetShapeSelector::getShape() const
{
- if (mySelectedObject) {
- std::string aName = mySelectedObject->data()->name();
+ GeomShapePtr aShape;
+ DataPtr aData = myFeature->data();
+ if (aData.get() == NULL)
+ return aShape;
- myTextLine->setText(QString::fromStdString(aName));
- } else
- myTextLine->setText("");
+ AttributeSelectionPtr aSelect = aData->selection(attributeID());
+ if (aSelect)
+ aShape = aSelect->value();
+
+ return aShape;
}
//********************************************************************
-void ModuleBase_WidgetShapeSelector::enableOthersControls(bool toEnable) const
+void ModuleBase_WidgetShapeSelector::updateSelectionName()
{
- QWidget* aParent = myContainer->parentWidget();
- QList<QWidget*> aChldList = aParent->findChildren<QWidget*>();
- foreach(QWidget* aWgt, aChldList)
- {
- if ((aWgt != myLabel) && (aWgt != myActivateBtn) && (aWgt != myTextLine)
- && (aWgt != myContainer))
- aWgt->setEnabled(toEnable);
+ DataPtr aData = myFeature->data();
+ if (aData.get() == NULL)
+ return;
+
+ bool isNameUpdated = false;
+ AttributeSelectionPtr aSelect = aData->selection(attributeID());
+ if (aSelect) {
+ myTextLine->setText(QString::fromStdString(aSelect->namingName()));
+ isNameUpdated = true;
+ }
+ if (!isNameUpdated) {
+ ObjectPtr anObject = GeomValidators_Tools::getObject(myFeature->attribute(attributeID()));
+ if (anObject.get() != NULL) {
+ std::string aName = anObject->data()->name();
+ myTextLine->setText(QString::fromStdString(aName));
+ } else {
+ AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
+ if (aRefAttr && aRefAttr->attr().get() != NULL) {
+ //myIsObject = aRefAttr->isObject();
+ AttributePtr anAttr = aRefAttr->attr();
+ if (anAttr.get() != NULL) {
+ std::stringstream aName;
+ aName <<anAttr->owner()->data()->name()<<"/"<<anAttr->id();
+ myTextLine->setText(QString::fromStdString(aName.str()));
+ }
+ }
+ else {
+ myTextLine->setText("");
+ }
+ }
}
}
+
//********************************************************************
void ModuleBase_WidgetShapeSelector::activateSelection(bool toActivate)
{
- enableOthersControls(!toActivate);
- //myTextLine->setEnabled(toActivate);
- if (toActivate)
- myTextLine->setPalette(myBasePalet);
- else
- myTextLine->setPalette(myInactivePalet);
-
- if (toActivate)
- connect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
- else
- disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
-
- myActivateBtn->setDown(toActivate);
+ updateSelectionName();
+
+ if (toActivate) {
+ QIntList aList;
+ foreach (QString aType, myShapeTypes) {
+ aList.append(ModuleBase_Tools::shapeType(aType));
+ }
+ myWorkshop->activateSubShapesSelection(aList);
+ } else {
+ myWorkshop->deactivateSubShapesSelection();
+ }
}
//********************************************************************
void ModuleBase_WidgetShapeSelector::raisePanel() const
{
- QWidget* aParent = myContainer->parentWidget();
+ QWidget* aParent = this->parentWidget();
QWidget* aLastPanel = 0;
while (!aParent->inherits("QDockWidget")) {
aLastPanel = aParent;
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::focusTo()
+void ModuleBase_WidgetShapeSelector::activateCustom()
{
- myActivateBtn->setChecked(true);
+ connect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
+ activateSelection(true);
+
+ // Restore selection in the viewer by the attribute selection list
+ myWorkshop->setSelected(getCurrentSelection());
+
+ activateFilters(myWorkshop, true);
+}
+
+//********************************************************************
+void ModuleBase_WidgetShapeSelector::storeAttributeValue()
+{
+ DataPtr aData = myFeature->data();
+ AttributePtr anAttribute = myFeature->attribute(attributeID());
+
+ myObject = GeomValidators_Tools::getObject(anAttribute);
+ myShape = getShape();
+ myRefAttribute = AttributePtr();
+ myIsObject = false;
+ AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
+ if (aRefAttr) {
+ myIsObject = aRefAttr->isObject();
+ myRefAttribute = aRefAttr->attr();
+ }
+}
+
+//********************************************************************
+void ModuleBase_WidgetShapeSelector::restoreAttributeValue(bool theValid)
+{
+ DataPtr aData = myFeature->data();
+ AttributePtr anAttribute = myFeature->attribute(attributeID());
+
+ setObject(myObject, myShape);
+ AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
+ if (aRefAttr) {
+ if (!myIsObject)
+ aRefAttr->setAttr(myRefAttribute);
+ }
+}
+
+//********************************************************************
+bool ModuleBase_WidgetShapeSelector::setSelectionCustom(const ModuleBase_ViewerPrs& thePrs)
+{
+ bool isDone = false;
+
+ // It should be checked by corresponded validator
+ ObjectPtr aObject = thePrs.object();
+ ObjectPtr aCurrentObject = GeomValidators_Tools::getObject(myFeature->attribute(attributeID()));
+ /*
+ if ((!aCurrentObject) && (!aObject))
+ return false;*/
+
+ // It should be checked by corresponded validator
+ // Check that the selected object is result (others can not be accepted)
+ ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aObject);
+ if (!aRes)
+ return false;
+ /*if (myFeature) {
+ // We can not select a result of our feature
+ const std::list<std::shared_ptr<ModelAPI_Result>>& aResList = myFeature->results();
+ std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aIt;
+ for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
+ if ((*aIt) == aRes)
+ return false;
+ }
+ }
+ */
+ // It should be checked by corresponded validator
+ /*
+ // Check that object belongs to active document or PartSet
+ DocumentPtr aDoc = aRes->document();
+ SessionPtr aMgr = ModelAPI_Session::get();
+ if (!(aDoc == aMgr->activeDocument()) && !(aDoc == aMgr->moduleDocument()))
+ return false;*/
+
+ // It should be checked by corresponded validator
+ // Check that the result has a shape
+ GeomShapePtr aShape = ModelAPI_Tools::shape(aRes);
+ if (!aShape)
+ return false;
+
+ // Get sub-shapes from local selection
+ if (!thePrs.shape().IsNull()) {
+ aShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape());
+ aShape->setImpl(new TopoDS_Shape(thePrs.shape()));
+ }
+ // Check that the selection corresponds to selection type
+ if (!acceptSubShape(aShape))
+ return false;
+
+ setObject(aObject, aShape);
return true;
}
+
+//********************************************************************
+void ModuleBase_WidgetShapeSelector::deactivate()
+{
+ activateSelection(false);
+ activateFilters(myWorkshop, false);
+ disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
+}