X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FPartSet%2FPartSet_Module.cpp;h=03b29dd75a50a4cc255b261bb270f8bd7b171b29;hb=013eda84dcea4a427d406c5c75ba7870f4124b7f;hp=860ca9e37bef4e2c5864635fd8e7acbd26a0109f;hpb=3031dbf1abc9330fc9dcffeecb5813e7fe71e87b;p=modules%2Fshaper.git diff --git a/src/PartSet/PartSet_Module.cpp b/src/PartSet/PartSet_Module.cpp old mode 100644 new mode 100755 index 860ca9e37..03b29dd75 --- a/src/PartSet/PartSet_Module.cpp +++ b/src/PartSet/PartSet_Module.cpp @@ -10,23 +10,35 @@ #include "PartSet_WidgetPoint2dAngle.h" #include "PartSet_WidgetMultiSelector.h" #include "PartSet_WidgetEditor.h" +#include "PartSet_WidgetFileSelector.h" +#include "PartSet_WidgetSketchCreator.h" #include "PartSet_SketcherMgr.h" #include "PartSet_MenuMgr.h" +#include "PartSet_CustomPrs.h" +#include "PartSet_IconFactory.h" + +#include "PartSet_Filters.h" +#include "PartSet_FilterInfinite.h" #include #include +#include #include #include #include #include #include +#include #include #include #include #include #include +#include +#include +#include #include @@ -35,6 +47,7 @@ #include #include #include +#include #include #include @@ -48,6 +61,8 @@ #include #include #include +#include +#include #include #include @@ -55,6 +70,7 @@ #include #include #include +#include #include #include #include @@ -87,12 +103,12 @@ #include #include +#include + #ifdef _DEBUG #include #endif - - /*!Create and return new instance of XGUI_Module*/ extern "C" PARTSET_EXPORT ModuleBase_IModule* createModule(ModuleBase_IWorkshop* theWshop) { @@ -103,6 +119,8 @@ PartSet_Module::PartSet_Module(ModuleBase_IWorkshop* theWshop) : ModuleBase_IModule(theWshop), myRestartingMode(RM_None), myVisualLayerId(0) { + new PartSet_IconFactory(); + mySketchMgr = new PartSet_SketcherMgr(this); myDataModel = new PartSet_DocumentDataModel(this); @@ -119,17 +137,48 @@ PartSet_Module::PartSet_Module(ModuleBase_IWorkshop* theWshop) this, SLOT(onKeyRelease(ModuleBase_IViewWindow*, QKeyEvent*))); connect(aViewer, SIGNAL(viewTransformed(int)), SLOT(onViewTransformed(int))); + connect(aViewer, SIGNAL(viewCreated(ModuleBase_IViewWindow*)), + SLOT(onViewCreated(ModuleBase_IViewWindow*))); myMenuMgr = new PartSet_MenuMgr(this); + myCustomPrs = new PartSet_CustomPrs(theWshop); Events_Loop* aLoop = Events_Loop::loop(); aLoop->registerListener(this, Events_Loop::eventByName(EVENT_DOCUMENT_CHANGED)); + + mySelectionFilters.Append(new PartSet_GlobalFilter(myWorkshop)); + mySelectionFilters.Append(new PartSet_FilterInfinite(myWorkshop)); } PartSet_Module::~PartSet_Module() { - if (!myDocumentShapeFilter.IsNull()) - myDocumentShapeFilter.Nullify(); + SelectMgr_ListIteratorOfListOfFilter aIt(mySelectionFilters); + for (; aIt.More(); aIt.Next()) { + Handle(SelectMgr_Filter) aFilter = aIt.Value(); + if (!aFilter.IsNull()) + aFilter.Nullify(); + } + delete myCustomPrs; +} + +void PartSet_Module::activateSelectionFilters() +{ + SelectMgr_ListIteratorOfListOfFilter aIt(mySelectionFilters); + for (; aIt.More(); aIt.Next()) { + Handle(SelectMgr_Filter) aFilter = aIt.Value(); + if (!aFilter.IsNull()) + myWorkshop->viewer()->addSelectionFilter(aFilter); + } +} + +void PartSet_Module::deactivateSelectionFilters() +{ + SelectMgr_ListIteratorOfListOfFilter aIt(mySelectionFilters); + for (; aIt.More(); aIt.Next()) { + Handle(SelectMgr_Filter) aFilter = aIt.Value(); + if (!aFilter.IsNull()) + myWorkshop->viewer()->removeSelectionFilter(aFilter); + } } void PartSet_Module::registerValidators() @@ -147,21 +196,33 @@ void PartSet_Module::registerValidators() aFactory->registerValidator("PartSet_HVDirSelection", new PartSet_HVDirSelection); aFactory->registerValidator("PartSet_TangentSelection", new PartSet_TangentSelection); aFactory->registerValidator("PartSet_FilletSelection", new PartSet_FilletSelection); + aFactory->registerValidator("PartSet_AngleSelection", new PartSet_AngleSelection); aFactory->registerValidator("PartSet_DifferentObjects", new PartSet_DifferentObjectsValidator); aFactory->registerValidator("PartSet_DifferentShapes", new ModelAPI_ShapeValidator); + aFactory->registerValidator("PartSet_CoincidentAttr", new PartSet_CoincidentAttr); + aFactory->registerValidator("GeomValidators_ShapeType", new GeomValidators_ShapeType); aFactory->registerValidator("GeomValidators_Face", new GeomValidators_Face); aFactory->registerValidator("GeomValidators_ConstructionComposite", new GeomValidators_ConstructionComposite); + aFactory->registerValidator("GeomValidators_ZeroOffset", + new GeomValidators_ZeroOffset); + + aFactory->registerValidator("GeomValidators_BooleanArguments", + new GeomValidators_BooleanArguments); + aFactory->registerValidator("PartSet_SketchEntityValidator", new PartSet_SketchEntityValidator); aFactory->registerValidator("PartSet_SameTypeAttr", new PartSet_SameTypeAttrValidator); + + aFactory->registerValidator("GeomValidators_Different", + new GeomValidators_Different); } void PartSet_Module::registerFilters() @@ -179,7 +240,7 @@ void PartSet_Module::registerProperties() Config_Prop::Integer, SKETCH_WIDTH); } -void PartSet_Module::operationCommitted(ModuleBase_Operation* theOperation) +void PartSet_Module::onOperationCommitted(ModuleBase_Operation* theOperation) { if (PartSet_SketcherMgr::isNestedSketchOperation(theOperation)) { mySketchMgr->commitNestedSketch(theOperation); @@ -190,9 +251,9 @@ void PartSet_Module::operationCommitted(ModuleBase_Operation* theOperation) // the selection is cleared after commit the create operation // in order to do not use the same selected objects in the restarted operation // for common behaviour, the selection is cleared even if the operation is not restarted - Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext(); - if (!aContext.IsNull()) - aContext->ClearSelected(); + XGUI_ModuleConnector* aConnector = dynamic_cast(workshop()); + XGUI_Workshop* aWorkshop = aConnector->workshop(); + aWorkshop->selector()->clearSelection(); /// Restart sketcher operations automatically FeaturePtr aFeature = theOperation->feature(); @@ -215,45 +276,33 @@ void PartSet_Module::breakOperationSequence() myRestartingMode = RM_None; } -void PartSet_Module::operationAborted(ModuleBase_Operation* theOperation) +void PartSet_Module::onOperationAborted(ModuleBase_Operation* theOperation) { breakOperationSequence(); } -void PartSet_Module::sendOperation(ModuleBase_Operation* theOperation) -{ - // Install the document filter before any other filter - if (myDocumentShapeFilter.IsNull()) - myDocumentShapeFilter = new PartSet_GlobalFilter(myWorkshop); - myWorkshop->viewer()->addSelectionFilter(myDocumentShapeFilter); - - ModuleBase_IModule::sendOperation(theOperation); -} - -void PartSet_Module::operationStarted(ModuleBase_Operation* theOperation) +void PartSet_Module::onOperationStarted(ModuleBase_Operation* theOperation) { if (PartSet_SketcherMgr::isSketchOperation(theOperation)) { - Handle(V3d_Viewer) aViewer = myWorkshop->viewer()->AISContext()->CurrentViewer(); - aViewer->AddZLayer(myVisualLayerId); mySketchMgr->startSketch(theOperation); } else if (PartSet_SketcherMgr::isNestedSketchOperation(theOperation)) { mySketchMgr->startNestedSketch(theOperation); } + + myCustomPrs->activate(theOperation->feature()); } -void PartSet_Module::operationStopped(ModuleBase_Operation* theOperation) +void PartSet_Module::onOperationStopped(ModuleBase_Operation* theOperation) { + myCustomPrs->deactivate(); + if (PartSet_SketcherMgr::isSketchOperation(theOperation)) { mySketchMgr->stopSketch(theOperation); - Handle(V3d_Viewer) aViewer = myWorkshop->viewer()->AISContext()->CurrentViewer(); - aViewer->RemoveZLayer(myVisualLayerId); - myVisualLayerId = 0; } else if (PartSet_SketcherMgr::isNestedSketchOperation(theOperation)) { mySketchMgr->stopNestedSketch(theOperation); } - myWorkshop->viewer()->removeSelectionFilter(myDocumentShapeFilter); } ModuleBase_Operation* PartSet_Module::currentOperation() const @@ -287,18 +336,57 @@ bool PartSet_Module::canRedo() const return aCanRedo; } +bool PartSet_Module::canApplyAction(const ObjectPtr& theObject, const QString& theActionId) const +{ + bool aValid = true; + if (theActionId == "DELETE_CMD" || theActionId == "MOVE_CMD") { + FeaturePtr aFeature = ModelAPI_Feature::feature(theObject); + if (aFeature) { + // part features are removed in the PartSet module only. + if (aFeature->getKind() == PartSetPlugin_Part::ID()) + aValid = false; + } + } + return aValid; +} + +bool PartSet_Module::canCommitOperation() const +{ + return mySketchMgr->canCommitOperation(); +} + bool PartSet_Module::canDisplayObject(const ObjectPtr& theObject) const { // the sketch manager put the restriction to the objects display return mySketchMgr->canDisplayObject(theObject); } +bool PartSet_Module::canActivateSelection(const ObjectPtr& theObject) const +{ + bool aCanActivate = ModuleBase_IModule::canActivateSelection(theObject); + + ModuleBase_Operation* anOperation = myWorkshop->currentOperation(); + bool isSketchOp = PartSet_SketcherMgr::isSketchOperation(anOperation), + isNestedOp = PartSet_SketcherMgr::isNestedSketchOperation(anOperation); + if (isSketchOp || isNestedOp) { + // in active sketch operation it is possible to activate operation object in selection + // in the edit operation, e.g. points of the line can be moved when the line is edited + aCanActivate = aCanActivate || anOperation->isEditOperation(); + } + return aCanActivate; +} bool PartSet_Module::addViewerMenu(QMenu* theMenu, const QMap& theStdActions) const { return myMenuMgr->addViewerMenu(theMenu, theStdActions); } +void PartSet_Module::updateViewerMenu(const QMap& theStdActions) +{ + myMenuMgr->updateViewerMenu(theStdActions); +} + + void PartSet_Module::activeSelectionModes(QIntList& theModes) { theModes.clear(); @@ -338,24 +426,6 @@ void PartSet_Module::propertyPanelDefined(ModuleBase_Operation* theOperation) } } } - } else { - // Start editing constraint - if (theOperation->isEditOperation()) { - // TODO: #391 - to be removed - std::string aId = theOperation->id().toStdString(); - if (PartSet_SketcherMgr::isNestedSketchOperation(theOperation) && - PartSet_SketcherMgr::isDistanceOperation(theOperation)) { - // Find and activate widget for management of point for dimension line position - QList aWidgets = aPanel->modelWidgets(); - foreach (ModuleBase_ModelWidget* aWgt, aWidgets) { - PartSet_WidgetPoint2D* aPntWgt = dynamic_cast(aWgt); - if (aPntWgt) { - aPanel->activateWidget(aPntWgt); - return; - } - } - } - } } } @@ -384,7 +454,8 @@ void PartSet_Module::onSelectionChanged() std::string aId = aFeature->getKind(); if ((aId == SketchPlugin_ConstraintRadius::ID()) || (aId == SketchPlugin_ConstraintLength::ID()) || - (aId == SketchPlugin_ConstraintDistance::ID())) { + (aId == SketchPlugin_ConstraintDistance::ID()) || + (aId == SketchPlugin_ConstraintAngle::ID())) { editFeature(aFeature); } } @@ -453,49 +524,51 @@ void PartSet_Module::onVertexSelected() ModuleBase_ModelWidget* PartSet_Module::createWidgetByType(const std::string& theType, QWidget* theParent, Config_WidgetAPI* theWidgetApi, std::string theParentId) { - XGUI_ModuleConnector* aConnector = dynamic_cast(workshop()); - XGUI_Workshop* aWorkshop = aConnector->workshop(); + ModuleBase_IWorkshop* aWorkshop = workshop(); + XGUI_ModuleConnector* aConnector = dynamic_cast(aWorkshop); + XGUI_Workshop* aXUIWorkshop = aConnector->workshop(); ModuleBase_ModelWidget* aWgt = NULL; if (theType == "sketch-start-label") { - PartSet_WidgetSketchLabel* aLabelWgt = new PartSet_WidgetSketchLabel(theParent, + PartSet_WidgetSketchLabel* aLabelWgt = new PartSet_WidgetSketchLabel(theParent, aWorkshop, theWidgetApi, theParentId, mySketchMgr->isConstraintsShown()); - aLabelWgt->setWorkshop(aWorkshop); connect(aLabelWgt, SIGNAL(planeSelected(const std::shared_ptr&)), mySketchMgr, SLOT(onPlaneSelected(const std::shared_ptr&))); connect(aLabelWgt, SIGNAL(showConstraintToggled(bool)), mySketchMgr, SLOT(onShowConstraintsToggle(bool))); aWgt = aLabelWgt; } else if (theType == "sketch-2dpoint_selector") { - PartSet_WidgetPoint2D* aPointWgt = new PartSet_WidgetPoint2D(theParent, theWidgetApi, theParentId); - aPointWgt->setWorkshop(aWorkshop); + PartSet_WidgetPoint2D* aPointWgt = new PartSet_WidgetPoint2D(theParent, aWorkshop, + theWidgetApi, theParentId); aPointWgt->setSketch(mySketchMgr->activeSketch()); connect(aPointWgt, SIGNAL(vertexSelected()), this, SLOT(onVertexSelected())); aWgt = aPointWgt; - } if (theType == "point2ddistance") { - PartSet_WidgetPoint2dDistance* aDistanceWgt = new PartSet_WidgetPoint2dDistance(theParent, theWidgetApi, theParentId); - aDistanceWgt->setWorkshop(aWorkshop); + } else if (theType == "point2ddistance") { + PartSet_WidgetPoint2dDistance* aDistanceWgt = new PartSet_WidgetPoint2dDistance(theParent, + aWorkshop, theWidgetApi, theParentId); aDistanceWgt->setSketch(mySketchMgr->activeSketch()); aWgt = aDistanceWgt; - } if(theType == "point2dangle") { - PartSet_WidgetPoint2dAngle* anAngleWgt = new PartSet_WidgetPoint2dAngle(theParent, theWidgetApi, theParentId); - anAngleWgt->setWorkshop(aWorkshop); + } else if(theType == "point2dangle") { + PartSet_WidgetPoint2dAngle* anAngleWgt = new PartSet_WidgetPoint2dAngle(theParent, + aWorkshop, theWidgetApi, theParentId); anAngleWgt->setSketch(mySketchMgr->activeSketch()); aWgt = anAngleWgt; - } - if (theType == "sketch_shape_selector") { + } else if (theType == "sketch_shape_selector") { PartSet_WidgetShapeSelector* aShapeSelectorWgt = - new PartSet_WidgetShapeSelector(theParent, workshop(), theWidgetApi, theParentId); + new PartSet_WidgetShapeSelector(theParent, aWorkshop, theWidgetApi, theParentId); aShapeSelectorWgt->setSketcher(mySketchMgr->activeSketch()); aWgt = aShapeSelectorWgt; - } if (theType == "sketch_multi_selector") { + } else if (theType == "sketch_multi_selector") { PartSet_WidgetMultiSelector* aShapeSelectorWgt = - new PartSet_WidgetMultiSelector(theParent, workshop(), theWidgetApi, theParentId); + new PartSet_WidgetMultiSelector(theParent, aWorkshop, theWidgetApi, theParentId); aShapeSelectorWgt->setSketcher(mySketchMgr->activeSketch()); aWgt = aShapeSelectorWgt; + } else if (theType == WDG_DOUBLEVALUE_EDITOR) { + aWgt = new PartSet_WidgetEditor(theParent, aWorkshop, theWidgetApi, theParentId); + } else if (theType == "export_file_selector") { + aWgt = new PartSet_WidgetFileSelector(theParent, aWorkshop, theWidgetApi, theParentId); + } else if (theType == "sketch_launcher") { + aWgt = new PartSet_WidgetSketchCreator(theParent, this, theWidgetApi, theParentId); } - if (theType == WDG_DOUBLEVALUE_EDITOR) { - aWgt = new PartSet_WidgetEditor(theParent, workshop(), theWidgetApi, theParentId); - } return aWgt; } @@ -507,7 +580,7 @@ bool PartSet_Module::deleteObjects() ModuleBase_Operation* anOperation = myWorkshop->currentOperation(); bool isSketchOp = PartSet_SketcherMgr::isSketchOperation(anOperation), isNestedOp = PartSet_SketcherMgr::isNestedSketchOperation(anOperation); - if (isSketchOp && isNestedOp) { + if (isSketchOp || isNestedOp) { // 2. find selected presentations // selected objects should be collected before the current operation abort because // the abort leads to selection lost on constraint objects. It can be corrected after #386 issue @@ -527,8 +600,12 @@ bool PartSet_Module::deleteObjects() QObjectPtrList::const_iterator anIt = aSelectedObj.begin(), aLast = aSelectedObj.end(); for ( ; anIt != aLast; anIt++) { ObjectPtr anObject = *anIt; - if (mySketchMgr->isObjectOfSketch(anObject)) - aSketchObjects.append(anObject); + if (mySketchMgr->isObjectOfSketch(anObject)) { + // sketch feature should be used in this list because workshop deletes features only + // results are skipped + FeaturePtr aSketchFeature = ModelAPI_Feature::feature(anObject); + aSketchObjects.append(aSketchFeature); + } } // if the selection contains only local selected presentations from other sketches, // the Delete operation should not be done at all @@ -546,14 +623,12 @@ bool PartSet_Module::deleteObjects() // 4. delete features // sketch feature should be skipped, only sub-features can be removed // when sketch operation is active - std::set anIgnoredFeatures; - anIgnoredFeatures.insert(mySketchMgr->activeSketch()); - aWorkshop->deleteFeatures(aSketchObjects, anIgnoredFeatures); - + aWorkshop->deleteFeatures(aSketchObjects); // 5. stop operation aWorkshop->displayer()->updateViewer(); aMgr->finishOperation(); } else { + bool isPartRemoved = false; // Delete part with help of PartSet plugin // TODO: the deleted objects has to be processed by multiselection QObjectPtrList aObjects = myWorkshop->selection()->selectedObjects(); @@ -561,19 +636,51 @@ bool PartSet_Module::deleteObjects() ObjectPtr aObj = aObjects.first(); FeaturePtr aFeature = std::dynamic_pointer_cast(aObj); if (aFeature.get() && (aFeature->getKind() == PartSetPlugin_Part::ID())) { - std::shared_ptr aDoc = aMgr->activeDocument(); - aMgr->startOperation(PartSetPlugin_Remove::ID()); - FeaturePtr aFeature = aDoc->addFeature(PartSetPlugin_Remove::ID()); - aFeature->execute(); - aMgr->finishOperation(); - } else - return false; - } else - return false; + // Remove feature should be created in the document of the part results + ResultPtr aPartResult = aFeature->firstResult(); + if (aPartResult.get()) { + std::shared_ptr aPart = + std::dynamic_pointer_cast(aPartResult); + DocumentPtr aPartDoc = aPart->partDoc(); + if (aPartDoc.get()) { + aMgr->startOperation(PartSetPlugin_Remove::ID()); + FeaturePtr aFeature = aPartDoc->addFeature(PartSetPlugin_Remove::ID()); + aFeature->execute(); + aMgr->finishOperation(); + isPartRemoved = true; + } + } + } + } + return isPartRemoved; } return true; } +void PartSet_Module::onFeatureTriggered() +{ + SessionPtr aMgr = ModelAPI_Session::get(); + // 1. check whether the delete should be processed in the module + ModuleBase_Operation* anOperation = myWorkshop->currentOperation(); + bool isNestedOp = PartSet_SketcherMgr::isNestedCreateOperation(anOperation); + if (isNestedOp) { + // in case if in the viewer nothing is displayed, the create operation should not be + // comitted even if all values of the feature are initialized + if (!mySketchMgr->canDisplayCurrentCreatedFeature()) { + QAction* aCmd = dynamic_cast(sender()); + //Do nothing on uncheck + if (aCmd->isCheckable() && !aCmd->isChecked()) + return; + + // the action information should be saved before the operation is aborted + // because this abort leads to update command status, which unchecks this action + anOperation->abort(); + + launchOperation(aCmd->data().toString()); + } + } + ModuleBase_IModule::onFeatureTriggered(); +} void PartSet_Module::onObjectDisplayed(ObjectPtr theObject, AISObjectPtr theAIS) { @@ -591,6 +698,14 @@ void PartSet_Module::onObjectDisplayed(ObjectPtr theObject, AISObjectPtr theAIS) } } +void PartSet_Module::onBeforeObjectErase(ObjectPtr theObject, AISObjectPtr theAIS) +{ + // it should be recomputed in order to disappear in the viewer if the corresponded object + // is erased + if (myCustomPrs->isActive()) + myCustomPrs->customize(theObject); +} + void PartSet_Module::onViewTransformed(int theTrsfType) { // Set length of arrows constant in pixel size @@ -623,7 +738,12 @@ void PartSet_Module::onViewTransformed(int theTrsfType) return; double aLen = aView->Convert(20); + double aPrevLen = SketcherPrs_Tools::getArrowSize(); SketcherPrs_Tools::setArrowSize(aLen); + const double aPrevScale = aViewer->Scale(aViewer->activeView()); + const double aCurScale = aViewer->activeView()->Camera()->Scale(); + aViewer->SetScale(aViewer->activeView(), aCurScale); + SketcherPrs_Tools::setTextHeight (aCurScale / aPrevScale * SketcherPrs_Tools::getTextHeight()); bool isModified = false; QList aPrsList = aDisplayer->displayedPresentations(); foreach (AISObjectPtr aAIS, aPrsList) { @@ -640,6 +760,11 @@ void PartSet_Module::onViewTransformed(int theTrsfType) aDisplayer->updateViewer(); } +void PartSet_Module::customizeObject(ObjectPtr theObject) +{ + if (myCustomPrs->isActive()) + myCustomPrs->customize(theObject); +} void PartSet_Module::customizeObjectBrowser(QWidget* theObjectBrowser) { @@ -649,11 +774,37 @@ void PartSet_Module::customizeObjectBrowser(QWidget* theObjectBrowser) QPalette aPalet = aLabel->palette(); aPalet.setColor(QPalette::Text, QColor(0, 72, 140)); aLabel->setPalette(aPalet); + aOB->treeView()->setExpandsOnDoubleClick(false); +#ifdef ModuleDataModel + connect(aOB->treeView(), SIGNAL(doubleClicked(const QModelIndex&)), + SLOT(onTreeViewDoubleClick(const QModelIndex&))); + connect(aOB, SIGNAL(headerMouseDblClicked(const QModelIndex&)), + SLOT(onTreeViewDoubleClick(const QModelIndex&))); connect(aOB->treeView(), SIGNAL(doubleClicked(const QModelIndex&)), myDataModel, SLOT(onMouseDoubleClick(const QModelIndex&))); +#endif } } +ObjectPtr PartSet_Module::findPresentedObject(const AISObjectPtr& theAIS) const +{ + ObjectPtr anObject; + ModuleBase_Operation* aOperation = myWorkshop->currentOperation(); + if (aOperation) { + /// If last line finished on vertex the lines creation sequence has to be break + ModuleBase_IPropertyPanel* aPanel = aOperation->propertyPanel(); + ModuleBase_ModelWidget* anActiveWidget = aPanel->activeWidget(); + // if there is an active widget, find the presented object in it + if (!anActiveWidget) + anActiveWidget = aPanel->preselectionWidget(); + + ModuleBase_WidgetValidated* aWidgetValidated = dynamic_cast + (anActiveWidget); + if (aWidgetValidated) + anObject = aWidgetValidated->findPresentedObject(theAIS); + } + return anObject; +} void PartSet_Module::addObjectBrowserMenu(QMenu* theMenu) const { @@ -664,13 +815,23 @@ void PartSet_Module::addObjectBrowserMenu(QMenu* theMenu) const bool hasResult = false; bool hasFeature = false; bool hasParameter = false; - ModuleBase_Tools::checkObjects(aObjects, hasResult, hasFeature, hasParameter); + bool hasSubFeature = false; + ModuleBase_Tools::checkObjects(aObjects, hasResult, hasFeature, hasParameter, hasSubFeature); ObjectPtr aObject = aObjects.first(); if (aObject) { ResultPartPtr aPart = std::dynamic_pointer_cast(aObject); - if (aPart) { - if (aMgr->activeDocument() == aPart->partDoc()) + FeaturePtr aPartFeature = std::dynamic_pointer_cast(aObject); + bool isPart = aPart.get() || + (aPartFeature.get() && (aPartFeature->getKind() == PartSetPlugin_Part::ID())); + if (isPart) { + DocumentPtr aPartDoc; + if (!aPart.get()) { + aPart = std::dynamic_pointer_cast(aPartFeature->firstResult()); + } + if (aPart.get()) // this may be null is Part feature is disabled + aPartDoc = aPart->partDoc(); + if (aMgr->activeDocument() == aPartDoc) theMenu->addAction(myMenuMgr->action("DEACTIVATE_PART_CMD")); else theMenu->addAction(myMenuMgr->action("ACTIVATE_PART_CMD")); @@ -678,20 +839,42 @@ void PartSet_Module::addObjectBrowserMenu(QMenu* theMenu) const if (hasParameter || hasFeature) theMenu->addAction(myMenuMgr->action("EDIT_CMD")); } + + ResultBodyPtr aResult = std::dynamic_pointer_cast(aObject); + if( aResult.get() ) + { + theMenu->addAction(myMenuMgr->action("SELECT_PARENT_CMD")); + } } else { // If feature is 0 the it means that selected root object (document) if (aMgr->activeDocument() != aMgr->moduleDocument()) theMenu->addAction(myMenuMgr->action("ACTIVATE_PARTSET_CMD")); } } else if (aSelected == 0) { // if there is no selection then it means that upper label is selected - if (aMgr->activeDocument() != aMgr->moduleDocument()) - theMenu->addAction(myMenuMgr->action("ACTIVATE_PARTSET_CMD")); + QModelIndexList aIndexes = myWorkshop->selection()->selectedIndexes(); + if (aIndexes.size() == 0) // it means that selection happens in top label outside of tree view + if (aMgr->activeDocument() != aMgr->moduleDocument()) + theMenu->addAction(myMenuMgr->action("ACTIVATE_PARTSET_CMD")); } + bool aCanDeactivate = (myWorkshop->currentOperation() == 0); + myMenuMgr->action("ACTIVATE_PARTSET_CMD")->setEnabled(aCanDeactivate); + myMenuMgr->action("DEACTIVATE_PART_CMD")->setEnabled(aCanDeactivate); + myMenuMgr->action("ACTIVATE_PART_CMD")->setEnabled(aCanDeactivate); } void PartSet_Module::processEvent(const std::shared_ptr& theMessage) { if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_DOCUMENT_CHANGED)) { + // Do not change activation of parts if an operation active + static QStringList aAllowActivationList; + if (aAllowActivationList.isEmpty()) + aAllowActivationList << + QString(PartSetPlugin_Part::ID().c_str()) << + QString(PartSetPlugin_Duplicate::ID().c_str()) << + QString(PartSetPlugin_Remove::ID().c_str()); + if (myWorkshop->currentOperation() && + (!aAllowActivationList.contains(myWorkshop->currentOperation()->id()))) + return; XGUI_ModuleConnector* aConnector = dynamic_cast(myWorkshop); XGUI_Workshop* aWorkshop = aConnector->workshop(); XGUI_DataTree* aTreeView = aWorkshop->objectBrowser()->treeView(); @@ -700,8 +883,9 @@ void PartSet_Module::processEvent(const std::shared_ptr& theMess SessionPtr aMgr = ModelAPI_Session::get(); DocumentPtr aActiveDoc = aMgr->activeDocument(); - DocumentPtr aDoc = aMgr->moduleDocument(); +#ifdef ModuleDataModel QModelIndex aOldIndex = myDataModel->activePartTree(); + DocumentPtr aDoc = aMgr->moduleDocument(); if (aActiveDoc == aDoc) { if (aOldIndex.isValid()) aTreeView->setExpanded(aOldIndex, false); @@ -714,7 +898,8 @@ void PartSet_Module::processEvent(const std::shared_ptr& theMess if (aPart->partDoc() == aActiveDoc) { QModelIndex aIndex = myDataModel->partIndex(aPart); if (myDataModel->activatePart(aIndex)) { - aTreeView->setExpanded(aOldIndex, false); + if (aOldIndex.isValid()) + aTreeView->setExpanded(aOldIndex, false); aTreeView->setExpanded(myDataModel->activePartTree(), true); aPalet.setColor(QPalette::Text, Qt::black); } @@ -722,6 +907,76 @@ void PartSet_Module::processEvent(const std::shared_ptr& theMess } } } +#else + // Problem with MPV: At first time on creation it doesn't work because Part feature + // creation event will be sent after + if (aActivePartIndex.isValid()) + aTreeView->setExpanded(aActivePartIndex, false); + XGUI_DataModel* aDataModel = aWorkshop->objectBrowser()->dataModel(); + aActivePartIndex = aDataModel->documentRootIndex(aActiveDoc); + if (aActivePartIndex.isValid()) + aTreeView->setExpanded(aActivePartIndex, true); +#endif aLabel->setPalette(aPalet); + aWorkshop->updateCommandStatus(); + + // Update displayed objects in order to update active color + XGUI_Displayer* aDisplayer = aWorkshop->displayer(); + QObjectPtrList aObjects = aDisplayer->displayedObjects(); + foreach(ObjectPtr aObj, aObjects) + aDisplayer->redisplay(aObj, false); + aDisplayer->updateViewer(); } -} \ No newline at end of file +} + +void PartSet_Module::onTreeViewDoubleClick(const QModelIndex& theIndex) +{ + if (myWorkshop->currentOperation()) // Do not change activation of parts if an operation active + return; + SessionPtr aMgr = ModelAPI_Session::get(); + if (!theIndex.isValid()) { + aMgr->setActiveDocument(aMgr->moduleDocument()); + return; + } + if (theIndex.column() != 0) // Use only first column + return; + ObjectPtr aObj = myDataModel->object(theIndex); + ResultPartPtr aPart = std::dynamic_pointer_cast(aObj); + if (!aPart.get()) { // Probably this is Feature + FeaturePtr aPartFeature = std::dynamic_pointer_cast(aObj); + if (aPartFeature.get() && (aPartFeature->getKind() == PartSetPlugin_Part::ID())) { + aPart = std::dynamic_pointer_cast(aPartFeature->firstResult()); + } + } + if (aPart.get()) { // if this is a part + if (aPart->partDoc() == aMgr->activeDocument()) { + aMgr->setActiveDocument(aMgr->moduleDocument()); + } else { + aPart->activate(); + } + } +} + + +void PartSet_Module::onViewCreated(ModuleBase_IViewWindow*) +{ + // z layer is created for all started operations in order to visualize operation AIS presentation + // over the object + Handle(V3d_Viewer) aViewer = myWorkshop->viewer()->AISContext()->CurrentViewer(); + if (myVisualLayerId == 0) { + if (myVisualLayerId == 0) + aViewer->AddZLayer(myVisualLayerId); + } else { + TColStd_SequenceOfInteger aZList; + aViewer->GetAllZLayers(aZList); + bool aFound = false; + for (int i = 1; i <= aZList.Length(); i++) { + if (aZList(i) == myVisualLayerId) { + aFound = true; + break; + } + } + if (!aFound) + aViewer->AddZLayer(myVisualLayerId); + } +}