Salome HOME
PAL10397 - Selection isn't cleared after apply
[modules/smesh.git] / src / SMESHFiltersSelection / SMESH_TypeFilter.cxx
1 #include "SMESH_TypeFilter.hxx"
2
3 #include <SUIT_Session.h>
4
5 #include <SalomeApp_Study.h>
6 #include <SalomeApp_DataOwner.h>
7
8 SMESH_TypeFilter::SMESH_TypeFilter (MeshObjectType theType) 
9 {
10   myType = theType;
11 }
12
13 SMESH_TypeFilter::~SMESH_TypeFilter() 
14 {
15 }
16
17 bool SMESH_TypeFilter::isOk (const SUIT_DataOwner* theDataOwner) const
18 {
19   bool Ok = false, extractReference = true;
20
21   const SalomeApp_DataOwner* owner =
22     dynamic_cast<const SalomeApp_DataOwner*>(theDataOwner);
23   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>
24     (SUIT_Session::session()->activeApplication()->activeStudy());
25
26   if (owner && appStudy) {
27     _PTR(Study) study = appStudy->studyDS();
28     QString entry = owner->entry();
29
30     _PTR(SObject) obj (study->FindObjectID(entry.latin1())), aRefSO;
31     if( extractReference && obj && obj->ReferencedObject( aRefSO ) )
32       obj = aRefSO;
33     if (!obj) return false;
34
35     _PTR(SObject) objFather = obj->GetFather();
36     _PTR(SComponent) objComponent = obj->GetFatherComponent();
37
38     if( objComponent->ComponentDataType()!="SMESH" )
39       return false;
40
41     int aLevel = obj->Depth() - objComponent->Depth();
42
43     // Max level under the component is 5:
44     //
45     // 0    Mesh Component
46     // 1    |- Hypotheses
47     // 2    |  |- Regular 1D
48     //      |- Algorithms
49     //      |- Mesh 1
50     //         |- * Main Shape
51     //         |- Applied Hypotheses
52     //         |- Applied Algorithms
53     //         |- Submeshes on Face
54     // 3       |  |- SubmeshFace
55     // 4       |     |- * Face 1
56     // 4       |     |- Applied algorithms ( selectable in Use Case Browser )
57     // 5       |          |- Regular 1D
58     //         |- Group Of Nodes
59
60     if (aLevel <= 0)
61       return false;
62
63     switch (myType)
64       {
65       case HYPOTHESIS:
66         {
67           if ( aLevel == 2 && ( objFather->Tag() == 1 )) // hypo definition
68             Ok = true;
69           else if ( aLevel == 3 && ( objFather->Tag() == 2 )) // applied global hypo
70             Ok = true;
71           else if ( aLevel == 5 && ( objFather->Tag() == 2 )) // applied local hypo
72             Ok = true;
73           break;
74         }
75       case ALGORITHM:
76         {
77           if ( aLevel == 2 && ( objFather->Tag() == 2 )) // algo definition
78             Ok = true;
79           else if ( aLevel == 3 && ( objFather->Tag() == 3 )) // applied global algo
80             Ok = true;
81           else if ( aLevel == 5 && ( objFather->Tag() == 3 )) // applied local algo
82             Ok = true;
83           break;
84         }
85       case MESH:
86         {
87           if ( aLevel == 1 && ( obj->Tag() >= 3 ))
88             Ok = true;
89           break;
90         }
91       case SUBMESH:
92         {
93           // see SMESH_Gen_i.cxx for tag numbers
94           if ( aLevel == 3 && ( objFather->Tag() >= 4 && objFather->Tag() <= 10 ))
95             Ok = true;
96           break;
97         }
98       case MESHorSUBMESH:
99         {
100           if ( aLevel == 1 && ( obj->Tag() >= 3 ))
101             Ok = true; // mesh
102           else if ( aLevel == 3 && ( objFather->Tag() >= 4 && objFather->Tag() <= 10 ))
103             Ok = true;
104           break;
105         }
106       case SUBMESH_VERTEX:  // Label "SubMeshes on vertexes"
107         {
108           if ( aLevel == 3 && ( objFather->Tag() == 4 ))
109             Ok = true;
110           break;
111         }
112       case SUBMESH_EDGE:
113         {
114           if ( aLevel == 3 && ( objFather->Tag() == 5 ))
115             Ok = true;
116           break;
117         }
118       case SUBMESH_FACE:
119         {
120           if ( aLevel == 3 && ( objFather->Tag() == 7 ))
121             Ok = true;
122           break;
123         }
124       case SUBMESH_SOLID:
125         {
126           if ( aLevel == 3 && ( objFather->Tag() == 9 ))
127             Ok = true;
128           break;
129         }
130       case SUBMESH_COMPOUND:
131         {
132           if ( aLevel == 3 && ( objFather->Tag() == 10 ))
133             Ok = true;
134           break;
135         }
136       case GROUP:
137         {
138           if ( aLevel == 3 && ( objFather->Tag() > 10 ))
139             Ok = true;
140           break;
141         }
142       }
143   }
144   return Ok;
145 }
146
147 MeshObjectType SMESH_TypeFilter::type() const 
148 {
149   return myType;
150 }