src/HYDROGUI/HYDROGUI_Polyline.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 <HYDROGUI_Polyline.h>
  20 #include <BRepAdaptor_Curve.hxx>
  21 #include <BRepBndLib.hxx>
  22 #include <BRep_Tool.hxx>
  23 #include <GCPnts_AbscissaPoint.hxx>
  24 #include <GCPnts_QuasiUniformDeflection.hxx>
  25 #include <Graphic3d_ArrayOfPolylines.hxx>
  26 #include <Prs3d_Arrow.hxx>
  27 #include <Prs3d_LineAspect.hxx>
  28 #include <Prs3d_Point.hxx>
  29 #include <Prs3d_Root.hxx>
  30 #include <TopExp.hxx>
  31 #include <TopExp_Explorer.hxx>
  32 #include <TopoDS.hxx>
  33 #include <TopoDS_Edge.hxx>
  34 #include <TopoDS_Vertex.hxx>
  35
  36 IMPLEMENT_STANDARD_RTTIEXT(HYDROGUI_Polyline, AIS_Shape)
  37
  38
  39 HYDROGUI_Polyline::HYDROGUI_Polyline(const TopoDS_Shape& shape)
  40   : AIS_Shape(shape)
  41 {
  42 }
  43
  44 HYDROGUI_Polyline::~HYDROGUI_Polyline()
  45 {
  46 }
  47
  48 Handle( Graphic3d_ArrayOfPolylines ) BuildEdgePresentation( const TopoDS_Edge& theEdge, double theDeviation )
  49 {
  50   BRepAdaptor_Curve aCurveAdaptor( theEdge );
  51   GCPnts_QuasiUniformDeflection aPnts( aCurveAdaptor, theDeviation );
  52
  53   Handle( Graphic3d_ArrayOfPolylines ) anArray;
  54   if( !aPnts.IsDone() )
  55     return anArray;
  56
  57   int n = aPnts.NbPoints();
  58   anArray = new Graphic3d_ArrayOfPolylines( n );
  59   for( int i=1; i<=n; i++ )
  60     anArray->AddVertex( aPnts.Value( i ) );
  61
  62   return anArray;
  63 }
  64
  65 void HYDROGUI_Polyline::Compute(const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
  66                             const Handle(Prs3d_Presentation)& aPrs,
  67                             const Standard_Integer aMode)
  68 {
  69   //AIS_Shape::Compute(aPresentationManager, aPrs, aMode);
  70   //return;
  71
  72   aPrs->Clear();
  73   Handle(Graphic3d_Group) aGroup = Prs3d_Root::CurrentGroup( aPrs );
  74   Quantity_Color aColor;
  75   Aspect_TypeOfLine aType;
  76   Standard_Real anWidth;
  77   Attributes()->LineAspect()->Aspect()->Values( aColor, aType, anWidth );
  78   anWidth =2;
  79   Handle(Graphic3d_AspectLine3d) anAspect = new Graphic3d_AspectLine3d( aColor, aType, anWidth );
  80
  81   TopExp_Explorer Exp1 ( myshape, TopAbs_EDGE );
  82   Bnd_Box BB;
  83   BRepBndLib::AddClose(myshape, BB);
  84   double xmin, xmax, ymin, ymax, zmin, zmax;
  85   double devCoeff = 0.05;
  86   if (!BB.IsVoid())
  87   {
  88     BB.Get(xmin, ymin, zmin, xmax, ymax, zmax); //ignore Z coord
  89     double minSide = Min(Abs(xmax - xmin), Abs(ymax - ymin));
  90     devCoeff = minSide > 50 ? 0.05 : minSide / 3000;
  91   }
  92   for ( ; Exp1.More(); Exp1.Next() )
  93   {
  94     TopoDS_Edge anEdge = TopoDS::Edge( Exp1.Current() );
  95     Handle( Graphic3d_ArrayOfPolylines ) anArray = BuildEdgePresentation( anEdge, devCoeff );
  96     if( !anArray.IsNull() )
  97     {
  98       aGroup->SetPrimitivesAspect( anAspect );
  99       aGroup->AddPrimitiveArray( anArray );
 100     }
 101   }
 102 }
 103
 104 QList<Handle(AIS_InteractiveObject)> HYDROGUI_Polyline::createPresentations
 105   ( const TopoDS_Shape& theShape, int theType, int theSize )
 106 {
 107   QList<Handle(AIS_InteractiveObject)> shapes;
 108
 109   // 1. Main shape
 110   shapes.append( new HYDROGUI_Polyline( theShape ) );
 111
 112   // 2. Shapes for direction arrows on edges
 113   TopExp_Explorer Exp ( theShape, TopAbs_EDGE );
 114   for ( ; Exp.More(); Exp.Next() )
 115   {
 116     TopoDS_Edge anEdge = TopoDS::Edge(Exp.Current());
 117     if ( !anEdge.IsNull() )
 118     {
 119       Handle(HYDROGUI_Arrow) arrow = new HYDROGUI_Arrow( anEdge );
 120       if( theType>=0 )
 121         arrow->SetType( (HYDROGUI_Arrow::Type)theType );
 122       if( theSize>=0 )
 123         arrow->SetSize( theSize );
 124       shapes.append( arrow );
 125     }
 126   }
 127
 128   return shapes;
 129 }
 130
 131 void HYDROGUI_Polyline::update( const QList<Handle(AIS_InteractiveObject)>& theObjects, int theType, int theSize )
 132 {
 133   foreach( Handle(AIS_InteractiveObject) obj, theObjects )
 134   {
 135     Handle(HYDROGUI_Arrow) arrow = Handle(HYDROGUI_Arrow)::DownCast( obj );
 136     if( !arrow.IsNull() )
 137     {
 138       if( theType>=0 )
 139         arrow->SetType( (HYDROGUI_Arrow::Type)theType );
 140       if( theSize>=0 )
 141         arrow->SetSize( theSize );
 142     }
 143   }
 144 }
 145
 146
 147
 148
 149 IMPLEMENT_STANDARD_RTTIEXT(HYDROGUI_Arrow, AIS_Shape)
 150
 151
 152 HYDROGUI_Arrow::HYDROGUI_Arrow( const TopoDS_Edge& edge )
 153   : AIS_Shape( edge ), myType( Cone ), mySize( 35 )
 154 {
 155 }
 156
 157 HYDROGUI_Arrow::~HYDROGUI_Arrow()
 158 {
 159 }
 160
 161 HYDROGUI_Arrow::Type HYDROGUI_Arrow::GetType() const
 162 {
 163   return myType;
 164 }
 165
 166 void HYDROGUI_Arrow::SetType( HYDROGUI_Arrow::Type theType )
 167 {
 168   myType = theType;
 169 }
 170
 171 int HYDROGUI_Arrow::GetSize() const
 172 {
 173   return mySize;
 174 }
 175
 176 void HYDROGUI_Arrow::SetSize( int theSize )
 177 {
 178   mySize = qMax( theSize, 0 );
 179 }
 180
 181 void HYDROGUI_Arrow::Compute( const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
 182                                                       const Handle(Prs3d_Presentation)& aPrs,
 183                                                       const Standard_Integer aMode )
 184 {
 185   aPrs->Clear();
 186   if( myType==None )
 187     return;
 188
 189   TopoDS_Edge anEdge = TopoDS::Edge( myshape );
 190   if( anEdge.IsNull() )
 191     return;
 192
 193   BRepAdaptor_Curve anAdaptor( anEdge );
 194   double curveLen = GCPnts_AbscissaPoint::Length( anAdaptor, anAdaptor.FirstParameter(), anAdaptor.LastParameter() );
 195   double arrowLen;
 196   if( mySize==0 )
 197     // if size==0, then the arrow length is proportional to curve length
 198     arrowLen = curveLen/10;
 199   else
 200   {
 201     //arrowLen = qMin( curveLen/10, (double)mySize );
 202     arrowLen = (double)mySize;
 203   }
 204
 205   double t = ( anAdaptor.FirstParameter() + anAdaptor.LastParameter() ) / 2;
 206   gp_Pnt P, P1;
 207   gp_Vec V;
 208   anAdaptor.D1( t, P, V );
 209
 210   bool isFixed = mySize>0;
 211
 212   if( isFixed )
 213   {
 214     gp_Trsf tr;
 215     tr.SetTranslation( -gp_Vec( gp_Pnt(), P ) );
 216     aPrs->SetTransformation( new Geom_Transformation( tr ) );
 217
 218     Handle(Graphic3d_TransformPers) tp = new Graphic3d_TransformPers( Graphic3d_TMF_ZoomPers, P );
 219     SetTransformPersistence( tp );
 220
 221     P1 = gp_Pnt();
 222   }
 223   else
 224   {
 225     aPrs->SetTransformation( Handle(Geom_Transformation)() );
 226     SetTransformPersistence( Handle(Graphic3d_TransformPers)() );
 227     P1 = P;
 228   }
 229
 230   Handle(Graphic3d_Group) aGroup = Prs3d_Root::CurrentGroup( aPrs );
 231   Quantity_Color aColor;
 232   Aspect_TypeOfLine aType;
 233   Standard_Real anWidth;
 234   Attributes()->LineAspect()->Aspect()->Values( aColor, aType, anWidth );
 235   anWidth = 1;
 236
 237   if( myType==Cone )
 238   {
 239     Handle(Graphic3d_AspectLine3d) anAspect = new Graphic3d_AspectLine3d( aColor, aType, anWidth );
 240     aGroup->SetPrimitivesAspect( anAspect );
 241
 242     Prs3d_Arrow::Draw( aGroup, P1, V, M_PI/180.*12., arrowLen );
 243   }
 244   else
 245   {
 246     Graphic3d_MaterialAspect m( Graphic3d_NOM_PLASTIC );
 247     Handle(Graphic3d_AspectFillArea3d) anAspect = new Graphic3d_AspectFillArea3d
 248       ( Aspect_IS_SOLID, aColor, aColor, Aspect_TOL_SOLID, 1, m, m );
 249     aGroup->SetPrimitivesAspect( anAspect );
 250
 251     gp_Vec d = -V.Normalized() * arrowLen;
 252     const double k = 5.0;
 253     gp_Vec n( -d.Y()/k, d.X()/k, 0 );
 254
 255     gp_Pnt Pa = P1.Translated( d );
 256     gp_Pnt P2 = Pa.Translated( n );
 257     gp_Pnt P3 = Pa.Translated( -n );
 258
 259     Handle(Graphic3d_ArrayOfTriangles) prim = new Graphic3d_ArrayOfTriangles(3);
 260     prim->AddVertex( P1 );
 261     prim->AddVertex( P2 );
 262     prim->AddVertex( P3 );
 263
 264     aGroup->AddPrimitiveArray( prim );
 265   }
 266 }