44 lines
1.3 KiB
C++
44 lines
1.3 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Containers/Set.h"
|
|
#include "EpicRtcTrack.h"
|
|
#include "VideoCapturer.h"
|
|
#include "VideoSink.h"
|
|
#include "RendererInterface.h"
|
|
|
|
#include "epic_rtc/core/video/video_frame.h"
|
|
|
|
#define UE_API PIXELSTREAMING2RTC_API
|
|
|
|
namespace UE::PixelStreaming2
|
|
{
|
|
/**
|
|
* Video sink class that receives a frame from EpicRtc and passes the frame to all added consumers
|
|
*/
|
|
class FEpicRtcVideoSink : public FVideoSink, public TEpicRtcTrack<EpicRtcVideoTrackInterface>, public TSharedFromThis<FEpicRtcVideoSink>
|
|
{
|
|
public:
|
|
static UE_API TSharedPtr<FEpicRtcVideoSink> Create(TRefCountPtr<EpicRtcVideoTrackInterface> InTrack);
|
|
// Note: destructor will call destroy on any attached video consumers
|
|
virtual ~FEpicRtcVideoSink() = default;
|
|
|
|
UE_API void OnEpicRtcFrame(const EpicRtcVideoFrame& Frame);
|
|
private:
|
|
UE_API FEpicRtcVideoSink(TRefCountPtr<EpicRtcVideoTrackInterface> InTrack);
|
|
|
|
UE_API void OnFrameCaptured();
|
|
|
|
FCriticalSection RenderSyncContext;
|
|
FPooledRenderTargetDesc RenderTargetDescriptor;
|
|
TRefCountPtr<IPooledRenderTarget> RenderTarget;
|
|
TArray<uint8_t> Buffer;
|
|
FTextureRHIRef SourceTexture;
|
|
|
|
TSharedPtr<FVideoCapturer> VideoCapturer;
|
|
};
|
|
} // namespace UE::PixelStreaming2
|
|
|
|
#undef UE_API
|