Salome HOME
Dump to python corrected.
[modules/hydro.git] / src / HYDROData / HYDROData_Obstacle.cxx
1
2 #include "HYDROData_Obstacle.h"
3
4 #include "HYDROData_Document.h"
5 #include "HYDROData_ShapesGroup.h"
6 #include "HYDROData_ShapesTool.h"
7
8 #include <Basics_Utils.hxx>
9
10 #include <BRepTools.hxx>
11 #include <BRep_Builder.hxx>
12
13 #include <IGESControl_Reader.hxx>
14 #include <IGESData_IGESModel.hxx>
15
16 #include <STEPControl_Reader.hxx>
17
18 #include <Interface_Static.hxx>
19
20 #include <TopoDS.hxx>
21 #include <TopoDS_Iterator.hxx>
22 #include <TopoDS_Shape.hxx>
23 #include <TopoDS_Edge.hxx>
24
25 #include <TDataStd_AsciiString.hxx>
26
27 #include <TColStd_SequenceOfAsciiString.hxx>
28
29 #include <TopExp_Explorer.hxx>
30
31 #include <QColor>
32 #include <QFile>
33 #include <QFileInfo>
34 #include <QStringList>
35
36 #include <Standard_ErrorHandler.hxx> // CAREFUL ! position of this file is critic
37
38 #include <HYDROData_Projection.h>
39
40 IMPLEMENT_STANDARD_HANDLE(HYDROData_Obstacle,HYDROData_ArtificialObject)
41 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Obstacle,HYDROData_ArtificialObject)
42
43
44 HYDROData_Obstacle::HYDROData_Obstacle()
45 : HYDROData_ArtificialObject()
46 {
47 }
48
49 HYDROData_Obstacle::~HYDROData_Obstacle()
50 {
51 }
52
53 QStringList HYDROData_Obstacle::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
54 {
55   QStringList aResList = dumpObjectCreation( theTreatedObjects );
56   QString aName = GetObjPyName();
57
58   // TODO
59
60   aResList << QString( "" );
61   aResList << QString( "%1.Update();" ).arg( aName );
62   aResList << QString( "" );
63
64   return aResList;
65 }
66
67 void HYDROData_Obstacle::Update()
68 {
69   removeGroupObjects();
70   createGroupObjects();
71   checkAndSetAltitudeObject();
72
73   HYDROData_Entity::Update();
74 }
75
76 TopoDS_Shape HYDROData_Obstacle::GetTopShape() const
77 {
78   return getTopShape();
79 }
80
81 TopoDS_Shape HYDROData_Obstacle::GetShape3D() const
82 {
83   return getShape3D();
84 }
85
86 void HYDROData_Obstacle::SetShape3D( const TopoDS_Shape& theShape )
87 {
88   TopoDS_Face aShape2d = HYDROData_Projection::MakeProjection( theShape );
89   HYDROData_ArtificialObject::SetShape3D( theShape );
90   HYDROData_ArtificialObject::SetTopShape( aShape2d );
91 }
92
93 QColor HYDROData_Obstacle::DefaultFillingColor()
94 {
95   return QColor( Qt::yellow );
96 }
97
98 QColor HYDROData_Obstacle::DefaultBorderColor()
99 {
100   return QColor( Qt::transparent );
101 }
102
103 bool HYDROData_Obstacle::ImportFromFile( const QString& theFilePath )
104 {
105   // Check the file existence
106   QFileInfo aFileInfo( theFilePath );
107   if ( !aFileInfo.exists() ) {
108     return false;
109   }
110
111   bool aRes = false;
112   TopoDS_Shape aShape;
113
114   // Import file
115   QString aFileSuf = aFileInfo.suffix().toLower();
116   if ( aFileSuf == "brep" ) {
117     aShape = ImportBREP( theFilePath );
118   } else if ( aFileSuf == "iges" || aFileSuf == "igs" ) {
119     aShape = ImportIGES( theFilePath );
120   } else if ( aFileSuf == "step" ) {
121     aShape = ImportSTEP( theFilePath );
122   }
123  
124   // Check the result shape
125   aRes = !aShape.IsNull();
126
127   // Set shape to the obstacle in case of success
128   if ( aRes ) {
129     SetShape3D( aShape );
130     SetFilePath( theFilePath );
131   }
132
133   return aRes;
134 }
135
136 void HYDROData_Obstacle::SetFilePath( const QString& theFilePath )
137 {
138   TCollection_AsciiString anAsciiStr( theFilePath.toStdString().c_str() );
139   TDataStd_AsciiString::Set( myLab.FindChild( DataTag_FilePath ), anAsciiStr );
140 }
141
142 QString HYDROData_Obstacle::GetFilePath() const
143 {
144   QString aRes;
145
146   Handle(TDataStd_AsciiString) anAsciiStr;
147   if ( myLab.FindChild( DataTag_FilePath ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
148     aRes = QString( anAsciiStr->Get().ToCString() );
149
150   return aRes;
151 }
152
153 void HYDROData_Obstacle::SetGeomObjectEntry( const QString& theEntry )
154 {
155   TCollection_AsciiString anAsciiStr( theEntry.toStdString().c_str() );
156   TDataStd_AsciiString::Set( myLab.FindChild( DataTag_GeomObjectEntry ), anAsciiStr );
157 }
158
159 QString HYDROData_Obstacle::GetGeomObjectEntry() const
160 {
161   QString aRes;
162
163   Handle(TDataStd_AsciiString) anAsciiStr;
164   if ( myLab.FindChild( DataTag_GeomObjectEntry ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
165     aRes = QString( anAsciiStr->Get().ToCString() );
166
167   return aRes;
168 }
169
170 TopoDS_Shape HYDROData_Obstacle::ImportBREP( const QString& theFilePath ) const
171 {
172   TopoDS_Shape aResShape;
173
174   BRep_Builder aBrepBuilder;
175   BRepTools::Read( aResShape, qPrintable(theFilePath), aBrepBuilder );
176
177   return aResShape;
178 }
179
180 TopoDS_Shape HYDROData_Obstacle::ImportIGES( const QString& theFilePath ) const
181 {
182   TopoDS_Shape aResShape;
183
184   // Set "C" numeric locale to save numbers correctly
185   Kernel_Utils::Localizer loc;
186
187   IGESControl_Reader aReader;
188
189   Interface_Static::SetCVal("xstep.cascade.unit","M");
190
191   try {
192     OCC_CATCH_SIGNALS;
193
194     IFSelect_ReturnStatus status = aReader.ReadFile(qPrintable(theFilePath));
195
196     if (status == IFSelect_RetDone) {
197       // Rescale units
198       Handle(IGESData_IGESModel) aModel =
199         Handle(IGESData_IGESModel)::DownCast(aReader.Model());
200       if (!aModel.IsNull()) {
201         IGESData_GlobalSection aGS = aModel->GlobalSection();
202         aGS.SetUnitFlag(6);
203         aModel->SetGlobalSection(aGS);
204       }
205     
206       aReader.ClearShapes();
207       aReader.TransferRoots();
208
209       aResShape = aReader.OneShape();
210     } 
211     else {
212       aResShape.Nullify();
213     }
214   }
215   catch(Standard_Failure) {
216     aResShape.Nullify();
217   }
218
219   return aResShape;
220 }
221
222 TopoDS_Shape HYDROData_Obstacle::ImportSTEP( const QString& theFilePath ) const
223 {
224   TopoDS_Shape aResShape;
225
226   // Set "C" numeric locale to save numbers correctly
227   Kernel_Utils::Localizer loc;
228
229   STEPControl_Reader aReader;
230
231   // Convert to METERS
232   Interface_Static::SetCVal("xstep.cascade.unit","M");
233   Interface_Static::SetIVal("read.step.ideas", 1);
234   Interface_Static::SetIVal("read.step.nonmanifold", 1);
235
236   BRep_Builder B;
237   TopoDS_Compound compound;
238   B.MakeCompound(compound);
239
240   try {
241     OCC_CATCH_SIGNALS;
242
243     IFSelect_ReturnStatus status = aReader.ReadFile( qPrintable(theFilePath) );
244
245     if (status == IFSelect_RetDone) {
246       // Rescale units
247       // set UnitFlag to units from file
248       TColStd_SequenceOfAsciiString anUnitLengthNames;
249       TColStd_SequenceOfAsciiString anUnitAngleNames;
250       TColStd_SequenceOfAsciiString anUnitSolidAngleNames;
251       aReader.FileUnits(anUnitLengthNames, anUnitAngleNames, anUnitSolidAngleNames);
252       if (anUnitLengthNames.Length() > 0) {
253         TCollection_AsciiString aLenUnits = anUnitLengthNames.First();
254         if (aLenUnits == "millimetre")
255           Interface_Static::SetCVal("xstep.cascade.unit", "MM");
256         else if (aLenUnits == "centimetre")
257           Interface_Static::SetCVal("xstep.cascade.unit", "CM");
258         else if (aLenUnits == "metre" || aLenUnits.IsEmpty())
259           Interface_Static::SetCVal("xstep.cascade.unit", "M");
260         else if (aLenUnits == "INCH")
261           Interface_Static::SetCVal("xstep.cascade.unit", "INCH");
262         else {
263           // The file contains not supported units
264           return aResShape;
265         }
266       }
267         
268       Standard_Boolean failsonly = Standard_False;
269       aReader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity);
270
271       // Root transfers
272       Standard_Integer nbr = aReader.NbRootsForTransfer();
273       aReader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity);
274
275       for (Standard_Integer n = 1; n <= nbr; n++) {
276         Standard_Boolean ok = aReader.TransferRoot(n);
277         // Collecting resulting entities
278         Standard_Integer nbs = aReader.NbShapes();
279         if (!ok || nbs == 0) {
280           continue; // skip empty root
281         } 
282         else if (nbr == 1 && nbs == 1) { // For a single entity
283           aResShape = aReader.Shape(1);
284           // ATTENTION: this is a workaround for mantis issue 0020442 remark 0010776
285           // It should be removed after patching OCCT for bug OCC22436
286           // (fix for OCCT is expected in service pack next to OCCT6.3sp12)
287           if (aResShape.ShapeType() == TopAbs_COMPOUND) {
288             int nbSub1 = 0;
289             TopoDS_Shape currShape;
290             TopoDS_Iterator It (aResShape, Standard_True, Standard_True);
291             for (; It.More(); It.Next()) {
292               nbSub1++;
293               currShape = It.Value();
294             }
295             if (nbSub1 == 1)
296               aResShape = currShape;
297           }
298           // END workaround
299           break;
300         }
301
302         for (Standard_Integer i = 1; i <= nbs; i++) {
303           TopoDS_Shape aShape = aReader.Shape(i);
304           if (aShape.IsNull()) {
305             continue;
306           }
307           else {
308             B.Add(compound, aShape);
309           }
310         }
311       }
312   
313       if (aResShape.IsNull())
314         aResShape = compound;
315
316       // Check if any BRep entity has been read, there must be at least a vertex
317       if ( !TopExp_Explorer( aResShape, TopAbs_VERTEX ).More() ) {
318         // No geometrical data in the imported file
319         return TopoDS_Shape();
320       }
321     }
322     else {
323       aResShape.Nullify();
324     }
325   }
326   catch (Standard_Failure) {
327     aResShape.Nullify();
328   }
329
330   return aResShape;
331 }
332
333 QColor HYDROData_Obstacle::getDefaultFillingColor() const
334 {
335   return DefaultFillingColor();
336 }
337
338 QColor HYDROData_Obstacle::getDefaultBorderColor() const
339 {
340   return DefaultBorderColor();
341 }
342
343 void HYDROData_Obstacle::createGroupObjects()
344 {
345   TopoDS_Shape anObstacleShape = GetTopShape();
346   if ( !anObstacleShape.IsNull() )
347   {
348     TopTools_SequenceOfShape aWireEdges;
349     HYDROData_ShapesTool::ExploreShapeToShapes( anObstacleShape, TopAbs_EDGE, aWireEdges );
350     if ( !aWireEdges.IsEmpty() )
351     {
352       QString aWireGroupName = GetName() + "_Outer_Wire";
353
354       Handle(HYDROData_ShapesGroup) anExtWireGroup = createGroupObject();
355       anExtWireGroup->SetName( aWireGroupName );
356      
357       anExtWireGroup->SetShapes( aWireEdges );
358     }
359   }
360 }
361
362 ObjectKind HYDROData_Obstacle::getAltitudeObjectType() const
363 {
364   return KIND_OBSTACLE_ALTITUDE;
365 }
366
367
368