]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROData/HYDROData_CalculationCase.cxx
Salome HOME
Merge branch 'BR_HYDRO_IMPS_2016' of ssh://gitolite3@git.salome-platform.org/modules...
[modules/hydro.git] / src / HYDROData / HYDROData_CalculationCase.cxx
1 // Copyright (C) 2014-2015  EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
6 //
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
10 // Lesser General Public License for more details.
11 //
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include "HYDROData_CalculationCase.h"
20 #include "HYDROData_ArtificialObject.h"
21 #include "HYDROData_IAltitudeObject.h"
22 #include "HYDROData_Document.h"
23 #include "HYDROData_ShapesGroup.h"
24 #include "HYDROData_Iterator.h"
25 #include "HYDROData_NaturalObject.h"
26 #include "HYDROData_PolylineXY.h"
27 #include "HYDROData_StricklerTable.h"
28 #include "HYDROData_LandCoverMap.h"
29 #include "HYDROData_SplitShapesGroup.h"
30 #include "HYDROData_Region.h"
31 #include "HYDROData_Tool.h"
32 #include "HYDROData_GeomTool.h"
33
34 #ifdef WIN32
35   #pragma warning ( disable: 4251 )
36 #endif
37
38 #ifndef LIGHT_MODE
39 #include <GEOMBase.h>
40 #endif
41
42 #ifdef WIN32
43   #pragma warning ( default: 4251 )
44 #endif
45
46 #include <QSet>
47
48 #include <TopoDS.hxx>
49 #include <TopoDS_Shell.hxx>
50 #include <TopoDS_Edge.hxx>
51
52 #include <BRep_Builder.hxx>
53 #include <BRepBuilderAPI_Sewing.hxx>
54 #include <BRepTopAdaptor_FClass2d.hxx>
55
56 #include <BRepTools.hxx>
57
58 #include <TopAbs.hxx>
59 #include <TopExp_Explorer.hxx>
60 #include <TopExp.hxx>
61 #include <TopTools_ListOfShape.hxx>
62 #include <TopTools_ListIteratorOfListOfShape.hxx>
63 #include <TDataStd_Integer.hxx>
64
65 //#define  DEB_CALCULATION 1
66 #ifdef DEB_CALCULATION
67 #include <BRepTools.hxx>
68 #include <TopLoc_Location.hxx>
69 #endif 
70 #define CALCULATION_REGIONS_PREF GetName() + "_Reg"
71 #define CALCULATION_ZONES_PREF GetName() + "_Zone"
72 #define CALCULATION_GROUPS_PREF GetName() + "_"
73 //#define DEB_CLASS2D 1
74 #ifdef DEB_CLASS2D
75 #include <BRepBuilderAPI_MakeVertex.hxx>
76 #endif
77
78 #define EXPORT_NAME "HYDRO_" + GetName()
79
80 #ifndef LIGHT_MODE
81 #include <SALOME_NamingService.hxx>
82 #include <SALOME_LifeCycleCORBA.hxx>
83 #endif
84
85 #define _DEVDEBUG_
86 #include "HYDRO_trace.hxx"
87
88 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_CalculationCase, HYDROData_Entity)
89
90 HYDROData_CalculationCase::HYDROData_CalculationCase()
91 : HYDROData_Entity( Geom_2d_and_groups )
92 {
93 }
94
95 HYDROData_CalculationCase::~HYDROData_CalculationCase()
96 {
97 }
98
99 void HYDROData_CalculationCase::SetName( const QString& theName )
100 {
101   QString anOldCaseName = GetName();
102   if ( anOldCaseName != theName )
103   {
104     // Update names of regions and its zones
105     UpdateRegionsNames( GetRegions(), anOldCaseName, theName );
106
107     HYDROData_SequenceOfObjects aGroups = GetSplitGroups();
108
109     HYDROData_SequenceOfObjects::Iterator anIter;
110     anIter.Init( aGroups );
111     for ( ; anIter.More(); anIter.Next() )
112     {
113       Handle(HYDROData_SplitShapesGroup) aGroup =
114         Handle(HYDROData_SplitShapesGroup)::DownCast( anIter.Value() );
115       if ( aGroup.IsNull() )
116         continue;
117
118       HYDROData_Tool::UpdateChildObjectName( anOldCaseName, theName, aGroup );
119     }
120   }
121
122   HYDROData_Entity::SetName( theName );
123 }
124
125 QStringList HYDROData_CalculationCase::DumpToPython( const QString&       thePyScriptPath,
126                                                      MapOfTreatedObjects& theTreatedObjects ) const
127 {
128   QStringList aResList = dumpObjectCreation( theTreatedObjects );
129   aResList.prepend( "# Calculation case" );
130
131   QString aCalculName = GetObjPyName();
132
133   AssignmentMode aMode = GetAssignmentMode();  
134   QString aModeStr = aMode==MANUAL ? "HYDROData_CalculationCase.MANUAL" : "HYDROData_CalculationCase.AUTOMATIC";
135   aResList << QString( "%0.SetAssignmentMode( %1 )" ).arg( aCalculName ).arg( aModeStr );
136
137   HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
138   HYDROData_SequenceOfObjects::Iterator anIter( aGeomObjects );
139   for ( ; anIter.More(); anIter.Next() )
140   {
141     Handle(HYDROData_Object) aRefGeomObj =
142       Handle(HYDROData_Object)::DownCast( anIter.Value() );
143     setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aRefGeomObj, "AddGeometryObject" );
144   }
145   aResList << QString( "" );
146
147   QString aGroupName = HYDROData_Tool::GenerateNameForPython( theTreatedObjects, "case_geom_group" );
148
149   HYDROData_SequenceOfObjects aGeomGroups = GetGeometryGroups();
150   anIter.Init( aGeomGroups );
151   for ( ; anIter.More(); anIter.Next() )
152   {
153     Handle(HYDROData_ShapesGroup) aGeomGroup =
154       Handle(HYDROData_ShapesGroup)::DownCast( anIter.Value() );
155     if ( aGeomGroup.IsNull() )
156       continue;
157
158     Handle(HYDROData_Object) aFatherGeom =
159       Handle(HYDROData_Object)::DownCast( aGeomGroup->GetFatherObject() );
160     if ( aFatherGeom.IsNull() )
161       continue;
162
163     int aGroupId = aFatherGeom->GetGroupId( aGeomGroup );
164     aResList << QString( "%1 = %2.GetGroup( %3 )" )
165               .arg( aGroupName ).arg( aFatherGeom->GetObjPyName() ).arg( aGroupId );
166
167     aResList << QString( "%1.AddGeometryGroup( %2 )" ).arg( aCalculName ).arg( aGroupName );
168   }
169
170   Handle(HYDROData_PolylineXY) aBoundaryPolyline = GetBoundaryPolyline();
171   setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aBoundaryPolyline, "SetBoundaryPolyline" );
172
173   Handle(HYDROData_StricklerTable) aStricklerTable = GetStricklerTable();
174   setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aStricklerTable, "SetStricklerTable" );
175
176   Handle(HYDROData_LandCoverMap) aLandCoverMap = GetLandCoverMap();
177   setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aLandCoverMap, "SetLandCoverMap" );
178
179   if( aMode==AUTOMATIC )
180     DumpRulesToPython( aCalculName, aResList );
181
182   aResList << QString( "" );
183   aResList << "# Start the algorithm of the partition and assignment";
184   aResList << QString( "%1.Update()" ).arg( aCalculName );
185
186   if( aMode==MANUAL )
187   {
188     // Now we restore the
189     // - regions and zones order
190     DumpRegionsToPython( aResList, thePyScriptPath, theTreatedObjects, GetRegions() );   
191   }
192
193   // Export calculation case
194   aResList << QString( "" );
195   aResList << "# Export of the calculation case";
196   QString aStudyName = "theStudy";
197   QString anEntryVar = aCalculName + "_entry";
198   aResList << QString( "%1 = %2.Export( %3._get_StudyId() )" ).arg( anEntryVar ).arg( aCalculName ).arg( aStudyName );
199
200   // Get geometry object and print debug information
201   aResList << "";
202   aResList << "# Get geometry shape and print debug information";
203   aResList << "import GEOM";
204   aResList << QString( "print \"Entry:\", %1" ).arg( anEntryVar );
205   QString aGeomShapeName = aCalculName + "_geom";
206   aResList << QString( "HYDRO_%1 = salome.IDToObject( str( %2 ) )" ).arg( GetName() ).arg( anEntryVar );
207   aResList << QString( "print \"Geom shape:\", HYDRO_%1" ).arg( GetName() );
208   aResList << QString( "print \"Geom shape name:\", HYDRO_%1.GetName()" ).arg( GetName() );
209
210
211   //DumpSampleMeshing( aResList, aStudyName, aGeomShapeName, aCalculName+"_mesh" );
212
213   aResList << QString( "" );
214   return aResList;
215 }
216
217 void HYDROData_CalculationCase::DumpSampleMeshing( QStringList& theResList,
218                                                    const QString& theStudyName,
219                                                    const QString& theGeomShapeName,
220                                                    const QString& theMeshName ) const
221 {
222   theResList << "";
223   theResList << "# Meshing";
224   theResList << "import SMESH, SALOMEDS";
225   theResList << "from salome.smesh import smeshBuilder";
226   theResList << "from salome.geom import geomBuilder";
227
228   theResList << QString( "smesh = smeshBuilder.New( %1 )" ).arg( theStudyName );
229   theResList << QString( "%1 = smesh.Mesh( %2 )" ).arg( theMeshName ).arg( theGeomShapeName );
230   theResList << QString( "MEFISTO_2D = %1.Triangle( algo=smeshBuilder.MEFISTO )" ).arg( theMeshName );
231   theResList << "Max_Element_Area_1 = MEFISTO_2D.MaxElementArea( 10 )";
232   theResList << QString( "Regular_1D = %1.Segment()" ).arg( theMeshName );
233   theResList << "Max_Size_1 = Regular_1D.MaxSize(10)";
234   theResList << QString( "isDone = %1.Compute()" ).arg( theMeshName );
235
236   theResList << "";
237   theResList << "# Set names of Mesh objects";
238   theResList << "smesh.SetName( MEFISTO_2D.GetAlgorithm(), 'MEFISTO_2D' )";
239   theResList << "smesh.SetName( Regular_1D.GetAlgorithm(), 'Regular_1D' )";
240   theResList << "smesh.SetName( Max_Size_1, 'Max Size_1' )";
241   theResList << "smesh.SetName( Max_Element_Area_1, 'Max. Element Area_1' )";
242   theResList << QString( "smesh.SetName( %1.GetMesh(), '%1' )" ).arg( theMeshName );
243
244   theResList << "";
245   theResList << "# Greate SMESH groups";
246   theResList << QString( "geompy = geomBuilder.New( %1 )" ).arg( theStudyName );
247   theResList << QString( "geom_groups = geompy.GetGroups( %1 )" ).arg( theGeomShapeName );
248   theResList << QString( "for group in geom_groups:" );
249   theResList << QString( "    smesh_group = %1.GroupOnGeom(group, group.GetName(), SMESH.EDGE)" )
250                 .arg( theMeshName );
251   theResList << QString( "    smesh.SetName(smesh_group, group.GetName())" );
252   theResList << QString( "    print \"SMESH group '%s': %s\" % (smesh_group.GetName(), smesh_group)" );
253 }
254
255 HYDROData_SequenceOfObjects HYDROData_CalculationCase::GetAllReferenceObjects() const
256 {
257   HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
258
259   Handle(HYDROData_PolylineXY) aBoundaryPolyline = GetBoundaryPolyline();
260   if ( !aBoundaryPolyline.IsNull() )
261     aResSeq.Append( aBoundaryPolyline );
262
263   HYDROData_SequenceOfObjects aSeqOfGeomObjs = GetGeometryObjects();
264   aResSeq.Append( aSeqOfGeomObjs );
265
266   // Regions
267   HYDROData_SequenceOfObjects aSeqOfRegions = GetRegions();
268   aResSeq.Append( aSeqOfRegions );
269
270   return aResSeq;
271 }
272
273 void HYDROData_CalculationCase::Update()
274 {
275   HYDROData_Entity::Update();
276   SetWarning();
277
278   // At first we remove previously created objects
279   RemoveRegions();
280   RemoveSplitGroups();
281
282   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
283   if ( aDocument.IsNull() )
284     return;
285
286   // Split to zones
287   HYDROData_SplitToZonesTool::SplitDataList aZonesList, anEdgesList;
288
289   Handle(HYDROData_PolylineXY) aBoundaryPolyline = GetBoundaryPolyline();
290
291   HYDROData_SequenceOfObjects InterPolys = GetInterPolyObjects();
292
293   HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
294   if ( !aGeomObjects.IsEmpty() ) {
295     HYDROData_SequenceOfObjects aGeomGroups = GetGeometryGroups();
296
297     HYDROData_SplitToZonesTool::SplitDataList aSplitObjects =
298       HYDROData_SplitToZonesTool::Split( aGeomObjects, aGeomGroups, aBoundaryPolyline, InterPolys );
299     if ( !aSplitObjects.isEmpty() ) {
300       HYDROData_SplitToZonesTool::SplitDataListIterator anIter( aSplitObjects );
301       while( anIter.hasNext() ) {
302         const HYDROData_SplitToZonesTool::SplitData& aSplitData = anIter.next();
303         if ( aSplitData.Type == HYDROData_SplitToZonesTool::SplitData::Data_Zone )
304           aZonesList.append( aSplitData );
305         else if ( aSplitData.Type == HYDROData_SplitToZonesTool::SplitData::Data_Edge ||
306                   aSplitData.Type == HYDROData_SplitToZonesTool::SplitData::Data_IntEdge)
307           anEdgesList.append( aSplitData );
308       }
309     }
310   }
311
312   switch( GetAssignmentMode() )
313   {
314   case MANUAL:
315     CreateRegionsDef( aDocument, aZonesList );    
316     break;
317   case AUTOMATIC:
318     CreateRegionsAuto( aDocument, aZonesList );    
319     break;
320   }
321
322   CreateEdgeGroupsDef( aDocument, anEdgesList );
323 }
324
325 void HYDROData_CalculationCase::CreateRegionsDef( const Handle(HYDROData_Document)& theDoc,
326                                                   const HYDROData_SplitToZonesTool::SplitDataList& theZones )
327 {
328   // Create result regions for case, by default one zone for one region
329   QString aRegsPref = CALCULATION_REGIONS_PREF;
330   QString aZonesPref = CALCULATION_ZONES_PREF;
331
332   HYDROData_SplitToZonesTool::SplitDataListIterator anIter( theZones );
333   while( anIter.hasNext() )
334   {
335     const HYDROData_SplitToZonesTool::SplitData& aSplitData = anIter.next();
336     // Create new region
337     Handle(HYDROData_Region) aRegion = addNewRegion( theDoc, aRegsPref );
338
339     // Add the zone for region
340     Handle(HYDROData_Zone) aRegionZone = aRegion->addNewZone( theDoc, aZonesPref, aSplitData.Face(), aSplitData.ObjectNames );
341   }
342 }
343
344 void HYDROData_CalculationCase::CreateRegionsAuto( const Handle(HYDROData_Document)& theDoc,
345                                                    const HYDROData_SplitToZonesTool::SplitDataList& theZones )
346 {
347   DEBTRACE("HYDROData_CalculationCase::CreateRegionsAuto");
348   QMap<QString, Handle(HYDROData_Region)> aRegionsMap; //object name to region
349   QMap<QString, QString> aRegionNameToObjNameMap;
350   QString aZonesPref = CALCULATION_ZONES_PREF;
351   HYDROData_PriorityQueue aPr( this, DataTag_CustomRules );
352
353   // 1. First we create a default region for each object included into the calculation case
354   HYDROData_SequenceOfObjects anObjects = GetGeometryObjects();
355   for( int i = anObjects.Lower(), n = anObjects.Upper(); i<=n; i++ )
356   {
357     Handle(HYDROData_Entity) anObj = anObjects.Value( i );
358     if( anObj.IsNull() )
359       continue;
360     QString anObjName = anObj->GetName();
361     QString aRegName = anObjName + "_reg";
362     Handle(HYDROData_Region) aRegion = addNewRegion( theDoc, aRegName, false );
363     aRegionsMap.insert( anObjName, aRegion );
364     aRegionNameToObjNameMap.insert( aRegName, anObjName );
365   }
366
367   // 2. Now for each zone it is necessary to determine the most priority object
368   //    and assign to zone to corresponding region
369   HYDROData_SplitToZonesTool::SplitDataListIterator anIter( theZones );
370   while( anIter.hasNext() )
371   {
372     const HYDROData_SplitToZonesTool::SplitData& aSplitData = anIter.next();
373     HYDROData_Zone::MergeType aMergeType;
374     Handle(HYDROData_Entity) aRegObj = aPr.GetMostPriorityObject( aSplitData.ObjectNames, aMergeType );
375     if( aRegObj.IsNull() )
376       continue;
377     Handle(HYDROData_Region) aRegion = aRegionsMap[aRegObj->GetName()];
378     if( aRegion.IsNull() )
379       continue;
380     Handle(HYDROData_Zone) aRegionZone = aRegion->addNewZone( theDoc, aZonesPref, aSplitData.Face(), aSplitData.ObjectNames );
381
382     if( aSplitData.ObjectNames.count() > 1 && aMergeType==HYDROData_Zone::Merge_UNKNOWN )
383     {
384       qDebug( "Error in algorithm: unresolved conflicts" );
385     }
386
387     Handle(HYDROData_Entity) aMergeEntity = aRegObj;
388     Handle(HYDROData_Object) aMergeObject = Handle(HYDROData_Object)::DownCast( aMergeEntity );
389     if ( !aMergeObject.IsNull() ) {
390       DEBTRACE("aMergeEntity " << aMergeEntity->GetName().toStdString());
391       aMergeEntity = aMergeObject->GetAltitudeObject();
392     }
393
394     switch( aMergeType )
395     {
396     case HYDROData_Zone::Merge_ZMIN:
397     case HYDROData_Zone::Merge_ZMAX:
398       aRegionZone->SetMergeType( aMergeType );
399       break;
400     case HYDROData_Zone::Merge_Object:
401       aRegionZone->SetMergeType( aMergeType );
402       aRegionZone->RemoveMergeObject();
403       aRegionZone->SetMergeObject( aMergeEntity );
404       break;
405     }
406   }
407
408   QStringList anObjectsWithEmptyRegions;
409   QMap<QString, Handle(HYDROData_Region)>::const_iterator
410     anIt = aRegionsMap.begin(), aLast = aRegionsMap.end();
411   for( ; anIt!=aLast; anIt++ )
412   {
413     Handle(HYDROData_Region) aRegion = anIt.value();
414     if( aRegion->GetZones().IsEmpty() )
415     {
416       QString aRegName = aRegion->GetName();
417       QString anObjName = aRegionNameToObjNameMap[aRegName];
418       anObjectsWithEmptyRegions.append( anObjName );
419     }
420   }
421   
422   if( !anObjectsWithEmptyRegions.empty() )
423   {
424     QString aData = anObjectsWithEmptyRegions.join( ", " );
425     SetWarning( WARN_EMPTY_REGIONS, aData );
426   }
427 }
428
429 void HYDROData_CalculationCase::CreateEdgeGroupsDef( const Handle(HYDROData_Document)& theDoc,
430                                                      const HYDROData_SplitToZonesTool::SplitDataList& theEdges )
431 {
432   QMap<QString,Handle(HYDROData_SplitShapesGroup)> aSplitEdgesGroupsMap;
433
434   HYDROData_SplitToZonesTool::SplitDataListIterator anIter( theEdges );
435   while( anIter.hasNext() )
436   {
437     const HYDROData_SplitToZonesTool::SplitData& aSplitData = anIter.next();
438     // Create new edges group
439     if ( aSplitData.ObjectNames.isEmpty() || aSplitData.Shape.IsNull() )
440       continue;
441
442     QString anObjName = aSplitData.ObjectNames.first();
443     if ( anObjName.isEmpty() )
444       continue;
445 #ifdef DEB_CALCULATION
446     QString aStr = aSplitData.ObjectNames.join(" "); 
447           cout << " CCase: Names = "<<aStr.toStdString() << " size = " <<aSplitData.ObjectNames.size() <<endl; 
448 #endif
449     Handle(HYDROData_SplitShapesGroup) aSplitGroup;
450     if ( !aSplitEdgesGroupsMap.contains( anObjName ) )
451     {
452       aSplitGroup = addNewSplitGroup( CALCULATION_GROUPS_PREF + anObjName );
453       aSplitEdgesGroupsMap.insert( anObjName, aSplitGroup );
454     }
455     else
456     {
457       aSplitGroup = aSplitEdgesGroupsMap[ anObjName ];
458     }
459     if ( aSplitGroup.IsNull() )
460       continue;
461
462     aSplitGroup->AddShape( aSplitData.Shape );
463
464     TopTools_SequenceOfShape theShapes;
465     aSplitGroup->GetShapes(theShapes);
466
467     if (aSplitData.Type == HYDROData_SplitToZonesTool::SplitData::Data_IntEdge)
468       aSplitGroup->SetInternal(true); 
469   }
470 }
471
472 bool HYDROData_CalculationCase::AddGeometryObject( const Handle(HYDROData_Object)& theObject )
473 {
474   if ( !HYDROData_Tool::IsGeometryObject( theObject ) )
475     return false; // Wrong type of object
476
477   if ( HasReference( theObject, DataTag_GeometryObject ) )
478     return false; // Object is already in reference list
479
480   AddReferenceObject( theObject, DataTag_GeometryObject );
481   
482   // Indicate model of the need to update splitting
483   Changed( Geom_2d );
484
485   return true;
486 }
487
488 HYDROData_SequenceOfObjects HYDROData_CalculationCase::GetGeometryObjects() const
489 {
490   return GetReferenceObjects( DataTag_GeometryObject );
491 }
492
493 void HYDROData_CalculationCase::RemoveGeometryObject( const Handle(HYDROData_Object)& theObject )
494 {
495   if ( theObject.IsNull() )
496     return;
497
498   RemoveReferenceObject( theObject->Label(), DataTag_GeometryObject );
499
500   // Indicate model of the need to update splitting
501   Changed( Geom_2d );
502 }
503
504 void HYDROData_CalculationCase::RemoveGeometryObjects()
505 {
506   ClearReferenceObjects( DataTag_GeometryObject );
507
508   // Indicate model of the need to update splitting
509   Changed( Geom_2d );
510 }
511
512 bool HYDROData_CalculationCase::AddGeometryGroup( const Handle(HYDROData_ShapesGroup)& theGroup )
513 {
514   if ( theGroup.IsNull() )
515     return false;
516
517   if ( HasReference( theGroup, DataTag_GeometryGroup ) )
518     return false; // Object is already in reference list
519
520   AddReferenceObject( theGroup, DataTag_GeometryGroup );
521   
522   // Indicate model of the need to update splitting
523   Changed( Geom_Groups );
524
525   return true;
526 }
527
528 HYDROData_SequenceOfObjects HYDROData_CalculationCase::GetGeometryGroups() const
529 {
530   return GetReferenceObjects( DataTag_GeometryGroup );
531 }
532
533 void HYDROData_CalculationCase::RemoveGeometryGroup( const Handle(HYDROData_ShapesGroup)& theGroup )
534 {
535   if ( theGroup.IsNull() )
536     return;
537
538   RemoveReferenceObject( theGroup->Label(), DataTag_GeometryGroup );
539
540   // Indicate model of the need to update splitting
541   Changed( Geom_Groups );
542 }
543
544 void HYDROData_CalculationCase::RemoveGeometryGroups()
545 {
546   ClearReferenceObjects( DataTag_GeometryGroup );
547
548   // Indicate model of the need to update splitting
549   Changed( Geom_Groups );
550 }
551
552 void HYDROData_CalculationCase::SetBoundaryPolyline( const Handle(HYDROData_PolylineXY)& thePolyline )
553 {
554   Handle(HYDROData_PolylineXY) aPrevPolyline = GetBoundaryPolyline();
555
556   SetReferenceObject( thePolyline, DataTag_Polyline );
557
558   // Indicate model of the need to update zones splitting
559   if( !IsEqual( aPrevPolyline, thePolyline ) )
560     Changed( Geom_2d );
561 }
562
563 Handle(HYDROData_PolylineXY) HYDROData_CalculationCase::GetBoundaryPolyline() const
564 {
565   return Handle(HYDROData_PolylineXY)::DownCast( 
566            GetReferenceObject( DataTag_Polyline ) );
567 }
568
569 void HYDROData_CalculationCase::RemoveBoundaryPolyline()
570 {
571   Handle(HYDROData_PolylineXY) aPrevPolyline = GetBoundaryPolyline();
572
573   ClearReferenceObjects( DataTag_Polyline );
574
575   // Indicate model of the need to update zones splitting
576   Changed( Geom_2d );
577 }
578
579 void HYDROData_CalculationCase::SetStricklerTable( const Handle(HYDROData_StricklerTable)& theStricklerTable )
580 {
581   Handle(HYDROData_StricklerTable) aPrevStricklerTable = GetStricklerTable();
582
583   SetReferenceObject( theStricklerTable, DataTag_StricklerTable );
584
585   // Indicate model of the need to update land covers partition
586   if( !IsEqual( aPrevStricklerTable, theStricklerTable ) )
587     Changed( Geom_No );
588 }
589
590 Handle(HYDROData_StricklerTable) HYDROData_CalculationCase::GetStricklerTable() const
591 {
592   return Handle(HYDROData_StricklerTable)::DownCast( 
593            GetReferenceObject( DataTag_StricklerTable ) );
594 }
595
596 void HYDROData_CalculationCase::RemoveStricklerTable()
597 {
598   Handle(HYDROData_StricklerTable) aPrevStricklerTable = GetStricklerTable();
599
600   ClearReferenceObjects( DataTag_StricklerTable );
601
602   // Indicate model of the need to update land covers partition
603   Changed( Geom_No );
604 }
605
606 Handle(HYDROData_LandCoverMap) HYDROData_CalculationCase::GetLandCoverMap() const
607 {
608   Handle(HYDROData_LandCoverMap) aMap = Handle(HYDROData_LandCoverMap)::DownCast(
609     GetReferenceObject( DataTag_LandCoverMap ) );
610   return aMap;
611 }
612
613 void HYDROData_CalculationCase::SetLandCoverMap( const Handle(HYDROData_LandCoverMap)& theMap )
614 {
615   SetReferenceObject( theMap, DataTag_LandCoverMap );
616 }
617
618 Handle(HYDROData_Region) HYDROData_CalculationCase::AddNewRegion( const Handle(HYDROData_Zone)& theZone )
619 {
620   Changed( Geom_No );
621   Changed( Geom_No );
622   Changed( Geom_No );
623   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
624   Handle(HYDROData_Region) aNewRegion = addNewRegion( aDocument, CALCULATION_REGIONS_PREF );
625   if ( aNewRegion.IsNull() )
626     return aNewRegion;
627
628   aNewRegion->AddZone( theZone );
629
630   return aNewRegion;
631 }
632
633 bool HYDROData_CalculationCase::AddRegion( const Handle(HYDROData_Region)& theRegion )
634 {
635   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
636
637   if ( theRegion.IsNull() )
638     return false;
639   
640   HYDROData_CalculationCase::DataTag aDataTag = DataTag_Region;
641
642   if ( HasReference( theRegion, aDataTag ) )
643     return false; // Object is already in reference list
644
645   // Move the region from other calculation
646   Handle(HYDROData_CalculationCase) aFatherCalc = 
647     Handle(HYDROData_CalculationCase)::DownCast( theRegion->GetFatherObject() );
648   if ( !aFatherCalc.IsNull() && aFatherCalc->Label() != myLab )
649   {
650     Handle(HYDROData_Region) aNewRegion = addNewRegion( aDocument, CALCULATION_REGIONS_PREF );
651     theRegion->CopyTo( aNewRegion, true );
652
653     aFatherCalc->RemoveRegion( theRegion );
654
655     theRegion->SetLabel( aNewRegion->Label() );
656   }
657   else
658   {
659     AddReferenceObject( theRegion, aDataTag );
660   }
661
662   return true;
663 }
664
665 HYDROData_SequenceOfObjects HYDROData_CalculationCase::GetRegions() const
666 {
667   return GetReferenceObjects( DataTag_Region );
668 }
669
670 void HYDROData_CalculationCase::UpdateRegionsOrder()
671 {
672   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
673   if ( aDocument.IsNull() )
674     return;
675
676   HYDROData_SequenceOfObjects aRegions = GetRegions();
677   HYDROData_SequenceOfObjects::Iterator anIter( aRegions );
678   for ( ; anIter.More(); anIter.Next() )
679   {
680     Handle(HYDROData_Region) aRegion =
681       Handle(HYDROData_Region)::DownCast( anIter.Value() );
682     if ( aRegion.IsNull() )
683       continue;
684
685     aRegion->SetName( "" );
686   }
687
688   QString aRegsPref = CALCULATION_REGIONS_PREF;
689
690   anIter.Init( aRegions );
691   for ( ; anIter.More(); anIter.Next() )
692   {
693     Handle(HYDROData_Region) aRegion =
694       Handle(HYDROData_Region)::DownCast( anIter.Value() );
695     if ( aRegion.IsNull() )
696       continue;
697
698     QString aRegionName = HYDROData_Tool::GenerateObjectName( aDocument, aRegsPref );
699     aRegion->SetName( aRegionName );
700   }
701 }
702
703 void HYDROData_CalculationCase::RemoveRegion( const Handle(HYDROData_Region)& theRegion )
704 {
705   if ( theRegion.IsNull() )
706     return;
707
708   HYDROData_CalculationCase::DataTag aDataTag = DataTag_Region;
709   RemoveReferenceObject( theRegion->Label(), aDataTag );
710
711   // Remove region from data model
712   Handle(HYDROData_CalculationCase) aFatherCalc = 
713     Handle(HYDROData_CalculationCase)::DownCast( theRegion->GetFatherObject() );
714   if ( !aFatherCalc.IsNull() && aFatherCalc->Label() == myLab )
715     theRegion->Remove();
716 }
717
718 void HYDROData_CalculationCase::RemoveRegions()
719 {
720   myLab.FindChild( DataTag_ChildRegion ).ForgetAllAttributes();
721 }
722
723 HYDROData_SequenceOfObjects HYDROData_CalculationCase::GetSplitGroups() const
724 {
725   return GetReferenceObjects( DataTag_SplitGroups );
726 }
727
728 void HYDROData_CalculationCase::RemoveSplitGroups()
729 {
730   myLab.FindChild( DataTag_SplitGroups ).ForgetAllAttributes();
731 }
732
733 double HYDROData_CalculationCase::GetAltitudeForPoint( const gp_XY& thePoint ) const
734 {
735   Handle(HYDROData_Zone) aZone = GetZoneFromPoint( thePoint );
736   return GetAltitudeForPoint( thePoint, aZone );
737 }
738
739 double HYDROData_CalculationCase::GetAltitudeForPoint( const gp_XY&                    thePoint,
740                                                        const Handle(HYDROData_Region)& theRegion,
741                                                        int theMethod) const
742 {
743   double aResAltitude = HYDROData_IAltitudeObject::GetInvalidAltitude();
744
745   Handle(HYDROData_Zone) aZone = GetZoneFromPoint( thePoint );
746   if ( !aZone.IsNull() )
747   {
748     //DEBTRACE("GetAltitudeForPoint Region " << theRegion->GetName().toStdString() << " Zone " << aZone->GetName().toStdString());
749     Handle(HYDROData_Region) aRefRegion = Handle(HYDROData_Region)::DownCast( aZone->GetFatherObject() );
750     if ( IsEqual( aRefRegion, theRegion ) )
751       aResAltitude = GetAltitudeForPoint( thePoint, aZone, theMethod );
752     else
753       {
754         //DEBTRACE("GetAltitudeForPoint Region " << aRefRegion->GetName().toStdString() << " Zone " << aZone->GetName().toStdString() << " ---------------------------");
755         aResAltitude = GetAltitudeForPoint( thePoint, aZone, theMethod );
756       }
757   }
758   else
759     {
760       //DEBTRACE(" --- GetAltitudeForPoint No Zone ---");
761     }
762
763   return aResAltitude;
764 }
765
766 double HYDROData_CalculationCase::GetAltitudeForPoint( const gp_XY&                  thePoint,
767                                                        const Handle(HYDROData_Zone)& theZone,
768                                                        int theMethod) const
769 {
770   //DEBTRACE("GetAltitudeForPoint Zone " << theZone->GetName().toStdString());
771   double aResAltitude = HYDROData_IAltitudeObject::GetInvalidAltitude();
772   if ( theZone.IsNull() )
773   {
774         DEBTRACE("Zone nulle");
775     return aResAltitude;
776   }
777
778   HYDROData_Zone::MergeType aZoneMergeType = theZone->GetMergeType();
779   //DEBTRACE("aZoneMergeType " << aZoneMergeType);
780   if ( !theZone->IsMergingNeed() )
781   {
782     aZoneMergeType = HYDROData_Zone::Merge_UNKNOWN;
783     //DEBTRACE("---");
784   }
785   else if ( aZoneMergeType == HYDROData_Zone::Merge_UNKNOWN )
786   {
787         //DEBTRACE("GetAltitudeForPoint Zone " << theZone->GetName().toStdString() << " Merge_UNKNOWN");
788     return aResAltitude;
789   }
790
791   HYDROData_IInterpolator* aZoneInterpolator = theZone->GetInterpolator();
792   if ( aZoneMergeType == HYDROData_Zone::Merge_Object )
793   {
794     Handle(HYDROData_IAltitudeObject) aMergeAltitude = 
795       Handle(HYDROData_IAltitudeObject)::DownCast( theZone->GetMergeObject() );
796     if ( !aMergeAltitude.IsNull() )
797     {
798       if ( aZoneInterpolator != NULL )
799       {
800         DEBTRACE("aZoneInterpolator != NULL");
801         aZoneInterpolator->SetAltitudeObject( aMergeAltitude );
802         aResAltitude = aZoneInterpolator->GetAltitudeForPoint( thePoint );
803       }
804       else
805       {
806         DEBTRACE("aZoneInterpolator == NULL");
807         aResAltitude = aMergeAltitude->GetAltitudeForPoint( thePoint );
808       }
809     }
810   }
811   else
812   {
813         //DEBTRACE("aZoneMergeType != HYDROData_Zone::Merge_Object");
814     HYDROData_SequenceOfObjects aZoneObjects = theZone->GetObjects();
815     HYDROData_SequenceOfObjects::Iterator anIter( aZoneObjects );
816     for ( ; anIter.More(); anIter.Next() )
817     {
818       Handle(HYDROData_Object) aZoneObj =
819         Handle(HYDROData_Object)::DownCast( anIter.Value() );
820       if ( aZoneObj.IsNull() )
821         continue;
822
823       Handle(HYDROData_IAltitudeObject) anObjAltitude = aZoneObj->GetAltitudeObject();
824       if ( anObjAltitude.IsNull() )
825         continue;
826
827       double aPointAltitude = 0.0;
828       if ( aZoneInterpolator != NULL )
829       {
830         DEBTRACE("aZoneInterpolator != NULL");
831         aZoneInterpolator->SetAltitudeObject( anObjAltitude );
832         aPointAltitude = aZoneInterpolator->GetAltitudeForPoint( thePoint );
833       }
834       else
835       {
836         //DEBTRACE("aZoneInterpolator == NULL");
837         aPointAltitude = anObjAltitude->GetAltitudeForPoint( thePoint, theMethod );
838       }
839
840       if ( ValuesEquals( aPointAltitude, HYDROData_IAltitudeObject::GetInvalidAltitude() ) )
841         continue;
842
843       if ( aZoneMergeType == HYDROData_Zone::Merge_UNKNOWN )
844       {
845         aResAltitude = aPointAltitude;
846         break;
847       }
848       else if ( aZoneMergeType == HYDROData_Zone::Merge_ZMIN )
849       {
850         if ( ValuesEquals( aResAltitude, HYDROData_IAltitudeObject::GetInvalidAltitude() ) ||
851              aResAltitude > aPointAltitude )
852         {
853           aResAltitude = aPointAltitude;
854         }
855       }
856       else if ( aZoneMergeType == HYDROData_Zone::Merge_ZMAX )
857       {
858         if ( ValuesEquals( aResAltitude, HYDROData_IAltitudeObject::GetInvalidAltitude() ) ||
859              aResAltitude < aPointAltitude )
860         {
861           aResAltitude = aPointAltitude;
862         }
863       }
864     }
865   }
866
867   return aResAltitude;
868 }
869
870 NCollection_Sequence<double> HYDROData_CalculationCase::GetAltitudesForPoints( 
871   const NCollection_Sequence<gp_XY>& thePoints,
872   const Handle(HYDROData_Region)&    theRegion,
873   int theMethod) const
874 {
875   DEBTRACE("HYDROData_CalculationCase::GetAltitudesForPoints " << theRegion->GetName().toStdString());
876   NCollection_Sequence<double> aResSeq;
877
878   for ( int i = 1, n = thePoints.Length(); i <= n; ++i )
879   {
880     const gp_XY& thePnt = thePoints.Value( i );
881     
882     double anAltitude = GetAltitudeForPoint( thePnt, theRegion, theMethod );
883     aResSeq.Append( anAltitude );
884   }
885
886   return aResSeq;
887 }
888
889 NCollection_Sequence<double> HYDROData_CalculationCase::GetAltitudesForPoints( 
890   const NCollection_Sequence<gp_XY>& thePoints,
891   const Handle(HYDROData_Zone)&      theZone,
892   int theMethod) const
893 {
894   NCollection_Sequence<double> aResSeq;
895
896   for ( int i = 1, n = thePoints.Length(); i <= n; ++i )
897   {
898     const gp_XY& thePnt = thePoints.Value( i );
899     
900     double anAltitude = GetAltitudeForPoint( thePnt, theZone, theMethod );
901     aResSeq.Append( anAltitude );
902   }
903
904   return aResSeq;
905 }
906
907 double HYDROData_CalculationCase::GetStricklerCoefficientForPoint( const gp_XY& thePoint ) const
908 {
909   Handle( HYDROData_LandCoverMap ) aMap = GetLandCoverMap();
910   Handle( HYDROData_StricklerTable ) aTable = GetStricklerTable();
911   if( aMap.IsNull() )
912     return 0.0;
913
914   QString aType;
915   aMap->FindByPoint( thePoint, aType );
916   double aCoeff = aTable->Get( aType, 0.0 );
917   return aCoeff;
918 }
919
920 std::vector<double> HYDROData_CalculationCase::GetStricklerCoefficientForPoints(const std::vector<gp_XY>& thePoints,
921   double DefValue, bool UseMax ) const
922 {
923   Handle( HYDROData_LandCoverMap ) aLCM = GetLandCoverMap();
924   Handle( HYDROData_StricklerTable ) aTable = GetStricklerTable();
925   std::vector<double> theCoeffs;
926   if( aLCM.IsNull() || aTable.IsNull() )
927     return theCoeffs;
928
929   aLCM->ClassifyPoints(thePoints, aTable, theCoeffs, DefValue, UseMax );
930
931   return theCoeffs;
932 }
933
934 std::vector<int> HYDROData_CalculationCase::GetStricklerTypeForPoints( const std::vector<gp_XY>& thePoints ) const
935 {
936   Handle( HYDROData_LandCoverMap ) aLCM = GetLandCoverMap();
937   Handle( HYDROData_StricklerTable ) aTable = GetStricklerTable();
938   std::vector<int> types;
939   if( aLCM.IsNull() || aTable.IsNull() )
940     return types;
941
942   aLCM->ClassifyPoints(thePoints, aTable, types );
943   return types;
944 }
945
946 Handle(HYDROData_Region) HYDROData_CalculationCase::GetRegionFromPoint( const gp_XY& thePoint ) const
947 {
948   Handle(HYDROData_Region) aResRegion;
949
950   Handle(HYDROData_Zone) aZone = GetZoneFromPoint( thePoint );
951   if ( !aZone.IsNull() )
952     aResRegion = Handle(HYDROData_Region)::DownCast( aZone->GetFatherObject() );
953
954   return aResRegion;
955 }
956
957 Handle(HYDROData_Zone) HYDROData_CalculationCase::GetZoneFromPoint( const gp_XY& thePoint ) const
958 {
959   Handle(HYDROData_Zone) aResZone;
960
961   HYDROData_SequenceOfObjects aRegions = GetRegions();
962
963   HYDROData_SequenceOfObjects::Iterator anIter( aRegions );
964   for ( ; anIter.More() && aResZone.IsNull(); anIter.Next() )
965   {
966     Handle(HYDROData_Region) aRegion =
967       Handle(HYDROData_Region)::DownCast( anIter.Value() );
968     if ( aRegion.IsNull() )
969       continue;
970
971     HYDROData_SequenceOfObjects aZones = aRegion->GetZones();
972     HYDROData_SequenceOfObjects::Iterator aZonesIter( aZones );
973     for ( ; aZonesIter.More() && aResZone.IsNull(); aZonesIter.Next() )
974     {
975       Handle(HYDROData_Zone) aRegZone =
976         Handle(HYDROData_Zone)::DownCast( aZonesIter.Value() );
977       if ( aRegZone.IsNull() )
978         continue;
979
980       PointClassification aPointRelation = GetPointClassification( thePoint, aRegZone );
981       if ( aPointRelation != POINT_OUT )
982         aResZone = aRegZone; // We found the desired zone
983     }
984   }
985
986   return aResZone;
987 }
988
989 HYDROData_CalculationCase::PointClassification HYDROData_CalculationCase::GetPointClassification(
990   const gp_XY&                  thePoint,
991   const Handle(HYDROData_Zone)& theZone ) const
992 {
993   PointClassification aRes = POINT_OUT;
994   if ( theZone.IsNull() )
995     return aRes;
996
997   TopoDS_Face aZoneFace = TopoDS::Face( theZone->GetShape() );
998   if ( aZoneFace.IsNull() )
999     return aRes;
1000 #ifdef DEB_CLASS2D      
1001           TopoDS_Compound aCmp;
1002       BRep_Builder aBB;
1003       aBB.MakeCompound(aCmp);
1004           aBB.Add(aCmp, aZoneFace);
1005           gp_Pnt aPnt (thePoint.X(), thePoint.Y(), 0.);
1006           BRepBuilderAPI_MakeVertex aMk(aPnt);
1007           aBB.Add(aCmp, aMk.Vertex());
1008           BRepTools::Write(aCmp, "FCL2d.brep");
1009 #endif  
1010   TopAbs_State State = HYDROData_Tool::ComputePointState(thePoint, aZoneFace);
1011   if (State == TopAbs_OUT)
1012     aRes =  POINT_OUT;
1013   else if(State == TopAbs_IN)
1014     aRes =  POINT_IN;
1015   else if(State == TopAbs_ON)
1016     aRes =  POINT_ON;
1017   return aRes;
1018 }
1019
1020 Handle(HYDROData_Region) HYDROData_CalculationCase::addNewRegion( const Handle(HYDROData_Document)& theDoc,
1021                                                                   const QString& thePrefixOrName,
1022                                                                   bool isPrefix )
1023 {
1024   TDF_Label aNewLab = myLab.FindChild( DataTag_ChildRegion ).NewChild();
1025   int aTag = aNewLab.Tag();
1026
1027   Handle(HYDROData_Region) aNewRegion =
1028     Handle(HYDROData_Region)::DownCast( HYDROData_Iterator::CreateObject( aNewLab, KIND_REGION ) );
1029   AddRegion( aNewRegion );
1030
1031   QString aRegionName = isPrefix ? HYDROData_Tool::GenerateObjectName( theDoc, thePrefixOrName ) : thePrefixOrName;
1032   aNewRegion->SetName( aRegionName, true );
1033
1034   return aNewRegion;
1035 }
1036
1037 Handle(HYDROData_SplitShapesGroup) HYDROData_CalculationCase::addNewSplitGroup( const QString& theName )
1038 {
1039   TDF_Label aNewLab = myLab.FindChild( DataTag_SplitGroups ).NewChild();
1040
1041   Handle(HYDROData_SplitShapesGroup) aNewGroup =
1042     Handle(HYDROData_SplitShapesGroup)::DownCast( 
1043       HYDROData_Iterator::CreateObject( aNewLab, KIND_SPLIT_GROUP ) );
1044   AddReferenceObject( aNewGroup, DataTag_SplitGroups );
1045
1046   aNewGroup->SetName( theName );
1047
1048   return aNewGroup;
1049 }
1050
1051 QString HYDROData_CalculationCase::Export( int theStudyId ) const
1052 {
1053 #ifdef LIGHT_MODE
1054   return "";
1055 #else
1056   GEOM::GEOM_Gen_var aGEOMEngine = HYDROData_GeomTool::GetGeomGen();
1057   SALOMEDS::Study_var aDSStudy = HYDROData_GeomTool::GetStudyByID( theStudyId );
1058   
1059   QString aGeomObjEntry, anErrorMsg;
1060   QString statMess;
1061   bool isOK = Export( aGEOMEngine, aDSStudy, aGeomObjEntry, anErrorMsg, statMess );
1062   return isOK ? aGeomObjEntry : QString();
1063 #endif
1064 }
1065
1066 #ifndef LIGHT_MODE
1067 bool HYDROData_CalculationCase::Export( GEOM::GEOM_Gen_var  theGeomEngine,
1068                                         SALOMEDS::Study_ptr theStudy,
1069                                         QString& theGeomObjEntry,
1070                                         QString& theErrorMsg,
1071                                         QString& statMess) const
1072 {
1073   HYDROData_ShapesGroup::SeqOfGroupsDefs aSeqOfGroupsDefs;
1074
1075   // Get groups definitions
1076   HYDROData_SequenceOfObjects aSplitGroups = GetSplitGroups();
1077
1078   TopTools_SequenceOfShape IntSh; //internal edges
1079   HYDROData_SequenceOfObjects::Iterator anIter( aSplitGroups );
1080   for ( ; anIter.More(); anIter.Next() )
1081   {
1082     // Get shapes group
1083     Handle(HYDROData_ShapesGroup) aGroup =
1084       Handle(HYDROData_ShapesGroup)::DownCast( anIter.Value() );
1085     if ( aGroup.IsNull() )
1086       continue;
1087
1088     HYDROData_ShapesGroup::GroupDefinition aGroupDef;
1089
1090     aGroupDef.Name = aGroup->GetName().toLatin1().constData();
1091     aGroup->GetShapes( aGroupDef.Shapes );
1092
1093     aSeqOfGroupsDefs.Append( aGroupDef );
1094
1095     Handle(HYDROData_SplitShapesGroup) aSSGroup = Handle(HYDROData_SplitShapesGroup)::DownCast( anIter.Value() );
1096     TopTools_SequenceOfShape dummy;
1097     if (!aSSGroup.IsNull())
1098       if (aSSGroup->GetInternal())
1099       {
1100         aSSGroup->GetShapes(dummy);
1101         IntSh.Append(dummy);
1102       }
1103
1104   }
1105   
1106   // Get faces
1107   bool isAllNotSubmersible = true;
1108   HYDROData_SequenceOfObjects aCaseRegions = GetRegions();
1109   HYDROData_SequenceOfObjects::Iterator aRegionIter( aCaseRegions );
1110   NCollection_IndexedDataMap<TopoDS_Shape, QString> aShToNames;
1111   for ( ; aRegionIter.More(); aRegionIter.Next() )
1112   {
1113     Handle(HYDROData_Region) aRegion =
1114       Handle(HYDROData_Region)::DownCast( aRegionIter.Value() );
1115     if( aRegion.IsNull() || !aRegion->IsSubmersible() )
1116       continue;
1117     
1118     if ( isAllNotSubmersible )
1119       isAllNotSubmersible = false;
1120
1121     TopoDS_Shape aRegionShape = aRegion->GetShape( &aSeqOfGroupsDefs, &IntSh );
1122     aShToNames.Add( aRegionShape, aRegion->GetName() );
1123   }
1124
1125   bool aRes = false;
1126
1127   if ( aCaseRegions.IsEmpty() ) {
1128     theErrorMsg = QString("the list of regions is empty.");
1129   } else if ( isAllNotSubmersible ) {
1130     theErrorMsg = QString("there are no submersible regions.");
1131   } else {
1132     aRes = Export( theGeomEngine, theStudy, aShToNames, aSeqOfGroupsDefs, theGeomObjEntry );;
1133   }
1134
1135   if( aRes && !GetLandCoverMap().IsNull() && !GetStricklerTable().IsNull() )
1136   {
1137     QString aTelemacFileName = GetName() + ".telemac";
1138     aRes = GetLandCoverMap()->ExportTelemac( aTelemacFileName, 1E-2, GetStricklerTable(), statMess );
1139     if (!aRes)
1140       theErrorMsg = QString( "The export to TELEMAC %1 failed" ).arg( aTelemacFileName );
1141   }
1142   return aRes;
1143 }
1144
1145 bool HYDROData_CalculationCase::Export( GEOM::GEOM_Gen_var                            theGeomEngine,
1146                                         SALOMEDS::Study_ptr                           theStudy,
1147                                         const NCollection_IndexedDataMap<TopoDS_Shape, QString>& aShToName,
1148                                         const HYDROData_ShapesGroup::SeqOfGroupsDefs& theGroupsDefs,
1149                                         QString& theGeomObjEntry ) const
1150 {
1151   // Sew faces
1152   BRepBuilderAPI_Sewing aSewing( Precision::Confusion() * 10.0 );
1153   aSewing.SetNonManifoldMode( Standard_False );
1154 #ifdef DEB_CALCULATION
1155   TCollection_AsciiString aNam("Sh_");
1156   int i=1;
1157 #endif
1158   TopTools_DataMapOfShapeListOfShape SH2M;
1159   for ( int i = 1; i <= aShToName.Extent(); i++ )
1160   {
1161     const TopoDS_Shape& aShape = aShToName.FindKey(i);
1162     if ( aShape.IsNull() )
1163       continue;
1164     
1165     SH2M.Bind(aShape, TopTools_ListOfShape());
1166     TopTools_ListOfShape& LM = SH2M.ChangeFind(aShape);
1167     if ( aShape.ShapeType() == TopAbs_FACE || aShape.ShapeType() == TopAbs_SHELL )
1168     {
1169       aSewing.Add( aShape );
1170       LM.Append(aShape);
1171     }
1172     else if (aShape.ShapeType() == TopAbs_COMPOUND)
1173     {
1174       TopExp_Explorer anExp( aShape, TopAbs_SHELL );
1175       for (; anExp.More(); anExp.Next() )
1176       {
1177         aSewing.Add( anExp.Current() );
1178         LM.Append(anExp.Current());
1179       }
1180       anExp.Init( aShape, TopAbs_FACE, TopAbs_SHELL );
1181       {
1182         aSewing.Add( anExp.Current() );
1183         LM.Append(anExp.Current());
1184       }
1185     }
1186   } // faces iterator
1187   
1188   aSewing.Perform();
1189   TopoDS_Shape aSewedShape = aSewing.SewedShape();
1190
1191   NCollection_IndexedDataMap<TopoDS_Shape, QString, TopTools_ShapeMapHasher> aFacesToNameModif;
1192
1193   for ( int i = 1; i <= aShToName.Extent(); i++ )
1194   {
1195     const TopoDS_Shape& CurShape = aShToName.FindKey(i);
1196     const QString& Qstr = aShToName.FindFromIndex(i);
1197     const TopTools_ListOfShape& LM = SH2M(CurShape);
1198     for (TopTools_ListIteratorOfListOfShape it(LM); it.More(); it.Next())
1199     {
1200       const TopoDS_Shape& csh = it.Value();
1201       if (aSewing.IsModified(csh))
1202         aFacesToNameModif.Add(aSewing.Modified(csh), Qstr);
1203       else
1204         aFacesToNameModif.Add(csh, Qstr);
1205     }
1206   }
1207
1208  
1209   // If the sewed shape is empty - return false
1210   if ( aSewedShape.IsNull() || !TopoDS_Iterator( aSewedShape ).More() )
1211     return false;
1212
1213 #ifdef DEB_CALCULATION
1214   BRepTools::Write(aSewedShape ,"Sew.brep");
1215 #endif
1216   // Publish the sewed shape
1217   QString aName = EXPORT_NAME;
1218   GEOM::GEOM_Object_ptr aMainShape = 
1219     HYDROData_GeomTool::ExplodeShapeInGEOMandPublish( theGeomEngine, theStudy, aSewedShape, aFacesToNameModif, aName, theGeomObjEntry );
1220
1221   if ( aMainShape->_is_nil() )  
1222     return false;
1223
1224   if ( theGroupsDefs.IsEmpty() )
1225     return true;
1226
1227   // Create groups
1228   TopTools_IndexedMapOfShape aMapOfSubShapes;
1229   TopExp::MapShapes( aSewedShape, aMapOfSubShapes );
1230
1231   NCollection_DataMap< TCollection_AsciiString, NCollection_Sequence<int> > aGroupsData;
1232
1233   for ( int aGrId = 1, nbGroups = theGroupsDefs.Length(); aGrId <= nbGroups; ++aGrId )
1234   {
1235     const HYDROData_ShapesGroup::GroupDefinition& aGroupDef = theGroupsDefs.Value( aGrId );
1236
1237     NCollection_Sequence<int> aGroupIndexes;
1238     for( int i = 1, n = aGroupDef.Shapes.Length(); i <= n; i++ )
1239     {
1240       const TopoDS_Shape& aShape = aGroupDef.Shapes.Value( i );
1241 #ifdef DEB_CALCULATION
1242       cout << "\nOld shape(" << i << ") = " << aShape.TShape() <<endl;
1243 #endif
1244       
1245       TopoDS_Shape aModifiedShape = aShape;
1246       if ( aSewing.IsModified( aShape ) )
1247         aModifiedShape = aSewing.Modified( aShape );
1248       else if ( aSewing.IsModifiedSubShape( aShape ) )
1249         aModifiedShape = aSewing.ModifiedSubShape( aShape );
1250
1251 #ifdef DEB_CALCULATION
1252       const TopLoc_Location& aL1 = aShape.Location();
1253       const TopLoc_Location& aL2 = aModifiedShape.Location();
1254       cout << "\nNew shape(" << i << ") = " << aModifiedShape.TShape() << " Location is Equal = " << aL1.IsEqual(aL2)<<endl;
1255 #endif
1256
1257       int anIndex = aMapOfSubShapes.FindIndex(aModifiedShape);
1258       if ( anIndex > 0 ) {
1259         aGroupIndexes.Append( anIndex );
1260       } else {
1261 #ifdef DEB_CALCULATION    
1262         TCollection_AsciiString aNam("Lost_");
1263         if(!aMapOfSubShapes.Contains(aModifiedShape)) {
1264         for ( int anIndex = 1; anIndex <= aMapOfSubShapes.Extent(); anIndex++ )
1265         {
1266            const TopoDS_Shape& aS = aMapOfSubShapes.FindKey( anIndex );
1267            if ( aModifiedShape.IsPartner( aS ) )
1268            {
1269              cout <<"\nIndex in Map = " << anIndex << "TShape = " << aS.TShape() <<endl;
1270              TCollection_AsciiString aName = aNam + i + "_" + anIndex + ".brep";
1271              BRepTools::Write(aS ,aName.ToCString());
1272             break;
1273            }
1274          }
1275         }
1276 #endif
1277       }
1278     }
1279     if ( !aGroupIndexes.IsEmpty() )
1280       aGroupsData.Bind( aGroupDef.Name, aGroupIndexes );
1281   }
1282  
1283   if ( !aGroupsData.IsEmpty() )
1284   {
1285     GEOM::GEOM_IGroupOperations_var aGroupOp = 
1286       theGeomEngine->GetIGroupOperations( theStudy->StudyId() );  
1287
1288     NCollection_DataMap< TCollection_AsciiString, NCollection_Sequence<int> >::Iterator aMapIt( aGroupsData );
1289     for ( ; aMapIt.More(); aMapIt.Next() )
1290     {
1291       const TCollection_AsciiString& aGroupName = aMapIt.Key(); 
1292       const NCollection_Sequence<int>& aGroupIndexes = aMapIt.Value();
1293
1294       GEOM::GEOM_Object_var aGeomGroup = aGroupOp->CreateGroup( aMainShape, TopAbs_EDGE );
1295       if ( CORBA::is_nil( aGeomGroup ) || !aGroupOp->IsDone() )
1296         continue;
1297
1298       GEOM::ListOfLong_var aGeomIndexes = new GEOM::ListOfLong;
1299       aGeomIndexes->length( aGroupIndexes.Length() );
1300
1301       for( int i = 1, n = aGroupIndexes.Length(); i <= n; i++ )
1302         aGeomIndexes[ i - 1 ] = aGroupIndexes.Value( i );
1303
1304       aGroupOp->UnionIDs( aGeomGroup, aGeomIndexes );
1305       if ( aGroupOp->IsDone() )
1306       {
1307         SALOMEDS::SObject_var aGroupSO = 
1308           theGeomEngine->AddInStudy( theStudy, aGeomGroup, aGroupName.ToCString(), aMainShape );
1309       }
1310     }
1311   }
1312
1313   return true;
1314 }
1315 #endif
1316
1317 void HYDROData_CalculationCase::ClearRules( HYDROData_CalculationCase::DataTag theDataTag,
1318                                             const bool theIsSetToUpdate )
1319 {
1320   TDF_Label aRulesLab = myLab.FindChild( theDataTag );
1321   HYDROData_PriorityQueue::ClearRules( aRulesLab );
1322
1323   // Indicate model of the need to update splitting
1324   if ( theIsSetToUpdate ) {
1325     Changed( Geom_2d );
1326   }
1327 }
1328
1329 void HYDROData_CalculationCase::AddRule( const Handle(HYDROData_Entity)&    theObject1,
1330                                          HYDROData_PriorityType             thePriority,
1331                                          const Handle(HYDROData_Entity)&    theObject2,
1332                                          HYDROData_Zone::MergeType          theMergeType,
1333                                          HYDROData_CalculationCase::DataTag theDataTag )
1334 {
1335   TDF_Label aRulesLab = myLab.FindChild( theDataTag );
1336   HYDROData_PriorityQueue::AddRule( aRulesLab, theObject1, thePriority, theObject2, theMergeType );
1337
1338   // Indicate model of the need to update splitting
1339   Changed( Geom_2d );
1340 }
1341
1342 QString HYDROData_CalculationCase::DumpRules() const
1343 {
1344   TDF_Label aRulesLab = myLab.FindChild( DataTag_CustomRules );
1345   return HYDROData_PriorityQueue::DumpRules( aRulesLab );
1346 }
1347
1348 void HYDROData_CalculationCase::SetAssignmentMode( AssignmentMode theMode )
1349 {
1350   TDF_Label aModeLab = myLab.FindChild( DataTag_AssignmentMode );
1351   TDataStd_Integer::Set( aModeLab, ( int ) theMode );
1352
1353   // Indicate model of the need to update splitting
1354   Changed( Geom_2d );
1355 }
1356
1357 HYDROData_CalculationCase::AssignmentMode HYDROData_CalculationCase::GetAssignmentMode() const
1358 {
1359   Handle(TDataStd_Integer) aModeAttr;
1360   bool isOK = myLab.FindChild( DataTag_AssignmentMode ).FindAttribute( TDataStd_Integer::GetID(), aModeAttr );
1361   if( isOK )
1362     return ( AssignmentMode ) aModeAttr->Get();
1363   else
1364     return MANUAL;
1365 }
1366
1367 void HYDROData_CalculationCase::DumpRulesToPython( const QString& theCalcCaseName,
1368                                                    QStringList& theScript ) const
1369 {
1370   TDF_Label aRulesLab = myLab.FindChild( DataTag_CustomRules );
1371   HYDROData_PriorityQueue::DumpRulesToPython( aRulesLab, theCalcCaseName, theScript );
1372 }
1373
1374 HYDROData_Warning HYDROData_CalculationCase::GetLastWarning() const
1375 {
1376   return myLastWarning;
1377 }
1378
1379 void HYDROData_CalculationCase::SetWarning( HYDROData_WarningType theType, const QString& theData )
1380 {
1381   myLastWarning.Type = theType;
1382   myLastWarning.Data = theData;
1383 }
1384
1385 void HYDROData_CalculationCase::UpdateRegionsNames( const HYDROData_SequenceOfObjects& theRegions,
1386                                                     const QString& theOldCaseName,
1387                                                     const QString& theName )
1388 {
1389   HYDROData_SequenceOfObjects::Iterator anIter( theRegions );
1390   for ( ; anIter.More(); anIter.Next() )
1391   {
1392     Handle(HYDROData_Region) aRegion =
1393       Handle(HYDROData_Region)::DownCast( anIter.Value() );
1394     if ( aRegion.IsNull() )
1395       continue;
1396
1397     HYDROData_Tool::UpdateChildObjectName( theOldCaseName, theName, aRegion );
1398
1399     HYDROData_SequenceOfObjects aZones = aRegion->GetZones();
1400     HYDROData_SequenceOfObjects::Iterator anIter( aZones );
1401     for ( ; anIter.More(); anIter.Next() )
1402     {
1403       Handle(HYDROData_Zone) aRegZone =
1404         Handle(HYDROData_Zone)::DownCast( anIter.Value() );
1405       if ( aRegZone.IsNull() )
1406         continue;
1407
1408       HYDROData_Tool::UpdateChildObjectName( theOldCaseName, theName, aRegZone );
1409     }
1410   }
1411 }
1412
1413 void HYDROData_CalculationCase::DumpRegionsToPython( QStringList& theResList,
1414                                                      const QString& thePyScriptPath,
1415                                                      MapOfTreatedObjects& theTreatedObjects,
1416                                                      const HYDROData_SequenceOfObjects& theRegions ) const
1417 {
1418   HYDROData_SequenceOfObjects::Iterator anIter;
1419   anIter.Init(theRegions);
1420   for (int ireg = 1; anIter.More(); anIter.Next(), ireg++)
1421     {
1422       Handle(HYDROData_Region) aRegion = Handle(HYDROData_Region)::DownCast(anIter.Value());
1423       if (aRegion.IsNull())
1424         continue;
1425       QString defRegName = this->GetName();
1426       QString regSuffix = QString("_Reg_%1").arg(ireg);
1427       defRegName += regSuffix;
1428       theTreatedObjects.insert(aRegion->GetName(), aRegion);
1429       QStringList aRegDump = aRegion->DumpToPython(thePyScriptPath, theTreatedObjects, defRegName);
1430       theResList << aRegDump;
1431     }
1432   for (anIter.Init(theRegions); anIter.More(); anIter.Next())
1433     {
1434       Handle(HYDROData_Region) aRegion = Handle(HYDROData_Region)::DownCast(anIter.Value());
1435       if (aRegion.IsNull())
1436         continue;
1437       QStringList aRegDump;
1438       aRegion->SetNameInDumpPython(aRegDump);
1439       theResList << aRegDump;
1440     }
1441 }
1442
1443 bool HYDROData_CalculationCase::GetRule( int theIndex, 
1444                                          Handle(HYDROData_Entity)&           theObject1,
1445                                          HYDROData_PriorityType&             thePriority,
1446                                          Handle(HYDROData_Entity)&           theObject2,
1447                                          HYDROData_Zone::MergeType&          theMergeType,
1448                                          HYDROData_CalculationCase::DataTag& theDataTag) const
1449 {
1450   TDF_Label aRulesLab = myLab.FindChild( theDataTag );
1451   return HYDROData_PriorityQueue::GetRule( aRulesLab, theIndex,
1452     theObject1, thePriority, theObject2, theMergeType );
1453 }
1454
1455 bool HYDROData_CalculationCase::AddInterPoly( const Handle(HYDROData_PolylineXY)& theInterPolyline )
1456 {
1457   HYDROData_CalculationCase::DataTag aDataTag = DataTag_InterPoly;
1458
1459   if ( HasReference( theInterPolyline, aDataTag ) )
1460     return false; 
1461
1462   AddReferenceObject( theInterPolyline, aDataTag );
1463
1464   Changed( Geom_2d );
1465
1466   return true;
1467 }
1468
1469 HYDROData_SequenceOfObjects HYDROData_CalculationCase::GetInterPolyObjects() const
1470 {
1471   return GetReferenceObjects( DataTag_InterPoly ); 
1472 }
1473
1474 void HYDROData_CalculationCase::RemoveInterPolyObject( const Handle(HYDROData_PolylineXY)& theInterPolyline ) 
1475 {
1476   if ( theInterPolyline.IsNull() )
1477     return;
1478
1479   RemoveReferenceObject( theInterPolyline->Label(), DataTag_InterPoly );
1480
1481   Changed( Geom_2d );
1482 }
1483
1484
1485