Code/Engine/Foundation/Platform/Win/PipeChannel_Platform.h - dev - redgpu/ezEngine

Code/Engine/Foundation/Platform/Win/PipeChannel_Platform.h

71 строка2 KB

Jan Krassnigg

Further refactor of platform specific code (2) (#1416)

25 окт 2024, 17:01

25 окт 2024, 17:01b3eeda7

#pragma once

#include <Foundation/FoundationInternal.h>
EZ_FOUNDATION_INTERNAL_HEADER

#if EZ_ENABLED(EZ_PLATFORM_WINDOWS_DESKTOP)

#  include <Foundation/Basics.h>
#  include <Foundation/Communication/IpcChannel.h>
#  include <Foundation/Platform/Win/Utils/IncludeWindows.h>

struct IOContext
{
  OVERLAPPED Overlapped;  ///< Must be first field in class so we can do a reinterpret cast from *Overlapped to *IOContext.
  ezIpcChannel* pChannel; ///< Owner of this IOContext.
};

class EZ_FOUNDATION_DLL ezPipeChannel_win : public ezIpcChannel
{
public:
  ezPipeChannel_win(ezStringView sAddress, Mode::Enum mode);
  ~ezPipeChannel_win();

private:
  friend class ezMessageLoop;
  friend class ezMessageLoop_win;

  bool CreatePipe(ezStringView sAddress);

  // All functions from here on down are run from worker thread only
  virtual void InternalConnect() override;
  virtual void InternalDisconnect() override;
  virtual void InternalSend() override;
  virtual bool NeedWakeup() const override;

  bool ProcessConnection();
  bool ProcessIncomingMessages(DWORD uiBytesRead);
  bool ProcessOutgoingMessages(DWORD uiBytesWritten);


protected:
  void OnIOCompleted(IOContext* pContext, DWORD uiBytesTransfered, DWORD uiError);

private:
  struct State
  {
    explicit State(ezPipeChannel_win* pChannel);
    ~State();
    IOContext Context;
    ezAtomicInteger32 IsPending = false; ///< Whether an async operation is in process.
  };

  enum Constants
  {
    BUFFER_SIZE = 4096,
  };

  // Shared data
  State m_InputState;
  State m_OutputState;

  // Setup in ctor
  HANDLE m_hPipeHandle = INVALID_HANDLE_VALUE;

  // Only accessed from worker thread
  ezUInt8 m_InputBuffer[BUFFER_SIZE];
};

using ezPipeChannel_Platform = ezPipeChannel_win;

#endif