src/SMESHFiltersSelection/SMESH_TypeFilter.cxx

   1 // Copyright (C) 2007-2014  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include "SMESH_TypeFilter.hxx"
  24
  25 #include <SUIT_Session.h>
  26
  27 #include <SalomeApp_Study.h>
  28 #include <LightApp_DataOwner.h>
  29
  30 #include <SALOMEconfig.h>
  31 #include CORBA_CLIENT_HEADER(SMESH_Gen)
  32
  33 SMESH_TypeFilter::SMESH_TypeFilter (SMESH::MeshObjectType theType)
  34 {
  35   myType = theType;
  36 }
  37
  38 SMESH_TypeFilter::~SMESH_TypeFilter()
  39 {
  40 }
  41
  42 bool SMESH_TypeFilter::isOk (const SUIT_DataOwner* theDataOwner) const
  43 {
  44   bool Ok = false, extractReference = true;
  45
  46   const LightApp_DataOwner* owner =
  47     dynamic_cast<const LightApp_DataOwner*>(theDataOwner);
  48   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>
  49     (SUIT_Session::session()->activeApplication()->activeStudy());
  50
  51   if (owner && appStudy) {
  52     _PTR(Study) study = appStudy->studyDS();
  53     QString entry = owner->entry();
  54
  55     _PTR(SObject) obj (study->FindObjectID(entry.toLatin1().data())), aRefSO;
  56     if( extractReference && obj && obj->ReferencedObject( aRefSO ) )
  57       obj = aRefSO;
  58     if (!obj) return false;
  59
  60     _PTR(SObject) objFather = obj->GetFather();
  61     _PTR(SComponent) objComponent = obj->GetFatherComponent();
  62
  63     if( objComponent->ComponentDataType()!="SMESH" )
  64       return false;
  65
  66     int aLevel = obj->Depth() - objComponent->Depth();
  67
  68     // Max level under the component is 5:
  69     //
  70     // 0    Mesh Component
  71     // 1    |- Hypotheses
  72     // 2    |  |- Regular 1D
  73     //      |- Algorithms
  74     //      |- Mesh 1
  75     //         |- * Main Shape
  76     //         |- Applied Hypotheses
  77     //         |- Applied Algorithms
  78     //         |- Submeshes on Face
  79     // 3       |  |- SubmeshFace
  80     // 4       |     |- * Face 1
  81     // 4       |     |- Applied algorithms ( selectable in Use Case Browser )
  82     // 5       |          |- Regular 1D
  83     //         |- Group Of Nodes
  84     //            |- Group 1
  85
  86     if (aLevel <= 0)
  87       return false;
  88
  89     switch (myType)
  90     {
  91       case SMESH::HYPOTHESIS:
  92         {
  93           if      (aLevel == 2 && (objFather->Tag() == SMESH::Tag_HypothesisRoot))
  94             // hypo definition
  95             Ok = true;
  96           else if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_RefOnAppliedHypothesis))
  97             // applied global hypo
  98             Ok = true;
  99           else if (aLevel == 5 && (objFather->Tag() == SMESH::Tag_RefOnAppliedHypothesis))
 100             // applied local hypo
 101             Ok = true;
 102           break;
 103         }
 104       case SMESH::ALGORITHM:
 105         {
 106           if      (aLevel == 2 && (objFather->Tag() == SMESH::Tag_AlgorithmsRoot))
 107             // algo definition
 108             Ok = true;
 109           else if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_RefOnAppliedAlgorithms))
 110             // applied global algo
 111             Ok = true;
 112           else if (aLevel == 5 && (objFather->Tag() == SMESH::Tag_RefOnAppliedAlgorithms))
 113             // applied local algo
 114             Ok = true;
 115           break;
 116         }
 117       case SMESH::MESH:
 118         {
 119           if (aLevel == 1 && (obj->Tag() >= SMESH::Tag_FirstMeshRoot))
 120             Ok = true;
 121           break;
 122         }
 123       case SMESH::SUBMESH:
 124         {
 125           // see SMESH_Gen_i.cxx for tag numbers
 126           if (aLevel == 3 && (objFather->Tag() >= SMESH::Tag_FirstSubMesh &&
 127                               objFather->Tag() <= SMESH::Tag_LastSubMesh))
 128             Ok = true;
 129           break;
 130         }
 131       case SMESH::MESHorSUBMESH:
 132         {
 133           if (aLevel == 1 && (obj->Tag() >= SMESH::Tag_FirstMeshRoot))
 134             Ok = true; // mesh
 135           else if (aLevel == 3 && (objFather->Tag() >= SMESH::Tag_FirstSubMesh &&
 136                                    objFather->Tag() <= SMESH::Tag_LastSubMesh))
 137             Ok = true;
 138           break;
 139         }
 140       case SMESH::SUBMESH_VERTEX: // Label "SubMeshes on vertexes"
 141         {
 142           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_SubMeshOnVertex))
 143             Ok = true;
 144           break;
 145         }
 146       case SMESH::SUBMESH_EDGE:
 147         {
 148           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_SubMeshOnEdge))
 149             Ok = true;
 150           break;
 151         }
 152       case SMESH::SUBMESH_FACE:
 153         {
 154           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_SubMeshOnFace))
 155             Ok = true;
 156           break;
 157         }
 158       case SMESH::SUBMESH_SOLID:
 159         {
 160           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_SubMeshOnSolid))
 161             Ok = true;
 162           break;
 163         }
 164       case SMESH::SUBMESH_COMPOUND:
 165         {
 166           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_SubMeshOnCompound))
 167             Ok = true;
 168           break;
 169         }
 170       case SMESH::GROUP:
 171         {
 172           if (aLevel == 3 && (objFather->Tag() >= SMESH::Tag_FirstGroup))
 173             Ok = true;
 174           break;
 175         }
 176       case SMESH::GROUP_NODE:
 177         {
 178           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_NodeGroups))
 179             Ok = true;
 180           break;
 181         }
 182       case SMESH::GROUP_EDGE:
 183         {
 184           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_EdgeGroups))
 185             Ok = true;
 186           break;
 187         }
 188       case SMESH::GROUP_FACE:
 189         {
 190           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_FaceGroups))
 191             Ok = true;
 192           break;
 193         }
 194       case SMESH::GROUP_VOLUME:
 195         {
 196           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_VolumeGroups))
 197             Ok = true;
 198           break;
 199         }
 200       case SMESH::GROUP_0D:
 201         {
 202           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_0DElementsGroups))
 203             Ok = true;
 204           break;
 205         }
 206       case SMESH::GROUP_BALL:
 207         {
 208           if (aLevel == 3 && (objFather->Tag() == SMESH::Tag_BallElementsGroups))
 209             Ok = true;
 210           break;
 211         }
 212       case SMESH::IDSOURCE:
 213         {
 214           Ok = ( SMESH_TypeFilter(SMESH::MESHorSUBMESH).isOk( theDataOwner ) ||
 215                  SMESH_TypeFilter(SMESH::GROUP)        .isOk( theDataOwner ));
 216           break;
 217         }
 218     }
 219   }
 220   return Ok;
 221 }
 222
 223 SMESH::MeshObjectType SMESH_TypeFilter::type() const
 224 {
 225   return myType;
 226 }