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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/*=========================================================================
Program: Visualization Toolkit
Module: vtkDiskSource.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 "vtkDiskSource.h"
#include "vtkCellArray.h"
#include "vtkMath.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPoints.h"
#include "vtkPolyData.h"
vtkStandardNewMacro(vtkDiskSource);
vtkDiskSource::vtkDiskSource()
{
this->InnerRadius = 0.25;
this->OuterRadius = 0.5;
this->RadialResolution = 1;
this->CircumferentialResolution = 6;
this->SetNumberOfInputPorts(0);
}
int vtkDiskSource::RequestData(
vtkInformation *vtkNotUsed(request),
vtkInformationVector **vtkNotUsed(inputVector),
vtkInformationVector *outputVector)
{
// get the info object
vtkInformation *outInfo = outputVector->GetInformationObject(0);
// get the ouptut
vtkPolyData *output = vtkPolyData::SafeDownCast(
outInfo->Get(vtkDataObject::DATA_OBJECT()));
vtkIdType numPolys, numPts;
double x[3];
int i, j;
vtkIdType pts[4];
double theta, deltaRadius;
double cosTheta, sinTheta;
vtkPoints *newPoints;
vtkCellArray *newPolys;
// Set things up; allocate memory
//
numPts = (this->RadialResolution + 1) *
(this->CircumferentialResolution + 1);
numPolys = this->RadialResolution * this->CircumferentialResolution;
newPoints = vtkPoints::New();
newPoints->Allocate(numPts);
newPolys = vtkCellArray::New();
newPolys->Allocate(newPolys->EstimateSize(numPolys,4));
// Create disk
//
theta = 2.0 * vtkMath::Pi() / this->CircumferentialResolution;
deltaRadius = (this->OuterRadius - this->InnerRadius)/this->RadialResolution;
for (i=0; i < this->CircumferentialResolution; i++)
{
cosTheta = cos(i*theta);
sinTheta = sin(i*theta);
for (j=0; j <= this->RadialResolution; j++)
{
x[0] = (this->InnerRadius + j*deltaRadius) * cosTheta;
x[1] = (this->InnerRadius + j*deltaRadius) * sinTheta;
x[2] = 0.0;
newPoints->InsertNextPoint(x);
}
}
// Create connectivity
//
for (i=0; i < this->CircumferentialResolution; i++)
{
for (j=0; j < this->RadialResolution; j++)
{
pts[0] = i*(this->RadialResolution+1) + j;
pts[1] = pts[0] + 1;
if ( i < (this->CircumferentialResolution-1) )
{
pts[2] = pts[1] + this->RadialResolution + 1;
}
else
{
pts[2] = j + 1;
}
pts[3] = pts[2] - 1;
newPolys->InsertNextCell(4,pts);
}
}
// Update ourselves and release memory
//
output->SetPoints(newPoints);
newPoints->Delete();
output->SetPolys(newPolys);
newPolys->Delete();
return 1;
}
void vtkDiskSource::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os,indent);
os << indent << "InnerRadius: " << this->InnerRadius << "\n";
os << indent << "OuterRadius: " << this->OuterRadius << "\n";
os << indent << "RadialResolution: " << this->RadialResolution << "\n";
os << indent << "CircumferentialResolution: " << this->CircumferentialResolution << "\n";
}
|
