Mercurial > hg > forks > libbpg
comparison jctvc/TLibCommon/ContextModel.h @ 0:772086c29cc7
Initial import.
author | Matti Hamalainen <ccr@tnsp.org> |
---|---|
date | Wed, 16 Nov 2016 11:16:33 +0200 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:772086c29cc7 |
---|---|
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 | |
35 /** \file ContextModel.h | |
36 \brief context model class (header) | |
37 */ | |
38 | |
39 #ifndef __CONTEXTMODEL__ | |
40 #define __CONTEXTMODEL__ | |
41 | |
42 #if _MSC_VER > 1000 | |
43 #pragma once | |
44 #endif // _MSC_VER > 1000 | |
45 | |
46 #include "CommonDef.h" | |
47 #include "TComRom.h" | |
48 | |
49 //! \ingroup TLibCommon | |
50 //! \{ | |
51 | |
52 // ==================================================================================================================== | |
53 // Class definition | |
54 // ==================================================================================================================== | |
55 | |
56 /// context model class | |
57 class ContextModel | |
58 { | |
59 public: | |
60 ContextModel () { m_ucState = 0; m_binsCoded = 0; } | |
61 ~ContextModel () {} | |
62 | |
63 UChar getState () { return ( m_ucState >> 1 ); } ///< get current state | |
64 UChar getMps () { return ( m_ucState & 1 ); } ///< get curret MPS | |
65 Void setStateAndMps( UChar ucState, UChar ucMPS) { m_ucState = (ucState << 1) + ucMPS; } ///< set state and MPS | |
66 | |
67 Void init ( Int qp, Int initValue ); ///< initialize state with initial probability | |
68 | |
69 Void updateLPS () | |
70 { | |
71 m_ucState = m_aucNextStateLPS[ m_ucState ]; | |
72 } | |
73 | |
74 Void updateMPS () | |
75 { | |
76 m_ucState = m_aucNextStateMPS[ m_ucState ]; | |
77 } | |
78 | |
79 Int getEntropyBits(Short val) { return m_entropyBits[m_ucState ^ val]; } | |
80 | |
81 #if FAST_BIT_EST | |
82 Void update( Int binVal ) | |
83 { | |
84 m_ucState = m_nextState[m_ucState][binVal]; | |
85 } | |
86 static Void buildNextStateTable(); | |
87 static Int getEntropyBitsTrm( Int val ) { return m_entropyBits[126 ^ val]; } | |
88 #endif | |
89 Void setBinsCoded(UInt val) { m_binsCoded = val; } | |
90 UInt getBinsCoded() { return m_binsCoded; } | |
91 | |
92 private: | |
93 UChar m_ucState; ///< internal state variable | |
94 | |
95 static const UInt m_totalStates = (1 << CONTEXT_STATE_BITS) * 2; //*2 for MPS = [0|1] | |
96 static const UChar m_aucNextStateMPS[m_totalStates]; | |
97 static const UChar m_aucNextStateLPS[m_totalStates]; | |
98 static const Int m_entropyBits [m_totalStates]; | |
99 #if FAST_BIT_EST | |
100 static UChar m_nextState[m_totalStates][2 /*MPS = [0|1]*/]; | |
101 #endif | |
102 UInt m_binsCoded; | |
103 }; | |
104 | |
105 //! \} | |
106 | |
107 #endif | |
108 |