From 1fec6a8ac4dfb5bf4ed5a2097527afc9d6048a37 Mon Sep 17 00:00:00 2001
From: Hendry <fnu.hendry@huawei.com>
Date: Tue, 17 Sep 2019 11:17:07 -0700
Subject: [PATCH] Modify the condition for having BP SEI msg
---
source/Lib/CommonLib/TypeDef.h | 2 ++
source/Lib/EncoderLib/EncGOP.cpp | 7 +++++++
2 files changed, 9 insertions(+)
diff --git a/source/Lib/CommonLib/TypeDef.h b/source/Lib/CommonLib/TypeDef.h
index 239ac95256..3e709eabf8 100644
--- a/source/Lib/CommonLib/TypeDef.h
+++ b/source/Lib/CommonLib/TypeDef.h
@@ -52,6 +52,8 @@
#define JVET_N0494_DRAP 1 // JVET-N0494: Dependent random access point indication SEI
+#define JVET_OO152_BP_SEI_GDR 1 // JVET-O0152: Allowing association of a buffering period SEI with GDR picture
+
#define JVET_OO147_LEADING_PIC_CHECKING 1 // JVET-O0147: Constraints on leading pictures
#define JVET_O0299_APS_SCALINGLIST 1 // JVET-O0299: Scaling List Matrices Support in APS
diff --git a/source/Lib/EncoderLib/EncGOP.cpp b/source/Lib/EncoderLib/EncGOP.cpp
index 57e526936d..9abf1b4576 100644
--- a/source/Lib/EncoderLib/EncGOP.cpp
+++ b/source/Lib/EncoderLib/EncGOP.cpp
@@ -696,10 +696,17 @@ void EncGOP::xCreateIRAPLeadingSEIMessages (SEIMessages& seiMessages, const SPS
void EncGOP::xCreatePerPictureSEIMessages (int picInGOP, SEIMessages& seiMessages, SEIMessages& nestedSeiMessages, Slice *slice)
{
+#if JVET_OO152_BP_SEI_GDR
+ if ((m_pcCfg->getBufferingPeriodSEIEnabled()) && (slice->isIRAP() || slice->getNalUnitType() == NAL_UNIT_CODED_SLICE_GDR) &&
+ (slice->getSPS()->getVuiParametersPresentFlag()) &&
+ ((slice->getSPS()->getHrdParameters()->getNalHrdParametersPresentFlag())
+ || (slice->getSPS()->getHrdParameters()->getVclHrdParametersPresentFlag())))
+#else
if( ( m_pcCfg->getBufferingPeriodSEIEnabled() ) && ( slice->getSliceType() == I_SLICE ) &&
( slice->getSPS()->getVuiParametersPresentFlag() ) &&
( ( slice->getSPS()->getHrdParameters()->getNalHrdParametersPresentFlag() )
|| ( slice->getSPS()->getHrdParameters()->getVclHrdParametersPresentFlag() ) ) )
+#endif
{
SEIBufferingPeriod *bufferingPeriodSEI = new SEIBufferingPeriod();
m_seiEncoder.initSEIBufferingPeriod(bufferingPeriodSEI);
--
GitLab