src/ModuleBase/ModuleBase_WidgetSelectionFilter.cpp

   1 // Copyright (C) 2014-2021  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include "ModuleBase_WidgetSelectionFilter.h"
  21 #include "ModuleBase_Tools.h"
  22 #include "ModuleBase_IWorkshop.h"
  23 #include "ModuleBase_ISelectionActivate.h"
  24 #include "ModuleBase_IModule.h"
  25 #include "ModuleBase_IViewer.h"
  26 #include "ModuleBase_IPropertyPanel.h"
  27 #include "ModuleBase_PageWidget.h"
  28 #include "ModuleBase_WidgetMultiSelector.h"
  29 #include "ModuleBase_ResultPrs.h"
  30 #include "ModuleBase_WidgetFactory.h"
  31
  32 #include <ModelAPI_Session.h>
  33 #include <ModelAPI_AttributeSelectionList.h>
  34 #include <ModelAPI_Events.h>
  35 #include <ModelAPI_FiltersFactory.h>
  36 #include <ModelAPI_ResultBody.h>
  37 #include <GeomAPI_ShapeExplorer.h>
  38
  39 #include <Events_Loop.h>
  40 #include <Config_ValidatorReader.h>
  41
  42 #include <AIS_InteractiveContext.hxx>
  43 #include <StdSelect_BRepOwner.hxx>
  44 #include <TopoDS_Compound.hxx>
  45 #include <BRep_Builder.hxx>
  46 #include <TopExp_Explorer.hxx>
  47
  48 #include <QLayout>
  49 #include <QPushButton>
  50 #include <QLabel>
  51 #include <QComboBox>
  52 #include <QGroupBox>
  53 #include <QDialog>
  54 #include <QToolButton>
  55 #include <QCheckBox>
  56 #include <QDir>
  57
  58 FeaturePtr ModuleBase_WidgetSelectionFilter::SelectorFeature;
  59 std::string ModuleBase_WidgetSelectionFilter::AttributeId;
  60
  61
  62 GeomAPI_Shape::ShapeType selectionType(const QString& theType)
  63 {
  64   QString aType = theType.toUpper();
  65   if ((aType == "VERTEX") || (aType == "VERTICES"))
  66     return GeomAPI_Shape::VERTEX;
  67   else if ((aType == "EDGE") || (aType == "EDGES"))
  68     return GeomAPI_Shape::EDGE;
  69   else if ((aType == "WIRE") || (aType == "WIRES"))
  70     return GeomAPI_Shape::WIRE;
  71   else if ((aType == "FACE") || (aType == "FACES"))
  72     return GeomAPI_Shape::FACE;
  73   else if ((aType == "SHELL") || (aType == "SHELLS"))
  74     return GeomAPI_Shape::SHELL;
  75   else if ((aType == "SOLID") || (aType == "SOLIDS"))
  76     return GeomAPI_Shape::SOLID;
  77   else if ((aType == "COMPSOLID") || (aType == "COMPSOLIDS"))
  78     return GeomAPI_Shape::COMPSOLID;
  79   else if ((aType == "COMPOUND") || (aType == "COMPOUNDS"))
  80     return GeomAPI_Shape::COMPOUND;
  81   else
  82     return GeomAPI_Shape::SHAPE;
  83 }
  84
  85
  86 ModuleBase_FilterStarter::ModuleBase_FilterStarter(const std::string& theFeature,
  87   QWidget* theParent, ModuleBase_IWorkshop* theWorkshop)
  88   : QWidget(theParent),
  89   myFeatureName(theFeature),
  90   myWorkshop(theWorkshop)
  91 {
  92   QHBoxLayout* aMainLayout = new QHBoxLayout(this);
  93   ModuleBase_Tools::adjustMargins(aMainLayout);
  94
  95   aMainLayout->addStretch(1);
  96   QPushButton* aLaunchBtn = new QPushButton(
  97       ModuleBase_Tools::translate("FiltersSelection", "Selection by filters"), this);
  98   connect(aLaunchBtn, SIGNAL(clicked()), SLOT(onFiltersLaunch()));
  99   aMainLayout->addWidget(aLaunchBtn);
 100 }
 101
 102 void ModuleBase_FilterStarter::onFiltersLaunch()
 103 {
 104   static QString aHelpFileName = QString("FiltersPlugin") + QDir::separator() +
 105     QString("FiltersPlugin.html");
 106
 107   ModuleBase_Operation* aParentOp = myWorkshop->currentOperation();
 108   ModuleBase_OperationFeature* aFeatureOp = dynamic_cast<ModuleBase_OperationFeature*>(aParentOp);
 109   if (aFeatureOp)
 110     // Open transaction on filters operation finish
 111     aFeatureOp->openTransactionOnResume();
 112
 113   QWidget* aParent = parentWidget();
 114   ModuleBase_WidgetMultiSelector* aSelector =
 115     dynamic_cast<ModuleBase_WidgetMultiSelector*>(aParent);
 116   while (!aSelector) {
 117     aParent = aParent->parentWidget();
 118     aSelector = dynamic_cast<ModuleBase_WidgetMultiSelector*>(aParent);
 119   }
 120   if (!aSelector)
 121     return;
 122   ModuleBase_WidgetSelectionFilter::SelectorFeature = aSelector->feature();
 123   ModuleBase_WidgetSelectionFilter::AttributeId = aSelector->attributeID();
 124
 125   // Launch Filters operation
 126   ModuleBase_OperationFeature* aFOperation = dynamic_cast<ModuleBase_OperationFeature*>
 127     (myWorkshop->module()->createOperation(myFeatureName));
 128
 129   AttributeSelectionListPtr aAttrList =
 130     ModuleBase_WidgetSelectionFilter::SelectorFeature->selectionList(
 131       ModuleBase_WidgetSelectionFilter::AttributeId);
 132   FiltersFeaturePtr aFilters = aAttrList->filters();
 133   if (aFilters.get())
 134     aFOperation->setFeature(aFilters);
 135   aFOperation->setHelpFileName(aHelpFileName);
 136   myWorkshop->processLaunchOperation(aFOperation);
 137 }
 138
 139 //*****************************************************************************
 140 //*****************************************************************************
 141 //*****************************************************************************
 142 ModuleBase_FilterItem::ModuleBase_FilterItem(
 143   const std::string& theFilter, ModuleBase_WidgetSelectionFilter* theParent)
 144   : QWidget(theParent->filtersWidget()), myFilterID(theFilter),
 145     mySelection(std::dynamic_pointer_cast<ModelAPI_FiltersFeature>(theParent->feature()))
 146 {
 147   FilterPtr aFilter = ModelAPI_Session::get()->filters()->filter(theFilter);
 148   std::string aXmlString = aFilter->xmlRepresentation();
 149   if (aXmlString.length() == 0)
 150     addItemRow(this);
 151   else {
 152     std::string anAttrPrefix; // this must be added to the attributes names for multiple filters
 153     std::string aFilterKind = ModelAPI_Session::get()->filters()->id(aFilter);
 154     if (theFilter != aFilterKind) {
 155       anAttrPrefix = theFilter.substr(0, theFilter.size() - aFilterKind.size());
 156     }
 157     ModuleBase_WidgetFactory aFactory(aXmlString, theParent->workshop(), anAttrPrefix);
 158     Config_ValidatorReader aValidatorReader(aXmlString, true);
 159     aValidatorReader.setFeatureId(mySelection->getKind());
 160     aValidatorReader.readAll();
 161
 162     QVBoxLayout* aLayout = new QVBoxLayout(this);
 163     ModuleBase_Tools::zeroMargins(aLayout);
 164
 165     QWidget* aItemRow = new QWidget(this);
 166     addItemRow(aItemRow);
 167     aLayout->addWidget(aItemRow);
 168
 169     ModuleBase_PageWidget* aParamsWgt = new ModuleBase_PageWidget(this);
 170     aParamsWgt->setFrameStyle(QFrame::Box | QFrame::Raised);
 171     aFactory.createWidget(aParamsWgt);
 172     ModuleBase_Tools::zeroMargins(aParamsWgt->layout());
 173     myWidgets = aFactory.getModelWidgets();
 174     foreach(ModuleBase_ModelWidget* aWidget, myWidgets) {
 175       aWidget->setFeature(theParent->feature());
 176       connect(aWidget, SIGNAL(focusInWidget(ModuleBase_ModelWidget*)),
 177         theParent, SIGNAL(focusInWidget(ModuleBase_ModelWidget*)));
 178       connect(aWidget, SIGNAL(focusOutWidget(ModuleBase_ModelWidget*)),
 179         theParent, SIGNAL(focusOutWidget(ModuleBase_ModelWidget*)));
 180       connect(aWidget, SIGNAL(objectUpdated()), theParent, SLOT(onObjectUpdated()));
 181     }
 182     aLayout->addWidget(aParamsWgt);
 183   }
 184 }
 185
 186 void ModuleBase_FilterItem::addItemRow(QWidget* theParent)
 187 {
 188   std::string aContext = mySelection->getKind();
 189   QHBoxLayout* aLayout = new QHBoxLayout(theParent);
 190   ModuleBase_Tools::zeroMargins(aLayout);
 191
 192   // Reverse filter button
 193   myRevBtn = new QToolButton(theParent);
 194   myRevBtn->setCheckable(true);
 195   bool isReversed = mySelection->isReversed(myFilterID);
 196   myRevBtn->setChecked(isReversed);
 197   myRevBtn->setAutoRaise(true);
 198   if (isReversed)
 199     myRevBtn->setIcon(QIcon(":pictures/reverce.png"));
 200   else
 201     myRevBtn->setIcon(QIcon(":pictures/add.png"));
 202   myRevBtn->setToolTip(ModuleBase_Tools::translate(aContext, "Reverse the filter"));
 203   connect(myRevBtn, SIGNAL(toggled(bool)), SLOT(onReverse(bool)));
 204   aLayout->addWidget(myRevBtn);
 205
 206   const std::string& aFilterName = ModelAPI_Session::get()->filters()->filter(myFilterID)->name();
 207   aLayout->addWidget(new QLabel(ModuleBase_Tools::translate(aContext, aFilterName), theParent), 1);
 208
 209   QToolButton* aDelBtn = new QToolButton(theParent);
 210   aDelBtn->setIcon(QIcon(":pictures/delete.png"));
 211   aDelBtn->setAutoRaise(true);
 212   aDelBtn->setToolTip(ModuleBase_Tools::translate(aContext, "Delete the filter"));
 213   connect(aDelBtn, SIGNAL(clicked(bool)), SLOT(onDelete()));
 214   aLayout->addWidget(aDelBtn);
 215 }
 216
 217 void ModuleBase_FilterItem::onReverse(bool theCheck)
 218 {
 219   mySelection->setReversed(myFilterID, theCheck);
 220   if (theCheck)
 221     myRevBtn->setIcon(QIcon(":pictures/reverce.png"));
 222   else
 223     myRevBtn->setIcon(QIcon(":pictures/add.png"));
 224   emit reversedItem(this);
 225 }
 226
 227 void ModuleBase_FilterItem::onDelete()
 228 {
 229   emit deleteItem(this);
 230 }
 231
 232
 233 //*****************************************************************************
 234 //*****************************************************************************
 235 //*****************************************************************************
 236 ModuleBase_WidgetSelectionFilter::ModuleBase_WidgetSelectionFilter(QWidget* theParent,
 237   ModuleBase_IWorkshop* theWorkshop, const Config_WidgetAPI* theData, bool theReadOnly)
 238   : ModuleBase_ModelWidget(theParent, theData),
 239   myWorkshop(theWorkshop),
 240   mySelectorFeature(ModuleBase_WidgetSelectionFilter::SelectorFeature),
 241   mySelectorAttribute(ModuleBase_WidgetSelectionFilter::AttributeId)
 242 {
 243   // Clear Old selection
 244     AttributeSelectionListPtr aAttrList = mySelectorFeature->selectionList(mySelectorAttribute);
 245     mySelectionType = selectionType(aAttrList->selectionType().c_str());
 246   if (!theReadOnly) {
 247     aAttrList->clear();
 248   }
 249
 250   // Define widgets
 251   QVBoxLayout* aMainLayout = new QVBoxLayout(this);
 252   ModuleBase_Tools::adjustMargins(aMainLayout);
 253
 254   QGroupBox* aFiltersGroup = new QGroupBox(translate("Filters"), this);
 255   QVBoxLayout* aGroupLayout = new QVBoxLayout(aFiltersGroup);
 256   aGroupLayout->setContentsMargins(0, 0, 0, 0);
 257   aGroupLayout->setSpacing(0);
 258
 259   myFiltersWgt = new QWidget();
 260   myFiltersLayout = new QVBoxLayout(myFiltersWgt);
 261   myFiltersLayout->setContentsMargins(0, 0, 0, 0);
 262   aGroupLayout->addWidget(myFiltersWgt);
 263
 264   myFiltersCombo = new QComboBox(aFiltersGroup);
 265   myFiltersCombo->addItem(translate("Add new filter..."));
 266   SessionPtr aSession = ModelAPI_Session::get();
 267   std::list<FilterPtr> allFilters =
 268     aSession->filters()->filters((GeomAPI_Shape::ShapeType) mySelectionType);
 269   storeFilters(allFilters);
 270   QStringList aItems;
 271   std::list<FilterPtr>::const_iterator aIt;
 272   for (aIt = allFilters.cbegin(); aIt != allFilters.cend(); aIt++) {
 273     aItems.push_back(translate((*aIt)->name().c_str()));
 274   }
 275   myFiltersCombo->addItems(aItems);
 276   connect(myFiltersCombo, SIGNAL(currentIndexChanged(int)), SLOT(onAddFilter(int)));
 277
 278   aGroupLayout->addWidget(myFiltersCombo);
 279   aMainLayout->addWidget(aFiltersGroup);
 280
 281   // Select Button
 282   QWidget* aBtnWgt = new QWidget(this);
 283   QHBoxLayout* aBtnLayout = new QHBoxLayout(aBtnWgt);
 284   ModuleBase_Tools::adjustMargins(aBtnLayout);
 285
 286   aBtnLayout->addStretch(1);
 287
 288   mySelectBtn = new QPushButton(translate("Select"), aBtnWgt);
 289   connect(mySelectBtn, SIGNAL(clicked()), SLOT(onSelect()));
 290   aBtnLayout->addWidget(mySelectBtn);
 291
 292   aMainLayout->addWidget(aBtnWgt);
 293
 294   // Label widgets
 295   QWidget* aLblWgt = new QWidget(this);
 296   QHBoxLayout* aLblLayout = new QHBoxLayout(aLblWgt);
 297   ModuleBase_Tools::zeroMargins(aLblLayout);
 298
 299   aLblLayout->addWidget(new QLabel(translate("Number of selected objects:"), aLblWgt));
 300
 301   myNbLbl = new QLabel("0", aLblWgt);
 302   aLblLayout->addWidget(myNbLbl);
 303
 304   // Show only button
 305   myShowBtn = new QCheckBox(translate("Show only"), this);
 306   connect(myShowBtn, SIGNAL(toggled(bool)), SLOT(onShowOnly(bool)));
 307   aLblLayout->addWidget(myShowBtn);
 308
 309   aMainLayout->addWidget(aLblWgt);
 310
 311   aMainLayout->addStretch(1);
 312
 313   updateSelectBtn();
 314   if (theReadOnly) {
 315     myFiltersCombo->hide();
 316     mySelectBtn->hide();
 317     aLblWgt->hide();
 318     myShowBtn->hide();
 319   }
 320 }
 321
 322 ModuleBase_WidgetSelectionFilter::~ModuleBase_WidgetSelectionFilter()
 323 {
 324   myValues.clear();
 325   if (!myPreview.IsNull()) {
 326     Handle(AIS_InteractiveContext) aCtx = myWorkshop->viewer()->AISContext();
 327     aCtx->Remove(myPreview, false);
 328     myPreview.Nullify();
 329     if (myListIO.Size() > 0) {
 330       aCtx = myWorkshop->viewer()->AISContext();
 331       AIS_ListOfInteractive::const_iterator aIt;
 332       Handle(AIS_Shape) aShapeIO;
 333       for (aIt = myListIO.cbegin(); aIt != myListIO.cend(); aIt++) {
 334         aShapeIO = Handle(AIS_Shape)::DownCast(*aIt);
 335         if (!aShapeIO.IsNull()) {
 336           aCtx->Display(aShapeIO, false);
 337           std::shared_ptr<GeomAPI_AISObject> anAISObj = AISObjectPtr(new GeomAPI_AISObject());
 338           anAISObj->setImpl(new Handle(AIS_InteractiveObject)(aShapeIO));
 339           myWorkshop->applyCurrentSelectionModes(anAISObj);
 340         }
 341       }
 342       myListIO.Clear();
 343       myShowBtn->setChecked(false);
 344     }
 345     myWorkshop->viewer()->update();
 346   }
 347   SelectorFeature = FeaturePtr();
 348   AttributeId = "";
 349 }
 350
 351 void ModuleBase_WidgetSelectionFilter::onAddFilter(int theIndex)
 352 {
 353   if (theIndex == 0)
 354     return;
 355
 356   ModelAPI_FiltersFactory* aFactory = ModelAPI_Session::get()->filters();
 357   FiltersFeaturePtr aFiltersFeature =
 358     std::dynamic_pointer_cast<ModelAPI_FiltersFeature>(myFeature);
 359
 360   std::string aText = myFiltersCombo->itemText(theIndex).toStdString();
 361
 362   std::list<FilterPtr>::iterator aIt;
 363   std::string aFilter;
 364   std::map<std::string, FilterPtr>::const_iterator aFound = myFilters.find(aText);
 365   if (aFound == myFilters.end()) {
 366     std::list<FilterPtr> aFilters = aFactory->filters((GeomAPI_Shape::ShapeType) mySelectionType);
 367     storeFilters(aFilters);
 368     aFound = myFilters.find(aText);
 369   }
 370   if (aFound != myFilters.end())
 371     aFilter = aFactory->id(aFound->second);
 372
 373   aFiltersFeature->addFilter(aFilter);
 374   updateObject(myFeature);
 375
 376   QList<ModuleBase_FilterItem*> aList = itemsList();
 377   if (!aList.isEmpty() && (aList.last()->widgets().size() > 0))
 378     aList.last()->widgets().first()->emitFocusInWidget();
 379   else
 380     emitFocusInWidget();
 381 }
 382
 383 ModuleBase_FilterItem* ModuleBase_WidgetSelectionFilter::onAddFilter(const std::string& theFilter)
 384 {
 385   if (theFilter.length() == 0)
 386     return 0;
 387   ModuleBase_FilterItem* aItem = new ModuleBase_FilterItem(theFilter, this);
 388   connect(aItem, SIGNAL(deleteItem(ModuleBase_FilterItem*)),
 389     SLOT(onDeleteItem(ModuleBase_FilterItem*)));
 390   connect(aItem, SIGNAL(reversedItem(ModuleBase_FilterItem*)),
 391     SLOT(onReverseItem(ModuleBase_FilterItem*)));
 392   myFiltersLayout->addWidget(aItem);
 393
 394   updateSelectBtn();
 395   clearCurrentSelection(true);
 396   updateNumberSelected();
 397   return aItem;
 398 }
 399
 400 void ModuleBase_WidgetSelectionFilter::onDeleteItem(ModuleBase_FilterItem* theItem)
 401 {
 402   std::string aFilter = theItem->filter();
 403   QList<ModuleBase_ModelWidget*> aWidgets = theItem->widgets();
 404   foreach(ModuleBase_ModelWidget* aWgt, aWidgets) {
 405     aWgt->deactivate();
 406   }
 407   myFiltersLayout->removeWidget(theItem);
 408   theItem->deleteLater();
 409
 410   ModelAPI_FiltersFactory* aFactory = ModelAPI_Session::get()->filters();
 411   if (!aFactory->filter(aFilter)->isMultiple()) {
 412     //myFilters.push_back(aFilter);
 413     myFiltersCombo->addItem(ModelAPI_Session::get()->filters()->filter(aFilter)->name().c_str());
 414   }
 415   FiltersFeaturePtr aFiltersFeature =
 416     std::dynamic_pointer_cast<ModelAPI_FiltersFeature>(myFeature);
 417   aFiltersFeature->removeFilter(aFilter);
 418
 419   updateSelectBtn();
 420   clearCurrentSelection(true);
 421   updateNumberSelected();
 422
 423   myWorkshop->deactivateCurrentSelector();
 424   myWorkshop->selectionActivate()->updateSelectionModes();
 425   myWorkshop->selectionActivate()->updateSelectionFilters();
 426   redisplayFeature();
 427   emitFocusInWidget();
 428   updateObject(myFeature);
 429 }
 430
 431
 432 void ModuleBase_WidgetSelectionFilter::redisplayFeature()
 433 {
 434   static Events_ID aDispEvent = Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY);
 435   ModelAPI_EventCreator::get()->sendUpdated(myFeature, aDispEvent);
 436   Events_Loop::loop()->flush(aDispEvent);
 437 }
 438
 439 void ModuleBase_WidgetSelectionFilter::onReverseItem(ModuleBase_FilterItem* theItem)
 440 {
 441   updateSelectBtn();
 442   clearCurrentSelection(true);
 443   updateNumberSelected();
 444 }
 445
 446 void ModuleBase_WidgetSelectionFilter::onSelect()
 447 {
 448   Handle(AIS_InteractiveContext) aCtx = myWorkshop->viewer()->AISContext();
 449   if (aCtx.IsNull())
 450     return;
 451
 452   clearCurrentSelection();
 453
 454   BRep_Builder aBuilder;
 455   TopoDS_Compound aComp;
 456   aBuilder.MakeCompound(aComp);
 457
 458   DocumentPtr aDoc = myFeature->document();
 459   int aNb = aDoc->size(ModelAPI_ResultBody::group());
 460   ObjectPtr aObj;
 461   ResultBodyPtr aBody;
 462   for (int i = 0; i < aNb; i++) {
 463     aObj = aDoc->object(ModelAPI_ResultBody::group(), i);
 464     aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aObj);
 465     GeomShapePtr aShape = aBody->shape();
 466     std::list<GeomShapePtr> aSubShapes =
 467       aShape->subShapes((GeomAPI_Shape::ShapeType)mySelectionType);
 468     TopTools_MapOfShape alreadyThere;
 469     std::list<GeomShapePtr>::const_iterator aShapesIt;
 470     for (aShapesIt = aSubShapes.cbegin(); aShapesIt != aSubShapes.cend(); aShapesIt++) {
 471       GeomShapePtr aSubShape = (*aShapesIt);
 472       TopoDS_Shape aTShape = aSubShape->impl<TopoDS_Shape>();
 473       if (!alreadyThere.Add(aTShape))
 474         continue;
 475       static SessionPtr aSession = ModelAPI_Session::get();
 476       bool isValid = aSession->filters()->isValid(myFeature, aBody, aSubShape);
 477       if (isValid) {
 478         aBuilder.Add(aComp, aTShape);
 479         // bos #24043: Naming on a compsolid works wrong.
 480         // Find a simple sub-result for the ViewerPrs context:
 481         ResultBodyPtr aContext = aBody;
 482         bool isComposite = aContext->numberOfSubs() > 0;
 483         while (isComposite) {
 484           isComposite = false;
 485           int nbSubs = aContext->numberOfSubs();
 486           for (int aSubIndex = 0; aSubIndex < nbSubs; aSubIndex++) {
 487             ResultBodyPtr aSubResult = aContext->subResult(aSubIndex);
 488             GeomShapePtr aSubResultShape = aSubResult->shape();
 489             if (aSubResultShape->isSubShape(aSubShape)) {
 490               aContext = aSubResult;
 491               isComposite = aContext->numberOfSubs() > 0;
 492               break;
 493             }
 494           }
 495         }
 496         ModuleBase_ViewerPrsPtr aValue(new ModuleBase_ViewerPrs(aContext, aSubShape));
 497         //ModuleBase_ViewerPrsPtr aValue(new ModuleBase_ViewerPrs(aObj, aSubShape));
 498         myValues.append(aValue);
 499       }
 500     }
 501   }
 502
 503   if (myValues.size() > 0)
 504     updatePreview(aComp);
 505   updateNumberSelected();
 506   updateObject(myFeature);
 507   onShowOnly(myShowBtn->isChecked());
 508 }
 509
 510 void ModuleBase_WidgetSelectionFilter::updatePreview(const TopoDS_Shape& theShape)
 511 {
 512   Handle(AIS_InteractiveContext) aCtx = myWorkshop->viewer()->AISContext();
 513   if (aCtx.IsNull())
 514     return;
 515
 516   if (myPreview.IsNull()) {
 517     myPreview = new AIS_Shape(theShape);
 518     myPreview->SetDisplayMode(AIS_Shaded);
 519     myPreview->SetColor(Quantity_NOC_BLUE1);
 520     Handle(Prs3d_Drawer) aDrawer = myPreview->Attributes();
 521     if (aDrawer->HasOwnPointAspect()) {
 522       aDrawer->PointAspect()->SetTypeOfMarker(Aspect_TOM_O_STAR);
 523       aDrawer->PointAspect()->SetColor(Quantity_NOC_BLUE1);
 524       aDrawer->PointAspect()->SetScale(2.);
 525     }
 526     else
 527       aDrawer->SetPointAspect(new Prs3d_PointAspect(Aspect_TOM_O_STAR, Quantity_NOC_BLUE1, 2.));
 528     myPreview->SetTransparency();
 529     aCtx->Display(myPreview, true);
 530     aCtx->Deactivate(myPreview);
 531   }
 532   else {
 533     myPreview->Set(theShape);
 534     aCtx->Redisplay(myPreview, true);
 535   }
 536 }
 537
 538
 539 void ModuleBase_WidgetSelectionFilter::onShowOnly(bool theShow)
 540 {
 541   Handle(AIS_InteractiveContext) aCtx = myWorkshop->viewer()->AISContext();
 542   if (theShow) {
 543     AIS_ListOfInteractive aList;
 544     aCtx->DisplayedObjects(AIS_KOI_Shape, -1, aList);
 545     if (!myPreview.IsNull())
 546       aList.Remove(myPreview);
 547     if (aList.Size() > 0)
 548       myListIO = aList;
 549   }
 550   AIS_ListOfInteractive::const_iterator aIt;
 551   Handle(AIS_Shape) aShapeIO;
 552   bool isModified = false;
 553   for (aIt = myListIO.cbegin(); aIt != myListIO.cend(); aIt++) {
 554     aShapeIO = Handle(AIS_Shape)::DownCast(*aIt);
 555     if (!aShapeIO.IsNull()) {
 556       if (theShow) {
 557         if (aCtx->IsDisplayed(aShapeIO)) {
 558           aCtx->Erase(aShapeIO, false);
 559           isModified = true;
 560         }
 561       }
 562       else {
 563         if (!aCtx->IsDisplayed(aShapeIO)) {
 564           aCtx->Display(aShapeIO, false);
 565           std::shared_ptr<GeomAPI_AISObject> anAISObj = AISObjectPtr(new GeomAPI_AISObject());
 566           anAISObj->setImpl(new Handle(AIS_InteractiveObject)(aShapeIO));
 567           myWorkshop->applyCurrentSelectionModes(anAISObj);
 568           isModified = true;
 569         }
 570       }
 571     }
 572   }
 573   if (isModified)
 574     myWorkshop->viewer()->update();
 575 }
 576
 577 void ModuleBase_WidgetSelectionFilter::updateSelectBtn()
 578 {
 579   FiltersFeaturePtr aFiltersFeature = std::dynamic_pointer_cast<ModelAPI_FiltersFeature>(myFeature);
 580   mySelectBtn->setEnabled(aFiltersFeature.get() && (aFiltersFeature->filters().size() > 0));
 581 }
 582
 583 void ModuleBase_WidgetSelectionFilter::updateNumberSelected()
 584 {
 585   int aNb = myValues.size();
 586   myNbLbl->setText(QString::number(aNb));
 587   if (aNb == 0) {
 588     myFeature->setError(translate("Selection is empty").toStdString(), false, false);
 589     myShowBtn->setChecked(false);
 590     onShowOnly(false);
 591     myShowBtn->setEnabled(false);
 592   }
 593   else {
 594     myFeature->setError("", false, false);
 595     myFeature->data()->execState(ModelAPI_StateDone);
 596     myShowBtn->setEnabled(true);
 597   }
 598 }
 599
 600 QList<QWidget*> ModuleBase_WidgetSelectionFilter::getControls() const
 601 {
 602   QList<QWidget*> aWidgets;
 603   aWidgets.append(myFiltersCombo);
 604   return aWidgets;
 605 }
 606
 607 void ModuleBase_WidgetSelectionFilter::clearCurrentSelection(bool toUpdate)
 608 {
 609   myValues.clear();
 610   if (!myPreview.IsNull()) {
 611     Handle(AIS_InteractiveContext) aCtx = myWorkshop->viewer()->AISContext();
 612     aCtx->Remove(myPreview, toUpdate);
 613     myPreview.Nullify();
 614   }
 615 }
 616
 617 void replaceSubShapesByResult(QList<ModuleBase_ViewerPrsPtr>& theResults, int theShapeType)
 618 {
 619   QMap<ObjectPtr, QList<GeomShapePtr>> myResShapes;
 620   // Sort sub-shapes by result
 621   foreach (ModuleBase_ViewerPrsPtr aPrs, theResults) {
 622     if (myResShapes.contains(aPrs->object()))
 623       myResShapes[aPrs->object()].append(aPrs->shape());
 624     else {
 625       QList<GeomShapePtr> aShapes;
 626       aShapes << aPrs->shape();
 627       myResShapes[aPrs->object()] = aShapes;
 628     }
 629   }
 630   // Find Results to replace by whole result
 631   QList<GeomShapePtr> aShapes;
 632   QList<ObjectPtr> aToReplace;
 633   std::list<GeomShapePtr> aSubShapes;
 634   foreach(ObjectPtr aObj, myResShapes.keys()) {
 635     aShapes = myResShapes[aObj];
 636     ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
 637     TopTools_MapOfShape aShapesMap;
 638     if (aRes.get()) {
 639       GeomShapePtr aSubShape = aRes->shape();
 640       const TopoDS_Shape& aShape = aSubShape->impl<TopoDS_Shape>();
 641       for (TopExp_Explorer anExp(aShape, (TopAbs_ShapeEnum)theShapeType);
 642         anExp.More(); anExp.Next()) {
 643         aShapesMap.Add(anExp.Current());
 644       }
 645     }
 646     if (aShapes.count() == aShapesMap.Size())
 647       aToReplace.append(aObj);
 648   }
 649   // Replace the found results
 650   QList<ModuleBase_ViewerPrsPtr>::iterator aIt;
 651   foreach(ObjectPtr aObj, aToReplace) {
 652     for (aIt = theResults.begin(); aIt != theResults.end(); aIt++) {
 653       if ((*aIt)->object() == aObj) {
 654         theResults.removeAll(*aIt);
 655         aIt--;
 656       }
 657     }
 658     ModuleBase_ViewerPrsPtr aValue(new ModuleBase_ViewerPrs(aObj));
 659     theResults.append(aValue);
 660   }
 661 }
 662
 663 void ModuleBase_WidgetSelectionFilter::onFeatureAccepted()
 664 {
 665   AttributePtr aAttr = mySelectorFeature->attribute(mySelectorAttribute);
 666   AttributeSelectionListPtr aSelListAttr =
 667     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(aAttr);
 668   aSelListAttr->clear();
 669   if (aSelListAttr->isWholeResultAllowed())
 670     replaceSubShapesByResult(myValues, selectionType(aSelListAttr->selectionType().c_str()));
 671   foreach(ModuleBase_ViewerPrsPtr aPrs, myValues) {
 672     aSelListAttr->append(aPrs->object(), aPrs->shape());
 673   }
 674 }
 675
 676 bool ModuleBase_WidgetSelectionFilter::storeValueCustom()
 677 {
 678   ModuleBase_ModelWidget* aActive = myWorkshop->propertyPanel()->activeWidget();
 679   if (aActive)
 680     return aActive->storeValue();
 681   updateObject(myFeature);
 682   return true;
 683 }
 684
 685 QList<ModuleBase_FilterItem*> ModuleBase_WidgetSelectionFilter::itemsList() const
 686 {
 687   return  myFiltersWgt->findChildren<ModuleBase_FilterItem*>();
 688 }
 689
 690
 691 bool ModuleBase_WidgetSelectionFilter::restoreValueCustom()
 692 {
 693   ModelAPI_FiltersFactory* aFactory = ModelAPI_Session::get()->filters();
 694   FiltersFeaturePtr aFiltersFeature = std::dynamic_pointer_cast<ModelAPI_FiltersFeature>(myFeature);
 695
 696   // Init filters member of the parent attribute
 697   AttributeSelectionListPtr aAttrList = mySelectorFeature->selectionList(mySelectorAttribute);
 698   if (aAttrList->filters() != aFiltersFeature) {
 699     aAttrList->setFilters(aFiltersFeature);
 700   }
 701
 702   QList<ModuleBase_FilterItem*> aItemsList = itemsList();
 703   std::list<std::string> aFilters = aFiltersFeature->filters();
 704
 705   std::list<std::string>::const_iterator aIt;
 706   int i = 0;
 707   int aNbItems = aItemsList.size();
 708   ModuleBase_FilterItem* aItem = 0;
 709   bool isBlocked = myFiltersCombo->blockSignals(true);
 710   for (aIt = aFilters.cbegin(); aIt != aFilters.cend(); aIt++, i++) {
 711     std::string aStr = (*aIt);
 712     aItem = 0;
 713     if (i >= aNbItems) {
 714       aItem = onAddFilter(aStr);
 715       FilterPtr aFilterObj = aFactory->filter(aStr);
 716       int aId = myFiltersCombo->findText(aFilterObj->name().c_str());
 717       if ((aId != -1) && !aFilterObj->isMultiple())
 718         myFiltersCombo->removeItem(aId);
 719       if (aItem) {
 720         QList<ModuleBase_ModelWidget*> aSubList = aItem->widgets();
 721         foreach(ModuleBase_ModelWidget* aWgt, aSubList) {
 722           aWgt->restoreValue();
 723         }
 724       }
 725     }
 726   }
 727   myFiltersCombo->setCurrentIndex(0);
 728   myFiltersCombo->blockSignals(isBlocked);
 729   return true;
 730 }
 731
 732 QString ModuleBase_WidgetSelectionFilter::getError(const bool theValueStateChecked) const
 733 {
 734   QString aErrorMsg = ModuleBase_ModelWidget::getError(theValueStateChecked);
 735   if (aErrorMsg.isEmpty()) {
 736     if (myValues.size() == 0)
 737       aErrorMsg = translate("Selection is empty");
 738   }
 739   return aErrorMsg;
 740 }
 741
 742 void ModuleBase_WidgetSelectionFilter::onObjectUpdated()
 743 {
 744   myShowBtn->setChecked(false);
 745   clearCurrentSelection(true);
 746   updateNumberSelected();
 747
 748   QList<ModuleBase_FilterItem*> aItemsList = itemsList();
 749   foreach(ModuleBase_FilterItem* aItem, aItemsList) {
 750     QList<ModuleBase_ModelWidget*> aWidgetsList = aItem->widgets();
 751     foreach(ModuleBase_ModelWidget* aWidget, aWidgetsList) {
 752       if (!aWidget->feature().get())
 753         aWidget->setFeature(myFeature);
 754       aWidget->restoreValue();
 755     }
 756   }
 757   updateObject(myFeature);
 758
 759   // Redisplay the feature on order to Customize presentations from filters with selectors
 760   static Events_ID EVENT_DISP = Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY);
 761   ModelAPI_EventCreator::get()->sendUpdated(myFeature, EVENT_DISP);
 762   Events_Loop::loop()->flush(EVENT_DISP);
 763 }
 764
 765 void ModuleBase_WidgetSelectionFilter::storeFilters(const std::list<FilterPtr>& theFilters)
 766 {
 767   for (std::list<FilterPtr>::const_iterator anIt = theFilters.begin();
 768        anIt != theFilters.end(); ++anIt) {
 769     std::string aName = translate((*anIt)->name()).toStdString();
 770     myFilters[aName] = *anIt;
 771   }
 772 }
 773
 774 QString ModuleBase_WidgetSelectionFilter::translate(const std::string& theString) const
 775 {
 776   return ModuleBase_Tools::translate(myFeatureId, theString);
 777 }