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
126
127
128
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkSocketController.h"
#include "vtkObjectFactory.h"
#include "vtkProcessGroup.h"
#include "vtkSocketCommunicator.h"
#if defined(_WIN32) && !defined(__CYGWIN__)
#define VTK_WINDOWS_FULL
#include "vtkWindows.h"
#define WSA_VERSION MAKEWORD(1, 1)
#endif
VTK_ABI_NAMESPACE_BEGIN
int vtkSocketController::Initialized = 0;
vtkStandardNewMacro(vtkSocketController);
//------------------------------------------------------------------------------
vtkSocketController::vtkSocketController()
{
this->Communicator = vtkSocketCommunicator::New();
this->RMICommunicator = this->Communicator;
}
//------------------------------------------------------------------------------
vtkSocketController::~vtkSocketController()
{
this->Communicator->Delete();
this->Communicator = this->RMICommunicator = nullptr;
}
//------------------------------------------------------------------------------
void vtkSocketController::Initialize(int*, char***)
{
if (vtkSocketController::Initialized)
{
vtkWarningMacro("Already initialized.");
return;
}
#if defined(_WIN32) && !defined(__CYGWIN__)
WSAData wsaData;
if (WSAStartup(WSA_VERSION, &wsaData))
{
vtkErrorMacro("Could not initialize sockets !");
}
#endif
vtkSocketController::Initialized = 1;
}
//------------------------------------------------------------------------------
void vtkSocketController::SetCommunicator(vtkSocketCommunicator* comm)
{
if (comm == this->Communicator)
{
return;
}
if (this->Communicator)
{
this->Communicator->UnRegister(this);
}
this->Communicator = comm;
this->RMICommunicator = comm;
if (comm)
{
comm->Register(this);
}
}
//------------------------------------------------------------------------------
void vtkSocketController::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
//------------------------------------------------------------------------------
int vtkSocketController::WaitForConnection(int port)
{
return vtkSocketCommunicator::SafeDownCast(this->Communicator)->WaitForConnection(port);
}
//------------------------------------------------------------------------------
void vtkSocketController::CloseConnection()
{
vtkSocketCommunicator::SafeDownCast(this->Communicator)->CloseConnection();
}
//------------------------------------------------------------------------------
int vtkSocketController::ConnectTo(const char* hostName, int port)
{
return vtkSocketCommunicator::SafeDownCast(this->Communicator)->ConnectTo(hostName, port);
}
//------------------------------------------------------------------------------
int vtkSocketController::GetSwapBytesInReceivedData()
{
return vtkSocketCommunicator::SafeDownCast(this->Communicator)->GetSwapBytesInReceivedData();
}
//------------------------------------------------------------------------------
vtkMultiProcessController* vtkSocketController::CreateCompliantController()
{
vtkProcessGroup* group = vtkProcessGroup::New();
group->Initialize(this->Communicator);
group->RemoveAllProcessIds();
// This hack creates sub controllers with differing orders of the processes
// that will map the ids to be unique on each process.
if (vtkSocketCommunicator::SafeDownCast(this->Communicator)->GetIsServer())
{
group->AddProcessId(1);
group->AddProcessId(0);
}
else
{
group->AddProcessId(0);
group->AddProcessId(1);
}
vtkMultiProcessController* compliantController = this->CreateSubController(group);
group->Delete();
return compliantController;
}
VTK_ABI_NAMESPACE_END
|
