src/HYDROData/HYDROData_Channel.cxx

   1
   2 #include "HYDROData_Channel.h"
   3
   4 #include "HYDROData_Document.h"
   5 #include "HYDROData_Polyline3D.h"
   6 #include "HYDROData_Profile.h"
   7 #include "HYDROData_PolylineXY.h"
   8 #include "HYDROData_Projection.h"
   9 #include "HYDROData_ShapesGroup.h"
  10 #include "HYDROData_ShapesTool.h"
  11 #include "HYDROData_Pipes.h"
  12
  13 #include <BRepOffsetAPI_MakePipeShell.hxx>
  14 #include <BRepOffsetAPI_MakePipe.hxx>
  15 #include <BRepCheck_Analyzer.hxx>
  16 #include <BRepTools.hxx>
  17
  18 #include <TopExp.hxx>
  19
  20 #include <TopoDS.hxx>
  21 #include <TopoDS_Wire.hxx>
  22 #include <TopoDS_Vertex.hxx>
  23 #include <TopoDS_Face.hxx>
  24
  25 //#define DEB_CHANNEL 1
  26 #ifdef DEB_CHANNEL
  27 #include <BRepTools.hxx>
  28 #endif
  29
  30 #include <QColor>
  31 #include <QStringList>
  32
  33 #define PYTHON_CHANNEL_ID "KIND_CHANNEL"
  34
  35 IMPLEMENT_STANDARD_HANDLE(HYDROData_Channel,HYDROData_ArtificialObject)
  36 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Channel,HYDROData_ArtificialObject)
  37
  38
  39 HYDROData_Channel::HYDROData_Channel()
  40 : HYDROData_ArtificialObject()
  41 {
  42 }
  43
  44 HYDROData_Channel::~HYDROData_Channel()
  45 {
  46 }
  47
  48 QStringList HYDROData_Channel::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
  49 {
  50   QStringList aResList;
  51
  52   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
  53   if ( aDocument.IsNull() )
  54     return aResList;
  55
  56   QString aDocName = aDocument->GetDocPyName();
  57   QString aChannelName = GetName();
  58
  59   aResList << QString( "%1 = %2.CreateObject( %3 );" )
  60               .arg( aChannelName ).arg( aDocName ).arg( PYTHON_CHANNEL_ID );
  61   aResList << QString( "%1.SetName( \"%2\" );" )
  62               .arg( aChannelName ).arg( aChannelName );
  63   aResList << QString( "" );
  64
  65   // TODO
  66
  67   return aResList;
  68 }
  69
  70 HYDROData_SequenceOfObjects HYDROData_Channel::GetAllReferenceObjects() const
  71 {
  72   HYDROData_SequenceOfObjects aResSeq = HYDROData_ArtificialObject::GetAllReferenceObjects();
  73
  74   Handle(HYDROData_Polyline3D) aGuideLine = GetGuideLine();
  75   if ( !aGuideLine.IsNull() )
  76     aResSeq.Append( aGuideLine );
  77
  78   Handle(HYDROData_Profile) aProfile = GetProfile();
  79   if ( !aProfile.IsNull() )
  80     aResSeq.Append( aProfile );
  81
  82   return aResSeq;
  83 }
  84
  85 TopoDS_Shape HYDROData_Channel::GetTopShape() const
  86 {
  87   return getTopShape();
  88 }
  89
  90 TopoDS_Shape HYDROData_Channel::GetShape3D() const
  91 {
  92   return getShape3D();
  93 }
  94
  95 void HYDROData_Channel::Update()
  96 {
  97   HYDROData_ArtificialObject::Update();
  98
  99   Handle(HYDROData_Polyline3D) aGuideLine = GetGuideLine();
 100   Handle(HYDROData_Profile) aProfile = GetProfile();
 101   if ( aGuideLine.IsNull() || aProfile.IsNull() )
 102     return;
 103
 104   // build 3d shape
 105   TopoDS_Wire aPathWire = TopoDS::Wire(aGuideLine->GetShape3D());
 106   if(aPathWire.IsNull())
 107     return;
 108   TopoDS_Wire aProfileWire = TopoDS::Wire( aProfile->GetShape3D() );
 109   if(aProfileWire.IsNull())
 110     return;
 111
 112   //BRepTools::Write( aPathWire, "guideline.brep" );
 113   //BRepTools::Write( aProfileWire, "profile.brep" );
 114
 115   HYDROData_Canal3dAnd2d aChannelConstructor( aProfileWire, aPathWire );
 116   aChannelConstructor.Create3dPresentation();
 117   aChannelConstructor.Create2dPresentation();
 118   SetShape3D( aChannelConstructor.Get3dPresentation() );
 119   SetTopShape( aChannelConstructor.Get2dPresentation() );
 120
 121   //BRepTools::Write( aPathWire, "guideline.brep" );
 122   //BRepTools::Write( aProfileWire, "profile.brep" );
 123   //BRepTools::Write( aChannelConstructor.Get2dPresentation(), "channel2d.brep" );
 124   //BRepTools::Write( aChannelConstructor.Get3dPresentation(), "channel3d.brep" );
 125
 126   // Create groups for channel
 127   TopoDS_Wire aLeftBank = aChannelConstructor.GetLeftBank();
 128   TopoDS_Wire aRightBank = aChannelConstructor.GetRightBank();
 129   TopoDS_Wire anInlet = aChannelConstructor.GetInlet();
 130   TopoDS_Wire anOutlet = aChannelConstructor.GetOutlet();
 131
 132   TopTools_SequenceOfShape aLeftBankEdges;
 133   HYDROData_ShapesTool::ExploreShapeToShapes( aLeftBank, TopAbs_EDGE, aLeftBankEdges );
 134
 135   TopTools_SequenceOfShape aRightBankEdges;
 136   HYDROData_ShapesTool::ExploreShapeToShapes( aRightBank, TopAbs_EDGE, aRightBankEdges );
 137
 138   TopTools_SequenceOfShape anInletEdges;
 139   HYDROData_ShapesTool::ExploreShapeToShapes( anInlet, TopAbs_EDGE, anInletEdges );
 140
 141   TopTools_SequenceOfShape anOutletEdges;
 142   HYDROData_ShapesTool::ExploreShapeToShapes( anOutlet, TopAbs_EDGE, anOutletEdges );
 143
 144   QString aLeftGroupName = GetName() + "_Left_Bank";
 145
 146   Handle(HYDROData_ShapesGroup) aLeftGroup = createGroupObject();
 147   aLeftGroup->SetName( aLeftGroupName );
 148   aLeftGroup->SetShapes( aLeftBankEdges );
 149
 150   QString aRightGroupName = GetName() + "_Right_Bank";
 151
 152   Handle(HYDROData_ShapesGroup) aRightGroup = createGroupObject();
 153   aRightGroup->SetName( aRightGroupName );
 154   aRightGroup->SetShapes( aRightBankEdges );
 155
 156   QString anInGroupName = GetName() + "_Inlet";
 157
 158   Handle(HYDROData_ShapesGroup) anInGroup = createGroupObject();
 159   anInGroup->SetName( anInGroupName );
 160   anInGroup->SetShapes( anInletEdges );
 161
 162   QString anOutGroupName = GetName() + "_Outlet";
 163
 164   Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject();
 165   anOutGroup->SetName( anOutGroupName );
 166   anOutGroup->SetShapes( anOutletEdges );
 167 }
 168
 169 QColor HYDROData_Channel::DefaultFillingColor()
 170 {
 171   return QColor( Qt::blue );
 172 }
 173
 174 QColor HYDROData_Channel::DefaultBorderColor()
 175 {
 176   return QColor( Qt::transparent );
 177 }
 178
 179 QColor HYDROData_Channel::getDefaultFillingColor() const
 180 {
 181   return DefaultFillingColor();
 182 }
 183
 184 QColor HYDROData_Channel::getDefaultBorderColor() const
 185 {
 186   return DefaultBorderColor();
 187 }
 188
 189 bool HYDROData_Channel::SetGuideLine( const Handle(HYDROData_Polyline3D)& theGuideLine )
 190 {
 191   Handle(HYDROData_Polyline3D) aPrevGuideLine = GetGuideLine();
 192
 193   if ( theGuideLine.IsNull() )
 194   {
 195     RemoveGuideLine();
 196     return !aPrevGuideLine.IsNull();
 197   }
 198
 199   if ( IsEqual( aPrevGuideLine, theGuideLine ) )
 200     return false;
 201
 202   TopoDS_Wire aHydraulicWire = TopoDS::Wire( theGuideLine->GetTopShape() );
 203   if ( aHydraulicWire.IsNull() )
 204     return false; // The polyline must be a single wire
 205
 206   SetReferenceObject( theGuideLine, DataTag_GuideLine );
 207
 208   // Indicate model of the need to update the chanel presentation
 209   SetToUpdate( true );
 210
 211   return true;
 212 }
 213
 214 Handle(HYDROData_Polyline3D) HYDROData_Channel::GetGuideLine() const
 215 {
 216   return Handle(HYDROData_Polyline3D)::DownCast(
 217            GetReferenceObject( DataTag_GuideLine ) );
 218 }
 219
 220 void HYDROData_Channel::RemoveGuideLine()
 221 {
 222   Handle(HYDROData_Polyline3D) aPrevGuideLine = GetGuideLine();
 223   if ( aPrevGuideLine.IsNull() )
 224     return;
 225
 226   ClearReferenceObjects( DataTag_GuideLine );
 227
 228   // Indicate model of the need to update the chanel presentation
 229   SetToUpdate( true );
 230 }
 231
 232 bool HYDROData_Channel::SetProfile( const Handle(HYDROData_Profile)& theProfile )
 233 {
 234   Handle(HYDROData_Profile) aPrevProfile = GetProfile();
 235
 236   if ( theProfile.IsNull() )
 237   {
 238     RemoveProfile();
 239     return !aPrevProfile.IsNull();
 240   }
 241
 242   if ( IsEqual( aPrevProfile, theProfile ) )
 243     return false;
 244
 245   SetReferenceObject( theProfile, DataTag_Profile );
 246
 247   // Indicate model of the need to update the chanel presentation
 248   SetToUpdate( true );
 249
 250   return true;
 251 }
 252
 253 Handle(HYDROData_Profile) HYDROData_Channel::GetProfile() const
 254 {
 255   return Handle(HYDROData_Profile)::DownCast(
 256            GetReferenceObject( DataTag_Profile ) );
 257 }
 258
 259 void HYDROData_Channel::RemoveProfile()
 260 {
 261   Handle(HYDROData_Profile) aPrevProfile = GetProfile();
 262   if ( aPrevProfile.IsNull() )
 263     return;
 264
 265   ClearReferenceObjects( DataTag_Profile );
 266
 267   // Indicate model of the need to update the chanel presentation
 268   SetToUpdate( true );
 269 }
 270
 271 ObjectKind HYDROData_Channel::getAltitudeObjectType() const
 272 {
 273   return KIND_OBSTACLE_ALTITUDE;
 274 }
 275