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: TestMySQLTableReadWrite.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 vtkTableToMySQLWriter and vtkMySQLToTableReader
// .SECTION Description
//
#include "vtksys/SystemTools.hxx"
#include "vtkSmartPointer.h"
#include "vtkMySQLDatabase.h"
#include "vtkSQLQuery.h"
#include "vtkTable.h"
#include "vtkTableReader.h"
#include "vtkTableWriter.h"
#include "vtkToolkits.h"
#include "vtkTableToMySQLWriter.h"
#include "vtkMySQLToTableReader.h"
#include "vtkIOMySQLTestingCxxConfigure.h"
int TestMySQLTableReadWrite(int argc, char *argv[])
{
if ( argc <= 1 )
{
cerr << "Usage: " << argv[0] << " <.vtk table file>" << endl;
return 1;
}
cerr << "reading a vtkTable from file" << endl;
vtkSmartPointer<vtkTableReader> tableFileReader =
vtkSmartPointer<vtkTableReader>::New();
tableFileReader->SetFileName(argv[1]);
vtkTable *table = tableFileReader->GetOutput();
tableFileReader->Update();
cerr << "opening a MySQL database connection" << endl;
vtkMySQLDatabase* db = vtkMySQLDatabase::SafeDownCast(
vtkSQLDatabase::CreateFromURL( VTK_MYSQL_TEST_URL ) );
bool status = db->Open();
if ( ! status )
{
cerr << "Couldn't open database.\n";
return 1;
}
cerr << "creating a MySQL table from a vtkTable" << endl;
vtkSmartPointer<vtkTableToMySQLWriter> writerToTest =
vtkSmartPointer<vtkTableToMySQLWriter>::New();
writerToTest->SetInputData(table);
writerToTest->SetDatabase(db);
writerToTest->SetTableName("tableTest");
writerToTest->Update();
cerr << "converting it back to a vtkTable" << endl;
vtkSmartPointer<vtkMySQLToTableReader> readerToTest =
vtkSmartPointer<vtkMySQLToTableReader>::New();
readerToTest->SetDatabase(db);
readerToTest->SetTableName("tableTest");
readerToTest->Update();
cerr << "writing the table out to disk" << endl;
vtkSmartPointer<vtkTableWriter> tableFileWriter =
vtkSmartPointer<vtkTableWriter>::New();
tableFileWriter->SetFileName("TestMySQLTableReadWrite.vtk");
tableFileWriter->SetInputConnection(readerToTest->GetOutputPort());
tableFileWriter->Update();
cerr << "verifying that it's the same as what we started with...";
int result = 0;
if(vtksys::SystemTools::FilesDiffer(argv[1], "TestMySQLTableReadWrite.vtk"))
{
cerr << "it's not." << endl;
result = 1;
}
else
{
cerr << "it is!" << endl;
}
//drop the table we created
vtkSQLQuery* query = db->GetQueryInstance();
query->SetQuery("DROP TABLE tableTest");
query->Execute();
//clean up memory
db->Delete();
query->Delete();
return result;
}
|
