Newer
Older
Karsten Suehring
committed
/* The copyright in this software is being made available under the BSD
* License, included below. This software may be subject to other third party
* and contributor rights, including patent rights, and no such rights are
* granted under this license.
*
* Copyright (c) 2010-2019, ITU/ISO/IEC
Karsten Suehring
committed
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
* be used to endorse or promote products derived from this software without
* specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#pragma once
#ifndef __SEIENCODER__
#define __SEIENCODER__
#include "CommonLib/SEI.h"
// forward declarations
class EncCfg;
class EncLib;
class EncGOP;
//! Initializes different SEI message types based on given encoder configuration parameters
class SEIEncoder
{
public:
SEIEncoder()
:m_pcCfg(NULL)
,m_pcEncLib(NULL)
,m_pcEncGOP(NULL)
,m_tl0Idx(0)
,m_rapIdx(0)
,m_isInitialized(false)
{};
virtual ~SEIEncoder(){};
void init(EncCfg* encCfg, EncLib *encTop, EncGOP *encGOP)
{
m_pcCfg = encCfg;
m_pcEncGOP = encGOP;
m_pcEncLib = encTop;
m_isInitialized = true;
};
// leading SEIs
#if HEVC_VPS
void initSEIActiveParameterSets (SEIActiveParameterSets *sei, const VPS *vps, const SPS *sps);
#else
void initSEIActiveParameterSets (SEIActiveParameterSets *sei, const SPS *sps);
#endif
void initSEIFramePacking(SEIFramePacking *sei, int currPicNum);
void initSEIDisplayOrientation(SEIDisplayOrientation *sei);
void initSEIToneMappingInfo(SEIToneMappingInfo *sei);
void initSEISOPDescription(SEISOPDescription *sei, Slice *slice, int picInGOP, int lastIdr, int currGOPSize);
void initSEIBufferingPeriod(SEIBufferingPeriod *sei, Slice *slice);
void initSEIScalableNesting(SEIScalableNesting *sei, SEIMessages &nestedSEIs);
void initSEIRecoveryPoint(SEIRecoveryPoint *sei, Slice *slice);
void initSEISegmentedRectFramePacking(SEISegmentedRectFramePacking *sei);
void initSEITempMotionConstrainedTileSets (SEITempMotionConstrainedTileSets *sei, const PPS *pps);
void initSEIKneeFunctionInfo(SEIKneeFunctionInfo *sei);
void initSEIChromaResamplingFilterHint(SEIChromaResamplingFilterHint *sei, int iHorFilterIndex, int iVerFilterIndex);
void initSEITimeCode(SEITimeCode *sei);
bool initSEIColourRemappingInfo(SEIColourRemappingInfo *sei, int currPOC); // returns true on success, false on failure.
#if U0033_ALTERNATIVE_TRANSFER_CHARACTERISTICS_SEI
void initSEIAlternativeTransferCharacteristics(SEIAlternativeTransferCharacteristics *sei);
#endif
// trailing SEIs
void initDecodedPictureHashSEI(SEIDecodedPictureHash *sei, PelUnitBuf& pic, std::string &rHashString, const BitDepths &bitDepths);
void initTemporalLevel0IndexSEI(SEITemporalLevel0Index *sei, Slice *slice);
void initSEIGreenMetadataInfo(SEIGreenMetadataInfo *sei, uint32_t u);
private:
EncCfg* m_pcCfg;
EncLib* m_pcEncLib;
EncGOP* m_pcEncGOP;
// for temporal level 0 index SEI
uint32_t m_tl0Idx;
uint32_t m_rapIdx;
bool m_isInitialized;
};
//! \}
#endif // __SEIENCODER__