Mercurial > hg > forks > libbpg
diff jctvc/TLibEncoder/AnnexBwrite.h @ 0:772086c29cc7
Initial import.
author | Matti Hamalainen <ccr@tnsp.org> |
---|---|
date | Wed, 16 Nov 2016 11:16:33 +0200 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/jctvc/TLibEncoder/AnnexBwrite.h Wed Nov 16 11:16:33 2016 +0200 @@ -0,0 +1,91 @@ +/* The copyright in this software is being made available under the BSD + * License, included below. This software may be subject to other third party + * and contributor rights, including patent rights, and no such rights are + * granted under this license. + * + * Copyright (c) 2010-2014, ITU/ISO/IEC + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may + * be used to endorse or promote products derived from this software without + * specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF + * THE POSSIBILITY OF SUCH DAMAGE. + */ + +#pragma once + +#ifndef __ANNEXBWRITE__ +#define __ANNEXBWRITE__ + +#include <ostream> +#include "TLibCommon/AccessUnit.h" +#include "NALwrite.h" + +//! \ingroup TLibEncoder +//! \{ + +/** + * write all NALunits in au to bytestream out in a manner satisfying + * AnnexB of AVC. NALunits are written in the order they are found in au. + * the zero_byte word is appended to: + * - the initial startcode in the access unit, + * - any SPS/PPS nal units + */ +static std::vector<UInt> writeAnnexB(std::ostream& out, const AccessUnit& au) +{ + std::vector<UInt> annexBsizes; + + for (AccessUnit::const_iterator it = au.begin(); it != au.end(); it++) + { + const NALUnitEBSP& nalu = **it; + UInt size = 0; /* size of annexB unit in bytes */ + + static const Char start_code_prefix[] = {0,0,0,1}; + if (it == au.begin() || nalu.m_nalUnitType == NAL_UNIT_VPS || nalu.m_nalUnitType == NAL_UNIT_SPS || nalu.m_nalUnitType == NAL_UNIT_PPS) + { + /* From AVC, When any of the following conditions are fulfilled, the + * zero_byte syntax element shall be present: + * - the nal_unit_type within the nal_unit() is equal to 7 (sequence + * parameter set) or 8 (picture parameter set), + * - the byte stream NAL unit syntax structure contains the first NAL + * unit of an access unit in decoding order, as specified by subclause + * 7.4.1.2.3. + */ + out.write(start_code_prefix, 4); + size += 4; + } + else + { + out.write(start_code_prefix+1, 3); + size += 3; + } + out << nalu.m_nalUnitData.str(); + size += UInt(nalu.m_nalUnitData.str().size()); + + annexBsizes.push_back(size); + } + + return annexBsizes; +} +//! \} + +#endif