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
|
/*=========================================================================
Program: ParaView
Module: pvpython.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 "vtkInitializationHelper.h"
#include "vtkMultiProcessController.h"
#include "vtkProcessModule.h"
#include "vtkPVConfig.h" // Required to get build options for paraview
#include "vtkPVPythonOptions.h"
#include "vtkSMSession.h"
#include "vtkToolkits.h" // For VTK_USE_MPI
#include "vtkPVPythonInterpretor.h"
#include <vector>
namespace ParaViewPython {
//---------------------------------------------------------------------------
char* clone(const char* str)
{
char *newStr = new char[ strlen(str) + 1 ];
strcpy(newStr, str);
return newStr;
}
//---------------------------------------------------------------------------
void ProcessArgsForPython( std::vector<char*> & pythonArgs,
const char *script,
int argc, char* argv[] )
{
pythonArgs.clear();
pythonArgs.push_back(clone(argv[0]));
if(script)
{
pythonArgs.push_back(clone(script));
}
else if (argc > 1)
{
pythonArgs.push_back(clone("-"));
}
for (int cc=1; cc < argc; cc++)
{
pythonArgs.push_back(clone(argv[cc]));
}
}
//---------------------------------------------------------------------------
int Run(int processType, int argc, char* argv[])
{
// Setup options
// Marking this static avoids the false leak messages from vtkDebugLeaks when
// using mpich. It appears that the root process which spawns all the
// main processes waits in MPI_Init() and calls exit() when
// the others are done, causing apparent memory leaks for any non-static objects
// created before MPI_Init().
static vtkSmartPointer<vtkPVPythonOptions> options =
vtkSmartPointer<vtkPVPythonOptions>::New();
vtkInitializationHelper::Initialize( argc, argv, processType, options );
if (options->GetTellVersion() || options->GetHelpSelected())
{
vtkInitializationHelper::Finalize();
return 1;
}
vtkProcessModule* pm = vtkProcessModule::GetProcessModule();
int ret_val = 0;
if (pm->GetSymmetricMPIMode() == false &&
pm->GetPartitionId() > 0)
{
vtkIdType sid = vtkSMSession::ConnectToSelf();
pm->GetGlobalController()->ProcessRMIs();
pm->UnRegisterSession(sid);
}
else
{
int remaining_argc;
char** remaining_argv;
options->GetRemainingArguments(&remaining_argc, &remaining_argv);
// Process arguments
std::vector<char*> pythonArgs;
ProcessArgsForPython(pythonArgs, options->GetPythonScriptName(),
remaining_argc, remaining_argv);
// Start interpretor
vtkPVPythonInterpretor* interpretor = vtkPVPythonInterpretor::New();
ret_val = interpretor->PyMain(pythonArgs.size(), &*pythonArgs.begin());
interpretor->Delete();
// Free python args
std::vector<char*>::iterator it = pythonArgs.begin();
while(it != pythonArgs.end())
{
delete [] *it;
++it;
}
}
// Exit application
vtkInitializationHelper::Finalize();
return ret_val;
}
}
|
