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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestThresholdTable.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.
=========================================================================*/
/*-------------------------------------------------------------------------
Copyright 2008 Sandia Corporation.
Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
#include "vtkArrayPrint.h"
#include "vtkDenseArray.h"
#include "vtkDoubleArray.h"
#include "vtkIntArray.h"
#include "vtkStringArray.h"
#include "vtkTable.h"
#include "vtkTableToArray.h"
#include "vtkSmartPointer.h"
#include <stdexcept>
#define VTK_CREATE(type, name) \
vtkSmartPointer<type> name = vtkSmartPointer<type>::New()
#define test_expression(expression) \
{ \
if(!(expression)) \
throw std::runtime_error("Expression failed: " #expression); \
}
int TestTableToArray(int vtkNotUsed(argc), char* vtkNotUsed(argv)[])
{
try
{
VTK_CREATE(vtkTable, table);
VTK_CREATE(vtkIntArray, int_array);
int_array->SetName("A");
int_array->InsertNextValue(1);
int_array->InsertNextValue(2);
int_array->InsertNextValue(3);
int_array->InsertNextValue(4);
table->AddColumn(int_array);
VTK_CREATE(vtkDoubleArray, double_array);
double_array->SetName("B");
double_array->InsertNextValue(1.1);
double_array->InsertNextValue(1.2);
double_array->InsertNextValue(1.3);
double_array->InsertNextValue(1.4);
table->AddColumn(double_array);
VTK_CREATE(vtkStringArray, string_array);
string_array->SetName("C");
string_array->InsertNextValue("11");
string_array->InsertNextValue("12");
string_array->InsertNextValue("13");
string_array->InsertNextValue("14");
table->AddColumn(string_array);
VTK_CREATE(vtkTableToArray, table_to_array);
table_to_array->SetInputData(0, table);
table_to_array->AddColumn("C");
table_to_array->AddColumn(1);
table_to_array->AddColumn(static_cast<vtkIdType>(0));
table_to_array->AddAllColumns();
table_to_array->Update();
test_expression(table_to_array->GetOutput());
test_expression(1 == table_to_array->GetOutput()->GetNumberOfArrays());
vtkDenseArray<double>* const array = vtkDenseArray<double>::SafeDownCast(table_to_array->GetOutput()->GetArray(0));
test_expression(array);
test_expression(2 == array->GetDimensions());
test_expression(4 == array->GetExtent(0).GetSize());
test_expression(6 == array->GetExtent(1).GetSize());
test_expression(11 == array->GetValue(0, 0));
test_expression(1.1 == array->GetValue(0, 1));
test_expression(1 == array->GetValue(0, 2));
test_expression(1 == array->GetValue(0, 3));
test_expression(1.1 == array->GetValue(0, 4));
test_expression(11 == array->GetValue(0, 5));
test_expression(14 == array->GetValue(3, 0));
vtkPrintMatrixFormat(std::cout, array);
return 0;
}
catch(std::exception& e)
{
cerr << e.what() << endl;
return 1;
}
}
|
