ImageWriteStream.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Containers/Array.h"
#include "Containers/ArrayView.h"
#include "HAL/CriticalSection.h"
#include "ImagePixelData.h"
#include "Misc/CoreDefines.h"
#include "Templates/Function.h"
#include "Templates/UniquePtr.h"
#include "Templates/UnrealTemplate.h"
 
struct FImagePixelData;
struct FImageStreamEndpoint;
 
struct FImagePixelPipe
{
    FImagePixelPipe()
    {}
 
    FImagePixelPipe(const TFunction<void(TUniquePtr<FImagePixelData>&&)>& InEndpoint)
    {
        AddEndpoint(InEndpoint);
    }
 
    IMAGEWRITEQUEUE_API void Push(TUniquePtr<FImagePixelData>&& InImagePixelData);
 
    IMAGEWRITEQUEUE_API void AddEndpoint(TUniquePtr<FImageStreamEndpoint>&& InEndpoint);
 
    IMAGEWRITEQUEUE_API void AddEndpoint(const TFunction<void(TUniquePtr<FImagePixelData>&& )>& InHandler);
 
    TArrayView<const TUniquePtr<FImageStreamEndpoint>> GetEndPoints() const
    {
        return EndPoints;
    }
 
    std::atomic_bool bIsExpecting32BitPixelData = false;
 
private:
 
    FCriticalSection EndPointLock;
 
    TArray<TUniquePtr<FImageStreamEndpoint>> EndPoints;
};
 
 
 
struct FImageStreamEndpoint
{
    virtual ~FImageStreamEndpoint(){}
 
    IMAGEWRITEQUEUE_API void PipeImage(TUniquePtr<FImagePixelData>&& InOwnedImage);
 
private:
 
    virtual void OnImageReceived(TUniquePtr<FImagePixelData>&& InOwnedImage) {}
};