Salome HOME
Switch to the newer compiled Linux SALOME environment.
[modules/shaper.git] / src / PartSet / PartSet_Filters.cpp
1 // Copyright (C) 2014-2017  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
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "PartSet_Filters.h"
22
23 #include <ModuleBase_IWorkshop.h>
24 #include "ModuleBase_IModule.h"
25 #include <ModuleBase_ResultPrs.h>
26
27 #include <ModelAPI_Feature.h>
28 #include <ModelAPI_ResultPart.h>
29 #include <ModelAPI_Session.h>
30
31 #include <AIS_InteractiveObject.hxx>
32 #include <AIS_Shape.hxx>
33 #include <StdSelect_BRepOwner.hxx>
34
35
36 IMPLEMENT_STANDARD_RTTIEXT(PartSet_GlobalFilter, ModuleBase_ShapeDocumentFilter);
37
38 Standard_Boolean PartSet_GlobalFilter::IsOk(const Handle(SelectMgr_EntityOwner)& theOwner) const
39 {
40   bool aValid = true;
41   ModuleBase_Operation* anOperation = myWorkshop->module()->currentOperation();
42   // the shapes from different documents should be provided if there is no started operation
43   // in order to show/hide results
44   if (anOperation) {
45     aValid = false;
46     if (ModuleBase_ShapeDocumentFilter::IsOk(theOwner)) {
47       std::shared_ptr<GeomAPI_AISObject> aAISObj = AISObjectPtr(new GeomAPI_AISObject());
48       if (theOwner->HasSelectable()) {
49         Handle(AIS_InteractiveObject) aAisObj =
50           Handle(AIS_InteractiveObject)::DownCast(theOwner->Selectable());
51         if (!aAisObj.IsNull()) {
52           aAISObj->setImpl(new Handle(AIS_InteractiveObject)(aAisObj));
53         }
54       }
55       ObjectPtr aObj = myWorkshop->findPresentedObject(aAISObj);
56       if (aObj) {
57         ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
58         // result of parts belongs to PartSet document and can be selected only when PartSet
59         //  is active in order to do not select the result of the active part.
60         if (aResult.get() && aResult->groupName() == ModelAPI_ResultPart::group()) {
61           SessionPtr aMgr = ModelAPI_Session::get();
62           aValid = aMgr->activeDocument() == aMgr->moduleDocument();
63         }
64         else {
65           FeaturePtr aFeature = ModelAPI_Feature::feature(aObj);
66           if (aFeature) {
67             aValid = aFeature->getKind() != "Group";
68           } else
69             aValid = Standard_True;
70         }
71       } else
72         // This is not object controlled by the filter
73         aValid = Standard_True;
74     }
75   }
76 #ifdef DEBUG_FILTERS
77   qDebug(QString("PartSet_GlobalFilter::IsOk = %1").arg(aValid).toStdString().c_str());
78 #endif
79   return aValid;
80 }
81
82 IMPLEMENT_STANDARD_RTTIEXT(PartSet_CirclePointFilter, SelectMgr_Filter);
83
84 Standard_Boolean
85   PartSet_CirclePointFilter::IsOk(const Handle(SelectMgr_EntityOwner)& theOwner) const
86 {
87   ModuleBase_Operation* anOperation = myWorkshop->module()->currentOperation();
88   if(!anOperation) {
89     return Standard_True;
90   }
91
92   if(theOwner->HasSelectable() == Standard_False) {
93     return Standard_True;
94   }
95
96   Handle(StdSelect_BRepOwner) aBrepOwner = Handle(StdSelect_BRepOwner)::DownCast(theOwner);
97   if(aBrepOwner.IsNull()) {
98     return Standard_True;
99   }
100
101   const TopoDS_Shape& aShape = aBrepOwner->Shape();
102   if(aShape.IsNull() || aShape.ShapeType() != TopAbs_VERTEX) {
103     return Standard_True;
104   }
105
106   Handle(ModuleBase_ResultPrs) aResultPrs =
107     Handle(ModuleBase_ResultPrs)::DownCast(theOwner->Selectable());
108   if(aResultPrs.IsNull()) {
109     return Standard_True;
110   }
111
112   std::shared_ptr<GeomAPI_AISObject> aGeomAISObj(new GeomAPI_AISObject());
113   aGeomAISObj->setImpl(new Handle(AIS_InteractiveObject)(aResultPrs));
114   ObjectPtr anObj = myWorkshop->findPresentedObject(aGeomAISObj);
115   if(!anObj.get()) {
116     return Standard_True;
117   }
118
119   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObj);
120   if(!aResult.get()) {
121     return Standard_True;
122   }
123
124   DocumentPtr aDocument = aResult->document();
125   if(!aDocument.get()) {
126     return Standard_True;
127   }
128
129   FeaturePtr aFeature = aDocument->feature(aResult);
130   if(!aFeature.get() || aFeature->getKind() != "SketchCircle") {
131     return Standard_True;
132   }
133
134   const TopoDS_Shape anOwnerShape = aResultPrs->Shape();
135   if(anOwnerShape.ShapeType() == TopAbs_EDGE) {
136     return Standard_False;
137   }
138
139 #ifdef DEBUG_FILTERS
140   qDebug(QString("PartSet_ShapeDocumentFilter::IsOk = %1").arg(aValid).toStdString().c_str());
141 #endif
142
143   return Standard_True;
144 }