source/encoder/framefilter.h

   1 /*****************************************************************************
   2  * Copyright (C) 2013 x265 project
   3  *
   4  * Authors: Chung Shin Yee <shinyee@multicorewareinc.com>
   5  *          Min Chen <chenm003@163.com>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  20  *
  21  * This program is also available under a commercial proprietary license.
  22  * For more information, contact us at license @ x265.com.
  23  *****************************************************************************/
  24
  25 #ifndef X265_FRAMEFILTER_H
  26 #define X265_FRAMEFILTER_H
  27
  28 #include "common.h"
  29 #include "frame.h"
  30 #include "deblock.h"
  31 #include "sao.h"
  32
  33 namespace x265 {
  34 // private x265 namespace
  35
  36 class Encoder;
  37 class Entropy;
  38 class FrameEncoder;
  39 struct ThreadLocalData;
  40
  41 // Manages the processing of a single frame loopfilter
  42 class FrameFilter
  43 {
  44 public:
  45
  46     x265_param*   m_param;
  47     Frame*        m_frame;
  48     FrameEncoder* m_frameEncoder;
  49     int           m_hChromaShift;
  50     int           m_vChromaShift;
  51     int           m_pad[2];
  52
  53     Deblock       m_deblock;
  54     SAO           m_sao;
  55     int           m_numRows;
  56     int           m_saoRowDelay;
  57     int           m_lastHeight;
  58
  59     void*         m_ssimBuf; /* Temp storage for ssim computation */
  60
  61     FrameFilter();
  62
  63     void init(Encoder *top, FrameEncoder *frame, int numRows);
  64     void destroy();
  65
  66     void start(Frame *pic, Entropy& initState, int qp);
  67
  68     void processRow(int row);
  69     void processRowPost(int row);
  70     void processSao(int row);
  71     uint32_t getCUHeight(int rowNum) const;
  72 };
  73 }
  74
  75 #endif // ifndef X265_FRAMEFILTER_H