2 #include "HYDROData_Obstacle.h"
4 #include "HYDROData_Document.h"
5 #include "HYDROData_ShapesGroup.h"
7 #include <Basics_Utils.hxx>
9 #include <BRepTools.hxx>
10 #include <BRep_Builder.hxx>
12 #include <IGESControl_Reader.hxx>
13 #include <IGESData_IGESModel.hxx>
15 #include <STEPControl_Reader.hxx>
17 #include <Interface_Static.hxx>
20 #include <TopoDS_Iterator.hxx>
21 #include <TopoDS_Shape.hxx>
22 #include <TopoDS_Edge.hxx>
24 #include <TDataStd_AsciiString.hxx>
26 #include <TColStd_SequenceOfAsciiString.hxx>
28 #include <TopExp_Explorer.hxx>
33 #include <QStringList>
35 #include <Standard_ErrorHandler.hxx> // CAREFUL ! position of this file is critic
37 #include <HYDROData_Projection.h>
39 #define PYTHON_OBSTACLE_ID "KIND_OBSTACLE"
41 IMPLEMENT_STANDARD_HANDLE(HYDROData_Obstacle,HYDROData_ArtificialObject)
42 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Obstacle,HYDROData_ArtificialObject)
45 HYDROData_Obstacle::HYDROData_Obstacle()
46 : HYDROData_ArtificialObject()
50 HYDROData_Obstacle::~HYDROData_Obstacle()
54 QStringList HYDROData_Obstacle::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
58 Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
59 if ( aDocument.IsNull() )
62 QString aDocName = aDocument->GetDocPyName();
63 QString anObstacleName = GetName();
65 aResList << QString( "%1 = %2.CreateObject( %3 );" )
66 .arg( anObstacleName ).arg( aDocName ).arg( PYTHON_OBSTACLE_ID );
67 aResList << QString( "%1.SetName( \"%2\" );" )
68 .arg( anObstacleName ).arg( anObstacleName );
69 aResList << QString( "" );
76 void HYDROData_Obstacle::Update()
78 HYDROData_Entity::Update();
84 TopoDS_Shape HYDROData_Obstacle::GetTopShape() const
89 TopoDS_Shape HYDROData_Obstacle::GetShape3D() const
94 void HYDROData_Obstacle::SetShape3D( const TopoDS_Shape& theShape )
96 TopoDS_Face aShape2d = HYDROData_Projection::MakeProjection( theShape );
97 HYDROData_ArtificialObject::SetShape3D( theShape );
98 HYDROData_ArtificialObject::SetTopShape( aShape2d );
101 QColor HYDROData_Obstacle::DefaultFillingColor()
103 return QColor( Qt::yellow );
106 QColor HYDROData_Obstacle::DefaultBorderColor()
108 return QColor( Qt::transparent );
111 bool HYDROData_Obstacle::ImportFromFile( const QString& theFilePath )
113 // Check the file existence
114 QFileInfo aFileInfo( theFilePath );
115 if ( !aFileInfo.exists() ) {
123 QString aFileSuf = aFileInfo.suffix().toLower();
124 if ( aFileSuf == "brep" ) {
125 aShape = ImportBREP( theFilePath );
126 } else if ( aFileSuf == "iges" || aFileSuf == "igs" ) {
127 aShape = ImportIGES( theFilePath );
128 } else if ( aFileSuf == "step" ) {
129 aShape = ImportSTEP( theFilePath );
132 // Check the result shape
133 aRes = !aShape.IsNull();
135 // Set shape to the obstacle in case of success
137 SetShape3D( aShape );
138 SetFilePath( theFilePath );
144 void HYDROData_Obstacle::SetFilePath( const QString& theFilePath )
146 TCollection_AsciiString anAsciiStr( theFilePath.toStdString().c_str() );
147 TDataStd_AsciiString::Set( myLab.FindChild( DataTag_FilePath ), anAsciiStr );
150 QString HYDROData_Obstacle::GetFilePath() const
154 Handle(TDataStd_AsciiString) anAsciiStr;
155 if ( myLab.FindChild( DataTag_FilePath ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
156 aRes = QString( anAsciiStr->Get().ToCString() );
161 void HYDROData_Obstacle::SetGeomObjectEntry( const QString& theEntry )
163 TCollection_AsciiString anAsciiStr( theEntry.toStdString().c_str() );
164 TDataStd_AsciiString::Set( myLab.FindChild( DataTag_GeomObjectEntry ), anAsciiStr );
167 QString HYDROData_Obstacle::GetGeomObjectEntry() const
171 Handle(TDataStd_AsciiString) anAsciiStr;
172 if ( myLab.FindChild( DataTag_GeomObjectEntry ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
173 aRes = QString( anAsciiStr->Get().ToCString() );
178 TopoDS_Shape HYDROData_Obstacle::ImportBREP( const QString& theFilePath ) const
180 TopoDS_Shape aResShape;
182 BRep_Builder aBrepBuilder;
183 BRepTools::Read( aResShape, qPrintable(theFilePath), aBrepBuilder );
188 TopoDS_Shape HYDROData_Obstacle::ImportIGES( const QString& theFilePath ) const
190 TopoDS_Shape aResShape;
192 // Set "C" numeric locale to save numbers correctly
193 Kernel_Utils::Localizer loc;
195 IGESControl_Reader aReader;
197 Interface_Static::SetCVal("xstep.cascade.unit","M");
202 IFSelect_ReturnStatus status = aReader.ReadFile(qPrintable(theFilePath));
204 if (status == IFSelect_RetDone) {
206 Handle(IGESData_IGESModel) aModel =
207 Handle(IGESData_IGESModel)::DownCast(aReader.Model());
208 if (!aModel.IsNull()) {
209 IGESData_GlobalSection aGS = aModel->GlobalSection();
211 aModel->SetGlobalSection(aGS);
214 aReader.ClearShapes();
215 aReader.TransferRoots();
217 aResShape = aReader.OneShape();
223 catch(Standard_Failure) {
230 TopoDS_Shape HYDROData_Obstacle::ImportSTEP( const QString& theFilePath ) const
232 TopoDS_Shape aResShape;
234 // Set "C" numeric locale to save numbers correctly
235 Kernel_Utils::Localizer loc;
237 STEPControl_Reader aReader;
240 Interface_Static::SetCVal("xstep.cascade.unit","M");
241 Interface_Static::SetIVal("read.step.ideas", 1);
242 Interface_Static::SetIVal("read.step.nonmanifold", 1);
245 TopoDS_Compound compound;
246 B.MakeCompound(compound);
251 IFSelect_ReturnStatus status = aReader.ReadFile( qPrintable(theFilePath) );
253 if (status == IFSelect_RetDone) {
255 // set UnitFlag to units from file
256 TColStd_SequenceOfAsciiString anUnitLengthNames;
257 TColStd_SequenceOfAsciiString anUnitAngleNames;
258 TColStd_SequenceOfAsciiString anUnitSolidAngleNames;
259 aReader.FileUnits(anUnitLengthNames, anUnitAngleNames, anUnitSolidAngleNames);
260 if (anUnitLengthNames.Length() > 0) {
261 TCollection_AsciiString aLenUnits = anUnitLengthNames.First();
262 if (aLenUnits == "millimetre")
263 Interface_Static::SetCVal("xstep.cascade.unit", "MM");
264 else if (aLenUnits == "centimetre")
265 Interface_Static::SetCVal("xstep.cascade.unit", "CM");
266 else if (aLenUnits == "metre" || aLenUnits.IsEmpty())
267 Interface_Static::SetCVal("xstep.cascade.unit", "M");
268 else if (aLenUnits == "INCH")
269 Interface_Static::SetCVal("xstep.cascade.unit", "INCH");
271 // The file contains not supported units
276 Standard_Boolean failsonly = Standard_False;
277 aReader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity);
280 Standard_Integer nbr = aReader.NbRootsForTransfer();
281 aReader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity);
283 for (Standard_Integer n = 1; n <= nbr; n++) {
284 Standard_Boolean ok = aReader.TransferRoot(n);
285 // Collecting resulting entities
286 Standard_Integer nbs = aReader.NbShapes();
287 if (!ok || nbs == 0) {
288 continue; // skip empty root
290 else if (nbr == 1 && nbs == 1) { // For a single entity
291 aResShape = aReader.Shape(1);
292 // ATTENTION: this is a workaround for mantis issue 0020442 remark 0010776
293 // It should be removed after patching OCCT for bug OCC22436
294 // (fix for OCCT is expected in service pack next to OCCT6.3sp12)
295 if (aResShape.ShapeType() == TopAbs_COMPOUND) {
297 TopoDS_Shape currShape;
298 TopoDS_Iterator It (aResShape, Standard_True, Standard_True);
299 for (; It.More(); It.Next()) {
301 currShape = It.Value();
304 aResShape = currShape;
310 for (Standard_Integer i = 1; i <= nbs; i++) {
311 TopoDS_Shape aShape = aReader.Shape(i);
312 if (aShape.IsNull()) {
316 B.Add(compound, aShape);
321 if (aResShape.IsNull())
322 aResShape = compound;
324 // Check if any BRep entity has been read, there must be at least a vertex
325 if ( !TopExp_Explorer( aResShape, TopAbs_VERTEX ).More() ) {
326 // No geometrical data in the imported file
327 return TopoDS_Shape();
334 catch (Standard_Failure) {
341 QColor HYDROData_Obstacle::getDefaultFillingColor() const
343 return DefaultFillingColor();
346 QColor HYDROData_Obstacle::getDefaultBorderColor() const
348 return DefaultBorderColor();
351 void HYDROData_Obstacle::createGroupObjects()
353 TopoDS_Shape anObstacleShape = GetTopShape();
354 if ( !anObstacleShape.IsNull() )
356 TopTools_SequenceOfShape aWireEdges;
358 TopExp_Explorer anExp( anObstacleShape, TopAbs_EDGE );
359 for ( ; anExp.More(); anExp.Next() )
361 TopoDS_Edge anEdge = TopoDS::Edge( anExp.Current() );
362 if ( anEdge.IsNull() )
365 aWireEdges.Append( anEdge );
368 if ( !aWireEdges.IsEmpty() )
370 QString aWireGroupName = GetName() + "_External_Wire";
372 Handle(HYDROData_ShapesGroup) anExtWireGroup = createGroupObject();
373 anExtWireGroup->SetName( aWireGroupName );
375 anExtWireGroup->SetShapes( aWireEdges );