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
|
/*=========================================================================
Program: ParaView
Module: vtkSMAnimationSceneGeometryWriter.cxx
Copyright (c) Kitware, Inc.
All rights reserved.
See Copyright.txt or http://www.paraview.org/HTML/Copyright.html 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.
=========================================================================*/
#include "vtkSMAnimationSceneGeometryWriter.h"
#include "vtkObjectFactory.h"
#include "vtkSMPropertyHelper.h"
#include "vtkSMProxy.h"
#include "vtkSMProxyManager.h"
#include "vtkSMSessionProxyManager.h"
#include "vtkSMProxyProperty.h"
#include "vtkSMRepresentationProxy.h"
#include "vtkSMStringVectorProperty.h"
#include "vtkSMPropertyHelper.h"
#include "vtkSMSession.h"
#include <assert.h>
vtkStandardNewMacro(vtkSMAnimationSceneGeometryWriter);
vtkCxxSetObjectMacro(vtkSMAnimationSceneGeometryWriter, ViewModule, vtkSMProxy);
//-----------------------------------------------------------------------------
vtkSMAnimationSceneGeometryWriter::vtkSMAnimationSceneGeometryWriter()
{
this->GeometryWriter = 0;
this->ViewModule = 0;
}
//-----------------------------------------------------------------------------
vtkSMAnimationSceneGeometryWriter::~vtkSMAnimationSceneGeometryWriter()
{
this->SetViewModule(0);
if (this->GeometryWriter)
{
this->GeometryWriter->Delete();
this->GeometryWriter = 0;
}
}
//-----------------------------------------------------------------------------
bool vtkSMAnimationSceneGeometryWriter::SaveInitialize(int vtkNotUsed(startCount))
{
if (!this->ViewModule)
{
vtkErrorMacro("No view from which to save the geometry is set.");
return false;
}
assert("The session should be set by now" && this->Session);
vtkSMSessionProxyManager* pxm = this->GetSessionProxyManager();
this->GeometryWriter = pxm->NewProxy("writers","XMLPVAnimationWriter");
vtkSMPropertyHelper(this->GeometryWriter,"FileName").Set(this->FileName);
vtkSMProxyProperty* pp = vtkSMProxyProperty::SafeDownCast(
this->ViewModule->GetProperty("Representations"));
vtkSMProxyProperty* gwInput = vtkSMProxyProperty::SafeDownCast(
this->GeometryWriter->GetProperty("Representations"));
gwInput->RemoveAllProxies();
for (unsigned int cc=0; cc < pp->GetNumberOfProxies(); ++cc)
{
vtkSMRepresentationProxy* repr =
vtkSMRepresentationProxy::SafeDownCast(
pp->GetProxy(cc));
if (repr && vtkSMPropertyHelper(repr,"Visibility", true).GetAsInt() != 0)
{
gwInput->AddProxy(repr);
}
}
this->GeometryWriter->UpdateVTKObjects();
this->GeometryWriter->InvokeCommand("Start");
return true;
}
//-----------------------------------------------------------------------------
bool vtkSMAnimationSceneGeometryWriter::SaveFrame(double time)
{
vtkSMPropertyHelper(this->GeometryWriter, "WriteTime").Set(time);
this->GeometryWriter->UpdateProperty("WriteTime", 1);
this->GeometryWriter->UpdatePropertyInformation();
if (vtkSMPropertyHelper(this->GeometryWriter, "ErrorCode").GetAsInt())
{
return false;
}
return true;
}
//-----------------------------------------------------------------------------
bool vtkSMAnimationSceneGeometryWriter::SaveFinalize()
{
if (!this->GeometryWriter)
{
return true;
}
this->GeometryWriter->InvokeCommand("Finish");
this->GeometryWriter->Delete();
this->GeometryWriter = 0;
return true;
}
//-----------------------------------------------------------------------------
void vtkSMAnimationSceneGeometryWriter::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
os << indent << "ViewModule: " << this->ViewModule << endl;
}
|
