src/OBJECT/SMESH_DeviceActor.cxx

   1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   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.
  10 //
  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.
  15 //
  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
  19 //
  20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 //  SMESH OBJECT : interactive object for SMESH visualization
  24 //  File   : SMESH_DeviceActor.cxx
  25 //  Author :
  26 //  Module : SMESH
  27 //
  28 #include "SMESH_DeviceActor.h"
  29 #include "SMESH_ScalarBarActor.h"
  30 #include "SMESH_ExtractGeometry.h"
  31 #include "SMESH_ControlsDef.hxx"
  32 #include "SMESH_ActorUtils.h"
  33 #include "SMESH_FaceOrientationFilter.h"
  34 #include "VTKViewer_CellLocationsArray.h"
  35 #include "VTKViewer_PolyDataMapper.h"
  36
  37 #include <VTKViewer_Transform.h>
  38 #include <VTKViewer_TransformFilter.h>
  39 #include <VTKViewer_ExtractUnstructuredGrid.h>
  40
  41 // VTK Includes
  42 #include <vtkObjectFactory.h>
  43 #include <vtkShrinkFilter.h>
  44 #include <vtkShrinkPolyData.h>
  45
  46 #include <vtkProperty.h>
  47 #include <vtkPolyData.h>
  48 #include <vtkMergeFilter.h>
  49 #include <vtkPolyDataMapper.h>
  50 #include <vtkUnstructuredGrid.h>
  51
  52 #include <vtkLookupTable.h>
  53 #include <vtkDoubleArray.h>
  54 #include <vtkCellData.h>
  55
  56 #include <vtkCell.h>
  57 #include <vtkIdList.h>
  58 #include <vtkCellArray.h>
  59 #include <vtkUnsignedCharArray.h>
  60
  61 #include <vtkImplicitBoolean.h>
  62 #include <vtkPassThroughFilter.h>
  63
  64 #include <vtkRenderer.h>
  65
  66 #include "utilities.h"
  67
  68 #ifdef _DEBUG_
  69 static int MYDEBUG = 0;
  70 #else
  71 static int MYDEBUG = 0;
  72 #endif
  73
  74 using namespace std;
  75
  76
  77 vtkStandardNewMacro(SMESH_DeviceActor);
  78
  79
  80 SMESH_DeviceActor
  81 ::SMESH_DeviceActor()
  82 {
  83   if(MYDEBUG) MESSAGE("SMESH_DeviceActor - "<<this);
  84
  85   myIsShrinkable = false;
  86   myIsShrunk = false;
  87   myIsHighlited = false;
  88
  89   myRepresentation = eSurface;
  90
  91   myProperty = vtkProperty::New();
  92   myMapper = VTKViewer_PolyDataMapper::New();
  93
  94   vtkMapper::GetResolveCoincidentTopologyPolygonOffsetParameters(myPolygonOffsetFactor,
  95                                                                  myPolygonOffsetUnits);
  96
  97   myMapper->UseLookupTableScalarRangeOn();
  98   myMapper->SetColorModeToMapScalars();
  99
 100   myShrinkFilter = vtkShrinkFilter::New();
 101
 102   myStoreClippingMapping = false;
 103
 104   myExtractGeometry = SMESH_ExtractGeometry::New();
 105   myExtractGeometry->SetReleaseDataFlag(true);
 106   myIsImplicitFunctionUsed = false;
 107
 108   myExtractUnstructuredGrid = VTKViewer_ExtractUnstructuredGrid::New();
 109
 110   myMergeFilter = vtkMergeFilter::New();
 111
 112   myGeomFilter = VTKViewer_GeometryFilter::New();
 113
 114   myTransformFilter = VTKViewer_TransformFilter::New();
 115
 116   for(int i = 0; i < 6; i++)
 117     myPassFilter.push_back(vtkPassThroughFilter::New());
 118
 119   // Orientation of faces
 120   myIsFacesOriented = false;
 121
 122   vtkFloatingPointType anRGB[3] = { 1, 1, 1 };
 123   SMESH::GetColor( "SMESH", "orientation_color", anRGB[0], anRGB[1], anRGB[2], QColor( 255, 255, 255 ) );
 124
 125   myFaceOrientationFilter = SMESH_FaceOrientationFilter::New();
 126
 127   myFaceOrientationDataMapper = vtkPolyDataMapper::New();
 128   myFaceOrientationDataMapper->SetInput(myFaceOrientationFilter->GetOutput());
 129
 130   myFaceOrientation = vtkActor::New();
 131   myFaceOrientation->SetMapper(myFaceOrientationDataMapper);
 132   myFaceOrientation->GetProperty()->SetColor(anRGB[0], anRGB[1], anRGB[2]);
 133 }
 134
 135
 136 SMESH_DeviceActor
 137 ::~SMESH_DeviceActor()
 138 {
 139   if(MYDEBUG) MESSAGE("~SMESH_DeviceActor - "<<this);
 140
 141   myProperty->Delete();
 142
 143   myMapper->Delete();
 144
 145   myShrinkFilter->Delete();
 146
 147   myExtractUnstructuredGrid->Delete();
 148
 149   myMergeFilter->Delete();
 150
 151   myGeomFilter->Delete();
 152
 153   myExtractGeometry->Delete();
 154
 155   myTransformFilter->Delete();
 156
 157   for(int i = 0, iEnd = myPassFilter.size(); i < iEnd; i++){
 158     myPassFilter[i]->Delete();
 159   }
 160
 161   // Orientation of faces
 162   myFaceOrientationFilter->Delete();
 163
 164   myFaceOrientationDataMapper->RemoveAllInputs();
 165   myFaceOrientationDataMapper->Delete();
 166
 167   myFaceOrientation->Delete();
 168 }
 169
 170
 171 void
 172 SMESH_DeviceActor
 173 ::SetStoreGemetryMapping(bool theStoreMapping)
 174 {
 175   myGeomFilter->SetStoreMapping(theStoreMapping);
 176   SetStoreClippingMapping(theStoreMapping);
 177 }
 178
 179
 180 void
 181 SMESH_DeviceActor
 182 ::SetStoreClippingMapping(bool theStoreMapping)
 183 {
 184   myStoreClippingMapping = theStoreMapping;
 185   myExtractGeometry->SetStoreMapping(theStoreMapping && myIsImplicitFunctionUsed);
 186   SetStoreIDMapping(theStoreMapping);
 187 }
 188
 189
 190 void
 191 SMESH_DeviceActor
 192 ::SetStoreIDMapping(bool theStoreMapping)
 193 {
 194   myExtractUnstructuredGrid->SetStoreMapping(theStoreMapping);
 195 }
 196
 197
 198 void
 199 SMESH_DeviceActor
 200 ::Init(TVisualObjPtr theVisualObj,
 201        vtkImplicitBoolean* theImplicitBoolean)
 202 {
 203   myVisualObj = theVisualObj;
 204   myExtractGeometry->SetImplicitFunction(theImplicitBoolean);
 205   SetUnstructuredGrid(myVisualObj->GetUnstructuredGrid());
 206 }
 207
 208
 209 void
 210 SMESH_DeviceActor
 211 ::SetImplicitFunctionUsed(bool theIsImplicitFunctionUsed)
 212 {
 213   int anId = 0;
 214   if(theIsImplicitFunctionUsed)
 215     myPassFilter[ anId ]->SetInput( myExtractGeometry->GetOutput() );
 216   else
 217     myPassFilter[ anId ]->SetInput( myMergeFilter->GetOutput() );
 218
 219   myIsImplicitFunctionUsed = theIsImplicitFunctionUsed;
 220   SetStoreClippingMapping(myStoreClippingMapping);
 221 }
 222
 223
 224 void
 225 SMESH_DeviceActor
 226 ::SetUnstructuredGrid(vtkUnstructuredGrid* theGrid)
 227 {
 228   if(theGrid){
 229     //myIsShrinkable = theGrid->GetNumberOfCells() > 10;
 230     myIsShrinkable = true;
 231
 232     myExtractUnstructuredGrid->SetInput(theGrid);
 233
 234     myMergeFilter->SetGeometry(myExtractUnstructuredGrid->GetOutput());
 235
 236     myExtractGeometry->SetInput(myMergeFilter->GetOutput());
 237
 238     int anId = 0;
 239     SetImplicitFunctionUsed(myIsImplicitFunctionUsed);
 240     myPassFilter[ anId + 1]->SetInput( myPassFilter[ anId ]->GetOutput() );
 241
 242     anId++; // 1
 243     myGeomFilter->SetInput( myPassFilter[ anId ]->GetOutput() );
 244
 245     anId++; // 2
 246     myPassFilter[ anId ]->SetInput( myGeomFilter->GetOutput() );
 247     myPassFilter[ anId + 1 ]->SetInput( myPassFilter[ anId ]->GetOutput() );
 248
 249     anId++; // 3
 250     myTransformFilter->SetInput( myPassFilter[ anId ]->GetPolyDataOutput() );
 251
 252     anId++; // 4
 253     myPassFilter[ anId ]->SetInput( myTransformFilter->GetOutput() );
 254     myPassFilter[ anId + 1 ]->SetInput( myPassFilter[ anId ]->GetOutput() );
 255
 256     anId++; // 5
 257     myMapper->SetInput( myPassFilter[ anId ]->GetPolyDataOutput() );
 258
 259     vtkLODActor::SetMapper( myMapper );
 260     Modified();
 261   }
 262 }
 263
 264
 265 VTKViewer_ExtractUnstructuredGrid*
 266 SMESH_DeviceActor
 267 ::GetExtractUnstructuredGrid()
 268 {
 269   return myExtractUnstructuredGrid;
 270 }
 271
 272
 273 vtkUnstructuredGrid*
 274 SMESH_DeviceActor
 275 ::GetUnstructuredGrid()
 276 {
 277   myExtractUnstructuredGrid->Update();
 278   return myExtractUnstructuredGrid->GetOutput();
 279 }
 280
 281
 282 void
 283 SMESH_DeviceActor
 284 ::SetControlMode(SMESH::Controls::FunctorPtr theFunctor,
 285                  SMESH_ScalarBarActor* theScalarBarActor,
 286                  vtkLookupTable* theLookupTable)
 287 {
 288   bool anIsInitialized = theFunctor;
 289   if(anIsInitialized){
 290     vtkUnstructuredGrid* aDataSet = vtkUnstructuredGrid::New();
 291
 292     SetStoreIDMapping(true);
 293     myExtractUnstructuredGrid->Update();
 294     vtkUnstructuredGrid* aGrid = myExtractUnstructuredGrid->GetOutput();
 295
 296     aDataSet->ShallowCopy(aGrid);
 297
 298     vtkDoubleArray *aScalars = vtkDoubleArray::New();
 299     vtkIdType aNbCells = aGrid->GetNumberOfCells();
 300     aScalars->SetNumberOfComponents(1);
 301     aScalars->SetNumberOfTuples(aNbCells);
 302
 303     myVisualObj->UpdateFunctor(theFunctor);
 304
 305     using namespace SMESH::Controls;
 306     if(NumericalFunctor* aNumericalFunctor = dynamic_cast<NumericalFunctor*>(theFunctor.get())){
 307       for(vtkIdType i = 0; i < aNbCells; i++){
 308         vtkIdType anId = myExtractUnstructuredGrid->GetInputId(i);
 309         vtkIdType anObjId = myVisualObj->GetElemObjId(anId);
 310         double aValue = aNumericalFunctor->GetValue(anObjId);
 311         aScalars->SetValue(i,aValue);
 312       }
 313     }else if(Predicate* aPredicate = dynamic_cast<Predicate*>(theFunctor.get())){
 314       for(vtkIdType i = 0; i < aNbCells; i++){
 315         vtkIdType anId = myExtractUnstructuredGrid->GetInputId(i);
 316         vtkIdType anObjId = myVisualObj->GetElemObjId(anId);
 317         bool aValue = aPredicate->IsSatisfy(anObjId);
 318         aScalars->SetValue(i,aValue);
 319       }
 320     }
 321
 322     aDataSet->GetCellData()->SetScalars(aScalars);
 323     aScalars->Delete();
 324
 325     theLookupTable->SetRange(aScalars->GetRange());
 326     theLookupTable->SetNumberOfTableValues(theScalarBarActor->GetMaximumNumberOfColors());
 327     theLookupTable->Build();
 328
 329     myMergeFilter->SetScalars(aDataSet);
 330     aDataSet->Delete();
 331   }
 332   GetMapper()->SetScalarVisibility(anIsInitialized);
 333   theScalarBarActor->SetVisibility(anIsInitialized);
 334 }
 335
 336 void
 337 SMESH_DeviceActor
 338 ::SetExtControlMode(SMESH::Controls::FunctorPtr theFunctor,
 339                     SMESH_ScalarBarActor* theScalarBarActor,
 340                     vtkLookupTable* theLookupTable)
 341 {
 342   bool anIsInitialized = theFunctor;
 343   myExtractUnstructuredGrid->ClearRegisteredCells();
 344   myExtractUnstructuredGrid->ClearRegisteredCellsWithType();
 345   myExtractUnstructuredGrid->SetModeOfChanging(VTKViewer_ExtractUnstructuredGrid::ePassAll);
 346   myVisualObj->UpdateFunctor(theFunctor);
 347
 348   using namespace SMESH::Controls;
 349   if (anIsInitialized){
 350     if (Length2D* aLength2D = dynamic_cast<Length2D*>(theFunctor.get())){
 351       SMESH::Controls::Length2D::TValues aValues;
 352
 353       aLength2D->GetValues(aValues);
 354       vtkUnstructuredGrid* aDataSet = vtkUnstructuredGrid::New();
 355       vtkUnstructuredGrid* aGrid = myVisualObj->GetUnstructuredGrid();
 356
 357       aDataSet->SetPoints(aGrid->GetPoints());
 358
 359       vtkIdType aNbCells = aValues.size();
 360
 361       vtkDoubleArray *aScalars = vtkDoubleArray::New();
 362       aScalars->SetNumberOfComponents(1);
 363       aScalars->SetNumberOfTuples(aNbCells);
 364
 365       vtkIdType aCellsSize = 3*aNbCells;
 366       vtkCellArray* aConnectivity = vtkCellArray::New();
 367       aConnectivity->Allocate( aCellsSize, 0 );
 368
 369       vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
 370       aCellTypesArray->SetNumberOfComponents( 1 );
 371       aCellTypesArray->Allocate( aNbCells * aCellTypesArray->GetNumberOfComponents() );
 372
 373       vtkIdList *anIdList = vtkIdList::New();
 374       anIdList->SetNumberOfIds(2);
 375
 376       Length2D::TValues::const_iterator anIter = aValues.begin();
 377       for(vtkIdType aVtkId = 0; anIter != aValues.end(); anIter++,aVtkId++){
 378         const Length2D::Value& aValue = *anIter;
 379         int aNode[2] = {
 380           myVisualObj->GetNodeVTKId(aValue.myPntId[0]),
 381           myVisualObj->GetNodeVTKId(aValue.myPntId[1])
 382         };
 383         if(aNode[0] >= 0 && aNode[1] >= 0){
 384           anIdList->SetId( 0, aNode[0] );
 385           anIdList->SetId( 1, aNode[1] );
 386           aConnectivity->InsertNextCell( anIdList );
 387           aCellTypesArray->InsertNextValue( VTK_LINE );
 388           aScalars->SetValue(aVtkId,aValue.myLength);
 389         }
 390       }
 391
 392       VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
 393       aCellLocationsArray->SetNumberOfComponents( 1 );
 394       aCellLocationsArray->SetNumberOfTuples( aNbCells );
 395
 396       aConnectivity->InitTraversal();
 397       for( vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell( npts, pts ); idType++ )
 398         aCellLocationsArray->SetValue( idType, aConnectivity->GetTraversalLocation( npts ) );
 399
 400       aDataSet->SetCells( aCellTypesArray, aCellLocationsArray,aConnectivity );
 401       SetUnstructuredGrid(aDataSet);
 402
 403       aDataSet->GetCellData()->SetScalars(aScalars);
 404       aScalars->Delete();
 405
 406       theLookupTable->SetRange(aScalars->GetRange());
 407       theLookupTable->Build();
 408
 409       myMergeFilter->SetScalars(aDataSet);
 410       aDataSet->Delete();
 411     }
 412     else if (MultiConnection2D* aMultiConnection2D = dynamic_cast<MultiConnection2D*>(theFunctor.get())){
 413       SMESH::Controls::MultiConnection2D::MValues aValues;
 414
 415       aMultiConnection2D->GetValues(aValues);
 416       vtkUnstructuredGrid* aDataSet = vtkUnstructuredGrid::New();
 417       vtkUnstructuredGrid* aGrid = myVisualObj->GetUnstructuredGrid();
 418       aDataSet->SetPoints(aGrid->GetPoints());
 419
 420       vtkIdType aNbCells = aValues.size();
 421       vtkDoubleArray *aScalars = vtkDoubleArray::New();
 422       aScalars->SetNumberOfComponents(1);
 423       aScalars->SetNumberOfTuples(aNbCells);
 424
 425       vtkIdType aCellsSize = 3*aNbCells;
 426       vtkCellArray* aConnectivity = vtkCellArray::New();
 427       aConnectivity->Allocate( aCellsSize, 0 );
 428
 429       vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
 430       aCellTypesArray->SetNumberOfComponents( 1 );
 431       aCellTypesArray->Allocate( aNbCells * aCellTypesArray->GetNumberOfComponents() );
 432
 433       vtkIdList *anIdList = vtkIdList::New();
 434       anIdList->SetNumberOfIds(2);
 435
 436       MultiConnection2D::MValues::const_iterator anIter = aValues.begin();
 437       for(vtkIdType aVtkId = 0; anIter != aValues.end(); anIter++,aVtkId++){
 438         const MultiConnection2D::Value& aValue = (*anIter).first;
 439         int aNode[2] = {
 440           myVisualObj->GetNodeVTKId(aValue.myPntId[0]),
 441           myVisualObj->GetNodeVTKId(aValue.myPntId[1])
 442         };
 443         if(aNode[0] >= 0 && aNode[1] >= 0){
 444           anIdList->SetId( 0, aNode[0] );
 445           anIdList->SetId( 1, aNode[1] );
 446           aConnectivity->InsertNextCell( anIdList );
 447           aCellTypesArray->InsertNextValue( VTK_LINE );
 448           aScalars->SetValue(aVtkId,(*anIter).second);
 449         }
 450       }
 451
 452       VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
 453       aCellLocationsArray->SetNumberOfComponents( 1 );
 454       aCellLocationsArray->SetNumberOfTuples( aNbCells );
 455
 456       aConnectivity->InitTraversal();
 457       for( vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell( npts, pts ); idType++ )
 458         aCellLocationsArray->SetValue( idType, aConnectivity->GetTraversalLocation( npts ) );
 459
 460       aDataSet->SetCells( aCellTypesArray, aCellLocationsArray,aConnectivity );
 461       SetUnstructuredGrid(aDataSet);
 462
 463       aDataSet->GetCellData()->SetScalars(aScalars);
 464       aScalars->Delete();
 465
 466       theLookupTable->SetRange(aScalars->GetRange());
 467       theLookupTable->Build();
 468
 469       myMergeFilter->SetScalars(aDataSet);
 470       aDataSet->Delete();
 471     }
 472   }
 473   GetMapper()->SetScalarVisibility(anIsInitialized);
 474   theScalarBarActor->SetVisibility(anIsInitialized);
 475 }
 476
 477 void
 478 SMESH_DeviceActor
 479 ::SetExtControlMode(SMESH::Controls::FunctorPtr theFunctor)
 480 {
 481   myExtractUnstructuredGrid->ClearRegisteredCells();
 482   myExtractUnstructuredGrid->ClearRegisteredCellsWithType();
 483   myExtractUnstructuredGrid->SetModeOfChanging(VTKViewer_ExtractUnstructuredGrid::ePassAll);
 484   myVisualObj->UpdateFunctor(theFunctor);
 485
 486   using namespace SMESH::Controls;
 487   if ( dynamic_cast<FreeBorders          *>(theFunctor.get()) ||
 488        dynamic_cast<FreeFaces            *>(theFunctor.get()) ||
 489        dynamic_cast<BareBorderVolume     *>(theFunctor.get()) ||
 490        dynamic_cast<BareBorderFace       *>(theFunctor.get()) ||
 491        dynamic_cast<OverConstrainedVolume*>(theFunctor.get()) ||
 492        dynamic_cast<OverConstrainedFace  *>(theFunctor.get()))
 493   {
 494     Predicate* aFreePredicate = dynamic_cast<Predicate*>(theFunctor.get());
 495     myExtractUnstructuredGrid->SetModeOfChanging(VTKViewer_ExtractUnstructuredGrid::eAdding);
 496     vtkUnstructuredGrid* aGrid = myVisualObj->GetUnstructuredGrid();
 497     vtkIdType aNbCells = aGrid->GetNumberOfCells();
 498     for( vtkIdType i = 0; i < aNbCells; i++ ){
 499       vtkIdType anObjId = myVisualObj->GetElemObjId(i);
 500       if(aFreePredicate->IsSatisfy(anObjId))
 501         myExtractUnstructuredGrid->RegisterCell(i);
 502     }
 503     if(!myExtractUnstructuredGrid->IsCellsRegistered())
 504       myExtractUnstructuredGrid->RegisterCell(-1);
 505     SetUnstructuredGrid(myVisualObj->GetUnstructuredGrid());
 506   }
 507   else if(FreeEdges* aFreeEdges = dynamic_cast<FreeEdges*>(theFunctor.get()))
 508   {
 509     SMESH::Controls::FreeEdges::TBorders aBorders;
 510     aFreeEdges->GetBoreders(aBorders);
 511     vtkUnstructuredGrid* aDataSet = vtkUnstructuredGrid::New();
 512     vtkUnstructuredGrid* aGrid = myVisualObj->GetUnstructuredGrid();
 513     aDataSet->SetPoints(aGrid->GetPoints());
 514
 515     vtkIdType aNbCells = aBorders.size();
 516     vtkIdType aCellsSize = 3*aNbCells;
 517     vtkCellArray* aConnectivity = vtkCellArray::New();
 518     aConnectivity->Allocate( aCellsSize, 0 );
 519
 520     vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
 521     aCellTypesArray->SetNumberOfComponents( 1 );
 522     aCellTypesArray->Allocate( aNbCells * aCellTypesArray->GetNumberOfComponents() );
 523
 524     vtkIdList *anIdList = vtkIdList::New();
 525     anIdList->SetNumberOfIds(2);
 526
 527     FreeEdges::TBorders::const_iterator anIter = aBorders.begin();
 528     for(; anIter != aBorders.end(); anIter++){
 529       const FreeEdges::Border& aBorder = *anIter;
 530       int aNode[2] = {
 531         myVisualObj->GetNodeVTKId(aBorder.myPntId[0]),
 532         myVisualObj->GetNodeVTKId(aBorder.myPntId[1])
 533       };
 534       //cout<<"aNode = "<<aBorder.myPntId[0]<<"; "<<aBorder.myPntId[1]<<endl;
 535       if(aNode[0] >= 0 && aNode[1] >= 0){
 536         anIdList->SetId( 0, aNode[0] );
 537         anIdList->SetId( 1, aNode[1] );
 538         aConnectivity->InsertNextCell( anIdList );
 539         aCellTypesArray->InsertNextValue( VTK_LINE );
 540       }
 541     }
 542
 543     VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
 544     aCellLocationsArray->SetNumberOfComponents( 1 );
 545     aCellLocationsArray->SetNumberOfTuples( aNbCells );
 546
 547     aConnectivity->InitTraversal();
 548     for( vtkIdType idType = 0, *pts, npts; aConnectivity->GetNextCell( npts, pts ); idType++ )
 549       aCellLocationsArray->SetValue( idType, aConnectivity->GetTraversalLocation( npts ) );
 550
 551     aDataSet->SetCells( aCellTypesArray, aCellLocationsArray,aConnectivity );
 552
 553     SetUnstructuredGrid(aDataSet);
 554     aDataSet->Delete();
 555   }
 556   else if(FreeNodes* aFreeNodes = dynamic_cast<FreeNodes*>(theFunctor.get()))
 557   {
 558     myExtractUnstructuredGrid->SetModeOfChanging(VTKViewer_ExtractUnstructuredGrid::eAdding);
 559     vtkIdType aNbNodes = myVisualObj->GetNbEntities(SMDSAbs_Node);
 560     for( vtkIdType i = 0; i < aNbNodes; i++ ){
 561       vtkIdType anObjId = myVisualObj->GetNodeObjId(i);
 562       if(aFreeNodes->IsSatisfy(anObjId))
 563         myExtractUnstructuredGrid->RegisterCell(i);
 564     }
 565     if(!myExtractUnstructuredGrid->IsCellsRegistered())
 566       myExtractUnstructuredGrid->RegisterCell(-1);
 567     SetUnstructuredGrid(myVisualObj->GetUnstructuredGrid());
 568   }
 569 }
 570
 571
 572
 573
 574 unsigned long int
 575 SMESH_DeviceActor
 576 ::GetMTime()
 577 {
 578   unsigned long mTime = this->Superclass::GetMTime();
 579   mTime = max(mTime,myExtractGeometry->GetMTime());
 580   mTime = max(mTime,myExtractUnstructuredGrid->GetMTime());
 581   mTime = max(mTime,myMergeFilter->GetMTime());
 582   mTime = max(mTime,myGeomFilter->GetMTime());
 583   mTime = max(mTime,myTransformFilter->GetMTime());
 584   mTime = max(mTime,myFaceOrientationFilter->GetMTime());
 585   return mTime;
 586 }
 587
 588
 589 void
 590 SMESH_DeviceActor
 591 ::SetTransform(VTKViewer_Transform* theTransform)
 592 {
 593   myTransformFilter->SetTransform(theTransform);
 594 }
 595
 596
 597 void
 598 SMESH_DeviceActor
 599 ::SetShrink()
 600 {
 601   if ( !myIsShrinkable ) return;
 602   if ( vtkDataSet* aDataSet = myPassFilter[ 0 ]->GetOutput() )
 603   {
 604     myShrinkFilter->SetInput( aDataSet );
 605     myPassFilter[ 1 ]->SetInput( myShrinkFilter->GetOutput() );
 606     myIsShrunk = true;
 607   }
 608 }
 609
 610 void
 611 SMESH_DeviceActor
 612 ::UnShrink()
 613 {
 614   if ( !myIsShrunk ) return;
 615   if ( vtkDataSet* aDataSet = myPassFilter[ 0 ]->GetOutput() )
 616   {
 617     myPassFilter[ 1 ]->SetInput( aDataSet );
 618     myPassFilter[ 1 ]->Modified();
 619     myIsShrunk = false;
 620     Modified();
 621   }
 622 }
 623
 624
 625 void
 626 SMESH_DeviceActor
 627 ::SetFacesOriented(bool theIsFacesOriented)
 628 {
 629   if ( vtkDataSet* aDataSet = myPassFilter[ 1 ]->GetOutput() )
 630   {
 631     myIsFacesOriented = theIsFacesOriented;
 632     if( theIsFacesOriented )
 633       myFaceOrientationFilter->SetInput( aDataSet );
 634     UpdateFaceOrientation();
 635   }
 636 }
 637
 638 void
 639 SMESH_DeviceActor
 640 ::SetFacesOrientationColor(vtkFloatingPointType theColor[3])
 641 {
 642   myFaceOrientation->GetProperty()->SetColor( theColor );
 643 }
 644
 645 void
 646 SMESH_DeviceActor
 647 ::GetFacesOrientationColor(vtkFloatingPointType theColor[3])
 648 {
 649   myFaceOrientation->GetProperty()->GetColor( theColor );
 650 }
 651
 652 void
 653 SMESH_DeviceActor
 654 ::SetFacesOrientationScale(vtkFloatingPointType theScale)
 655 {
 656   myFaceOrientationFilter->SetOrientationScale( theScale );
 657 }
 658
 659 vtkFloatingPointType
 660 SMESH_DeviceActor
 661 ::GetFacesOrientationScale()
 662 {
 663   return myFaceOrientationFilter->GetOrientationScale();
 664 }
 665
 666 void
 667 SMESH_DeviceActor
 668 ::SetFacesOrientation3DVectors(bool theState)
 669 {
 670   myFaceOrientationFilter->Set3dVectors( theState );
 671 }
 672
 673 bool
 674 SMESH_DeviceActor
 675 ::GetFacesOrientation3DVectors()
 676 {
 677   return myFaceOrientationFilter->Get3dVectors();
 678 }
 679
 680 void
 681 SMESH_DeviceActor
 682 ::UpdateFaceOrientation()
 683 {
 684   bool aShowFaceOrientation = myIsFacesOriented;
 685   aShowFaceOrientation &= GetVisibility();
 686   aShowFaceOrientation &= myRepresentation == eSurface;
 687   myFaceOrientation->SetVisibility(aShowFaceOrientation);
 688 }
 689
 690
 691 void
 692 SMESH_DeviceActor
 693 ::SetRepresentation(EReperesent theMode)
 694 {
 695   switch(theMode){
 696   case ePoint:
 697     myGeomFilter->SetInside(true);
 698     myGeomFilter->SetWireframeMode(false);
 699     GetProperty()->SetRepresentation(0);
 700     break;
 701   case eWireframe:
 702     myGeomFilter->SetInside(false);
 703     myGeomFilter->SetWireframeMode(true);
 704     GetProperty()->SetRepresentation(theMode);
 705     break;
 706   case eInsideframe:
 707     myGeomFilter->SetInside(true);
 708     myGeomFilter->SetWireframeMode(true);
 709     GetProperty()->SetRepresentation(1);
 710     break;
 711   case eSurface:
 712     myGeomFilter->SetInside(false);
 713     myGeomFilter->SetWireframeMode(false);
 714     GetProperty()->SetRepresentation(theMode);
 715   }
 716   SetMarkerEnabled(theMode == ePoint);
 717   myRepresentation = theMode;
 718   UpdateFaceOrientation();
 719   GetProperty()->Modified();
 720   myMapper->Modified();
 721   Modified();
 722 }
 723
 724
 725 void
 726 SMESH_DeviceActor
 727 ::SetVisibility(int theMode)
 728 {
 729   if(!myExtractUnstructuredGrid->GetInput() ||
 730      GetUnstructuredGrid()->GetNumberOfCells())
 731   {
 732     vtkLODActor::SetVisibility(theMode);
 733   }else{
 734     vtkLODActor::SetVisibility(false);
 735   }
 736   UpdateFaceOrientation();
 737 }
 738
 739
 740 int
 741 SMESH_DeviceActor
 742 ::GetVisibility()
 743 {
 744   if(!GetUnstructuredGrid()->GetNumberOfCells()){
 745     vtkLODActor::SetVisibility(false);
 746   }
 747   return vtkLODActor::GetVisibility();
 748 }
 749
 750
 751 void
 752 SMESH_DeviceActor
 753 ::AddToRender(vtkRenderer* theRenderer)
 754 {
 755   theRenderer->AddActor(this);
 756   theRenderer->AddActor(myFaceOrientation);
 757 }
 758
 759 void
 760 SMESH_DeviceActor
 761 ::RemoveFromRender(vtkRenderer* theRenderer)
 762 {
 763   theRenderer->RemoveActor(this);
 764   theRenderer->RemoveActor(myFaceOrientation);
 765 }
 766
 767
 768 int
 769 SMESH_DeviceActor
 770 ::GetNodeObjId(int theVtkID)
 771 {
 772   vtkIdType anID = theVtkID;
 773
 774   if(IsImplicitFunctionUsed())
 775     anID = myExtractGeometry->GetNodeObjId(theVtkID);
 776
 777   vtkIdType aRetID = myVisualObj->GetNodeObjId(anID);
 778   if(MYDEBUG) MESSAGE("GetNodeObjId - theVtkID = "<<theVtkID<<"; anID = "<<anID<<"; aRetID = "<<aRetID);
 779   return aRetID;
 780 }
 781
 782 vtkFloatingPointType*
 783 SMESH_DeviceActor
 784 ::GetNodeCoord(int theObjID)
 785 {
 786   vtkDataSet* aDataSet = myMergeFilter->GetOutput();
 787   vtkIdType anID = myVisualObj->GetNodeVTKId(theObjID);
 788   vtkFloatingPointType* aCoord = (anID >=0) ? aDataSet->GetPoint(anID) : NULL;
 789   if(MYDEBUG) MESSAGE("GetNodeCoord - theObjID = "<<theObjID<<"; anID = "<<anID);
 790   return aCoord;
 791 }
 792
 793
 794 int
 795 SMESH_DeviceActor
 796 ::GetElemObjId(int theVtkID)
 797 {
 798   vtkIdType anId = myGeomFilter->GetElemObjId(theVtkID);
 799   if(anId < 0)
 800     return -1;
 801
 802   vtkIdType anId2 = anId;
 803   if(IsImplicitFunctionUsed())
 804     anId2 = myExtractGeometry->GetElemObjId(anId);
 805   if(anId2 < 0)
 806     return -1;
 807
 808   vtkIdType anId3 = myExtractUnstructuredGrid->GetInputId(anId2);
 809   if(anId3 < 0)
 810     return -1;
 811
 812   vtkIdType aRetID = myVisualObj->GetElemObjId(anId3);
 813   if(MYDEBUG)
 814      MESSAGE("GetElemObjId - theVtkID = "<<theVtkID<<"; anId2 = "<<anId2<<"; anId3 = "<<anId3<<"; aRetID = "<<aRetID);
 815   return aRetID;
 816 }
 817
 818 vtkCell*
 819 SMESH_DeviceActor
 820 ::GetElemCell(int theObjID)
 821 {
 822   vtkDataSet* aDataSet = myVisualObj->GetUnstructuredGrid();
 823   vtkIdType aGridID = myVisualObj->GetElemVTKId(theObjID);
 824   vtkCell* aCell = (aGridID >= 0 ) ? aDataSet->GetCell(aGridID) : NULL;
 825   if(MYDEBUG)
 826     MESSAGE("GetElemCell - theObjID = "<<theObjID<<"; aGridID = "<<aGridID);
 827   return aCell;
 828 }
 829
 830
 831 vtkFloatingPointType
 832 SMESH_DeviceActor
 833 ::GetShrinkFactor()
 834 {
 835   return myShrinkFilter->GetShrinkFactor();
 836 }
 837
 838 void
 839 SMESH_DeviceActor
 840 ::SetShrinkFactor(vtkFloatingPointType theValue)
 841 {
 842   theValue = theValue > 0.1? theValue: 0.8;
 843   myShrinkFilter->SetShrinkFactor(theValue);
 844   Modified();
 845 }
 846
 847
 848 void
 849 SMESH_DeviceActor
 850 ::SetHighlited(bool theIsHighlited)
 851 {
 852   if ( myIsHighlited == theIsHighlited )
 853     return;
 854   myIsHighlited = theIsHighlited;
 855   Modified();
 856 }
 857
 858 void
 859 SMESH_DeviceActor
 860 ::Render(vtkRenderer *ren, vtkMapper* m)
 861 {
 862   int aResolveCoincidentTopology = vtkMapper::GetResolveCoincidentTopology();
 863   vtkFloatingPointType aStoredFactor, aStoredUnit;
 864   vtkMapper::GetResolveCoincidentTopologyPolygonOffsetParameters(aStoredFactor,aStoredUnit);
 865
 866   vtkMapper::SetResolveCoincidentTopologyToPolygonOffset();
 867   vtkFloatingPointType aFactor = myPolygonOffsetFactor, aUnits = myPolygonOffsetUnits;
 868   if(myIsHighlited){
 869     static vtkFloatingPointType EPS = .01;
 870     aUnits *= (1.0-EPS);
 871   }
 872   vtkMapper::SetResolveCoincidentTopologyPolygonOffsetParameters(aFactor,aUnits);
 873   vtkLODActor::Render(ren,m);
 874
 875   vtkMapper::SetResolveCoincidentTopologyPolygonOffsetParameters(aStoredFactor,aStoredUnit);
 876   vtkMapper::SetResolveCoincidentTopology(aResolveCoincidentTopology);
 877 }
 878
 879
 880 void
 881 SMESH_DeviceActor
 882 ::SetPolygonOffsetParameters(vtkFloatingPointType factor,
 883                              vtkFloatingPointType units)
 884 {
 885   myPolygonOffsetFactor = factor;
 886   myPolygonOffsetUnits = units;
 887 }
 888
 889 /*!
 890  * On/Off representation 2D quadratic element as arked polygon
 891  */
 892 void SMESH_DeviceActor::SetQuadraticArcMode(bool theFlag){
 893   myGeomFilter->SetQuadraticArcMode(theFlag);
 894 }
 895
 896 /*!
 897  * Return true if 2D quadratic element displayed as arked polygon
 898  */
 899 bool SMESH_DeviceActor::GetQuadraticArcMode(){
 900   return myGeomFilter->GetQuadraticArcMode();
 901 }
 902 /*!
 903  * Set Max angle for representation 2D quadratic element as arked polygon
 904  */
 905 void SMESH_DeviceActor::SetQuadraticArcAngle(vtkFloatingPointType theMaxAngle){
 906   myGeomFilter->SetQuadraticArcAngle(theMaxAngle);
 907 }
 908
 909 /*!
 910  * Return Max angle of the representation 2D quadratic element as arked polygon
 911  */
 912 vtkFloatingPointType SMESH_DeviceActor::GetQuadraticArcAngle(){
 913   return myGeomFilter->GetQuadraticArcAngle();
 914 }
 915
 916 /*!
 917  * Set point marker enabled
 918  * \param theMarkerEnabled flag to enable/disable point marker
 919  */
 920 void SMESH_DeviceActor::SetMarkerEnabled( bool theMarkerEnabled )
 921 {
 922   myMapper->SetMarkerEnabled( theMarkerEnabled );
 923 }
 924
 925 /*!
 926  * Set standard point marker
 927  * \param theMarkerType type of the marker
 928  */
 929 void SMESH_DeviceActor::SetMarkerStd( VTK::MarkerType theMarkerType, VTK::MarkerScale theMarkerScale )
 930 {
 931   myMapper->SetMarkerStd( theMarkerType, theMarkerScale );
 932 }
 933
 934 /*!
 935  * Set custom point marker
 936  * \param theMarkerId id of the marker texture
 937  * \param theMarkerTexture marker texture
 938  */
 939 void SMESH_DeviceActor::SetMarkerTexture( int theMarkerId, VTK::MarkerTexture theMarkerTexture )
 940 {
 941   myMapper->SetMarkerTexture( theMarkerId, theMarkerTexture );
 942 }
 943
 944 /*!
 945  * Get type of the point marker
 946  * \return type of the point marker
 947  */
 948 VTK::MarkerType SMESH_DeviceActor::GetMarkerType()
 949 {
 950   return myMapper->GetMarkerType();
 951 }
 952
 953 /*!
 954   Get scale of the point marker
 955   \return scale of the point marker
 956 */
 957 VTK::MarkerScale SMESH_DeviceActor::GetMarkerScale()
 958 {
 959   return myMapper->GetMarkerScale();
 960 }
 961
 962 /*!
 963  * Get texture identifier of the point marker
 964  * \return texture identifier of the point marker
 965  */
 966 int SMESH_DeviceActor::GetMarkerTexture()
 967 {
 968   return myMapper->GetMarkerTexture();
 969 }