Cloned library of VTK-5.0.0 with extra build files for internal package management.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

107 lines
3.0 KiB

2 years ago
/*=========================================================================
Program: Visualization Toolkit
Module: $RCSfile: task2.cxx,v $
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 "TaskParallelism.h"
#include "vtkImageData.h"
#include "vtkPolyDataMapper.h"
// Task 2 for TaskParallelism.
// See TaskParallelism.cxx for more information.
vtkPolyDataMapper* task2(vtkRenderWindow* renWin, double data,
vtkCamera* cam)
{
double extent = data;
int iextent = static_cast<int>(data);
// The pipeline
// Synthetic image source.
vtkRTAnalyticSource* source1 = vtkRTAnalyticSource::New();
source1->SetWholeExtent (-1*iextent, iextent, -1*iextent, iextent,
-1*iextent, iextent );
source1->SetCenter(0, 0, 0);
source1->SetStandardDeviation( 0.5 );
source1->SetMaximum( 255.0 );
source1->SetXFreq( 60 );
source1->SetXMag( 10 );
source1->SetYFreq( 30 );
source1->SetYMag( 18 );
source1->SetZFreq( 40 );
source1->SetZMag( 5 );
source1->GetOutput()->SetSpacing(2.0/extent,2.0/extent,2.0/extent);
// Gradient vector.
vtkImageGradient* grad = vtkImageGradient::New();
grad->SetDimensionality( 3 );
grad->SetInputConnection(source1->GetOutputPort());
vtkImageShrink3D* mask = vtkImageShrink3D::New();
mask->SetInputConnection(grad->GetOutputPort());
mask->SetShrinkFactors(5, 5, 5);
// Label the scalar field as the active vectors.
vtkAssignAttribute* aa = vtkAssignAttribute::New();
aa->SetInputConnection(mask->GetOutputPort());
aa->Assign(vtkDataSetAttributes::SCALARS, vtkDataSetAttributes::VECTORS,
vtkAssignAttribute::POINT_DATA);
vtkGlyphSource2D* arrow = vtkGlyphSource2D::New();
arrow->SetGlyphTypeToArrow();
arrow->SetScale(0.2);
arrow->FilledOff();
// Glyph the gradient vector (with arrows)
vtkGlyph3D* glyph = vtkGlyph3D::New();
glyph->SetInputConnection(aa->GetOutputPort());
glyph->SetSource(arrow->GetOutput());
glyph->ScalingOff();
glyph->OrientOn();
glyph->SetVectorModeToUseVector();
glyph->SetColorModeToColorByVector();
// Rendering objects.
vtkPolyDataMapper* mapper = vtkPolyDataMapper::New();
mapper->SetInputConnection(glyph->GetOutputPort());
mapper->SetScalarRange(50, 180);
mapper->ImmediateModeRenderingOn();
vtkActor* actor = vtkActor::New();
actor->SetMapper(mapper);
vtkRenderer* ren = vtkRenderer::New();
renWin->AddRenderer(ren);
ren->AddActor(actor);
ren->SetActiveCamera( cam );
// Cleanup
source1->Delete();
grad->Delete();
aa->Delete();
mask->Delete();
glyph->Delete();
arrow->Delete();
actor->Delete();
ren->Delete();
return mapper;
}