1 /* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license.
5 *
6 * Copyright (c) 2010-2014, ITU/ISO/IEC
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 *
12 * * Redistributions of source code must retain the above copyright notice,
13 * this list of conditions and the following disclaimer.
14 * * Redistributions in binary form must reproduce the above copyright notice,
15 * this list of conditions and the following disclaimer in the documentation
16 * and/or other materials provided with the distribution.
17 * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18 * be used to endorse or promote products derived from this software without
19 * specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34 /** \file TComSampleAdaptiveOffset.h
35 \brief sample adaptive offset class (header)
36 */
37
38 #ifndef __TCOMSAMPLEADAPTIVEOFFSET__
39 #define __TCOMSAMPLEADAPTIVEOFFSET__
40
41 #include "CommonDef.h"
42 #include "TComPic.h"
43
44 //! \ingroup TLibCommon
45 //! \{
46
47
48 // ====================================================================================================================
49 // Constants
50 // ====================================================================================================================
51
52 #define MAX_SAO_TRUNCATED_BITDEPTH 10
53
54 // ====================================================================================================================
55 // Class definition
56 // ====================================================================================================================
57 extern UInt g_saoMaxOffsetQVal[MAX_NUM_COMPONENT];
58
sgn(T val)59 template <typename T> Int sgn(T val)
60 {
61 return (T(0) < val) - (val < T(0));
62 }
63
64 class TComSampleAdaptiveOffset
65 {
66 public:
67 TComSampleAdaptiveOffset();
68 virtual ~TComSampleAdaptiveOffset();
69 Void SAOProcess(TComPic* pDecPic);
70 Void create( Int picWidth, Int picHeight, ChromaFormat format, UInt maxCUWidth, UInt maxCUHeight, UInt maxCUDepth, UInt lumaBitShift, UInt chromaBitShift );
71 Void destroy();
72 Void reconstructBlkSAOParams(TComPic* pic, SAOBlkParam* saoBlkParams);
73 Void PCMLFDisableProcess (TComPic* pcPic);
74 protected:
75 Void offsetBlock(ComponentID compIdx, Int typeIdx, Int* offset, Pel* srcBlk, Pel* resBlk, Int srcStride, Int resStride, Int width, Int height
76 , Bool isLeftAvail, Bool isRightAvail, Bool isAboveAvail, Bool isBelowAvail, Bool isAboveLeftAvail, Bool isAboveRightAvail, Bool isBelowLeftAvail, Bool isBelowRightAvail);
77 Void invertQuantOffsets(ComponentID compIdx, Int typeIdc, Int typeAuxInfo, Int* dstOffsets, Int* srcOffsets);
78 Void reconstructBlkSAOParam(SAOBlkParam& recParam, SAOBlkParam* mergeList[NUM_SAO_MERGE_TYPES]);
79 Int getMergeList(TComPic* pic, Int ctuRsAddr, SAOBlkParam* blkParams, SAOBlkParam* mergeList[NUM_SAO_MERGE_TYPES]);
80 Void offsetCTU(Int ctuRsAddr, TComPicYuv* srcYuv, TComPicYuv* resYuv, SAOBlkParam& saoblkParam, TComPic* pPic);
81 Void xPCMRestoration(TComPic* pcPic);
82 Void xPCMCURestoration ( TComDataCU* pcCU, UInt uiAbsZorderIdx, UInt uiDepth );
83 Void xPCMSampleRestoration (TComDataCU* pcCU, UInt uiAbsZorderIdx, UInt uiDepth, ComponentID component);
84 protected:
85 UInt m_offsetStepLog2[MAX_NUM_COMPONENT]; //offset step
86 TComPicYuv* m_tempPicYuv; //temporary buffer
87 Int m_picWidth;
88 Int m_picHeight;
89 Int m_maxCUWidth;
90 Int m_maxCUHeight;
91 Int m_numCTUInWidth;
92 Int m_numCTUInHeight;
93 Int m_numCTUsPic;
94
95
96 Int m_lineBufWidth;
97 Char* m_signLineBuf1;
98 Char* m_signLineBuf2;
99 ChromaFormat m_chromaFormatIDC;
100 private:
101 Bool m_picSAOEnabled[MAX_NUM_COMPONENT];
102 };
103
104 //! \}
105 #endif
106
107