1 // Copyright (C) 2007-2008 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 // File : GEOM_VTKTrihedron.cxx
24 // Author : Sergey LITONIN
28 #include "GEOM_VTKTrihedron.hxx"
30 #include <Geom_Axis2Placement.hxx>
35 #include <vtkMapper.h>
36 #include <vtkRenderer.h>
37 #include <vtkObjectFactory.h>
39 #include <vtkActorCollection.h>
40 #include <vtkProperty.h>
41 #include <vtkLineSource.h>
42 #include <vtkPolyDataMapper.h>
43 #include <vtkAppendPolyData.h>
44 #include <vtkMatrix4x4.h>
45 #include <vtkTransform.h>
46 #include <vtkVectorText.h>
47 #include <vtkCoordinate.h>
48 #include <vtkTextActor.h>
49 #include <vtkTextMapper.h>
52 Class : GEOM_VTKTrihedron
53 Description : Class for displaying axis of trihedron in VTK viewer.
54 Placement of trihedron is chahged with SetPlacement() method
56 class GEOM_VTKTrihedronAxis : public VTKViewer_Axis
60 GEOM_VTKTrihedronAxis();
61 GEOM_VTKTrihedronAxis( const GEOM_VTKTrihedronAxis& );
62 ~GEOM_VTKTrihedronAxis();
66 vtkTypeMacro( GEOM_VTKTrihedronAxis, VTKViewer_Axis );
67 static GEOM_VTKTrihedronAxis* New();
69 void SetAxis( const gp_Ax1& theAxis, const int theRot, vtkFloatingPointType theColor[ 3 ] );
70 virtual void Render( vtkRenderer* theRenderer );
71 virtual void SetSize( vtkFloatingPointType theSize );
72 gp_Pnt GetOri() const;
73 void SetColor( const vtkFloatingPointType theColor[ 3 ] );
77 vtkFloatingPointType myOri[ 3 ];
78 vtkMatrix4x4* myMatrix;
82 vtkStandardNewMacro( GEOM_VTKTrihedronAxis );
84 GEOM_VTKTrihedronAxis::GEOM_VTKTrihedronAxis()
86 VTKViewer_Axis::SetColor( 0.0, 0.0, 1.0 );
87 myMatrix = vtkMatrix4x4::New();
88 myTrsf = vtkTransform::New();
91 GEOM_VTKTrihedronAxis::~GEOM_VTKTrihedronAxis()
97 void GEOM_VTKTrihedronAxis::SetSize( vtkFloatingPointType theSize )
99 vtkFloatingPointType aPosition[ 3 ] = { myOri[ 0 ] + myDir[ 0 ] * theSize,
100 myOri[ 1 ] + myDir[ 1 ] * theSize,
101 myOri[ 2 ] + myDir[ 2 ] * theSize };
103 myLineSource->SetPoint1( myOri[ 0 ], myOri[ 1 ], myOri[ 2 ] );
104 myLineSource->SetPoint2( aPosition );
106 myArrowActor->SetPosition( 0, 0, 0 );
107 myArrowActor->SetOrientation( 0, 0, 0 );
108 myArrowActor->SetUserMatrix( 0 );
112 aTrsf.SetDisplacement( gp_Ax3( gp_Pnt( 0, 0, 0 ), gp_Dir( 1, 0, 0 ) ),
113 gp_Ax3( GetOri(), gp_Dir( myDir[ 0 ], myDir[ 1 ], myDir[ 2 ] ) ) );
115 const gp_XYZ& aTrans = aTrsf.TranslationPart();
116 gp_Mat aRot = aTrsf.VectorialPart();
118 for ( int i = 1; i <= 3; i++ )
119 for ( int j = 1; j <= 3; j++ )
120 myMatrix->SetElement( i - 1, j - 1, aRot( i, j ) );
122 myArrowActor->SetUserMatrix( myMatrix );
123 myArrowActor->SetPosition( aPosition );
126 if( vtkCoordinate* aCoord = myLabelActor->GetPositionCoordinate()->GetReferenceCoordinate() )
127 aCoord->SetValue( aPosition );
129 myLabelActor->SetPosition( 0, 0, 0 );
130 myLabelActor->AddPosition( aPosition );
134 void GEOM_VTKTrihedronAxis::Render(vtkRenderer* theRenderer)
136 myLineActor->Render( theRenderer );
137 myArrowActor->Render( theRenderer );
139 myLabelActor->Render( theRenderer );
142 vtkCamera* aCamera = theRenderer->GetActiveCamera();
143 SetCamera( aCamera );
146 gp_Pnt GEOM_VTKTrihedronAxis::GetOri() const
148 return gp_Pnt( myOri[ 0 ], myOri[ 1 ], myOri[ 2 ] );
151 void GEOM_VTKTrihedronAxis::SetAxis( const gp_Ax1& theAxis,
153 vtkFloatingPointType theColor[ 3 ] )
155 gp_Pnt aLoc = theAxis.Location();
156 gp_Dir aDir = theAxis.Direction();
158 myOri[ 0 ] = aLoc.X();
159 myOri[ 1 ] = aLoc.Y();
160 myOri[ 2 ] = aLoc.Z();
162 myDir[ 0 ] = aDir.X();
163 myDir[ 1 ] = aDir.Y();
164 myDir[ 2 ] = aDir.Z();
166 vtkFloatingPointType aColor[ 3 ] = { 0, 0, 0 };
167 aColor[ theRot ] = 1;
168 if ( theColor[ 0 ] == -1 )
169 VTKViewer_Axis::SetColor( aColor[ 0 ], aColor[ 1 ], aColor[ 2 ] );
171 VTKViewer_Axis::SetColor( theColor[ 0 ], theColor[ 1 ], theColor[ 2 ] );
174 if ( theRot == 0 ) myTextMapper->SetInput( "X" );
175 else if ( theRot == 1 ) myTextMapper->SetInput( "Y" );
176 else if ( theRot == 2 ) myTextMapper->SetInput( "Z" );
178 if ( theRot == 0 ) myVectorText->SetText( "X" );
179 else if ( theRot == 1 ) myVectorText->SetText( "Y" );
180 else if ( theRot == 2 ) myVectorText->SetText( "Z" );
184 void GEOM_VTKTrihedronAxis::SetColor( const vtkFloatingPointType theColor[ 3 ] )
186 VTKViewer_Axis::SetColor( theColor[ 0 ], theColor[ 1 ], theColor[ 2 ] );
191 Class : GEOM_VTKTrihedron
192 Description : Class for displaying trihedron of local CS in VTK viewer.
193 Placement of trihedron is chahged with SetPlacement() method
196 vtkStandardNewMacro( GEOM_VTKTrihedron );
198 GEOM_VTKTrihedron::GEOM_VTKTrihedron()
200 myMapper = vtkPolyDataMapper::New();
201 myAxis[ 0 ] = myAxis[ 1 ] = myAxis[ 2 ] = 0;
204 myColor[ 0 ] = myColor[ 1 ] = myColor[ 2 ] = -1;
206 myDefaultColor[ 0 ] = myDefaultColor[ 1 ] = myDefaultColor[ 2 ] = 1;
208 myPreHighlightColor[ 0 ] = 0;
209 myPreHighlightColor[ 1 ] = myPreHighlightColor[ 2 ] = 1;
211 myHighlightColor[ 0 ] = myHighlightColor[ 1 ] = myHighlightColor[ 2 ] = 1;
213 //SetInfinitive( true );
217 GEOM_VTKTrihedron::~GEOM_VTKTrihedron()
219 for ( int i = 0; i < 3; i++ )
226 void GEOM_VTKTrihedron::SetSize( vtkFloatingPointType theSize )
229 for ( int i = 0; i < 3; i++ )
230 myAxis[i]->SetSize( theSize );
233 // Calculate new sensitive area
234 gp_Pnt anOri = ( (GEOM_VTKTrihedronAxis*)myAxis[ 0 ] )->GetOri();
235 gp_Pnt aEndX( myLocation.XYZ() + myDirX.XYZ() * mySize );
236 gp_Pnt aEndY( myLocation.XYZ() + myDirY.XYZ() * mySize );
237 gp_Pnt aEndZ( myLocation.XYZ() + myDirZ.XYZ() * mySize );
239 vtkLineSource* aSrcX = vtkLineSource::New();
240 vtkLineSource* aSrcY = vtkLineSource::New();
241 vtkLineSource* aSrcZ = vtkLineSource::New();
243 aSrcX->SetPoint1( anOri.X(), anOri.Y(), anOri.Z() );
244 aSrcX->SetPoint2( aEndX.X(), aEndX.Y(), aEndX.Z() );
246 aSrcY->SetPoint1( anOri.X(), anOri.Y(), anOri.Z() );
247 aSrcY->SetPoint2( aEndY.X(), aEndY.Y(), aEndY.Z() );
249 aSrcZ->SetPoint1( anOri.X(), anOri.Y(), anOri.Z() );
250 aSrcZ->SetPoint2( aEndZ.X(), aEndZ.Y(), aEndZ.Z() );
252 vtkAppendPolyData* aRes = vtkAppendPolyData::New();
253 aRes->AddInput( aSrcX->GetOutput() );
254 aRes->AddInput( aSrcY->GetOutput() );
255 aRes->AddInput( aSrcZ->GetOutput() );
257 myMapper->SetInput( aRes->GetOutput() );
258 SALOME_Actor::SetMapper( myMapper );
266 void GEOM_VTKTrihedron::SetVisibility( int theVisibility )
268 Superclass::SetVisibility( theVisibility );
269 SetVisibility( theVisibility == 1 ? VTKViewer_Trihedron::eOn : VTKViewer_Trihedron::eOff );
272 void GEOM_VTKTrihedron::SetVisibility( VTKViewer_Trihedron::TVisibility theVis )
274 for ( int i = 0; i < 3; i++ )
275 myAxis[i]->SetVisibility( theVis );
278 void GEOM_VTKTrihedron::AddToRender( vtkRenderer* theRenderer )
280 vtkCamera* aCamera = theRenderer->GetActiveCamera();
281 for(int i = 0; i < 3; i++)
283 myAxis[i]->AddToRender(theRenderer);
284 myAxis[i]->SetCamera(aCamera);
287 theRenderer->AddActor( this );
290 void GEOM_VTKTrihedron::RemoveFromRender( vtkRenderer* theRenderer )
292 for(int i = 0; i < 3; i++)
294 myAxis[i]->RemoveFromRender(theRenderer);
297 theRenderer->RemoveActor( this );
300 int GEOM_VTKTrihedron::GetVisibleActorCount(vtkRenderer* theRenderer)
302 vtkActorCollection* aCollection = theRenderer->GetActors();
303 aCollection->InitTraversal();
305 while( vtkActor* prop = aCollection->GetNextActor() )
306 if ( prop->GetVisibility() )
307 if ( SALOME_Actor* anActor = SALOME_Actor::SafeDownCast( prop ) )
308 if ( !anActor->IsInfinitive() )
311 vtkCamera* aCamera = theRenderer->GetActiveCamera();
312 for(int i = 0; i < 3; i++)
313 myAxis[i]->SetCamera( aCamera );
318 void GEOM_VTKTrihedron::SetCamera( vtkCamera* theCamera )
320 for(int i = 0; i < 3; i++)
321 myAxis[i]->SetCamera( theCamera );
324 void GEOM_VTKTrihedron::SetPlacement( const Handle(Geom_Axis2Placement)& thePlc )
326 gp_Ax2 anAx2 = thePlc->Ax2();
328 myDirX = anAx2.XDirection();
329 myDirY = anAx2.YDirection();
330 myDirZ = anAx2.Direction();
332 myLocation = anAx2.Axis().Location();
334 if ( myAxis[ 0 ] != 0 ) myAxis[ 0 ]->Delete();
335 if ( myAxis[ 1 ] != 0 ) myAxis[ 1 ]->Delete();
336 if ( myAxis[ 2 ] != 0 ) myAxis[ 2 ]->Delete();
338 myAxis[ 0 ] = GEOM_VTKTrihedronAxis::New();
339 myAxis[ 1 ] = GEOM_VTKTrihedronAxis::New();
340 myAxis[ 2 ] = GEOM_VTKTrihedronAxis::New();
342 ( (GEOM_VTKTrihedronAxis*)myAxis[ 0 ] )->SetAxis( gp_Ax1( myLocation, myDirX ), 0, myColor );
343 ( (GEOM_VTKTrihedronAxis*)myAxis[ 1 ] )->SetAxis( gp_Ax1( myLocation, myDirY ), 1, myColor );
344 ( (GEOM_VTKTrihedronAxis*)myAxis[ 2 ] )->SetAxis( gp_Ax1( myLocation, myDirZ ), 2, myColor );
349 vtkMapper* GEOM_VTKTrihedron::GetMapper()
354 void GEOM_VTKTrihedron::Render(vtkRenderer* r, vtkMapper *)
356 ( (GEOM_VTKTrihedronAxis*)myAxis[ 0 ] )->Render( r );
357 ( (GEOM_VTKTrihedronAxis*)myAxis[ 1 ] )->Render( r );
358 ( (GEOM_VTKTrihedronAxis*)myAxis[ 2 ] )->Render( r );
361 void GEOM_VTKTrihedron::SetColor( vtkFloatingPointType r, vtkFloatingPointType g, vtkFloatingPointType b )
367 if ( myAxis[ 0 ] ) ( (GEOM_VTKTrihedronAxis*)myAxis[ 0 ] )->SetColor( myColor );
368 if ( myAxis[ 1 ] ) ( (GEOM_VTKTrihedronAxis*)myAxis[ 1 ] )->SetColor( myColor );
369 if ( myAxis[ 2 ] ) ( (GEOM_VTKTrihedronAxis*)myAxis[ 2 ] )->SetColor( myColor );
372 void GEOM_VTKTrihedron::GetColor( vtkFloatingPointType& r, vtkFloatingPointType& g, vtkFloatingPointType& b )
379 bool GEOM_VTKTrihedron::IsSetCamera() const
384 bool GEOM_VTKTrihedron::IsResizable() const
389 void GEOM_VTKTrihedron::Highlight( bool theIsHighlight )
392 SetAxesColors( myHighlightColor );
396 Superclass::Highlight( theIsHighlight );
399 bool GEOM_VTKTrihedron::PreHighlight( vtkInteractorStyle *theInteractorStyle,
400 SVTK_SelectionEvent* theSelectionEvent,
401 bool theIsHighlight )
403 if ( !GetPickable() )
406 if ( !isHighlighted() )
409 SetAxesColors( myPreHighlightColor );
414 return Superclass::PreHighlight( theInteractorStyle, theSelectionEvent, theIsHighlight );
417 void GEOM_VTKTrihedron::ResetAxesColors()
419 if( myColor[0] != -1 )
420 SetAxesColors( myColor );
422 SetAxesColors( myDefaultColor, true );
425 void GEOM_VTKTrihedron::SetAxesColors( vtkFloatingPointType theColor[3], bool theIsDiffuse )
427 myAxis[ 0 ]->SetColor( theColor[0], theIsDiffuse ? 0.0 : theColor[1], theIsDiffuse ? 0.0 : theColor[2] );
428 myAxis[ 1 ]->SetColor( theIsDiffuse ? 0.0 : theColor[0], theColor[1], theIsDiffuse ? 0.0 : theColor[2] );
429 myAxis[ 2 ]->SetColor( theIsDiffuse ? 0.0 : theColor[0], theIsDiffuse ? 0.0 : theColor[1], theColor[2] );