]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROData/HYDROData_LandCover.cxx
Salome HOME
merge BR_v14_rc
[modules/hydro.git] / src / HYDROData / HYDROData_LandCover.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_LandCover.h"
20
21 #include "HYDROData_PolylineXY.h"
22
23 #include <BRep_Builder.hxx>
24 #include <BRepBuilderAPI_MakeFace.hxx>
25 #include <BRepBuilderAPI_MakeWire.hxx>
26 #include <TDataStd_AsciiString.hxx>
27 #include <TNaming_Builder.hxx>
28 #include <TNaming_NamedShape.hxx>
29 #include <TopoDS.hxx>
30 #include <TopoDS_Shape.hxx>
31 #include <TopoDS_Wire.hxx>
32 #include <TopoDS_Face.hxx>
33 #include <TopExp_Explorer.hxx>
34 #include <TopTools_ListOfShape.hxx>
35 #include <TopTools_ListIteratorOfListOfShape.hxx>
36 #include <NCollection_IncAllocator.hxx>
37 #include <BOPAlgo_BOP.hxx>
38 #include <ShapeAnalysis_Wire.hxx>
39 #include <Precision.hxx>
40 #include <TopTools_SequenceOfShape.hxx>
41
42 #include <QColor>
43 #include <QStringList>
44
45 IMPLEMENT_STANDARD_HANDLE( HYDROData_LandCover, HYDROData_Entity )
46 IMPLEMENT_STANDARD_RTTIEXT( HYDROData_LandCover, HYDROData_Entity )
47
48 HYDROData_LandCover::HYDROData_LandCover()
49 : HYDROData_Entity( Geom_2d )
50 {
51 }
52
53 HYDROData_LandCover::~HYDROData_LandCover()
54 {
55 }
56
57 const ObjectKind HYDROData_LandCover::GetKind() const
58 {
59   return KIND_LAND_COVER;
60 }
61
62 QStringList HYDROData_LandCover::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
63 {
64   QStringList aResList = dumpObjectCreation( theTreatedObjects );
65   QString aName = GetObjPyName();
66   
67   // Set Strickler type
68   QString aType = GetStricklerType();
69   if ( !aType.isEmpty() ) {
70     aResList << QString( "" );
71     ///< \TODO to be implemented:
72     // aResList << QString( "%1.SetStricklerType( \"%2\" );" ).arg( aName ).arg( aType );
73     aResList << QString( "" );
74   }
75
76   // Set polylines
77   ///< \TODO to be implemented: 
78   
79   aResList << QString( "" );
80   aResList << QString( "%1.Update();" ).arg( aName );
81   aResList << QString( "" );
82
83   return aResList;
84 }
85
86 HYDROData_SequenceOfObjects HYDROData_LandCover::GetAllReferenceObjects() const
87 {
88   HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
89
90   HYDROData_SequenceOfObjects aSeqOfPolylines = GetPolylines();
91   aResSeq.Append( aSeqOfPolylines );
92
93   return aResSeq;
94 }
95
96 bool HYDROData_LandCover::IsHas2dPrs() const
97 {
98   return true;
99 }
100
101 void HYDROData_LandCover::SetStricklerType( const QString& theType )
102 {
103   TCollection_AsciiString anAsciiStr( theType.toStdString().c_str() );
104   TDataStd_AsciiString::Set( myLab.FindChild( DataTag_StricklerType ), anAsciiStr );
105 }
106
107 QString HYDROData_LandCover::GetStricklerType() const
108 {
109   QString aType;
110
111   TDF_Label aLabel = myLab.FindChild( DataTag_StricklerType, false );
112   if ( !aLabel.IsNull() ) {
113     Handle(TDataStd_AsciiString) anAsciiStr;
114     if ( aLabel.FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) ) {
115       aType = QString( anAsciiStr->Get().ToCString() );
116     }
117   }
118
119   return aType;
120 }
121
122 void HYDROData_LandCover::Update()
123 {
124   HYDROData_Entity::Update();
125   
126   removeShape();
127
128   TCollection_AsciiString anErrorMsg;
129   TopoDS_Shape aResShape = buildShape( GetPolylines(), anErrorMsg );
130   
131   setShape( aResShape );
132 }
133
134 void HYDROData_LandCover::SetPolylines( const HYDROData_SequenceOfObjects& thePolylines )
135 {
136   SetReferenceObjects( thePolylines, DataTag_Polylines );
137   Changed( Geom_2d );
138 }
139
140 HYDROData_SequenceOfObjects HYDROData_LandCover::GetPolylines() const
141 {
142   return GetReferenceObjects( DataTag_Polylines );
143 }
144
145 TopoDS_Shape HYDROData_LandCover::GetShape() const
146 {
147   TopoDS_Shape aShape;
148
149   TDF_Label aLabel = myLab.FindChild( DataTag_Shape, false );
150   if ( !aLabel.IsNull() )
151   {
152     Handle(TNaming_NamedShape) aNamedShape;
153     if( aLabel.FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) ) {
154       aShape = aNamedShape->Get();
155     }
156   }
157
158   return aShape;
159 }
160
161 void HYDROData_LandCover::SetFillingColor( const QColor& theColor )
162 {
163   SetColor( theColor, DataTag_FillingColor );
164 }
165
166 QColor HYDROData_LandCover::GetFillingColor() const
167 {
168   return GetColor( DefaultFillingColor(), DataTag_FillingColor );
169 }
170
171 void HYDROData_LandCover::SetBorderColor( const QColor& theColor )
172 {
173   SetColor( theColor, DataTag_BorderColor );
174 }
175
176 QColor HYDROData_LandCover::GetBorderColor() const
177 {
178   return GetColor( DefaultBorderColor(), DataTag_BorderColor );
179 }
180
181 QColor HYDROData_LandCover::DefaultFillingColor()
182 {
183   return QColor( Qt::magenta );
184 }
185
186 QColor HYDROData_LandCover::DefaultBorderColor()
187 {
188   return QColor( Qt::transparent );
189 }
190
191 void HYDROData_LandCover::setShape( const TopoDS_Shape& theShape )
192 {
193   TNaming_Builder aBuilder( myLab.FindChild( DataTag_Shape ) );
194   aBuilder.Generated( theShape );
195 }
196
197 void HYDROData_LandCover::removeShape()
198 {
199   TDF_Label aLabel = myLab.FindChild( DataTag_Shape, false );
200   if ( !aLabel.IsNull() ) {
201     aLabel.ForgetAllAttributes();
202   }
203 }
204
205 TopoDS_Shape HYDROData_LandCover::buildShape( const HYDROData_SequenceOfObjects& thePolylines,
206                                               TCollection_AsciiString& theErrorMsg )
207 {
208   theErrorMsg.Clear();
209   TopoDS_Shape aResShape;
210
211   BRepBuilderAPI_MakeWire aMakeWire;
212   
213   TopTools_ListOfShape aClosedWires;
214
215   int aNbPolylines = thePolylines.Length();
216   for ( int i = 1; i <= aNbPolylines; ++i ) {
217     Handle(HYDROData_PolylineXY) aPolyline = 
218       Handle(HYDROData_PolylineXY)::DownCast( thePolylines.Value( i ) );
219     
220     if ( aPolyline.IsNull() ) {
221       continue;
222     }
223
224     TopoDS_Shape aPolyShape = aPolyline->GetShape();
225     if ( aPolyShape.IsNull() ) {
226       continue;
227     }
228
229     // Extract polyline wire(s)
230     TopTools_ListOfShape aPolylineWires;
231       
232     if ( aPolyShape.ShapeType() == TopAbs_WIRE ) {
233       const TopoDS_Wire& aPolylineWire = TopoDS::Wire( aPolyShape );
234       if ( !aPolylineWire.IsNull() ) {
235         aPolylineWires.Append( aPolylineWire );
236       }
237     } else if ( aPolyShape.ShapeType() == TopAbs_COMPOUND ) {
238       TopExp_Explorer anExp( aPolyShape, TopAbs_WIRE );
239       for (; anExp.More(); anExp.Next() ) {
240         if(!anExp.Current().IsNull()) {
241           const TopoDS_Wire& aWire = TopoDS::Wire( anExp.Current() );
242           aPolylineWires.Append( aWire );
243         }
244       }
245     }
246     
247     TopTools_ListIteratorOfListOfShape anIt( aPolylineWires );
248     for ( ; anIt.More(); anIt.Next() ) {
249       TopoDS_Wire& aWire = TopoDS::Wire( anIt.Value() );
250       
251       if ( aWire.Closed() ) {
252         aClosedWires.Append( aWire );
253       } else {
254         aMakeWire.Add( aWire );
255         aMakeWire.Build();
256         if ( aMakeWire.IsDone() ) {
257           if ( aMakeWire.Wire().Closed() ) {
258             aClosedWires.Append( aMakeWire.Wire() );
259             aMakeWire = BRepBuilderAPI_MakeWire();
260           }
261         }
262       }
263     }
264   }
265
266   if ( aClosedWires.Extent() == 1 ) {
267     // make face
268     TopoDS_Wire aW = TopoDS::Wire( aClosedWires.First());
269     BRepBuilderAPI_MakeFace aMakeFace( aW );
270     aMakeFace.Build();
271     if( aMakeFace.IsDone() ) 
272     {
273       Handle(ShapeAnalysis_Wire) aSAW = new ShapeAnalysis_Wire(aW, aMakeFace.Face(), Precision::Confusion());
274       if (!aSAW->CheckSelfIntersection())
275         aResShape = aMakeFace.Face();
276       else
277         theErrorMsg = "Can't create landcover on the given polyline\nSelf-intersection of wire have been detected";
278     }
279   } else if ( aClosedWires.Extent() > 1 ) {
280     // make compound
281     BRep_Builder aBuilder;
282     TopoDS_Compound aCompound;
283     aBuilder.MakeCompound( aCompound );
284     TopTools_SequenceOfShape aSeq;
285     TopTools_ListIteratorOfListOfShape aWiresIter( aClosedWires );
286     bool anErrStat = false;
287     for ( ; aWiresIter.More() && !anErrStat; aWiresIter.Next() )
288     {
289       TopoDS_Wire aW = TopoDS::Wire( aWiresIter.Value() );
290       BRepBuilderAPI_MakeFace aMakeFace( aW );
291       aMakeFace.Build();
292       if( aMakeFace.IsDone() ) 
293       {
294         Handle(ShapeAnalysis_Wire) aSAW = new ShapeAnalysis_Wire(aW, aMakeFace.Face(), Precision::Confusion());
295         if (!aSAW->CheckSelfIntersection())
296           aSeq.Append( aMakeFace.Face() );
297         else
298         {
299           anErrStat = true;
300           theErrorMsg = "Can't create landcover on the given polyline\nSelf-intersection of wire(s) have been detected";
301         }
302       }
303     }
304     if (!anErrStat)
305     {
306       for (int i = 1; i <= aSeq.Length(); i++)
307         aBuilder.Add( aCompound, aSeq(i) );
308       aResShape = aCompound;
309     }
310     else
311       aResShape = TopoDS_Shape();
312   } else if ( aNbPolylines > 0 ) {
313     TCollection_AsciiString aSourceName = aNbPolylines > 1 ? "polylines" : "polyline";
314     theErrorMsg = "Can't build closed contour on the given ";
315     theErrorMsg += aSourceName;
316   }
317
318   ///< \TODO to be reimplemented
319   /*
320   TopoDS_Shape anArgShape;
321   TopTools_ListOfShape aToolShapes;
322   
323   HYDROData_SequenceOfObjects aRefPolylines = GetPolylines();
324   for ( int i = 1, n = aRefPolylines.Length(); i <= n; ++i ) {
325     Handle(HYDROData_PolylineXY) aPolyline = 
326       Handle(HYDROData_PolylineXY)::DownCast( aRefPolylines.Value( i ) );
327     
328     if ( aPolyline.IsNull() ) {
329       continue;
330     }
331
332     if ( !aPolyline->IsClosed() ) {
333       continue;
334     }
335
336     TopoDS_Shape aPolyShape = aPolyline->GetShape();
337     if ( aPolyShape.IsNull() || aPolyShape.ShapeType() != TopAbs_WIRE ) {
338       continue;
339     }
340
341     const TopoDS_Wire& aPolylineWire = TopoDS::Wire( aPolyShape );
342     if ( aPolylineWire.IsNull() ) {
343       continue;
344     }
345
346     TopoDS_Face aResultFace = TopoDS_Face();
347     BRepBuilderAPI_MakeFace aMakeFace( aPolylineWire, Standard_True );
348     aMakeFace.Build();
349     if( aMakeFace.IsDone() ) {
350       aResultFace = aMakeFace.Face();
351     }
352
353     if( aResultFace.IsNull() ) {
354       continue;
355     }
356
357     if ( anArgShape.IsNull() ) {
358       anArgShape = aResultFace;
359     } else {
360       aToolShapes.Append( aResultFace );
361     }
362   }
363
364   aResShape = anArgShape;
365
366   if ( !anArgShape.IsNull() && aToolShapes.Extent() > 0 ) {
367     Handle(NCollection_BaseAllocator)aAL=new NCollection_IncAllocator;
368     BOPAlgo_BOP aBOP(aAL);
369  
370     aBOP.AddArgument( anArgShape );
371
372     TopTools_ListIteratorOfListOfShape anIt(aToolShapes);
373     for( ; anIt.More(); anIt.Next() ) {
374       aBOP.AddTool( anIt.Value() );
375     }
376
377     aBOP.SetOperation( BOPAlgo_CUT );
378     aBOP.Perform();
379
380     if ( !aBOP.Shape().IsNull() ) {
381       aResShape = aBOP.Shape();
382     }
383   }
384   */
385   
386   return aResShape;
387 }