/*****************************************************************************

* Copyright (C) 2013-2020 MulticoreWare, Inc

*

* Authors: Steve Borho <steve@borho.org>

*

* This program is free software; you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation; either version 2 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program; if not, write to the Free Software

* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.

*

* This program is also available under a commercial proprietary license.

* For more information, contact us at license @ x265.com.

*****************************************************************************/

#ifndef X265_SEI_H

#define X265_SEI_H

#include "common.h"

#include "bitstream.h"

#include "slice.h"

#include "nal.h"

#include "md5.h"

namespace X265_NS {

// private namespace

class SEI : public SyntaxElementWriter

{

public:

    /* SEI users call writeSEImessages() to marshal an SEI to a bitstream.

    * The writeSEImessages() method calls writeSEI() which encodes the header */

    void writeSEImessages(Bitstream& bs, const SPS& sps, NalUnitType nalUnitType, NALList& list, int isNested, int layerId = 0);

    void setSize(uint32_t size);

    static char* base64Decode(char encodedString[], int base64EncodeLength, char* base64Decode);

    virtual ~SEI() {}

protected:

    SEIPayloadType  m_payloadType;

    uint32_t        m_payloadSize;

    virtual void writeSEI(const SPS&) = 0;

    void writeByteAlign();

};

//seongnam.oh@samsung.com :: for the Creative Intent Meta Data Encoding

class SEIuserDataRegistered : public SEI

{

public:

    SEIuserDataRegistered()

    {

        m_payloadType = USER_DATA_REGISTERED_ITU_T_T35;

        m_payloadSize = 0;

    }

    uint8_t *m_userData;

    // daniel.vt@samsung.com :: for the Creative Intent Meta Data Encoding ( seongnam.oh@samsung.com )

    void writeSEI(const SPS&)

    {

        if (!m_userData)

            return;

        uint32_t i = 0;

        for (; i < m_payloadSize; ++i)

            WRITE_CODE(m_userData[i], 8, "creative_intent_metadata");

    }

};

/* Film grain characteristics */

class FilmGrainCharacteristics : public SEI

{

  public:

    FilmGrainCharacteristics()

    {

        m_payloadType = FILM_GRAIN_CHARACTERISTICS;

        m_payloadSize = 0;

    }

    struct CompModelIntensityValues

    {

        uint8_t intensityIntervalLowerBound;

        uint8_t intensityIntervalUpperBound;

        int*    compModelValue;

    };

    struct CompModel

    {

        bool    bPresentFlag;

        uint8_t numModelValues;

        uint8_t m_filmGrainNumIntensityIntervalMinus1;

        CompModelIntensityValues* intensityValues;

    };

    CompModel   m_compModel[MAX_NUM_COMPONENT];

    bool        m_filmGrainCharacteristicsPersistenceFlag;

    bool        m_filmGrainCharacteristicsCancelFlag;

    bool        m_separateColourDescriptionPresentFlag;

    bool        m_filmGrainFullRangeFlag;

    uint8_t     m_filmGrainModelId;

    uint8_t     m_blendingModeId;

    uint8_t     m_log2ScaleFactor;

    uint8_t     m_filmGrainBitDepthLumaMinus8;

    uint8_t     m_filmGrainBitDepthChromaMinus8;

    uint8_t     m_filmGrainColourPrimaries;

    uint8_t     m_filmGrainTransferCharacteristics;

    uint8_t     m_filmGrainMatrixCoeffs;

    void writeSEI(const SPS&)

    {

        WRITE_FLAG(m_filmGrainCharacteristicsCancelFlag, "film_grain_characteristics_cancel_flag");

        if (!m_filmGrainCharacteristicsCancelFlag)

        {

            WRITE_CODE(m_filmGrainModelId, 2, "film_grain_model_id");

            WRITE_FLAG(m_separateColourDescriptionPresentFlag, "separate_colour_description_present_flag");

            if (m_separateColourDescriptionPresentFlag)

            {

                WRITE_CODE(m_filmGrainBitDepthLumaMinus8, 3, "film_grain_bit_depth_luma_minus8");

                WRITE_CODE(m_filmGrainBitDepthChromaMinus8, 3, "film_grain_bit_depth_chroma_minus8");

                WRITE_FLAG(m_filmGrainFullRangeFlag, "film_grain_full_range_flag");

                WRITE_CODE(m_filmGrainColourPrimaries, X265_BYTE, "film_grain_colour_primaries");

                WRITE_CODE(m_filmGrainTransferCharacteristics, X265_BYTE, "film_grain_transfer_characteristics");

                WRITE_CODE(m_filmGrainMatrixCoeffs, X265_BYTE, "film_grain_matrix_coeffs");

            }

            WRITE_CODE(m_blendingModeId, 2, "blending_mode_id");

            WRITE_CODE(m_log2ScaleFactor, 4, "log2_scale_factor");

            for (uint8_t c = 0; c < 3; c++)

            {

                WRITE_FLAG(m_compModel[c].bPresentFlag && m_compModel[c].m_filmGrainNumIntensityIntervalMinus1 + 1 > 0 && m_compModel[c].numModelValues > 0, "comp_model_present_flag[c]");

            }

            for (uint8_t c = 0; c < 3; c++)

            {

                if (m_compModel[c].bPresentFlag && m_compModel[c].m_filmGrainNumIntensityIntervalMinus1 + 1 > 0 && m_compModel[c].numModelValues > 0)

                {

                    assert(m_compModel[c].m_filmGrainNumIntensityIntervalMinus1 + 1 <= 256);

                    assert(m_compModel[c].numModelValues <= X265_BYTE);

                    WRITE_CODE(m_compModel[c].m_filmGrainNumIntensityIntervalMinus1 , X265_BYTE, "num_intensity_intervals_minus1[c]");

                    WRITE_CODE(m_compModel[c].numModelValues - 1, 3, "num_model_values_minus1[c]");

                    for (uint8_t interval = 0; interval < m_compModel[c].m_filmGrainNumIntensityIntervalMinus1 + 1; interval++)

                    {

                        WRITE_CODE(m_compModel[c].intensityValues[interval].intensityIntervalLowerBound, X265_BYTE, "intensity_interval_lower_bound[c][i]");

                        WRITE_CODE(m_compModel[c].intensityValues[interval].intensityIntervalUpperBound, X265_BYTE, "intensity_interval_upper_bound[c][i]");

                        for (uint8_t j = 0; j < m_compModel[c].numModelValues; j++)

                        {

                            WRITE_SVLC(m_compModel[c].intensityValues[interval].compModelValue[j],"comp_model_value[c][i]");

                        }

                    }

                }

            }

            WRITE_FLAG(m_filmGrainCharacteristicsPersistenceFlag, "film_grain_characteristics_persistence_flag");

        }

        if (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

        {

            WRITE_FLAG(1, "payload_bit_equal_to_one");

            while (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

            {

                WRITE_FLAG(0, "payload_bit_equal_to_zero");

            }

        }

    }

};

class AomFilmGrainCharacteristics : public SEI {

public:

    AomFilmGrainCharacteristics()

    {

        m_payloadType = USER_DATA_REGISTERED_ITU_T_T35;

        m_payloadSize = 0;

    }

    int32_t     m_apply_grain;

    int32_t     m_update_grain;

    int32_t     m_scaling_points_y[14][2];

    int32_t     m_num_y_points;

    int32_t     m_scaling_points_cb[10][2];

    int32_t     m_num_cb_points;

    int32_t     m_scaling_points_cr[10][2];

    int32_t     m_num_cr_points;

    int32_t     m_scaling_shift;

    int32_t     m_ar_coeff_lag;

    int32_t     m_ar_coeffs_y[24];

    int32_t     m_ar_coeffs_cb[25];

    int32_t     m_ar_coeffs_cr[25];

    int32_t     m_ar_coeff_shift;

    int32_t     m_cb_mult;

    int32_t     m_cb_luma_mult;

    int32_t     m_cb_offset;

    int32_t     m_cr_mult;

    int32_t     m_cr_luma_mult;

    int32_t     m_cr_offset;

    int32_t     m_overlap_flag;

    int32_t     m_clip_to_restricted_range;

    int32_t     m_bitDepth;

    int32_t     m_chroma_scaling_from_luma;

    int32_t     m_grain_scale_shift;

    uint16_t    m_grain_seed;

    void writeSEI(const SPS&)

    {

        WRITE_CODE(0x26, 8, "country_code");

        WRITE_CODE(0x5890, 16, "provider_code");

        WRITE_CODE(0x0001, 16, "provider_oriented_code");

        WRITE_FLAG(m_apply_grain, "afgs1_enable_flag");

        WRITE_CODE(m_grain_seed, 16, "grain_seed");

        WRITE_CODE(0, 3, "film_grain_param_set_idx");

        WRITE_CODE(m_update_grain, 1, "update_grain");

        WRITE_CODE(m_num_y_points, 4, "num_y_points");

        if (m_num_y_points)

        {

            for (int i = 0; i < m_num_y_points; i++)

            {

                for (int j = 0; j < 2; j++)

                {

                    WRITE_CODE(m_scaling_points_y[i][j], 8, "scaling_points_y[i][j]");

                }

            }

        }

        WRITE_FLAG(m_num_cb_points == 0 && m_num_cr_points == 0, "luma_only_flag");

        WRITE_FLAG(0, "chroma_scaling_from_luma");

        WRITE_CODE(m_num_cb_points, 4, "num_cb_points");

        if (m_num_cb_points)

        {

            for (int i = 0; i < m_num_cb_points; i++)

            {

                for (int j = 0; j < 2; j++)

                {

                    WRITE_CODE(m_scaling_points_cb[i][j], 8, "scaling_points_cb[i][j]");

                }

            }

        }

        WRITE_CODE(m_num_cr_points, 4, "num_cr_points");

        if (m_num_cr_points)

        {

            for (int i = 0; i < m_num_cr_points; i++)

            {

                for (int j = 0; j < 2; j++)

                {

                    WRITE_CODE(m_scaling_points_cr[i][j], 8, "scaling_points_cr[i][j]");

                }

            }

        }

        WRITE_CODE(m_scaling_shift - 8, 2, "scaling_shift");

        WRITE_CODE(m_ar_coeff_lag, 2, "ar_coeff_lag");

        if (m_num_y_points)

        {

            for (int i = 0; i < 24; i++)

            {

                WRITE_CODE(m_ar_coeffs_y[i] + 128, 8, "ar_coeff_y[i]");

            }

        }

        if (m_num_cb_points || m_chroma_scaling_from_luma)

        {

            for (int i = 0; i < 25; i++)

            {

                WRITE_CODE(m_ar_coeffs_cb[i] + 128, 8, "ar_coeff_cb[i]");

            }

        }

        if (m_num_cr_points || m_chroma_scaling_from_luma)

        {

            for (int i = 0; i < 25; i++)

            {

                WRITE_CODE(m_ar_coeffs_cr[i] + 128, 8, "ar_coeff_cr[i]");

            }

        }

        WRITE_CODE(m_ar_coeff_shift - 6, 2, "ar_coeff_shift");

        WRITE_CODE(m_grain_scale_shift, 2, "grain_scale_shift");

        if (m_num_cb_points)

        {

            WRITE_CODE(m_cb_mult, 8, "cb_mult");

            WRITE_CODE(m_cb_luma_mult, 8, "cb_luma_mult");

            WRITE_CODE(m_cb_offset, 9, "cb_offset");

        }

        if (m_num_cr_points)

        {

            WRITE_CODE(m_cr_mult, 8, "cr_mult");

            WRITE_CODE(m_cr_luma_mult, 8, "cr_luma_mult");

            WRITE_CODE(m_cr_offset, 9, "cr_offset");

        }

        WRITE_FLAG(m_overlap_flag, "overlap_flag");

        WRITE_FLAG(m_clip_to_restricted_range, "clip_to_restricted_range");

        if (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

        {

            WRITE_FLAG(1, "payload_bit_equal_to_one");

            while (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

            {

                WRITE_FLAG(0, "payload_bit_equal_to_zero");

            }

        }

    }

};

static const uint32_t ISO_IEC_11578_LEN = 16;

class SEIuserDataUnregistered : public SEI

{

public:

    SEIuserDataUnregistered() : m_userData(NULL)

    {

        m_payloadType = USER_DATA_UNREGISTERED;

        m_payloadSize = 0;

    }

    static const uint8_t m_uuid_iso_iec_11578[ISO_IEC_11578_LEN];

    uint8_t *m_userData;

    void writeSEI(const SPS&)

    {

        for (uint32_t i = 0; i < ISO_IEC_11578_LEN; i++)

            WRITE_CODE(m_uuid_iso_iec_11578[i], 8, "sei.uuid_iso_iec_11578[i]");

        for (uint32_t i = 0; i < m_payloadSize; i++)

            WRITE_CODE(m_userData[i], 8, "user_data");

    }

};

#if ENABLE_ALPHA

class SEIAlphaChannelInfo : public SEI

{

public:

    SEIAlphaChannelInfo()

    {

        m_payloadType = ALPHA_CHANNEL_INFO;

        m_payloadSize = 0;

    }

    bool alpha_channel_cancel_flag;

    void writeSEI(const SPS&)

    {

        WRITE_CODE(alpha_channel_cancel_flag, 1, "alpha_channel_cancel_flag");

        if (!alpha_channel_cancel_flag)

        {

            WRITE_CODE(0, 3, "alpha_channel_use_idc");

            WRITE_CODE(0, 3, "alpha_channel_bit_depth_minus8");

            WRITE_CODE(0, 9, "alpha_transparent_value");

            WRITE_CODE(255, 9, "alpha_opaque_value");

            WRITE_CODE(0, 1, "alpha_channel_incr_flag");

            WRITE_CODE(0, 1, "alpha_channel_clip_flag");

        }

        if (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

        {

            WRITE_FLAG(1, "payload_bit_equal_to_one");

            while (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

            {

                WRITE_FLAG(0, "payload_bit_equal_to_zero");

            }

        }

    }

};

#endif

#if ENABLE_MULTIVIEW

class SEIThreeDimensionalReferenceDisplaysInfo : public SEI

{

public:

    SEIThreeDimensionalReferenceDisplaysInfo()

        : m_numRefDisplaysMinus1(0),

        m_refViewingDistanceFlag(false),

        m_additionalShiftPresentFlag(false)

    {

        m_payloadType = THREE_DIMENSIONAL_REFERENCE_DISPLAYS_INFO;

        m_payloadSize = 0;

    }

    int m_numRefDisplaysMinus1;

    bool m_refViewingDistanceFlag;

    bool m_additionalShiftPresentFlag;

    void writeSEI(const SPS&)

    {

        WRITE_UVLC(31, "prec_ref_display_width");

        WRITE_FLAG(m_refViewingDistanceFlag, "ref_viewing_distance_flag");

        if (m_refViewingDistanceFlag)

        {

            WRITE_UVLC(0, "prec_ref_viewing_dist");

        }

        WRITE_UVLC(0, "num_ref_displays_minus1");

        for (int i = 0; i <= m_numRefDisplaysMinus1; i++)

        {

            WRITE_UVLC(0, "left_view_id");

            WRITE_UVLC(1, "right_view_id");

            WRITE_CODE(0, 6, "exponent_ref_display_width");

            WRITE_CODE(0, 2, "mantissa_ref_display_width");

            if (m_refViewingDistanceFlag)

            {

                WRITE_CODE(0, 6, "exponent_ref_viewing_distance");

                WRITE_CODE(0, 1, "mantissa_ref_viewing_distance");

            }

            WRITE_FLAG(m_additionalShiftPresentFlag, "additional_shift_present_flag");

            if (m_additionalShiftPresentFlag)

            {

                WRITE_CODE(0, 10, "num_sample_shift_plus512");

            }

        }

        WRITE_FLAG(0, "three_dimensional_reference_displays_extension_flag");

        if (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

        {

            WRITE_FLAG(1, "payload_bit_equal_to_one");

            while (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

            {

                WRITE_FLAG(0, "payload_bit_equal_to_zero");

            }

        }

    }

};

class SEIMultiviewSceneInfo : public SEI

{

public:

    SEIMultiviewSceneInfo()

    {

        m_payloadType = MULTIVIEW_SCENE_INFO;

        m_payloadSize = 0;

    }

    void writeSEI(const SPS&)

    {

        WRITE_SVLC(-333, "min_disparity");

        WRITE_UVLC(2047, "max_disparity_range");

        if (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

        {

            WRITE_FLAG(1, "payload_bit_equal_to_one");

            while (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

            {

                WRITE_FLAG(0, "payload_bit_equal_to_zero");

            }

        }

    }

};

static int temp_sign_r[3][3] = { {0,1,0},{1,0,0},{0,1,1} };

static int temp_exponent_r[3][3] = { {10,20,11},{10,5,11},{2,20,11} };

static int temp_mantissa_r[3][3] = { {4,9,1},{0,3,4},{3,3,7} };

static int temp_sign_t[1][3] = { { 0,1,0  } };

static int temp_exponent_t[1][3] = { { 0,10,5 } };

static int temp_mantissa_t[1][3] = { { 1,8,9 } };

static int temp_length_mantissa_r[3][3] = { {10,20,11},{10,5,11},{2,20,11} };

static int temp_length_mantissa_t[1][3] = { { 1,10,5 } };

class SEIMultiviewAcquisitionInfo : public SEI

{

public:

    SEIMultiviewAcquisitionInfo()

        :sign_r(),

        exponent_r(),

        mantissa_r(),

        sign_t(),

        exponent_t(),

        mantissa_t(),

        length_mantissa_r(),

        length_mantissa_t(),

        m_intrinsicParamFlag(true),

        m_extrinsicParamFlag(true),

        m_intrinsicParamsEqualFlag(true)

    {

        m_payloadType = MULTIVIEW_ACQUISITION_INFO;

        m_payloadSize = 0;

        for (int i = 0; i <= 0; i++)

        {

            for (int j = 0; j <= 2; j++)  /* row */

            {

                for (int k = 0; k <= 2; k++)  /* column */

                {

                    sign_r[j][k] = temp_sign_r[j][k];

                    exponent_r[j][k] = temp_exponent_r[j][k];

                    mantissa_r[j][k] = temp_mantissa_r[j][k];

                    length_mantissa_r[j][k] = temp_length_mantissa_r[j][k];

                }

                sign_t[i][j] = temp_sign_t[i][j];

                exponent_t[i][j] = temp_exponent_t[i][j];

                mantissa_t[i][j] = temp_mantissa_t[i][j];

                length_mantissa_t[i][j] = temp_length_mantissa_t[i][j];

            }

        }

    }

    int sign_r[3][3];

    int exponent_r[3][3];

    int mantissa_r[3][3];

    int sign_t[1][3];

    int exponent_t[1][3];

    int mantissa_t[1][3];

    int length_mantissa_r[3][3];

    int length_mantissa_t[1][3];

    bool m_intrinsicParamFlag;

    bool m_extrinsicParamFlag;

    bool m_intrinsicParamsEqualFlag;

    void writeSEI(const SPS& sps)

    {

        WRITE_FLAG(m_intrinsicParamFlag, "intrinsic_param_flag");

        WRITE_FLAG(m_extrinsicParamFlag, "extrinsic_param_flag");

        if (m_intrinsicParamFlag)

        {

            WRITE_FLAG(m_intrinsicParamsEqualFlag, "intrinsic_params_equal_flag");

            WRITE_UVLC(31, "prec_focal_length");

            WRITE_UVLC(31, "prec_principal_point");

            WRITE_UVLC(31, "prec_skew_factor");

            for (int i = 0; i <= (m_intrinsicParamsEqualFlag ? 0 : sps.maxViews - 1); i++)

            {

                WRITE_FLAG(0, "sign_focal_length_x");

                WRITE_CODE(0, 6, "exponent_focal_length_x");

                WRITE_CODE(0, 1, "mantissa_focal_length_x");

                WRITE_FLAG(0, "sign_focal_length_y");

                WRITE_CODE(0, 6, "exponent_focal_length_y");

                WRITE_CODE(0, 1, "mantissa_focal_length_y");

                WRITE_FLAG(0, "sign_principal_point_x");

                WRITE_CODE(0, 6, "exponent_principal_point_x");

                WRITE_CODE(0, 1, "mantissa_principal_point_x");

                WRITE_FLAG(0, "sign_principal_point_y");

                WRITE_CODE(0, 6, "exponent_principal_point_y");

                WRITE_CODE(0, 1, "mantissa_principal_point_y");

                WRITE_FLAG(0, "sign_skew_factor");

                WRITE_CODE(0, 6, "exponent_skew_factor");

                WRITE_CODE(0, 1, "mantissa_skew_factor");

            }

        }

        if (m_extrinsicParamFlag)

        {

            WRITE_UVLC(31, "prec_rotation_param");

            WRITE_UVLC(31, "prec_translation_param");

            for (int i = 0; i <= 0; i++)

            {

                for (int j = 0; j <= 2; j++)  /* row */

                {

                    for (int k = 0; k <= 2; k++)  /* column */

                    {

                        WRITE_FLAG(sign_r[j][k], "sign_r");

                        WRITE_CODE(exponent_r[j][k], 6, "exponent_r");

                        WRITE_CODE(mantissa_r[j][k], length_mantissa_r[j][k], "mantissa_r");

                    }

                    WRITE_FLAG(sign_t[i][j], "sign_t");

                    WRITE_CODE(exponent_t[i][j], 6, "exponent_t");

                    WRITE_CODE(mantissa_t[i][j], length_mantissa_t[i][j], "mantissa_t");

                }

            }

        }

        if (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

        {

            WRITE_FLAG(1, "payload_bit_equal_to_one");

            while (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

            {

                WRITE_FLAG(0, "payload_bit_equal_to_zero");

            }

        }

    }

};

class SEIMultiviewViewPosition : public SEI

{

public:

    SEIMultiviewViewPosition()

    {

        m_payloadType = MULTIVIEW_VIEW_POSITION;

        m_payloadSize = 0;

    }

    void writeSEI(const SPS& sps)

    {

        WRITE_UVLC(sps.maxViews - 1, "num_views_minus1");

        for (int i = 0; i <= sps.maxViews - 1; i++)

        {

            WRITE_UVLC(!i, "view_position");

        }

        if (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

        {

            WRITE_FLAG(1, "payload_bit_equal_to_one");

            while (m_bitIf->getNumberOfWrittenBits() % X265_BYTE != 0)

            {

                WRITE_FLAG(0, "payload_bit_equal_to_zero");

            }

        }

    }

};

#endif

class SEIMasteringDisplayColorVolume : public SEI

{

public:

    SEIMasteringDisplayColorVolume()

    {

        m_payloadType = MASTERING_DISPLAY_INFO;

        m_payloadSize = (8 * 2 + 2 * 4);

    }

    uint16_t displayPrimaryX[3];

    uint16_t displayPrimaryY[3];

    uint16_t whitePointX, whitePointY;

    uint32_t maxDisplayMasteringLuminance;

    uint32_t minDisplayMasteringLuminance;

    bool parse(const char* value)

    {

        return sscanf(value, "G(%hu,%hu)B(%hu,%hu)R(%hu,%hu)WP(%hu,%hu)L(%u,%u)",

                      &displayPrimaryX[0], &displayPrimaryY[0],

                      &displayPrimaryX[1], &displayPrimaryY[1],

                      &displayPrimaryX[2], &displayPrimaryY[2],

                      &whitePointX, &whitePointY,

                      &maxDisplayMasteringLuminance, &minDisplayMasteringLuminance) == 10;

    }

    void writeSEI(const SPS&)

    {

        for (uint32_t i = 0; i < 3; i++)

        {

            WRITE_CODE(displayPrimaryX[i], 16, "display_primaries_x[ c ]");

            WRITE_CODE(displayPrimaryY[i], 16, "display_primaries_y[ c ]");

        }

        WRITE_CODE(whitePointX, 16, "white_point_x");

        WRITE_CODE(whitePointY, 16, "white_point_y");

        WRITE_CODE(maxDisplayMasteringLuminance, 32, "max_display_mastering_luminance");

        WRITE_CODE(minDisplayMasteringLuminance, 32, "min_display_mastering_luminance");

    }

};

class SEIContentLightLevel : public SEI

{

public:

    SEIContentLightLevel()

    {

        m_payloadType = CONTENT_LIGHT_LEVEL_INFO;

        m_payloadSize = 4;

    }

    uint16_t max_content_light_level;

    uint16_t max_pic_average_light_level;

    void writeSEI(const SPS&)

    {

        WRITE_CODE(max_content_light_level,     16, "max_content_light_level");

        WRITE_CODE(max_pic_average_light_level, 16, "max_pic_average_light_level");

    }

};

class SEIDecodedPictureHash : public SEI

{

public:

    SEIDecodedPictureHash()

    {

        m_payloadType = DECODED_PICTURE_HASH;

        m_payloadSize = 0;

    }

    enum Method

    {

        MD5,

        CRC,

        CHECKSUM,

    } m_method;

    MD5Context m_state[3];

    uint32_t   m_crc[3];

    uint32_t   m_checksum[3];

    uint8_t    m_digest[3][16];

    void writeSEI(const SPS& sps)

    {

        int planes = (sps.chromaFormatIdc != X265_CSP_I400) ? 3 : 1;

        WRITE_CODE(m_method, 8, "hash_type");

        for (int yuvIdx = 0; yuvIdx < planes; yuvIdx++)

        {

            if (m_method == MD5)

            {

                for (uint32_t i = 0; i < 16; i++)

                    WRITE_CODE(m_digest[yuvIdx][i], 8, "picture_md5");

            }

            else if (m_method == CRC)

            {

                uint32_t val = (m_digest[yuvIdx][0] << 8) + m_digest[yuvIdx][1];

                WRITE_CODE(val, 16, "picture_crc");

            }

            else if (m_method == CHECKSUM)

            {

                uint32_t val = (m_digest[yuvIdx][0] << 24) + (m_digest[yuvIdx][1] << 16) + (m_digest[yuvIdx][2] << 8) + m_digest[yuvIdx][3];

                WRITE_CODE(val, 32, "picture_checksum");

            }

        }

    }

};

class SEIActiveParameterSets : public SEI

{

public:

    SEIActiveParameterSets()

    {

        m_payloadType = ACTIVE_PARAMETER_SETS;

        m_payloadSize = 0;

    }

    bool m_selfContainedCvsFlag;

    bool m_noParamSetUpdateFlag;

    void writeSEI(const SPS&)

    {

        WRITE_CODE(0, 4, "active_vps_id");

        WRITE_FLAG(m_selfContainedCvsFlag, "self_contained_cvs_flag");

        WRITE_FLAG(m_noParamSetUpdateFlag, "no_param_set_update_flag");

        WRITE_UVLC(0, "num_sps_ids_minus1");

        WRITE_UVLC(0, "active_seq_param_set_id");

        writeByteAlign();

    }

};

class SEIBufferingPeriod : public SEI

{

public:

    SEIBufferingPeriod()

        : m_cpbDelayOffset(0)

        , m_dpbDelayOffset(0)

        , m_concatenationFlag(0)

        , m_auCpbRemovalDelayDelta(1)

    {

        m_payloadType = BUFFERING_PERIOD;

        m_payloadSize = 0;

    }

    bool     m_cpbDelayOffset;

    bool     m_dpbDelayOffset;

    bool     m_concatenationFlag;

    uint32_t m_initialCpbRemovalDelay;

    uint32_t m_initialCpbRemovalDelayOffset;

    uint32_t m_auCpbRemovalDelayDelta;

    void writeSEI(const SPS& sps)

    {

        const HRDInfo& hrd = sps.vuiParameters.hrdParameters;

        WRITE_UVLC(0, "bp_seq_parameter_set_id");

        WRITE_FLAG(0, "rap_cpb_params_present_flag");

        WRITE_FLAG(m_concatenationFlag, "concatenation_flag");

        WRITE_CODE(m_auCpbRemovalDelayDelta - 1,   hrd.cpbRemovalDelayLength,       "au_cpb_removal_delay_delta_minus1");

        WRITE_CODE(m_initialCpbRemovalDelay,       hrd.initialCpbRemovalDelayLength,        "initial_cpb_removal_delay");

        WRITE_CODE(m_initialCpbRemovalDelayOffset, hrd.initialCpbRemovalDelayLength, "initial_cpb_removal_delay_offset");

        writeByteAlign();

    }

};

class SEIPictureTiming : public SEI

{

public:

    SEIPictureTiming()

    {

        m_payloadType = PICTURE_TIMING;

        m_payloadSize = 0;

    }

    uint32_t  m_picStruct;

    uint32_t  m_sourceScanType;

    bool      m_duplicateFlag;

    uint32_t  m_auCpbRemovalDelay;

    uint32_t  m_picDpbOutputDelay;

    void writeSEI(const SPS& sps)

    {

        const VUI *vui = &sps.vuiParameters;

        const HRDInfo *hrd = &vui->hrdParameters;

        if (vui->frameFieldInfoPresentFlag)

        {

            WRITE_CODE(m_picStruct, 4,          "pic_struct");

            WRITE_CODE(m_sourceScanType, 2,     "source_scan_type");

            WRITE_FLAG(m_duplicateFlag,         "duplicate_flag");

        }

        if (vui->hrdParametersPresentFlag)

        {

            WRITE_CODE(m_auCpbRemovalDelay - 1, hrd->cpbRemovalDelayLength, "au_cpb_removal_delay_minus1");

            WRITE_CODE(m_picDpbOutputDelay, hrd->dpbOutputDelayLength, "pic_dpb_output_delay");

            /* Removed sub-pic signaling June 2014 */

        }

        writeByteAlign();

    }

};

class SEIRecoveryPoint : public SEI

{

public:

    SEIRecoveryPoint()

    {

        m_payloadType = RECOVERY_POINT;

        m_payloadSize = 0;

    }

    int  m_recoveryPocCnt;

    bool m_exactMatchingFlag;

    bool m_brokenLinkFlag;

    void writeSEI(const SPS&)

    {

        WRITE_SVLC(m_recoveryPocCnt,    "recovery_poc_cnt");

        WRITE_FLAG(m_exactMatchingFlag, "exact_matching_flag");

        WRITE_FLAG(m_brokenLinkFlag,    "broken_link_flag");

        writeByteAlign();

    }

};

class SEIAlternativeTC : public SEI

{

public:

    int m_preferredTransferCharacteristics;

    SEIAlternativeTC()

    {

        m_payloadType = ALTERNATIVE_TRANSFER_CHARACTERISTICS;

        m_payloadSize = 0;

        m_preferredTransferCharacteristics = -1;

    }

    void writeSEI(const SPS&)

    {

        WRITE_CODE(m_preferredTransferCharacteristics, 8, "Preferred transfer characteristics");

    }

};

}

#endif // ifndef X265_SEI_H