src/HYDROData/HYDROData_ImmersibleZone.cxx

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