/*

 * Copyright (c) 2019, Alliance for Open Media. All rights reserved.

 *

 * This source code is subject to the terms of the BSD 2 Clause License and

 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License

 * was not distributed with this source code in the LICENSE file, you can

 * obtain it at www.aomedia.org/license/software. If the Alliance for Open

 * Media Patent License 1.0 was not distributed with this source code in the

 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.

 */

#include "av1/encoder/encoder.h"

#include "av1/encoder/level.h"

#define UNDEFINED_LEVEL     \

  { .level = SEQ_LEVEL_MAX, \

    .max_picture_size = 0,  \

    .max_h_size = 0,        \

    .max_v_size = 0,        \

    .max_display_rate = 0,  \

    .max_decode_rate = 0,   \

    .max_header_rate = 0,   \

    .main_mbps = 0,         \

    .high_mbps = 0,         \

    .main_cr = 0,           \

    .high_cr = 0,           \

    .max_tiles = 0,         \

    .max_tile_cols = 0 }

static const AV1LevelSpec av1_level_defs[SEQ_LEVELS] = {

  { .level = SEQ_LEVEL_2_0,

    .max_picture_size = 147456,

    .max_h_size = 2048,

    .max_v_size = 1152,

    .max_display_rate = 4423680L,

    .max_decode_rate = 5529600L,

    .max_header_rate = 150,

    .main_mbps = 1.5,

    .high_mbps = 0,

    .main_cr = 2.0,

    .high_cr = 0,

    .max_tiles = 8,

    .max_tile_cols = 4 },

  { .level = SEQ_LEVEL_2_1,

    .max_picture_size = 278784,

    .max_h_size = 2816,

    .max_v_size = 1584,

    .max_display_rate = 8363520L,

    .max_decode_rate = 10454400L,

    .max_header_rate = 150,

    .main_mbps = 3.0,

    .high_mbps = 0,

    .main_cr = 2.0,

    .high_cr = 0,

    .max_tiles = 8,

    .max_tile_cols = 4 },

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  { .level = SEQ_LEVEL_3_0,

    .max_picture_size = 665856,

    .max_h_size = 4352,

    .max_v_size = 2448,

    .max_display_rate = 19975680L,

    .max_decode_rate = 24969600L,

    .max_header_rate = 150,

    .main_mbps = 6.0,

    .high_mbps = 0,

    .main_cr = 2.0,

    .high_cr = 0,

    .max_tiles = 16,

    .max_tile_cols = 6 },

  { .level = SEQ_LEVEL_3_1,

    .max_picture_size = 1065024,

    .max_h_size = 5504,

    .max_v_size = 3096,

    .max_display_rate = 31950720L,

    .max_decode_rate = 39938400L,

    .max_header_rate = 150,

    .main_mbps = 10.0,

    .high_mbps = 0,

    .main_cr = 2.0,

    .high_cr = 0,

    .max_tiles = 16,

    .max_tile_cols = 6 },

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  { .level = SEQ_LEVEL_4_0,

    .max_picture_size = 2359296,

    .max_h_size = 6144,

    .max_v_size = 3456,

    .max_display_rate = 70778880L,

    .max_decode_rate = 77856768L,

    .max_header_rate = 300,

    .main_mbps = 12.0,

    .high_mbps = 30.0,

    .main_cr = 4.0,

    .high_cr = 4.0,

    .max_tiles = 32,

    .max_tile_cols = 8 },

  { .level = SEQ_LEVEL_4_1,

    .max_picture_size = 2359296,

    .max_h_size = 6144,

    .max_v_size = 3456,

    .max_display_rate = 141557760L,

    .max_decode_rate = 155713536L,

    .max_header_rate = 300,

    .main_mbps = 20.0,

    .high_mbps = 50.0,

    .main_cr = 4.0,

    .high_cr = 4.0,

    .max_tiles = 32,

    .max_tile_cols = 8 },

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  { .level = SEQ_LEVEL_5_0,

    .max_picture_size = 8912896,

    .max_h_size = 8192,

    .max_v_size = 4352,

    .max_display_rate = 267386880L,

    .max_decode_rate = 273715200L,

    .max_header_rate = 300,

    .main_mbps = 30.0,

    .high_mbps = 100.0,

    .main_cr = 6.0,

    .high_cr = 4.0,

    .max_tiles = 64,

    .max_tile_cols = 8 },

  { .level = SEQ_LEVEL_5_1,

    .max_picture_size = 8912896,

    .max_h_size = 8192,

    .max_v_size = 4352,

    .max_display_rate = 534773760L,

    .max_decode_rate = 547430400L,

    .max_header_rate = 300,

    .main_mbps = 40.0,

    .high_mbps = 160.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 64,

    .max_tile_cols = 8 },

  { .level = SEQ_LEVEL_5_2,

    .max_picture_size = 8912896,

    .max_h_size = 8192,

    .max_v_size = 4352,

    .max_display_rate = 1069547520L,

    .max_decode_rate = 1094860800L,

    .max_header_rate = 300,

    .main_mbps = 60.0,

    .high_mbps = 240.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 64,

    .max_tile_cols = 8 },

  { .level = SEQ_LEVEL_5_3,

    .max_picture_size = 8912896,

    .max_h_size = 8192,

    .max_v_size = 4352,

    .max_display_rate = 1069547520L,

    .max_decode_rate = 1176502272L,

    .max_header_rate = 300,

    .main_mbps = 60.0,

    .high_mbps = 240.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 64,

    .max_tile_cols = 8 },

  { .level = SEQ_LEVEL_6_0,

    .max_picture_size = 35651584,

    .max_h_size = 16384,

    .max_v_size = 8704,

    .max_display_rate = 1069547520L,

    .max_decode_rate = 1176502272L,

    .max_header_rate = 300,

    .main_mbps = 60.0,

    .high_mbps = 240.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 128,

    .max_tile_cols = 16 },

  { .level = SEQ_LEVEL_6_1,

    .max_picture_size = 35651584,

    .max_h_size = 16384,

    .max_v_size = 8704,

    .max_display_rate = 2139095040L,

    .max_decode_rate = 2189721600L,

    .max_header_rate = 300,

    .main_mbps = 100.0,

    .high_mbps = 480.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 128,

    .max_tile_cols = 16 },

  { .level = SEQ_LEVEL_6_2,

    .max_picture_size = 35651584,

    .max_h_size = 16384,

    .max_v_size = 8704,

    .max_display_rate = 4278190080L,

    .max_decode_rate = 4379443200L,

    .max_header_rate = 300,

    .main_mbps = 160.0,

    .high_mbps = 800.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 128,

    .max_tile_cols = 16 },

  { .level = SEQ_LEVEL_6_3,

    .max_picture_size = 35651584,

    .max_h_size = 16384,

    .max_v_size = 8704,

    .max_display_rate = 4278190080L,

    .max_decode_rate = 4706009088L,

    .max_header_rate = 300,

    .main_mbps = 160.0,

    .high_mbps = 800.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 128,

    .max_tile_cols = 16 },

#if CONFIG_CWG_C013

  { .level = SEQ_LEVEL_7_0,

    .max_picture_size = 142606336,

    .max_h_size = 32768,

    .max_v_size = 17408,

    .max_display_rate = 4278190080L,

    .max_decode_rate = 4706009088L,

    .max_header_rate = 300,

    .main_mbps = 160.0,

    .high_mbps = 800.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 256,

    .max_tile_cols = 32 },

  { .level = SEQ_LEVEL_7_1,

    .max_picture_size = 142606336,

    .max_h_size = 32768,

    .max_v_size = 17408,

    .max_display_rate = 8556380160L,

    .max_decode_rate = 8758886400L,

    .max_header_rate = 300,

    .main_mbps = 200.0,

    .high_mbps = 960.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 256,

    .max_tile_cols = 32 },

  { .level = SEQ_LEVEL_7_2,

    .max_picture_size = 142606336,

    .max_h_size = 32768,

    .max_v_size = 17408,

    .max_display_rate = 17112760320L,

    .max_decode_rate = 17517772800L,

    .max_header_rate = 300,

    .main_mbps = 320.0,

    .high_mbps = 1600.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 256,

    .max_tile_cols = 32 },

  { .level = SEQ_LEVEL_7_3,

    .max_picture_size = 142606336,

    .max_h_size = 32768,

    .max_v_size = 17408,

    .max_display_rate = 17112760320L,

    .max_decode_rate = 18824036352L,

    .max_header_rate = 300,

    .main_mbps = 320.0,

    .high_mbps = 1600.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 256,

    .max_tile_cols = 32 },

  { .level = SEQ_LEVEL_8_0,

    .max_picture_size = 530841600,

    .max_h_size = 65536,

    .max_v_size = 34816,

    .max_display_rate = 17112760320L,

    .max_decode_rate = 18824036352L,

    .max_header_rate = 300,

    .main_mbps = 320.0,

    .high_mbps = 1600.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 512,

    .max_tile_cols = 64 },

  { .level = SEQ_LEVEL_8_1,

    .max_picture_size = 530841600,

    .max_h_size = 65536,

    .max_v_size = 34816,

    .max_display_rate = 34225520640L,

    .max_decode_rate = 34910031052L,

    .max_header_rate = 300,

    .main_mbps = 400.0,

    .high_mbps = 1920.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 512,

    .max_tile_cols = 64 },

  { .level = SEQ_LEVEL_8_2,

    .max_picture_size = 530841600,

    .max_h_size = 65536,

    .max_v_size = 34816,

    .max_display_rate = 68451041280L,

    .max_decode_rate = 69820062105L,

    .max_header_rate = 300,

    .main_mbps = 640.0,

    .high_mbps = 3200.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 512,

    .max_tile_cols = 64 },

  { .level = SEQ_LEVEL_8_3,

    .max_picture_size = 530841600,

    .max_h_size = 65536,

    .max_v_size = 34816,

    .max_display_rate = 68451041280L,

    .max_decode_rate = 75296145408L,

    .max_header_rate = 300,

    .main_mbps = 640.0,

    .high_mbps = 3200.0,

    .main_cr = 8.0,

    .high_cr = 4.0,

    .max_tiles = 512,

    .max_tile_cols = 64 },

#else   // !CONFIG_CWG_C013

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

  UNDEFINED_LEVEL,

#endif  // CONFIG_CWG_C013

};

typedef enum {

  LUMA_PIC_SIZE_TOO_LARGE,

  LUMA_PIC_H_SIZE_TOO_LARGE,

  LUMA_PIC_V_SIZE_TOO_LARGE,

  LUMA_PIC_H_SIZE_TOO_SMALL,

  LUMA_PIC_V_SIZE_TOO_SMALL,

  TOO_MANY_TILE_COLUMNS,

  TOO_MANY_TILES,

  TILE_RATE_TOO_HIGH,

  TILE_TOO_LARGE,

  SUPERRES_TILE_WIDTH_TOO_LARGE,

  CROPPED_TILE_WIDTH_TOO_SMALL,

  CROPPED_TILE_HEIGHT_TOO_SMALL,

  TILE_WIDTH_INVALID,

  FRAME_HEADER_RATE_TOO_HIGH,

  DISPLAY_RATE_TOO_HIGH,

  DECODE_RATE_TOO_HIGH,

  CR_TOO_SMALL,

  TILE_SIZE_HEADER_RATE_TOO_HIGH,

  BITRATE_TOO_HIGH,

  DECODER_MODEL_FAIL,

  TARGET_LEVEL_FAIL_IDS,

  TARGET_LEVEL_OK,

} TARGET_LEVEL_FAIL_ID;

static const char *level_fail_messages[TARGET_LEVEL_FAIL_IDS] = {

  "The picture size is too large.",

  "The picture width is too large.",

  "The picture height is too large.",

  "The picture width is too small.",

  "The picture height is too small.",

  "Too many tile columns are used.",

  "Too many tiles are used.",

  "The tile rate is too high.",

  "The tile size is too large.",

  "The superres tile width is too large.",

  "The cropped tile width is less than 8.",

  "The cropped tile height is less than 8.",

  "The tile width is invalid.",

  "The frame header rate is too high.",

  "The display luma sample rate is too high.",

  "The decoded luma sample rate is too high.",

  "The compression ratio is too small.",

  "The product of max tile size and header rate is too high.",

  "The bitrate is too high.",

  "The decoder model fails.",

};

static double get_max_bitrate(const AV1LevelSpec *const level_spec, int tier,

                              BITSTREAM_PROFILE profile) {

  if (level_spec->level < SEQ_LEVEL_4_0) tier = 0;

  const double bitrate_basis =

      (tier ? level_spec->high_mbps : level_spec->main_mbps) * 1e6;

  const double bitrate_profile_factor =

      profile == PROFILE_0 ? 1.0 : (profile == PROFILE_1 ? 2.0 : 3.0);

  return bitrate_basis * bitrate_profile_factor;

}

double av1_get_max_bitrate_for_level(AV1_LEVEL level_index, int tier,

                                     BITSTREAM_PROFILE profile) {

  assert(is_valid_seq_level_idx(level_index));

  return get_max_bitrate(&av1_level_defs[level_index], tier, profile);

}

void av1_get_max_tiles_for_level(AV1_LEVEL level_index, int *const max_tiles,

                                 int *const max_tile_cols) {

  assert(is_valid_seq_level_idx(level_index));

  const AV1LevelSpec *const level_spec = &av1_level_defs[level_index];

  *max_tiles = level_spec->max_tiles;

  *max_tile_cols = level_spec->max_tile_cols;

}

// We assume time t to be valid if and only if t >= 0.0.

// So INVALID_TIME can be defined as anything less than 0.

#define INVALID_TIME (-1.0)

// This corresponds to "free_buffer" in the spec.

static void release_buffer(DECODER_MODEL *const decoder_model, int idx) {

  assert(idx >= 0 && idx < BUFFER_POOL_MAX_SIZE);

  FRAME_BUFFER *const this_buffer = &decoder_model->frame_buffer_pool[idx];

  this_buffer->decoder_ref_count = 0;

  this_buffer->player_ref_count = 0;

  this_buffer->display_index = -1;

  this_buffer->presentation_time = INVALID_TIME;

}

static void initialize_buffer_pool(DECODER_MODEL *const decoder_model) {

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

    release_buffer(decoder_model, i);

  }

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

    decoder_model->vbi[i] = -1;

  }

}

static int get_free_buffer(DECODER_MODEL *const decoder_model) {

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

    const FRAME_BUFFER *const this_buffer =

        &decoder_model->frame_buffer_pool[i];

    if (this_buffer->decoder_ref_count == 0 &&

        this_buffer->player_ref_count == 0)

      return i;

  }

  return -1;

}

static void update_ref_buffers(DECODER_MODEL *const decoder_model, int idx,

                               int refresh_frame_flags) {

  FRAME_BUFFER *const this_buffer = &decoder_model->frame_buffer_pool[idx];

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

    if (refresh_frame_flags & (1 << i)) {

      const int pre_idx = decoder_model->vbi[i];

      if (pre_idx != -1) {

        --decoder_model->frame_buffer_pool[pre_idx].decoder_ref_count;

      }

      decoder_model->vbi[i] = idx;

      ++this_buffer->decoder_ref_count;

    }

  }

}

// The time (in seconds) required to decode a frame.

static double time_to_decode_frame(const AV1_COMMON *const cm,

                                   int64_t max_decode_rate) {

  if (cm->show_existing_frame) return 0.0;

  const FRAME_TYPE frame_type = cm->current_frame.frame_type;

  int luma_samples = 0;

  if (frame_type == KEY_FRAME || frame_type == INTRA_ONLY_FRAME) {

    luma_samples = cm->superres_upscaled_width * cm->height;

  } else {

    const int spatial_layer_dimensions_present_flag = 0;

    if (spatial_layer_dimensions_present_flag) {

      assert(0 && "Spatial layer dimensions not supported yet.");

    } else {

      const SequenceHeader *const seq_params = cm->seq_params;

      const int max_frame_width = seq_params->max_frame_width;

      const int max_frame_height = seq_params->max_frame_height;

      luma_samples = max_frame_width * max_frame_height;

    }

  }

  return luma_samples / (double)max_decode_rate;

}

// Release frame buffers that are no longer needed for decode or display.

// It corresponds to "start_decode_at_removal_time" in the spec.

static void release_processed_frames(DECODER_MODEL *const decoder_model,

                                     double removal_time) {

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

    FRAME_BUFFER *const this_buffer = &decoder_model->frame_buffer_pool[i];

    if (this_buffer->player_ref_count > 0) {

      if (this_buffer->presentation_time >= 0.0 &&

          this_buffer->presentation_time <= removal_time) {

        this_buffer->player_ref_count = 0;

        if (this_buffer->decoder_ref_count == 0) {

          release_buffer(decoder_model, i);

        }

      }

    }

  }

}

static int frames_in_buffer_pool(const DECODER_MODEL *const decoder_model) {

  int frames_in_pool = 0;

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

    const FRAME_BUFFER *const this_buffer =

        &decoder_model->frame_buffer_pool[i];

    if (this_buffer->decoder_ref_count > 0 ||

        this_buffer->player_ref_count > 0) {

      ++frames_in_pool;

    }

  }

  return frames_in_pool;

}

static double get_presentation_time(const DECODER_MODEL *const decoder_model,

                                    int display_index) {

  if (decoder_model->mode == SCHEDULE_MODE) {

    assert(0 && "SCHEDULE_MODE NOT SUPPORTED");

    return INVALID_TIME;

  } else {

    const double initial_presentation_delay =

        decoder_model->initial_presentation_delay;

    // Can't decide presentation time until the initial presentation delay is

    // known.

    if (initial_presentation_delay < 0.0) return INVALID_TIME;

    return initial_presentation_delay +

           display_index * decoder_model->num_ticks_per_picture *

               decoder_model->display_clock_tick;

  }

}

#define MAX_TIME 1e16

static double time_next_buffer_is_free(int num_decoded_frame,

                                       int decoder_buffer_delay,

                                       const FRAME_BUFFER *frame_buffer_pool,

                                       double current_time) {

  if (num_decoded_frame == 0) {

    return (double)decoder_buffer_delay / 90000.0;

  }

  double buf_free_time = MAX_TIME;

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

    const FRAME_BUFFER *const this_buffer = &frame_buffer_pool[i];

    if (this_buffer->decoder_ref_count == 0) {

      if (this_buffer->player_ref_count == 0) {

        return current_time;

      }

      const double presentation_time = this_buffer->presentation_time;

      if (presentation_time >= 0.0 && presentation_time < buf_free_time) {

        buf_free_time = presentation_time;

      }

    }

  }

  return buf_free_time < MAX_TIME ? buf_free_time : INVALID_TIME;

}

#undef MAX_TIME

static double get_removal_time(int mode, int num_decoded_frame,

                               int decoder_buffer_delay,

                               const FRAME_BUFFER *frame_buffer_pool,

                               double current_time) {

  if (mode == SCHEDULE_MODE) {

    assert(0 && "SCHEDULE_MODE IS NOT SUPPORTED YET");

    return INVALID_TIME;

  } else {

    return time_next_buffer_is_free(num_decoded_frame, decoder_buffer_delay,

                                    frame_buffer_pool, current_time);

  }

}

#if 0

// Print the status of the decoder model (for debugging).

void av1_decoder_model_print_status(const DECODER_MODEL *const decoder_model) {

  printf(

      "\n status %d, num_frame %3d, num_decoded_frame %3d, "

      "num_shown_frame %3d, current time %6.2f, frames in buffer %2d, "

      "presentation delay %6.2f, total interval %6.2f\n",

      decoder_model->status, decoder_model->num_frame,

      decoder_model->num_decoded_frame, decoder_model->num_shown_frame,

      decoder_model->current_time, frames_in_buffer_pool(decoder_model),

      decoder_model->initial_presentation_delay,

      decoder_model->dfg_interval_queue.total_interval);

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

    const FRAME_BUFFER *const this_buffer =

        &decoder_model->frame_buffer_pool[i];

    printf("buffer %d, decode count %d, display count %d, present time %6.4f\n",

           i, this_buffer->decoder_ref_count, this_buffer->player_ref_count,

           this_buffer->presentation_time);

  }

}

#endif

// op_index is the operating point index.

static void decoder_model_init(const AV1_COMP *const cpi, AV1_LEVEL level,

                               int op_index,

                               DECODER_MODEL *const decoder_model) {

  decoder_model->status = DECODER_MODEL_OK;

  decoder_model->level = level;

  const AV1_COMMON *const cm = &cpi->common;

  const SequenceHeader *const seq_params = cm->seq_params;

  decoder_model->bit_rate = get_max_bitrate(

      av1_level_defs + level, seq_params->tier[op_index], seq_params->profile);

  // TODO(huisu or anyone): implement SCHEDULE_MODE.

  decoder_model->mode = RESOURCE_MODE;

  decoder_model->encoder_buffer_delay = 20000;

  decoder_model->decoder_buffer_delay = 70000;

  decoder_model->is_low_delay_mode = false;

  decoder_model->first_bit_arrival_time = 0.0;

  decoder_model->last_bit_arrival_time = 0.0;

  decoder_model->coded_bits = 0;

  decoder_model->removal_time = INVALID_TIME;

  decoder_model->presentation_time = INVALID_TIME;

  decoder_model->decode_samples = 0;

  decoder_model->display_samples = 0;

  decoder_model->max_decode_rate = 0.0;

  decoder_model->max_display_rate = 0.0;

  decoder_model->num_frame = -1;

  decoder_model->num_decoded_frame = -1;

  decoder_model->num_shown_frame = -1;

  decoder_model->current_time = 0.0;

  initialize_buffer_pool(decoder_model);

  DFG_INTERVAL_QUEUE *const dfg_interval_queue =

      &decoder_model->dfg_interval_queue;

  dfg_interval_queue->total_interval = 0.0;

  dfg_interval_queue->head = 0;

  dfg_interval_queue->size = 0;

  if (seq_params->timing_info_present) {

    decoder_model->num_ticks_per_picture =

        seq_params->timing_info.num_ticks_per_picture;

    decoder_model->display_clock_tick =

        seq_params->timing_info.num_units_in_display_tick /

        seq_params->timing_info.time_scale;

  } else {

    decoder_model->num_ticks_per_picture = 1;

    decoder_model->display_clock_tick = 1.0 / cpi->framerate;

  }

  decoder_model->initial_display_delay =

      seq_params->op_params[op_index].initial_display_delay;

  decoder_model->initial_presentation_delay = INVALID_TIME;

  decoder_model->decode_rate = av1_level_defs[level].max_decode_rate;

}

DECODER_MODEL_STATUS av1_decoder_model_try_smooth_buf(

    const AV1_COMP *const cpi, size_t coded_bits,

    const DECODER_MODEL *const decoder_model) {

  DECODER_MODEL_STATUS status = DECODER_MODEL_OK;

  if (!decoder_model || decoder_model->status != DECODER_MODEL_OK) {

    return status;

  }

  const AV1_COMMON *const cm = &cpi->common;

  const int show_existing_frame = cm->show_existing_frame;

  size_t cur_coded_bits = decoder_model->coded_bits + coded_bits;

  int num_decoded_frame = decoder_model->num_decoded_frame;

  if (!show_existing_frame) ++num_decoded_frame;

  if (show_existing_frame) {

    return status;

  } else {

    const double removal_time = get_removal_time(

        decoder_model->mode, num_decoded_frame,

        decoder_model->decoder_buffer_delay, decoder_model->frame_buffer_pool,

        decoder_model->current_time);

    if (removal_time < 0.0) {

      status = DECODE_FRAME_BUF_UNAVAILABLE;

      return status;

    }

    // A frame with show_existing_frame being false indicates the end of a DFG.

    // Update the bits arrival time of this DFG.

    const double buffer_delay = (decoder_model->encoder_buffer_delay +

                                 decoder_model->decoder_buffer_delay) /

                                90000.0;

    const double latest_arrival_time = removal_time - buffer_delay;

    const double first_bit_arrival_time =

        AOMMAX(decoder_model->last_bit_arrival_time, latest_arrival_time);

    const double last_bit_arrival_time =

        first_bit_arrival_time +

        (double)cur_coded_bits / decoder_model->bit_rate;

    // Smoothing buffer underflows if the last bit arrives after the removal

    // time.

    if (last_bit_arrival_time > removal_time &&

        !decoder_model->is_low_delay_mode) {

      status = SMOOTHING_BUFFER_UNDERFLOW;

      return status;

    }

    // Check if the smoothing buffer overflows.

    const DFG_INTERVAL_QUEUE *const queue = &decoder_model->dfg_interval_queue;

    if (queue->size >= DFG_INTERVAL_QUEUE_SIZE) {

      assert(0);

    }

    double total_interval = queue->total_interval;

    int qhead = queue->head;

    int qsize = queue->size;

    // Remove the DFGs with removal time earlier than last_bit_arrival_time.

    while (queue->buf[qhead].removal_time <= last_bit_arrival_time &&

           qsize > 0) {

      if (queue->buf[qhead].removal_time - first_bit_arrival_time +

              total_interval >

          1.0) {

        status = SMOOTHING_BUFFER_OVERFLOW;

        return status;

      }

      total_interval -= queue->buf[qhead].last_bit_arrival_time -

                        queue->buf[qhead].first_bit_arrival_time;

      qhead = (qhead + 1) % DFG_INTERVAL_QUEUE_SIZE;

      --qsize;

    }

    total_interval += last_bit_arrival_time - first_bit_arrival_time;

    // The smoothing buffer can hold at most "bit_rate" bits, which is

    // equivalent to 1 second of total interval.

    if (total_interval > 1.0) {

      status = SMOOTHING_BUFFER_OVERFLOW;

      return status;

    }

    return status;

  }

}

static void decoder_model_process_frame(const AV1_COMP *const cpi,

                                        size_t coded_bits,

                                        DECODER_MODEL *const decoder_model) {

  if (!decoder_model || decoder_model->status != DECODER_MODEL_OK) return;

  const AV1_COMMON *const cm = &cpi->common;

  const int luma_pic_size = cm->superres_upscaled_width * cm->height;

  const int show_existing_frame = cm->show_existing_frame;

  const int show_frame = cm->show_frame || show_existing_frame;

  ++decoder_model->num_frame;

  if (!show_existing_frame) ++decoder_model->num_decoded_frame;

  if (show_frame) ++decoder_model->num_shown_frame;

  decoder_model->coded_bits += coded_bits;

  int display_idx = -1;

  if (show_existing_frame) {

    display_idx = decoder_model->vbi[cpi->existing_fb_idx_to_show];

    if (display_idx < 0) {

      decoder_model->status = DECODE_EXISTING_FRAME_BUF_EMPTY;

      return;

    }

    if (decoder_model->frame_buffer_pool[display_idx].frame_type == KEY_FRAME) {

      update_ref_buffers(decoder_model, display_idx, 0xFF);

    }

  } else {

    const double removal_time = get_removal_time(

        decoder_model->mode, decoder_model->num_decoded_frame,

        decoder_model->decoder_buffer_delay, decoder_model->frame_buffer_pool,

        decoder_model->current_time);

    if (removal_time < 0.0) {

      decoder_model->status = DECODE_FRAME_BUF_UNAVAILABLE;

      return;

    }

    const int previous_decode_samples = decoder_model->decode_samples;

    const double previous_removal_time = decoder_model->removal_time;

    assert(previous_removal_time < removal_time);

    decoder_model->removal_time = removal_time;

    decoder_model->decode_samples = luma_pic_size;

    const double this_decode_rate =

        previous_decode_samples / (removal_time - previous_removal_time);

    decoder_model->max_decode_rate =

        AOMMAX(decoder_model->max_decode_rate, this_decode_rate);

    // A frame with show_existing_frame being false indicates the end of a DFG.

    // Update the bits arrival time of this DFG.

    const double buffer_delay = (decoder_model->encoder_buffer_delay +

                                 decoder_model->decoder_buffer_delay) /

                                90000.0;

    const double latest_arrival_time = removal_time - buffer_delay;

    decoder_model->first_bit_arrival_time =

        AOMMAX(decoder_model->last_bit_arrival_time, latest_arrival_time);

    decoder_model->last_bit_arrival_time =

        decoder_model->first_bit_arrival_time +

        (double)decoder_model->coded_bits / decoder_model->bit_rate;

    // Smoothing buffer underflows if the last bit arrives after the removal

    // time.

    if (decoder_model->last_bit_arrival_time > removal_time &&

        !decoder_model->is_low_delay_mode) {

      decoder_model->status = SMOOTHING_BUFFER_UNDERFLOW;

      return;

    }

    // Reset the coded bits for the next DFG.

    decoder_model->coded_bits = 0;

    // Check if the smoothing buffer overflows.

    DFG_INTERVAL_QUEUE *const queue = &decoder_model->dfg_interval_queue;

    if (queue->size >= DFG_INTERVAL_QUEUE_SIZE) {

      assert(0);

    }

    const double first_bit_arrival_time = decoder_model->first_bit_arrival_time;

    const double last_bit_arrival_time = decoder_model->last_bit_arrival_time;

    // Remove the DFGs with removal time earlier than last_bit_arrival_time.

    while (queue->buf[queue->head].removal_time <= last_bit_arrival_time &&

           queue->size > 0) {

      if (queue->buf[queue->head].removal_time - first_bit_arrival_time +

              queue->total_interval >

          1.0) {

        decoder_model->status = SMOOTHING_BUFFER_OVERFLOW;

        return;

      }

      queue->total_interval -= queue->buf[queue->head].last_bit_arrival_time -

                               queue->buf[queue->head].first_bit_arrival_time;

      queue->head = (queue->head + 1) % DFG_INTERVAL_QUEUE_SIZE;

      --queue->size;

    }

    // Push current DFG into the queue.

    const int queue_index =

        (queue->head + queue->size++) % DFG_INTERVAL_QUEUE_SIZE;

    queue->buf[queue_index].first_bit_arrival_time = first_bit_arrival_time;

    queue->buf[queue_index].last_bit_arrival_time = last_bit_arrival_time;

    queue->buf[queue_index].removal_time = removal_time;

    queue->total_interval += last_bit_arrival_time - first_bit_arrival_time;

    // The smoothing buffer can hold at most "bit_rate" bits, which is

    // equivalent to 1 second of total interval.

    if (queue->total_interval > 1.0) {

      decoder_model->status = SMOOTHING_BUFFER_OVERFLOW;

      return;

    }

    release_processed_frames(decoder_model, removal_time);

    decoder_model->current_time =

        removal_time + time_to_decode_frame(cm, decoder_model->decode_rate);

    const int cfbi = get_free_buffer(decoder_model);

    if (cfbi < 0) {

      decoder_model->status = DECODE_FRAME_BUF_UNAVAILABLE;

      return;

    }

    const CurrentFrame *const current_frame = &cm->current_frame;

    decoder_model->frame_buffer_pool[cfbi].frame_type =

        cm->current_frame.frame_type;

    display_idx = cfbi;

    update_ref_buffers(decoder_model, cfbi, current_frame->refresh_frame_flags);

    if (decoder_model->initial_presentation_delay < 0.0) {

      // Display can begin after required number of frames have been buffered.

      if (frames_in_buffer_pool(decoder_model) >=

          decoder_model->initial_display_delay - 1) {

        decoder_model->initial_presentation_delay = decoder_model->current_time;

        // Update presentation time for each shown frame in the frame buffer.

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

          FRAME_BUFFER *const this_buffer =

              &decoder_model->frame_buffer_pool[i];

          if (this_buffer->player_ref_count == 0) continue;

          assert(this_buffer->display_index >= 0);

          this_buffer->presentation_time =

              get_presentation_time(decoder_model, this_buffer->display_index);

        }

      }

    }

  }

  // Display.

  if (show_frame) {

    assert(display_idx >= 0 && display_idx < BUFFER_POOL_MAX_SIZE);

    FRAME_BUFFER *const this_buffer =

        &decoder_model->frame_buffer_pool[display_idx];

    ++this_buffer->player_ref_count;

    this_buffer->display_index = decoder_model->num_shown_frame;

    const double presentation_time =

        get_presentation_time(decoder_model, this_buffer->display_index);

    this_buffer->presentation_time = presentation_time;

    if (presentation_time >= 0.0 &&

        decoder_model->current_time > presentation_time) {

      decoder_model->status = DISPLAY_FRAME_LATE;

      return;

    }

    const int previous_display_samples = decoder_model->display_samples;

    const double previous_presentation_time = decoder_model->presentation_time;

    decoder_model->display_samples = luma_pic_size;

    decoder_model->presentation_time = presentation_time;

    if (presentation_time >= 0.0 && previous_presentation_time >= 0.0) {

      assert(previous_presentation_time < presentation_time);

      const double this_display_rate =

          previous_display_samples /

          (presentation_time - previous_presentation_time);

      decoder_model->max_display_rate =

          AOMMAX(decoder_model->max_display_rate, this_display_rate);

    }

  }

}

void av1_init_level_info(AV1_COMP *cpi) {

  for (int op_index = 0; op_index < MAX_NUM_OPERATING_POINTS; ++op_index) {

    AV1LevelInfo *const this_level_info =

        cpi->ppi->level_params.level_info[op_index];

    if (!this_level_info) continue;

    memset(this_level_info, 0, sizeof(*this_level_info));

    AV1LevelSpec *const level_spec = &this_level_info->level_spec;

    level_spec->level = SEQ_LEVEL_MAX;

    AV1LevelStats *const level_stats = &this_level_info->level_stats;

    level_stats->min_cropped_tile_width = INT_MAX;

    level_stats->min_cropped_tile_height = INT_MAX;

    level_stats->min_frame_width = INT_MAX;

    level_stats->min_frame_height = INT_MAX;

    level_stats->tile_width_is_valid = 1;

    level_stats->min_cr = 1e8;

    FrameWindowBuffer *const frame_window_buffer =

        &this_level_info->frame_window_buffer;

    frame_window_buffer->num = 0;

    frame_window_buffer->start = 0;

    const AV1_COMMON *const cm = &cpi->common;

    const int upscaled_width = cm->superres_upscaled_width;

    const int height = cm->height;

    const int pic_size = upscaled_width * height;

    for (AV1_LEVEL level = SEQ_LEVEL_2_0; level < SEQ_LEVELS; ++level) {

      DECODER_MODEL *const this_model = &this_level_info->decoder_models[level];

      const AV1LevelSpec *const spec = &av1_level_defs[level];

      if (upscaled_width > spec->max_h_size || height > spec->max_v_size ||

          pic_size > spec->max_picture_size) {

        // Turn off decoder model for this level as the frame size already

        // exceeds level constraints.

        this_model->status = DECODER_MODEL_DISABLED;

      } else {

        decoder_model_init(cpi, level, op_index, this_model);

      }

    }

  }

}

static double get_min_cr(const AV1LevelSpec *const level_spec, int tier,

                         int is_still_picture, int64_t decoded_sample_rate) {

  if (is_still_picture) return 0.8;

  if (level_spec->level < SEQ_LEVEL_4_0) tier = 0;

  const double min_cr_basis = tier ? level_spec->high_cr : level_spec->main_cr;

  const double speed_adj =

      (double)decoded_sample_rate / level_spec->max_display_rate;

  return AOMMAX(min_cr_basis * speed_adj, 0.8);

}

double av1_get_min_cr_for_level(AV1_LEVEL level_index, int tier,

                                int is_still_picture) {

  assert(is_valid_seq_level_idx(level_index));

  const AV1LevelSpec *const level_spec = &av1_level_defs[level_index];

  return get_min_cr(level_spec, tier, is_still_picture,

                    level_spec->max_decode_rate);

}

static void get_temporal_parallel_params(int scalability_mode_idc,

                                         int *temporal_parallel_num,