source: 3DVCSoftware/branches/HTM-5.1-dev2-Mediatek/source/Lib/TLibEncoder/NALwrite.h @ 852

Last change on this file since 852 was 210, checked in by tech, 12 years ago

Reintegrated /branches/HTM-5.0-dev0 rev. 207.

File size: 3.9 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license.
5 *
6 * Copyright (c) 2010-2012, ITU/ISO/IEC
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 *
12 *  * Redistributions of source code must retain the above copyright notice,
13 *    this list of conditions and the following disclaimer.
14 *  * Redistributions in binary form must reproduce the above copyright notice,
15 *    this list of conditions and the following disclaimer in the documentation
16 *    and/or other materials provided with the distribution.
17 *  * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18 *    be used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34#pragma once
35
36#include <ostream>
37
38#include "TLibCommon/TypeDef.h"
39#include "TLibCommon/TComBitStream.h"
40#include "TLibCommon/NAL.h"
41
42//! \ingroup TLibEncoder
43//! \{
44
45/**
46 * A convenience wrapper to NALUnit that also provides a
47 * bitstream object.
48 */
49struct OutputNALUnit : public NALUnit
50{
51  /**
52   * construct an OutputNALunit structure with given header values and
53   * storage for a bitstream.  Upon construction the NALunit header is
54   * written to the bitstream.
55   */
56#if H0388
57  OutputNALUnit(
58    NalUnitType nalUnitType,
59#if NAL_REF_FLAG
60    Bool nalRefFlag,
61#else
62    NalRefIdc nalRefIDC,
63#endif
64#if VIDYO_VPS_INTEGRATION|QC_MVHEVC_B0046
65    unsigned layerId,
66#else
67    Int viewId,
68    Bool isDepth,
69#endif
70    unsigned temporalID = 0)
71#if NAL_REF_FLAG
72#if VIDYO_VPS_INTEGRATION|QC_MVHEVC_B0046
73  : NALUnit(nalUnitType, nalRefFlag, layerId, temporalID)
74#else
75  : NALUnit(nalUnitType, nalRefFlag, viewId, isDepth, temporalID)
76#endif
77#else
78#if VIDYO_VPS_INTEGRATION
79  : NALUnit(nalUnitType, nalRefIDC, layerId, temporalID)
80#else
81  : NALUnit(nalUnitType, nalRefIDC, viewId, isDepth, temporalID)
82#endif
83#endif
84  , m_Bitstream()
85  {}
86#else
87  OutputNALUnit(
88    NalUnitType nalUnitType,
89    NalRefIdc nalRefIDC,
90    Int viewId,
91    Bool isDepth,
92    unsigned temporalID = 0,
93    bool outputFlag = true)
94  : NALUnit(nalUnitType, nalRefIDC, viewId, isDepth, temporalID, outputFlag)
95  , m_Bitstream()
96  {}
97#endif
98
99  OutputNALUnit(const NALUnit& src)
100  {
101    m_Bitstream.clear();
102    static_cast<NALUnit*>(this)->operator=(src);
103  }
104
105  OutputNALUnit& operator=(const NALUnit& src)
106  {
107    m_Bitstream.clear();
108    static_cast<NALUnit*>(this)->operator=(src);
109    return *this;
110  }
111
112  TComOutputBitstream m_Bitstream;
113};
114
115void write(std::ostream& out, OutputNALUnit& nalu);
116void writeRBSPTrailingBits(TComOutputBitstream& bs);
117
118inline NALUnitEBSP::NALUnitEBSP(OutputNALUnit& nalu)
119  : NALUnit(nalu)
120{
121  write(m_nalUnitData, nalu);
122}
123
124void copyNaluData(OutputNALUnit& naluDest, const OutputNALUnit& naluSrc);
125
126//! \}
Note: See TracBrowser for help on using the repository browser.