src/SMESHFiltersSelection/SMESH_TypeFilter.cxx

   1 #include "SMESH_TypeFilter.hxx"
   2
   3 #include <SUIT_Session.h>
   4
   5 #include <SalomeApp_Study.h>
   6 #include <LightApp_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 LightApp_DataOwner* owner =
  22     dynamic_cast<const LightApp_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 }