[5] | 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 |
---|
[56] | 4 | * granted under this license. |
---|
[5] | 5 | * |
---|
[608] | 6 | * Copyright (c) 2010-2013, ITU/ISO/IEC |
---|
[5] | 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. |
---|
[56] | 17 | * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may |
---|
[5] | 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 | */ |
---|
[2] | 33 | |
---|
| 34 | /** \file ContextModel3DBuffer.h |
---|
| 35 | \brief context model 3D buffer class (header) |
---|
| 36 | */ |
---|
| 37 | |
---|
[56] | 38 | #ifndef _HM_CONTEXT_MODEL_3DBUFFER_H_ |
---|
| 39 | #define _HM_CONTEXT_MODEL_3DBUFFER_H_ |
---|
[2] | 40 | |
---|
| 41 | #include <stdio.h> |
---|
| 42 | #include <assert.h> |
---|
| 43 | #include <memory.h> |
---|
| 44 | |
---|
| 45 | #include "CommonDef.h" |
---|
| 46 | #include "ContextModel.h" |
---|
| 47 | |
---|
[56] | 48 | //! \ingroup TLibCommon |
---|
| 49 | //! \{ |
---|
| 50 | |
---|
[2] | 51 | // ==================================================================================================================== |
---|
| 52 | // Class definition |
---|
| 53 | // ==================================================================================================================== |
---|
| 54 | |
---|
| 55 | /// context model 3D buffer class |
---|
| 56 | class ContextModel3DBuffer |
---|
| 57 | { |
---|
| 58 | protected: |
---|
[56] | 59 | ContextModel* m_contextModel; ///< array of context models |
---|
| 60 | const UInt m_sizeX; ///< X size of 3D buffer |
---|
| 61 | const UInt m_sizeXY; ///< X times Y size of 3D buffer |
---|
| 62 | const UInt m_sizeXYZ; ///< total size of 3D buffer |
---|
[2] | 63 | |
---|
| 64 | public: |
---|
[56] | 65 | ContextModel3DBuffer ( UInt uiSizeZ, UInt uiSizeY, UInt uiSizeX, ContextModel *basePtr, Int &count ); |
---|
| 66 | ~ContextModel3DBuffer () {} |
---|
[2] | 67 | |
---|
| 68 | // access functions |
---|
| 69 | ContextModel& get( UInt uiZ, UInt uiY, UInt uiX ) |
---|
| 70 | { |
---|
[56] | 71 | return m_contextModel[ uiZ * m_sizeXY + uiY * m_sizeX + uiX ]; |
---|
[2] | 72 | } |
---|
| 73 | ContextModel* get( UInt uiZ, UInt uiY ) |
---|
| 74 | { |
---|
[56] | 75 | return &m_contextModel[ uiZ * m_sizeXY + uiY * m_sizeX ]; |
---|
[2] | 76 | } |
---|
| 77 | ContextModel* get( UInt uiZ ) |
---|
| 78 | { |
---|
[56] | 79 | return &m_contextModel[ uiZ * m_sizeXY ]; |
---|
[2] | 80 | } |
---|
| 81 | |
---|
| 82 | // initialization & copy functions |
---|
[56] | 83 | Void initBuffer( SliceType eSliceType, Int iQp, UChar* ctxModel ); ///< initialize 3D buffer by slice type & QP |
---|
| 84 | |
---|
| 85 | UInt calcCost( SliceType sliceType, Int qp, UChar* ctxModel ); ///< determine cost of choosing a probability table based on current probabilities |
---|
| 86 | /** copy from another buffer |
---|
| 87 | * \param src buffer to copy from |
---|
| 88 | */ |
---|
| 89 | Void copyFrom( ContextModel3DBuffer* src ) |
---|
| 90 | { |
---|
| 91 | assert( m_sizeXYZ == src->m_sizeXYZ ); |
---|
| 92 | ::memcpy( m_contextModel, src->m_contextModel, sizeof(ContextModel) * m_sizeXYZ ); |
---|
| 93 | } |
---|
[2] | 94 | }; |
---|
| 95 | |
---|
[56] | 96 | //! \} |
---|
[2] | 97 | |
---|
[56] | 98 | #endif // _HM_CONTEXT_MODEL_3DBUFFER_H_ |
---|