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