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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestProStarReader.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.
=========================================================================*/
// .NAME Test of vtkProStarReader
// .SECTION Description
//
#include "vtkProStarReader.h"
#include "vtkDebugLeaks.h"
#include "vtkActor.h"
#include "vtkCamera.h"
#include "vtkGeometryFilter.h"
#include "vtkIdList.h"
#include "vtkPolyDataMapper.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRegressionTestImage.h"
#include "vtkSmartPointer.h"
#include "vtkTestUtilities.h"
#include "vtkUnstructuredGrid.h"
#include "vtkWindowToImageFilter.h"
#include "vtkPNGWriter.h"
int TestProStarReader( int argc, char *argv[] )
{
// Read file name.
char* fname = vtkTestUtilities::ExpandDataFileName(argc, argv, "Data/prostar.vrt");
// Create the reader.
vtkSmartPointer<vtkProStarReader> reader = vtkSmartPointer<vtkProStarReader>::New();
reader->SetFileName(fname);
reader->Update();
delete [] fname;
vtkUnstructuredGrid* grid = vtkUnstructuredGrid::SafeDownCast(reader->GetOutput());
if(grid->GetNumberOfPoints() != 44)
{
vtkGenericWarningMacro("Input grid has " << grid->GetNumberOfPoints()
<< " but should have 44.");
return 1;
}
if(grid->GetNumberOfCells() != 10)
{
vtkGenericWarningMacro("Input grid has " << grid->GetNumberOfCells()
<< " but should have 10.");
return 1;
}
// There are render issues with 7 and 9 so we ignore those
// for the tests.
vtkUnstructuredGrid* newGrid = vtkUnstructuredGrid::New();
newGrid->SetPoints(grid->GetPoints());
newGrid->Allocate(8);
vtkIdList* cellIds = vtkIdList::New();
for(vtkIdType i=0;i<grid->GetNumberOfCells();i++)
{
if(i != 8 && i != 9)
{
grid->GetCellPoints(i, cellIds);
newGrid->InsertNextCell(grid->GetCellType(i), cellIds);
}
}
cellIds->Delete();
// Convert to PolyData.
vtkGeometryFilter* geometryFilter = vtkGeometryFilter::New();
geometryFilter->SetInputData(newGrid);
newGrid->Delete();
// Create a mapper.
vtkPolyDataMapper* mapper = vtkPolyDataMapper::New();
mapper->SetInputConnection(geometryFilter->GetOutputPort());
mapper->ScalarVisibilityOn();
// Create the actor.
vtkActor* actor = vtkActor::New();
actor->SetMapper(mapper);
// Basic visualisation.
vtkRenderWindow* renWin = vtkRenderWindow::New();
vtkRenderer* ren = vtkRenderer::New();
renWin->AddRenderer(ren);
vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New();
iren->SetRenderWindow(renWin);
ren->AddActor(actor);
ren->SetBackground(0,0,0);
renWin->SetSize(300,300);
// interact with data
renWin->Render();
int retVal = vtkRegressionTestImage( renWin );
if ( retVal == vtkRegressionTester::DO_INTERACTOR)
{
iren->Start();
}
actor->Delete();
mapper->Delete();
geometryFilter->Delete();
renWin->Delete();
ren->Delete();
iren->Delete();
return !retVal;
}
|