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 127 128
|
/*=========================================================================
Program: Visualization Toolkit
Module: vtkSectorSource.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 "vtkSectorSource.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPolyData.h"
//#include "vtkDiskSource.h"
#include "vtkLineSource.h"
#include "vtkRotationalExtrusionFilter.h"
#include "vtkMath.h"
#include "vtkStreamingDemandDrivenPipeline.h"
#include "vtkSmartPointer.h"
#define VTK_CREATE(type, name) \
vtkSmartPointer<type> name = vtkSmartPointer<type>::New()
vtkStandardNewMacro(vtkSectorSource);
vtkSectorSource::vtkSectorSource()
{
this->InnerRadius = 1.0;
this->OuterRadius = 2.0;
this->ZCoord = 0.0;
this->StartAngle = 0.0;
this->EndAngle = 90.0;
this->RadialResolution = 1;
this->CircumferentialResolution = 6;
this->SetNumberOfInputPorts(0);
}
int vtkSectorSource::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()));
int piece, numPieces;
piece = outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_PIECE_NUMBER());
numPieces = outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_NUMBER_OF_PIECES());
outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_NUMBER_OF_GHOST_LEVELS());
// if( (this->StartAngle == 0. && this->EndAngle == 360.) ||
// (this->StartAngle == 360. && this->EndAngle == 0. ) )
// {
// //use vtkDiskSource
// VTK_CREATE(vtkDiskSource, diskSource );
// diskSource->SetCircumferentialResolution( this->CircumferentialResolution );
// diskSource->SetRadialResolution( this->RadialResolution );
// diskSource->SetInnerRadius( this->InnerRadius );
// diskSource->SetOuterRadius( this->OuterRadius );
// if (output->GetUpdatePiece() == 0 && numPieces > 0)
// {
// diskSource->Update();
// output->ShallowCopy(diskSource->GetOutput());
// }
// output->SetUpdatePiece(piece);
// output->SetUpdateNumberOfPieces(numPieces);
// output->SetUpdateGhostLevel(ghostLevel);
// }
// else
// {
VTK_CREATE(vtkLineSource, lineSource);
lineSource->SetResolution( this->RadialResolution );
//set vertex 1, adjust for start angle
//set vertex 2, adjust for start angle
double x1[3], x2[3];
x1[0] = this->InnerRadius * cos( vtkMath::RadiansFromDegrees( this->StartAngle ) );
x1[1] = this->InnerRadius * sin( vtkMath::RadiansFromDegrees( this->StartAngle ) );
x1[2] = this->ZCoord;
x2[0] = this->OuterRadius * cos( vtkMath::RadiansFromDegrees( this->StartAngle ) );
x2[1] = this->OuterRadius * sin( vtkMath::RadiansFromDegrees( this->StartAngle ) );
x2[2] = this->ZCoord;
lineSource->SetPoint1(x1);
lineSource->SetPoint2(x2);
lineSource->Update();
VTK_CREATE(vtkRotationalExtrusionFilter, rotateFilter);
rotateFilter->SetResolution( this->CircumferentialResolution );
rotateFilter->SetInputConnection(lineSource->GetOutputPort());
rotateFilter->SetAngle( this->EndAngle - this->StartAngle );
if (piece == 0 && numPieces > 0)
{
rotateFilter->Update();
output->ShallowCopy(rotateFilter->GetOutput());
}
// }
return 1;
}
void vtkSectorSource::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os,indent);
os << indent << "InnerRadius: " << this->InnerRadius << "\n";
os << indent << "OuterRadius: " << this->OuterRadius << "\n";
os << indent << "ZCoord: " << this->ZCoord << "\n";
os << indent << "StartAngle: " << this->StartAngle << "\n";
os << indent << "EndAngle: " << this->EndAngle << "\n";;
os << indent << "CircumferentialResolution: " << this->CircumferentialResolution << "\n";
os << indent << "RadialResolution: " << this->RadialResolution << "\n";
}
|