vtkMPIMToNSocketConnection.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkMPIMToNSocketConnection_h
#define vtkMPIMToNSocketConnection_h

#include "vtkObject.h"
#include "vtkRemotingCoreModule.h" //needed for exports
class vtkMultiProcessController;
class vtkServerSocket;
class vtkSocketCommunicator;
class vtkMPIMToNSocketConnectionPortInformation;
class vtkMPIMToNSocketConnectionInternals;

class VTKREMOTINGCORE_EXPORT vtkMPIMToNSocketConnection : public vtkObject
{
public:
  static vtkMPIMToNSocketConnection* New();
  vtkTypeMacro(vtkMPIMToNSocketConnection, vtkObject);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  void Initialize(int waiting_process_type);

  void ConnectMtoN();


  void SetNumberOfConnections(int);
  vtkGetMacro(NumberOfConnections, int);

  void SetPortInformation(unsigned int processNumber, int portNumber, const char* hostName);


  vtkGetObjectMacro(SocketCommunicator, vtkSocketCommunicator);

  void GetPortInformation(vtkMPIMToNSocketConnectionPortInformation*);


  vtkGetMacro(PortNumber, int);

protected:
  vtkSetMacro(PortNumber, int);

  void SetupWaitForConnection();

  void WaitForConnection();

  void Connect();

  virtual void SetController(vtkMultiProcessController*);
  virtual void SetSocketCommunicator(vtkSocketCommunicator*);
  vtkMPIMToNSocketConnection();
  ~vtkMPIMToNSocketConnection() override;

private:
  int PortNumber;
  int Socket;
  vtkServerSocket* ServerSocket;
  int NumberOfConnections;
  vtkMPIMToNSocketConnectionInternals* Internals;
  vtkMultiProcessController* Controller;
  vtkSocketCommunicator* SocketCommunicator;
  vtkMPIMToNSocketConnection(const vtkMPIMToNSocketConnection&) = delete;
  void operator=(const vtkMPIMToNSocketConnection&) = delete;
  bool IsWaiting;
};

#endif