1 // SMESH SMESH_I : idl implementation based on 'SMESH' unit's classes
3 // Copyright (C) 2004 CEA
5 // This library is free software; you can redistribute it and/or
6 // modify it under the terms of the GNU Lesser General Public
7 // License as published by the Free Software Foundation; either
8 // version 2.1 of the License.
10 // This library is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 // Lesser General Public License for more details.
15 // You should have received a copy of the GNU Lesser General Public
16 // License along with this library; if not, write to the Free Software
17 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 // See http://www.salome-platform.org or email : webmaster.salome@opencascade.org
23 // File : SMESH_Group_i.cxx
24 // Author : Sergey ANIKIN, OCC
29 #include "SMESH_Group_i.hxx"
30 #include "SMESH_Mesh_i.hxx"
31 #include "SMESH_Gen_i.hxx"
32 #include "SMESH_Group.hxx"
33 #include "SMESHDS_Group.hxx"
34 #include "SMESHDS_GroupOnGeom.hxx"
35 #include "SMDSAbs_ElementType.hxx"
37 #include "SMESH_Filter_i.hxx"
38 #include "SMESH_PythonDump.hxx"
40 #include "utilities.h"
42 using namespace SMESH;
44 //=============================================================================
48 //=============================================================================
50 SMESH_GroupBase_i::SMESH_GroupBase_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
51 : SALOME::GenericObj_i( thePOA ),
52 myMeshServant( theMeshServant ),
53 myLocalID( theLocalID )
55 // PAL7962: san -- To ensure correct mapping of servant and correct reference counting in GenericObj_i,
56 // servant activation is performed by SMESH_Mesh_i::createGroup()
57 // thePOA->activate_object( this );
60 SMESH_Group_i::SMESH_Group_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
61 : SALOME::GenericObj_i( thePOA ),
62 SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
64 MESSAGE("SMESH_Group_i; this = "<<this );
67 SMESH_GroupOnGeom_i::SMESH_GroupOnGeom_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
68 : SALOME::GenericObj_i( thePOA ),
69 SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
71 MESSAGE("SMESH_GroupOnGeom_i; this = "<<this );
74 //=============================================================================
78 //=============================================================================
80 SMESH_GroupBase_i::~SMESH_GroupBase_i()
82 MESSAGE("~SMESH_GroupBase_i; this = "<<this );
84 myMeshServant->removeGroup(myLocalID);
87 //=======================================================================
88 //function : GetSmeshGroup
90 //=======================================================================
92 ::SMESH_Group* SMESH_GroupBase_i::GetSmeshGroup() const
94 if ( myMeshServant ) {
95 ::SMESH_Mesh& aMesh = myMeshServant->GetImpl();
96 return aMesh.GetGroup(myLocalID);
101 //=======================================================================
102 //function : GetGroupDS
104 //=======================================================================
106 SMESHDS_GroupBase* SMESH_GroupBase_i::GetGroupDS() const
108 ::SMESH_Group* aGroup = GetSmeshGroup();
110 return aGroup->GetGroupDS();
114 //=============================================================================
118 //=============================================================================
120 void SMESH_GroupBase_i::SetName( const char* theName )
122 // Update Python script
123 TCollection_AsciiString aStr, aStrName ((char*)theName);
124 SMESH_Gen_i::AddObject(aStr, _this()) += ".SetName(\"";
125 aStr += aStrName + ")";
127 SMESH_Gen_i::AddToCurrentPyScript(aStr);
130 ::SMESH_Group* aGroup = GetSmeshGroup();
132 aGroup->SetName(theName);
134 // Update group name in a study
135 SMESH_Gen_i* aGen = myMeshServant->GetGen();
136 aGen->SetName( aGen->ObjectToSObject( aGen->GetCurrentStudy(), _this() ), theName );
139 MESSAGE("can't set name of a vague group");
142 //=============================================================================
146 //=============================================================================
148 char* SMESH_GroupBase_i::GetName()
150 ::SMESH_Group* aGroup = GetSmeshGroup();
152 return CORBA::string_dup (aGroup->GetName());
153 MESSAGE("get name of a vague group");
154 return CORBA::string_dup( "NO_NAME" );
157 //=============================================================================
161 //=============================================================================
163 SMESH::ElementType SMESH_GroupBase_i::GetType()
165 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
167 SMDSAbs_ElementType aSMDSType = aGroupDS->GetType();
168 SMESH::ElementType aType;
170 case SMDSAbs_Node: aType = SMESH::NODE; break;
171 case SMDSAbs_Edge: aType = SMESH::EDGE; break;
172 case SMDSAbs_Face: aType = SMESH::FACE; break;
173 case SMDSAbs_Volume: aType = SMESH::VOLUME; break;
174 default: aType = SMESH::ALL; break;
178 MESSAGE("get type of a vague group");
183 //=============================================================================
187 //=============================================================================
189 CORBA::Long SMESH_GroupBase_i::Size()
191 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
193 return aGroupDS->Extent();
194 MESSAGE("get size of a vague group");
198 //=============================================================================
202 //=============================================================================
204 CORBA::Boolean SMESH_GroupBase_i::IsEmpty()
206 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
208 return aGroupDS->IsEmpty();
209 MESSAGE("checking IsEmpty of a vague group");
213 //=============================================================================
217 //=============================================================================
219 void SMESH_Group_i::Clear()
221 // Update Python script
222 TCollection_AsciiString aStr;
223 SMESH_Gen_i::AddObject(aStr, _this()) += ".Clear()";
225 SMESH_Gen_i::AddToCurrentPyScript(aStr);
228 SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
233 MESSAGE("attempt to clear a vague group");
236 //=============================================================================
240 //=============================================================================
242 CORBA::Boolean SMESH_GroupBase_i::Contains( CORBA::Long theID )
244 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
246 return aGroupDS->Contains(theID);
247 MESSAGE("attempt to check contents of a vague group");
251 //=============================================================================
255 //=============================================================================
257 CORBA::Long SMESH_Group_i::Add( const SMESH::long_array& theIDs )
259 // Update Python script
260 TCollection_AsciiString aStr ("nbAdd = ");
261 SMESH_Gen_i::AddObject(aStr, _this()) += ".Add(";
262 SMESH_Gen_i::AddArray(aStr, theIDs) += ")";
264 SMESH_Gen_i::AddToCurrentPyScript(aStr);
266 // Add elements to the group
267 SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
270 for (int i = 0; i < theIDs.length(); i++) {
271 int anID = (int) theIDs[i];
272 if (aGroupDS->Add(anID))
277 MESSAGE("attempt to add elements to a vague group");
281 //=============================================================================
285 //=============================================================================
287 CORBA::Long SMESH_Group_i::Remove( const SMESH::long_array& theIDs )
289 // Update Python script
290 TCollection_AsciiString aStr ("nbDel = ");
291 SMESH_Gen_i::AddObject(aStr, _this()) += ".Remove(";
292 SMESH_Gen_i::AddArray(aStr, theIDs) += ")";
294 SMESH_Gen_i::AddToCurrentPyScript(aStr);
296 // Remove elements from the group
297 SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
300 for (int i = 0; i < theIDs.length(); i++) {
301 int anID = (int) theIDs[i];
302 if (aGroupDS->Remove(anID))
307 MESSAGE("attempt to remove elements from a vague group");
311 //=============================================================================
315 //=============================================================================
317 typedef bool (SMESHDS_Group::*TFunChangeGroup)(const int);
320 ChangeByPredicate( SMESH::Predicate_i* thePredicate,
321 SMESHDS_GroupBase* theGroupBase,
322 TFunChangeGroup theFun)
325 if(SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>(theGroupBase)){
326 SMESH::Controls::Filter::TIdSequence aSequence;
327 const SMDS_Mesh* aMesh = theGroupBase->GetMesh();
328 SMESH::Filter_i::GetElementsId(thePredicate,aMesh,aSequence);
330 CORBA::Long i = 0, iEnd = aSequence.size();
332 if((aGroupDS->*theFun)(aSequence[i]))
341 AddByPredicate( SMESH::Predicate_ptr thePredicate )
343 if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
344 TPythonDump()<<_this()<<".AddByPredicate("<<aPredicate<<")";
345 return ChangeByPredicate(aPredicate,GetGroupDS(),&SMESHDS_Group::Add);
352 RemoveByPredicate( SMESH::Predicate_ptr thePredicate )
354 if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
355 TPythonDump()<<_this()<<".RemoveByPredicate("<<aPredicate<<")";
356 return ChangeByPredicate(aPredicate,GetGroupDS(),&SMESHDS_Group::Remove);
361 //=============================================================================
365 //=============================================================================
367 CORBA::Long SMESH_GroupBase_i::GetID( CORBA::Long theIndex )
369 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
371 return aGroupDS->GetID(theIndex);
372 MESSAGE("attempt to iterate on a vague group");
376 //=============================================================================
380 //=============================================================================
382 SMESH::long_array* SMESH_GroupBase_i::GetListOfID()
384 SMESH::long_array_var aRes = new SMESH::long_array();
385 SMESHDS_GroupBase* aGroupDS = GetGroupDS();
387 int aSize = aGroupDS->Extent();
389 for (int i = 0; i < aSize; i++)
390 aRes[i] = aGroupDS->GetID(i+1);
393 MESSAGE("get list of IDs of a vague group");
397 //=============================================================================
401 //=============================================================================
402 SMESH::SMESH_Mesh_ptr SMESH_GroupBase_i::GetMesh()
404 SMESH::SMESH_Mesh_var aMesh;
406 aMesh = SMESH::SMESH_Mesh::_narrow( myMeshServant->_this() );
407 return aMesh._retn();
410 //=============================================================================
414 //=============================================================================
415 SMESH::long_array* SMESH_GroupBase_i::GetIDs()
417 SMESH::long_array_var aResult = GetListOfID();
418 return aResult._retn();
421 //=======================================================================
422 //function : GetShape
424 //=======================================================================
426 GEOM::GEOM_Object_ptr SMESH_GroupOnGeom_i::GetShape()
428 GEOM::GEOM_Object_var aGeomObj;
429 SMESHDS_GroupOnGeom* aGroupDS = dynamic_cast<SMESHDS_GroupOnGeom*>( GetGroupDS() );
431 SMESH_Gen_i* aGen = GetMeshServant()->GetGen();
432 aGeomObj = aGen->ShapeToGeomObject( aGroupDS->GetShape() );
434 return aGeomObj._retn();