h264bsd_macroblock_layer.h revision 0c1bc742181ded4930842b46e9507372f0b1b963 
1/*
2 * Copyright (C) 2009 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17/*------------------------------------------------------------------------------
18
19    Table of contents
20
21    1. Include headers
22    2. Module defines
23    3. Data types
24    4. Function prototypes
25
26------------------------------------------------------------------------------*/
27
28#ifndef H264SWDEC_MACROBLOCK_LAYER_H
29#define H264SWDEC_MACROBLOCK_LAYER_H
30
31/*------------------------------------------------------------------------------
32    1. Include headers
33------------------------------------------------------------------------------*/
34
35#include "basetype.h"
36#include "h264bsd_stream.h"
37#include "h264bsd_image.h"
38#include "h264bsd_dpb.h"
39
40/*------------------------------------------------------------------------------
41    2. Module defines
42------------------------------------------------------------------------------*/
43
44/* Macro to determine if a mb is an intra mb */
45#define IS_INTRA_MB(a) ((a).mbType > 5)
46
47/* Macro to determine if a mb is an I_PCM mb */
48#define IS_I_PCM_MB(a) ((a).mbType == 31)
49
50typedef enum {
51    P_Skip          = 0,
52    P_L0_16x16      = 1,
53    P_L0_L0_16x8    = 2,
54    P_L0_L0_8x16    = 3,
55    P_8x8           = 4,
56    P_8x8ref0       = 5,
57    I_4x4           = 6,
58    I_16x16_0_0_0   = 7,
59    I_16x16_1_0_0   = 8,
60    I_16x16_2_0_0   = 9,
61    I_16x16_3_0_0   = 10,
62    I_16x16_0_1_0   = 11,
63    I_16x16_1_1_0   = 12,
64    I_16x16_2_1_0   = 13,
65    I_16x16_3_1_0   = 14,
66    I_16x16_0_2_0   = 15,
67    I_16x16_1_2_0   = 16,
68    I_16x16_2_2_0   = 17,
69    I_16x16_3_2_0   = 18,
70    I_16x16_0_0_1   = 19,
71    I_16x16_1_0_1   = 20,
72    I_16x16_2_0_1   = 21,
73    I_16x16_3_0_1   = 22,
74    I_16x16_0_1_1   = 23,
75    I_16x16_1_1_1   = 24,
76    I_16x16_2_1_1   = 25,
77    I_16x16_3_1_1   = 26,
78    I_16x16_0_2_1   = 27,
79    I_16x16_1_2_1   = 28,
80    I_16x16_2_2_1   = 29,
81    I_16x16_3_2_1   = 30,
82    I_PCM           = 31
83} mbType_e;
84
85typedef enum {
86    P_L0_8x8 = 0,
87    P_L0_8x4 = 1,
88    P_L0_4x8 = 2,
89    P_L0_4x4 = 3
90} subMbType_e;
91
92typedef enum {
93    MB_P_16x16 = 0,
94    MB_P_16x8,
95    MB_P_8x16,
96    MB_P_8x8
97} mbPartMode_e;
98
99typedef enum {
100    MB_SP_8x8 = 0,
101    MB_SP_8x4,
102    MB_SP_4x8,
103    MB_SP_4x4
104} subMbPartMode_e;
105
106typedef enum {
107    PRED_MODE_INTRA4x4 = 0,
108    PRED_MODE_INTRA16x16  ,
109    PRED_MODE_INTER
110} mbPartPredMode_e;
111
112/*------------------------------------------------------------------------------
113    3. Data types
114------------------------------------------------------------------------------*/
115
116typedef struct
117{
118    /* MvPrediction16x16 assumes that MVs are 16bits */
119    i16 hor;
120    i16 ver;
121} mv_t;
122
123typedef struct
124{
125    u32 prevIntra4x4PredModeFlag[16];
126    u32 remIntra4x4PredMode[16];
127    u32 intraChromaPredMode;
128    u32 refIdxL0[4];
129    mv_t mvdL0[4];
130} mbPred_t;
131
132typedef struct
133{
134    subMbType_e subMbType[4];
135    u32 refIdxL0[4];
136    mv_t mvdL0[4][4];
137} subMbPred_t;
138
139typedef struct
140{
141#ifdef H264DEC_OMXDL
142    u8 posCoefBuf[27*16*3];
143    u8 totalCoeff[27];
144#else
145    i16 totalCoeff[27];
146#endif
147    i32 level[26][16];
148    u32 coeffMap[24];
149} residual_t;
150
151typedef struct
152{
153    mbType_e mbType;
154    u32 codedBlockPattern;
155    i32 mbQpDelta;
156    mbPred_t mbPred;
157    subMbPred_t subMbPred;
158    residual_t residual;
159} macroblockLayer_t;
160
161typedef struct mbStorage
162{
163    mbType_e mbType;
164    u32 sliceId;
165    u32 disableDeblockingFilterIdc;
166    i32 filterOffsetA;
167    i32 filterOffsetB;
168    u32 qpY;
169    i32 chromaQpIndexOffset;
170#ifdef H264DEC_OMXDL
171    u8 totalCoeff[27];
172#else
173    i16 totalCoeff[27];
174#endif
175    u8 intra4x4PredMode[16];
176    u32 refPic[4];
177    u8* refAddr[4];
178    mv_t mv[16];
179    u32 decoded;
180    struct mbStorage *mbA;
181    struct mbStorage *mbB;
182    struct mbStorage *mbC;
183    struct mbStorage *mbD;
184} mbStorage_t;
185
186/*------------------------------------------------------------------------------
187    4. Function prototypes
188------------------------------------------------------------------------------*/
189
190u32 h264bsdDecodeMacroblockLayer(strmData_t *pStrmData,
191    macroblockLayer_t *pMbLayer, mbStorage_t *pMb, u32 sliceType,
192    u32 numRefIdxActive);
193
194u32 h264bsdNumMbPart(mbType_e mbType);
195u32 h264bsdNumSubMbPart(subMbType_e subMbType);
196
197subMbPartMode_e h264bsdSubMbPartMode(subMbType_e subMbType);
198
199u32 h264bsdDecodeMacroblock(mbStorage_t *pMb, macroblockLayer_t *pMbLayer,
200    image_t *currImage, dpbStorage_t *dpb, i32 *qpY, u32 mbNum,
201    u32 constrainedIntraPredFlag, u8* data);
202
203u32 h264bsdPredModeIntra16x16(mbType_e mbType);
204
205mbPartPredMode_e h264bsdMbPartPredMode(mbType_e mbType);
206#ifdef H264DEC_NEON
207u32 h264bsdClearMbLayer(macroblockLayer_t *pMbLayer, u32 size);
208#endif
209
210#endif /* #ifdef H264SWDEC_MACROBLOCK_LAYER_H */
211
212
213