libbpg-0.9.6

2015-10-27 11:46:00 +01:00 · 2015-10-27 11:46:00 +01:00 · 35a8402710
commit 35a8402710
parent 3035b41edf
248 changed files with 232891 additions and 100 deletions
--- a/x265/source/encoder/sao.h
+++ b/x265/source/encoder/sao.h
@ -0,0 +1,154 @@
+/*****************************************************************************
+ * Copyright (C) 2013 x265 project
+ *
+ * Authors: Steve Borho <steve@borho.org>
+ *          Min Chen <chenm003@163.com>
+ *          Praveen Kumar Tiwari <praveen@multicorewareinc.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
+ *
+ * This program is also available under a commercial proprietary license.
+ * For more information, contact us at license @ x265.com.
+ *****************************************************************************/
+
+#ifndef X265_SAO_H
+#define X265_SAO_H
+
+#include "common.h"
+#include "frame.h"
+#include "entropy.h"
+
+namespace X265_NS {
+// private namespace
+
+enum SAOTypeLen
+{
+    SAO_EO_LEN = 4,
+    SAO_BO_LEN = 4,
+    SAO_NUM_BO_CLASSES = 32
+};
+
+enum SAOType
+{
+    SAO_EO_0 = 0,
+    SAO_EO_1,
+    SAO_EO_2,
+    SAO_EO_3,
+    SAO_BO,
+    MAX_NUM_SAO_TYPE
+};
+
+class SAO
+{
+public:
+
+    enum { SAO_MAX_DEPTH = 4 };
+    enum { SAO_BO_BITS  = 5 };
+    enum { MAX_NUM_SAO_CLASS = 33 };
+    enum { SAO_BIT_INC = 0 }; /* in HM12.0, it wrote as X265_MAX(X265_DEPTH - 10, 0) */
+    enum { OFFSET_THRESH = 1 << X265_MIN(X265_DEPTH - 5, 5) };
+    enum { NUM_EDGETYPE = 5 };
+    enum { NUM_PLANE = 3 };
+    enum { NUM_MERGE_MODE = 3 };
+
+    static const uint32_t s_eoTable[NUM_EDGETYPE];
+
+    typedef int32_t (PerClass[MAX_NUM_SAO_TYPE][MAX_NUM_SAO_CLASS]);
+    typedef int32_t (PerPlane[NUM_PLANE][MAX_NUM_SAO_TYPE][MAX_NUM_SAO_CLASS]);
+
+protected:
+
+    /* allocated per part */
+    PerClass*   m_count;
+    PerClass*   m_offset;
+    PerClass*   m_offsetOrg;
+
+    /* allocated per CTU */
+    PerPlane*   m_countPreDblk;
+    PerPlane*   m_offsetOrgPreDblk;
+
+    double      m_depthSaoRate[2][4];
+    int8_t      m_offsetBo[SAO_NUM_BO_CLASSES];
+    int8_t      m_offsetEo[NUM_EDGETYPE];
+
+    int         m_numCuInWidth;
+    int         m_numCuInHeight;
+    int         m_numPlanes;
+    int         m_hChromaShift;
+    int         m_vChromaShift;
+
+    pixel*      m_clipTable;
+    pixel*      m_clipTableBase;
+
+    pixel*      m_tmpU1[3];
+    pixel*      m_tmpU2[3];
+    pixel*      m_tmpL1;
+    pixel*      m_tmpL2;
+
+public:
+
+    struct SAOContexts
+    {
+        Entropy cur;
+        Entropy next;
+        Entropy temp;
+    };
+
+    Frame*      m_frame;
+    Entropy     m_entropyCoder;
+    SAOContexts m_rdContexts;
+
+    x265_param* m_param;
+    int         m_refDepth;
+    int         m_numNoSao[2];
+
+    double      m_lumaLambda;
+    double      m_chromaLambda;
+    /* TODO: No doubles for distortion */
+
+    SAO();
+
+    bool create(x265_param* param);
+    void destroy();
+
+    void allocSaoParam(SAOParam* saoParam) const;
+
+    void startSlice(Frame* pic, Entropy& initState, int qp);
+    void resetStats();
+    void resetSaoUnit(SaoCtuParam* saoUnit);
+
+    // CTU-based SAO process without slice granularity
+    void processSaoCu(int addr, int typeIdx, int plane);
+    void processSaoUnitRow(SaoCtuParam* ctuParam, int idxY, int plane);
+
+    void copySaoUnit(SaoCtuParam* saoUnitDst, const SaoCtuParam* saoUnitSrc);
+
+    void calcSaoStatsCu(int addr, int plane);
+    void calcSaoStatsCu_BeforeDblk(Frame* pic, int idxX, int idxY);
+
+    void saoComponentParamDist(SAOParam* saoParam, int addr, int addrUp, int addrLeft, SaoCtuParam mergeSaoParam[2], double* mergeDist);
+    void sao2ChromaParamDist(SAOParam* saoParam, int addr, int addrUp, int addrLeft, SaoCtuParam mergeSaoParam[][2], double* mergeDist);
+
+    inline int estIterOffset(int typeIdx, int classIdx, double lambda, int offset, int32_t count, int32_t offsetOrg,
+                             int32_t* currentDistortionTableBo, double* currentRdCostTableBo);
+    inline int64_t estSaoTypeDist(int plane, int typeIdx, double lambda, int32_t* currentDistortionTableBo, double* currentRdCostTableBo);
+
+    void rdoSaoUnitRowEnd(const SAOParam* saoParam, int numctus);
+    void rdoSaoUnitRow(SAOParam* saoParam, int idxY);
+};
+
+}
+
+#endif // ifndef X265_SAO_H