source/encoder/bitcost.h

   1 /*****************************************************************************
   2  * Copyright (C) 2013 x265 project
   3  *
   4  * Authors: Steve Borho <steve@borho.org>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  19  *
  20  * This program is also available under a commercial proprietary license.
  21  * For more information, contact us at license @ x265.com.
  22  *****************************************************************************/
  23
  24 #ifndef X265_BITCOST_H
  25 #define X265_BITCOST_H
  26
  27 #include "common.h"
  28 #include "threading.h"
  29 #include "mv.h"
  30
  31 namespace x265 {
  32 // private x265 namespace
  33
  34 class BitCost
  35 {
  36 public:
  37
  38     BitCost() : m_cost_mvx(0), m_cost_mvy(0), m_cost(0), m_mvp(0) {}
  39
  40     void setQP(unsigned int qp);
  41
  42     void setMVP(const MV& mvp)                      { m_mvp = mvp; m_cost_mvx = m_cost - mvp.x; m_cost_mvy = m_cost - mvp.y; }
  43
  44     // return bit cost of motion vector difference, multiplied by lambda
  45     inline uint16_t mvcost(const MV& mv) const      { return m_cost_mvx[mv.x] + m_cost_mvy[mv.y]; }
  46
  47     // return bit cost of motion vector difference, without lambda
  48     inline uint32_t bitcost(const MV& mv) const
  49     {
  50         return (uint32_t)(s_bitsizes[abs(mv.x - m_mvp.x)] +
  51                           s_bitsizes[abs(mv.y - m_mvp.y)] + 0.5f);
  52     }
  53
  54     static inline uint32_t bitcost(const MV& mv, const MV& mvp)
  55     {
  56         return (uint32_t)(s_bitsizes[abs(mv.x - mvp.x)] +
  57                           s_bitsizes[abs(mv.y - mvp.y)] + 0.5f);
  58     }
  59
  60     static void destroy();
  61
  62 protected:
  63
  64     uint16_t *m_cost_mvx;
  65
  66     uint16_t *m_cost_mvy;
  67
  68     uint16_t *m_cost;
  69
  70     MV        m_mvp;
  71
  72     BitCost& operator =(const BitCost&);
  73
  74 private:
  75
  76     /* default log2_max_mv_length_horizontal and log2_max_mv_length_horizontal
  77      * are 15, specified in quarter-pel luma sample units. making the maximum
  78      * signaled ful-pel motion distance 4096, max qpel is 32768 */
  79     enum { BC_MAX_MV = (1 << 15) };
  80
  81     enum { BC_MAX_QP = 82 };
  82
  83     static float *s_bitsizes;
  84
  85     static uint16_t *s_costs[BC_MAX_QP];
  86
  87     static Lock s_costCalcLock;
  88
  89     static void CalculateLogs();
  90 };
  91 }
  92
  93 #endif // ifndef X265_BITCOST_H