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