#ifndef TABLES_H_
#define TABLES_H_
/*****************************************************************************
 * This file is part of Kvazaar HEVC encoder.
 *
 * Copyright (C) 2013-2014 Tampere University of Technology and others (see
 * COPYING file).
 *
 * Kvazaar is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as published
 * by the Free Software Foundation.
 *
 * Kvazaar 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 Kvazaar.  If not, see <http://www.gnu.org/licenses/>.
 ****************************************************************************/

#include "global.h"

//4 8 16 32 64 128
//0 1  2  3  4   5
static const uint8_t g_to_bits[129] =
{
  0,
  0,0,0,0,
  0,0,0,1,
  0,0,0,0,0,0,0,2,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,3,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,4,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,5
};
#define TOBITS(len) g_to_bits[len]

/***
  * List of indices for 4x4 coefficient groups within 8x8 transform block.
  * First index: 0 = diagonal, 1 = vertical, 2 horizontal scan pattern.
  * Second index: (log2 - 2) size of transform block. 4x4 .. 32x32
  */
static const uint32_t g_sig_last_scan_8x8[3][4] =
{ {0, 2, 1, 3},
  {0, 1, 2, 3},
  {0, 2, 1, 3}
};

/***
  * List of indices for 4x4 coefficient groups within 16x16 transform block.
  */
static const uint32_t g_sig_last_scan_16x16[16] = {
  0,  4,  1,  8,
  5,  2, 12,  9,
  6,  3, 13, 10,
  7, 14, 11, 15
};

/***
  * List of indices for 4x4 coefficient groups within 32x32 transform block.
  */
static const uint32_t g_sig_last_scan_32x32[64] = {
  0,   8,  1, 16,  9,  2, 24, 17,
  10,  3, 32, 25, 18, 11,  4, 40,
  33, 26, 19, 12,  5, 48, 41, 34,
  27, 20, 13,  6, 56, 49, 42, 35,
  28, 21, 14,  7, 57, 50, 43, 36,
  29, 22, 15, 58, 51, 44, 37, 30,
  23, 59, 52, 45, 38, 31, 60, 53,
  46, 39, 61, 54, 47, 62, 55, 63
};

/**
 * List of pointers to coefficient group mappings.
 * First index: (log2 - 2) of transform block size
 * Second index: scan pattern 0 = diagonal, 1 = horizontal, 2 = vertical
 */
static const uint32_t * const g_sig_last_scan_cg[4][3] = {
  { g_sig_last_scan_8x8[0], g_sig_last_scan_8x8[1], g_sig_last_scan_8x8[2] },  // 4x4, only first element is used
  { g_sig_last_scan_8x8[0], g_sig_last_scan_8x8[1], g_sig_last_scan_8x8[2] },
  { g_sig_last_scan_16x16, 0, 0 },
  { g_sig_last_scan_32x32, 0, 0 }
};


typedef enum
{
  SCAN_DIAG = 0, // up-right diagonal scan
  SCAN_HOR,      // horizontal first scan
  SCAN_VER       // vertical first scan
} coeff_scan_order_t;


/**
 * List of mappings for coefficients within a transform block.
 * First index: scan pattern 0 = diagonal, 1 = horizontal, 2 = vertical
 * Second index: (log2 - 1) size of transform block. 2x2 .. 32x32
 */
extern const uint32_t* const g_sig_last_scan[3][5];
extern const int8_t g_convert_to_bit[LCU_WIDTH + 1];

#endif //TABLES_H_