131 lines
6.0 KiB
C++
131 lines
6.0 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "CoreTypes.h"
|
|
#include "Templates/SharedPointer.h"
|
|
#include "Misc/Optional.h"
|
|
#include "Utils/MPEG/ElectraUtilsMPEGVideo.h"
|
|
#include "MediaVideoDecoderOutput.h"
|
|
#include "ParameterDictionary.h"
|
|
|
|
namespace Electra
|
|
{
|
|
|
|
namespace MPEG
|
|
{
|
|
|
|
class FVideoDecoderTimecode : public IVideoDecoderTimecode
|
|
{
|
|
public:
|
|
virtual ~FVideoDecoderTimecode() = default;
|
|
virtual FMPEGDefinition const* GetMPEGDefinition() const override
|
|
{ return &DecoderTimecode; }
|
|
void Update(const IVideoDecoderTimecode::FMPEGDefinition& InTimecode)
|
|
{ DecoderTimecode = InTimecode; }
|
|
void UpdateWith(const ElectraDecodersUtil::MPEG::FCommonPictureTiming& InTiming)
|
|
{
|
|
DecoderTimecode.clockTimestamp = InTiming.clockTimestamp;
|
|
DecoderTimecode.num_units_in_tick = InTiming.num_units_in_tick;
|
|
DecoderTimecode.time_scale = InTiming.time_scale;
|
|
DecoderTimecode.time_offset = InTiming.time_offset;
|
|
DecoderTimecode.n_frames = InTiming.n_frames;
|
|
DecoderTimecode.timing_info_present_flag = InTiming.timing_info_present_flag;
|
|
DecoderTimecode.clock_timestamp_flag = InTiming.clock_timestamp_flag;
|
|
DecoderTimecode.ct_type = InTiming.ct_type;
|
|
DecoderTimecode.nuit_field_based_flag = InTiming.nuit_field_based_flag;
|
|
DecoderTimecode.counting_type = InTiming.counting_type;
|
|
DecoderTimecode.full_timestamp_flag = InTiming.full_timestamp_flag;
|
|
DecoderTimecode.discontinuity_flag = InTiming.discontinuity_flag;
|
|
DecoderTimecode.cnt_dropped_flag = InTiming.cnt_dropped_flag;
|
|
DecoderTimecode.seconds_value = InTiming.seconds_value;
|
|
DecoderTimecode.minutes_value = InTiming.minutes_value;
|
|
DecoderTimecode.hours_value = InTiming.hours_value;
|
|
DecoderTimecode.FromH26x = InTiming.FromH26x;
|
|
}
|
|
private:
|
|
IVideoDecoderTimecode::FMPEGDefinition DecoderTimecode;
|
|
};
|
|
|
|
|
|
|
|
class FColorimetryHelper
|
|
{
|
|
public:
|
|
ELECTRAPLAYERRUNTIME_API void Reset();
|
|
ELECTRAPLAYERRUNTIME_API void Update(uint8 colour_primaries, uint8 transfer_characteristics, uint8 matrix_coeffs, uint8 video_full_range_flag, uint8 video_format);
|
|
ELECTRAPLAYERRUNTIME_API void Update(const TArray<uint8>& InFromCOLRBox);
|
|
ELECTRAPLAYERRUNTIME_API void UpdateParamDict(FParamDict& InOutDictionary);
|
|
ELECTRAPLAYERRUNTIME_API bool GetCurrentValues(uint8& colour_primaries, uint8& transfer_characteristics, uint8& matrix_coeffs) const;
|
|
|
|
private:
|
|
class FVideoDecoderColorimetry : public IVideoDecoderColorimetry
|
|
{
|
|
public:
|
|
FVideoDecoderColorimetry(uint8 colour_primaries, uint8 transfer_characteristics, uint8 matrix_coeffs, uint8 video_full_range_flag, uint8 video_format)
|
|
{
|
|
Colorimetry.ColourPrimaries = colour_primaries;
|
|
Colorimetry.TransferCharacteristics = transfer_characteristics;
|
|
Colorimetry.MatrixCoefficients = matrix_coeffs;
|
|
Colorimetry.VideoFullRangeFlag = video_full_range_flag;
|
|
Colorimetry.VideoFormat = video_format;
|
|
}
|
|
virtual ~FVideoDecoderColorimetry() = default;
|
|
IVideoDecoderColorimetry::FMPEGDefinition const* GetMPEGDefinition() const override
|
|
{ return &Colorimetry; }
|
|
IVideoDecoderColorimetry::FMPEGDefinition Colorimetry;
|
|
};
|
|
|
|
TSharedPtr<FVideoDecoderColorimetry, ESPMode::ThreadSafe> CurrentColorimetry;
|
|
};
|
|
|
|
|
|
|
|
class FHDRHelper
|
|
{
|
|
public:
|
|
ELECTRAPLAYERRUNTIME_API void Reset();
|
|
ELECTRAPLAYERRUNTIME_API void Update(int32 BitDepth, const FColorimetryHelper& InColorimetry, const TArray<ElectraDecodersUtil::MPEG::FSEIMessage>& InGlobalPrefixSEIs, const TArray<ElectraDecodersUtil::MPEG::FSEIMessage>& InLocalPrefixSEIs, bool bIsNewCLVS);
|
|
ELECTRAPLAYERRUNTIME_API void UpdateFromMPEGBoxes(int32 BitDepth, const FColorimetryHelper& InColorimetry, const TArray<uint8>& InMDCVBox, const TArray<uint8>& InCLLIBox);
|
|
ELECTRAPLAYERRUNTIME_API void Update(int32 BitDepth, const FColorimetryHelper& InColorimetry, const TOptional<FVideoDecoderHDRMetadata_mastering_display_colour_volume>& InMDCV, const TOptional<FVideoDecoderHDRMetadata_content_light_level_info>& InCLLI);
|
|
ELECTRAPLAYERRUNTIME_API void UpdateWith(const ElectraDecodersUtil::MPEG::FSEImastering_display_colour_volume& InSEI);
|
|
ELECTRAPLAYERRUNTIME_API void UpdateWith(const ElectraDecodersUtil::MPEG::FSEIcontent_light_level_info& InSEI);
|
|
ELECTRAPLAYERRUNTIME_API void UpdateWith(const ElectraDecodersUtil::MPEG::FSEIalternative_transfer_characteristics& InSEI);
|
|
ELECTRAPLAYERRUNTIME_API void SetHDRType(int32 BitDepth, const FColorimetryHelper& InColorimetry);
|
|
ELECTRAPLAYERRUNTIME_API void UpdateParamDict(FParamDict& InOutDictionary);
|
|
private:
|
|
class FVideoDecoderHDRInformation : public IVideoDecoderHDRInformation
|
|
{
|
|
public:
|
|
virtual ~FVideoDecoderHDRInformation() = default;
|
|
IVideoDecoderHDRInformation::EType GetHDRType() const override
|
|
{ return HDRType; }
|
|
const FVideoDecoderHDRMetadata_mastering_display_colour_volume* GetMasteringDisplayColourVolume() const override
|
|
{ return MasteringDisplayColourVolume.GetPtrOrNull(); }
|
|
const FVideoDecoderHDRMetadata_content_light_level_info* GetContentLightLevelInfo() const override
|
|
{ return ContentLightLevelInfo.GetPtrOrNull(); }
|
|
void SetHDRType(IVideoDecoderHDRInformation::EType InHDRType)
|
|
{ HDRType = InHDRType; }
|
|
void SetMasteringDisplayColourVolume(const FVideoDecoderHDRMetadata_mastering_display_colour_volume& In)
|
|
{ MasteringDisplayColourVolume = In; }
|
|
void SetContentLightLevelInfo(const FVideoDecoderHDRMetadata_content_light_level_info& In)
|
|
{ ContentLightLevelInfo = In; }
|
|
private:
|
|
IVideoDecoderHDRInformation::EType HDRType = IVideoDecoderHDRInformation::EType::Unknown;
|
|
TOptional<FVideoDecoderHDRMetadata_mastering_display_colour_volume> MasteringDisplayColourVolume;
|
|
TOptional<FVideoDecoderHDRMetadata_content_light_level_info> ContentLightLevelInfo;
|
|
};
|
|
TSharedPtr<FVideoDecoderHDRInformation, ESPMode::ThreadSafe> CurrentHDRInfo;
|
|
TOptional<ElectraDecodersUtil::MPEG::FSEIMessage> ActiveMasteringDisplayColourVolume;
|
|
TOptional<ElectraDecodersUtil::MPEG::FSEIMessage> ActiveContentLightLevelInfo;
|
|
TOptional<ElectraDecodersUtil::MPEG::FSEIMessage> ActiveAlternativeTransferCharacteristics;
|
|
bool bIsFirst = true;
|
|
int32 CurrentAlternativeTransferCharacteristics = -1;
|
|
};
|
|
|
|
|
|
|
|
} // namespace MPEG
|
|
} // namespace Electra
|
|
|