]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROData/HYDROData_ImmersibleZone.cxx
Salome HOME
Merge remote-tracking branch 'origin/BR_IMPROVEMENTS' into BR_v14_rc
[modules/hydro.git] / src / HYDROData / HYDROData_ImmersibleZone.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_ImmersibleZone.h"
20
21 #include "HYDROData_IAltitudeObject.h"
22 #include "HYDROData_Document.h"
23 #include "HYDROData_ShapesGroup.h"
24 #include "HYDROData_PolylineXY.h"
25 #include "HYDROData_ShapesTool.h"
26
27 #include <BRepBuilderAPI_MakeFace.hxx>
28
29 #include <TopoDS.hxx>
30 #include <TopoDS_Face.hxx>
31 #include <TopoDS_Wire.hxx>
32 #include <TopoDS_Compound.hxx>
33 #include <TopExp_Explorer.hxx>
34 #include <TopTools_ListIteratorOfListOfShape.hxx>
35
36 #include <BRep_Builder.hxx>
37 #include <BRepAlgo_FaceRestrictor.hxx>
38 #include <BRepCheck_Analyzer.hxx>
39
40 #include <ShapeAnalysis.hxx>
41
42 #include <QColor>
43 #include <QStringList>
44
45 //#define HYDRODATA_IMZONE_DEB 1
46
47 IMPLEMENT_STANDARD_HANDLE(HYDROData_ImmersibleZone,HYDROData_NaturalObject)
48 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_ImmersibleZone,HYDROData_NaturalObject)
49
50
51 HYDROData_ImmersibleZone::HYDROData_ImmersibleZone()
52 : HYDROData_NaturalObject()
53 {
54 }
55
56 HYDROData_ImmersibleZone::~HYDROData_ImmersibleZone()
57 {
58 }
59
60 QStringList HYDROData_ImmersibleZone::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
61 {
62   QStringList aResList = dumpObjectCreation( theTreatedObjects );
63   
64   QString aZoneName = GetObjPyName();
65
66   Handle(HYDROData_IAltitudeObject) aRefAltitude = GetAltitudeObject();
67   setPythonReferenceObject( theTreatedObjects, aResList, aRefAltitude, "SetAltitudeObject" );
68
69   Handle(HYDROData_PolylineXY) aRefPolyline = GetPolyline();
70   setPythonReferenceObject( theTreatedObjects, aResList, aRefPolyline, "SetPolyline" );
71
72   aResList << QString( "" );
73
74   aResList << QString( "%1.Update();" ).arg( aZoneName );
75   aResList << QString( "" );
76
77   return aResList;
78 }
79
80 HYDROData_SequenceOfObjects HYDROData_ImmersibleZone::GetAllReferenceObjects() const
81 {
82   HYDROData_SequenceOfObjects aResSeq = HYDROData_NaturalObject::GetAllReferenceObjects();
83
84   Handle(HYDROData_PolylineXY) aRefPolyline = GetPolyline();
85   if ( !aRefPolyline.IsNull() )
86     aResSeq.Append( aRefPolyline );
87
88   return aResSeq;
89 }
90
91 TopoDS_Shape HYDROData_ImmersibleZone::GetTopShape() const
92 {
93   return getTopShape();
94 }
95
96 void HYDROData_ImmersibleZone::Update()
97 {
98   HYDROData_NaturalObject::Update();
99   
100   TopoDS_Shape aResShape = generateTopShape();
101   SetTopShape( aResShape );
102
103   createGroupObjects();
104 }
105
106 bool HYDROData_ImmersibleZone::IsHas2dPrs() const
107 {
108   return true;
109 }
110
111 TopoDS_Shape HYDROData_ImmersibleZone::generateTopShape() const
112 {
113   return generateTopShape( GetPolyline() );
114 }
115
116 TopoDS_Shape HYDROData_ImmersibleZone::generateTopShape( const Handle(HYDROData_PolylineXY)& aPolyline )
117 {
118   TopoDS_Face aResultFace = TopoDS_Face();
119
120   if( !aPolyline.IsNull() )
121   {
122     TopoDS_Shape aPolylineShape = aPolyline->GetShape();
123     TopTools_ListOfShape aWiresList;
124
125     if ( !aPolylineShape.IsNull() && 
126          aPolylineShape.ShapeType() == TopAbs_WIRE ) {
127       const TopoDS_Wire& aPolylineWire = TopoDS::Wire( aPolylineShape );
128       if ( !aPolylineWire.IsNull() ) {
129         BRepBuilderAPI_MakeFace aMakeFace( aPolylineWire, Standard_True );
130         aMakeFace.Build();
131         if( aMakeFace.IsDone() ) {
132           aResultFace = aMakeFace.Face();
133         }
134       }
135     } else {
136       TopExp_Explorer anExp( aPolylineShape, TopAbs_WIRE );
137       for ( ; anExp.More(); anExp.Next() ) {
138         if(!anExp.Current().IsNull()) {
139           const TopoDS_Wire& aWire = TopoDS::Wire( anExp.Current() );
140           aWiresList.Append( aWire );
141         }
142       }
143       if(aWiresList.IsEmpty())
144          return aResultFace;
145
146       BRepAlgo_FaceRestrictor aFR;
147       TopoDS_Face aRefFace;      
148       TopoDS_Shape aS = aWiresList.First();
149       BRepBuilderAPI_MakeFace aMakeFace( TopoDS::Wire(aWiresList.First()), Standard_True );
150       aMakeFace.Build();
151       if( aMakeFace.IsDone() ) {
152         aRefFace = aMakeFace.Face();
153       }
154       if(aRefFace.IsNull())
155         return aResultFace;
156
157       aFR.Init(aRefFace,Standard_False, Standard_True);
158       TopTools_ListIteratorOfListOfShape anIt( aWiresList );
159       for ( ; anIt.More(); anIt.Next() ) {
160         TopoDS_Wire& aWire = TopoDS::Wire( anIt.Value() );
161         if ( aWire.IsNull() ) 
162             continue;
163         aFR.Add(aWire);
164       }
165       aFR.Perform();
166       if (aFR.IsDone()) {
167         for (; aFR.More(); aFR.Next()) {
168           aResultFace = aFR.Current();
169           break;
170         }
171       }
172     }
173   }
174
175   if( aResultFace.IsNull() )
176     return aResultFace;
177
178   BRepCheck_Analyzer anAnalyzer( aResultFace );
179   if( anAnalyzer.IsValid() && aResultFace.ShapeType()==TopAbs_FACE )
180     return aResultFace;
181   else
182     return TopoDS_Face();
183 }
184
185 void HYDROData_ImmersibleZone::createGroupObjects()
186 {
187   TopoDS_Shape aZoneShape = GetTopShape();
188   
189   // Temporary solution while the restriction for polylines is not implemented
190   // and shape for zone can be compound and not face only
191   if ( !aZoneShape.IsNull() && aZoneShape.ShapeType() != TopAbs_FACE )
192   {
193     TopExp_Explorer aZoneFaceExp( aZoneShape, TopAbs_FACE );
194     if ( aZoneFaceExp.More() )
195       aZoneShape = aZoneFaceExp.Current(); // Take only first face into account
196   }
197
198   if ( aZoneShape.IsNull() || aZoneShape.ShapeType() != TopAbs_FACE )
199     return;
200
201   TopoDS_Face aZoneFace = TopoDS::Face( aZoneShape );
202   
203   TopoDS_Wire aZoneOuterWire = ShapeAnalysis::OuterWire( aZoneFace );
204
205   // Create outer edges group
206   QString anOutWiresGroupName = GetName() + "_Outer";
207
208   Handle(HYDROData_ShapesGroup) anOutWiresGroup = createGroupObject();
209   anOutWiresGroup->SetName( anOutWiresGroupName );
210
211   TopTools_SequenceOfShape anOuterEdges;
212   HYDROData_ShapesTool::ExploreShapeToShapes( aZoneOuterWire, TopAbs_EDGE, anOuterEdges );
213   anOutWiresGroup->SetShapes( anOuterEdges );
214
215   int anInnerCounter = 1;
216   TopExp_Explorer aZoneFaceExp( aZoneFace, TopAbs_WIRE );
217   for ( ; aZoneFaceExp.More(); aZoneFaceExp.Next() )
218   {
219     TopoDS_Wire aZoneWire = TopoDS::Wire( aZoneFaceExp.Current() );
220     if ( aZoneWire.IsEqual( aZoneOuterWire ) )
221       continue; // Skip the outer wire
222
223     TopTools_SequenceOfShape anInnerEdges;
224     HYDROData_ShapesTool::ExploreShapeToShapes( aZoneWire, TopAbs_EDGE, anInnerEdges );
225     if ( anInnerEdges.IsEmpty() )
226       continue;
227
228     QString anInWiresGroupName = GetName() + "_Inner_" + QString::number( anInnerCounter++ );
229
230     Handle(HYDROData_ShapesGroup) anInWiresGroup = createGroupObject();
231     anInWiresGroup->SetName( anInWiresGroupName );
232
233     anInWiresGroup->SetShapes( anInnerEdges );
234   }  
235 }
236
237 TopoDS_Shape HYDROData_ImmersibleZone::GetShape3D() const
238 {
239   return getTopShape();
240 }
241
242 QColor HYDROData_ImmersibleZone::DefaultFillingColor()
243 {
244   return QColor( Qt::darkBlue );
245 }
246
247 QColor HYDROData_ImmersibleZone::DefaultBorderColor()
248 {
249   return QColor( Qt::transparent );
250 }
251
252 QColor HYDROData_ImmersibleZone::getDefaultFillingColor() const
253 {
254   return DefaultFillingColor();
255 }
256
257 QColor HYDROData_ImmersibleZone::getDefaultBorderColor() const
258 {
259   return DefaultBorderColor();
260 }
261
262 void HYDROData_ImmersibleZone::SetPolyline( const Handle(HYDROData_PolylineXY)& thePolyline )
263 {
264   SetReferenceObject( thePolyline, DataTag_Polyline );
265   SetToUpdate( true );
266 }
267
268 Handle(HYDROData_PolylineXY) HYDROData_ImmersibleZone::GetPolyline() const
269 {
270   return Handle(HYDROData_PolylineXY)::DownCast( 
271            GetReferenceObject( DataTag_Polyline ) );
272 }
273
274 void HYDROData_ImmersibleZone::RemovePolyline()
275 {
276   ClearReferenceObjects( DataTag_Polyline );
277   SetToUpdate( true );
278 }
279
280