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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
// .NAME Test of vtkTableToMySQLWriter and vtkMySQLToTableReader
// .SECTION Description
//
#include "vtkMySQLDatabase.h"
#include "vtkSQLQuery.h"
#include "vtkSmartPointer.h"
#include "vtkTable.h"
#include "vtkTableReader.h"
#include "vtkTableWriter.h"
#include "vtksys/SystemTools.hxx"
#include "vtkIOMySQLTestingCxxConfigure.h"
#include "vtkMySQLToTableReader.h"
#include "vtkTableToMySQLWriter.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;
}
|
