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
122
123
124
125
|
#include "vtkSMEnsembleDataReaderProxy.h"
#include "vtkClientServerStream.h"
#include "vtkNew.h"
#include "vtkObjectFactory.h"
#include "vtkPVEnsembleDataReaderInformation.h"
#include "vtkSmartPointer.h"
#include "vtkSMCoreUtilities.h"
#include "vtkSMPropertyHelper.h"
#include "vtkSMProxyManager.h"
#include "vtkSMReaderFactory.h"
#include "vtkSMSession.h"
#include "vtkSMSessionProxyManager.h"
#include <cassert>
#include <vector>
vtkStandardNewMacro(vtkSMEnsembleDataReaderProxy);
//-----------------------------------------------------------------------------
vtkSMEnsembleDataReaderProxy::vtkSMEnsembleDataReaderProxy()
{
this->FileNamePotentiallyModified = false;
}
//-----------------------------------------------------------------------------
vtkSMEnsembleDataReaderProxy::~vtkSMEnsembleDataReaderProxy()
{
}
//-----------------------------------------------------------------------------
void vtkSMEnsembleDataReaderProxy::SetPropertyModifiedFlag(
const char *name, int flag)
{
this->Superclass::SetPropertyModifiedFlag(name, flag);
if (strcmp(name, "FileName") == 0)
{
this->FileNamePotentiallyModified = true;
}
}
//-----------------------------------------------------------------------------
void vtkSMEnsembleDataReaderProxy::UpdateVTKObjects()
{
this->Superclass::UpdateVTKObjects();
if (this->FileNamePotentiallyModified)
{
this->FetchFileNames();
this->FileNamePotentiallyModified = false;
}
}
//-----------------------------------------------------------------------------
bool vtkSMEnsembleDataReaderProxy::FetchFileNames()
{
// Gather information
vtkNew<vtkPVEnsembleDataReaderInformation> info;
assert(info.GetPointer());
this->GatherInformation(info.GetPointer());
// Create reader factory
vtkSMReaderFactory *readerFactory = vtkSMProxyManager::GetProxyManager()->GetReaderFactory();
if (!readerFactory || readerFactory->GetNumberOfRegisteredPrototypes() == 0)
{
vtkErrorMacro("No reader factory found. Cannot determine reader types!!!");
return false;
}
if (info->GetFileCount() == 0)
{
return false;
}
// Get session and session proxy manager
vtkSMSession *session = this->GetSession();
assert(session);
vtkSMSessionProxyManager *spm = session->GetSessionProxyManager();
assert(spm);
// Stream reader proxies to VTK object
std::vector<vtkSmartPointer<vtkSMProxy> > proxies;
vtkClientServerStream stream;
stream << vtkClientServerStream::Invoke
<< VTKOBJECT(this)
<< "ResetReaders"
<< vtkClientServerStream::End;
for (unsigned int i = 0, max = info->GetFileCount(); i < max; ++i)
{
vtkStdString filePath = info->GetFilePath(i);
if (readerFactory->CanReadFile(filePath.c_str(), session))
{
// Create reader proxy
const char *readerGroup = readerFactory->GetReaderGroup();
const char *readerName = readerFactory->GetReaderName();
vtkSMProxy *proxy = spm->NewProxy(readerGroup, readerName);
if (!proxy)
{
return false;
}
const char *fileNameProperty = vtkSMCoreUtilities::GetFileNameProperty(proxy);
assert(fileNameProperty);
vtkSMPropertyHelper(proxy, fileNameProperty).Set(filePath);
proxy->UpdateVTKObjects();
// Push to stream
stream << vtkClientServerStream::Invoke
<< VTKOBJECT(this) << "SetReader" << i << VTKOBJECT(proxy)
<< vtkClientServerStream::End;
proxies.push_back(proxy);
proxy->Delete();
}
else
{
vtkErrorMacro("Cannot create a reader for: " << filePath.c_str());
}
}
this->ExecuteStream(stream);
return true;
}
//-----------------------------------------------------------------------------
void vtkSMEnsembleDataReaderProxy::PrintSelf(ostream &os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
|
