1 //<MStar Software>
2 //******************************************************************************
3 // MStar Software
4 // Copyright (c) 2010 - 2012 MStar Semiconductor, Inc. All rights reserved.
5 // All software, firmware and related documentation herein ("MStar Software") are
6 // intellectual property of MStar Semiconductor, Inc. ("MStar") and protected by
7 // law, including, but not limited to, copyright law and international treaties.
8 // Any use, modification, reproduction, retransmission, or republication of all
9 // or part of MStar Software is expressly prohibited, unless prior written
10 // permission has been granted by MStar.
11 //
12 // By accessing, browsing and/or using MStar Software, you acknowledge that you
13 // have read, understood, and agree, to be bound by below terms ("Terms") and to
14 // comply with all applicable laws and regulations:
15 //
16 // 1. MStar shall retain any and all right, ownership and interest to MStar
17 // Software and any modification/derivatives thereof.
18 // No right, ownership, or interest to MStar Software and any
19 // modification/derivatives thereof is transferred to you under Terms.
20 //
21 // 2. You understand that MStar Software might include, incorporate or be
22 // supplied together with third party`s software and the use of MStar
23 // Software may require additional licenses from third parties.
24 // Therefore, you hereby agree it is your sole responsibility to separately
25 // obtain any and all third party right and license necessary for your use of
26 // such third party`s software.
27 //
28 // 3. MStar Software and any modification/derivatives thereof shall be deemed as
29 // MStar`s confidential information and you agree to keep MStar`s
30 // confidential information in strictest confidence and not disclose to any
31 // third party.
32 //
33 // 4. MStar Software is provided on an "AS IS" basis without warranties of any
34 // kind. Any warranties are hereby expressly disclaimed by MStar, including
35 // without limitation, any warranties of merchantability, non-infringement of
36 // intellectual property rights, fitness for a particular purpose, error free
37 // and in conformity with any international standard. You agree to waive any
38 // claim against MStar for any loss, damage, cost or expense that you may
39 // incur related to your use of MStar Software.
40 // In no event shall MStar be liable for any direct, indirect, incidental or
41 // consequential damages, including without limitation, lost of profit or
42 // revenues, lost or damage of data, and unauthorized system use.
43 // You agree that this Section 4 shall still apply without being affected
44 // even if MStar Software has been modified by MStar in accordance with your
45 // request or instruction for your use, except otherwise agreed by both
46 // parties in writing.
47 //
48 // 5. If requested, MStar may from time to time provide technical supports or
49 // services in relation with MStar Software to you for your use of
50 // MStar Software in conjunction with your or your customer`s product
51 // ("Services").
52 // You understand and agree that, except otherwise agreed by both parties in
53 // writing, Services are provided on an "AS IS" basis and the warranty
54 // disclaimer set forth in Section 4 above shall apply.
55 //
56 // 6. Nothing contained herein shall be construed as by implication, estoppels
57 // or otherwise:
58 // (a) conferring any license or right to use MStar name, trademark, service
59 // mark, symbol or any other identification;
60 // (b) obligating MStar or any of its affiliates to furnish any person,
61 // including without limitation, you and your customers, any assistance
62 // of any kind whatsoever, or any information; or
63 // (c) conferring any license or right under any intellectual property right.
64 //
65 // 7. These terms shall be governed by and construed in accordance with the laws
66 // of Taiwan, R.O.C., excluding its conflict of law rules.
67 // Any and all dispute arising out hereof or related hereto shall be finally
68 // settled by arbitration referred to the Chinese Arbitration Association,
69 // Taipei in accordance with the ROC Arbitration Law and the Arbitration
70 // Rules of the Association by three (3) arbitrators appointed in accordance
71 // with the said Rules.
72 // The place of arbitration shall be in Taipei, Taiwan and the language shall
73 // be English.
74 // The arbitration award shall be final and binding to both parties.
75 //
76 //******************************************************************************
77 //<MStar Software>
78
79
80
81 #include "MFE_chip.h"
82 #include "mfe_type.h"
83 #include "mfe_common.h"
84 #include "ms_dprintf.h"
85
86 #include "OutStrm.h"
87
88 /********************************************************/
89 /* Implementation of class OutStream */
90 /********************************************************/
91
osCreate(OutStream * pStream)92 void osCreate(OutStream* pStream)
93 {
94 // Check endian
95 MS_U32 tmp = 0x04030201;
96 if (*(MS_U8*)&tmp == 0x1)
97 pStream->m_bUseLittleEndian = 1;
98 else if (*(MS_U8*)&tmp == 0x4)
99 pStream->m_bUseLittleEndian = 0;
100 else {
101 MS_ASSERT(0);
102 pStream->m_bUseLittleEndian = 0;
103 }
104 ms_dprintk(DRV_L2,"OutStream: LittleEndian=%d\n", pStream->m_bUseLittleEndian);
105
106 pStream->m_pbFrameBuffer = 0;
107 pStream->m_nByteCount = 0;
108 pStream->m_nBufferEmptyBits = 32;
109 pStream->m_SmallBuffer = 0;
110 pStream->BC_nCumulativeBits = 0;
111 }
112
osDestroy(OutStream * pStream)113 void osDestroy(OutStream* pStream)
114 {
115 }
116
osSetWriteBuffer(OutStream * pStream,MS_U8 * pWBuf)117 void osSetWriteBuffer(OutStream* pStream, MS_U8 *pWBuf)
118 {
119 pStream->m_SmallBuffer = 0;
120 pStream->m_nByteCount = 0;
121 pStream->m_nBufferEmptyBits = 32;
122 pStream->m_pbFrameBuffer = pWBuf;
123 }
124
osReset(OutStream * pStream)125 void osReset(OutStream* pStream)
126 {
127 pStream->m_SmallBuffer = 0;
128 pStream->m_nByteCount = 0;
129 pStream->m_nBufferEmptyBits = 32;
130 pStream->BC_nCumulativeBits = 0;
131 }
132
osFlushAll(OutStream * pStream)133 void osFlushAll(OutStream* pStream)
134 {
135 MS_S32 nUnusedByteCount = 0;
136 MS_S32 nUnusedBits = pStream->m_nBufferEmptyBits;
137 if (pStream->m_nBufferEmptyBits != 32)
138 {
139 if (pStream->m_nBufferEmptyBits>=24)
140 nUnusedByteCount = 3;
141 else if (pStream->m_nBufferEmptyBits>=16)
142 nUnusedByteCount = 2;
143 else if (pStream->m_nBufferEmptyBits>=8)
144 nUnusedByteCount = 1;
145 osPutBits(pStream, 0, pStream->m_nBufferEmptyBits, (MS_S8*)(""));
146 }
147 pStream->BC_nCumulativeBits -= nUnusedBits;
148 pStream->m_nByteCount -= nUnusedByteCount;
149 }
150
osPutBits(OutStream * pStream,MS_U32 value,MS_S32 bits,MS_S8 * name)151 void osPutBits(OutStream* pStream, MS_U32 value, MS_S32 bits, MS_S8* name)
152 {
153 // MS_U32 *ptwrite;
154 MS_U32 tlongbuf;
155
156 // BitRate Control
157 pStream->BC_nCumulativeBits += bits;
158 //printf("PutBits() %s Val=%x Bits=%d\n", name, value, bits);
159
160 MS_ASSERT(bits!=0);
161 // if (bits) {
162 //
163 // Less than empty bits
164 //
165 if (bits < pStream->m_nBufferEmptyBits) {
166 pStream->m_nBufferEmptyBits -= bits;
167 pStream->m_SmallBuffer += value<<pStream->m_nBufferEmptyBits;
168 }
169 //
170 // overflow of 32 bits. so value should be segement.
171 //
172 else if (bits>=pStream->m_nBufferEmptyBits)
173 {
174 // to reduce code size, no shortcut if bits == m_nBufferEmptyBits
175 // since the probability is low.
176 pStream->m_SmallBuffer+=value>>(bits-pStream->m_nBufferEmptyBits); //shift right
177
178 // ptwrite = (MS_U32 *)(pStream->m_pbFrameBuffer+pStream->m_nByteCount);
179 tlongbuf = pStream->m_SmallBuffer;
180
181 // ------ Little Endian in PC --------
182 if (pStream->m_bUseLittleEndian) {
183 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 0] = (MS_U8)(tlongbuf >> 24);
184 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 1] = (MS_U8)(tlongbuf >> 16);
185 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 2] = (MS_U8)(tlongbuf >> 8);
186 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 3] = (MS_U8)(tlongbuf);
187 }
188 else {
189 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 3] = (MS_U8)(tlongbuf >> 24);
190 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 2] = (MS_U8)(tlongbuf >> 16);
191 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 1] = (MS_U8)(tlongbuf >> 8);
192 pStream->m_pbFrameBuffer[pStream->m_nByteCount + 0] = (MS_U8)(tlongbuf);
193 }
194
195 // -----------------------------------
196 pStream->m_nByteCount+=4;
197 if (bits==pStream->m_nBufferEmptyBits) {
198 pStream->m_nBufferEmptyBits = 32;
199 pStream->m_SmallBuffer = 0;
200 } else {
201 pStream->m_nBufferEmptyBits = 32 - bits + pStream->m_nBufferEmptyBits;
202 pStream->m_SmallBuffer = value << pStream->m_nBufferEmptyBits;
203 }
204 }
205 // }
206 }
207
208 // 14496-2
osNextStartCode(OutStream * pStream)209 void osNextStartCode(OutStream* pStream)
210 {
211 osPutBits(pStream, 0, 1, (MS_S8*)("NextStartCode"));
212 osStuffing(pStream);
213 }
214
GetBitsCount(OutStream * pStream)215 MS_U32 GetBitsCount(OutStream* pStream)
216 {
217 return pStream->BC_nCumulativeBits;
218 }
219
osStuffing(OutStream * pStream)220 void osStuffing(OutStream* pStream)
221 {
222 MS_S32 nStuffingBits = pStream->m_nBufferEmptyBits&7;
223 if (pStream->m_nBufferEmptyBits & 7)
224 osPutBits(pStream, 0xFF>>(8-nStuffingBits), nStuffingBits, (MS_S8*)("stuffing"));
225 }
226