/*

 * H.265 video codec.

 * Copyright (c) 2013-2014 struktur AG, Dirk Farin <farin@struktur.de>

 *

 * Authors: struktur AG, Dirk Farin <farin@struktur.de>

 *          Min Chen <chenm003@163.com>

 *

 * This file is part of libde265.

 *

 * libde265 is free software: you can redistribute it and/or modify

 * it under the terms of the GNU Lesser General Public License as

 * published by the Free Software Foundation, either version 3 of

 * the License, or (at your option) any later version.

 *

 * libde265 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 Lesser General Public License for more details.

 *

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

 * along with libde265.  If not, see <http://www.gnu.org/licenses/>.

 */

#ifndef DE265_SLICE_H

#define DE265_SLICE_H

#include "libde265/cabac.h"

#include "libde265/de265.h"

#include "libde265/util.h"

#include "libde265/refpic.h"

#include "libde265/threads.h"

#include "contextmodel.h"

#include <vector>

#include <string.h>

#include <memory>

class decoder_context;

class thread_context;

class error_queue;

class seq_parameter_set;

class pic_parameter_set;

enum SliceType

  {

    SLICE_TYPE_B = 0,

    SLICE_TYPE_P = 1,

    SLICE_TYPE_I = 2

  };

/*

        2Nx2N           2NxN             Nx2N            NxN

      +-------+       +-------+       +---+---+       +---+---+

      |       |       |       |       |   |   |       |   |   |

      |       |       |_______|       |   |   |       |___|___|

      |       |       |       |       |   |   |       |   |   |

      |       |       |       |       |   |   |       |   |   |

      +-------+       +-------+       +---+---+       +---+---+

        2NxnU           2NxnD           nLx2N           nRx2N

      +-------+       +-------+       +-+-----+       +-----+-+

      |_______|       |       |       | |     |       |     | |

      |       |       |       |       | |     |       |     | |

      |       |       |_______|       | |     |       |     | |

      |       |       |       |       | |     |       |     | |

      +-------+       +-------+       +-+-----+       +-----+-+

      - AMP only if CU size > min CU size -> minimum PU size = CUsize/2

      - NxN only if size >= 16x16 (-> minimum block size = 8x8)

      - minimum block size for Bi-Pred is 8x8 (wikipedia: Coding_tree_unit)

*/

enum PartMode

  {

    PART_2Nx2N = 0,

    PART_2NxN  = 1,

    PART_Nx2N  = 2,

    PART_NxN   = 3,

    PART_2NxnU = 4,

    PART_2NxnD = 5,

    PART_nLx2N = 6,

    PART_nRx2N = 7

  };

const char* part_mode_name(PartMode);

enum PredMode

  {

    MODE_INTRA, MODE_INTER, MODE_SKIP

  };

enum IntraPredMode

  {

    INTRA_PLANAR = 0,

    INTRA_DC = 1,

    INTRA_ANGULAR_2 = 2,    INTRA_ANGULAR_3 = 3,    INTRA_ANGULAR_4 = 4,    INTRA_ANGULAR_5 = 5,

    INTRA_ANGULAR_6 = 6,    INTRA_ANGULAR_7 = 7,    INTRA_ANGULAR_8 = 8,    INTRA_ANGULAR_9 = 9,

    INTRA_ANGULAR_10 = 10,  INTRA_ANGULAR_11 = 11,  INTRA_ANGULAR_12 = 12,  INTRA_ANGULAR_13 = 13,

    INTRA_ANGULAR_14 = 14,  INTRA_ANGULAR_15 = 15,  INTRA_ANGULAR_16 = 16,  INTRA_ANGULAR_17 = 17,

    INTRA_ANGULAR_18 = 18,  INTRA_ANGULAR_19 = 19,  INTRA_ANGULAR_20 = 20,  INTRA_ANGULAR_21 = 21,

    INTRA_ANGULAR_22 = 22,  INTRA_ANGULAR_23 = 23,  INTRA_ANGULAR_24 = 24,  INTRA_ANGULAR_25 = 25,

    INTRA_ANGULAR_26 = 26,  INTRA_ANGULAR_27 = 27,  INTRA_ANGULAR_28 = 28,  INTRA_ANGULAR_29 = 29,

    INTRA_ANGULAR_30 = 30,  INTRA_ANGULAR_31 = 31,  INTRA_ANGULAR_32 = 32,  INTRA_ANGULAR_33 = 33,

    INTRA_ANGULAR_34 = 34

  };

enum IntraChromaPredMode

  {

    INTRA_CHROMA_PLANAR_OR_34     = 0,

    INTRA_CHROMA_ANGULAR_26_OR_34 = 1,

    INTRA_CHROMA_ANGULAR_10_OR_34 = 2,

    INTRA_CHROMA_DC_OR_34         = 3,

    INTRA_CHROMA_LIKE_LUMA  = 4

  };

enum InterPredIdc

  {

    // note: values have to match the decoding function decode_inter_pred_idc()

    PRED_L0=1,

    PRED_L1=2,

    PRED_BI=3

  };

class slice_segment_header {

public:

  slice_segment_header() {

    reset();

  }

  de265_error read(bitreader* br, decoder_context*, bool* continueDecoding);

  de265_error write(error_queue*, CABAC_encoder&,

                    const seq_parameter_set* sps,

                    const pic_parameter_set* pps,

                    uint8_t nal_unit_type);

  void dump_slice_segment_header(const decoder_context*, int fd) const;

  void set_defaults();

  void reset();

  int  slice_index; // index through all slices in a picture  (internal only)

  std::shared_ptr<const pic_parameter_set> pps;

  bool first_slice_segment_in_pic_flag;

  bool no_output_of_prior_pics_flag;

  uint8_t slice_pic_parameter_set_id; // [0;63]

  bool dependent_slice_segment_flag;

  uint32_t slice_segment_address;

  uint8_t slice_type;                  // [0;2]

  bool pic_output_flag;

  char colour_plane_id;

  int  slice_pic_order_cnt_lsb;

  bool short_term_ref_pic_set_sps_flag;

  ref_pic_set slice_ref_pic_set;

  uint8_t short_term_ref_pic_set_idx;

  uint8_t num_long_term_sps;        // [0;32]

  uint8_t num_long_term_pics;       // [0;32]

  uint8_t lt_idx_sps[MAX_NUM_REF_PICS];

  int     poc_lsb_lt[MAX_NUM_REF_PICS];

  bool    used_by_curr_pic_lt_flag[MAX_NUM_REF_PICS];

  bool delta_poc_msb_present_flag[MAX_NUM_REF_PICS];

  uint32_t delta_poc_msb_cycle_lt[MAX_NUM_REF_PICS];

  bool slice_temporal_mvp_enabled_flag;

  bool slice_sao_luma_flag;

  bool slice_sao_chroma_flag;

  bool num_ref_idx_active_override_flag;

  uint8_t num_ref_idx_l0_active; // [1;16]

  uint8_t num_ref_idx_l1_active; // [1;16]

  bool ref_pic_list_modification_flag_l0;

  bool ref_pic_list_modification_flag_l1;

  uint8_t list_entry_l0[16];

  uint8_t list_entry_l1[16];

  bool mvd_l1_zero_flag;

  bool cabac_init_flag;

  bool collocated_from_l0_flag;

  uint8_t collocated_ref_idx;         // [0;15]

  // --- pred_weight_table ---

  uint8_t luma_log2_weight_denom; // [0;7]

  uint8_t ChromaLog2WeightDenom;  // [0;7]

  // first index is L0/L1

  uint8_t luma_weight_flag[2][16];   // bool

  uint8_t chroma_weight_flag[2][16]; // bool

  int16_t LumaWeight[2][16];

  int16_t luma_offset[2][16];

  int16_t ChromaWeight[2][16][2];

  int16_t ChromaOffset[2][16][2];

  uint8_t five_minus_max_num_merge_cand; // [0;5]

  int  slice_qp_delta;

  int  slice_cb_qp_offset;

  int  slice_cr_qp_offset;

  bool cu_chroma_qp_offset_enabled_flag;

  bool deblocking_filter_override_flag;

  bool slice_deblocking_filter_disabled_flag;

  int8_t slice_beta_offset; // [-12;12], = pps->beta_offset if undefined

  int8_t slice_tc_offset;   // [-12;12], = pps->tc_offset if undefined

  bool slice_loop_filter_across_slices_enabled_flag;

  uint32_t num_entry_point_offsets;

  int      offset_len;

  std::vector<uint32_t> entry_point_offset;

  int  slice_segment_header_extension_length;

  // --- derived data ---

  int SliceQPY;

  int initType;

  void compute_derived_values(const pic_parameter_set* pps);

  // --- data for external modules ---

  uint32_t SliceAddrRS;  // slice_segment_address of last independent slice

  int MaxNumMergeCand;  // directly derived from 'five_minus_max_num_merge_cand'

  int CurrRpsIdx;

  ref_pic_set CurrRps;  // the active reference-picture set

  int NumPocTotalCurr;

  // number of entries: num_ref_idx_l0_active / num_ref_idx_l1_active

  int RefPicList[2][MAX_NUM_REF_PICS]; // contains buffer IDs (D:indices into DPB/E:frame number)

  int RefPicList_POC[2][MAX_NUM_REF_PICS];

  int RefPicList_PicState[2][MAX_NUM_REF_PICS]; /* We have to save the PicState because the decoding

                                                   of an image may be delayed and the PicState can

                                                   change in the mean-time (e.g. from ShortTerm to

                                                   LongTerm). PicState is used in motion.cc */

  bool LongTermRefPic[2][MAX_NUM_REF_PICS]; /* Flag whether the picture at this ref-pic-list

                                               is a long-term picture. */

  // context storage for dependent slices (stores CABAC model at end of slice segment)

  context_model_table ctx_model_storage;

  // StatCoeff[] (persistent_rice_adaptation state) saved alongside ctx_model_storage

  uint8_t ctx_model_storage_StatCoeff[4];

  bool ctx_model_storage_defined; // whether there is valid data in ctx_model_storage

  std::vector<int> RemoveReferencesList; // images that can be removed from the DPB before decoding this slice

};

struct sao_info {

  // TODO: we could combine SaoTypeIdx and SaoEoClass into one byte to make the struct 16 bytes only

  unsigned char SaoTypeIdx; // use with (SaoTypeIdx>>(2*cIdx)) & 0x3

  unsigned char SaoEoClass; // use with (SaoTypeIdx>>(2*cIdx)) & 0x3

  uint8_t sao_band_position[3];

  int8_t  saoOffsetVal[3][4]; // index with [][idx-1] as saoOffsetVal[][0]==0 always

};

de265_error read_slice_segment_data(thread_context* tctx);

bool alloc_and_init_significant_coeff_ctxIdx_lookupTable();

void free_significant_coeff_ctxIdx_lookupTable();

class thread_task_ctb_row : public thread_task

{

public:

  bool   firstSliceSubstream;

  uint16_t debug_startCtbRow;

  thread_context* tctx;

  void work() override;

  std::string name() const override;

};

class thread_task_slice_segment : public thread_task

{

public:

  bool   firstSliceSubstream;

  uint16_t debug_startCtbX, debug_startCtbY;

  thread_context* tctx;

  void work() override;

  std::string name() const override;

};

int check_CTB_available(const de265_image* img,

                        int xC,int yC, int xN,int yN);

#endif