Salome HOME
52457: Addition of hypotheses is 8 time longer than meshing.
[modules/smesh.git] / src / SMESH / SMESH_Mesh.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 //  File   : SMESH_Mesh.cxx
24 //  Author : Paul RASCLE, EDF
25 //  Module : SMESH
26 //
27 #include "SMESH_Mesh.hxx"
28 #include "SMESH_MesherHelper.hxx"
29 #include "SMDS_MeshVolume.hxx"
30 #include "SMDS_SetIterator.hxx"
31 #include "SMESHDS_Document.hxx"
32 #include "SMESHDS_Group.hxx"
33 #include "SMESHDS_GroupOnGeom.hxx"
34 #include "SMESHDS_Script.hxx"
35 #include "SMESHDS_TSubMeshHolder.hxx"
36 #include "SMESH_Gen.hxx"
37 #include "SMESH_Group.hxx"
38 #include "SMESH_HypoFilter.hxx"
39 #include "SMESH_Hypothesis.hxx"
40 #include "SMESH_subMesh.hxx"
41
42 #include "utilities.h"
43
44 #include "DriverDAT_W_SMDS_Mesh.h"
45 #include "DriverGMF_Read.hxx"
46 #include "DriverGMF_Write.hxx"
47 #include "DriverMED_R_SMESHDS_Mesh.h"
48 #include "DriverMED_W_SMESHDS_Mesh.h"
49 #include "DriverSTL_R_SMDS_Mesh.h"
50 #include "DriverSTL_W_SMDS_Mesh.h"
51 #include "DriverUNV_R_SMDS_Mesh.h"
52 #include "DriverUNV_W_SMDS_Mesh.h"
53 #ifdef WITH_CGNS
54 #include "DriverCGNS_Read.hxx"
55 #include "DriverCGNS_Write.hxx"
56 #endif
57
58 #undef _Precision_HeaderFile
59 #include <BRepBndLib.hxx>
60 #include <BRepPrimAPI_MakeBox.hxx>
61 #include <Bnd_Box.hxx>
62 #include <TColStd_MapOfInteger.hxx>
63 #include <TopExp.hxx>
64 #include <TopExp_Explorer.hxx>
65 #include <TopTools_ListIteratorOfListOfShape.hxx>
66 #include <TopTools_ListOfShape.hxx>
67 #include <TopTools_MapOfShape.hxx>
68 #include <TopoDS_Iterator.hxx>
69
70 #include "SMESH_TryCatch.hxx" // include after OCCT headers!
71
72 #include "Utils_ExceptHandlers.hxx"
73 #ifndef WIN32
74 #include <boost/thread/thread.hpp>
75 #include <boost/bind.hpp>
76 #else 
77 #include <pthread.h> 
78 #endif
79
80 using namespace std;
81
82 // maximum stored group name length in MED file
83 #define MAX_MED_GROUP_NAME_LENGTH 80
84
85 #ifdef _DEBUG_
86 static int MYDEBUG = 0;
87 #else
88 static int MYDEBUG = 0;
89 #endif
90
91 #define cSMESH_Hyp(h) static_cast<const SMESH_Hypothesis*>(h)
92
93 typedef SMESH_HypoFilter THypType;
94
95 class SMESH_Mesh::SubMeshHolder : public SMESHDS_TSubMeshHolder< SMESH_subMesh >
96 {
97 };
98
99 //=============================================================================
100 /*!
101  * 
102  */
103 //=============================================================================
104
105 SMESH_Mesh::SMESH_Mesh(int               theLocalId, 
106                        int               theStudyId, 
107                        SMESH_Gen*        theGen,
108                        bool              theIsEmbeddedMode,
109                        SMESHDS_Document* theDocument):
110   _groupId( 0 ), _nbSubShapes( 0 )
111 {
112   MESSAGE("SMESH_Mesh::SMESH_Mesh(int localId)");
113   _id            = theLocalId;
114   _studyId       = theStudyId;
115   _gen           = theGen;
116   _myDocument    = theDocument;
117   _myMeshDS      = theDocument->NewMesh(theIsEmbeddedMode,theLocalId);
118   _isShapeToMesh = false;
119   _isAutoColor   = false;
120   _isModified    = false;
121   _shapeDiagonal = 0.0;
122   _callUp        = NULL;
123   _myMeshDS->ShapeToMesh( PseudoShape() );
124   _subMeshHolder = new SubMeshHolder;
125 }
126
127 //================================================================================
128 /*!
129  * \brief Constructor of SMESH_Mesh being a base of some descendant class
130  */
131 //================================================================================
132
133 SMESH_Mesh::SMESH_Mesh():
134   _id(-1),
135   _studyId(-1),
136   _groupId( 0 ),
137   _nbSubShapes( 0 ),
138   _isShapeToMesh( false ),
139   _myDocument( 0 ),
140   _myMeshDS( 0 ),
141   _gen( 0 ),
142   _isAutoColor( false ),
143   _isModified( false ),
144   _shapeDiagonal( 0.0 ),
145   _callUp( 0 )
146 {
147   _subMeshHolder = new SubMeshHolder;
148 }
149
150 namespace
151 {
152 #ifndef WIN32
153   void deleteMeshDS(SMESHDS_Mesh* meshDS)
154   {
155     //cout << "deleteMeshDS( " << meshDS << endl;
156     delete meshDS;
157   }
158 #else
159   static void* deleteMeshDS(void* meshDS)
160   {
161     //cout << "deleteMeshDS( " << meshDS << endl;
162     SMESHDS_Mesh* m = (SMESHDS_Mesh*)meshDS;
163     if(m) {
164       delete m;
165     }
166     return 0;
167   }
168 #endif
169 }
170
171 //=============================================================================
172 /*!
173  *
174  */
175 //=============================================================================
176
177 SMESH_Mesh::~SMESH_Mesh()
178 {
179   MESSAGE("SMESH_Mesh::~SMESH_Mesh");
180
181   // issue 0020340: EDF 1022 SMESH : Crash with FindNodeClosestTo in a second new study
182   //   Notify event listeners at least that something happens
183   if ( SMESH_subMesh * sm = GetSubMeshContaining(1))
184     sm->ComputeStateEngine( SMESH_subMesh::MESH_ENTITY_REMOVED );
185
186   // delete groups
187   map < int, SMESH_Group * >::iterator itg;
188   for (itg = _mapGroup.begin(); itg != _mapGroup.end(); itg++) {
189     SMESH_Group *aGroup = (*itg).second;
190     delete aGroup;
191   }
192   _mapGroup.clear();
193
194   // delete sub-meshes
195   delete _subMeshHolder;
196
197   if ( _callUp) delete _callUp;
198   _callUp = 0;
199
200   // remove self from studyContext
201   if ( _gen )
202   {
203     StudyContextStruct * studyContext = _gen->GetStudyContext( _studyId );
204     studyContext->mapMesh.erase( _id );
205   }
206   if ( _myDocument )
207     _myDocument->RemoveMesh( _id );
208   _myDocument = 0;
209
210   if ( _myMeshDS ) {
211     // delete _myMeshDS, in a thread in order not to block closing a study with large meshes
212 #ifndef WIN32
213     boost::thread aThread(boost::bind( & deleteMeshDS, _myMeshDS ));
214 #else
215     pthread_t thread;
216     int result=pthread_create(&thread, NULL, deleteMeshDS, (void*)_myMeshDS);
217 #endif
218   }
219 }
220
221 //================================================================================
222 /*!
223  * \brief Return true if a mesh with given id exists
224  */
225 //================================================================================
226
227 bool SMESH_Mesh::MeshExists( int meshId ) const
228 {
229   return _myDocument ? _myDocument->GetMesh( meshId ) : false;
230 }
231
232 //=============================================================================
233 /*!
234  * \brief Set geometry to be meshed
235  */
236 //=============================================================================
237
238 void SMESH_Mesh::ShapeToMesh(const TopoDS_Shape & aShape)
239 {
240   if(MYDEBUG) MESSAGE("SMESH_Mesh::ShapeToMesh");
241
242   if ( !aShape.IsNull() && _isShapeToMesh ) {
243     if ( aShape.ShapeType() != TopAbs_COMPOUND && // group contents is allowed to change
244          _myMeshDS->ShapeToMesh().ShapeType() != TopAbs_COMPOUND )
245       throw SALOME_Exception(LOCALIZED ("a shape to mesh has already been defined"));
246   }
247   // clear current data
248   if ( !_myMeshDS->ShapeToMesh().IsNull() )
249   {
250     // removal of a shape to mesh, delete objects referring to sub-shapes:
251     // - sub-meshes
252     _subMeshHolder->DeleteAll();
253     //  - groups on geometry
254     map <int, SMESH_Group *>::iterator i_gr = _mapGroup.begin();
255     while ( i_gr != _mapGroup.end() ) {
256       if ( dynamic_cast<SMESHDS_GroupOnGeom*>( i_gr->second->GetGroupDS() )) {
257         _myMeshDS->RemoveGroup( i_gr->second->GetGroupDS() );
258         delete i_gr->second;
259         _mapGroup.erase( i_gr++ );
260       }
261       else
262         i_gr++;
263     }
264     _mapAncestors.Clear();
265
266     // clear SMESHDS
267     TopoDS_Shape aNullShape;
268     _myMeshDS->ShapeToMesh( aNullShape );
269
270     _shapeDiagonal = 0.0;
271   }
272
273   // set a new geometry
274   if ( !aShape.IsNull() )
275   {
276     _myMeshDS->ShapeToMesh(aShape);
277     _isShapeToMesh = true;
278     _nbSubShapes = _myMeshDS->MaxShapeIndex();
279
280     // fill map of ancestors
281     fillAncestorsMap(aShape);
282   }
283   else
284   {
285     _isShapeToMesh = false;
286     _shapeDiagonal = 0.0;
287     _myMeshDS->ShapeToMesh( PseudoShape() );
288   }
289   _isModified = false;
290 }
291
292 //=======================================================================
293 /*!
294  * \brief Return geometry to be meshed. (It may be a PseudoShape()!)
295  */
296 //=======================================================================
297
298 TopoDS_Shape SMESH_Mesh::GetShapeToMesh() const
299 {
300   return _myMeshDS->ShapeToMesh();
301 }
302
303 //=======================================================================
304 /*!
305  * \brief Return a solid which is returned by GetShapeToMesh() if
306  *        a real geometry to be meshed was not set
307  */
308 //=======================================================================
309
310 const TopoDS_Solid& SMESH_Mesh::PseudoShape()
311 {
312   static TopoDS_Solid aSolid;
313   if ( aSolid.IsNull() )
314   {
315     aSolid = BRepPrimAPI_MakeBox(1,1,1);
316   }
317   return aSolid;
318 }
319
320 //=======================================================================
321 /*!
322  * \brief Return diagonal size of bounding box of a shape
323  */
324 //=======================================================================
325
326 double SMESH_Mesh::GetShapeDiagonalSize(const TopoDS_Shape & aShape)
327 {
328   if ( !aShape.IsNull() ) {
329     Bnd_Box Box;
330     BRepBndLib::Add(aShape, Box);
331     return sqrt( Box.SquareExtent() );
332   }
333   return 0;
334 }
335
336 //=======================================================================
337 /*!
338  * \brief Return diagonal size of bounding box of shape to mesh
339  */
340 //=======================================================================
341
342 double SMESH_Mesh::GetShapeDiagonalSize() const
343 {
344   if ( _shapeDiagonal == 0. && _isShapeToMesh )
345     const_cast<SMESH_Mesh*>(this)->_shapeDiagonal = GetShapeDiagonalSize( GetShapeToMesh() );
346
347   return _shapeDiagonal;
348 }
349
350 //================================================================================
351 /*!
352  * \brief Load mesh from study file
353  */
354 //================================================================================
355
356 void SMESH_Mesh::Load()
357 {
358   if (_callUp)
359     _callUp->Load();
360 }
361
362 //=======================================================================
363 /*!
364  * \brief Remove all nodes and elements
365  */
366 //=======================================================================
367
368 void SMESH_Mesh::Clear()
369 {
370   if ( HasShapeToMesh() ) // remove all nodes and elements
371   {
372     // clear mesh data
373     _myMeshDS->ClearMesh();
374
375     // update compute state of submeshes
376     if ( SMESH_subMesh *sm = GetSubMeshContaining( GetShapeToMesh() ) )
377     {
378       sm->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
379       sm->ComputeSubMeshStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
380       sm->ComputeStateEngine( SMESH_subMesh::CLEAN ); // for event listeners (issue 0020918)
381       sm->ComputeSubMeshStateEngine( SMESH_subMesh::CLEAN );
382     }
383   }
384   else // remove only nodes/elements computed by algorithms
385   {
386     if ( SMESH_subMesh *sm = GetSubMeshContaining( GetShapeToMesh() ) )
387     {
388       sm->ComputeStateEngine( SMESH_subMesh::CLEAN );
389       sm->ComputeSubMeshStateEngine( SMESH_subMesh::CLEAN );
390       sm->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
391       sm->ComputeSubMeshStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
392     }
393   }
394   _isModified = false;
395 }
396
397 //=======================================================================
398 /*!
399  * \brief Remove all nodes and elements of indicated shape
400  */
401 //=======================================================================
402
403 void SMESH_Mesh::ClearSubMesh(const int theShapeId)
404 {
405   // clear sub-meshes; get ready to re-compute as a side-effect 
406   if ( SMESH_subMesh *sm = GetSubMeshContaining( theShapeId ) )
407   {
408     SMESH_subMeshIteratorPtr smIt = sm->getDependsOnIterator(/*includeSelf=*/true,
409                                                              /*complexShapeFirst=*/false);
410     while ( smIt->more() )
411     {
412       sm = smIt->next();
413       TopAbs_ShapeEnum shapeType = sm->GetSubShape().ShapeType();      
414       if ( shapeType == TopAbs_VERTEX || shapeType < TopAbs_SOLID )
415         // all other shapes depends on vertices so they are already cleaned
416         sm->ComputeStateEngine( SMESH_subMesh::CLEAN );
417       // to recompute even if failed
418       sm->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
419     }
420   }
421 }
422
423 //=======================================================================
424 //function : UNVToMesh
425 //purpose  : 
426 //=======================================================================
427
428 int SMESH_Mesh::UNVToMesh(const char* theFileName)
429 {
430   if(MYDEBUG) MESSAGE("UNVToMesh - theFileName = "<<theFileName);
431   if(_isShapeToMesh)
432     throw SALOME_Exception(LOCALIZED("a shape to mesh has already been defined"));
433   _isShapeToMesh = false;
434   DriverUNV_R_SMDS_Mesh myReader;
435   myReader.SetMesh(_myMeshDS);
436   myReader.SetFile(theFileName);
437   myReader.SetMeshId(-1);
438   myReader.Perform();
439   if(MYDEBUG){
440     MESSAGE("UNVToMesh - _myMeshDS->NbNodes() = "<<_myMeshDS->NbNodes());
441     MESSAGE("UNVToMesh - _myMeshDS->NbEdges() = "<<_myMeshDS->NbEdges());
442     MESSAGE("UNVToMesh - _myMeshDS->NbFaces() = "<<_myMeshDS->NbFaces());
443     MESSAGE("UNVToMesh - _myMeshDS->NbVolumes() = "<<_myMeshDS->NbVolumes());
444   }
445   SMDS_MeshGroup* aGroup = (SMDS_MeshGroup*) myReader.GetGroup();
446   if (aGroup != 0) {
447     TGroupNamesMap aGroupNames = myReader.GetGroupNamesMap();
448     //const TGroupIdMap& aGroupId = myReader.GetGroupIdMap();
449     aGroup->InitSubGroupsIterator();
450     while (aGroup->MoreSubGroups()) {
451       SMDS_MeshGroup* aSubGroup = (SMDS_MeshGroup*) aGroup->NextSubGroup();
452       string aName = aGroupNames[aSubGroup];
453       int aId;
454
455       SMESH_Group* aSMESHGroup = AddGroup( aSubGroup->GetType(), aName.c_str(), aId );
456       if ( aSMESHGroup ) {
457         if(MYDEBUG) MESSAGE("UNVToMesh - group added: "<<aName);      
458         SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( aSMESHGroup->GetGroupDS() );
459         if ( aGroupDS ) {
460           aGroupDS->SetStoreName(aName.c_str());
461           aSubGroup->InitIterator();
462           const SMDS_MeshElement* aElement = 0;
463           while (aSubGroup->More()) {
464             aElement = aSubGroup->Next();
465             if (aElement) {
466               aGroupDS->SMDSGroup().Add(aElement);
467             }
468           }
469           if (aElement)
470             aGroupDS->SetType(aElement->GetType());
471         }
472       }
473     }
474   }
475   return 1;
476 }
477
478 //=======================================================================
479 //function : MEDToMesh
480 //purpose  : 
481 //=======================================================================
482
483 int SMESH_Mesh::MEDToMesh(const char* theFileName, const char* theMeshName)
484 {
485   if(MYDEBUG) MESSAGE("MEDToMesh - theFileName = "<<theFileName<<", mesh name = "<<theMeshName);
486   if(_isShapeToMesh)
487     throw SALOME_Exception(LOCALIZED("a shape to mesh has already been defined"));
488   _isShapeToMesh = false;
489   DriverMED_R_SMESHDS_Mesh myReader;
490   myReader.SetMesh(_myMeshDS);
491   myReader.SetMeshId(-1);
492   myReader.SetFile(theFileName);
493   myReader.SetMeshName(theMeshName);
494   Driver_Mesh::Status status = myReader.Perform();
495   if(MYDEBUG){
496     MESSAGE("MEDToMesh - _myMeshDS->NbNodes() = "<<_myMeshDS->NbNodes());
497     MESSAGE("MEDToMesh - _myMeshDS->NbEdges() = "<<_myMeshDS->NbEdges());
498     MESSAGE("MEDToMesh - _myMeshDS->NbFaces() = "<<_myMeshDS->NbFaces());
499     MESSAGE("MEDToMesh - _myMeshDS->NbVolumes() = "<<_myMeshDS->NbVolumes());
500   }
501
502   // Reading groups (sub-meshes are out of scope of MED import functionality)
503   list<TNameAndType> aGroupNames = myReader.GetGroupNamesAndTypes();
504   if(MYDEBUG) MESSAGE("MEDToMesh - Nb groups = "<<aGroupNames.size()); 
505   int anId;
506   list<TNameAndType>::iterator name_type = aGroupNames.begin();
507   for ( ; name_type != aGroupNames.end(); name_type++ ) {
508     SMESH_Group* aGroup = AddGroup( name_type->second, name_type->first.c_str(), anId );
509     if ( aGroup ) {
510       if(MYDEBUG) MESSAGE("MEDToMesh - group added: "<<name_type->first.c_str());      
511       SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( aGroup->GetGroupDS() );
512       if ( aGroupDS ) {
513         aGroupDS->SetStoreName( name_type->first.c_str() );
514         myReader.GetGroup( aGroupDS );
515       }
516     }
517   }
518   return (int) status;
519 }
520
521 //=======================================================================
522 //function : STLToMesh
523 //purpose  : 
524 //=======================================================================
525
526 int SMESH_Mesh::STLToMesh(const char* theFileName)
527 {
528   if(MYDEBUG) MESSAGE("STLToMesh - theFileName = "<<theFileName);
529   if(_isShapeToMesh)
530     throw SALOME_Exception(LOCALIZED("a shape to mesh has already been defined"));
531   _isShapeToMesh = false;
532   DriverSTL_R_SMDS_Mesh myReader;
533   myReader.SetMesh(_myMeshDS);
534   myReader.SetFile(theFileName);
535   myReader.SetMeshId(-1);
536   myReader.Perform();
537   if(MYDEBUG){
538     MESSAGE("STLToMesh - _myMeshDS->NbNodes() = "<<_myMeshDS->NbNodes());
539     MESSAGE("STLToMesh - _myMeshDS->NbEdges() = "<<_myMeshDS->NbEdges());
540     MESSAGE("STLToMesh - _myMeshDS->NbFaces() = "<<_myMeshDS->NbFaces());
541     MESSAGE("STLToMesh - _myMeshDS->NbVolumes() = "<<_myMeshDS->NbVolumes());
542   }
543   return 1;
544 }
545
546 //================================================================================
547 /*!
548  * \brief Reads the given mesh from the CGNS file
549  *  \param theFileName - name of the file
550  *  \retval int - Driver_Mesh::Status
551  */
552 //================================================================================
553
554 int SMESH_Mesh::CGNSToMesh(const char*  theFileName,
555                            const int    theMeshIndex,
556                            std::string& theMeshName)
557 {
558   int res = Driver_Mesh::DRS_FAIL;
559 #ifdef WITH_CGNS
560
561   DriverCGNS_Read myReader;
562   myReader.SetMesh(_myMeshDS);
563   myReader.SetFile(theFileName);
564   myReader.SetMeshId(theMeshIndex);
565   res = myReader.Perform();
566   theMeshName = myReader.GetMeshName();
567
568   // create groups
569   SynchronizeGroups();
570
571 #endif
572   return res;
573 }
574
575 //================================================================================
576 /*!
577  * \brief Fill its data by reading a GMF file
578  */
579 //================================================================================
580
581 SMESH_ComputeErrorPtr SMESH_Mesh::GMFToMesh(const char* theFileName,
582                                             bool        theMakeRequiredGroups)
583 {
584   DriverGMF_Read myReader;
585   myReader.SetMesh(_myMeshDS);
586   myReader.SetFile(theFileName);
587   myReader.SetMakeRequiredGroups( theMakeRequiredGroups );
588   myReader.Perform();
589   //theMeshName = myReader.GetMeshName();
590
591   // create groups
592   SynchronizeGroups();
593
594   return myReader.GetError();
595 }
596
597 //=============================================================================
598 /*!
599  * 
600  */
601 //=============================================================================
602
603 SMESH_Hypothesis::Hypothesis_Status
604   SMESH_Mesh::AddHypothesis(const TopoDS_Shape & aSubShape,
605                             int                  anHypId  ) throw(SALOME_Exception)
606 {
607   Unexpect aCatch(SalomeException);
608   if(MYDEBUG) MESSAGE("SMESH_Mesh::AddHypothesis");
609
610   SMESH_subMesh *subMesh = GetSubMesh(aSubShape);
611   if ( !subMesh || !subMesh->GetId())
612     return SMESH_Hypothesis::HYP_BAD_SUBSHAPE;
613
614   SMESH_Hypothesis *anHyp = GetHypothesis( anHypId );
615   if ( !anHyp )
616     throw SALOME_Exception(LOCALIZED("hypothesis does not exist"));
617
618   bool isGlobalHyp = IsMainShape( aSubShape );
619
620   // NotConformAllowed can be only global
621   if ( !isGlobalHyp )
622   {
623     // NOTE: this is not a correct way to check a name of hypothesis,
624     // there should be an attribute of hypothesis saying that it can/can't
625     // be global/local
626     string hypName = anHyp->GetName();
627     if ( hypName == "NotConformAllowed" )
628     {
629       if(MYDEBUG) MESSAGE( "Hypotesis <NotConformAllowed> can be only global" );
630       return SMESH_Hypothesis::HYP_INCOMPATIBLE;
631     }
632   }
633
634   // shape 
635
636   bool isAlgo = ( !anHyp->GetType() == SMESHDS_Hypothesis::PARAM_ALGO );
637   int event = isAlgo ? SMESH_subMesh::ADD_ALGO : SMESH_subMesh::ADD_HYP;
638
639   SMESH_Hypothesis::Hypothesis_Status ret = subMesh->AlgoStateEngine(event, anHyp);
640
641   // sub-shapes
642   if (!SMESH_Hypothesis::IsStatusFatal(ret) &&
643       anHyp->GetDim() <= SMESH_Gen::GetShapeDim(aSubShape)) // is added on father
644   {
645     event = isAlgo ? SMESH_subMesh::ADD_FATHER_ALGO : SMESH_subMesh::ADD_FATHER_HYP;
646
647     SMESH_Hypothesis::Hypothesis_Status ret2 =
648       subMesh->SubMeshesAlgoStateEngine(event, anHyp);
649     if (ret2 > ret)
650       ret = ret2;
651
652     // check concurent hypotheses on ancestors
653     if (ret < SMESH_Hypothesis::HYP_CONCURENT && !isGlobalHyp )
654     {
655       SMESH_subMeshIteratorPtr smIt = subMesh->getDependsOnIterator(false,false);
656       while ( smIt->more() ) {
657         SMESH_subMesh* sm = smIt->next();
658         if ( sm->IsApplicableHypotesis( anHyp )) {
659           ret2 = sm->CheckConcurentHypothesis( anHyp->GetType() );
660           if (ret2 > ret) {
661             ret = ret2;
662             break;
663           }
664         }
665       }
666     }
667   }
668   HasModificationsToDiscard(); // to reset _isModified flag if a mesh becomes empty
669
670   if(MYDEBUG) subMesh->DumpAlgoState(true);
671   if(MYDEBUG) SCRUTE(ret);
672   return ret;
673 }
674
675 //=============================================================================
676 /*!
677  * 
678  */
679 //=============================================================================
680
681 SMESH_Hypothesis::Hypothesis_Status
682   SMESH_Mesh::RemoveHypothesis(const TopoDS_Shape & aSubShape,
683                                int anHypId)throw(SALOME_Exception)
684 {
685   Unexpect aCatch(SalomeException);
686   if(MYDEBUG) MESSAGE("SMESH_Mesh::RemoveHypothesis");
687   
688   StudyContextStruct *sc = _gen->GetStudyContext(_studyId);
689   if (sc->mapHypothesis.find(anHypId) == sc->mapHypothesis.end())
690     throw SALOME_Exception(LOCALIZED("hypothesis does not exist"));
691   
692   SMESH_Hypothesis *anHyp = sc->mapHypothesis[anHypId];
693   if(MYDEBUG) {
694     SCRUTE(anHyp->GetType());
695   }
696   
697   // shape 
698   
699   bool isAlgo = ( !anHyp->GetType() == SMESHDS_Hypothesis::PARAM_ALGO );
700   int event = isAlgo ? SMESH_subMesh::REMOVE_ALGO : SMESH_subMesh::REMOVE_HYP;
701
702   SMESH_subMesh *subMesh = GetSubMesh(aSubShape);
703
704   SMESH_Hypothesis::Hypothesis_Status ret = subMesh->AlgoStateEngine(event, anHyp);
705
706   // there may appear concurrent hyps that were covered by the removed hyp
707   if (ret < SMESH_Hypothesis::HYP_CONCURENT &&
708       subMesh->IsApplicableHypotesis( anHyp ) &&
709       subMesh->CheckConcurentHypothesis( anHyp->GetType() ) != SMESH_Hypothesis::HYP_OK)
710     ret = SMESH_Hypothesis::HYP_CONCURENT;
711
712   // sub-shapes
713   if (!SMESH_Hypothesis::IsStatusFatal(ret) &&
714       anHyp->GetDim() <= SMESH_Gen::GetShapeDim(aSubShape)) // is removed from father
715   {
716     event = isAlgo ? SMESH_subMesh::REMOVE_FATHER_ALGO : SMESH_subMesh::REMOVE_FATHER_HYP;
717
718     SMESH_Hypothesis::Hypothesis_Status ret2 =
719       subMesh->SubMeshesAlgoStateEngine(event, anHyp);
720     if (ret2 > ret) // more severe
721       ret = ret2;
722
723     // check concurent hypotheses on ancestors
724     if (ret < SMESH_Hypothesis::HYP_CONCURENT && !IsMainShape( aSubShape ) )
725     {
726       SMESH_subMeshIteratorPtr smIt = subMesh->getDependsOnIterator(false,false);
727       while ( smIt->more() ) {
728         SMESH_subMesh* sm = smIt->next();
729         if ( sm->IsApplicableHypotesis( anHyp )) {
730           ret2 = sm->CheckConcurentHypothesis( anHyp->GetType() );
731           if (ret2 > ret) {
732             ret = ret2;
733             break;
734           }
735         }
736       }
737     }
738   }
739
740   HasModificationsToDiscard(); // to reset _isModified flag if mesh become empty
741
742   if(MYDEBUG) subMesh->DumpAlgoState(true);
743   if(MYDEBUG) SCRUTE(ret);
744   return ret;
745 }
746
747 //=============================================================================
748 /*!
749  * 
750  */
751 //=============================================================================
752
753 const list<const SMESHDS_Hypothesis*>&
754 SMESH_Mesh::GetHypothesisList(const TopoDS_Shape & aSubShape) const
755   throw(SALOME_Exception)
756 {
757   return _myMeshDS->GetHypothesis(aSubShape);
758 }
759
760 //=======================================================================
761 /*!
762  * \brief Return the hypothesis assigned to the shape
763  *  \param aSubShape    - the shape to check
764  *  \param aFilter      - the hypothesis filter
765  *  \param andAncestors - flag to check hypos assigned to ancestors of the shape
766  *  \param assignedTo   - to return the shape the found hypo is assigned to
767  *  \retval SMESH_Hypothesis* - the first hypo passed through aFilter
768  */
769 //=======================================================================
770
771 const SMESH_Hypothesis * SMESH_Mesh::GetHypothesis(const TopoDS_Shape &    aSubShape,
772                                                    const SMESH_HypoFilter& aFilter,
773                                                    const bool              andAncestors,
774                                                    TopoDS_Shape*           assignedTo) const
775 {
776   return GetHypothesis( const_cast< SMESH_Mesh* >(this)->GetSubMesh( aSubShape ),
777                         aFilter, andAncestors, assignedTo );
778 }
779
780 //=======================================================================
781 /*!
782  * \brief Return the hypothesis assigned to the shape of a sub-mesh
783  *  \param aSubMesh     - the sub-mesh to check
784  *  \param aFilter      - the hypothesis filter
785  *  \param andAncestors - flag to check hypos assigned to ancestors of the shape
786  *  \param assignedTo   - to return the shape the found hypo is assigned to
787  *  \retval SMESH_Hypothesis* - the first hypo passed through aFilter
788  */
789 //=======================================================================
790
791 const SMESH_Hypothesis * SMESH_Mesh::GetHypothesis(const SMESH_subMesh *   aSubMesh,
792                                                    const SMESH_HypoFilter& aFilter,
793                                                    const bool              andAncestors,
794                                                    TopoDS_Shape*           assignedTo) const
795 {
796   if ( !aSubMesh ) return 0;
797
798   {
799     const TopoDS_Shape & aSubShape = aSubMesh->GetSubShape();
800     const list<const SMESHDS_Hypothesis*>& hypList = _myMeshDS->GetHypothesis(aSubShape);
801     list<const SMESHDS_Hypothesis*>::const_iterator hyp = hypList.begin();
802     for ( ; hyp != hypList.end(); hyp++ ) {
803       const SMESH_Hypothesis * h = cSMESH_Hyp( *hyp );
804       if ( aFilter.IsOk( h, aSubShape)) {
805         if ( assignedTo ) *assignedTo = aSubShape;
806         return h;
807       }
808     }
809   }
810   if ( andAncestors )
811   {
812     // user sorted submeshes of ancestors, according to stored submesh priority
813     std::vector< SMESH_subMesh * > & ancestors =
814       const_cast< std::vector< SMESH_subMesh * > & > ( aSubMesh->GetAncestors() );
815     SortByMeshOrder( ancestors );
816
817     vector<SMESH_subMesh*>::const_iterator smIt = ancestors.begin(); 
818     for ( ; smIt != ancestors.end(); smIt++ )
819     {
820       const TopoDS_Shape& curSh = (*smIt)->GetSubShape();
821       const list<const SMESHDS_Hypothesis*>& hypList = _myMeshDS->GetHypothesis(curSh);
822       list<const SMESHDS_Hypothesis*>::const_iterator hyp = hypList.begin();
823       for ( ; hyp != hypList.end(); hyp++ ) {
824         const SMESH_Hypothesis * h = cSMESH_Hyp( *hyp );
825         if (aFilter.IsOk( h, curSh )) {
826           if ( assignedTo ) *assignedTo = curSh;
827           return h;
828         }
829       }
830     }
831   }
832   return 0;
833 }
834
835 //================================================================================
836 /*!
837  * \brief Return hypotheses assigned to the shape
838   * \param aSubShape - the shape to check
839   * \param aFilter - the hypothesis filter
840   * \param aHypList - the list of the found hypotheses
841   * \param andAncestors - flag to check hypos assigned to ancestors of the shape
842   * \retval int - number of unique hypos in aHypList
843  */
844 //================================================================================
845
846 int SMESH_Mesh::GetHypotheses(const TopoDS_Shape &                aSubShape,
847                               const SMESH_HypoFilter&             aFilter,
848                               list <const SMESHDS_Hypothesis * >& aHypList,
849                               const bool                          andAncestors,
850                               list< TopoDS_Shape > *              assignedTo/*=0*/) const
851 {
852   return GetHypotheses( const_cast< SMESH_Mesh* >(this)->GetSubMesh( aSubShape ),
853                         aFilter, aHypList, andAncestors, assignedTo );
854 }
855
856 //================================================================================
857 /*!
858  * \brief Return hypotheses assigned to the shape of a sub-mesh
859   * \param aSubShape - the sub-mesh to check
860   * \param aFilter - the hypothesis filter
861   * \param aHypList - the list of the found hypotheses
862   * \param andAncestors - flag to check hypos assigned to ancestors of the shape
863   * \retval int - number of unique hypos in aHypList
864  */
865 //================================================================================
866
867 int SMESH_Mesh::GetHypotheses(const SMESH_subMesh *               aSubMesh,
868                               const SMESH_HypoFilter&             aFilter,
869                               list <const SMESHDS_Hypothesis * >& aHypList,
870                               const bool                          andAncestors,
871                               list< TopoDS_Shape > *              assignedTo/*=0*/) const
872 {
873   if ( !aSubMesh ) return 0;
874
875   set<string> hypTypes; // to exclude same type hypos from the result list
876   int nbHyps = 0;
877
878   // only one main hypothesis is allowed
879   bool mainHypFound = false;
880
881   // fill in hypTypes
882   list<const SMESHDS_Hypothesis*>::const_iterator hyp;
883   for ( hyp = aHypList.begin(); hyp != aHypList.end(); hyp++ ) {
884     if ( hypTypes.insert( (*hyp)->GetName() ).second )
885       nbHyps++;
886     if ( !cSMESH_Hyp(*hyp)->IsAuxiliary() )
887       mainHypFound = true;
888   }
889
890   // get hypos from aSubShape
891   {
892     const TopoDS_Shape & aSubShape = aSubMesh->GetSubShape();
893     const list<const SMESHDS_Hypothesis*>& hypList = _myMeshDS->GetHypothesis(aSubShape);
894     for ( hyp = hypList.begin(); hyp != hypList.end(); hyp++ )
895       if ( aFilter.IsOk (cSMESH_Hyp( *hyp ), aSubShape) &&
896            ( cSMESH_Hyp(*hyp)->IsAuxiliary() || !mainHypFound ) &&
897            hypTypes.insert( (*hyp)->GetName() ).second )
898       {
899         aHypList.push_back( *hyp );
900         nbHyps++;
901         if ( !cSMESH_Hyp(*hyp)->IsAuxiliary() )
902           mainHypFound = true;
903         if ( assignedTo ) assignedTo->push_back( aSubShape );
904       }
905   }
906
907   // get hypos from ancestors of aSubShape
908   if ( andAncestors )
909   {
910     // user sorted submeshes of ancestors, according to stored submesh priority
911     std::vector< SMESH_subMesh * > & ancestors =
912       const_cast< std::vector< SMESH_subMesh * > & > ( aSubMesh->GetAncestors() );
913     SortByMeshOrder( ancestors );
914
915     vector<SMESH_subMesh*>::const_iterator smIt = ancestors.begin();
916     for ( ; smIt != ancestors.end(); smIt++ )
917     {
918       const TopoDS_Shape& curSh = (*smIt)->GetSubShape();
919       const list<const SMESHDS_Hypothesis*>& hypList = _myMeshDS->GetHypothesis(curSh);
920       for ( hyp = hypList.begin(); hyp != hypList.end(); hyp++ )
921         if (( aFilter.IsOk( cSMESH_Hyp( *hyp ), curSh ))         &&
922             ( cSMESH_Hyp(*hyp)->IsAuxiliary() || !mainHypFound ) &&
923             ( hypTypes.insert( (*hyp)->GetName() ).second          ))
924         {
925           aHypList.push_back( *hyp );
926           nbHyps++;
927           if ( !cSMESH_Hyp(*hyp)->IsAuxiliary() )
928             mainHypFound = true;
929           if ( assignedTo ) assignedTo->push_back( curSh );
930         }
931     }
932   }
933   return nbHyps;
934 }
935
936 //================================================================================
937 /*!
938  * \brief Return a hypothesis by its ID
939  */
940 //================================================================================
941
942 SMESH_Hypothesis * SMESH_Mesh::GetHypothesis(const int anHypId) const
943 {
944   StudyContextStruct *sc = _gen->GetStudyContext(_studyId);
945   if (sc->mapHypothesis.find(anHypId) == sc->mapHypothesis.end())
946     return NULL;
947
948   SMESH_Hypothesis *anHyp = sc->mapHypothesis[anHypId];
949   return anHyp;
950 }
951
952 //=============================================================================
953 /*!
954  * 
955  */
956 //=============================================================================
957
958 const list<SMESHDS_Command*> & SMESH_Mesh::GetLog() throw(SALOME_Exception)
959 {
960   Unexpect aCatch(SalomeException);
961   if(MYDEBUG) MESSAGE("SMESH_Mesh::GetLog");
962   return _myMeshDS->GetScript()->GetCommands();
963 }
964
965 //=============================================================================
966 /*!
967  * 
968  */
969 //=============================================================================
970 void SMESH_Mesh::ClearLog() throw(SALOME_Exception)
971 {
972   Unexpect aCatch(SalomeException);
973   if(MYDEBUG) MESSAGE("SMESH_Mesh::ClearLog");
974   _myMeshDS->GetScript()->Clear();
975 }
976
977 //=============================================================================
978 /*!
979  * Get or Create the SMESH_subMesh object implementation
980  */
981 //=============================================================================
982
983 SMESH_subMesh *SMESH_Mesh::GetSubMesh(const TopoDS_Shape & aSubShape)
984   throw(SALOME_Exception)
985 {
986   int index = _myMeshDS->ShapeToIndex(aSubShape);
987
988   // for submeshes on GEOM Group
989   if (( !index || index > _nbSubShapes ) && aSubShape.ShapeType() == TopAbs_COMPOUND ) {
990     TopoDS_Iterator it( aSubShape );
991     if ( it.More() )
992     {
993       index = _myMeshDS->AddCompoundSubmesh( aSubShape, it.Value().ShapeType() );
994       // fill map of Ancestors
995       while ( _nbSubShapes < index )
996         fillAncestorsMap( _myMeshDS->IndexToShape( ++_nbSubShapes ));
997     }
998   }
999 //   if ( !index )
1000 //     return NULL; // neither sub-shape nor a group
1001
1002   SMESH_subMesh* aSubMesh = _subMeshHolder->Get( index );
1003   if ( !aSubMesh )
1004   {
1005     aSubMesh = new SMESH_subMesh(index, this, _myMeshDS, aSubShape);
1006     _subMeshHolder->Add( index, aSubMesh );
1007   }
1008   return aSubMesh;
1009 }
1010
1011 //=============================================================================
1012 /*!
1013  * Get the SMESH_subMesh object implementation. Dont create it, return null
1014  * if it does not exist.
1015  */
1016 //=============================================================================
1017
1018 SMESH_subMesh *SMESH_Mesh::GetSubMeshContaining(const TopoDS_Shape & aSubShape) const
1019   throw(SALOME_Exception)
1020 {
1021   int index = _myMeshDS->ShapeToIndex(aSubShape);
1022   return GetSubMeshContaining( index );
1023 }
1024
1025 //=============================================================================
1026 /*!
1027  * Get the SMESH_subMesh object implementation. Dont create it, return null
1028  * if it does not exist.
1029  */
1030 //=============================================================================
1031
1032 SMESH_subMesh *SMESH_Mesh::GetSubMeshContaining(const int aShapeID) const
1033 throw(SALOME_Exception)
1034 {
1035   SMESH_subMesh *aSubMesh = _subMeshHolder->Get( aShapeID );
1036
1037   return aSubMesh;
1038 }
1039
1040 //================================================================================
1041 /*!
1042  * \brief Return submeshes of groups containing the given sub-shape
1043  */
1044 //================================================================================
1045
1046 list<SMESH_subMesh*>
1047 SMESH_Mesh::GetGroupSubMeshesContaining(const TopoDS_Shape & aSubShape) const
1048   throw(SALOME_Exception)
1049 {
1050   list<SMESH_subMesh*> found;
1051
1052   SMESH_subMesh * subMesh = GetSubMeshContaining(aSubShape);
1053   if ( !subMesh )
1054     return found;
1055
1056   // submeshes of groups have max IDs, so search from the map end
1057 SMESH_subMeshIteratorPtr smIt( _subMeshHolder->GetIterator( /*reverse=*/true ) );
1058   while ( smIt->more() ) {
1059     SMESH_subMesh*    sm = smIt->next();
1060     SMESHDS_SubMesh * ds = sm->GetSubMeshDS();
1061     if ( ds && ds->IsComplexSubmesh() ) {
1062       if ( SMESH_MesherHelper::IsSubShape( aSubShape, sm->GetSubShape() ))
1063       {
1064         found.push_back( sm );
1065         //break;
1066       }
1067     } else {
1068       break; // the rest sub-meshes are not those of groups
1069     }
1070   }
1071
1072   if ( found.empty() ) // maybe the main shape is a COMPOUND (issue 0021530)
1073   {
1074     if ( SMESH_subMesh * mainSM = GetSubMeshContaining(1) )
1075       if ( mainSM->GetSubShape().ShapeType() == TopAbs_COMPOUND )
1076       {
1077         TopoDS_Iterator it( mainSM->GetSubShape() );
1078         if ( it.Value().ShapeType() == aSubShape.ShapeType() &&
1079              SMESH_MesherHelper::IsSubShape( aSubShape, mainSM->GetSubShape() ))
1080           found.push_back( mainSM );
1081       }
1082   }
1083   return found;
1084 }
1085 //=======================================================================
1086 //function : IsUsedHypothesis
1087 //purpose  : Return True if anHyp is used to mesh aSubShape
1088 //=======================================================================
1089
1090 bool SMESH_Mesh::IsUsedHypothesis(SMESHDS_Hypothesis * anHyp,
1091                                   const SMESH_subMesh* aSubMesh)
1092 {
1093   SMESH_Hypothesis* hyp = static_cast<SMESH_Hypothesis*>(anHyp);
1094
1095   // check if anHyp can be used to mesh aSubMesh
1096   if ( !aSubMesh || !aSubMesh->IsApplicableHypotesis( hyp ))
1097     return false;
1098
1099   SMESH_Algo *algo = aSubMesh->GetAlgo();
1100
1101   // algorithm
1102   if (anHyp->GetType() > SMESHDS_Hypothesis::PARAM_ALGO)
1103     return ( anHyp == algo );
1104
1105   // algorithm parameter
1106   if (algo)
1107   {
1108     // look trough hypotheses used by algo
1109     const SMESH_HypoFilter* hypoKind;
1110     if (( hypoKind = algo->GetCompatibleHypoFilter( !hyp->IsAuxiliary() ))) {
1111       list <const SMESHDS_Hypothesis * > usedHyps;
1112       if ( GetHypotheses( aSubMesh, *hypoKind, usedHyps, true ))
1113         return ( find( usedHyps.begin(), usedHyps.end(), anHyp ) != usedHyps.end() );
1114     }
1115   }
1116
1117   return false;
1118 }
1119
1120 //=============================================================================
1121 /*!
1122  *
1123  */
1124 //=============================================================================
1125
1126 // const list < SMESH_subMesh * >&
1127 // SMESH_Mesh::GetSubMeshUsingHypothesis(SMESHDS_Hypothesis * anHyp)
1128 //   throw(SALOME_Exception)
1129 // {
1130 //   _subMeshesUsingHypothesisList.clear();
1131 //   SMESH_subMeshIteratorPtr smIt( _subMeshHolder->GetIterator() );
1132 //   while ( smIt->more() )
1133 //   {
1134 //     SMESH_subMesh* aSubMesh = smIt->next();
1135 //     if ( IsUsedHypothesis ( anHyp, aSubMesh ))
1136 //       _subMeshesUsingHypothesisList.push_back( aSubMesh );
1137 //   }
1138 //   return _subMeshesUsingHypothesisList;
1139 // }
1140
1141 //=======================================================================
1142 //function : NotifySubMeshesHypothesisModification
1143 //purpose  : Say all submeshes using theChangedHyp that it has been modified
1144 //=======================================================================
1145
1146 void SMESH_Mesh::NotifySubMeshesHypothesisModification(const SMESH_Hypothesis* hyp)
1147 {
1148   Unexpect aCatch(SalomeException);
1149
1150   if ( !GetMeshDS()->IsUsedHypothesis( hyp ))
1151     return;
1152
1153   if (_callUp)
1154     _callUp->HypothesisModified();
1155
1156   SMESH_Algo *algo;
1157   const SMESH_HypoFilter* compatibleHypoKind;
1158   list <const SMESHDS_Hypothesis * > usedHyps;
1159
1160   SMESH_subMeshIteratorPtr smIt( _subMeshHolder->GetIterator() );
1161   while ( smIt->more() )
1162   {
1163     SMESH_subMesh* aSubMesh = smIt->next();
1164
1165     // if aSubMesh meshing depends on hyp,
1166     // we call aSubMesh->AlgoStateEngine( MODIF_HYP, hyp ) that causes either
1167     // 1) clearing of already computed aSubMesh or
1168     // 2) changing algo_state from MISSING_HYP to HYP_OK when parameters of hyp becomes valid,
1169     // other possible changes are not interesting. (IPAL0052457 - assigning hyp performance pb)
1170     if ( aSubMesh->GetComputeState() != SMESH_subMesh::COMPUTE_OK &&
1171          aSubMesh->GetComputeState() != SMESH_subMesh::FAILED_TO_COMPUTE &&
1172          aSubMesh->GetAlgoState()    != SMESH_subMesh::MISSING_HYP )
1173       continue;
1174
1175     const TopoDS_Shape & aSubShape = aSubMesh->GetSubShape();
1176
1177     if (( aSubMesh->IsApplicableHypotesis( hyp )) &&
1178         ( algo = aSubMesh->GetAlgo() )            &&
1179         ( compatibleHypoKind = algo->GetCompatibleHypoFilter( !hyp->IsAuxiliary() )) &&
1180         ( compatibleHypoKind->IsOk( hyp, aSubShape )))
1181     {
1182       // check if hyp is used by algo
1183       usedHyps.clear();
1184       if ( GetHypotheses( aSubMesh, *compatibleHypoKind, usedHyps, true ) &&
1185            find( usedHyps.begin(), usedHyps.end(), hyp ) != usedHyps.end() )
1186       {
1187         aSubMesh->AlgoStateEngine(SMESH_subMesh::MODIF_HYP,
1188                                   const_cast< SMESH_Hypothesis*>( hyp ));
1189       }
1190     }
1191   }
1192   HasModificationsToDiscard(); // to reset _isModified flag if mesh becomes empty
1193   GetMeshDS()->Modified();
1194 }
1195
1196 //=============================================================================
1197 /*!
1198  *  Auto color functionality
1199  */
1200 //=============================================================================
1201 void SMESH_Mesh::SetAutoColor(bool theAutoColor) throw(SALOME_Exception)
1202 {
1203   Unexpect aCatch(SalomeException);
1204   _isAutoColor = theAutoColor;
1205 }
1206
1207 bool SMESH_Mesh::GetAutoColor() throw(SALOME_Exception)
1208 {
1209   Unexpect aCatch(SalomeException);
1210   return _isAutoColor;
1211 }
1212
1213 //=======================================================================
1214 //function : SetIsModified
1215 //purpose  : Set the flag meaning that the mesh has been edited "manually"
1216 //=======================================================================
1217
1218 void SMESH_Mesh::SetIsModified(bool isModified)
1219 {
1220   _isModified = isModified;
1221
1222   if ( _isModified )
1223     // check if mesh becomes empty as result of modification
1224     HasModificationsToDiscard();
1225 }
1226
1227 //=======================================================================
1228 //function : HasModificationsToDiscard
1229 //purpose  : Return true if the mesh has been edited since a total re-compute
1230 //           and those modifications may prevent successful partial re-compute.
1231 //           As a side effect reset _isModified flag if mesh is empty
1232 //issue    : 0020693
1233 //=======================================================================
1234
1235 bool SMESH_Mesh::HasModificationsToDiscard() const
1236 {
1237   if ( ! _isModified )
1238     return false;
1239
1240   // return true if the next Compute() will be partial and
1241   // existing but changed elements may prevent successful re-compute
1242   bool hasComputed = false, hasNotComputed = false;
1243 SMESH_subMeshIteratorPtr smIt( _subMeshHolder->GetIterator() );
1244   while ( smIt->more() )
1245   {
1246     const SMESH_subMesh* aSubMesh = smIt->next();
1247     switch ( aSubMesh->GetSubShape().ShapeType() )
1248     {
1249     case TopAbs_EDGE:
1250     case TopAbs_FACE:
1251     case TopAbs_SOLID:
1252       if ( aSubMesh->IsMeshComputed() )
1253         hasComputed = true;
1254       else
1255         hasNotComputed = true;
1256       if ( hasComputed && hasNotComputed)
1257         return true;
1258     }
1259   }
1260   if ( NbNodes() < 1 )
1261     const_cast<SMESH_Mesh*>(this)->_isModified = false;
1262
1263   return false;
1264 }
1265
1266 //================================================================================
1267 /*!
1268  * \brief Check if any groups of the same type have equal names
1269  */
1270 //================================================================================
1271
1272 bool SMESH_Mesh::HasDuplicatedGroupNamesMED()
1273 {
1274   //set<string> aGroupNames; // Corrected for Mantis issue 0020028
1275   map< SMDSAbs_ElementType, set<string> > aGroupNames;
1276   for ( map<int, SMESH_Group*>::iterator it = _mapGroup.begin(); it != _mapGroup.end(); it++ )
1277   {
1278     SMESH_Group* aGroup = it->second;
1279     SMDSAbs_ElementType aType = aGroup->GetGroupDS()->GetType();
1280     string aGroupName = aGroup->GetName();
1281     aGroupName.resize(MAX_MED_GROUP_NAME_LENGTH);
1282     if (!aGroupNames[aType].insert(aGroupName).second)
1283       return true;
1284   }
1285
1286   return false;
1287 }
1288
1289 //================================================================================
1290 /*!
1291  * \brief Export the mesh to a med file
1292  *  \param [in] file - name of the MED file
1293  *  \param [in] theMeshName - name of this mesh
1294  *  \param [in] theAutoGroups - boolean parameter for creating/not creating
1295  *              the groups Group_On_All_Nodes, Group_On_All_Faces, ... ;
1296  *              the typical use is auto_groups=false.
1297  *  \param [in] theVersion - defines the version of format of MED file, that will be created
1298  *  \param [in] meshPart - mesh data to export
1299  *  \param [in] theAutoDimension - if \c true, a space dimension of a MED mesh can be either
1300      *         - 1D if all mesh nodes lie on OX coordinate axis, or
1301      *         - 2D if all mesh nodes lie on XOY coordinate plane, or
1302      *         - 3D in the rest cases.
1303      *         If \a theAutoDimension is \c false, the space dimension is always 3.
1304  *  \return int - mesh index in the file
1305  */
1306 //================================================================================
1307
1308 void SMESH_Mesh::ExportMED(const char *        file, 
1309                            const char*         theMeshName, 
1310                            bool                theAutoGroups,
1311                            int                 theVersion,
1312                            const SMESHDS_Mesh* meshPart,
1313                            bool                theAutoDimension,
1314                            bool                theAddODOnVertices)
1315   throw(SALOME_Exception)
1316 {
1317   SMESH_TRY;
1318
1319   DriverMED_W_SMESHDS_Mesh myWriter;
1320   myWriter.SetFile         ( file, MED::EVersion(theVersion) );
1321   myWriter.SetMesh         ( meshPart ? (SMESHDS_Mesh*) meshPart : _myMeshDS   );
1322   myWriter.SetAutoDimension( theAutoDimension );
1323   myWriter.AddODOnVertices ( theAddODOnVertices );
1324   if ( !theMeshName ) 
1325     myWriter.SetMeshId     ( _id         );
1326   else {
1327     myWriter.SetMeshId     ( -1          );
1328     myWriter.SetMeshName   ( theMeshName );
1329   }
1330
1331   if ( theAutoGroups ) {
1332     myWriter.AddGroupOfNodes();
1333     myWriter.AddGroupOfEdges();
1334     myWriter.AddGroupOfFaces();
1335     myWriter.AddGroupOfVolumes();
1336   }
1337
1338   // Pass groups to writer. Provide unique group names.
1339   //set<string> aGroupNames; // Corrected for Mantis issue 0020028
1340   if ( !meshPart )
1341   {
1342     map< SMDSAbs_ElementType, set<string> > aGroupNames;
1343     char aString [256];
1344     int maxNbIter = 10000; // to guarantee cycle finish
1345     for ( map<int, SMESH_Group*>::iterator it = _mapGroup.begin(); it != _mapGroup.end(); it++ ) {
1346       SMESH_Group*       aGroup   = it->second;
1347       SMESHDS_GroupBase* aGroupDS = aGroup->GetGroupDS();
1348       if ( aGroupDS ) {
1349         SMDSAbs_ElementType aType = aGroupDS->GetType();
1350         string aGroupName0 = aGroup->GetName();
1351         aGroupName0.resize(MAX_MED_GROUP_NAME_LENGTH);
1352         string aGroupName = aGroupName0;
1353         for (int i = 1; !aGroupNames[aType].insert(aGroupName).second && i < maxNbIter; i++) {
1354           sprintf(&aString[0], "GR_%d_%s", i, aGroupName0.c_str());
1355           aGroupName = aString;
1356           aGroupName.resize(MAX_MED_GROUP_NAME_LENGTH);
1357         }
1358         aGroupDS->SetStoreName( aGroupName.c_str() );
1359         myWriter.AddGroup( aGroupDS );
1360       }
1361     }
1362   }
1363   // Perform export
1364   myWriter.Perform();
1365
1366   SMESH_CATCH( SMESH::throwSalomeEx );
1367 }
1368
1369 //================================================================================
1370 /*!
1371  * \brief Export the mesh to a SAUV file
1372  */
1373 //================================================================================
1374
1375 void SMESH_Mesh::ExportSAUV(const char *file, 
1376                             const char* theMeshName, 
1377                             bool theAutoGroups)
1378   throw(SALOME_Exception)
1379 {
1380   std::string medfilename(file);
1381   medfilename += ".med";
1382   std::string cmd;
1383 #ifdef WIN32
1384   cmd = "%PYTHONBIN% ";
1385 #else
1386   cmd = "python ";
1387 #endif
1388   cmd += "-c \"";
1389   cmd += "from medutilities import my_remove ; my_remove(r'" + medfilename + "')";
1390   cmd += "\"";
1391   system(cmd.c_str());
1392   ExportMED(medfilename.c_str(), theMeshName, theAutoGroups, 1);
1393 #ifdef WIN32
1394   cmd = "%PYTHONBIN% ";
1395 #else
1396   cmd = "python ";
1397 #endif
1398   cmd += "-c \"";
1399   cmd += "from medutilities import convert ; convert(r'" + medfilename + "', 'MED', 'GIBI', 1, r'" + file + "')";
1400   cmd += "\"";
1401   system(cmd.c_str());
1402 #ifdef WIN32
1403   cmd = "%PYTHONBIN% ";
1404 #else
1405   cmd = "python ";
1406 #endif
1407   cmd += "-c \"";
1408   cmd += "from medutilities import my_remove ; my_remove(r'" + medfilename + "')";
1409   cmd += "\"";
1410   system(cmd.c_str());
1411 }
1412
1413 //================================================================================
1414 /*!
1415  * \brief Export the mesh to a DAT file
1416  */
1417 //================================================================================
1418
1419 void SMESH_Mesh::ExportDAT(const char *        file,
1420                            const SMESHDS_Mesh* meshPart) throw(SALOME_Exception)
1421 {
1422   Unexpect aCatch(SalomeException);
1423   DriverDAT_W_SMDS_Mesh myWriter;
1424   myWriter.SetFile( file );
1425   myWriter.SetMesh( meshPart ? (SMESHDS_Mesh*) meshPart : _myMeshDS );
1426   myWriter.SetMeshId(_id);
1427   myWriter.Perform();
1428 }
1429
1430 //================================================================================
1431 /*!
1432  * \brief Export the mesh to an UNV file
1433  */
1434 //================================================================================
1435
1436 void SMESH_Mesh::ExportUNV(const char *        file,
1437                            const SMESHDS_Mesh* meshPart) throw(SALOME_Exception)
1438 {
1439   Unexpect aCatch(SalomeException);
1440   DriverUNV_W_SMDS_Mesh myWriter;
1441   myWriter.SetFile( file );
1442   myWriter.SetMesh( meshPart ? (SMESHDS_Mesh*) meshPart : _myMeshDS );
1443   myWriter.SetMeshId(_id);
1444   //  myWriter.SetGroups(_mapGroup);
1445
1446   if ( !meshPart )
1447   {
1448     for ( map<int, SMESH_Group*>::iterator it = _mapGroup.begin(); it != _mapGroup.end(); it++ ) {
1449       SMESH_Group*       aGroup   = it->second;
1450       SMESHDS_GroupBase* aGroupDS = aGroup->GetGroupDS();
1451       if ( aGroupDS ) {
1452         string aGroupName = aGroup->GetName();
1453         aGroupDS->SetStoreName( aGroupName.c_str() );
1454         myWriter.AddGroup( aGroupDS );
1455       }
1456     }
1457   }
1458   myWriter.Perform();
1459 }
1460
1461 //================================================================================
1462 /*!
1463  * \brief Export the mesh to an STL file
1464  */
1465 //================================================================================
1466
1467 void SMESH_Mesh::ExportSTL(const char *        file,
1468                            const bool          isascii,
1469                            const SMESHDS_Mesh* meshPart) throw(SALOME_Exception)
1470 {
1471   Unexpect aCatch(SalomeException);
1472   DriverSTL_W_SMDS_Mesh myWriter;
1473   myWriter.SetFile( file );
1474   myWriter.SetIsAscii( isascii );
1475   myWriter.SetMesh( meshPart ? (SMESHDS_Mesh*) meshPart : _myMeshDS);
1476   myWriter.SetMeshId(_id);
1477   myWriter.Perform();
1478 }
1479
1480 //================================================================================
1481 /*!
1482  * \brief Export the mesh to the CGNS file
1483  */
1484 //================================================================================
1485
1486 void SMESH_Mesh::ExportCGNS(const char *        file,
1487                             const SMESHDS_Mesh* meshDS)
1488 {
1489   int res = Driver_Mesh::DRS_FAIL;
1490 #ifdef WITH_CGNS
1491   DriverCGNS_Write myWriter;
1492   myWriter.SetFile( file );
1493   myWriter.SetMesh( const_cast<SMESHDS_Mesh*>( meshDS ));
1494   myWriter.SetMeshName( SMESH_Comment("Mesh_") << meshDS->GetPersistentId());
1495   res = myWriter.Perform();
1496 #endif
1497   if ( res != Driver_Mesh::DRS_OK )
1498     throw SALOME_Exception("Export failed");
1499 }
1500
1501 //================================================================================
1502 /*!
1503  * \brief Export the mesh to a GMF file
1504  */
1505 //================================================================================
1506
1507 void SMESH_Mesh::ExportGMF(const char *        file,
1508                            const SMESHDS_Mesh* meshDS,
1509                            bool                withRequiredGroups)
1510 {
1511   DriverGMF_Write myWriter;
1512   myWriter.SetFile( file );
1513   myWriter.SetMesh( const_cast<SMESHDS_Mesh*>( meshDS ));
1514   myWriter.SetExportRequiredGroups( withRequiredGroups );
1515
1516   myWriter.Perform();
1517 }
1518
1519 //================================================================================
1520 /*!
1521  * \brief Return a ratio of "compute cost" of computed sub-meshes to the whole
1522  *        "compute cost".
1523  */
1524 //================================================================================
1525
1526 double SMESH_Mesh::GetComputeProgress() const
1527 {
1528   double totalCost = 1e-100, computedCost = 0;
1529   const SMESH_subMesh* curSM = _gen->GetCurrentSubMesh();
1530
1531   // get progress of a current algo
1532   TColStd_MapOfInteger currentSubIds; 
1533   if ( curSM )
1534     if ( SMESH_Algo* algo = curSM->GetAlgo() )
1535     {
1536       int algoNotDoneCost = 0, algoDoneCost = 0;
1537       const std::vector<SMESH_subMesh*>& smToCompute = algo->SubMeshesToCompute();
1538       for ( size_t i = 0; i < smToCompute.size(); ++i )
1539       {
1540         if ( smToCompute[i]->IsEmpty() )
1541           algoNotDoneCost += smToCompute[i]->GetComputeCost();
1542         else
1543           algoDoneCost += smToCompute[i]->GetComputeCost();
1544         currentSubIds.Add( smToCompute[i]->GetId() );
1545       }
1546       double rate = 0;
1547       try
1548       {
1549         OCC_CATCH_SIGNALS;
1550         rate = algo->GetProgress();
1551       }
1552       catch (...) {
1553 #ifdef _DEBUG_
1554         cerr << "Exception in " << algo->GetName() << "::GetProgress()" << endl;
1555 #endif
1556       }
1557       if ( 0. < rate && rate < 1.001 )
1558       {
1559         computedCost += rate * ( algoDoneCost + algoNotDoneCost );
1560       }
1561       else
1562       {
1563         rate = algo->GetProgressByTic();
1564         computedCost += algoDoneCost + rate * algoNotDoneCost;
1565       }
1566       // cout << "rate: "<<rate << " algoNotDoneCost: " << algoNotDoneCost << endl;
1567     }
1568
1569   // get cost of already treated sub-meshes
1570   if ( SMESH_subMesh* mainSM = GetSubMeshContaining( 1 ))
1571   {
1572     SMESH_subMeshIteratorPtr smIt = mainSM->getDependsOnIterator(/*includeSelf=*/true);
1573     while ( smIt->more() )
1574     {
1575       const SMESH_subMesh* sm = smIt->next();
1576       const int smCost = sm->GetComputeCost();
1577       totalCost += smCost;
1578       if ( !currentSubIds.Contains( sm->GetId() ) )
1579       {
1580         if (( !sm->IsEmpty() ) ||
1581             ( sm->GetComputeState() == SMESH_subMesh::FAILED_TO_COMPUTE &&
1582               !sm->DependsOn( curSM ) ))
1583           computedCost += smCost;
1584       }
1585     }
1586   }
1587   // cout << "Total: " << totalCost
1588   //      << " computed: " << computedCost << " progress: " << computedCost / totalCost
1589   //      << " nbElems: " << GetMeshDS()->GetMeshInfo().NbElements() << endl;
1590   return computedCost / totalCost;
1591 }
1592
1593 //================================================================================
1594 /*!
1595  * \brief Return number of nodes in the mesh
1596  */
1597 //================================================================================
1598
1599 int SMESH_Mesh::NbNodes() const throw(SALOME_Exception)
1600 {
1601   Unexpect aCatch(SalomeException);
1602   return _myMeshDS->NbNodes();
1603 }
1604
1605 //================================================================================
1606 /*!
1607  * \brief  Return number of edges of given order in the mesh
1608  */
1609 //================================================================================
1610
1611 int SMESH_Mesh::Nb0DElements() const throw(SALOME_Exception)
1612 {
1613   Unexpect aCatch(SalomeException);
1614   return _myMeshDS->GetMeshInfo().Nb0DElements();
1615 }
1616
1617 //================================================================================
1618 /*!
1619  * \brief  Return number of edges of given order in the mesh
1620  */
1621 //================================================================================
1622
1623 int SMESH_Mesh::NbEdges(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1624 {
1625   Unexpect aCatch(SalomeException);
1626   return _myMeshDS->GetMeshInfo().NbEdges(order);
1627 }
1628
1629 //================================================================================
1630 /*!
1631  * \brief Return number of faces of given order in the mesh
1632  */
1633 //================================================================================
1634
1635 int SMESH_Mesh::NbFaces(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1636 {
1637   Unexpect aCatch(SalomeException);
1638   return _myMeshDS->GetMeshInfo().NbFaces(order);
1639 }
1640
1641 //================================================================================
1642 /*!
1643  * \brief Return the number of faces in the mesh
1644  */
1645 //================================================================================
1646
1647 int SMESH_Mesh::NbTriangles(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1648 {
1649   Unexpect aCatch(SalomeException);
1650   return _myMeshDS->GetMeshInfo().NbTriangles(order);
1651 }
1652
1653 //================================================================================
1654 /*!
1655  * \brief Return number of biquadratic triangles in the mesh
1656  */
1657 //================================================================================
1658
1659 int SMESH_Mesh::NbBiQuadTriangles() const throw(SALOME_Exception)
1660 {
1661   Unexpect aCatch(SalomeException);
1662   return _myMeshDS->GetMeshInfo().NbBiQuadTriangles();
1663 }
1664
1665 //================================================================================
1666 /*!
1667  * \brief Return the number nodes faces in the mesh
1668  */
1669 //================================================================================
1670
1671 int SMESH_Mesh::NbQuadrangles(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1672 {
1673   Unexpect aCatch(SalomeException);
1674   return _myMeshDS->GetMeshInfo().NbQuadrangles(order);
1675 }
1676
1677 //================================================================================
1678 /*!
1679  * \brief Return number of biquadratic quadrangles in the mesh
1680  */
1681 //================================================================================
1682
1683 int SMESH_Mesh::NbBiQuadQuadrangles() const throw(SALOME_Exception)
1684 {
1685   Unexpect aCatch(SalomeException);
1686   return _myMeshDS->GetMeshInfo().NbBiQuadQuadrangles();
1687 }
1688
1689 //================================================================================
1690 /*!
1691  * \brief Return the number of polygonal faces in the mesh
1692  */
1693 //================================================================================
1694
1695 int SMESH_Mesh::NbPolygons() const throw(SALOME_Exception)
1696 {
1697   Unexpect aCatch(SalomeException);
1698   return _myMeshDS->GetMeshInfo().NbPolygons();
1699 }
1700
1701 //================================================================================
1702 /*!
1703  * \brief Return number of volumes of given order in the mesh
1704  */
1705 //================================================================================
1706
1707 int SMESH_Mesh::NbVolumes(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1708 {
1709   Unexpect aCatch(SalomeException);
1710   return _myMeshDS->GetMeshInfo().NbVolumes(order);
1711 }
1712
1713 //================================================================================
1714 /*!
1715  * \brief  Return number of tetrahedrons of given order in the mesh
1716  */
1717 //================================================================================
1718
1719 int SMESH_Mesh::NbTetras(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1720 {
1721   Unexpect aCatch(SalomeException);
1722   return _myMeshDS->GetMeshInfo().NbTetras(order);
1723 }
1724
1725 //================================================================================
1726 /*!
1727  * \brief  Return number of hexahedrons of given order in the mesh
1728  */
1729 //================================================================================
1730
1731 int SMESH_Mesh::NbHexas(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1732 {
1733   Unexpect aCatch(SalomeException);
1734   return _myMeshDS->GetMeshInfo().NbHexas(order);
1735 }
1736
1737 //================================================================================
1738 /*!
1739  * \brief  Return number of triquadratic hexahedrons in the mesh
1740  */
1741 //================================================================================
1742
1743 int SMESH_Mesh::NbTriQuadraticHexas() const throw(SALOME_Exception)
1744 {
1745   Unexpect aCatch(SalomeException);
1746   return _myMeshDS->GetMeshInfo().NbTriQuadHexas();
1747 }
1748
1749 //================================================================================
1750 /*!
1751  * \brief  Return number of pyramids of given order in the mesh
1752  */
1753 //================================================================================
1754
1755 int SMESH_Mesh::NbPyramids(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1756 {
1757   Unexpect aCatch(SalomeException);
1758   return _myMeshDS->GetMeshInfo().NbPyramids(order);
1759 }
1760
1761 //================================================================================
1762 /*!
1763  * \brief  Return number of prisms (penthahedrons) of given order in the mesh
1764  */
1765 //================================================================================
1766
1767 int SMESH_Mesh::NbPrisms(SMDSAbs_ElementOrder order) const throw(SALOME_Exception)
1768 {
1769   Unexpect aCatch(SalomeException);
1770   return _myMeshDS->GetMeshInfo().NbPrisms(order);
1771 }
1772
1773 //================================================================================
1774 /*!
1775  * \brief  Return number of hexagonal prisms in the mesh
1776  */
1777 //================================================================================
1778
1779 int SMESH_Mesh::NbHexagonalPrisms() const throw(SALOME_Exception)
1780 {
1781   Unexpect aCatch(SalomeException);
1782   return _myMeshDS->GetMeshInfo().NbHexPrisms();
1783 }
1784
1785 //================================================================================
1786 /*!
1787  * \brief  Return number of polyhedrons in the mesh
1788  */
1789 //================================================================================
1790
1791 int SMESH_Mesh::NbPolyhedrons() const throw(SALOME_Exception)
1792 {
1793   Unexpect aCatch(SalomeException);
1794   return _myMeshDS->GetMeshInfo().NbPolyhedrons();
1795 }
1796
1797 //================================================================================
1798 /*!
1799  * \brief  Return number of ball elements in the mesh
1800  */
1801 //================================================================================
1802
1803 int SMESH_Mesh::NbBalls() const throw(SALOME_Exception)
1804 {
1805   Unexpect aCatch(SalomeException);
1806   return _myMeshDS->GetMeshInfo().NbBalls();
1807 }
1808
1809 //================================================================================
1810 /*!
1811  * \brief  Return number of submeshes in the mesh
1812  */
1813 //================================================================================
1814
1815 int SMESH_Mesh::NbSubMesh() const throw(SALOME_Exception)
1816 {
1817   Unexpect aCatch(SalomeException);
1818   return _myMeshDS->NbSubMesh();
1819 }
1820
1821 //=======================================================================
1822 //function : IsNotConformAllowed
1823 //purpose  : check if a hypothesis alowing notconform mesh is present
1824 //=======================================================================
1825
1826 bool SMESH_Mesh::IsNotConformAllowed() const
1827 {
1828   if(MYDEBUG) MESSAGE("SMESH_Mesh::IsNotConformAllowed");
1829
1830   static SMESH_HypoFilter filter( SMESH_HypoFilter::HasName( "NotConformAllowed" ));
1831   return GetHypothesis( _myMeshDS->ShapeToMesh(), filter, false );
1832 }
1833
1834 //=======================================================================
1835 //function : IsMainShape
1836 //purpose  : 
1837 //=======================================================================
1838
1839 bool SMESH_Mesh::IsMainShape(const TopoDS_Shape& theShape) const
1840 {
1841   return theShape.IsSame(_myMeshDS->ShapeToMesh() );
1842 }
1843
1844 //=============================================================================
1845 /*!
1846  *  
1847  */
1848 //=============================================================================
1849
1850 SMESH_Group* SMESH_Mesh::AddGroup (const SMDSAbs_ElementType theType,
1851                                    const char*               theName,
1852                                    int&                      theId,
1853                                    const TopoDS_Shape&       theShape,
1854                                    const SMESH_PredicatePtr& thePredicate)
1855 {
1856   if (_mapGroup.count(_groupId))
1857     return NULL;
1858   theId = _groupId;
1859   SMESH_Group* aGroup = new SMESH_Group (theId, this, theType, theName, theShape, thePredicate);
1860   GetMeshDS()->AddGroup( aGroup->GetGroupDS() );
1861   _mapGroup[_groupId++] = aGroup;
1862   return aGroup;
1863 }
1864
1865 //================================================================================
1866 /*!
1867  * \brief Creates a group based on an existing SMESHDS group. Group ID should be unique
1868  */
1869 //================================================================================
1870
1871 SMESH_Group* SMESH_Mesh::AddGroup (SMESHDS_GroupBase* groupDS) throw(SALOME_Exception)
1872 {
1873   if ( !groupDS ) 
1874     throw SALOME_Exception(LOCALIZED ("SMESH_Mesh::AddGroup(): NULL SMESHDS_GroupBase"));
1875
1876   map <int, SMESH_Group*>::iterator i_g = _mapGroup.find( groupDS->GetID() );
1877   if ( i_g != _mapGroup.end() && i_g->second )
1878   {
1879     if ( i_g->second->GetGroupDS() == groupDS )
1880       return i_g->second;
1881     else
1882       throw SALOME_Exception(LOCALIZED ("SMESH_Mesh::AddGroup() wrong ID of SMESHDS_GroupBase"));
1883   }
1884   SMESH_Group* aGroup = new SMESH_Group (groupDS);
1885   _mapGroup[ groupDS->GetID() ] = aGroup;
1886   GetMeshDS()->AddGroup( aGroup->GetGroupDS() );
1887
1888   _groupId = 1 + _mapGroup.rbegin()->first;
1889
1890   return aGroup;
1891 }
1892
1893
1894 //================================================================================
1895 /*!
1896  * \brief Creates SMESH_Groups for not wrapped SMESHDS_Groups
1897  *  \retval bool - true if new SMESH_Groups have been created
1898  * 
1899  */
1900 //================================================================================
1901
1902 bool SMESH_Mesh::SynchronizeGroups()
1903 {
1904   int nbGroups = _mapGroup.size();
1905   const set<SMESHDS_GroupBase*>& groups = _myMeshDS->GetGroups();
1906   set<SMESHDS_GroupBase*>::const_iterator gIt = groups.begin();
1907   for ( ; gIt != groups.end(); ++gIt )
1908   {
1909     SMESHDS_GroupBase* groupDS = (SMESHDS_GroupBase*) *gIt;
1910     _groupId = groupDS->GetID();
1911     if ( !_mapGroup.count( _groupId ))
1912       _mapGroup[_groupId] = new SMESH_Group( groupDS );
1913   }
1914   if ( !_mapGroup.empty() )
1915     _groupId = _mapGroup.rbegin()->first + 1;
1916
1917   return nbGroups < _mapGroup.size();
1918 }
1919
1920 //================================================================================
1921 /*!
1922  * \brief Return iterator on all existing groups
1923  */
1924 //================================================================================
1925
1926 SMESH_Mesh::GroupIteratorPtr SMESH_Mesh::GetGroups() const
1927 {
1928   typedef map <int, SMESH_Group *> TMap;
1929   return GroupIteratorPtr( new SMDS_mapIterator<TMap>( _mapGroup ));
1930 }
1931
1932 //=============================================================================
1933 /*!
1934  * \brief Return a group by ID
1935  */
1936 //=============================================================================
1937
1938 SMESH_Group* SMESH_Mesh::GetGroup (const int theGroupID)
1939 {
1940   if (_mapGroup.find(theGroupID) == _mapGroup.end())
1941     return NULL;
1942   return _mapGroup[theGroupID];
1943 }
1944
1945
1946 //=============================================================================
1947 /*!
1948  * \brief Return IDs of all groups
1949  */
1950 //=============================================================================
1951
1952 list<int> SMESH_Mesh::GetGroupIds() const
1953 {
1954   list<int> anIds;
1955   for ( map<int, SMESH_Group*>::const_iterator it = _mapGroup.begin(); it != _mapGroup.end(); it++ )
1956     anIds.push_back( it->first );
1957   
1958   return anIds;
1959 }
1960
1961 //================================================================================
1962 /*!
1963  * \brief Set a caller of methods at level of CORBA API implementation.
1964  * The set upCaller will be deleted by SMESH_Mesh
1965  */
1966 //================================================================================
1967
1968 void SMESH_Mesh::SetCallUp( TCallUp* upCaller )
1969 {
1970   if ( _callUp ) delete _callUp;
1971   _callUp = upCaller;
1972 }
1973
1974 //=============================================================================
1975 /*!
1976  *  
1977  */
1978 //=============================================================================
1979
1980 bool SMESH_Mesh::RemoveGroup (const int theGroupID)
1981 {
1982   if (_mapGroup.find(theGroupID) == _mapGroup.end())
1983     return false;
1984   GetMeshDS()->RemoveGroup( _mapGroup[theGroupID]->GetGroupDS() );
1985   delete _mapGroup[theGroupID];
1986   _mapGroup.erase (theGroupID);
1987   if (_callUp)
1988     _callUp->RemoveGroup( theGroupID );
1989   return true;
1990 }
1991
1992 //=======================================================================
1993 //function : GetAncestors
1994 //purpose  : return list of ancestors of theSubShape in the order
1995 //           that lower dimention shapes come first.
1996 //=======================================================================
1997
1998 const TopTools_ListOfShape& SMESH_Mesh::GetAncestors(const TopoDS_Shape& theS) const
1999 {
2000   if ( _mapAncestors.Contains( theS ) )
2001     return _mapAncestors.FindFromKey( theS );
2002
2003   static TopTools_ListOfShape emptyList;
2004   return emptyList;
2005 }
2006
2007 //=======================================================================
2008 //function : Dump
2009 //purpose  : dumps contents of mesh to stream [ debug purposes ]
2010 //=======================================================================
2011
2012 ostream& SMESH_Mesh::Dump(ostream& save)
2013 {
2014   int clause = 0;
2015   save << "========================== Dump contents of mesh ==========================" << endl << endl;
2016   save << ++clause << ") Total number of nodes:   \t"    << NbNodes() << endl;
2017   save << ++clause << ") Total number of edges:   \t"    << NbEdges() << endl;
2018   save << ++clause << ") Total number of faces:   \t"    << NbFaces() << endl;
2019   save << ++clause << ") Total number of polygons:\t"    << NbPolygons() << endl;
2020   save << ++clause << ") Total number of volumes:\t"     << NbVolumes() << endl;
2021   save << ++clause << ") Total number of polyhedrons:\t" << NbPolyhedrons() << endl << endl;
2022   for ( int isQuadratic = 0; isQuadratic < 2; ++isQuadratic )
2023   {
2024     string orderStr = isQuadratic ? "quadratic" : "linear";
2025     SMDSAbs_ElementOrder order  = isQuadratic ? ORDER_QUADRATIC : ORDER_LINEAR;
2026
2027     save << ++clause << ") Total number of " << orderStr << " edges:\t" << NbEdges(order) << endl;
2028     save << ++clause << ") Total number of " << orderStr << " faces:\t" << NbFaces(order) << endl;
2029     if ( NbFaces(order) > 0 ) {
2030       int nb3 = NbTriangles(order);
2031       int nb4 = NbQuadrangles(order);
2032       save << clause << ".1) Number of " << orderStr << " triangles:  \t" << nb3 << endl;
2033       save << clause << ".2) Number of " << orderStr << " quadrangles:\t" << nb4 << endl;
2034       if ( nb3 + nb4 !=  NbFaces(order) ) {
2035         map<int,int> myFaceMap;
2036         SMDS_FaceIteratorPtr itFaces=_myMeshDS->facesIterator();
2037         while( itFaces->more( ) ) {
2038           int nbNodes = itFaces->next()->NbNodes();
2039           if ( myFaceMap.find( nbNodes ) == myFaceMap.end() )
2040             myFaceMap[ nbNodes ] = 0;
2041           myFaceMap[ nbNodes ] = myFaceMap[ nbNodes ] + 1;
2042         }
2043         save << clause << ".3) Faces in detail: " << endl;
2044         map <int,int>::iterator itF;
2045         for (itF = myFaceMap.begin(); itF != myFaceMap.end(); itF++)
2046           save << "--> nb nodes: " << itF->first << " - nb elemens:\t" << itF->second << endl;
2047       }
2048     }
2049     save << ++clause << ") Total number of " << orderStr << " volumes:\t" << NbVolumes(order) << endl;
2050     if ( NbVolumes(order) > 0 ) {
2051       int nb8 = NbHexas(order);
2052       int nb4 = NbTetras(order);
2053       int nb5 = NbPyramids(order);
2054       int nb6 = NbPrisms(order);
2055       save << clause << ".1) Number of " << orderStr << " hexahedrons:\t" << nb8 << endl;
2056       save << clause << ".2) Number of " << orderStr << " tetrahedrons:\t" << nb4 << endl;
2057       save << clause << ".3) Number of " << orderStr << " prisms:      \t" << nb6 << endl;
2058       save << clause << ".4) Number of " << orderStr << " pyramids:\t" << nb5 << endl;
2059       if ( nb8 + nb4 + nb5 + nb6 != NbVolumes(order) ) {
2060         map<int,int> myVolumesMap;
2061         SMDS_VolumeIteratorPtr itVolumes=_myMeshDS->volumesIterator();
2062         while( itVolumes->more( ) ) {
2063           int nbNodes = itVolumes->next()->NbNodes();
2064           if ( myVolumesMap.find( nbNodes ) == myVolumesMap.end() )
2065             myVolumesMap[ nbNodes ] = 0;
2066           myVolumesMap[ nbNodes ] = myVolumesMap[ nbNodes ] + 1;
2067         }
2068         save << clause << ".5) Volumes in detail: " << endl;
2069         map <int,int>::iterator itV;
2070         for (itV = myVolumesMap.begin(); itV != myVolumesMap.end(); itV++)
2071           save << "--> nb nodes: " << itV->first << " - nb elemens:\t" << itV->second << endl;
2072       }
2073     }
2074     save << endl;
2075   }
2076   save << "===========================================================================" << endl;
2077   return save;
2078 }
2079
2080 //=======================================================================
2081 //function : GetElementType
2082 //purpose  : Returns type of mesh element with certain id
2083 //=======================================================================
2084
2085 SMDSAbs_ElementType SMESH_Mesh::GetElementType( const int id, const bool iselem )
2086 {
2087   return _myMeshDS->GetElementType( id, iselem );
2088 }
2089
2090 //=============================================================================
2091 /*!
2092  *  \brief Convert group on geometry into standalone group
2093  */
2094 //=============================================================================
2095
2096 SMESH_Group* SMESH_Mesh::ConvertToStandalone ( int theGroupID )
2097 {
2098   SMESH_Group* aGroup = 0;
2099   map < int, SMESH_Group * >::iterator itg = _mapGroup.find( theGroupID );
2100   if ( itg == _mapGroup.end() )
2101     return aGroup;
2102
2103   SMESH_Group* anOldGrp = (*itg).second;
2104   SMESHDS_GroupBase* anOldGrpDS = anOldGrp->GetGroupDS();
2105   if ( !anOldGrp || !anOldGrpDS )
2106     return aGroup;
2107
2108   // create new standalone group
2109   aGroup = new SMESH_Group (theGroupID, this, anOldGrpDS->GetType(), anOldGrp->GetName() );
2110   _mapGroup[theGroupID] = aGroup;
2111
2112   SMESHDS_Group* aNewGrpDS = dynamic_cast<SMESHDS_Group*>( aGroup->GetGroupDS() );
2113   GetMeshDS()->RemoveGroup( anOldGrpDS );
2114   GetMeshDS()->AddGroup( aNewGrpDS );
2115
2116   // add elements (or nodes) into new created group
2117   SMDS_ElemIteratorPtr anItr = anOldGrpDS->GetElements();
2118   while ( anItr->more() )
2119     aNewGrpDS->Add( (anItr->next())->GetID() );
2120
2121   // set color
2122   aNewGrpDS->SetColor( anOldGrpDS->GetColor() );
2123
2124   // remove old group
2125   delete anOldGrp;
2126
2127   return aGroup;
2128 }
2129
2130 //=============================================================================
2131 /*!
2132  *  \brief remove submesh order  from Mesh
2133  */
2134 //=============================================================================
2135
2136 void SMESH_Mesh::ClearMeshOrder()
2137 {
2138   _mySubMeshOrder.clear();
2139 }
2140
2141 //=============================================================================
2142 /*!
2143  *  \brief remove submesh order  from Mesh
2144  */
2145 //=============================================================================
2146
2147 void SMESH_Mesh::SetMeshOrder(const TListOfListOfInt& theOrder )
2148 {
2149   _mySubMeshOrder = theOrder;
2150 }
2151
2152 //=============================================================================
2153 /*!
2154  *  \brief return submesh order if any
2155  */
2156 //=============================================================================
2157
2158 const TListOfListOfInt& SMESH_Mesh::GetMeshOrder() const
2159 {
2160   return _mySubMeshOrder;
2161 }
2162
2163 //=============================================================================
2164 /*!
2165  *  \brief fill _mapAncestors
2166  */
2167 //=============================================================================
2168
2169 void SMESH_Mesh::fillAncestorsMap(const TopoDS_Shape& theShape)
2170 {
2171   int desType, ancType;
2172   if ( !theShape.IsSame( GetShapeToMesh()) && theShape.ShapeType() == TopAbs_COMPOUND )
2173   {
2174     // a geom group is added. Insert it into lists of ancestors before
2175     // the first ancestor more complex than group members
2176     TopoDS_Iterator subIt( theShape );
2177     if ( !subIt.More() ) return;
2178     int memberType = subIt.Value().ShapeType();
2179     for ( desType = TopAbs_VERTEX; desType >= memberType; desType-- )
2180       for (TopExp_Explorer des( theShape, TopAbs_ShapeEnum( desType )); des.More(); des.Next())
2181       {
2182         if ( !_mapAncestors.Contains( des.Current() )) continue;// issue 0020982
2183         TopTools_ListOfShape& ancList = _mapAncestors.ChangeFromKey( des.Current() );
2184         TopTools_ListIteratorOfListOfShape ancIt (ancList);
2185         while ( ancIt.More() && ancIt.Value().ShapeType() >= memberType )
2186           ancIt.Next();
2187         if ( ancIt.More() )
2188         {
2189           ancList.InsertBefore( theShape, ancIt );
2190           GetSubMesh( des.Current() )->ClearAncestors(); // to re-fill _ancestors
2191         }
2192       }
2193   }
2194   {
2195     for ( desType = TopAbs_VERTEX; desType > TopAbs_COMPOUND; desType-- )
2196       for ( ancType = desType - 1; ancType >= TopAbs_COMPOUND; ancType-- )
2197         TopExp::MapShapesAndAncestors ( theShape,
2198                                         (TopAbs_ShapeEnum) desType,
2199                                         (TopAbs_ShapeEnum) ancType,
2200                                         _mapAncestors );
2201   }
2202   // visit COMPOUNDs inside a COMPOUND that are not reachable by TopExp_Explorer
2203   if ( theShape.ShapeType() == TopAbs_COMPOUND )
2204   {
2205     for ( TopoDS_Iterator sIt(theShape); sIt.More(); sIt.Next() )
2206       if ( sIt.Value().ShapeType() == TopAbs_COMPOUND )
2207         fillAncestorsMap( sIt.Value() );
2208   }
2209 }
2210
2211 //=============================================================================
2212 /*!
2213  * \brief sort submeshes according to stored mesh order
2214  * \param theListToSort in out list to be sorted
2215  * \return FALSE if nothing sorted
2216  */
2217 //=============================================================================
2218
2219  bool SMESH_Mesh::SortByMeshOrder(std::vector<SMESH_subMesh*>& theListToSort) const
2220 {
2221   if ( !_mySubMeshOrder.size() || theListToSort.size() < 2)
2222     return true;
2223   
2224   bool res = false;
2225   vector<SMESH_subMesh*> onlyOrderedList;
2226   // collect all ordered submeshes in one list as pointers
2227   // and get their positions within theListToSort
2228   typedef vector<SMESH_subMesh*>::iterator TPosInList;
2229   map< int, TPosInList > sortedPos;
2230   TPosInList smBeg = theListToSort.begin(), smEnd = theListToSort.end();
2231   TListOfListOfInt::const_iterator listIdsIt = _mySubMeshOrder.begin();
2232   for( ; listIdsIt != _mySubMeshOrder.end(); listIdsIt++)
2233   {
2234     const TListOfInt& listOfId = *listIdsIt;
2235     // convert sm ids to sm's
2236     vector<SMESH_subMesh*> smVec;
2237     TListOfInt::const_iterator idIt = listOfId.begin();
2238     for ( ; idIt != listOfId.end(); idIt++ ) {
2239       if ( SMESH_subMesh * sm = GetSubMeshContaining( *idIt )) {
2240         if ( sm->GetSubMeshDS() && sm->GetSubMeshDS()->IsComplexSubmesh() )
2241         {
2242           SMESHDS_SubMeshIteratorPtr smdsIt = sm->GetSubMeshDS()->GetSubMeshIterator();
2243           while ( smdsIt->more() )
2244           {
2245             const SMESHDS_SubMesh* smDS = smdsIt->next();
2246             if (( sm = GetSubMeshContaining( smDS->GetID() )))
2247               smVec.push_back( sm );
2248           }
2249         }
2250         else
2251         {
2252           smVec.push_back( sm );
2253         }
2254       }
2255     }
2256     // find smVec items in theListToSort
2257     for ( size_t i = 0; i < smVec.size(); ++i )
2258     {
2259       TPosInList smPos = find( smBeg, smEnd, smVec[i] );
2260       if ( smPos != smEnd ) {
2261         onlyOrderedList.push_back( smVec[i] );
2262         sortedPos[ distance( smBeg, smPos )] = smPos;
2263       }
2264     }
2265   }
2266   if (onlyOrderedList.size() < 2)
2267     return res;
2268   res = true;
2269
2270   vector<SMESH_subMesh*>::iterator onlyBIt = onlyOrderedList.begin();
2271   vector<SMESH_subMesh*>::iterator onlyEIt = onlyOrderedList.end();
2272
2273   // iterate on ordered submeshes and insert them in detected positions
2274   map< int, TPosInList >::iterator i_pos = sortedPos.begin();
2275   for ( ; onlyBIt != onlyEIt; ++onlyBIt, ++i_pos )
2276     *(i_pos->second) = *onlyBIt;
2277
2278   return res;
2279 }
2280
2281 //================================================================================
2282 /*!
2283  * \brief Return true if given order of sub-meshes is OK
2284  */
2285 //================================================================================
2286
2287 bool SMESH_Mesh::IsOrderOK( const SMESH_subMesh* smBefore,
2288                             const SMESH_subMesh* smAfter ) const
2289 {
2290   TListOfListOfInt::const_iterator listIdsIt = _mySubMeshOrder.begin();
2291   TListOfInt::const_iterator idBef, idAft;
2292   for( ; listIdsIt != _mySubMeshOrder.end(); listIdsIt++)
2293   {
2294     const TListOfInt& listOfId = *listIdsIt;
2295     idBef = std::find( listOfId.begin(), listOfId.end(), smBefore->GetId() );
2296     if ( idBef != listOfId.end() )
2297       idAft = std::find( listOfId.begin(), listOfId.end(), smAfter->GetId() );
2298     if ( idAft != listOfId.end () )
2299       return ( std::distance( listOfId.begin(), idBef ) <
2300                std::distance( listOfId.begin(), idAft )   );
2301   }
2302   return true; // no order imposed to given submeshes
2303
2304
2305 //=============================================================================
2306 /*!
2307  * \brief sort submeshes according to stored mesh order
2308  * \param theListToSort in out list to be sorted
2309  * \return FALSE if nothing sorted
2310  */
2311 //=============================================================================
2312
2313 void SMESH_Mesh::getAncestorsSubMeshes (const TopoDS_Shape&            theSubShape,
2314                                         std::vector< SMESH_subMesh* >& theSubMeshes) const
2315 {
2316   theSubMeshes.clear();
2317   TopTools_ListIteratorOfListOfShape it( GetAncestors( theSubShape ));
2318   for (; it.More(); it.Next() )
2319     if ( SMESH_subMesh* sm = GetSubMeshContaining( it.Value() ))
2320       theSubMeshes.push_back(sm);
2321
2322   // sort submeshes according to stored mesh order
2323   SortByMeshOrder( theSubMeshes );
2324 }