+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
// File: ModuleBase_WidgetShapeSelector.h
// Created: 2 June 2014
// Author: Vitaly Smetannikov
#include <ModuleBase_IWorkshop.h>
#include <ModuleBase_IViewer.h>
#include <ModuleBase_Tools.h>
-#include <ModuleBase_WidgetValueFeature.h>
+#include <ModuleBase_FilterFactory.h>
+#include <ModuleBase_Filter.h>
#include <Config_WidgetAPI.h>
#include <Events_Loop.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_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 <QEvent>
#include <QDockWidget>
#include <QApplication>
+#include <QFormLayout>
#include <TopExp_Explorer.hxx>
#include <TopoDS_Shape.hxx>
#include <list>
#include <string>
-typedef QMap<QString, TopAbs_ShapeEnum> ShapeTypes;
-static ShapeTypes MyShapeTypes;
-
-TopAbs_ShapeEnum ModuleBase_WidgetShapeSelector::shapeType(const QString& theType)
-{
- if (MyShapeTypes.count() == 0) {
- MyShapeTypes["face"] = TopAbs_FACE;
- MyShapeTypes["faces"] = TopAbs_FACE;
- MyShapeTypes["vertex"] = TopAbs_VERTEX;
- MyShapeTypes["vertices"] = TopAbs_VERTEX;
- MyShapeTypes["wire"] = TopAbs_WIRE;
- MyShapeTypes["edge"] = TopAbs_EDGE;
- MyShapeTypes["edges"] = TopAbs_EDGE;
- MyShapeTypes["shell"] = TopAbs_SHELL;
- MyShapeTypes["solid"] = TopAbs_SOLID;
- MyShapeTypes["solids"] = TopAbs_SOLID;
- }
- QString aType = theType.toLower();
- if (MyShapeTypes.contains(aType))
- return MyShapeTypes[aType];
- Events_Error::send("Shape type defined in XML is not implemented!");
- return TopAbs_SHAPE;
-}
ModuleBase_WidgetShapeSelector::ModuleBase_WidgetShapeSelector(QWidget* theParent,
ModuleBase_IWorkshop* theWorkshop,
const Config_WidgetAPI* theData,
const std::string& theParentId)
- : ModuleBase_ModelWidget(theParent, theData, theParentId),
- myWorkshop(theWorkshop), myIsActive(false), myUseSubShapes(false)
+ : 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);
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);
- aLayout->addWidget(myTextLine, 1);
+ aLayout->addRow(myLabel, myTextLine);
std::string aTypes = theData->getProperty("shape_types");
myShapeTypes = QString(aTypes.c_str()).split(' ', QString::SkipEmptyParts);
-
- std::string aObjTypes = theData->getProperty("object_types");
- myObjectTypes = QString(aObjTypes.c_str()).split(' ', QString::SkipEmptyParts);
-
- myUseSubShapes = theData->getBooleanAttribute("use_subshapes", false);
}
//********************************************************************
ModuleBase_WidgetShapeSelector::~ModuleBase_WidgetShapeSelector()
{
activateSelection(false);
+ activateFilters(myWorkshop, false);
+}
+
+//********************************************************************
+bool ModuleBase_WidgetShapeSelector::storeValueCustom() const
+{
+ // the value is stored on the selection changed signal processing
+ return true;
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::storeValue() const
+bool ModuleBase_WidgetShapeSelector::setObject(ObjectPtr theSelectedObject,
+ GeomShapePtr theShape)
{
- FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(mySelectedObject);
+ 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();
- if (myUseSubShapes) {
- ResultPtr aBody = std::dynamic_pointer_cast<ModelAPI_Result>(mySelectedObject);
- if (aBody) {
- AttributePtr aAttr = aData->attribute(attributeID());
-
- AttributeSelectionPtr aSelectAttr =
- std::dynamic_pointer_cast<ModelAPI_AttributeSelection>(aAttr);
- if (aSelectAttr)
- aSelectAttr->setValue(aBody, myShape);
- updateObject(myFeature);
- return true;
+ AttributeReferencePtr aRef = aData->reference(attributeID());
+ if (aRef) {
+ ObjectPtr aObject = aRef->value();
+ if (!(aObject && aObject->isSame(theSelectedObject))) {
+ aRef->setValue(theSelectedObject);
+ isChanged = true;
}
} else {
- AttributeReferencePtr aRef = aData->reference(attributeID());
- if (aRef) {
- ObjectPtr aObject = aRef->value();
- if (!(aObject && aObject->isSame(mySelectedObject))) {
- aRef->setValue(mySelectedObject);
- updateObject(myFeature);
- return true;
+ AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
+ if (aRefAttr) {
+ ObjectPtr aObject = aRefAttr->object();
+ if (!(aObject && aObject->isSame(theSelectedObject))) {
+ aRefAttr->setObject(theSelectedObject);
+ isChanged = true;
}
} else {
- AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
- if (aRefAttr) {
- ObjectPtr aObject = aRefAttr->object();
- if (!(aObject && aObject->isSame(mySelectedObject))) {
- aRefAttr->setObject(mySelectedObject);
- updateObject(myFeature);
- return true;
- }
+ 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 false;
+ return isChanged;
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::restoreValue()
+QList<ModuleBase_ViewerPrs> ModuleBase_WidgetShapeSelector::getCurrentSelection() const
{
- DataPtr aData = myFeature->data();
- bool isBlocked = this->blockSignals(true);
- if (myUseSubShapes) {
- AttributeSelectionPtr aSelect = aData->selection(attributeID());
- if (aSelect) {
- mySelectedObject = aSelect->context();
- myShape = aSelect->value();
- }
- } else {
- AttributeReferencePtr aRef = aData->reference(attributeID());
- if (aRef)
- mySelectedObject = aRef->value();
- else {
- AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
- if (aRefAttr)
- mySelectedObject = aRefAttr->object();
+ 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);
}
- updateSelectionName();
-
- this->blockSignals(isBlocked);
- return true;
+ return aSelected;
}
//********************************************************************
-QList<QWidget*> ModuleBase_WidgetShapeSelector::getControls() const
+void ModuleBase_WidgetShapeSelector::clearAttribute()
{
- QList<QWidget*> aControls;
- aControls.append(myTextLine);
- return aControls;
+ DataPtr aData = myFeature->data();
+ 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());
+ }
}
//********************************************************************
-void ModuleBase_WidgetShapeSelector::onSelectionChanged()
+bool ModuleBase_WidgetShapeSelector::restoreValue()
{
- QObjectPtrList aObjects = myWorkshop->selection()->selectedPresentations();
- if (aObjects.size() > 0) {
- ObjectPtr aObject = aObjects.first();
- if ((!mySelectedObject) && (!aObject))
- return;
-
- // Check that the selected object is result (others can not be accepted)
- ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aObject);
- if (!aRes)
- return;
-
- 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;
- }
- }
- // 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;
-
- // Check that the result has a shape
- GeomShapePtr aShape = ModelAPI_Tools::shape(aRes);
- if (!aShape)
- return;
-
- /// Check that object has acceptable type
- if (!acceptObjectType(aObject))
- return;
-
- // Get sub-shapes from local selection
- if (myUseSubShapes) {
- NCollection_List<TopoDS_Shape> aShapeList;
- std::list<ObjectPtr> aOwners;
- myWorkshop->selection()->selectedShapes(aShapeList, aOwners);
- if (aShapeList.Extent() > 0) {
- aShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape());
- aShape->setImpl(new TopoDS_Shape(aShapeList.First()));
- }
- }
+ bool isBlocked = this->blockSignals(true);
+ updateSelectionName();
+ this->blockSignals(isBlocked);
- // Check that the selection corresponds to selection type
- if (myUseSubShapes) {
- if (!acceptSubShape(aShape))
- return;
- } else {
- if (!acceptObjectShape(aObject))
- return;
- }
- setObject(aObject, aShape);
- emit focusOutWidget(this);
- }
+ return true;
}
//********************************************************************
-void ModuleBase_WidgetShapeSelector::setObject(ObjectPtr theObj, std::shared_ptr<GeomAPI_Shape> theShape)
+QList<QWidget*> ModuleBase_WidgetShapeSelector::getControls() const
{
- mySelectedObject = theObj;
- myShape = theShape;
- if (mySelectedObject) {
- raisePanel();
- }
- updateSelectionName();
- emit valuesChanged();
+ QList<QWidget*> aControls;
+ aControls.append(myTextLine);
+ return aControls;
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::acceptObjectShape(const ObjectPtr theResult) const
+void ModuleBase_WidgetShapeSelector::onSelectionChanged()
{
- ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theResult);
-
- // Check that the shape of necessary type
- std::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;
+ // 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;
}
}
- return false;
+ // 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);
}
//********************************************************************
{
TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
foreach (QString aType, myShapeTypes) {
- if (aShape.ShapeType() == shapeType(aType))
+ if (aShape.ShapeType() == ModuleBase_Tools::shapeType(aType))
return true;
}
return false;
}
//********************************************************************
-bool ModuleBase_WidgetShapeSelector::acceptObjectType(const ObjectPtr theObject) const
+GeomShapePtr ModuleBase_WidgetShapeSelector::getShape() const
{
- // Definition of types is not obligatory. If types are not defined then
- // it means that accepted any type
- if (myObjectTypes.isEmpty())
- return true;
-
- foreach (QString aType, myObjectTypes) {
- if (aType.toLower() == "construction") {
- ResultConstructionPtr aConstr =
- std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(theObject);
- return (aConstr != NULL);
- } // ToDo: Process other types of objects
- }
- // Object type is defined but not found
- return false;
-}
+ GeomShapePtr aShape;
+ DataPtr aData = myFeature->data();
+ if (aData.get() == NULL)
+ return aShape;
+ AttributeSelectionPtr aSelect = aData->selection(attributeID());
+ if (aSelect)
+ aShape = aSelect->value();
+
+ return aShape;
+}
//********************************************************************
void ModuleBase_WidgetShapeSelector::updateSelectionName()
{
- if (mySelectedObject) {
- std::string aName = mySelectedObject->data()->name();
- myTextLine->setText(QString::fromStdString(aName));
- } else {
- if (myIsActive) {
- myTextLine->setText("");
+ 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)
{
- if (myIsActive == toActivate)
- return;
- myIsActive = toActivate;
updateSelectionName();
- if (myIsActive) {
- connect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
- if (myUseSubShapes) {
-
- QIntList aList;
- foreach (QString aType, myShapeTypes)
- aList.append(shapeType(aType));
- myWorkshop->activateSubShapesSelection(aList);
- if (!myObjectTypes.isEmpty()) {
- myObjTypeFilter = new ModuleBase_ObjectTypesFilter(myWorkshop, myObjectTypes);
- myWorkshop->viewer()->clearSelectionFilters();
- myWorkshop->viewer()->addSelectionFilter(myObjTypeFilter);
- }
+ if (toActivate) {
+ QIntList aList;
+ foreach (QString aType, myShapeTypes) {
+ aList.append(ModuleBase_Tools::shapeType(aType));
}
+ myWorkshop->activateSubShapesSelection(aList);
} else {
- disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
- if (myUseSubShapes) {
- if (!myObjTypeFilter.IsNull()) {
- myWorkshop->viewer()->removeSelectionFilter(myObjTypeFilter);
- myObjTypeFilter.Nullify();
- }
- myWorkshop->deactivateSubShapesSelection();
- }
+ 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::setValue(ModuleBase_WidgetValue* theValue)
+void ModuleBase_WidgetShapeSelector::activateCustom()
{
- if (theValue) {
- ModuleBase_WidgetValueFeature* aFeatureValue =
- dynamic_cast<ModuleBase_WidgetValueFeature*>(theValue);
- if (aFeatureValue && aFeatureValue->object()) {
- ObjectPtr aObject = aFeatureValue->object();
- if (acceptObjectShape(aObject)) {
- setObject(aObject);
- return 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();
}
- return false;
}
//********************************************************************
-void ModuleBase_WidgetShapeSelector::activate()
+void ModuleBase_WidgetShapeSelector::restoreAttributeValue(bool theValid)
{
- activateSelection(true);
+ 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()));
}