| /** |
| * Copyright (C) 2022 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <ImsMediaBitWriter.h> |
| #include <ImsMediaTrace.h> |
| #include <string.h> |
| |
| ImsMediaBitWriter::ImsMediaBitWriter() |
| { |
| mBuffer = NULL; |
| mMaxBufferSize = 0; |
| mBytePos = 0; |
| mBitPos = 0; |
| mBitBuffer = 0; |
| mBufferFull = false; |
| } |
| |
| ImsMediaBitWriter::~ImsMediaBitWriter() {} |
| |
| void ImsMediaBitWriter::SetBuffer(uint8_t* pbBuffer, uint32_t nBufferSize) |
| { |
| mBytePos = 0; |
| mBitPos = 0; |
| mBitBuffer = 0; |
| mBufferFull = false; |
| mBuffer = pbBuffer; |
| mMaxBufferSize = nBufferSize; |
| } |
| |
| void ImsMediaBitWriter::Write(uint32_t nValue, uint32_t nSize) |
| { |
| if (nSize == 0) |
| return; |
| |
| if (mBuffer == NULL || nSize > 24 || mBufferFull) |
| { |
| IMLOGE2("[Write] nSize[%d], BufferFull[%d]", nSize, mBufferFull); |
| return; |
| } |
| |
| // write to bit buffer |
| mBitBuffer += (nValue << (32 - nSize) >> mBitPos); |
| mBitPos += nSize; |
| |
| // write to byte buffer |
| while (mBitPos >= 8) |
| { |
| mBuffer[mBytePos++] = (uint8_t)(mBitBuffer >> 24); |
| mBitBuffer <<= 8; |
| mBitPos -= 8; |
| } |
| |
| if (mBytePos >= mMaxBufferSize) |
| { |
| mBufferFull = true; |
| } |
| } |
| |
| void ImsMediaBitWriter::WriteByteBuffer(uint8_t* pbSrc, uint32_t nBitSize) |
| { |
| uint32_t nByteSize; |
| uint32_t nRemainBitSize; |
| nByteSize = nBitSize >> 3; |
| nRemainBitSize = nBitSize & 0x07; |
| |
| if (mBitPos == 0) |
| { |
| memcpy(mBuffer + mBytePos, pbSrc, nByteSize); |
| mBytePos += nByteSize; |
| } |
| else |
| { |
| uint32_t i; |
| |
| for (i = 0; i < nByteSize; i++) |
| { |
| Write(pbSrc[i], 8); |
| } |
| } |
| |
| if (nRemainBitSize > 0) |
| { |
| uint32_t v = pbSrc[nByteSize]; |
| v >>= (8 - nRemainBitSize); |
| Write(v, nRemainBitSize); |
| } |
| } |
| |
| void ImsMediaBitWriter::WriteByteBuffer(uint32_t value) |
| { |
| uint32_t nRemainBitSize = 32; |
| |
| for (int32_t i = 0; i < 4; i++) |
| { |
| nRemainBitSize -= 8; |
| uint8_t v = (value >> nRemainBitSize) & 0x00ff; |
| Write(v, 8); |
| } |
| } |
| |
| void ImsMediaBitWriter::Seek(uint32_t nSize) |
| { |
| Flush(); |
| mBitPos += nSize; |
| |
| while (mBitPos >= 8) |
| { |
| mBytePos++; |
| mBitPos -= 8; |
| } |
| } |
| |
| void ImsMediaBitWriter::AddPadding() |
| { |
| if (mBitPos > 0) |
| { |
| Write(0, 8 - mBitPos); |
| } |
| } |
| |
| uint32_t ImsMediaBitWriter::GetBufferSize() |
| { |
| uint32_t nSize; |
| nSize = (mBitPos + 7) >> 3; |
| nSize += mBytePos; |
| return nSize; |
| } |
| |
| void ImsMediaBitWriter::Flush() |
| { |
| if (mBitPos > 0) |
| { |
| mBuffer[mBytePos] += (uint8_t)(mBitBuffer >> 24); |
| mBitBuffer = 0; |
| } |
| } |