void CodingConstraintsBox::writeBox(BitStream& bitstr)
{
writeFullBoxHeader(bitstr);
bitstr.writeBits(mAllRefPicsIntra, 1);
bitstr.writeBits(mIntraPredUsed, 1);
bitstr.writeBits(mMaxRefPicUsed, 4);
bitstr.writeBits(0, 26); // reserved int(26)
updateSize(bitstr);
}
int Golomb::encode(uint m, int value, BitStream& bs){
int q, r, i;
int nbits = (int) floor(log2(m)) + 1;
int err;
if (value < 0 )
{
value = 2*abs(value) - 1;
}
else
{
value *= 2;
}
q = value/m;
r = value%m;
for(i=0; i<q;i++)
{
if ((err = bs.writeBit(1))!= 0)
return err;
}
if ((err = bs.writeBit(0))!= 0)
return err;
if ((err = bs.writeBits(r, nbits))!= 0){
return err;
}
return 0;
}
void rbucEncodeCPUOneLevel(BitStream& bitStream, std::vector<Symbol16>& lengths, const std::vector<Symbol16>& symbols, uint branchFactor)
{
rbucEncodeCPUBuildLengthsOneLevel(lengths, symbols, branchFactor);
uint symbolCount = (uint)symbols.size();
for(uint i = 0; i < symbolCount; i++) {
uint length = lengths[i / branchFactor];
bitStream.writeBits(symbols[i], length);
}
}
bool huffmanEncodeCPU(BitStream& bitStream, const std::vector<Symbol16>& symbolStream, const HuffmanEncodeTableCPU& encodeTable, std::vector<uint>& offsets, uint codingBlockSize)
{
uint offsetBase = bitStream.getBitPosition();
for(uint i = 0; i < symbolStream.size(); i++) {
if(codingBlockSize > 0 && i % codingBlockSize == 0) {
offsets.push_back(bitStream.getBitPosition() - offsetBase);
}
Symbol16 symbol = symbolStream[i];
const HuffmanEntry& entry = encodeTable.m_encodeTable[symbol];
bitStream.writeBits(entry.m_codeword, entry.m_codewordLength);
}
return true;
}
void rbucEncodeCPUWriteTree(BitStream& bitStream, std::vector<uint>& offsets, const std::vector<std::vector<Symbol16>>& tree, const std::vector<uint>& branchFactors, uint level, uint index)
{
if(level+2 == tree.size()) {
offsets.push_back(bitStream.getBitPosition());
}
uint bits = (level+1 >= tree.size() ? 8 : tree[level+1][index / branchFactors[level]]);
bitStream.writeBits(tree[level][index], bits);
if(level > 0) {
uint branchFactor = branchFactors[level - 1];
uint childrenCount = std::min(branchFactor, (uint)tree[level - 1].size() - index * branchFactor);
for(uint i = 0; i < childrenCount; i++) {
rbucEncodeCPUWriteTree(bitStream, offsets, tree, branchFactors, level - 1, index * branchFactor + i);
}
}
}
