1 /*****************************************************************
2 |
3 |    AP4 - dvcC Atoms
4 |
5 |    Copyright 2002-2016 Axiomatic Systems, LLC
6 |
7 |
8 |    This file is part of Bento4/AP4 (MP4 Atom Processing Library).
9 |
10 |    Unless you have obtained Bento4 under a difference license,
11 |    this version of Bento4 is Bento4|GPL.
12 |    Bento4|GPL is free software; you can redistribute it and/or modify
13 |    it under the terms of the GNU General Public License as published by
14 |    the Free Software Foundation; either version 2, or (at your option)
15 |    any later version.
16 |
17 |    Bento4|GPL is distributed in the hope that it will be useful,
18 |    but WITHOUT ANY WARRANTY; without even the implied warranty of
19 |    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20 |    GNU General Public License for more details.
21 |
22 |    You should have received a copy of the GNU General Public License
23 |    along with Bento4|GPL; see the file COPYING.  If not, write to the
24 |    Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
25 |    02111-1307, USA.
26 |
27 ****************************************************************/
28 
29 /*----------------------------------------------------------------------
30 |   includes
31 +---------------------------------------------------------------------*/
32 #include "Ap4DvccAtom.h"
33 #include "Ap4AtomFactory.h"
34 #include "Ap4Utils.h"
35 #include "Ap4Types.h"
36 #include "Ap4SampleDescription.h"
37 
38 /*----------------------------------------------------------------------
39 |   dynamic cast support
40 +---------------------------------------------------------------------*/
AP4_DEFINE_DYNAMIC_CAST_ANCHOR(AP4_DvccAtom) const41 AP4_DEFINE_DYNAMIC_CAST_ANCHOR(AP4_DvccAtom)
42 
43 /*----------------------------------------------------------------------
44 |   AP4_DvccAtom::GetProfileName
45 +---------------------------------------------------------------------*/
46 const char*
47 AP4_DvccAtom::GetProfileName(AP4_UI08 profile)
48 {
49     switch (profile) {
50         case AP4_DV_PROFILE_DVAV_PER: return "dvav.per";
51         case AP4_DV_PROFILE_DVAV_PEN: return "dvav.pen";
52         case AP4_DV_PROFILE_DVHE_DER: return "dvhe.der";
53         case AP4_DV_PROFILE_DVHE_DEN: return "dvhe.den";
54         case AP4_DV_PROFILE_DVHE_DTR: return "dvhe.dtr";
55         case AP4_DV_PROFILE_DVHE_STN: return "dvhe.stn";
56         case AP4_DV_PROFILE_DVHE_DTH: return "dvhe.dth";
57         case AP4_DV_PROFILE_DVHE_DTB: return "dvhr.dtb";
58         case AP4_DV_PROFILE_DVHE_ST : return "dvhe.st";
59         case AP4_DV_PROFILE_DVHE_SE : return "dvav.se";
60     }
61 
62     return NULL;
63 }
64 
65 /*----------------------------------------------------------------------
66 |   AP4_DvccAtom::Create
67 +---------------------------------------------------------------------*/
68 AP4_DvccAtom*
Create(AP4_Size size,AP4_ByteStream & stream)69 AP4_DvccAtom::Create(AP4_Size size, AP4_ByteStream& stream)
70 {
71     if (size < AP4_ATOM_HEADER_SIZE+24) return NULL;
72 
73     AP4_UI08 payload[24];
74     AP4_Result result = stream.Read(payload, 24);
75     if (AP4_FAILED(result)) return NULL;
76 
77     return new AP4_DvccAtom(payload[0],
78                             payload[1],
79                             payload[2]>>1,
80                             ((payload[2]&1)<<5) | (payload[3]>>3),
81                             (payload[3]&4) != 0,
82                             (payload[3]&2) != 0,
83                             (payload[3]&1) != 0,
84                             payload[4]>>4);
85 }
86 
87 /*----------------------------------------------------------------------
88 |   AP4_DvccAtom::AP4_DvccAtom
89 +---------------------------------------------------------------------*/
AP4_DvccAtom()90 AP4_DvccAtom::AP4_DvccAtom() :
91     AP4_Atom(AP4_ATOM_TYPE_DVCC, AP4_ATOM_HEADER_SIZE+24),
92     m_DvVersionMajor(0),
93     m_DvVersionMinor(0),
94     m_DvProfile(0),
95     m_DvLevel(0),
96     m_RpuPresentFlag(false),
97     m_ElPresentFlag(false),
98     m_BlPresentFlag(false),
99     m_DvBlSignalCompatibilityID(0)
100 {
101 }
102 
103 /*----------------------------------------------------------------------
104 |   AP4_DvccAtom::AP4_DvccAtom
105 +---------------------------------------------------------------------*/
AP4_DvccAtom(AP4_UI08 dv_version_major,AP4_UI08 dv_version_minor,AP4_UI08 dv_profile,AP4_UI08 dv_level,bool rpu_present_flag,bool el_present_flag,bool bl_present_flag,AP4_UI08 dv_bl_signal_compatibility_id)106 AP4_DvccAtom::AP4_DvccAtom(AP4_UI08 dv_version_major,
107                            AP4_UI08 dv_version_minor,
108                            AP4_UI08 dv_profile,
109                            AP4_UI08 dv_level,
110                            bool     rpu_present_flag,
111                            bool     el_present_flag,
112                            bool     bl_present_flag,
113                            AP4_UI08 dv_bl_signal_compatibility_id) :
114     AP4_Atom((dv_profile>7) ? AP4_ATOM_TYPE_DVVC : AP4_ATOM_TYPE_DVCC, AP4_ATOM_HEADER_SIZE+24),
115     m_DvVersionMajor(dv_version_major),
116     m_DvVersionMinor(dv_version_minor),
117     m_DvProfile(dv_profile),
118     m_DvLevel(dv_level),
119     m_RpuPresentFlag(rpu_present_flag),
120     m_ElPresentFlag(el_present_flag),
121     m_BlPresentFlag(bl_present_flag),
122     m_DvBlSignalCompatibilityID(dv_bl_signal_compatibility_id)
123 {
124 }
125 
126 /*----------------------------------------------------------------------
127 |   AP4_DvccAtom::GetCodecString
128 +---------------------------------------------------------------------*/
129 AP4_Result
GetCodecString(AP4_SampleDescription * parent,AP4_String & codec)130 AP4_DvccAtom::GetCodecString(AP4_SampleDescription* parent, AP4_String& codec)
131 {
132     char workspace[64];
133 
134     AP4_UI32 format = parent->GetFormat();
135     if (format == AP4_ATOM_TYPE_DVAV ||
136         format == AP4_ATOM_TYPE_DVA1 ||
137         format == AP4_ATOM_TYPE_DVHE ||
138         format == AP4_ATOM_TYPE_DVH1) {
139         /* Non backward-compatible */
140         char coding[5];
141         AP4_FormatFourChars(coding, format);
142         AP4_FormatString(workspace,
143                          sizeof(workspace),
144                          "%s.%02d.%02d",
145                          coding,
146                          GetDvProfile(),
147                          GetDvLevel());
148         codec = workspace;
149     } else {
150         /* Backward-compatible */
151         switch (format) {
152           case AP4_ATOM_TYPE_AVC1:
153             format = AP4_ATOM_TYPE_DVA1;
154             break;
155 
156           case AP4_ATOM_TYPE_AVC3:
157             format = AP4_ATOM_TYPE_DVAV;
158             break;
159 
160           case AP4_ATOM_TYPE_HEV1:
161             format = AP4_ATOM_TYPE_DVHE;
162             break;
163 
164           case AP4_ATOM_TYPE_HVC1:
165             format = AP4_ATOM_TYPE_DVH1;
166             break;
167         }
168         char coding[5];
169         AP4_FormatFourChars(coding, format);
170         AP4_String parent_codec;
171         parent->GetCodecString(parent_codec);
172         AP4_FormatString(workspace,
173                          sizeof(workspace),
174                          "%s,%s.%02d.%02d",
175                          parent_codec.GetChars(),
176                          coding,
177                          GetDvProfile(),
178                          GetDvLevel());
179         codec = workspace;
180     }
181 
182     return AP4_SUCCESS;
183 }
184 
185 /*----------------------------------------------------------------------
186 |   AP4_DvccAtom::WriteFields
187 +---------------------------------------------------------------------*/
188 AP4_Result
WriteFields(AP4_ByteStream & stream)189 AP4_DvccAtom::WriteFields(AP4_ByteStream& stream)
190 {
191     AP4_UI08 payload[24];
192     AP4_SetMemory(payload, 0, 24);
193 
194     payload[0] = m_DvVersionMajor;
195     payload[1] = m_DvVersionMinor;
196     payload[2] = (m_DvProfile<<1) | ((m_DvLevel&0x20)>>5);
197     payload[3] = (m_DvLevel<<3) | (m_RpuPresentFlag?4:0) | (m_ElPresentFlag?2:0) | (m_BlPresentFlag?1:0);
198     payload[4] = m_DvBlSignalCompatibilityID<<4;
199 
200     return stream.Write(payload, 24);
201 }
202 
203 /*----------------------------------------------------------------------
204 |   AP4_DvccAtom::InspectFields
205 +---------------------------------------------------------------------*/
206 AP4_Result
InspectFields(AP4_AtomInspector & inspector)207 AP4_DvccAtom::InspectFields(AP4_AtomInspector& inspector)
208 {
209     inspector.AddField("dv_version_major", m_DvVersionMajor);
210     inspector.AddField("dv_version_minor", m_DvVersionMinor);
211     inspector.AddField("dv_profile", m_DvProfile);
212     const char* profile_name = GetProfileName(m_DvProfile);
213     if (profile_name) {
214         inspector.AddField("dv_profile_name", profile_name);
215     } else {
216         inspector.AddField("dv_profile_name", "unknown");
217     }
218     inspector.AddField("dv_level", m_DvLevel);
219     inspector.AddField("rpu_present_flag", m_RpuPresentFlag);
220     inspector.AddField("el_present_flag",  m_ElPresentFlag);
221     inspector.AddField("bl_present_flag",  m_BlPresentFlag);
222     inspector.AddField("dv_bl_signal_compatibility_id", m_DvBlSignalCompatibilityID);
223     return AP4_SUCCESS;
224 }
225