Salome HOME
*** empty log message ***
[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;
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()));
31     if (!obj) return false;
32
33     _PTR(SObject) objFather = obj->GetFather();
34     _PTR(SComponent) objComponent = obj->GetFatherComponent();
35
36     if( objComponent->ComponentDataType()!="SMESH" )
37       return false;
38
39     int aLevel = obj->Depth() - objComponent->Depth();
40
41     // Max level under the component is 4:
42     //
43     // 0    Mesh Component
44     // 1    |- Hypotheses
45     // 2    |  |- Regular 1D
46     //      |- Algorithms
47     //      |- Mesh 1
48     //         |- Applied Hypotheses
49     //         |- Applied Algorithms
50     //         |- Submeshes on Face
51     // 3       |  |- SubmeshFace
52     // 4       |     |- Applied algorithms ( selectable in Use Case Browser )
53     //         |- Group Of Nodes
54
55     if (aLevel <= 0)
56       return false;
57
58     switch (myType)
59       {
60       case HYPOTHESIS:
61         {
62           if ( aLevel == 2 && ( objFather->Tag() == 1 ))
63             Ok = true;
64           break;
65         }
66       case ALGORITHM:
67         {
68           if ( aLevel == 2 && ( objFather->Tag() == 2 ))
69             Ok = true;
70           break;
71         }
72       case MESH:
73         {
74           if ( aLevel == 1 && ( obj->Tag() >= 3 ))
75             Ok = true;
76           break;
77         }
78       case SUBMESH:
79         {
80           // see SMESH_Gen_i.cxx for tag numbers
81           if ( aLevel == 3 && ( objFather->Tag() >= 4 && objFather->Tag() <= 10 ))
82             Ok = true;
83           break;
84         }
85       case MESHorSUBMESH:
86         {
87           if ( aLevel == 1 && ( obj->Tag() >= 3 ))
88             Ok = true; // mesh
89           else if ( aLevel == 3 && ( objFather->Tag() >= 4 && objFather->Tag() <= 10 ))
90             Ok = true;
91           break;
92         }
93       case SUBMESH_VERTEX:  // Label "SubMeshes on vertexes"
94         {
95           if ( aLevel == 3 && ( objFather->Tag() == 4 ))
96             Ok = true;
97           break;
98         }
99       case SUBMESH_EDGE:
100         {
101           if ( aLevel == 3 && ( objFather->Tag() == 5 ))
102             Ok = true;
103           break;
104         }
105       case SUBMESH_FACE:
106         {
107           if ( aLevel == 3 && ( objFather->Tag() == 7 ))
108             Ok = true;
109           break;
110         }
111       case SUBMESH_SOLID:
112         {
113           if ( aLevel == 3 && ( objFather->Tag() == 9 ))
114             Ok = true;
115           break;
116         }
117       case SUBMESH_COMPOUND:
118         {
119           if ( aLevel == 3 && ( objFather->Tag() == 10 ))
120             Ok = true;
121           break;
122         }
123       case GROUP:
124         {
125           if ( aLevel == 3 && ( objFather->Tag() > 10 ))
126             Ok = true;
127           break;
128         }
129       }
130   }
131   return Ok;
132 }
133
134 MeshObjectType SMESH_TypeFilter::type() const 
135 {
136   return myType;
137 }