vtkFrameRateWidget/Test2.cpp at master · daviddoria/vtkFrameRateWidget · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include <vtkSmartPointer.h>
#include <vtkMath.h>
#include <vtkPolyDataMapper.h>
#include <vtkActor.h>
#include <vtkRenderWindow.h>
#include <vtkRenderer.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkPolyData.h>
#include <vtkSphereSource.h>
#include <vtkGlyph3D.h>
#include <vtkPoints.h>
#include <iostream>
#include <sstream>

#include "vtkFrameRateWidget.h"

int main(int argc, char *argv[])
{
  unsigned nPoints = 0;
  // Check command line arguments
  if (argc !=2)
    {
    nPoints = 100;
    }
  else
    {
    std::istringstream(argv[1]) >> nPoints;
    }

  // Create the renderer
  vtkSmartPointer<vtkRenderer> renderer =
      vtkSmartPointer<vtkRenderer>::New();

  // Create some random points
  vtkSmartPointer<vtkPoints> points =
    vtkSmartPointer<vtkPoints>::New();
  for (unsigned i=0; i<nPoints; ++i)
  {
      points->InsertNextPoint(vtkMath::Random(0,nPoints),
                              vtkMath::Random(0,nPoints),
                              vtkMath::Random(0,nPoints));
  }

  vtkSmartPointer<vtkPolyData> data =
    vtkSmartPointer<vtkPolyData>::New();
  data->SetPoints(points);


  //
  // Add some spheres representing the atoms
  //
  vtkSmartPointer<vtkSphereSource> source =
    vtkSmartPointer<vtkSphereSource>::New();
  source->SetRadius(nPoints*0.01);
  vtkSmartPointer<vtkGlyph3D> glyph =
    vtkSmartPointer<vtkGlyph3D>::New();
  glyph->SetInput(data);
  glyph->SetSourceConnection(source->GetOutputPort());
  glyph->SetVectorModeToUseNormal();
  glyph->SetScaleModeToScaleByVector();
  glyph->SetColorModeToColorByScalar();

  vtkSmartPointer<vtkPolyDataMapper> mapper =
    vtkSmartPointer<vtkPolyDataMapper>::New();
  vtkSmartPointer<vtkActor> actor =
    vtkSmartPointer<vtkActor>::New();
  mapper->SetInputConnection(glyph->GetOutputPort());
  actor->SetMapper(mapper);
  renderer->AddActor(actor);

  vtkSmartPointer<vtkRenderWindow> renderWindow =
    vtkSmartPointer<vtkRenderWindow>::New();
  renderWindow->AddRenderer(renderer);

  vtkSmartPointer<vtkRenderWindowInteractor> interactor =
    vtkSmartPointer<vtkRenderWindowInteractor>::New();
  interactor->SetRenderWindow(renderWindow);

  vtkSmartPointer<vtkFrameRateWidget> frameRateWidget =
    vtkSmartPointer<vtkFrameRateWidget>::New();
  frameRateWidget->SetInteractor(interactor);
  frameRateWidget->SetRenderer(renderer);
  frameRateWidget->SelectableOff();
  frameRateWidget->Init();

  renderer->SetBackground(1,1,1); // Background color white
  renderWindow->Render();
  frameRateWidget->On();
  interactor->Start();

  return EXIT_SUCCESS;
}