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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestMovieWriter.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 vtkMPEG2Writer
// .SECTION Description
//
#include "vtkImageCast.h"
#include "vtkImageData.h"
#include "vtkImageMandelbrotSource.h"
#include "vtkImageMapToColors.h"
#include "vtkLookupTable.h"
#include "vtkMPEG2Writer.h"
#include "vtksys/SystemTools.hxx"
int TestMovieWriter(int vtkNotUsed(argc), char* vtkNotUsed(argv)[])
{
int err = 0;
int cc = 0;
int exists = 0;
unsigned long length = 0;
vtkImageMandelbrotSource* Fractal0 = vtkImageMandelbrotSource::New();
Fractal0->SetWholeExtent( 0, 247, 0, 247, 0, 0 );
Fractal0->SetProjectionAxes( 0, 1, 2 );
Fractal0->SetOriginCX( -1.75, -1.25, 0, 0 );
Fractal0->SetSizeCX( 2.5, 2.5, 2, 1.5 );
Fractal0->SetMaximumNumberOfIterations( 100);
vtkImageCast* cast = vtkImageCast::New();
cast->SetInputConnection(Fractal0->GetOutputPort());
cast->SetOutputScalarTypeToUnsignedChar();
vtkLookupTable* table = vtkLookupTable::New();
table->SetTableRange(0, 100);
table->SetNumberOfColors(100);
table->Build();
table->SetTableValue(99, 0, 0, 0);
vtkImageMapToColors* colorize = vtkImageMapToColors::New();
colorize->SetOutputFormatToRGB();
colorize->SetLookupTable(table);
colorize->SetInputConnection(cast->GetOutputPort());
vtkMPEG2Writer *w = vtkMPEG2Writer::New();
w->SetInputConnection(colorize->GetOutputPort());
w->SetFileName("TestMovieWriter.mpg");
cout << "Writing file TestMovieWriter.mpg..." << endl;
w->Start();
for ( cc = 2; cc < 99; cc ++ )
{
cout << ".";
Fractal0->SetMaximumNumberOfIterations(cc);
table->SetTableRange(0, cc);
table->SetNumberOfColors(cc);
table->ForceBuild();
table->SetTableValue(cc-1, 0, 0, 0);
w->Write();
}
w->End();
cout << endl;
cout << "Done writing file TestMovieWriter.mpg..." << endl;
w->Delete();
exists = (int) vtksys::SystemTools::FileExists("TestMovieWriter.mpg");
length = vtksys::SystemTools::FileLength("TestMovieWriter.mpg");
cout << "TestMovieWriter.mpg file exists: " << exists << endl;
cout << "TestMovieWriter.mpg file length: " << length << endl;
if (!exists)
{
err = 1;
cerr << "ERROR: 1 - Test failing because TestMovieWriter.mpg file doesn't exist..." << endl;
}
if (0==length)
{
err = 2;
cerr << "ERROR: 2 - Test failing because TestMovieWriter.mpg file has zero length..." << endl;
}
colorize->Delete();
table->Delete();
cast->Delete();
Fractal0->Delete();
// err == 0 means test passes...
//
return err;
}
|
