1 /*=========================================================================
2 
3   Program:   Visualization Toolkit
4   Module:    TestChartsOn3D.cxx
5 
6   Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
7   All rights reserved.
8   See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
9 
10      This software is distributed WITHOUT ANY WARRANTY; without even
11      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
12      PURPOSE.  See the above copyright notice for more information.
13 
14 =========================================================================*/
15 
16 #include "vtkActor.h"
17 #include "vtkChartXY.h"
18 #include "vtkContextScene.h"
19 #include "vtkContextActor.h"
20 #include "vtkCubeSource.h"
21 #include "vtkFloatArray.h"
22 #include "vtkPlotPoints.h"
23 #include "vtkPolyDataMapper.h"
24 #include "vtkProperty.h"
25 #include "vtkRenderer.h"
26 #include "vtkRenderWindow.h"
27 #include "vtkRenderWindowInteractor.h"
28 #include "vtkNew.h"
29 #include "vtkTable.h"
30 #include "vtkCamera.h"
31 
32 //----------------------------------------------------------------------------
TestChartsOn3D(int,char * [])33 int TestChartsOn3D(int , char * [])
34 {
35 
36   vtkNew<vtkRenderWindow> renwin;
37   renwin->SetMultiSamples(4);
38   renwin->SetSize(600, 400);
39 
40   vtkNew<vtkRenderWindowInteractor> iren;
41   iren->SetRenderWindow(renwin);
42 
43   vtkNew<vtkRenderer> renderer;
44   renderer->SetBackground(0.8, 0.8, 0.8);
45   renwin->AddRenderer(renderer);
46 
47   renderer->ResetCamera();
48   renderer->GetActiveCamera()->SetPosition(1.0, 1.0, -4.0);
49   renderer->GetActiveCamera()->Azimuth(40);
50 
51   // Cube Source 1
52   vtkNew<vtkCubeSource> cube;
53   vtkNew<vtkPolyDataMapper> cubeMapper;
54   vtkNew<vtkActor> cubeActor;
55 
56   cubeMapper->SetInputConnection(cube->GetOutputPort());
57   cubeActor->SetMapper(cubeMapper);
58   cubeActor->GetProperty()->SetColor(1.0, 0.0, 0.0);
59   renderer->AddActor(cubeActor);
60   cubeActor->GetProperty()->SetRepresentationToSurface();
61 
62   // Now the chart
63   vtkNew<vtkChartXY> chart;
64   vtkNew<vtkContextScene> chartScene;
65   vtkNew<vtkContextActor> chartActor;
66 
67   chart->SetAutoSize(false);
68   chart->SetSize(vtkRectf(0.0, 0.0, 300, 200));
69 
70   chartScene->AddItem(chart);
71   chartActor->SetScene(chartScene);
72 
73   //both needed
74   renderer->AddActor(chartActor);
75   chartScene->SetRenderer(renderer);
76 
77   // Create a table with some points in it...
78   vtkNew<vtkTable> table;
79   vtkNew<vtkFloatArray> arrX;
80   arrX->SetName("X Axis");
81   table->AddColumn(arrX);
82   vtkNew<vtkFloatArray> arrC;
83   arrC->SetName("Cosine");
84   table->AddColumn(arrC);
85   vtkNew<vtkFloatArray> arrS;
86   arrS->SetName("Sine");
87   table->AddColumn(arrS);
88   vtkNew<vtkFloatArray> arrT;
89   arrT->SetName("Tan");
90   table->AddColumn(arrT);
91   // Test charting with a few more points...
92   int numPoints = 69;
93   float inc = 7.5 / (numPoints-1);
94   table->SetNumberOfRows(numPoints);
95   table->SetNumberOfRows(numPoints);
96   for (int i = 0; i < numPoints; ++i)
97   {
98     table->SetValue(i, 0, i * inc);
99     table->SetValue(i, 1, cos(i * inc) + 0.0);
100     table->SetValue(i, 2, sin(i * inc) + 0.0);
101     table->SetValue(i, 3, tan(i * inc) + 0.5);
102   }
103 
104   // Add multiple line plots, setting the colors etc
105   vtkPlot *points = chart->AddPlot(vtkChart::POINTS);
106   points->SetInputData(table, 0, 1);
107   points->SetColor(0, 0, 0, 255);
108   points->SetWidth(1.0);
109   vtkPlotPoints::SafeDownCast(points)->SetMarkerStyle(vtkPlotPoints::CROSS);
110   points = chart->AddPlot(vtkChart::POINTS);
111   points->SetInputData(table, 0, 2);
112   points->SetColor(0, 0, 0, 255);
113   points->SetWidth(1.0);
114   vtkPlotPoints::SafeDownCast(points)->SetMarkerStyle(vtkPlotPoints::PLUS);
115   points = chart->AddPlot(vtkChart::POINTS);
116   points->SetInputData(table, 0, 3);
117   points->SetColor(0, 0, 255, 255);
118   points->SetWidth(4.0);
119 
120   renwin->SetMultiSamples(0);
121   iren->Initialize();
122   iren->Start();
123 
124   return EXIT_SUCCESS;
125 }
126