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
93
94
95
96
97
98
99
100
101
102
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestDistancePolyDataFilter.cxx
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#include "vtkActor.h"
#include "vtkArrayCalculator.h"
#include "vtkDataSetMapper.h"
#include "vtkMath.h"
#include "vtkNew.h"
#include "vtkProperty.h"
#include "vtkPolyData.h"
#include "vtkPoints.h"
#include "vtkPointSource.h"
#include "vtkRegressionTestImage.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRTAnalyticSource.h"
#include "vtkStreamTracer.h"
#include "vtkWarpScalar.h"
int TestStreamTracerSurface(int argc, char* argv[])
{
vtkNew<vtkRTAnalyticSource> wavelet;
wavelet->SetWholeExtent(-10, 100, -10, 100, 0, 0);
vtkNew<vtkWarpScalar> warp;
warp->SetScaleFactor(0.1);
warp->SetInputConnection(wavelet->GetOutputPort());
vtkNew<vtkArrayCalculator> calc;
calc->AddScalarArrayName("RTData");
calc->SetFunction("abs(RTData)*iHat + abs(RTData)*jHat");
calc->SetInputConnection(warp->GetOutputPort());
calc->Update();
vtkNew<vtkPoints> points;
vtkDataSet* calcData = calc->GetOutput();
vtkIdType nLine = static_cast<vtkIdType>(sqrt(static_cast<double>(calcData->GetNumberOfPoints())));
for (vtkIdType i = 0; i < nLine; i += 10)
{
points->InsertNextPoint(calcData->GetPoint(i * (nLine - 1) + nLine));
}
vtkNew<vtkPolyData> pointsPolydata;
pointsPolydata->SetPoints(points.Get());
vtkNew<vtkStreamTracer> stream;
stream->SurfaceStreamlinesOn();
stream->SetMaximumPropagation(210);
stream->SetIntegrationDirection(vtkStreamTracer::BOTH);
stream->SetInputConnection(calc->GetOutputPort());
stream->SetSourceData(pointsPolydata.Get());
vtkNew<vtkDataSetMapper> streamMapper;
streamMapper->SetInputConnection(stream->GetOutputPort());
streamMapper->ScalarVisibilityOff();
vtkNew<vtkDataSetMapper> surfaceMapper;
surfaceMapper->SetInputConnection(calc->GetOutputPort());
vtkNew<vtkActor> streamActor;
streamActor->SetMapper(streamMapper.Get());
streamActor->GetProperty()->SetColor(1, 1, 1);
streamActor->GetProperty()->SetLineWidth(4.);
streamActor->SetPosition(0, 0, 1);
vtkNew<vtkActor> surfaceActor;
surfaceActor->SetMapper(surfaceMapper.Get());
surfaceActor->GetProperty()->SetRepresentationToSurface();
vtkNew<vtkRenderer> renderer;
renderer->AddActor(surfaceActor.GetPointer());
renderer->AddActor(streamActor.GetPointer());
renderer->ResetCamera();
renderer->SetBackground(1., 1., 1.);
vtkNew<vtkRenderWindow> renWin;
renWin->AddRenderer(renderer.GetPointer());
renWin->SetMultiSamples(0);
renWin->SetSize(300, 300);
vtkNew<vtkRenderWindowInteractor> iren;
iren->SetRenderWindow(renWin.GetPointer());
int retVal = vtkRegressionTestImage(renWin.GetPointer());
if (retVal == vtkRegressionTester::DO_INTERACTOR)
{
iren->Start();
}
return !retVal;
}
|
