src/SMESHFiltersSelection/SMESH_TypeFilter.cxx

   1 // Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   3 //
   4 // This library is free software; you can redistribute it and/or
   5 // modify it under the terms of the GNU Lesser General Public
   6 // License as published by the Free Software Foundation; either
   7 // version 2.1 of the License.
   8 //
   9 // This library is distributed in the hope that it will be useful
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 // Lesser General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Lesser General Public
  15 // License along with this library; if not, write to the Free Software
  16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 //
  18 // See http://www.salome-platform.org/
  19 //
  20 #include "SMESH_TypeFilter.hxx"
  21
  22 #include <SUIT_Session.h>
  23
  24 #include <SalomeApp_Study.h>
  25 #include <LightApp_DataOwner.h>
  26
  27 SMESH_TypeFilter::SMESH_TypeFilter (MeshObjectType theType)
  28 {
  29   myType = theType;
  30 }
  31
  32 SMESH_TypeFilter::~SMESH_TypeFilter()
  33 {
  34 }
  35
  36 bool SMESH_TypeFilter::isOk (const SUIT_DataOwner* theDataOwner) const
  37 {
  38   bool Ok = false, extractReference = true;
  39
  40   const LightApp_DataOwner* owner =
  41     dynamic_cast<const LightApp_DataOwner*>(theDataOwner);
  42   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>
  43     (SUIT_Session::session()->activeApplication()->activeStudy());
  44
  45   if (owner && appStudy) {
  46     _PTR(Study) study = appStudy->studyDS();
  47     QString entry = owner->entry();
  48
  49     _PTR(SObject) obj (study->FindObjectID(entry.latin1())), aRefSO;
  50     if( extractReference && obj && obj->ReferencedObject( aRefSO ) )
  51       obj = aRefSO;
  52     if (!obj) return false;
  53
  54     _PTR(SObject) objFather = obj->GetFather();
  55     _PTR(SComponent) objComponent = obj->GetFatherComponent();
  56
  57     if( objComponent->ComponentDataType()!="SMESH" )
  58       return false;
  59
  60     int aLevel = obj->Depth() - objComponent->Depth();
  61
  62     // Max level under the component is 5:
  63     //
  64     // 0    Mesh Component
  65     // 1    |- Hypotheses
  66     // 2    |  |- Regular 1D
  67     //      |- Algorithms
  68     //      |- Mesh 1
  69     //         |- * Main Shape
  70     //         |- Applied Hypotheses
  71     //         |- Applied Algorithms
  72     //         |- Submeshes on Face
  73     // 3       |  |- SubmeshFace
  74     // 4       |     |- * Face 1
  75     // 4       |     |- Applied algorithms ( selectable in Use Case Browser )
  76     // 5       |          |- Regular 1D
  77     //         |- Group Of Nodes
  78
  79     if (aLevel <= 0)
  80       return false;
  81
  82     switch (myType)
  83       {
  84       case HYPOTHESIS:
  85         {
  86           if ( aLevel == 2 && ( objFather->Tag() == 1 )) // hypo definition
  87             Ok = true;
  88           else if ( aLevel == 3 && ( objFather->Tag() == 2 )) // applied global hypo
  89             Ok = true;
  90           else if ( aLevel == 5 && ( objFather->Tag() == 2 )) // applied local hypo
  91             Ok = true;
  92           break;
  93         }
  94       case ALGORITHM:
  95         {
  96           if ( aLevel == 2 && ( objFather->Tag() == 2 )) // algo definition
  97             Ok = true;
  98           else if ( aLevel == 3 && ( objFather->Tag() == 3 )) // applied global algo
  99             Ok = true;
 100           else if ( aLevel == 5 && ( objFather->Tag() == 3 )) // applied local algo
 101             Ok = true;
 102           break;
 103         }
 104       case MESH:
 105         {
 106           if ( aLevel == 1 && ( obj->Tag() >= 3 ))
 107             Ok = true;
 108           break;
 109         }
 110       case SUBMESH:
 111         {
 112           // see SMESH_Gen_i.cxx for tag numbers
 113           if ( aLevel == 3 && ( objFather->Tag() >= 4 && objFather->Tag() <= 10 ))
 114             Ok = true;
 115           break;
 116         }
 117       case MESHorSUBMESH:
 118         {
 119           if ( aLevel == 1 && ( obj->Tag() >= 3 ))
 120             Ok = true; // mesh
 121           else if ( aLevel == 3 && ( objFather->Tag() >= 4 && objFather->Tag() <= 10 ))
 122             Ok = true;
 123           break;
 124         }
 125       case SUBMESH_VERTEX:  // Label "SubMeshes on vertexes"
 126         {
 127           if ( aLevel == 3 && ( objFather->Tag() == 4 ))
 128             Ok = true;
 129           break;
 130         }
 131       case SUBMESH_EDGE:
 132         {
 133           if ( aLevel == 3 && ( objFather->Tag() == 5 ))
 134             Ok = true;
 135           break;
 136         }
 137       case SUBMESH_FACE:
 138         {
 139           if ( aLevel == 3 && ( objFather->Tag() == 7 ))
 140             Ok = true;
 141           break;
 142         }
 143       case SUBMESH_SOLID:
 144         {
 145           if ( aLevel == 3 && ( objFather->Tag() == 9 ))
 146             Ok = true;
 147           break;
 148         }
 149       case SUBMESH_COMPOUND:
 150         {
 151           if ( aLevel == 3 && ( objFather->Tag() == 10 ))
 152             Ok = true;
 153           break;
 154         }
 155       case GROUP:
 156         {
 157           if ( aLevel == 3 && ( objFather->Tag() > 10 ))
 158             Ok = true;
 159           break;
 160         }
 161       }
 162   }
 163   return Ok;
 164 }
 165
 166 MeshObjectType SMESH_TypeFilter::type() const
 167 {
 168   return myType;
 169 }