1 // Copyright (C) 2007-2008 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 // SMESH SMESH_I : idl implementation based on 'SMESH' unit's classes
23 // File : SMESH_Group_i.cxx
24 // Author : Sergey ANIKIN, OCC
28 #include "SMESH_Group_i.hxx"
29 #include "SMESH_Mesh_i.hxx"
30 #include "SMESH_Gen_i.hxx"
31 #include "SMESH_Group.hxx"
32 #include "SMESHDS_Group.hxx"
33 #include "SMESHDS_GroupOnGeom.hxx"
34 #include "SMDSAbs_ElementType.hxx"
36 #include "SMESH_Filter_i.hxx"
37 #include "SMESH_PythonDump.hxx"
39 #include "utilities.h"
41 using namespace SMESH;
43 //=============================================================================
47 //=============================================================================
49 SMESH_GroupBase_i::SMESH_GroupBase_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
50 : SALOME::GenericObj_i( thePOA ),
51 myMeshServant( theMeshServant ),
52 myLocalID( theLocalID )
54 // PAL7962: san -- To ensure correct mapping of servant and correct reference counting in GenericObj_i,
55 // servant activation is performed by SMESH_Mesh_i::createGroup()
56 // thePOA->activate_object( this );
59 SMESH_Group_i::SMESH_Group_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
60 : SALOME::GenericObj_i( thePOA ),
61 SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
63 //MESSAGE("SMESH_Group_i; this = "<<this );
66 SMESH_GroupOnGeom_i::SMESH_GroupOnGeom_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
67 : SALOME::GenericObj_i( thePOA ),
68 SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
70 //MESSAGE("SMESH_GroupOnGeom_i; this = "<<this );
73 //=============================================================================
77 //=============================================================================
79 SMESH_GroupBase_i::~SMESH_GroupBase_i()
81 MESSAGE("~SMESH_GroupBase_i; this = "<<this );
83 myMeshServant->removeGroup(myLocalID);
86 //=======================================================================
87 //function : GetSmeshGroup
89 //=======================================================================
91 ::SMESH_Group* SMESH_GroupBase_i::GetSmeshGroup() const
93 if ( myMeshServant ) {
94 ::SMESH_Mesh& aMesh = myMeshServant->GetImpl();
95 return aMesh.GetGroup(myLocalID);
100 //=======================================================================
101 //function : GetGroupDS
103 //=======================================================================
105 SMESHDS_GroupBase* SMESH_GroupBase_i::GetGroupDS() const
107 ::SMESH_Group* aGroup = GetSmeshGroup();
109 return aGroup->GetGroupDS();
113 //=============================================================================
117 //=============================================================================
119 void SMESH_GroupBase_i::SetName( const char* theName )
121 // Update Python script
122 TPythonDump() << _this() << ".SetName( '" << theName << "' )";
125 ::SMESH_Group* aGroup = GetSmeshGroup();
127 aGroup->SetName(theName);
129 // Update group name in a study
130 SMESH_Gen_i* aGen = myMeshServant->GetGen();
131 aGen->SetName( aGen->ObjectToSObject( aGen->GetCurrentStudy(), _this() ), theName );
134 MESSAGE("can't set name of a vague group");
137 //=============================================================================
141 //=============================================================================
143 char* SMESH_GroupBase_i::GetName()
145 ::SMESH_Group* aGroup = GetSmeshGroup();
147 return CORBA::string_dup (aGroup->GetName());
148 MESSAGE("get name of a vague group");
149 return CORBA::string_dup( "NO_NAME" );
152 //=============================================================================
156 //=============================================================================
158 SMESH::ElementType SMESH_GroupBase_i::GetType()
160 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
162 SMDSAbs_ElementType aSMDSType = aGroupDS->GetType();
163 SMESH::ElementType aType;
165 case SMDSAbs_Node: aType = SMESH::NODE; break;
166 case SMDSAbs_Edge: aType = SMESH::EDGE; break;
167 case SMDSAbs_Face: aType = SMESH::FACE; break;
168 case SMDSAbs_Volume: aType = SMESH::VOLUME; break;
169 default: aType = SMESH::ALL; break;
173 MESSAGE("get type of a vague group");
178 //=============================================================================
182 //=============================================================================
184 CORBA::Long SMESH_GroupBase_i::Size()
186 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
188 return aGroupDS->Extent();
189 MESSAGE("get size of a vague group");
193 //=============================================================================
197 //=============================================================================
199 CORBA::Boolean SMESH_GroupBase_i::IsEmpty()
201 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
203 return aGroupDS->IsEmpty();
204 MESSAGE("checking IsEmpty of a vague group");
208 //=============================================================================
212 //=============================================================================
214 void SMESH_Group_i::Clear()
216 // Update Python script
217 TPythonDump() << _this() << ".Clear()";
220 SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
225 MESSAGE("attempt to clear a vague group");
228 //=============================================================================
232 //=============================================================================
234 CORBA::Boolean SMESH_GroupBase_i::Contains( CORBA::Long theID )
236 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
238 return aGroupDS->Contains(theID);
239 MESSAGE("attempt to check contents of a vague group");
243 //=============================================================================
247 //=============================================================================
249 CORBA::Long SMESH_Group_i::Add( const SMESH::long_array& theIDs )
251 // Update Python script
252 TPythonDump() << "nbAdd = " << _this() << ".Add( " << theIDs << " )";
254 // Add elements to the group
255 SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
258 for (int i = 0; i < theIDs.length(); i++) {
259 int anID = (int) theIDs[i];
260 if (aGroupDS->Add(anID))
265 MESSAGE("attempt to add elements to a vague group");
269 //=============================================================================
273 //=============================================================================
275 CORBA::Long SMESH_Group_i::Remove( const SMESH::long_array& theIDs )
277 // Update Python script
278 TPythonDump() << "nbDel = " << _this() << ".Remove( " << theIDs << " )";
280 // Remove elements from the group
281 SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
284 for (int i = 0; i < theIDs.length(); i++) {
285 int anID = (int) theIDs[i];
286 if (aGroupDS->Remove(anID))
291 MESSAGE("attempt to remove elements from a vague group");
295 //=============================================================================
299 //=============================================================================
301 typedef bool (SMESHDS_Group::*TFunChangeGroup)(const int);
304 ChangeByPredicate( SMESH::Predicate_i* thePredicate,
305 SMESHDS_GroupBase* theGroupBase,
306 TFunChangeGroup theFun)
309 if(SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>(theGroupBase)){
310 SMESH::Controls::Filter::TIdSequence aSequence;
311 const SMDS_Mesh* aMesh = theGroupBase->GetMesh();
312 SMESH::Filter_i::GetElementsId(thePredicate,aMesh,aSequence);
314 CORBA::Long i = 0, iEnd = aSequence.size();
316 if((aGroupDS->*theFun)(aSequence[i]))
325 AddByPredicate( SMESH::Predicate_ptr thePredicate )
327 if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
328 TPythonDump()<<_this()<<".AddByPredicate("<<aPredicate<<")";
329 return ChangeByPredicate(aPredicate,GetGroupDS(),&SMESHDS_Group::Add);
336 RemoveByPredicate( SMESH::Predicate_ptr thePredicate )
338 if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
339 TPythonDump()<<_this()<<".RemoveByPredicate("<<aPredicate<<")";
340 return ChangeByPredicate(aPredicate,GetGroupDS(),&SMESHDS_Group::Remove);
345 //=============================================================================
349 //=============================================================================
351 CORBA::Long SMESH_GroupBase_i::GetID( CORBA::Long theIndex )
353 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
355 return aGroupDS->GetID(theIndex);
356 MESSAGE("attempt to iterate on a vague group");
360 //=============================================================================
364 //=============================================================================
366 SMESH::long_array* SMESH_GroupBase_i::GetListOfID()
368 SMESH::long_array_var aRes = new SMESH::long_array();
369 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
371 int aSize = aGroupDS->Extent();
373 for (int i = 0; i < aSize; i++)
374 aRes[i] = aGroupDS->GetID(i+1);
377 MESSAGE("get list of IDs of a vague group");
381 //=============================================================================
385 //=============================================================================
386 SMESH::SMESH_Mesh_ptr SMESH_GroupBase_i::GetMesh()
388 SMESH::SMESH_Mesh_var aMesh;
390 aMesh = SMESH::SMESH_Mesh::_narrow( myMeshServant->_this() );
391 return aMesh._retn();
394 //=============================================================================
398 //=============================================================================
399 SMESH::long_array* SMESH_GroupBase_i::GetIDs()
401 SMESH::long_array_var aResult = GetListOfID();
402 return aResult._retn();
405 //=======================================================================
406 //function : GetShape
408 //=======================================================================
410 GEOM::GEOM_Object_ptr SMESH_GroupOnGeom_i::GetShape()
412 GEOM::GEOM_Object_var aGeomObj;
413 SMESHDS_GroupOnGeom* aGroupDS = dynamic_cast<SMESHDS_GroupOnGeom*>( GetGroupDS() );
415 SMESH_Gen_i* aGen = GetMeshServant()->GetGen();
416 aGeomObj = aGen->ShapeToGeomObject( aGroupDS->GetShape() );
418 return aGeomObj._retn();
421 //=============================================================================
425 //=============================================================================
426 SALOMEDS::Color SMESH_GroupBase_i::GetColor()
428 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
431 Quantity_Color aQColor = aGroupDS->GetColor();
432 SALOMEDS::Color aColor;
433 aColor.R = aQColor.Red();
434 aColor.G = aQColor.Green();
435 aColor.B = aQColor.Blue();
439 MESSAGE("get color of a group");
440 return SALOMEDS::Color();
443 //=============================================================================
447 //=============================================================================
448 void SMESH_GroupBase_i::SetColor(const SALOMEDS::Color& color)
450 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
453 Quantity_Color aQColor( color.R, color.G, color.B, Quantity_TOC_RGB );
454 return aGroupDS->SetColor(aQColor);
456 MESSAGE("set color of a group");
460 //=============================================================================
464 //=============================================================================
465 CORBA::Long SMESH_GroupBase_i::GetColorNumber()
467 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
469 return aGroupDS->GetColorGroup();
470 MESSAGE("get color number of a group");
474 //=============================================================================
478 //=============================================================================
479 void SMESH_GroupBase_i::SetColorNumber(CORBA::Long color)
481 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
483 return aGroupDS->SetColorGroup(color);
484 MESSAGE("set color number of a group");