gstreamer/gst/isomp4/atoms.c
Alicia Boya García 5fcb7f715a qtmux: round to nearest when computing mehd and tkhd duration
This fixes a bug where in some files mehd.fragment_duration is one unit
less than the actual duration of the fragmented movie, as explained below:

mehd.fragment_duration is computed by scaling the end timestamp of
the last frame of the movie in (in nanoseconds) by the movie timescale.

In some situations, the end timestamp is innacurate due to lossy conversion to
fixed point required by GstBuffer upstream.

Take for instance a movie with 3 frames at exactly 3 fps.

$ gst-launch-1.0 -v videotestsrc num-buffers=3 \
  ! video/x-raw, framerate="(fraction)3/1" \
  ! x264enc \
  ! fakesink silent=false

dts: 999:59:59.333333334,  pts: 1000:00:00.000000000, duration: 0:00:00.333333333
dts: 999:59:59.666666667,  pts: 1000:00:00.666666666, duration: 0:00:00.333333334
dts: 1000:00:00.000000000, pts: 1000:00:00.333333333, duration: 0:00:00.333333333

The end timestamp is calculated by qtmux in this way:

end timestamp = last frame DTS + last frame DUR - first frame DTS =
  = 1000:00:00.000000000 + 0:00:00.333333333 - 999:59:59.333333334 =
  = 0:00:00.999999999

qtmux needs to round this timestamp to the declared movie timescale, which can
ameliorate this distortion, but it's important that round-neareast is used;
otherwise it would backfire badly.

Take for example a movie with a timescale of 30 units/s.

0.999999999 s * 30 units/s = 29.999999970 units

A round-floor (as it was done before this patch) would set fragment_duration to
29 units, amplifying the original distorsion from 1 nanosecond up to 33
milliseconds less than the correct value. The greatest distortion would occur
in the case where timescale = framerate, where an entire frame duration would
be subtracted.

Also, rounding is added to tkhd duration computation too, which
potentially has the same problem.

https://bugzilla.gnome.org/show_bug.cgi?id=793959
2018-10-27 13:12:56 +01:00

5658 lines
143 KiB
C

/* Quicktime muxer plugin for GStreamer
* Copyright (C) 2008-2010 Thiago Santos <thiagoss@embedded.ufcg.edu.br>
* Copyright (C) 2008 Mark Nauwelaerts <mnauw@users.sf.net>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
/*
* Unless otherwise indicated, Source Code is licensed under MIT license.
* See further explanation attached in License Statement (distributed in the file
* LICENSE).
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
* of the Software, and to permit persons to whom the Software is furnished to do
* so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include "atoms.h"
#include <string.h>
#include <glib.h>
#include <gst/gst.h>
#include <gst/base/gstbytewriter.h>
#include <gst/tag/tag.h>
#include <gst/video/video.h>
/*
* Creates a new AtomsContext for the given flavor.
*/
AtomsContext *
atoms_context_new (AtomsTreeFlavor flavor)
{
AtomsContext *context = g_new0 (AtomsContext, 1);
context->flavor = flavor;
return context;
}
/*
* Frees an AtomsContext and all memory associated with it
*/
void
atoms_context_free (AtomsContext * context)
{
g_free (context);
}
/* -- creation, initialization, clear and free functions -- */
#define SECS_PER_DAY (24 * 60 * 60)
#define LEAP_YEARS_FROM_1904_TO_1970 17
guint64
atoms_get_current_qt_time (void)
{
GTimeVal timeval;
g_get_current_time (&timeval);
/* FIXME this should use UTC coordinated time */
return timeval.tv_sec + (((1970 - 1904) * (guint64) 365) +
LEAP_YEARS_FROM_1904_TO_1970) * SECS_PER_DAY;
}
static void
common_time_info_init (TimeInfo * ti)
{
ti->creation_time = ti->modification_time = atoms_get_current_qt_time ();
ti->timescale = 0;
ti->duration = 0;
}
static void
atom_header_set (Atom * header, guint32 fourcc, gint32 size, gint64 ext_size)
{
header->type = fourcc;
header->size = size;
header->extended_size = ext_size;
}
static void
atom_clear (Atom * atom)
{
}
static void
atom_full_init (AtomFull * full, guint32 fourcc, gint32 size, gint64 ext_size,
guint8 version, guint8 flags[3])
{
atom_header_set (&(full->header), fourcc, size, ext_size);
full->version = version;
full->flags[0] = flags[0];
full->flags[1] = flags[1];
full->flags[2] = flags[2];
}
static void
atom_full_clear (AtomFull * full)
{
atom_clear (&full->header);
}
static void
atom_full_free (AtomFull * full)
{
atom_full_clear (full);
g_free (full);
}
static guint32
atom_full_get_flags_as_uint (AtomFull * full)
{
return full->flags[0] << 16 | full->flags[1] << 8 | full->flags[2];
}
static void
atom_full_set_flags_as_uint (AtomFull * full, guint32 flags_as_uint)
{
full->flags[2] = flags_as_uint & 0xFF;
full->flags[1] = (flags_as_uint & 0xFF00) >> 8;
full->flags[0] = (flags_as_uint & 0xFF0000) >> 16;
}
static AtomInfo *
build_atom_info_wrapper (Atom * atom, gpointer copy_func, gpointer free_func)
{
AtomInfo *info = NULL;
if (atom) {
info = g_new0 (AtomInfo, 1);
info->atom = atom;
info->copy_data_func = copy_func;
info->free_func = free_func;
}
return info;
}
static GList *
atom_info_list_prepend_atom (GList * ai, Atom * atom,
AtomCopyDataFunc copy_func, AtomFreeFunc free_func)
{
if (atom)
return g_list_prepend (ai,
build_atom_info_wrapper (atom, copy_func, free_func));
else
return ai;
}
static void
atom_info_list_free (GList * ai)
{
while (ai) {
AtomInfo *info = (AtomInfo *) ai->data;
info->free_func (info->atom);
g_free (info);
ai = g_list_delete_link (ai, ai);
}
}
static AtomData *
atom_data_new (guint32 fourcc)
{
AtomData *data = g_new0 (AtomData, 1);
atom_header_set (&data->header, fourcc, 0, 0);
return data;
}
static void
atom_data_alloc_mem (AtomData * data, guint32 size)
{
g_free (data->data);
data->data = g_new0 (guint8, size);
data->datalen = size;
}
static AtomData *
atom_data_new_from_data (guint32 fourcc, const guint8 * mem, gsize size)
{
AtomData *data = atom_data_new (fourcc);
atom_data_alloc_mem (data, size);
memcpy (data->data, mem, size);
return data;
}
static AtomData *
atom_data_new_from_gst_buffer (guint32 fourcc, const GstBuffer * buf)
{
AtomData *data = atom_data_new (fourcc);
gsize size = gst_buffer_get_size ((GstBuffer *) buf);
atom_data_alloc_mem (data, size);
gst_buffer_extract ((GstBuffer *) buf, 0, data->data, size);
return data;
}
static void
atom_data_free (AtomData * data)
{
atom_clear (&data->header);
g_free (data->data);
g_free (data);
}
static AtomUUID *
atom_uuid_new (void)
{
AtomUUID *uuid = g_new0 (AtomUUID, 1);
atom_header_set (&uuid->header, FOURCC_uuid, 0, 0);
return uuid;
}
static void
atom_uuid_free (AtomUUID * data)
{
atom_clear (&data->header);
g_free (data->data);
g_free (data);
}
static void
atom_ftyp_init (AtomFTYP * ftyp, guint32 major, guint32 version, GList * brands)
{
gint index;
GList *it = NULL;
atom_header_set (&ftyp->header, FOURCC_ftyp, 16, 0);
ftyp->major_brand = major;
ftyp->version = version;
/* always include major brand as compatible brand */
ftyp->compatible_brands_size = g_list_length (brands) + 1;
ftyp->compatible_brands = g_new (guint32, ftyp->compatible_brands_size);
ftyp->compatible_brands[0] = major;
index = 1;
for (it = brands; it != NULL; it = g_list_next (it)) {
ftyp->compatible_brands[index++] = GPOINTER_TO_UINT (it->data);
}
}
AtomFTYP *
atom_ftyp_new (AtomsContext * context, guint32 major, guint32 version,
GList * brands)
{
AtomFTYP *ftyp = g_new0 (AtomFTYP, 1);
atom_ftyp_init (ftyp, major, version, brands);
return ftyp;
}
void
atom_ftyp_free (AtomFTYP * ftyp)
{
atom_clear (&ftyp->header);
g_free (ftyp->compatible_brands);
ftyp->compatible_brands = NULL;
g_free (ftyp);
}
static void
atom_esds_init (AtomESDS * esds)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&esds->header, FOURCC_esds, 0, 0, 0, flags);
desc_es_init (&esds->es);
}
static AtomESDS *
atom_esds_new (void)
{
AtomESDS *esds = g_new0 (AtomESDS, 1);
atom_esds_init (esds);
return esds;
}
static void
atom_esds_free (AtomESDS * esds)
{
atom_full_clear (&esds->header);
desc_es_descriptor_clear (&esds->es);
g_free (esds);
}
static AtomFRMA *
atom_frma_new (void)
{
AtomFRMA *frma = g_new0 (AtomFRMA, 1);
atom_header_set (&frma->header, FOURCC_frma, 0, 0);
return frma;
}
static void
atom_frma_free (AtomFRMA * frma)
{
atom_clear (&frma->header);
g_free (frma);
}
static AtomWAVE *
atom_wave_new (void)
{
AtomWAVE *wave = g_new0 (AtomWAVE, 1);
atom_header_set (&wave->header, FOURCC_wave, 0, 0);
return wave;
}
static void
atom_wave_free (AtomWAVE * wave)
{
atom_clear (&wave->header);
atom_info_list_free (wave->extension_atoms);
g_free (wave);
}
static void
atom_elst_init (AtomELST * elst)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&elst->header, FOURCC_elst, 0, 0, 0, flags);
elst->entries = 0;
}
static void
atom_elst_clear (AtomELST * elst)
{
GSList *walker;
atom_full_clear (&elst->header);
walker = elst->entries;
while (walker) {
g_free ((EditListEntry *) walker->data);
walker = g_slist_next (walker);
}
g_slist_free (elst->entries);
}
static void
atom_edts_init (AtomEDTS * edts)
{
atom_header_set (&edts->header, FOURCC_edts, 0, 0);
atom_elst_init (&edts->elst);
}
static void
atom_edts_clear (AtomEDTS * edts)
{
atom_clear (&edts->header);
atom_elst_clear (&edts->elst);
}
static AtomEDTS *
atom_edts_new (void)
{
AtomEDTS *edts = g_new0 (AtomEDTS, 1);
atom_edts_init (edts);
return edts;
}
static void
atom_edts_free (AtomEDTS * edts)
{
atom_edts_clear (edts);
g_free (edts);
}
static void
atom_tcmi_init (AtomTCMI * tcmi)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&tcmi->header, FOURCC_tcmi, 0, 0, 0, flags);
}
static void
atom_tcmi_clear (AtomTCMI * tcmi)
{
atom_full_clear (&tcmi->header);
tcmi->text_font = 0;
tcmi->text_face = 0;
tcmi->text_size = 0;
tcmi->text_color[0] = 0;
tcmi->text_color[1] = 0;
tcmi->text_color[2] = 0;
tcmi->bg_color[0] = 0;
tcmi->bg_color[1] = 0;
tcmi->bg_color[2] = 0;
g_free (tcmi->font_name);
tcmi->font_name = NULL;
}
static AtomTMCD *
atom_tmcd_new (void)
{
AtomTMCD *tmcd = g_new0 (AtomTMCD, 1);
atom_header_set (&tmcd->header, FOURCC_tmcd, 0, 0);
atom_tcmi_init (&tmcd->tcmi);
return tmcd;
}
static void
atom_tmcd_free (AtomTMCD * tmcd)
{
atom_clear (&tmcd->header);
atom_tcmi_clear (&tmcd->tcmi);
g_free (tmcd);
}
static void
atom_gmin_init (AtomGMIN * gmin)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&gmin->header, FOURCC_gmin, 0, 0, 0, flags);
}
static void
atom_gmin_clear (AtomGMIN * gmin)
{
atom_full_clear (&gmin->header);
gmin->graphics_mode = 0;
gmin->opcolor[0] = 0;
gmin->opcolor[1] = 0;
gmin->opcolor[2] = 0;
gmin->balance = 0;
gmin->reserved = 0;
}
static void
atom_gmhd_init (AtomGMHD * gmhd)
{
atom_header_set (&gmhd->header, FOURCC_gmhd, 0, 0);
atom_gmin_init (&gmhd->gmin);
}
static void
atom_gmhd_clear (AtomGMHD * gmhd)
{
atom_clear (&gmhd->header);
atom_gmin_clear (&gmhd->gmin);
if (gmhd->tmcd) {
atom_tmcd_free (gmhd->tmcd);
gmhd->tmcd = NULL;
}
}
static AtomGMHD *
atom_gmhd_new (void)
{
AtomGMHD *gmhd = g_new0 (AtomGMHD, 1);
atom_gmhd_init (gmhd);
return gmhd;
}
static void
atom_gmhd_free (AtomGMHD * gmhd)
{
atom_gmhd_clear (gmhd);
g_free (gmhd);
}
static void
atom_sample_entry_init (SampleTableEntry * se, guint32 type)
{
atom_header_set (&se->header, type, 0, 0);
memset (se->reserved, 0, sizeof (guint8) * 6);
se->data_reference_index = 0;
}
static void
atom_sample_entry_free (SampleTableEntry * se)
{
atom_clear (&se->header);
}
static void
sample_entry_mp4a_init (SampleTableEntryMP4A * mp4a)
{
atom_sample_entry_init (&mp4a->se, FOURCC_mp4a);
mp4a->version = 0;
mp4a->revision_level = 0;
mp4a->vendor = 0;
mp4a->channels = 2;
mp4a->sample_size = 16;
mp4a->compression_id = 0;
mp4a->packet_size = 0;
mp4a->sample_rate = 0;
/* following only used if version is 1 */
mp4a->samples_per_packet = 0;
mp4a->bytes_per_packet = 0;
mp4a->bytes_per_frame = 0;
mp4a->bytes_per_sample = 0;
mp4a->extension_atoms = NULL;
}
static SampleTableEntryMP4A *
sample_entry_mp4a_new (void)
{
SampleTableEntryMP4A *mp4a = g_new0 (SampleTableEntryMP4A, 1);
sample_entry_mp4a_init (mp4a);
return mp4a;
}
static void
sample_entry_mp4a_free (SampleTableEntryMP4A * mp4a)
{
atom_sample_entry_free (&mp4a->se);
atom_info_list_free (mp4a->extension_atoms);
g_free (mp4a);
}
static void
sample_entry_tmcd_init (SampleTableEntryTMCD * tmcd)
{
atom_sample_entry_init (&tmcd->se, FOURCC_tmcd);
tmcd->tc_flags = 0;
tmcd->timescale = 0;
tmcd->frame_duration = 0;
tmcd->n_frames = 0;
tmcd->name.language_code = 0;
g_free (tmcd->name.name);
tmcd->name.name = NULL;
}
static SampleTableEntryTMCD *
sample_entry_tmcd_new (void)
{
SampleTableEntryTMCD *tmcd = g_new0 (SampleTableEntryTMCD, 1);
sample_entry_tmcd_init (tmcd);
return tmcd;
}
static void
sample_entry_tmcd_free (SampleTableEntryTMCD * tmcd)
{
atom_sample_entry_free (&tmcd->se);
g_free (tmcd->name.name);
g_free (tmcd);
}
static void
sample_entry_mp4v_init (SampleTableEntryMP4V * mp4v, AtomsContext * context)
{
atom_sample_entry_init (&mp4v->se, FOURCC_mp4v);
mp4v->version = 0;
mp4v->revision_level = 0;
mp4v->vendor = 0;
mp4v->temporal_quality = 0;
mp4v->spatial_quality = 0;
/* qt and ISO base media do not contradict, and examples agree */
mp4v->horizontal_resolution = 0x00480000;
mp4v->vertical_resolution = 0x00480000;
mp4v->datasize = 0;
mp4v->frame_count = 1;
memset (mp4v->compressor, 0, sizeof (guint8) * 32);
mp4v->depth = 0;
mp4v->color_table_id = 0;
mp4v->extension_atoms = NULL;
}
static void
sample_entry_mp4v_free (SampleTableEntryMP4V * mp4v)
{
atom_sample_entry_free (&mp4v->se);
atom_info_list_free (mp4v->extension_atoms);
g_free (mp4v);
}
static SampleTableEntryMP4V *
sample_entry_mp4v_new (AtomsContext * context)
{
SampleTableEntryMP4V *mp4v = g_new0 (SampleTableEntryMP4V, 1);
sample_entry_mp4v_init (mp4v, context);
return mp4v;
}
static void
sample_entry_tx3g_init (SampleTableEntryTX3G * tx3g)
{
atom_sample_entry_init (&tx3g->se, FOURCC_tx3g);
tx3g->display_flags = 0;
tx3g->font_id = 1; /* must be 1 as there is a single font */
tx3g->font_face = 0;
tx3g->foreground_color_rgba = 0xFFFFFFFF; /* white, opaque */
/* can't set this now */
tx3g->default_text_box = 0;
tx3g->font_size = 0;
}
static void
sample_entry_tx3g_free (SampleTableEntryTX3G * tx3g)
{
atom_sample_entry_free (&tx3g->se);
g_free (tx3g);
}
static SampleTableEntryTX3G *
sample_entry_tx3g_new (void)
{
SampleTableEntryTX3G *tx3g = g_new0 (SampleTableEntryTX3G, 1);
sample_entry_tx3g_init (tx3g);
return tx3g;
}
static void
atom_stsd_init (AtomSTSD * stsd)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&stsd->header, FOURCC_stsd, 0, 0, 0, flags);
stsd->entries = NULL;
stsd->n_entries = 0;
}
static void
atom_stsd_remove_entries (AtomSTSD * stsd)
{
GList *walker;
walker = stsd->entries;
while (walker) {
GList *aux = walker;
SampleTableEntry *se = (SampleTableEntry *) aux->data;
walker = g_list_next (walker);
stsd->entries = g_list_remove_link (stsd->entries, aux);
switch (se->kind) {
case AUDIO:
sample_entry_mp4a_free ((SampleTableEntryMP4A *) se);
break;
case VIDEO:
sample_entry_mp4v_free ((SampleTableEntryMP4V *) se);
break;
case SUBTITLE:
sample_entry_tx3g_free ((SampleTableEntryTX3G *) se);
break;
case TIMECODE:
sample_entry_tmcd_free ((SampleTableEntryTMCD *) se);
break;
case CLOSEDCAPTION:
default:
/* best possible cleanup */
atom_sample_entry_free (se);
}
g_list_free (aux);
}
stsd->n_entries = 0;
}
static void
atom_stsd_clear (AtomSTSD * stsd)
{
atom_stsd_remove_entries (stsd);
atom_full_clear (&stsd->header);
}
static void
atom_ctts_init (AtomCTTS * ctts)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&ctts->header, FOURCC_ctts, 0, 0, 0, flags);
atom_array_init (&ctts->entries, 128);
ctts->do_pts = FALSE;
}
static AtomCTTS *
atom_ctts_new (void)
{
AtomCTTS *ctts = g_new0 (AtomCTTS, 1);
atom_ctts_init (ctts);
return ctts;
}
static void
atom_ctts_free (AtomCTTS * ctts)
{
atom_full_clear (&ctts->header);
atom_array_clear (&ctts->entries);
g_free (ctts);
}
static void
atom_svmi_init (AtomSVMI * svmi)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&svmi->header, FOURCC_svmi, 0, 0, 0, flags);
svmi->stereoscopic_composition_type = 0x00;
svmi->is_left_first = FALSE;
}
AtomSVMI *
atom_svmi_new (guint8 stereoscopic_composition_type, gboolean is_left_first)
{
AtomSVMI *svmi = g_new0 (AtomSVMI, 1);
atom_svmi_init (svmi);
svmi->stereoscopic_composition_type = stereoscopic_composition_type;
svmi->is_left_first = is_left_first;
return svmi;
}
static void
atom_svmi_free (AtomSVMI * svmi)
{
g_free (svmi);
}
static void
atom_stts_init (AtomSTTS * stts)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&stts->header, FOURCC_stts, 0, 0, 0, flags);
atom_array_init (&stts->entries, 512);
}
static void
atom_stts_clear (AtomSTTS * stts)
{
atom_full_clear (&stts->header);
atom_array_clear (&stts->entries);
}
static void
atom_stsz_init (AtomSTSZ * stsz)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&stsz->header, FOURCC_stsz, 0, 0, 0, flags);
atom_array_init (&stsz->entries, 1024);
stsz->sample_size = 0;
stsz->table_size = 0;
}
static void
atom_stsz_clear (AtomSTSZ * stsz)
{
atom_full_clear (&stsz->header);
atom_array_clear (&stsz->entries);
stsz->table_size = 0;
}
static void
atom_stsc_init (AtomSTSC * stsc)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&stsc->header, FOURCC_stsc, 0, 0, 0, flags);
atom_array_init (&stsc->entries, 128);
}
static void
atom_stsc_clear (AtomSTSC * stsc)
{
atom_full_clear (&stsc->header);
atom_array_clear (&stsc->entries);
}
static void
atom_co64_init (AtomSTCO64 * co64)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&co64->header, FOURCC_stco, 0, 0, 0, flags);
atom_array_init (&co64->entries, 256);
}
static void
atom_stco64_clear (AtomSTCO64 * stco64)
{
atom_full_clear (&stco64->header);
atom_array_clear (&stco64->entries);
}
static void
atom_stss_init (AtomSTSS * stss)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&stss->header, FOURCC_stss, 0, 0, 0, flags);
atom_array_init (&stss->entries, 128);
}
static void
atom_stss_clear (AtomSTSS * stss)
{
atom_full_clear (&stss->header);
atom_array_clear (&stss->entries);
}
void
atom_stbl_init (AtomSTBL * stbl)
{
atom_header_set (&stbl->header, FOURCC_stbl, 0, 0);
atom_stts_init (&stbl->stts);
atom_stss_init (&stbl->stss);
atom_stsd_init (&stbl->stsd);
atom_stsz_init (&stbl->stsz);
atom_stsc_init (&stbl->stsc);
stbl->ctts = NULL;
stbl->svmi = NULL;
atom_co64_init (&stbl->stco64);
}
void
atom_stbl_clear (AtomSTBL * stbl)
{
atom_clear (&stbl->header);
atom_stsd_clear (&stbl->stsd);
atom_stts_clear (&stbl->stts);
atom_stss_clear (&stbl->stss);
atom_stsc_clear (&stbl->stsc);
atom_stsz_clear (&stbl->stsz);
if (stbl->ctts) {
atom_ctts_free (stbl->ctts);
}
if (stbl->svmi) {
atom_svmi_free (stbl->svmi);
}
atom_stco64_clear (&stbl->stco64);
}
static void
atom_vmhd_init (AtomVMHD * vmhd, AtomsContext * context)
{
guint8 flags[3] = { 0, 0, 1 };
atom_full_init (&vmhd->header, FOURCC_vmhd, 0, 0, 0, flags);
vmhd->graphics_mode = 0x0;
memset (vmhd->opcolor, 0, sizeof (guint16) * 3);
if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
vmhd->graphics_mode = 0x40;
vmhd->opcolor[0] = 32768;
vmhd->opcolor[1] = 32768;
vmhd->opcolor[2] = 32768;
}
}
static AtomVMHD *
atom_vmhd_new (AtomsContext * context)
{
AtomVMHD *vmhd = g_new0 (AtomVMHD, 1);
atom_vmhd_init (vmhd, context);
return vmhd;
}
static void
atom_vmhd_free (AtomVMHD * vmhd)
{
atom_full_clear (&vmhd->header);
g_free (vmhd);
}
static void
atom_smhd_init (AtomSMHD * smhd)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&smhd->header, FOURCC_smhd, 0, 0, 0, flags);
smhd->balance = 0;
smhd->reserved = 0;
}
static AtomSMHD *
atom_smhd_new (void)
{
AtomSMHD *smhd = g_new0 (AtomSMHD, 1);
atom_smhd_init (smhd);
return smhd;
}
static void
atom_smhd_free (AtomSMHD * smhd)
{
atom_full_clear (&smhd->header);
g_free (smhd);
}
static void
atom_hmhd_free (AtomHMHD * hmhd)
{
atom_full_clear (&hmhd->header);
g_free (hmhd);
}
static void
atom_hdlr_init (AtomHDLR * hdlr, AtomsContext * context)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&hdlr->header, FOURCC_hdlr, 0, 0, 0, flags);
hdlr->component_type = 0;
hdlr->handler_type = 0;
hdlr->manufacturer = 0;
hdlr->flags = 0;
hdlr->flags_mask = 0;
hdlr->name = g_strdup ("");
/* Store the flavor to know how to serialize the 'name' string */
hdlr->flavor = context->flavor;
}
static AtomHDLR *
atom_hdlr_new (AtomsContext * context)
{
AtomHDLR *hdlr = g_new0 (AtomHDLR, 1);
atom_hdlr_init (hdlr, context);
return hdlr;
}
static void
atom_hdlr_clear (AtomHDLR * hdlr)
{
atom_full_clear (&hdlr->header);
if (hdlr->name) {
g_free (hdlr->name);
hdlr->name = NULL;
}
}
static void
atom_hdlr_free (AtomHDLR * hdlr)
{
atom_hdlr_clear (hdlr);
g_free (hdlr);
}
static void
atom_url_init (AtomURL * url)
{
guint8 flags[3] = { 0, 0, 1 };
atom_full_init (&url->header, FOURCC_url_, 0, 0, 0, flags);
url->location = NULL;
}
static void
atom_url_free (AtomURL * url)
{
atom_full_clear (&url->header);
if (url->location) {
g_free (url->location);
url->location = NULL;
}
g_free (url);
}
static AtomURL *
atom_url_new (void)
{
AtomURL *url = g_new0 (AtomURL, 1);
atom_url_init (url);
return url;
}
static AtomFull *
atom_alis_new (void)
{
guint8 flags[3] = { 0, 0, 1 };
AtomFull *alis = g_new0 (AtomFull, 1);
atom_full_init (alis, FOURCC_alis, 0, 0, 0, flags);
return alis;
}
static void
atom_dref_init (AtomDREF * dref, AtomsContext * context)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&dref->header, FOURCC_dref, 0, 0, 0, flags);
/* in either case, alis or url init arranges to set self-contained flag */
if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
/* alis dref for qt */
AtomFull *alis = atom_alis_new ();
dref->entries = g_list_append (dref->entries, alis);
} else {
/* url for iso spec, as 'alis' not specified there */
AtomURL *url = atom_url_new ();
dref->entries = g_list_append (dref->entries, url);
}
}
static void
atom_dref_clear (AtomDREF * dref)
{
GList *walker;
atom_full_clear (&dref->header);
walker = dref->entries;
while (walker) {
GList *aux = walker;
Atom *atom = (Atom *) aux->data;
walker = g_list_next (walker);
dref->entries = g_list_remove_link (dref->entries, aux);
switch (atom->type) {
case FOURCC_alis:
atom_full_free ((AtomFull *) atom);
break;
case FOURCC_url_:
atom_url_free ((AtomURL *) atom);
break;
default:
/* we do nothing, better leak than crash */
break;
}
g_list_free (aux);
}
}
static void
atom_dinf_init (AtomDINF * dinf, AtomsContext * context)
{
atom_header_set (&dinf->header, FOURCC_dinf, 0, 0);
atom_dref_init (&dinf->dref, context);
}
static void
atom_dinf_clear (AtomDINF * dinf)
{
atom_clear (&dinf->header);
atom_dref_clear (&dinf->dref);
}
static void
atom_minf_init (AtomMINF * minf, AtomsContext * context)
{
atom_header_set (&minf->header, FOURCC_minf, 0, 0);
minf->vmhd = NULL;
minf->smhd = NULL;
minf->hmhd = NULL;
minf->gmhd = NULL;
if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
minf->hdlr = atom_hdlr_new (context);
minf->hdlr->component_type = FOURCC_dhlr;
minf->hdlr->handler_type = FOURCC_alis;
} else {
minf->hdlr = NULL;
}
atom_dinf_init (&minf->dinf, context);
atom_stbl_init (&minf->stbl);
}
static void
atom_minf_clear_handlers (AtomMINF * minf)
{
if (minf->vmhd) {
atom_vmhd_free (minf->vmhd);
minf->vmhd = NULL;
}
if (minf->smhd) {
atom_smhd_free (minf->smhd);
minf->smhd = NULL;
}
if (minf->hmhd) {
atom_hmhd_free (minf->hmhd);
minf->hmhd = NULL;
}
if (minf->gmhd) {
atom_gmhd_free (minf->gmhd);
minf->gmhd = NULL;
}
}
static void
atom_minf_clear (AtomMINF * minf)
{
atom_clear (&minf->header);
atom_minf_clear_handlers (minf);
if (minf->hdlr) {
atom_hdlr_free (minf->hdlr);
}
atom_dinf_clear (&minf->dinf);
atom_stbl_clear (&minf->stbl);
}
static void
atom_mdhd_init (AtomMDHD * mdhd)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&mdhd->header, FOURCC_mdhd, 0, 0, 0, flags);
common_time_info_init (&mdhd->time_info);
/* tempting as it may be to simply 0-initialize,
* that will have the demuxer (correctly) come up with 'eng' as language
* so explicitly specify undefined instead */
mdhd->language_code = language_code ("und");
mdhd->quality = 0;
}
static void
atom_mdhd_clear (AtomMDHD * mdhd)
{
atom_full_clear (&mdhd->header);
}
static void
atom_mdia_init (AtomMDIA * mdia, AtomsContext * context)
{
atom_header_set (&mdia->header, FOURCC_mdia, 0, 0);
atom_mdhd_init (&mdia->mdhd);
atom_hdlr_init (&mdia->hdlr, context);
atom_minf_init (&mdia->minf, context);
}
static void
atom_mdia_clear (AtomMDIA * mdia)
{
atom_clear (&mdia->header);
atom_mdhd_clear (&mdia->mdhd);
atom_hdlr_clear (&mdia->hdlr);
atom_minf_clear (&mdia->minf);
}
static void
atom_tkhd_init (AtomTKHD * tkhd, AtomsContext * context)
{
/*
* flags info
* 1 -> track enabled
* 2 -> track in movie
* 4 -> track in preview
*/
guint8 flags[3] = { 0, 0, 7 };
atom_full_init (&tkhd->header, FOURCC_tkhd, 0, 0, 0, flags);
tkhd->creation_time = tkhd->modification_time = atoms_get_current_qt_time ();
tkhd->duration = 0;
tkhd->track_ID = 0;
tkhd->reserved = 0;
tkhd->reserved2[0] = tkhd->reserved2[1] = 0;
tkhd->layer = 0;
tkhd->alternate_group = 0;
tkhd->volume = 0;
tkhd->reserved3 = 0;
memset (tkhd->matrix, 0, sizeof (guint32) * 9);
tkhd->matrix[0] = 1 << 16;
tkhd->matrix[4] = 1 << 16;
tkhd->matrix[8] = 16384 << 16;
tkhd->width = 0;
tkhd->height = 0;
}
static void
atom_tkhd_clear (AtomTKHD * tkhd)
{
atom_full_clear (&tkhd->header);
}
static void
atom_ilst_init (AtomILST * ilst)
{
atom_header_set (&ilst->header, FOURCC_ilst, 0, 0);
ilst->entries = NULL;
}
static AtomILST *
atom_ilst_new (void)
{
AtomILST *ilst = g_new0 (AtomILST, 1);
atom_ilst_init (ilst);
return ilst;
}
static void
atom_ilst_free (AtomILST * ilst)
{
if (ilst->entries)
atom_info_list_free (ilst->entries);
atom_clear (&ilst->header);
g_free (ilst);
}
static void
atom_meta_init (AtomMETA * meta, AtomsContext * context)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&meta->header, FOURCC_meta, 0, 0, 0, flags);
atom_hdlr_init (&meta->hdlr, context);
/* FIXME (ISOM says this is always 0) */
meta->hdlr.component_type = FOURCC_mhlr;
meta->hdlr.handler_type = FOURCC_mdir;
meta->ilst = NULL;
}
static AtomMETA *
atom_meta_new (AtomsContext * context)
{
AtomMETA *meta = g_new0 (AtomMETA, 1);
atom_meta_init (meta, context);
return meta;
}
static void
atom_meta_free (AtomMETA * meta)
{
atom_full_clear (&meta->header);
atom_hdlr_clear (&meta->hdlr);
if (meta->ilst)
atom_ilst_free (meta->ilst);
meta->ilst = NULL;
g_free (meta);
}
static void
atom_udta_init_metatags (AtomUDTA * udta, AtomsContext * context)
{
if (context->flavor != ATOMS_TREE_FLAVOR_3GP) {
if (!udta->meta) {
udta->meta = atom_meta_new (context);
}
if (!udta->meta->ilst) {
udta->meta->ilst = atom_ilst_new ();
}
}
}
static void
atom_udta_init (AtomUDTA * udta, AtomsContext * context)
{
atom_header_set (&udta->header, FOURCC_udta, 0, 0);
udta->meta = NULL;
udta->context = context;
atom_udta_init_metatags (udta, context);
}
static void
atom_udta_clear (AtomUDTA * udta)
{
atom_clear (&udta->header);
if (udta->meta)
atom_meta_free (udta->meta);
udta->meta = NULL;
if (udta->entries)
atom_info_list_free (udta->entries);
}
static void
atom_tref_init (AtomTREF * tref, guint32 reftype)
{
atom_header_set (&tref->header, FOURCC_tref, 0, 0);
tref->reftype = reftype;
atom_array_init (&tref->entries, 128);
}
static void
atom_tref_clear (AtomTREF * tref)
{
atom_clear (&tref->header);
tref->reftype = 0;
atom_array_clear (&tref->entries);
}
AtomTREF *
atom_tref_new (guint32 reftype)
{
AtomTREF *tref;
tref = g_new0 (AtomTREF, 1);
atom_tref_init (tref, reftype);
return tref;
}
static void
atom_tref_free (AtomTREF * tref)
{
atom_tref_clear (tref);
g_free (tref);
}
/* Clear added tags, but keep the context/flavor the same */
void
atom_udta_clear_tags (AtomUDTA * udta)
{
if (udta->entries) {
atom_info_list_free (udta->entries);
udta->entries = NULL;
}
if (udta->meta && udta->meta->ilst->entries) {
atom_info_list_free (udta->meta->ilst->entries);
udta->meta->ilst->entries = NULL;
}
}
static void
atom_tag_data_init (AtomTagData * data)
{
guint8 flags[] = { 0, 0, 0 };
atom_full_init (&data->header, FOURCC_data, 0, 0, 0, flags);
}
static void
atom_tag_data_clear (AtomTagData * data)
{
atom_full_clear (&data->header);
g_free (data->data);
data->datalen = 0;
}
/*
* Fourcc is the tag fourcc
* flags will be truncated to 24bits
*/
static AtomTag *
atom_tag_new (guint32 fourcc, guint32 flags_as_uint)
{
AtomTag *tag = g_new0 (AtomTag, 1);
tag->header.type = fourcc;
atom_tag_data_init (&tag->data);
atom_full_set_flags_as_uint (&tag->data.header, flags_as_uint);
return tag;
}
static void
atom_tag_free (AtomTag * tag)
{
atom_clear (&tag->header);
atom_tag_data_clear (&tag->data);
g_free (tag);
}
static void
atom_mvhd_init (AtomMVHD * mvhd)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&(mvhd->header), FOURCC_mvhd, sizeof (AtomMVHD), 0, 0, flags);
common_time_info_init (&mvhd->time_info);
mvhd->prefered_rate = 1 << 16;
mvhd->volume = 1 << 8;
mvhd->reserved3 = 0;
memset (mvhd->reserved4, 0, sizeof (guint32[2]));
memset (mvhd->matrix, 0, sizeof (guint32[9]));
mvhd->matrix[0] = 1 << 16;
mvhd->matrix[4] = 1 << 16;
mvhd->matrix[8] = 16384 << 16;
mvhd->preview_time = 0;
mvhd->preview_duration = 0;
mvhd->poster_time = 0;
mvhd->selection_time = 0;
mvhd->selection_duration = 0;
mvhd->current_time = 0;
mvhd->next_track_id = 1;
}
static void
atom_mvhd_clear (AtomMVHD * mvhd)
{
atom_full_clear (&mvhd->header);
}
static void
atom_mehd_init (AtomMEHD * mehd)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&mehd->header, FOURCC_mehd, 0, 0, 1, flags);
mehd->fragment_duration = 0;
}
static void
atom_mvex_init (AtomMVEX * mvex)
{
atom_header_set (&mvex->header, FOURCC_mvex, 0, 0);
atom_mehd_init (&mvex->mehd);
mvex->trexs = NULL;
}
static void
atom_trak_init (AtomTRAK * trak, AtomsContext * context)
{
atom_header_set (&trak->header, FOURCC_trak, 0, 0);
atom_tkhd_init (&trak->tkhd, context);
trak->context = context;
atom_udta_init (&trak->udta, context);
trak->edts = NULL;
atom_mdia_init (&trak->mdia, context);
trak->tref = NULL;
}
AtomTRAK *
atom_trak_new (AtomsContext * context)
{
AtomTRAK *trak = g_new0 (AtomTRAK, 1);
atom_trak_init (trak, context);
return trak;
}
static void
atom_trak_clear (AtomTRAK * trak)
{
atom_clear (&trak->header);
atom_tkhd_clear (&trak->tkhd);
if (trak->edts)
atom_edts_free (trak->edts);
atom_udta_clear (&trak->udta);
atom_mdia_clear (&trak->mdia);
if (trak->tref)
atom_tref_free (trak->tref);
}
static void
atom_trak_free (AtomTRAK * trak)
{
atom_trak_clear (trak);
g_free (trak);
}
static void
atom_moov_init (AtomMOOV * moov, AtomsContext * context)
{
atom_header_set (&(moov->header), FOURCC_moov, 0, 0);
atom_mvhd_init (&(moov->mvhd));
atom_mvex_init (&(moov->mvex));
atom_udta_init (&moov->udta, context);
moov->traks = NULL;
moov->context = *context;
}
AtomMOOV *
atom_moov_new (AtomsContext * context)
{
AtomMOOV *moov = g_new0 (AtomMOOV, 1);
atom_moov_init (moov, context);
return moov;
}
static void
atom_trex_free (AtomTREX * trex)
{
atom_full_clear (&trex->header);
g_free (trex);
}
static void
atom_mvex_clear (AtomMVEX * mvex)
{
GList *walker;
atom_clear (&mvex->header);
walker = mvex->trexs;
while (walker) {
atom_trex_free ((AtomTREX *) walker->data);
walker = g_list_next (walker);
}
g_list_free (mvex->trexs);
mvex->trexs = NULL;
}
void
atom_moov_free (AtomMOOV * moov)
{
GList *walker;
atom_clear (&moov->header);
atom_mvhd_clear (&moov->mvhd);
walker = moov->traks;
while (walker) {
atom_trak_free ((AtomTRAK *) walker->data);
walker = g_list_next (walker);
}
g_list_free (moov->traks);
moov->traks = NULL;
atom_udta_clear (&moov->udta);
atom_mvex_clear (&moov->mvex);
g_free (moov);
}
/* -- end of init / free -- */
/* -- copy data functions -- */
static guint8
atom_full_get_version (AtomFull * full)
{
return full->version;
}
static guint64
common_time_info_copy_data (TimeInfo * ti, gboolean trunc_to_32,
guint8 ** buffer, guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (trunc_to_32) {
prop_copy_uint32 ((guint32) ti->creation_time, buffer, size, offset);
prop_copy_uint32 ((guint32) ti->modification_time, buffer, size, offset);
prop_copy_uint32 (ti->timescale, buffer, size, offset);
prop_copy_uint32 ((guint32) ti->duration, buffer, size, offset);
} else {
prop_copy_uint64 (ti->creation_time, buffer, size, offset);
prop_copy_uint64 (ti->modification_time, buffer, size, offset);
prop_copy_uint32 (ti->timescale, buffer, size, offset);
prop_copy_uint64 (ti->duration, buffer, size, offset);
}
return *offset - original_offset;
}
static void
atom_write_size (guint8 ** buffer, guint64 * size, guint64 * offset,
guint64 atom_pos)
{
/* this only works for non-extended atom size, which is OK
* (though it could be made to do mem_move, etc and write extended size) */
prop_copy_uint32 (*offset - atom_pos, buffer, size, &atom_pos);
}
static guint64
atom_copy_empty (Atom * atom, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
prop_copy_uint32 (0, buffer, size, offset);
return *offset - original_offset;
}
guint64
atom_copy_data (Atom * atom, guint8 ** buffer, guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
/* copies type and size */
prop_copy_uint32 (atom->size, buffer, size, offset);
prop_copy_fourcc (atom->type, buffer, size, offset);
/* extended size needed */
if (atom->size == 1) {
/* really should not happen other than with mdat atom;
* would be a problem for size (re)write code, not to mention memory */
g_return_val_if_fail (atom->type == FOURCC_mdat, 0);
prop_copy_uint64 (atom->extended_size, buffer, size, offset);
}
return *offset - original_offset;
}
static guint64
atom_full_copy_data (AtomFull * atom, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&atom->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint8 (atom->version, buffer, size, offset);
prop_copy_uint8_array (atom->flags, 3, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_info_list_copy_data (GList * ai, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
while (ai) {
AtomInfo *info = (AtomInfo *) ai->data;
if (!info->copy_data_func (info->atom, buffer, size, offset)) {
return 0;
}
ai = g_list_next (ai);
}
return *offset - original_offset;
}
static guint64
atom_data_copy_data (AtomData * data, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&data->header, buffer, size, offset)) {
return 0;
}
if (data->datalen)
prop_copy_uint8_array (data->data, data->datalen, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_uuid_copy_data (AtomUUID * uuid, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&uuid->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint8_array (uuid->uuid, 16, buffer, size, offset);
if (uuid->datalen)
prop_copy_uint8_array (uuid->data, uuid->datalen, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_ftyp_copy_data (AtomFTYP * ftyp, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&ftyp->header, buffer, size, offset)) {
return 0;
}
prop_copy_fourcc (ftyp->major_brand, buffer, size, offset);
prop_copy_uint32 (ftyp->version, buffer, size, offset);
prop_copy_fourcc_array (ftyp->compatible_brands, ftyp->compatible_brands_size,
buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_mvhd_copy_data (AtomMVHD * atom, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint8 version;
guint64 original_offset = *offset;
if (!atom_full_copy_data (&(atom->header), buffer, size, offset)) {
return 0;
}
version = atom_full_get_version (&(atom->header));
if (version == 0) {
common_time_info_copy_data (&atom->time_info, TRUE, buffer, size, offset);
} else if (version == 1) {
common_time_info_copy_data (&atom->time_info, FALSE, buffer, size, offset);
} else {
*offset = original_offset;
return 0;
}
prop_copy_uint32 (atom->prefered_rate, buffer, size, offset);
prop_copy_uint16 (atom->volume, buffer, size, offset);
prop_copy_uint16 (atom->reserved3, buffer, size, offset);
prop_copy_uint32_array (atom->reserved4, 2, buffer, size, offset);
prop_copy_uint32_array (atom->matrix, 9, buffer, size, offset);
prop_copy_uint32 (atom->preview_time, buffer, size, offset);
prop_copy_uint32 (atom->preview_duration, buffer, size, offset);
prop_copy_uint32 (atom->poster_time, buffer, size, offset);
prop_copy_uint32 (atom->selection_time, buffer, size, offset);
prop_copy_uint32 (atom->selection_duration, buffer, size, offset);
prop_copy_uint32 (atom->current_time, buffer, size, offset);
prop_copy_uint32 (atom->next_track_id, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_tkhd_copy_data (AtomTKHD * tkhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&tkhd->header, buffer, size, offset)) {
return 0;
}
if (atom_full_get_version (&tkhd->header) == 0) {
prop_copy_uint32 ((guint32) tkhd->creation_time, buffer, size, offset);
prop_copy_uint32 ((guint32) tkhd->modification_time, buffer, size, offset);
prop_copy_uint32 (tkhd->track_ID, buffer, size, offset);
prop_copy_uint32 (tkhd->reserved, buffer, size, offset);
prop_copy_uint32 ((guint32) tkhd->duration, buffer, size, offset);
} else {
prop_copy_uint64 (tkhd->creation_time, buffer, size, offset);
prop_copy_uint64 (tkhd->modification_time, buffer, size, offset);
prop_copy_uint32 (tkhd->track_ID, buffer, size, offset);
prop_copy_uint32 (tkhd->reserved, buffer, size, offset);
prop_copy_uint64 (tkhd->duration, buffer, size, offset);
}
prop_copy_uint32_array (tkhd->reserved2, 2, buffer, size, offset);
prop_copy_uint16 (tkhd->layer, buffer, size, offset);
prop_copy_uint16 (tkhd->alternate_group, buffer, size, offset);
prop_copy_uint16 (tkhd->volume, buffer, size, offset);
prop_copy_uint16 (tkhd->reserved3, buffer, size, offset);
prop_copy_uint32_array (tkhd->matrix, 9, buffer, size, offset);
prop_copy_uint32 (tkhd->width, buffer, size, offset);
prop_copy_uint32 (tkhd->height, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_hdlr_copy_data (AtomHDLR * hdlr, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&hdlr->header, buffer, size, offset)) {
return 0;
}
prop_copy_fourcc (hdlr->component_type, buffer, size, offset);
prop_copy_fourcc (hdlr->handler_type, buffer, size, offset);
prop_copy_fourcc (hdlr->manufacturer, buffer, size, offset);
prop_copy_uint32 (hdlr->flags, buffer, size, offset);
prop_copy_uint32 (hdlr->flags_mask, buffer, size, offset);
if (hdlr->flavor == ATOMS_TREE_FLAVOR_MOV) {
prop_copy_size_string ((guint8 *) hdlr->name, strlen (hdlr->name), buffer,
size, offset);
} else {
/* assume isomedia base is more generic and use null terminated */
prop_copy_null_terminated_string (hdlr->name, buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_vmhd_copy_data (AtomVMHD * vmhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&vmhd->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (vmhd->graphics_mode, buffer, size, offset);
prop_copy_uint16_array (vmhd->opcolor, 3, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_smhd_copy_data (AtomSMHD * smhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&smhd->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (smhd->balance, buffer, size, offset);
prop_copy_uint16 (smhd->reserved, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_hmhd_copy_data (AtomHMHD * hmhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&hmhd->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (hmhd->max_pdu_size, buffer, size, offset);
prop_copy_uint16 (hmhd->avg_pdu_size, buffer, size, offset);
prop_copy_uint32 (hmhd->max_bitrate, buffer, size, offset);
prop_copy_uint32 (hmhd->avg_bitrate, buffer, size, offset);
prop_copy_uint32 (hmhd->sliding_avg_bitrate, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_tcmi_copy_data (AtomTCMI * tcmi, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&tcmi->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (tcmi->text_font, buffer, size, offset);
prop_copy_uint16 (tcmi->text_face, buffer, size, offset);
prop_copy_uint16 (tcmi->text_size, buffer, size, offset);
prop_copy_uint16 (tcmi->text_color[0], buffer, size, offset);
prop_copy_uint16 (tcmi->text_color[1], buffer, size, offset);
prop_copy_uint16 (tcmi->text_color[2], buffer, size, offset);
prop_copy_uint16 (tcmi->bg_color[0], buffer, size, offset);
prop_copy_uint16 (tcmi->bg_color[1], buffer, size, offset);
prop_copy_uint16 (tcmi->bg_color[2], buffer, size, offset);
/* reserved */
prop_copy_uint16 (0, buffer, size, offset);
prop_copy_size_string ((guint8 *) tcmi->font_name, strlen (tcmi->font_name),
buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_tmcd_copy_data (AtomTMCD * tmcd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&tmcd->header, buffer, size, offset)) {
return 0;
}
if (!atom_tcmi_copy_data (&tmcd->tcmi, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_gmin_copy_data (AtomGMIN * gmin, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&gmin->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (gmin->graphics_mode, buffer, size, offset);
prop_copy_uint16 (gmin->opcolor[0], buffer, size, offset);
prop_copy_uint16 (gmin->opcolor[1], buffer, size, offset);
prop_copy_uint16 (gmin->opcolor[2], buffer, size, offset);
prop_copy_uint8 (gmin->balance, buffer, size, offset);
/* reserved */
prop_copy_uint8 (0, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_gmhd_copy_data (AtomGMHD * gmhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&gmhd->header, buffer, size, offset)) {
return 0;
}
if (!atom_gmin_copy_data (&gmhd->gmin, buffer, size, offset)) {
return 0;
}
if (gmhd->tmcd && !atom_tmcd_copy_data (gmhd->tmcd, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static gboolean
atom_url_same_file_flag (AtomURL * url)
{
return (url->header.flags[2] & 0x1) == 1;
}
static guint64
atom_url_copy_data (AtomURL * url, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&url->header, buffer, size, offset)) {
return 0;
}
if (!atom_url_same_file_flag (url)) {
prop_copy_null_terminated_string (url->location, buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
guint64
atom_stts_copy_data (AtomSTTS * stts, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint i;
if (!atom_full_copy_data (&stts->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (atom_array_get_len (&stts->entries), buffer, size, offset);
/* minimize realloc */
prop_copy_ensure_buffer (buffer, size, offset,
8 * atom_array_get_len (&stts->entries));
for (i = 0; i < atom_array_get_len (&stts->entries); i++) {
STTSEntry *entry = &atom_array_index (&stts->entries, i);
prop_copy_uint32 (entry->sample_count, buffer, size, offset);
prop_copy_int32 (entry->sample_delta, buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_sample_entry_copy_data (SampleTableEntry * se, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&se->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint8_array (se->reserved, 6, buffer, size, offset);
prop_copy_uint16 (se->data_reference_index, buffer, size, offset);
return *offset - original_offset;
}
static guint64
atom_esds_copy_data (AtomESDS * esds, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&esds->header, buffer, size, offset)) {
return 0;
}
if (!desc_es_descriptor_copy_data (&esds->es, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_frma_copy_data (AtomFRMA * frma, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&(frma->header), buffer, size, offset))
return 0;
prop_copy_fourcc (frma->media_type, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_hint_sample_entry_copy_data (AtomHintSampleEntry * hse, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_sample_entry_copy_data (&hse->se, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (hse->size, buffer, size, offset);
prop_copy_uint8_array (hse->data, hse->size, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
sample_entry_mp4a_copy_data (SampleTableEntryMP4A * mp4a, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_sample_entry_copy_data (&mp4a->se, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (mp4a->version, buffer, size, offset);
prop_copy_uint16 (mp4a->revision_level, buffer, size, offset);
prop_copy_uint32 (mp4a->vendor, buffer, size, offset);
prop_copy_uint16 (mp4a->channels, buffer, size, offset);
prop_copy_uint16 (mp4a->sample_size, buffer, size, offset);
prop_copy_uint16 (mp4a->compression_id, buffer, size, offset);
prop_copy_uint16 (mp4a->packet_size, buffer, size, offset);
prop_copy_uint32 (mp4a->sample_rate, buffer, size, offset);
/* this should always be 0 for mp4 flavor */
if (mp4a->version == 1) {
prop_copy_uint32 (mp4a->samples_per_packet, buffer, size, offset);
prop_copy_uint32 (mp4a->bytes_per_packet, buffer, size, offset);
prop_copy_uint32 (mp4a->bytes_per_frame, buffer, size, offset);
prop_copy_uint32 (mp4a->bytes_per_sample, buffer, size, offset);
}
if (mp4a->extension_atoms) {
if (!atom_info_list_copy_data (mp4a->extension_atoms, buffer, size, offset))
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
sample_entry_mp4v_copy_data (SampleTableEntryMP4V * mp4v, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_sample_entry_copy_data (&mp4v->se, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (mp4v->version, buffer, size, offset);
prop_copy_uint16 (mp4v->revision_level, buffer, size, offset);
prop_copy_fourcc (mp4v->vendor, buffer, size, offset);
prop_copy_uint32 (mp4v->temporal_quality, buffer, size, offset);
prop_copy_uint32 (mp4v->spatial_quality, buffer, size, offset);
prop_copy_uint16 (mp4v->width, buffer, size, offset);
prop_copy_uint16 (mp4v->height, buffer, size, offset);
prop_copy_uint32 (mp4v->horizontal_resolution, buffer, size, offset);
prop_copy_uint32 (mp4v->vertical_resolution, buffer, size, offset);
prop_copy_uint32 (mp4v->datasize, buffer, size, offset);
prop_copy_uint16 (mp4v->frame_count, buffer, size, offset);
prop_copy_fixed_size_string ((guint8 *) mp4v->compressor, 32, buffer, size,
offset);
prop_copy_uint16 (mp4v->depth, buffer, size, offset);
prop_copy_uint16 (mp4v->color_table_id, buffer, size, offset);
/* extra atoms */
if (mp4v->extension_atoms &&
!atom_info_list_copy_data (mp4v->extension_atoms, buffer, size, offset))
return 0;
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
sample_entry_tx3g_copy_data (SampleTableEntryTX3G * tx3g, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_sample_entry_copy_data (&tx3g->se, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (tx3g->display_flags, buffer, size, offset);
/* reserved */
prop_copy_uint8 (1, buffer, size, offset);
prop_copy_uint8 (-1, buffer, size, offset);
prop_copy_uint32 (0, buffer, size, offset);
prop_copy_uint64 (tx3g->default_text_box, buffer, size, offset);
/* reserved */
prop_copy_uint32 (0, buffer, size, offset);
prop_copy_uint16 (tx3g->font_id, buffer, size, offset);
prop_copy_uint8 (tx3g->font_face, buffer, size, offset);
prop_copy_uint8 (tx3g->font_size, buffer, size, offset);
prop_copy_uint32 (tx3g->foreground_color_rgba, buffer, size, offset);
/* it must have a fonttable atom */
{
Atom atom;
atom_header_set (&atom, FOURCC_ftab, 18, 0);
if (!atom_copy_data (&atom, buffer, size, offset))
return 0;
prop_copy_uint16 (1, buffer, size, offset); /* Count must be 1 */
prop_copy_uint16 (1, buffer, size, offset); /* Font id: 1 */
prop_copy_size_string ((guint8 *) "Serif", 5, buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
sample_entry_tmcd_copy_data (SampleTableEntryTMCD * tmcd, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_sample_entry_copy_data (&tmcd->se, buffer, size, offset)) {
return 0;
}
/* reserved */
prop_copy_uint32 (0, buffer, size, offset);
prop_copy_uint32 (tmcd->tc_flags, buffer, size, offset);
prop_copy_uint32 (tmcd->timescale, buffer, size, offset);
prop_copy_uint32 (tmcd->frame_duration, buffer, size, offset);
prop_copy_uint8 (tmcd->n_frames, buffer, size, offset);
/* reserved */
prop_copy_uint8 (0, buffer, size, offset);
{
Atom atom;
guint64 name_offset = *offset;
atom_header_set (&atom, FOURCC_name, 0, 0);
if (!atom_copy_data (&atom, buffer, size, offset))
return 0;
prop_copy_uint16 (strlen (tmcd->name.name), buffer, size, offset);
prop_copy_uint16 (tmcd->name.language_code, buffer, size, offset);
prop_copy_fixed_size_string ((guint8 *) tmcd->name.name,
strlen (tmcd->name.name), buffer, size, offset);
atom_write_size (buffer, size, offset, name_offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
sample_entry_generic_copy_data (SampleTableEntry * entry, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_sample_entry_copy_data (entry, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_stsz_copy_data (AtomSTSZ * stsz, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint i;
if (!atom_full_copy_data (&stsz->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (stsz->sample_size, buffer, size, offset);
prop_copy_uint32 (stsz->table_size, buffer, size, offset);
if (stsz->sample_size == 0) {
/* minimize realloc */
prop_copy_ensure_buffer (buffer, size, offset, 4 * stsz->table_size);
/* entry count must match sample count */
g_assert (atom_array_get_len (&stsz->entries) == stsz->table_size);
for (i = 0; i < atom_array_get_len (&stsz->entries); i++) {
prop_copy_uint32 (atom_array_index (&stsz->entries, i), buffer, size,
offset);
}
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_stsc_copy_data (AtomSTSC * stsc, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint i, len;
gboolean last_entries_merged = FALSE;
if (!atom_full_copy_data (&stsc->header, buffer, size, offset)) {
return 0;
}
/* Last two entries might be the same size here as we only merge once the
* next chunk is started */
if ((len = atom_array_get_len (&stsc->entries)) > 1 &&
((atom_array_index (&stsc->entries, len - 1)).samples_per_chunk ==
(atom_array_index (&stsc->entries, len - 2)).samples_per_chunk)) {
stsc->entries.len--;
last_entries_merged = TRUE;
}
prop_copy_uint32 (atom_array_get_len (&stsc->entries), buffer, size, offset);
/* minimize realloc */
prop_copy_ensure_buffer (buffer, size, offset,
12 * atom_array_get_len (&stsc->entries));
for (i = 0; i < atom_array_get_len (&stsc->entries); i++) {
STSCEntry *entry = &atom_array_index (&stsc->entries, i);
prop_copy_uint32 (entry->first_chunk, buffer, size, offset);
prop_copy_uint32 (entry->samples_per_chunk, buffer, size, offset);
prop_copy_uint32 (entry->sample_description_index, buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
/* Need to add the last entry again as in "robust" muxing mode we will most
* likely add new samples to the last chunk, thus making the
* samples_per_chunk in the last one different to the second to last one,
* and thus making it wrong to keep them merged
*/
if (last_entries_merged)
stsc->entries.len++;
return *offset - original_offset;
}
guint64
atom_ctts_copy_data (AtomCTTS * ctts, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint i;
if (!atom_full_copy_data (&ctts->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (atom_array_get_len (&ctts->entries), buffer, size, offset);
/* minimize realloc */
prop_copy_ensure_buffer (buffer, size, offset,
8 * atom_array_get_len (&ctts->entries));
for (i = 0; i < atom_array_get_len (&ctts->entries); i++) {
CTTSEntry *entry = &atom_array_index (&ctts->entries, i);
prop_copy_uint32 (entry->samplecount, buffer, size, offset);
prop_copy_uint32 (entry->sampleoffset, buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_svmi_copy_data (AtomSVMI * svmi, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&svmi->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint8 (svmi->stereoscopic_composition_type, buffer, size, offset);
prop_copy_uint8 (svmi->is_left_first ? 1 : 0, buffer, size, offset);
/* stereo-mono change count */
prop_copy_uint32 (0, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_stco64_copy_data (AtomSTCO64 * stco64, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint i;
gboolean trunc_to_32 = stco64->header.header.type == FOURCC_stco;
if (!atom_full_copy_data (&stco64->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (atom_array_get_len (&stco64->entries), buffer, size,
offset);
/* minimize realloc */
prop_copy_ensure_buffer (buffer, size, offset,
8 * atom_array_get_len (&stco64->entries));
for (i = 0; i < atom_array_get_len (&stco64->entries); i++) {
guint64 value =
atom_array_index (&stco64->entries, i) + stco64->chunk_offset;
if (trunc_to_32) {
prop_copy_uint32 ((guint32) value, buffer, size, offset);
} else {
prop_copy_uint64 (value, buffer, size, offset);
}
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_stss_copy_data (AtomSTSS * stss, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint i;
if (atom_array_get_len (&stss->entries) == 0) {
/* FIXME not needing this atom might be confused with error while copying */
return 0;
}
if (!atom_full_copy_data (&stss->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (atom_array_get_len (&stss->entries), buffer, size, offset);
/* minimize realloc */
prop_copy_ensure_buffer (buffer, size, offset,
4 * atom_array_get_len (&stss->entries));
for (i = 0; i < atom_array_get_len (&stss->entries); i++) {
prop_copy_uint32 (atom_array_index (&stss->entries, i), buffer, size,
offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_stsd_copy_data (AtomSTSD * stsd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
GList *walker;
if (!atom_full_copy_data (&stsd->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (stsd->n_entries, buffer, size, offset);
for (walker = g_list_last (stsd->entries); walker != NULL;
walker = g_list_previous (walker)) {
SampleTableEntry *se = (SampleTableEntry *) walker->data;
switch (((Atom *) walker->data)->type) {
case FOURCC_mp4a:
if (!sample_entry_mp4a_copy_data ((SampleTableEntryMP4A *) walker->data,
buffer, size, offset)) {
return 0;
}
break;
case FOURCC_mp4v:
if (!sample_entry_mp4v_copy_data ((SampleTableEntryMP4V *) walker->data,
buffer, size, offset)) {
return 0;
}
break;
default:
if (se->kind == VIDEO) {
if (!sample_entry_mp4v_copy_data ((SampleTableEntryMP4V *)
walker->data, buffer, size, offset)) {
return 0;
}
} else if (se->kind == AUDIO) {
if (!sample_entry_mp4a_copy_data ((SampleTableEntryMP4A *)
walker->data, buffer, size, offset)) {
return 0;
}
} else if (se->kind == SUBTITLE) {
if (!sample_entry_tx3g_copy_data ((SampleTableEntryTX3G *)
walker->data, buffer, size, offset)) {
return 0;
}
} else if (se->kind == TIMECODE) {
if (!sample_entry_tmcd_copy_data ((SampleTableEntryTMCD *)
walker->data, buffer, size, offset)) {
return 0;
}
} else if (se->kind == CLOSEDCAPTION) {
if (!sample_entry_generic_copy_data ((SampleTableEntry *)
walker->data, buffer, size, offset)) {
return 0;
}
} else {
if (!atom_hint_sample_entry_copy_data (
(AtomHintSampleEntry *) walker->data, buffer, size, offset)) {
return 0;
}
}
break;
}
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_stbl_copy_data (AtomSTBL * stbl, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&stbl->header, buffer, size, offset)) {
return 0;
}
if (!atom_stsd_copy_data (&stbl->stsd, buffer, size, offset)) {
return 0;
}
if (!atom_stts_copy_data (&stbl->stts, buffer, size, offset)) {
return 0;
}
/* this atom is optional, so let's check if we need it
* (to avoid false error) */
if (atom_array_get_len (&stbl->stss.entries)) {
if (!atom_stss_copy_data (&stbl->stss, buffer, size, offset)) {
return 0;
}
}
if (!atom_stsc_copy_data (&stbl->stsc, buffer, size, offset)) {
return 0;
}
if (!atom_stsz_copy_data (&stbl->stsz, buffer, size, offset)) {
return 0;
}
if (stbl->ctts && stbl->ctts->do_pts) {
if (!atom_ctts_copy_data (stbl->ctts, buffer, size, offset)) {
return 0;
}
}
if (stbl->svmi) {
if (!atom_svmi_copy_data (stbl->svmi, buffer, size, offset)) {
return 0;
}
}
if (!atom_stco64_copy_data (&stbl->stco64, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_dref_copy_data (AtomDREF * dref, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
GList *walker;
if (!atom_full_copy_data (&dref->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (g_list_length (dref->entries), buffer, size, offset);
walker = dref->entries;
while (walker != NULL) {
Atom *atom = (Atom *) walker->data;
if (atom->type == FOURCC_url_) {
if (!atom_url_copy_data ((AtomURL *) atom, buffer, size, offset))
return 0;
} else if (atom->type == FOURCC_alis) {
if (!atom_full_copy_data ((AtomFull *) atom, buffer, size, offset))
return 0;
} else {
g_error ("Unsupported atom used inside dref atom");
}
walker = g_list_next (walker);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_dinf_copy_data (AtomDINF * dinf, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&dinf->header, buffer, size, offset)) {
return 0;
}
if (!atom_dref_copy_data (&dinf->dref, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return original_offset - *offset;
}
static guint64
atom_minf_copy_data (AtomMINF * minf, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&minf->header, buffer, size, offset)) {
return 0;
}
if (minf->vmhd) {
if (!atom_vmhd_copy_data (minf->vmhd, buffer, size, offset)) {
return 0;
}
} else if (minf->smhd) {
if (!atom_smhd_copy_data (minf->smhd, buffer, size, offset)) {
return 0;
}
} else if (minf->hmhd) {
if (!atom_hmhd_copy_data (minf->hmhd, buffer, size, offset)) {
return 0;
}
} else if (minf->gmhd) {
if (!atom_gmhd_copy_data (minf->gmhd, buffer, size, offset)) {
return 0;
}
}
if (minf->hdlr) {
if (!atom_hdlr_copy_data (minf->hdlr, buffer, size, offset)) {
return 0;
}
}
if (!atom_dinf_copy_data (&minf->dinf, buffer, size, offset)) {
return 0;
}
if (!atom_stbl_copy_data (&minf->stbl, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_mdhd_copy_data (AtomMDHD * mdhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&mdhd->header, buffer, size, offset)) {
return 0;
}
if (!common_time_info_copy_data (&mdhd->time_info,
atom_full_get_version (&mdhd->header) == 0, buffer, size, offset)) {
return 0;
}
prop_copy_uint16 (mdhd->language_code, buffer, size, offset);
prop_copy_uint16 (mdhd->quality, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_mdia_copy_data (AtomMDIA * mdia, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&mdia->header, buffer, size, offset)) {
return 0;
}
if (!atom_mdhd_copy_data (&mdia->mdhd, buffer, size, offset)) {
return 0;
}
if (!atom_hdlr_copy_data (&mdia->hdlr, buffer, size, offset)) {
return 0;
}
if (!atom_minf_copy_data (&mdia->minf, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_elst_copy_data (AtomELST * elst, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
GSList *walker;
if (!atom_full_copy_data (&elst->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (g_slist_length (elst->entries), buffer, size, offset);
for (walker = elst->entries; walker != NULL; walker = g_slist_next (walker)) {
EditListEntry *entry = (EditListEntry *) walker->data;
prop_copy_uint32 (entry->duration, buffer, size, offset);
prop_copy_uint32 (entry->media_time, buffer, size, offset);
prop_copy_uint32 (entry->media_rate, buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_tref_copy_data (AtomTREF * tref, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint i;
g_assert (atom_array_get_len (&tref->entries) > 0);
if (!atom_copy_data (&tref->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (8 + 4 * atom_array_get_len (&tref->entries), buffer, size,
offset);
prop_copy_fourcc (tref->reftype, buffer, size, offset);
/* minimize realloc */
prop_copy_ensure_buffer (buffer, size, offset,
4 * atom_array_get_len (&tref->entries));
for (i = 0; i < atom_array_get_len (&tref->entries); i++) {
prop_copy_uint32 (atom_array_index (&tref->entries, i), buffer, size,
offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_edts_copy_data (AtomEDTS * edts, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&(edts->header), buffer, size, offset))
return 0;
if (!atom_elst_copy_data (&(edts->elst), buffer, size, offset))
return 0;
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_tag_data_copy_data (AtomTagData * data, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&data->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (data->reserved, buffer, size, offset);
prop_copy_uint8_array (data->data, data->datalen, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_tag_copy_data (AtomTag * tag, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&tag->header, buffer, size, offset)) {
return 0;
}
if (!atom_tag_data_copy_data (&tag->data, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_ilst_copy_data (AtomILST * ilst, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&ilst->header, buffer, size, offset)) {
return 0;
}
/* extra atoms */
if (ilst->entries &&
!atom_info_list_copy_data (ilst->entries, buffer, size, offset))
return 0;
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_meta_copy_data (AtomMETA * meta, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&meta->header, buffer, size, offset)) {
return 0;
}
if (!atom_hdlr_copy_data (&meta->hdlr, buffer, size, offset)) {
return 0;
}
if (meta->ilst) {
if (!atom_ilst_copy_data (meta->ilst, buffer, size, offset)) {
return 0;
}
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_udta_copy_data (AtomUDTA * udta, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&udta->header, buffer, size, offset)) {
return 0;
}
if (udta->meta) {
if (!atom_meta_copy_data (udta->meta, buffer, size, offset)) {
return 0;
}
}
if (udta->entries) {
/* extra atoms */
if (!atom_info_list_copy_data (udta->entries, buffer, size, offset))
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_mehd_copy_data (AtomMEHD * mehd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&mehd->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint64 (mehd->fragment_duration, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_trex_copy_data (AtomTREX * trex, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&trex->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (trex->track_ID, buffer, size, offset);
prop_copy_uint32 (trex->default_sample_description_index, buffer, size,
offset);
prop_copy_uint32 (trex->default_sample_duration, buffer, size, offset);
prop_copy_uint32 (trex->default_sample_size, buffer, size, offset);
prop_copy_uint32 (trex->default_sample_flags, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_mvex_copy_data (AtomMVEX * mvex, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
GList *walker;
if (!atom_copy_data (&mvex->header, buffer, size, offset)) {
return 0;
}
if (!atom_mehd_copy_data (&mvex->mehd, buffer, size, offset)) {
return 0;
}
walker = g_list_first (mvex->trexs);
while (walker != NULL) {
if (!atom_trex_copy_data ((AtomTREX *) walker->data, buffer, size, offset)) {
return 0;
}
walker = g_list_next (walker);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_trak_copy_data (AtomTRAK * trak, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&trak->header, buffer, size, offset)) {
return 0;
}
if (!atom_tkhd_copy_data (&trak->tkhd, buffer, size, offset)) {
return 0;
}
if (trak->tapt) {
if (!trak->tapt->copy_data_func (trak->tapt->atom, buffer, size, offset)) {
return 0;
}
}
if (trak->edts) {
if (!atom_edts_copy_data (trak->edts, buffer, size, offset)) {
return 0;
}
}
if (trak->tref) {
/* Make sure we need this atom (there is a referenced track */
if (atom_array_get_len (&trak->tref->entries) > 0) {
if (!atom_tref_copy_data (trak->tref, buffer, size, offset)) {
return 0;
}
}
}
if (!atom_mdia_copy_data (&trak->mdia, buffer, size, offset)) {
return 0;
}
if (!atom_udta_copy_data (&trak->udta, buffer, size, offset)) {
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_moov_copy_data (AtomMOOV * atom, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
GList *walker;
if (!atom_copy_data (&(atom->header), buffer, size, offset))
return 0;
if (!atom_mvhd_copy_data (&(atom->mvhd), buffer, size, offset))
return 0;
walker = g_list_first (atom->traks);
while (walker != NULL) {
if (!atom_trak_copy_data ((AtomTRAK *) walker->data, buffer, size, offset)) {
return 0;
}
walker = g_list_next (walker);
}
if (!atom_udta_copy_data (&atom->udta, buffer, size, offset)) {
return 0;
}
if (atom->fragmented) {
if (!atom_mvex_copy_data (&atom->mvex, buffer, size, offset)) {
return 0;
}
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_wave_copy_data (AtomWAVE * wave, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_copy_data (&(wave->header), buffer, size, offset))
return 0;
if (wave->extension_atoms) {
if (!atom_info_list_copy_data (wave->extension_atoms, buffer, size, offset))
return 0;
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
/* -- end of copy data functions -- */
/* -- general functions, API and support functions */
/* add samples to tables */
void
atom_stsc_add_new_entry (AtomSTSC * stsc, guint32 first_chunk, guint32 nsamples)
{
gint len;
if ((len = atom_array_get_len (&stsc->entries)) > 1 &&
((atom_array_index (&stsc->entries, len - 1)).samples_per_chunk ==
(atom_array_index (&stsc->entries, len - 2)).samples_per_chunk)) {
STSCEntry *nentry;
/* Merge last two entries as they have the same number of samples per chunk */
nentry = &atom_array_index (&stsc->entries, len - 1);
nentry->first_chunk = first_chunk;
nentry->samples_per_chunk = nsamples;
nentry->sample_description_index = 1;
} else {
STSCEntry nentry;
nentry.first_chunk = first_chunk;
nentry.samples_per_chunk = nsamples;
nentry.sample_description_index = 1;
atom_array_append (&stsc->entries, nentry, 128);
}
}
static void
atom_stsc_update_entry (AtomSTSC * stsc, guint32 first_chunk, guint32 nsamples)
{
gint len;
len = atom_array_get_len (&stsc->entries);
g_assert (len != 0);
g_assert (atom_array_index (&stsc->entries,
len - 1).first_chunk == first_chunk);
atom_array_index (&stsc->entries, len - 1).samples_per_chunk += nsamples;
}
static void
atom_stts_add_entry (AtomSTTS * stts, guint32 sample_count, gint32 sample_delta)
{
STTSEntry *entry = NULL;
if (G_LIKELY (atom_array_get_len (&stts->entries) != 0))
entry = &atom_array_index (&stts->entries,
atom_array_get_len (&stts->entries) - 1);
if (entry && entry->sample_delta == sample_delta) {
entry->sample_count += sample_count;
} else {
STTSEntry nentry;
nentry.sample_count = sample_count;
nentry.sample_delta = sample_delta;
atom_array_append (&stts->entries, nentry, 256);
}
}
static void
atom_stsz_add_entry (AtomSTSZ * stsz, guint32 nsamples, guint32 size)
{
guint32 i;
stsz->table_size += nsamples;
if (stsz->sample_size != 0) {
/* it is constant size, we don't need entries */
return;
}
for (i = 0; i < nsamples; i++) {
atom_array_append (&stsz->entries, size, 1024);
}
}
static guint32
atom_stco64_get_entry_count (AtomSTCO64 * stco64)
{
return atom_array_get_len (&stco64->entries);
}
/* returns TRUE if a new entry was added */
static gboolean
atom_stco64_add_entry (AtomSTCO64 * stco64, guint64 entry)
{
guint32 len;
/* Only add a new entry if the chunk offset changed */
if ((len = atom_array_get_len (&stco64->entries)) &&
((atom_array_index (&stco64->entries, len - 1)) == entry))
return FALSE;
atom_array_append (&stco64->entries, entry, 256);
if (entry > G_MAXUINT32)
stco64->header.header.type = FOURCC_co64;
return TRUE;
}
void
atom_tref_add_entry (AtomTREF * tref, guint32 sample)
{
atom_array_append (&tref->entries, sample, 512);
}
static void
atom_stss_add_entry (AtomSTSS * stss, guint32 sample)
{
atom_array_append (&stss->entries, sample, 512);
}
static void
atom_stbl_add_stss_entry (AtomSTBL * stbl)
{
guint32 sample_index = stbl->stsz.table_size;
atom_stss_add_entry (&stbl->stss, sample_index);
}
static void
atom_ctts_add_entry (AtomCTTS * ctts, guint32 nsamples, guint32 offset)
{
CTTSEntry *entry = NULL;
if (G_LIKELY (atom_array_get_len (&ctts->entries) != 0))
entry = &atom_array_index (&ctts->entries,
atom_array_get_len (&ctts->entries) - 1);
if (entry == NULL || entry->sampleoffset != offset) {
CTTSEntry nentry;
nentry.samplecount = nsamples;
nentry.sampleoffset = offset;
atom_array_append (&ctts->entries, nentry, 256);
if (offset != 0)
ctts->do_pts = TRUE;
} else {
entry->samplecount += nsamples;
}
}
static void
atom_stbl_add_ctts_entry (AtomSTBL * stbl, guint32 nsamples, guint32 offset)
{
if (stbl->ctts == NULL) {
stbl->ctts = atom_ctts_new ();
}
atom_ctts_add_entry (stbl->ctts, nsamples, offset);
}
void
atom_stbl_add_samples (AtomSTBL * stbl, guint32 nsamples, guint32 delta,
guint32 size, guint64 chunk_offset, gboolean sync, gint64 pts_offset)
{
atom_stts_add_entry (&stbl->stts, nsamples, delta);
atom_stsz_add_entry (&stbl->stsz, nsamples, size);
if (atom_stco64_add_entry (&stbl->stco64, chunk_offset)) {
atom_stsc_add_new_entry (&stbl->stsc,
atom_stco64_get_entry_count (&stbl->stco64), nsamples);
} else {
atom_stsc_update_entry (&stbl->stsc,
atom_stco64_get_entry_count (&stbl->stco64), nsamples);
}
if (sync)
atom_stbl_add_stss_entry (stbl);
/* always store to arrange for consistent content */
atom_stbl_add_ctts_entry (stbl, nsamples, pts_offset);
}
void
atom_trak_add_samples (AtomTRAK * trak, guint32 nsamples, guint32 delta,
guint32 size, guint64 chunk_offset, gboolean sync, gint64 pts_offset)
{
AtomSTBL *stbl = &trak->mdia.minf.stbl;
atom_stbl_add_samples (stbl, nsamples, delta, size, chunk_offset, sync,
pts_offset);
}
/* trak and moov molding */
guint32
atom_trak_get_timescale (AtomTRAK * trak)
{
return trak->mdia.mdhd.time_info.timescale;
}
guint32
atom_trak_get_id (AtomTRAK * trak)
{
return trak->tkhd.track_ID;
}
static void
atom_trak_set_id (AtomTRAK * trak, guint32 id)
{
trak->tkhd.track_ID = id;
}
static void
atom_moov_add_trex (AtomMOOV * moov, AtomTREX * trex)
{
moov->mvex.trexs = g_list_append (moov->mvex.trexs, trex);
}
static AtomTREX *
atom_trex_new (AtomTRAK * trak)
{
guint8 flags[3] = { 0, 0, 0 };
AtomTREX *trex = g_new0 (AtomTREX, 1);
atom_full_init (&trex->header, FOURCC_trex, 0, 0, 0, flags);
trex->track_ID = trak->tkhd.track_ID;
trex->default_sample_description_index = 1;
trex->default_sample_duration = 0;
trex->default_sample_size = 0;
trex->default_sample_flags = 0;
return trex;
}
void
atom_moov_add_trak (AtomMOOV * moov, AtomTRAK * trak)
{
atom_trak_set_id (trak, moov->mvhd.next_track_id++);
moov->traks = g_list_append (moov->traks, trak);
/* additional trak means also new trex */
atom_moov_add_trex (moov, atom_trex_new (trak));
}
guint
atom_moov_get_trak_count (AtomMOOV * moov)
{
return g_list_length (moov->traks);
}
static guint64
atom_trak_get_duration (AtomTRAK * trak)
{
return trak->tkhd.duration;
}
static guint64
atom_stts_get_total_duration (AtomSTTS * stts)
{
guint i;
guint64 sum = 0;
for (i = 0; i < atom_array_get_len (&stts->entries); i++) {
STTSEntry *entry = &atom_array_index (&stts->entries, i);
sum += (guint64) (entry->sample_count) * entry->sample_delta;
}
return sum;
}
static void
atom_trak_update_duration (AtomTRAK * trak, guint64 moov_timescale)
{
trak->mdia.mdhd.time_info.duration =
atom_stts_get_total_duration (&trak->mdia.minf.stbl.stts);
if (trak->mdia.mdhd.time_info.timescale != 0) {
trak->tkhd.duration =
gst_util_uint64_scale_round (trak->mdia.mdhd.time_info.duration,
moov_timescale, trak->mdia.mdhd.time_info.timescale);
} else {
trak->tkhd.duration = 0;
}
}
static void
timecode_atom_trak_set_duration (AtomTRAK * trak, guint64 duration,
guint64 timescale)
{
STTSEntry *entry;
GList *iter;
/* Sanity checks to ensure we have a timecode */
g_assert (trak->mdia.minf.gmhd != NULL);
g_assert (atom_array_get_len (&trak->mdia.minf.stbl.stts.entries) == 1);
for (iter = trak->mdia.minf.stbl.stsd.entries; iter;
iter = g_list_next (iter)) {
SampleTableEntry *entry = iter->data;
if (entry->kind == TIMECODE) {
SampleTableEntryTMCD *tmcd = (SampleTableEntryTMCD *) entry;
duration = duration * tmcd->timescale / timescale;
timescale = tmcd->timescale;
break;
}
}
trak->tkhd.duration = duration;
trak->mdia.mdhd.time_info.duration = duration;
trak->mdia.mdhd.time_info.timescale = timescale;
entry = &atom_array_index (&trak->mdia.minf.stbl.stts.entries, 0);
entry->sample_delta = duration;
}
static guint32
atom_moov_get_timescale (AtomMOOV * moov)
{
return moov->mvhd.time_info.timescale;
}
void
atom_moov_update_timescale (AtomMOOV * moov, guint32 timescale)
{
moov->mvhd.time_info.timescale = timescale;
}
void
atom_moov_update_duration (AtomMOOV * moov)
{
GList *traks = moov->traks;
guint64 dur, duration = 0;
while (traks) {
AtomTRAK *trak = (AtomTRAK *) traks->data;
/* Skip timecodes for now: they have a placeholder duration */
if (trak->mdia.minf.gmhd == NULL || trak->mdia.minf.gmhd->tmcd == NULL) {
atom_trak_update_duration (trak, atom_moov_get_timescale (moov));
dur = atom_trak_get_duration (trak);
if (dur > duration)
duration = dur;
}
traks = g_list_next (traks);
}
/* Now update the duration of the timecodes */
traks = moov->traks;
while (traks) {
AtomTRAK *trak = (AtomTRAK *) traks->data;
if (trak->mdia.minf.gmhd != NULL && trak->mdia.minf.gmhd->tmcd != NULL)
timecode_atom_trak_set_duration (trak, duration,
atom_moov_get_timescale (moov));
traks = g_list_next (traks);
}
moov->mvhd.time_info.duration = duration;
moov->mvex.mehd.fragment_duration = duration;
}
void
atom_moov_set_fragmented (AtomMOOV * moov, gboolean fragmented)
{
moov->fragmented = fragmented;
}
void
atom_stco64_chunks_set_offset (AtomSTCO64 * stco64, guint32 offset)
{
stco64->chunk_offset = offset;
}
void
atom_moov_chunks_set_offset (AtomMOOV * moov, guint32 offset)
{
GList *traks = moov->traks;
if (offset == moov->chunks_offset)
return; /* Nothing to do */
while (traks) {
AtomTRAK *trak = (AtomTRAK *) traks->data;
atom_stco64_chunks_set_offset (&trak->mdia.minf.stbl.stco64, offset);
traks = g_list_next (traks);
}
moov->chunks_offset = offset;
}
void
atom_trak_update_bitrates (AtomTRAK * trak, guint32 avg_bitrate,
guint32 max_bitrate)
{
AtomESDS *esds = NULL;
AtomData *btrt = NULL;
AtomWAVE *wave = NULL;
AtomSTSD *stsd;
GList *iter;
GList *extensioniter = NULL;
g_return_if_fail (trak != NULL);
if (avg_bitrate == 0 && max_bitrate == 0)
return;
stsd = &trak->mdia.minf.stbl.stsd;
for (iter = stsd->entries; iter; iter = g_list_next (iter)) {
SampleTableEntry *entry = iter->data;
switch (entry->kind) {
case AUDIO:{
SampleTableEntryMP4A *audioentry = (SampleTableEntryMP4A *) entry;
extensioniter = audioentry->extension_atoms;
break;
}
case VIDEO:{
SampleTableEntryMP4V *videoentry = (SampleTableEntryMP4V *) entry;
extensioniter = videoentry->extension_atoms;
break;
}
default:
break;
}
}
for (; extensioniter; extensioniter = g_list_next (extensioniter)) {
AtomInfo *atominfo = extensioniter->data;
if (atominfo->atom->type == FOURCC_esds) {
esds = (AtomESDS *) atominfo->atom;
} else if (atominfo->atom->type == FOURCC_btrt) {
btrt = (AtomData *) atominfo->atom;
} else if (atominfo->atom->type == FOURCC_wave) {
wave = (AtomWAVE *) atominfo->atom;
}
}
/* wave might have an esds internally */
if (wave) {
for (extensioniter = wave->extension_atoms; extensioniter;
extensioniter = g_list_next (extensioniter)) {
AtomInfo *atominfo = extensioniter->data;
if (atominfo->atom->type == FOURCC_esds) {
esds = (AtomESDS *) atominfo->atom;
break;
}
}
}
if (esds) {
if (avg_bitrate && esds->es.dec_conf_desc.avg_bitrate == 0)
esds->es.dec_conf_desc.avg_bitrate = avg_bitrate;
if (max_bitrate && esds->es.dec_conf_desc.max_bitrate == 0)
esds->es.dec_conf_desc.max_bitrate = max_bitrate;
}
if (btrt) {
/* type(4bytes) + size(4bytes) + buffersize(4bytes) +
* maxbitrate(bytes) + avgbitrate(bytes) */
if (max_bitrate && GST_READ_UINT32_BE (btrt->data + 4) == 0)
GST_WRITE_UINT32_BE (btrt->data + 4, max_bitrate);
if (avg_bitrate && GST_READ_UINT32_BE (btrt->data + 8) == 0)
GST_WRITE_UINT32_BE (btrt->data + 8, avg_bitrate);
}
}
void
atom_trak_tx3g_update_dimension (AtomTRAK * trak, guint32 width, guint32 height)
{
AtomSTSD *stsd;
GList *iter;
SampleTableEntryTX3G *tx3g = NULL;
stsd = &trak->mdia.minf.stbl.stsd;
for (iter = stsd->entries; iter && tx3g == NULL; iter = g_list_next (iter)) {
SampleTableEntry *entry = iter->data;
switch (entry->kind) {
case SUBTITLE:{
tx3g = (SampleTableEntryTX3G *) entry;
break;
}
default:
break;
}
}
/* Currently we never set the vertical placement flag, so we don't
* check for it to set the dimensions differently as the spec says.
* Always do it for the not set case */
if (tx3g) {
tx3g->font_size = 0.05 * height;
height = 0.15 * height;
trak->tkhd.width = width << 16;
trak->tkhd.height = height << 16;
tx3g->default_text_box = width | (height << 16);
}
}
/*
* Meta tags functions
*/
static void
atom_tag_data_alloc_data (AtomTagData * data, guint size)
{
g_free (data->data);
data->data = g_new0 (guint8, size);
data->datalen = size;
}
static void
atom_udta_append_tag (AtomUDTA * udta, AtomInfo * tag)
{
GList **entries;
if (udta->meta)
entries = &udta->meta->ilst->entries;
else
entries = &udta->entries;
*entries = g_list_append (*entries, tag);
}
void
atom_udta_add_tag (AtomUDTA * udta, guint32 fourcc, guint32 flags,
const guint8 * data, guint size)
{
AtomTag *tag;
AtomTagData *tdata;
tag = atom_tag_new (fourcc, flags);
tdata = &tag->data;
atom_tag_data_alloc_data (tdata, size);
memmove (tdata->data, data, size);
atom_udta_append_tag (udta,
build_atom_info_wrapper ((Atom *) tag, atom_tag_copy_data,
atom_tag_free));
}
void
atom_udta_add_str_tag (AtomUDTA * udta, guint32 fourcc, const gchar * value)
{
gint len = strlen (value);
if (len > 0)
atom_udta_add_tag (udta, fourcc, METADATA_TEXT_FLAG, (guint8 *) value, len);
}
void
atom_udta_add_uint_tag (AtomUDTA * udta, guint32 fourcc, guint32 flags,
guint32 value)
{
guint8 data[8] = { 0, };
if (flags) {
GST_WRITE_UINT16_BE (data, value);
atom_udta_add_tag (udta, fourcc, flags, data, 2);
} else {
GST_WRITE_UINT32_BE (data + 2, value);
atom_udta_add_tag (udta, fourcc, flags, data, 8);
}
}
#define GST_BUFFER_NEW_READONLY(mem, size) \
gst_buffer_new_wrapped_full (GST_MEMORY_FLAG_READONLY, mem, size, \
0, size, mem, NULL)
void
atom_udta_add_blob_tag (AtomUDTA * udta, guint8 * data, guint size)
{
AtomData *data_atom;
guint len;
guint32 fourcc;
if (size < 8)
return;
/* blob is unparsed atom;
* extract size and fourcc, and wrap remainder in data atom */
len = GST_READ_UINT32_BE (data);
fourcc = GST_READ_UINT32_LE (data + 4);
if (len > size)
return;
data_atom = atom_data_new_from_data (fourcc, data + 8, len - 8);
atom_udta_append_tag (udta,
build_atom_info_wrapper ((Atom *) data_atom, atom_data_copy_data,
atom_data_free));
}
void
atom_udta_add_3gp_tag (AtomUDTA * udta, guint32 fourcc, guint8 * data,
guint size)
{
AtomData *data_atom;
data_atom = atom_data_new (fourcc);
/* need full atom */
atom_data_alloc_mem (data_atom, size + 4);
/* full atom: version and flags */
GST_WRITE_UINT32_BE (data_atom->data, 0);
memcpy (data_atom->data + 4, data, size);
atom_udta_append_tag (udta,
build_atom_info_wrapper ((Atom *) data_atom, atom_data_copy_data,
atom_data_free));
}
guint16
language_code (const char *lang)
{
g_return_val_if_fail (lang != NULL, 0);
g_return_val_if_fail (strlen (lang) == 3, 0);
return (((lang[0] - 0x60) & 0x1F) << 10) + (((lang[1] - 0x60) & 0x1F) << 5) +
((lang[2] - 0x60) & 0x1F);
}
void
atom_udta_add_3gp_str_int_tag (AtomUDTA * udta, guint32 fourcc,
const gchar * value, gint16 ivalue)
{
gint len = 0, size = 0;
guint8 *data;
if (value) {
len = strlen (value);
size = len + 3;
}
if (ivalue >= 0)
size += 2;
data = g_malloc (size + 3);
/* language tag and null-terminated UTF-8 string */
if (value) {
GST_WRITE_UINT16_BE (data, language_code (GST_QT_MUX_DEFAULT_TAG_LANGUAGE));
/* include 0 terminator */
memcpy (data + 2, value, len + 1);
}
/* 16-bit unsigned int if standalone, otherwise 8-bit */
if (ivalue >= 0) {
if (size == 2)
GST_WRITE_UINT16_BE (data + size - 2, ivalue);
else {
GST_WRITE_UINT8 (data + size - 2, ivalue & 0xFF);
size--;
}
}
atom_udta_add_3gp_tag (udta, fourcc, data, size);
g_free (data);
}
void
atom_udta_add_3gp_str_tag (AtomUDTA * udta, guint32 fourcc, const gchar * value)
{
atom_udta_add_3gp_str_int_tag (udta, fourcc, value, -1);
}
void
atom_udta_add_3gp_uint_tag (AtomUDTA * udta, guint32 fourcc, guint16 value)
{
atom_udta_add_3gp_str_int_tag (udta, fourcc, NULL, value);
}
void
atom_udta_add_xmp_tags (AtomUDTA * udta, GstBuffer * xmpbuffer)
{
AtomData *data_atom = NULL;
if (udta->context->flavor == ATOMS_TREE_FLAVOR_MOV) {
if (xmpbuffer) {
data_atom = atom_data_new_from_gst_buffer (FOURCC_XMP_, xmpbuffer);
udta->entries = g_list_append (udta->entries,
build_atom_info_wrapper ((Atom *) data_atom, atom_data_copy_data,
atom_data_free));
}
} else {
GST_DEBUG ("Not adding xmp to moov atom, it is only used in 'mov' format");
}
}
/*
* Functions for specifying media types
*/
static void
atom_minf_set_audio (AtomMINF * minf)
{
atom_minf_clear_handlers (minf);
minf->smhd = atom_smhd_new ();
}
static void
atom_minf_set_video (AtomMINF * minf, AtomsContext * context)
{
atom_minf_clear_handlers (minf);
minf->vmhd = atom_vmhd_new (context);
}
static void
atom_minf_set_subtitle (AtomMINF * minf)
{
atom_minf_clear_handlers (minf);
}
static void
atom_hdlr_set_type (AtomHDLR * hdlr, AtomsContext * context, guint32 comp_type,
guint32 hdlr_type)
{
if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
hdlr->component_type = comp_type;
}
hdlr->handler_type = hdlr_type;
}
static void
atom_hdlr_set_name (AtomHDLR * hdlr, const char *name)
{
g_free (hdlr->name);
hdlr->name = g_strdup (name);
}
static void
atom_mdia_set_hdlr_type_audio (AtomMDIA * mdia, AtomsContext * context)
{
atom_hdlr_set_type (&mdia->hdlr, context, FOURCC_mhlr, FOURCC_soun);
/* Some players (low-end hardware) check for this name, which is what
* QuickTime itself sets */
atom_hdlr_set_name (&mdia->hdlr, "SoundHandler");
}
static void
atom_mdia_set_hdlr_type_video (AtomMDIA * mdia, AtomsContext * context)
{
atom_hdlr_set_type (&mdia->hdlr, context, FOURCC_mhlr, FOURCC_vide);
/* Some players (low-end hardware) check for this name, which is what
* QuickTime itself sets */
atom_hdlr_set_name (&mdia->hdlr, "VideoHandler");
}
static void
atom_mdia_set_hdlr_type_subtitle (AtomMDIA * mdia, AtomsContext * context)
{
atom_hdlr_set_type (&mdia->hdlr, context, FOURCC_mhlr, FOURCC_sbtl);
/* Just follows the pattern from video and audio above */
atom_hdlr_set_name (&mdia->hdlr, "SubtitleHandler");
}
static void
atom_mdia_set_audio (AtomMDIA * mdia, AtomsContext * context)
{
atom_mdia_set_hdlr_type_audio (mdia, context);
atom_minf_set_audio (&mdia->minf);
}
static void
atom_mdia_set_video (AtomMDIA * mdia, AtomsContext * context)
{
atom_mdia_set_hdlr_type_video (mdia, context);
atom_minf_set_video (&mdia->minf, context);
}
static void
atom_mdia_set_subtitle (AtomMDIA * mdia, AtomsContext * context)
{
atom_mdia_set_hdlr_type_subtitle (mdia, context);
atom_minf_set_subtitle (&mdia->minf);
}
static void
atom_tkhd_set_audio (AtomTKHD * tkhd)
{
tkhd->volume = 0x0100;
tkhd->width = tkhd->height = 0;
}
static void
atom_tkhd_set_video (AtomTKHD * tkhd, AtomsContext * context, guint32 width,
guint32 height)
{
tkhd->volume = 0;
/* qt and ISO base media do not contradict, and examples agree */
tkhd->width = width;
tkhd->height = height;
}
static void
atom_tkhd_set_subtitle (AtomTKHD * tkhd, AtomsContext * context, guint32 width,
guint32 height)
{
tkhd->volume = 0;
/* qt and ISO base media do not contradict, and examples agree */
tkhd->width = width;
tkhd->height = height;
}
static void
atom_edts_add_entry (AtomEDTS * edts, gint index, EditListEntry * entry)
{
EditListEntry *e =
(EditListEntry *) g_slist_nth_data (edts->elst.entries, index);
/* Create a new entry if missing (appends to the list if index is larger) */
if (e == NULL) {
e = g_new (EditListEntry, 1);
edts->elst.entries = g_slist_insert (edts->elst.entries, e, index);
}
/* Update the entry */
*e = *entry;
}
void
atom_trak_edts_clear (AtomTRAK * trak)
{
if (trak->edts) {
atom_edts_free (trak->edts);
trak->edts = NULL;
}
}
/*
* Update an entry in this trak edits list, creating it if needed.
* index is the index of the entry to update, or create if it's past the end.
* duration is in the moov's timescale
* media_time is the offset in the media time to start from (media's timescale)
* rate is a 32 bits fixed-point
*/
void
atom_trak_set_elst_entry (AtomTRAK * trak, gint index,
guint32 duration, guint32 media_time, guint32 rate)
{
EditListEntry entry;
entry.duration = duration;
entry.media_time = media_time;
entry.media_rate = rate;
if (trak->edts == NULL)
trak->edts = atom_edts_new ();
atom_edts_add_entry (trak->edts, index, &entry);
}
/* re-negotiation is prevented at top-level, so only 1 entry expected.
* Quite some more care here and elsewhere may be needed to
* support several entries */
static SampleTableEntryMP4A *
atom_trak_add_audio_entry (AtomTRAK * trak, AtomsContext * context,
guint32 type)
{
AtomSTSD *stsd = &trak->mdia.minf.stbl.stsd;
SampleTableEntryMP4A *mp4a = sample_entry_mp4a_new ();
mp4a->se.header.type = type;
mp4a->se.kind = AUDIO;
mp4a->compression_id = -1;
mp4a->se.data_reference_index = 1;
stsd->entries = g_list_prepend (stsd->entries, mp4a);
stsd->n_entries++;
return mp4a;
}
/* return number of centiframes per second */
guint
atom_framerate_to_timescale (gint n, gint d)
{
if (n == 0)
return 10000;
if (d != 1 && d != 1001) {
/* otherwise there are probably rounding errors and we should rather guess
* if it's close enough to a well known framerate */
gst_video_guess_framerate (gst_util_uint64_scale (d, GST_SECOND, n), &n,
&d);
}
return gst_util_uint64_scale (n, 100, d);
}
static SampleTableEntryTMCD *
atom_trak_add_timecode_entry (AtomTRAK * trak, AtomsContext * context,
guint32 trak_timescale, GstVideoTimeCode * tc)
{
AtomSTSD *stsd = &trak->mdia.minf.stbl.stsd;
SampleTableEntryTMCD *tmcd = sample_entry_tmcd_new ();
g_assert (trak_timescale != 0);
trak->mdia.hdlr.component_type = FOURCC_mhlr;
trak->mdia.hdlr.handler_type = FOURCC_tmcd;
g_free (trak->mdia.hdlr.name);
trak->mdia.hdlr.name = g_strdup ("Time Code Media Handler");
trak->mdia.mdhd.time_info.timescale = trak_timescale;
tmcd->se.kind = TIMECODE;
tmcd->se.data_reference_index = 1;
tmcd->tc_flags = TC_24H_MAX;
if (tc->config.flags &= GST_VIDEO_TIME_CODE_FLAGS_DROP_FRAME)
tmcd->tc_flags |= TC_DROP_FRAME;
tmcd->name.language_code = 0;
tmcd->name.name = g_strdup ("Tape");
tmcd->timescale = trak_timescale;
tmcd->frame_duration =
gst_util_uint64_scale (tmcd->timescale, tc->config.fps_d,
tc->config.fps_n);
if (tc->config.fps_d == 1001)
tmcd->n_frames = tc->config.fps_n / 1000;
else
tmcd->n_frames = tc->config.fps_n / tc->config.fps_d;
stsd->entries = g_list_prepend (stsd->entries, tmcd);
stsd->n_entries++;
return tmcd;
}
static SampleTableEntryMP4V *
atom_trak_add_video_entry (AtomTRAK * trak, AtomsContext * context,
guint32 type)
{
SampleTableEntryMP4V *mp4v = sample_entry_mp4v_new (context);
AtomSTSD *stsd = &trak->mdia.minf.stbl.stsd;
mp4v->se.header.type = type;
mp4v->se.kind = VIDEO;
mp4v->se.data_reference_index = 1;
mp4v->horizontal_resolution = 72 << 16;
mp4v->vertical_resolution = 72 << 16;
if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
mp4v->spatial_quality = 512;
mp4v->temporal_quality = 512;
}
stsd->entries = g_list_prepend (stsd->entries, mp4v);
stsd->n_entries++;
return mp4v;
}
static SampleTableEntryTX3G *
atom_trak_add_subtitle_entry (AtomTRAK * trak, AtomsContext * context,
guint32 type)
{
SampleTableEntryTX3G *tx3g = sample_entry_tx3g_new ();
AtomSTSD *stsd = &trak->mdia.minf.stbl.stsd;
tx3g->se.header.type = type;
tx3g->se.kind = SUBTITLE;
tx3g->se.data_reference_index = 1;
stsd->entries = g_list_prepend (stsd->entries, tx3g);
stsd->n_entries++;
return tx3g;
}
void
atom_trak_set_constant_size_samples (AtomTRAK * trak, guint32 sample_size)
{
trak->mdia.minf.stbl.stsz.sample_size = sample_size;
}
static void
atom_trak_set_audio (AtomTRAK * trak, AtomsContext * context)
{
atom_tkhd_set_audio (&trak->tkhd);
atom_mdia_set_audio (&trak->mdia, context);
}
static void
atom_trak_set_video (AtomTRAK * trak, AtomsContext * context, guint32 width,
guint32 height)
{
atom_tkhd_set_video (&trak->tkhd, context, width, height);
atom_mdia_set_video (&trak->mdia, context);
}
static void
atom_trak_set_subtitle (AtomTRAK * trak, AtomsContext * context)
{
atom_tkhd_set_subtitle (&trak->tkhd, context, 0, 0);
atom_mdia_set_subtitle (&trak->mdia, context);
}
static void
atom_trak_set_audio_commons (AtomTRAK * trak, AtomsContext * context,
guint32 rate)
{
atom_trak_set_audio (trak, context);
trak->mdia.mdhd.time_info.timescale = rate;
}
static void
atom_trak_set_video_commons (AtomTRAK * trak, AtomsContext * context,
guint32 rate, guint32 width, guint32 height)
{
atom_trak_set_video (trak, context, width, height);
trak->mdia.mdhd.time_info.timescale = rate;
trak->tkhd.width = width << 16;
trak->tkhd.height = height << 16;
}
static void
atom_trak_set_subtitle_commons (AtomTRAK * trak, AtomsContext * context)
{
atom_trak_set_subtitle (trak, context);
trak->mdia.mdhd.time_info.timescale = 1000;
trak->tkhd.alternate_group = 2; /* same for all subtitles */
trak->tkhd.layer = -1; /* above video (layer 0) */
}
void
sample_table_entry_add_ext_atom (SampleTableEntry * ste, AtomInfo * ext)
{
GList **list = NULL;
if (ste->kind == AUDIO) {
list = &(((SampleTableEntryMP4A *) ste)->extension_atoms);
} else if (ste->kind == VIDEO) {
list = &(((SampleTableEntryMP4V *) ste)->extension_atoms);
} else {
g_assert_not_reached ();
return;
}
*list = g_list_prepend (*list, ext);
}
SampleTableEntryMP4A *
atom_trak_set_audio_type (AtomTRAK * trak, AtomsContext * context,
AudioSampleEntry * entry, guint32 scale, AtomInfo * ext, gint sample_size)
{
SampleTableEntryMP4A *ste;
atom_trak_set_audio_commons (trak, context, scale);
atom_stsd_remove_entries (&trak->mdia.minf.stbl.stsd);
ste = atom_trak_add_audio_entry (trak, context, entry->fourcc);
trak->is_video = FALSE;
trak->is_h264 = FALSE;
ste->version = entry->version;
ste->compression_id = entry->compression_id;
ste->sample_size = entry->sample_size;
ste->sample_rate = entry->sample_rate << 16;
ste->channels = entry->channels;
ste->samples_per_packet = entry->samples_per_packet;
ste->bytes_per_sample = entry->bytes_per_sample;
ste->bytes_per_packet = entry->bytes_per_packet;
ste->bytes_per_frame = entry->bytes_per_frame;
if (ext)
ste->extension_atoms = g_list_prepend (ste->extension_atoms, ext);
/* 0 size means variable size */
atom_trak_set_constant_size_samples (trak, sample_size);
return ste;
}
SampleTableEntryTMCD *
atom_trak_set_timecode_type (AtomTRAK * trak, AtomsContext * context,
guint32 trak_timescale, GstVideoTimeCode * tc)
{
SampleTableEntryTMCD *ste;
AtomGMHD *gmhd = trak->mdia.minf.gmhd;
if (context->flavor != ATOMS_TREE_FLAVOR_MOV) {
return NULL;
}
ste = atom_trak_add_timecode_entry (trak, context, trak_timescale, tc);
gmhd = atom_gmhd_new ();
gmhd->gmin.graphics_mode = 0x0040;
gmhd->gmin.opcolor[0] = 0x8000;
gmhd->gmin.opcolor[1] = 0x8000;
gmhd->gmin.opcolor[2] = 0x8000;
gmhd->tmcd = atom_tmcd_new ();
gmhd->tmcd->tcmi.text_size = 12;
gmhd->tmcd->tcmi.font_name = g_strdup ("Chicago"); /* Pascal string */
trak->mdia.minf.gmhd = gmhd;
trak->is_video = FALSE;
trak->is_h264 = FALSE;
return ste;
}
SampleTableEntry *
atom_trak_set_caption_type (AtomTRAK * trak, AtomsContext * context,
guint32 trak_timescale, guint32 caption_type)
{
SampleTableEntry *ste;
AtomGMHD *gmhd = trak->mdia.minf.gmhd;
AtomSTSD *stsd = &trak->mdia.minf.stbl.stsd;
if (context->flavor != ATOMS_TREE_FLAVOR_MOV) {
return NULL;
}
trak->mdia.mdhd.time_info.timescale = trak_timescale;
trak->mdia.hdlr.component_type = FOURCC_mhlr;
trak->mdia.hdlr.handler_type = FOURCC_clcp;
g_free (trak->mdia.hdlr.name);
trak->mdia.hdlr.name = g_strdup ("Closed Caption Media Handler");
ste = g_new0 (SampleTableEntry, 1);
atom_sample_entry_init (ste, caption_type);
ste->kind = CLOSEDCAPTION;
ste->data_reference_index = 1;
stsd->entries = g_list_prepend (stsd->entries, ste);
stsd->n_entries++;
gmhd = atom_gmhd_new ();
gmhd->gmin.graphics_mode = 0x0040;
gmhd->gmin.opcolor[0] = 0x8000;
gmhd->gmin.opcolor[1] = 0x8000;
gmhd->gmin.opcolor[2] = 0x8000;
trak->mdia.minf.gmhd = gmhd;
trak->is_video = FALSE;
trak->is_h264 = FALSE;
return ste;
}
static AtomInfo *
build_pasp_extension (gint par_width, gint par_height)
{
AtomData *atom_data = atom_data_new (FOURCC_pasp);
guint8 *data;
atom_data_alloc_mem (atom_data, 8);
data = atom_data->data;
/* ihdr = image header box */
GST_WRITE_UINT32_BE (data, par_width);
GST_WRITE_UINT32_BE (data + 4, par_height);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_fiel_extension (GstVideoInterlaceMode mode, GstVideoFieldOrder order)
{
AtomData *atom_data = atom_data_new (FOURCC_fiel);
guint8 *data;
gint field_order;
gint interlace;
atom_data_alloc_mem (atom_data, 2);
data = atom_data->data;
if (mode == GST_VIDEO_INTERLACE_MODE_PROGRESSIVE) {
interlace = 1;
field_order = 0;
} else if (mode == GST_VIDEO_INTERLACE_MODE_INTERLEAVED) {
interlace = 2;
field_order = order == GST_VIDEO_FIELD_ORDER_TOP_FIELD_FIRST ? 9 : 14;
} else {
interlace = 0;
field_order = 0;
}
GST_WRITE_UINT8 (data, interlace);
GST_WRITE_UINT8 (data + 1, field_order);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_colr_extension (const GstVideoColorimetry * colorimetry, gboolean is_mp4)
{
AtomData *atom_data = atom_data_new (FOURCC_colr);
guint8 *data;
guint16 primaries;
guint16 transfer_function;
guint16 matrix;
switch (colorimetry->primaries) {
case GST_VIDEO_COLOR_PRIMARIES_BT709:
primaries = 1;
break;
case GST_VIDEO_COLOR_PRIMARIES_BT470BG:
primaries = 5;
break;
case GST_VIDEO_COLOR_PRIMARIES_SMPTE170M:
case GST_VIDEO_COLOR_PRIMARIES_SMPTE240M:
primaries = 6;
break;
case GST_VIDEO_COLOR_PRIMARIES_BT2020:
primaries = 9;
break;
case GST_VIDEO_COLOR_PRIMARIES_UNKNOWN:
default:
primaries = 2;
break;
}
switch (colorimetry->transfer) {
case GST_VIDEO_TRANSFER_BT709:
transfer_function = 1;
break;
case GST_VIDEO_TRANSFER_SMPTE240M:
transfer_function = 7;
break;
case GST_VIDEO_TRANSFER_UNKNOWN:
default:
transfer_function = 2;
break;
}
switch (colorimetry->matrix) {
case GST_VIDEO_COLOR_MATRIX_BT709:
matrix = 1;
break;
case GST_VIDEO_COLOR_MATRIX_BT601:
matrix = 6;
break;
case GST_VIDEO_COLOR_MATRIX_SMPTE240M:
matrix = 7;
break;
case GST_VIDEO_COLOR_MATRIX_BT2020:
matrix = 9;
break;
case GST_VIDEO_COLOR_MATRIX_UNKNOWN:
default:
matrix = 2;
break;
}
atom_data_alloc_mem (atom_data, 10 + (is_mp4 ? 1 : 0));
data = atom_data->data;
/* colour specification box */
if (is_mp4)
GST_WRITE_UINT32_LE (data, FOURCC_nclx);
else
GST_WRITE_UINT32_LE (data, FOURCC_nclc);
GST_WRITE_UINT16_BE (data + 4, primaries);
GST_WRITE_UINT16_BE (data + 6, transfer_function);
GST_WRITE_UINT16_BE (data + 8, matrix);
if (is_mp4) {
GST_WRITE_UINT8 (data + 10,
colorimetry->range == GST_VIDEO_COLOR_RANGE_0_255 ? 0x80 : 0x00);
}
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_clap_extension (gint width_n, gint width_d, gint height_n, gint height_d,
gint h_off_n, gint h_off_d, gint v_off_n, gint v_off_d)
{
AtomData *atom_data = atom_data_new (FOURCC_clap);
guint8 *data;
atom_data_alloc_mem (atom_data, 32);
data = atom_data->data;
GST_WRITE_UINT32_BE (data, width_n);
GST_WRITE_UINT32_BE (data + 4, width_d);
GST_WRITE_UINT32_BE (data + 8, height_n);
GST_WRITE_UINT32_BE (data + 12, height_d);
GST_WRITE_UINT32_BE (data + 16, h_off_n);
GST_WRITE_UINT32_BE (data + 20, h_off_d);
GST_WRITE_UINT32_BE (data + 24, v_off_n);
GST_WRITE_UINT32_BE (data + 28, v_off_d);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_tapt_extension (gint clef_width, gint clef_height, gint prof_width,
gint prof_height, gint enof_width, gint enof_height)
{
AtomData *atom_data = atom_data_new (FOURCC_tapt);
guint8 *data;
atom_data_alloc_mem (atom_data, 60);
data = atom_data->data;
GST_WRITE_UINT32_BE (data, 20);
GST_WRITE_UINT32_LE (data + 4, FOURCC_clef);
GST_WRITE_UINT32_BE (data + 8, 0);
GST_WRITE_UINT32_BE (data + 12, clef_width);
GST_WRITE_UINT32_BE (data + 16, clef_height);
GST_WRITE_UINT32_BE (data + 20, 20);
GST_WRITE_UINT32_LE (data + 24, FOURCC_prof);
GST_WRITE_UINT32_BE (data + 28, 0);
GST_WRITE_UINT32_BE (data + 32, prof_width);
GST_WRITE_UINT32_BE (data + 36, prof_height);
GST_WRITE_UINT32_BE (data + 40, 20);
GST_WRITE_UINT32_LE (data + 44, FOURCC_enof);
GST_WRITE_UINT32_BE (data + 48, 0);
GST_WRITE_UINT32_BE (data + 52, enof_width);
GST_WRITE_UINT32_BE (data + 56, enof_height);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
static AtomInfo *
build_mov_video_sample_description_padding_extension (void)
{
AtomData *atom_data = atom_data_new (FOURCC_clap);
return build_atom_info_wrapper ((Atom *) atom_data, atom_copy_empty,
atom_data_free);
}
SampleTableEntryMP4V *
atom_trak_set_video_type (AtomTRAK * trak, AtomsContext * context,
VisualSampleEntry * entry, guint32 scale, GList * ext_atoms_list)
{
SampleTableEntryMP4V *ste;
guint dwidth, dheight;
gint par_n = 0, par_d = 0;
par_n = entry->par_n;
par_d = entry->par_d;
dwidth = entry->width;
dheight = entry->height;
/* ISO file spec says track header w/h indicates track's visual presentation
* (so this together with pixels w/h implicitly defines PAR) */
if (par_n && (context->flavor != ATOMS_TREE_FLAVOR_MOV)) {
dwidth = entry->width * par_n / par_d;
dheight = entry->height;
}
atom_trak_set_video_commons (trak, context, scale, dwidth, dheight);
atom_stsd_remove_entries (&trak->mdia.minf.stbl.stsd);
ste = atom_trak_add_video_entry (trak, context, entry->fourcc);
trak->is_video = TRUE;
trak->is_h264 = (entry->fourcc == FOURCC_avc1
|| entry->fourcc == FOURCC_avc3);
ste->version = entry->version;
ste->width = entry->width;
ste->height = entry->height;
ste->depth = entry->depth;
ste->color_table_id = entry->color_table_id;
ste->frame_count = entry->frame_count;
if (ext_atoms_list)
ste->extension_atoms = g_list_concat (ste->extension_atoms, ext_atoms_list);
ste->extension_atoms = g_list_append (ste->extension_atoms,
build_pasp_extension (par_n, par_d));
if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
/* append 0 as a terminator "length" to work around some broken software */
ste->extension_atoms =
g_list_append (ste->extension_atoms,
build_mov_video_sample_description_padding_extension ());
}
return ste;
}
void
subtitle_sample_entry_init (SubtitleSampleEntry * entry)
{
entry->font_size = 0;
entry->font_face = 0;
entry->foreground_color_rgba = 0xFFFFFFFF; /* all white, opaque */
}
SampleTableEntryTX3G *
atom_trak_set_subtitle_type (AtomTRAK * trak, AtomsContext * context,
SubtitleSampleEntry * entry)
{
SampleTableEntryTX3G *tx3g;
atom_trak_set_subtitle_commons (trak, context);
atom_stsd_remove_entries (&trak->mdia.minf.stbl.stsd);
tx3g = atom_trak_add_subtitle_entry (trak, context, entry->fourcc);
tx3g->font_face = entry->font_face;
tx3g->font_size = entry->font_size;
tx3g->foreground_color_rgba = entry->foreground_color_rgba;
trak->is_video = FALSE;
trak->is_h264 = FALSE;
return tx3g;
}
static void
atom_mfhd_init (AtomMFHD * mfhd, guint32 sequence_number)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&(mfhd->header), FOURCC_mfhd, 0, 0, 0, flags);
mfhd->sequence_number = sequence_number;
}
static void
atom_moof_init (AtomMOOF * moof, AtomsContext * context,
guint32 sequence_number)
{
atom_header_set (&moof->header, FOURCC_moof, 0, 0);
atom_mfhd_init (&moof->mfhd, sequence_number);
moof->trafs = NULL;
}
AtomMOOF *
atom_moof_new (AtomsContext * context, guint32 sequence_number)
{
AtomMOOF *moof = g_new0 (AtomMOOF, 1);
atom_moof_init (moof, context, sequence_number);
return moof;
}
static void
atom_trun_free (AtomTRUN * trun)
{
atom_full_clear (&trun->header);
atom_array_clear (&trun->entries);
g_free (trun);
}
static void
atom_sdtp_free (AtomSDTP * sdtp)
{
atom_full_clear (&sdtp->header);
atom_array_clear (&sdtp->entries);
g_free (sdtp);
}
void
atom_traf_free (AtomTRAF * traf)
{
GList *walker;
walker = traf->truns;
while (walker) {
atom_trun_free ((AtomTRUN *) walker->data);
walker = g_list_next (walker);
}
g_list_free (traf->truns);
traf->truns = NULL;
walker = traf->sdtps;
while (walker) {
atom_sdtp_free ((AtomSDTP *) walker->data);
walker = g_list_next (walker);
}
g_list_free (traf->sdtps);
traf->sdtps = NULL;
g_free (traf);
}
void
atom_moof_free (AtomMOOF * moof)
{
GList *walker;
walker = moof->trafs;
while (walker) {
atom_traf_free ((AtomTRAF *) walker->data);
walker = g_list_next (walker);
}
g_list_free (moof->trafs);
moof->trafs = NULL;
g_free (moof);
}
static guint64
atom_mfhd_copy_data (AtomMFHD * mfhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&mfhd->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (mfhd->sequence_number, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_tfhd_copy_data (AtomTFHD * tfhd, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint32 flags;
if (!atom_full_copy_data (&tfhd->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (tfhd->track_ID, buffer, size, offset);
flags = atom_full_get_flags_as_uint (&tfhd->header);
if (flags & TF_BASE_DATA_OFFSET)
prop_copy_uint64 (tfhd->base_data_offset, buffer, size, offset);
if (flags & TF_SAMPLE_DESCRIPTION_INDEX)
prop_copy_uint32 (tfhd->sample_description_index, buffer, size, offset);
if (flags & TF_DEFAULT_SAMPLE_DURATION)
prop_copy_uint32 (tfhd->default_sample_duration, buffer, size, offset);
if (flags & TF_DEFAULT_SAMPLE_SIZE)
prop_copy_uint32 (tfhd->default_sample_size, buffer, size, offset);
if (flags & TF_DEFAULT_SAMPLE_FLAGS)
prop_copy_uint32 (tfhd->default_sample_flags, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_tfdt_copy_data (AtomTFDT * tfdt, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&tfdt->header, buffer, size, offset)) {
return 0;
}
/* 32-bit time if version == 0 else 64-bit: */
if (tfdt->header.version == 0)
prop_copy_uint32 (tfdt->base_media_decode_time, buffer, size, offset);
else
prop_copy_uint64 (tfdt->base_media_decode_time, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_trun_copy_data (AtomTRUN * trun, guint8 ** buffer, guint64 * size,
guint64 * offset, guint32 * data_offset)
{
guint64 original_offset = *offset;
guint32 flags, i;
flags = atom_full_get_flags_as_uint (&trun->header);
/* if first trun in moof, forcibly add data_offset and record
* where it must be written later on */
if (data_offset && !*data_offset) {
flags |= TR_DATA_OFFSET;
} else {
flags &= ~TR_DATA_OFFSET;
}
atom_full_set_flags_as_uint (&trun->header, flags);
if (!atom_full_copy_data (&trun->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (trun->sample_count, buffer, size, offset);
if (flags & TR_DATA_OFFSET) {
*data_offset = *offset;
prop_copy_int32 (trun->data_offset, buffer, size, offset);
}
if (flags & TR_FIRST_SAMPLE_FLAGS)
prop_copy_uint32 (trun->first_sample_flags, buffer, size, offset);
for (i = 0; i < atom_array_get_len (&trun->entries); i++) {
TRUNSampleEntry *entry = &atom_array_index (&trun->entries, i);
if (flags & TR_SAMPLE_DURATION)
prop_copy_uint32 (entry->sample_duration, buffer, size, offset);
if (flags & TR_SAMPLE_SIZE)
prop_copy_uint32 (entry->sample_size, buffer, size, offset);
if (flags & TR_SAMPLE_FLAGS)
prop_copy_uint32 (entry->sample_flags, buffer, size, offset);
if (flags & TR_COMPOSITION_TIME_OFFSETS)
prop_copy_uint32 (entry->sample_composition_time_offset,
buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_sdtp_copy_data (AtomSDTP * sdtp, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
if (!atom_full_copy_data (&sdtp->header, buffer, size, offset)) {
return 0;
}
/* all entries at once */
prop_copy_fixed_size_string (&atom_array_index (&sdtp->entries, 0),
atom_array_get_len (&sdtp->entries), buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_traf_copy_data (AtomTRAF * traf, guint8 ** buffer, guint64 * size,
guint64 * offset, guint32 * data_offset)
{
guint64 original_offset = *offset;
GList *walker;
if (!atom_copy_data (&traf->header, buffer, size, offset)) {
return 0;
}
if (!atom_tfhd_copy_data (&traf->tfhd, buffer, size, offset)) {
return 0;
}
if (!atom_tfdt_copy_data (&traf->tfdt, buffer, size, offset)) {
return 0;
}
walker = g_list_first (traf->truns);
while (walker != NULL) {
if (!atom_trun_copy_data ((AtomTRUN *) walker->data, buffer, size, offset,
data_offset)) {
return 0;
}
walker = g_list_next (walker);
}
walker = g_list_first (traf->sdtps);
while (walker != NULL) {
if (!atom_sdtp_copy_data ((AtomSDTP *) walker->data, buffer, size, offset)) {
return 0;
}
walker = g_list_next (walker);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
/* creates moof atom; metadata is written expecting actual buffer data
* is in mdata directly after moof, and is consecutively written per trak */
guint64
atom_moof_copy_data (AtomMOOF * moof, guint8 ** buffer,
guint64 * size, guint64 * offset)
{
guint64 original_offset = *offset;
GList *walker;
guint32 data_offset = 0;
if (!atom_copy_data (&moof->header, buffer, size, offset))
return 0;
if (!atom_mfhd_copy_data (&moof->mfhd, buffer, size, offset))
return 0;
walker = g_list_first (moof->trafs);
while (walker != NULL) {
if (!atom_traf_copy_data ((AtomTRAF *) walker->data, buffer, size, offset,
&data_offset)) {
return 0;
}
walker = g_list_next (walker);
}
atom_write_size (buffer, size, offset, original_offset);
if (*buffer && data_offset) {
/* first trun needs a data-offset relative to moof start
* = moof size + mdat prefix */
GST_WRITE_UINT32_BE (*buffer + data_offset, *offset - original_offset + 8);
}
return *offset - original_offset;
}
static void
atom_tfhd_init (AtomTFHD * tfhd, guint32 track_ID)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&tfhd->header, FOURCC_tfhd, 0, 0, 0, flags);
tfhd->track_ID = track_ID;
tfhd->base_data_offset = 0;
tfhd->sample_description_index = 1;
tfhd->default_sample_duration = 0;
tfhd->default_sample_size = 0;
tfhd->default_sample_flags = 0;
}
static void
atom_tfdt_init (AtomTFDT * tfdt)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&tfdt->header, FOURCC_tfdt, 0, 0, 0, flags);
tfdt->base_media_decode_time = 0;
}
static void
atom_trun_init (AtomTRUN * trun)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&trun->header, FOURCC_trun, 0, 0, 0, flags);
trun->sample_count = 0;
trun->data_offset = 0;
trun->first_sample_flags = 0;
atom_array_init (&trun->entries, 512);
}
static AtomTRUN *
atom_trun_new (void)
{
AtomTRUN *trun = g_new0 (AtomTRUN, 1);
atom_trun_init (trun);
return trun;
}
static void
atom_sdtp_init (AtomSDTP * sdtp)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&sdtp->header, FOURCC_sdtp, 0, 0, 0, flags);
atom_array_init (&sdtp->entries, 512);
}
static AtomSDTP *
atom_sdtp_new (AtomsContext * context)
{
AtomSDTP *sdtp = g_new0 (AtomSDTP, 1);
atom_sdtp_init (sdtp);
return sdtp;
}
static void
atom_traf_add_sdtp (AtomTRAF * traf, AtomSDTP * sdtp)
{
traf->sdtps = g_list_append (traf->sdtps, sdtp);
}
static void
atom_sdtp_add_samples (AtomSDTP * sdtp, guint8 val)
{
/* it does not make much/any sense according to specs,
* but that's how MS isml samples seem to do it */
atom_array_append (&sdtp->entries, val, 256);
}
static void
atom_trun_add_samples (AtomTRUN * trun, guint32 delta, guint32 size,
guint32 flags, gint64 pts_offset)
{
TRUNSampleEntry nentry;
if (pts_offset != 0)
trun->header.flags[1] |= (TR_COMPOSITION_TIME_OFFSETS >> 8);
nentry.sample_duration = delta;
nentry.sample_size = size;
nentry.sample_flags = flags;
nentry.sample_composition_time_offset = pts_offset;
atom_array_append (&trun->entries, nentry, 256);
trun->sample_count++;
}
static void
atom_traf_init (AtomTRAF * traf, AtomsContext * context, guint32 track_ID)
{
atom_header_set (&traf->header, FOURCC_traf, 0, 0);
atom_tfdt_init (&traf->tfdt);
atom_tfhd_init (&traf->tfhd, track_ID);
traf->truns = NULL;
if (context->flavor == ATOMS_TREE_FLAVOR_ISML)
atom_traf_add_sdtp (traf, atom_sdtp_new (context));
}
AtomTRAF *
atom_traf_new (AtomsContext * context, guint32 track_ID)
{
AtomTRAF *traf = g_new0 (AtomTRAF, 1);
atom_traf_init (traf, context, track_ID);
return traf;
}
void
atom_traf_set_base_decode_time (AtomTRAF * traf, guint64 base_decode_time)
{
traf->tfdt.base_media_decode_time = base_decode_time;
/* If we need to write a 64-bit tfdt, set the atom version */
if (base_decode_time > G_MAXUINT32)
traf->tfdt.header.version = 1;
}
static void
atom_traf_add_trun (AtomTRAF * traf, AtomTRUN * trun)
{
traf->truns = g_list_append (traf->truns, trun);
}
void
atom_traf_add_samples (AtomTRAF * traf, guint32 delta, guint32 size,
gboolean sync, gint64 pts_offset, gboolean sdtp_sync)
{
AtomTRUN *trun;
guint32 flags;
/* 0x10000 is sample-is-difference-sample flag
* low byte stuff is what ismv uses */
flags = (sync ? 0x0 : 0x10000) | (sdtp_sync ? 0x40 : 0xc0);
if (G_UNLIKELY (!traf->truns)) {
trun = atom_trun_new ();
atom_traf_add_trun (traf, trun);
/* optimistic; indicate all defaults present in tfhd */
traf->tfhd.header.flags[2] = TF_DEFAULT_SAMPLE_DURATION |
TF_DEFAULT_SAMPLE_SIZE | TF_DEFAULT_SAMPLE_FLAGS;
traf->tfhd.default_sample_duration = delta;
traf->tfhd.default_sample_size = size;
traf->tfhd.default_sample_flags = flags;
trun->first_sample_flags = flags;
}
trun = traf->truns->data;
/* check if still matching defaults,
* if not, abandon default and need entry for each sample */
if (traf->tfhd.default_sample_duration != delta) {
traf->tfhd.header.flags[2] &= ~TF_DEFAULT_SAMPLE_DURATION;
trun->header.flags[1] |= (TR_SAMPLE_DURATION >> 8);
}
if (traf->tfhd.default_sample_size != size) {
traf->tfhd.header.flags[2] &= ~TF_DEFAULT_SAMPLE_SIZE;
trun->header.flags[1] |= (TR_SAMPLE_SIZE >> 8);
}
if (traf->tfhd.default_sample_flags != flags) {
if (trun->sample_count == 1) {
/* at least will need first sample flag */
traf->tfhd.default_sample_flags = flags;
trun->header.flags[2] |= TR_FIRST_SAMPLE_FLAGS;
} else {
/* now we need sample flags for each sample */
traf->tfhd.header.flags[2] &= ~TF_DEFAULT_SAMPLE_FLAGS;
trun->header.flags[1] |= (TR_SAMPLE_FLAGS >> 8);
trun->header.flags[2] &= ~TR_FIRST_SAMPLE_FLAGS;
}
}
atom_trun_add_samples (traf->truns->data, delta, size, flags, pts_offset);
if (traf->sdtps)
atom_sdtp_add_samples (traf->sdtps->data, 0x10 | ((flags & 0xff) >> 4));
}
guint32
atom_traf_get_sample_num (AtomTRAF * traf)
{
AtomTRUN *trun;
if (G_UNLIKELY (!traf->truns))
return 0;
trun = traf->truns->data;
return atom_array_get_len (&trun->entries);
}
void
atom_moof_add_traf (AtomMOOF * moof, AtomTRAF * traf)
{
moof->trafs = g_list_append (moof->trafs, traf);
}
static void
atom_tfra_free (AtomTFRA * tfra)
{
atom_full_clear (&tfra->header);
atom_array_clear (&tfra->entries);
g_free (tfra);
}
AtomMFRA *
atom_mfra_new (AtomsContext * context)
{
AtomMFRA *mfra = g_new0 (AtomMFRA, 1);
atom_header_set (&mfra->header, FOURCC_mfra, 0, 0);
return mfra;
}
void
atom_mfra_add_tfra (AtomMFRA * mfra, AtomTFRA * tfra)
{
mfra->tfras = g_list_append (mfra->tfras, tfra);
}
void
atom_mfra_free (AtomMFRA * mfra)
{
GList *walker;
walker = mfra->tfras;
while (walker) {
atom_tfra_free ((AtomTFRA *) walker->data);
walker = g_list_next (walker);
}
g_list_free (mfra->tfras);
mfra->tfras = NULL;
atom_clear (&mfra->header);
g_free (mfra);
}
static void
atom_tfra_init (AtomTFRA * tfra, guint32 track_ID)
{
guint8 flags[3] = { 0, 0, 0 };
atom_full_init (&tfra->header, FOURCC_tfra, 0, 0, 0, flags);
tfra->track_ID = track_ID;
atom_array_init (&tfra->entries, 512);
}
AtomTFRA *
atom_tfra_new (AtomsContext * context, guint32 track_ID)
{
AtomTFRA *tfra = g_new0 (AtomTFRA, 1);
atom_tfra_init (tfra, track_ID);
return tfra;
}
static inline gint
need_bytes (guint32 num)
{
gint n = 0;
while (num >>= 8)
n++;
return n;
}
void
atom_tfra_add_entry (AtomTFRA * tfra, guint64 dts, guint32 sample_num)
{
TFRAEntry entry;
entry.time = dts;
/* fill in later */
entry.moof_offset = 0;
/* always write a single trun in a single traf */
entry.traf_number = 1;
entry.trun_number = 1;
entry.sample_number = sample_num;
/* auto-use 64 bits if needed */
if (dts > G_MAXUINT32)
tfra->header.version = 1;
/* 1 byte will always do for traf and trun number,
* check how much sample_num needs */
tfra->lengths = (tfra->lengths & 0xfc) ||
MAX (tfra->lengths, need_bytes (sample_num));
atom_array_append (&tfra->entries, entry, 256);
}
void
atom_tfra_update_offset (AtomTFRA * tfra, guint64 offset)
{
gint i;
/* auto-use 64 bits if needed */
if (offset > G_MAXUINT32)
tfra->header.version = 1;
for (i = atom_array_get_len (&tfra->entries) - 1; i >= 0; i--) {
TFRAEntry *entry = &atom_array_index (&tfra->entries, i);
if (entry->moof_offset)
break;
entry->moof_offset = offset;
}
}
static guint64
atom_tfra_copy_data (AtomTFRA * tfra, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint32 i;
TFRAEntry *entry;
guint32 data;
guint bytes;
guint version;
if (!atom_full_copy_data (&tfra->header, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (tfra->track_ID, buffer, size, offset);
prop_copy_uint32 (tfra->lengths, buffer, size, offset);
prop_copy_uint32 (atom_array_get_len (&tfra->entries), buffer, size, offset);
version = tfra->header.version;
for (i = 0; i < atom_array_get_len (&tfra->entries); ++i) {
entry = &atom_array_index (&tfra->entries, i);
if (version) {
prop_copy_uint64 (entry->time, buffer, size, offset);
prop_copy_uint64 (entry->moof_offset, buffer, size, offset);
} else {
prop_copy_uint32 (entry->time, buffer, size, offset);
prop_copy_uint32 (entry->moof_offset, buffer, size, offset);
}
bytes = (tfra->lengths & (0x3 << 4)) + 1;
data = GUINT32_TO_BE (entry->traf_number);
prop_copy_fixed_size_string (((guint8 *) & data) + 4 - bytes, bytes,
buffer, size, offset);
bytes = (tfra->lengths & (0x3 << 2)) + 1;
data = GUINT32_TO_BE (entry->trun_number);
prop_copy_fixed_size_string (((guint8 *) & data) + 4 - bytes, bytes,
buffer, size, offset);
bytes = (tfra->lengths & (0x3)) + 1;
data = GUINT32_TO_BE (entry->sample_number);
prop_copy_fixed_size_string (((guint8 *) & data) + 4 - bytes, bytes,
buffer, size, offset);
}
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
static guint64
atom_mfro_copy_data (guint32 s, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
guint8 flags[3] = { 0, 0, 0 };
AtomFull mfro;
atom_full_init (&mfro, FOURCC_mfro, 0, 0, 0, flags);
if (!atom_full_copy_data (&mfro, buffer, size, offset)) {
return 0;
}
prop_copy_uint32 (s, buffer, size, offset);
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
guint64
atom_mfra_copy_data (AtomMFRA * mfra, guint8 ** buffer, guint64 * size,
guint64 * offset)
{
guint64 original_offset = *offset;
GList *walker;
if (!atom_copy_data (&mfra->header, buffer, size, offset))
return 0;
walker = g_list_first (mfra->tfras);
while (walker != NULL) {
if (!atom_tfra_copy_data ((AtomTFRA *) walker->data, buffer, size, offset)) {
return 0;
}
walker = g_list_next (walker);
}
/* 16 is the size of the mfro atom */
if (!atom_mfro_copy_data (*offset - original_offset + 16, buffer,
size, offset))
return 0;
atom_write_size (buffer, size, offset, original_offset);
return *offset - original_offset;
}
/* some sample description construction helpers */
AtomInfo *
build_esds_extension (AtomTRAK * trak, guint8 object_type, guint8 stream_type,
const GstBuffer * codec_data, guint32 avg_bitrate, guint32 max_bitrate)
{
guint32 track_id;
AtomESDS *esds;
track_id = trak->tkhd.track_ID;
esds = atom_esds_new ();
esds->es.id = track_id & 0xFFFF;
esds->es.dec_conf_desc.object_type = object_type;
esds->es.dec_conf_desc.stream_type = stream_type << 2 | 0x01;
if (avg_bitrate > 0)
esds->es.dec_conf_desc.avg_bitrate = avg_bitrate;
if (max_bitrate > 0)
esds->es.dec_conf_desc.max_bitrate = max_bitrate;
/* optional DecoderSpecificInfo */
if (codec_data) {
DecoderSpecificInfoDescriptor *desc;
gsize size;
esds->es.dec_conf_desc.dec_specific_info = desc =
desc_dec_specific_info_new ();
size = gst_buffer_get_size ((GstBuffer *) codec_data);
desc_dec_specific_info_alloc_data (desc, size);
gst_buffer_extract ((GstBuffer *) codec_data, 0, desc->data, size);
}
return build_atom_info_wrapper ((Atom *) esds, atom_esds_copy_data,
atom_esds_free);
}
AtomInfo *
build_btrt_extension (guint32 buffer_size_db, guint32 avg_bitrate,
guint32 max_bitrate)
{
AtomData *atom_data = atom_data_new (FOURCC_btrt);
guint8 *data;
atom_data_alloc_mem (atom_data, 12);
data = atom_data->data;
GST_WRITE_UINT32_BE (data, buffer_size_db);
GST_WRITE_UINT32_BE (data + 4, max_bitrate);
GST_WRITE_UINT32_BE (data + 8, avg_bitrate);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
static AtomInfo *
build_mov_wave_extension (guint32 fourcc, AtomInfo * atom1, AtomInfo * atom2,
gboolean terminator)
{
AtomWAVE *wave;
AtomFRMA *frma;
Atom *ext_atom;
/* Build WAVE atom for sample table entry */
wave = atom_wave_new ();
/* Prepend Terminator atom to the WAVE list first, so it ends up last */
if (terminator) {
ext_atom = (Atom *) atom_data_new (FOURCC_null);
wave->extension_atoms =
atom_info_list_prepend_atom (wave->extension_atoms, (Atom *) ext_atom,
(AtomCopyDataFunc) atom_data_copy_data, (AtomFreeFunc) atom_data_free);
}
/* Add supplied atoms to WAVE */
if (atom2)
wave->extension_atoms = g_list_prepend (wave->extension_atoms, atom2);
if (atom1)
wave->extension_atoms = g_list_prepend (wave->extension_atoms, atom1);
/* Add FRMA to the WAVE */
frma = atom_frma_new ();
frma->media_type = fourcc;
wave->extension_atoms =
atom_info_list_prepend_atom (wave->extension_atoms, (Atom *) frma,
(AtomCopyDataFunc) atom_frma_copy_data, (AtomFreeFunc) atom_frma_free);
return build_atom_info_wrapper ((Atom *) wave, atom_wave_copy_data,
atom_wave_free);
}
AtomInfo *
build_mov_aac_extension (AtomTRAK * trak, const GstBuffer * codec_data,
guint32 avg_bitrate, guint32 max_bitrate)
{
AtomInfo *esds, *mp4a;
GstBuffer *buf;
guint32 tmp = 0;
/* Add ESDS atom to WAVE */
esds = build_esds_extension (trak, ESDS_OBJECT_TYPE_MPEG4_P3,
ESDS_STREAM_TYPE_AUDIO, codec_data, avg_bitrate, max_bitrate);
/* Add MP4A atom to the WAVE:
* not really in spec, but makes offset based players happy */
buf = GST_BUFFER_NEW_READONLY (&tmp, 4);
mp4a = build_codec_data_extension (FOURCC_mp4a, buf);
gst_buffer_unref (buf);
return build_mov_wave_extension (FOURCC_mp4a, mp4a, esds, TRUE);
}
AtomInfo *
build_mov_alac_extension (const GstBuffer * codec_data)
{
AtomInfo *alac;
alac = build_codec_data_extension (FOURCC_alac, codec_data);
return build_mov_wave_extension (FOURCC_alac, NULL, alac, TRUE);
}
AtomInfo *
build_jp2x_extension (const GstBuffer * prefix)
{
AtomData *atom_data;
if (!prefix) {
return NULL;
}
atom_data = atom_data_new_from_gst_buffer (FOURCC_jp2x, prefix);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_jp2h_extension (gint width, gint height, const gchar * colorspace,
gint ncomp, const GValue * cmap_array, const GValue * cdef_array)
{
AtomData *atom_data;
GstBuffer *buf;
guint8 cenum;
gint i;
gint idhr_size = 22;
gint colr_size = 15;
gint cmap_size = 0, cdef_size = 0;
gint cmap_array_size = 0;
gint cdef_array_size = 0;
GstByteWriter writer;
g_return_val_if_fail (cmap_array == NULL ||
GST_VALUE_HOLDS_ARRAY (cmap_array), NULL);
g_return_val_if_fail (cdef_array == NULL ||
GST_VALUE_HOLDS_ARRAY (cdef_array), NULL);
if (g_str_equal (colorspace, "sRGB")) {
cenum = 0x10;
if (ncomp == 0)
ncomp = 3;
} else if (g_str_equal (colorspace, "GRAY")) {
cenum = 0x11;
if (ncomp == 0)
ncomp = 1;
} else if (g_str_equal (colorspace, "sYUV")) {
cenum = 0x12;
if (ncomp == 0)
ncomp = 3;
} else
return NULL;
if (cmap_array) {
cmap_array_size = gst_value_array_get_size (cmap_array);
cmap_size = 8 + cmap_array_size * 4;
}
if (cdef_array) {
cdef_array_size = gst_value_array_get_size (cdef_array);
cdef_size = 8 + 2 + cdef_array_size * 6;
}
gst_byte_writer_init_with_size (&writer,
idhr_size + colr_size + cmap_size + cdef_size, TRUE);
/* ihdr = image header box */
gst_byte_writer_put_uint32_be_unchecked (&writer, 22);
gst_byte_writer_put_uint32_le_unchecked (&writer, FOURCC_ihdr);
gst_byte_writer_put_uint32_be_unchecked (&writer, height);
gst_byte_writer_put_uint32_be_unchecked (&writer, width);
gst_byte_writer_put_uint16_be_unchecked (&writer, ncomp);
/* 8 bits per component, unsigned */
gst_byte_writer_put_uint8_unchecked (&writer, 0x7);
/* compression type; reserved */
gst_byte_writer_put_uint8_unchecked (&writer, 0x7);
/* colour space (un)known */
gst_byte_writer_put_uint8_unchecked (&writer, 0x0);
/* intellectual property right (box present) */
gst_byte_writer_put_uint8_unchecked (&writer, 0x0);
/* colour specification box */
gst_byte_writer_put_uint32_be_unchecked (&writer, 15);
gst_byte_writer_put_uint32_le_unchecked (&writer, FOURCC_colr);
/* specification method: enumerated */
gst_byte_writer_put_uint8_unchecked (&writer, 0x1);
/* precedence; reserved */
gst_byte_writer_put_uint8_unchecked (&writer, 0x0);
/* approximation; reserved */
gst_byte_writer_put_uint8_unchecked (&writer, 0x0);
/* enumerated colourspace */
gst_byte_writer_put_uint32_be_unchecked (&writer, cenum);
if (cmap_array) {
gst_byte_writer_put_uint32_be_unchecked (&writer, cmap_size);
gst_byte_writer_put_uint32_le_unchecked (&writer, FOURCC_cmap);
for (i = 0; i < cmap_array_size; i++) {
const GValue *item;
gint value;
guint16 cmp;
guint8 mtyp;
guint8 pcol;
item = gst_value_array_get_value (cmap_array, i);
value = g_value_get_int (item);
/* value is '(mtyp << 24) | (pcol << 16) | cmp' */
cmp = value & 0xFFFF;
mtyp = value >> 24;
pcol = (value >> 16) & 0xFF;
if (mtyp == 1)
GST_WARNING ("MTYP of cmap atom signals Pallete Mapping, but we don't "
"handle Pallete mapping atoms yet");
gst_byte_writer_put_uint16_be_unchecked (&writer, cmp);
gst_byte_writer_put_uint8_unchecked (&writer, mtyp);
gst_byte_writer_put_uint8_unchecked (&writer, pcol);
}
}
if (cdef_array) {
gst_byte_writer_put_uint32_be_unchecked (&writer, cdef_size);
gst_byte_writer_put_uint32_le_unchecked (&writer, FOURCC_cdef);
gst_byte_writer_put_uint16_be_unchecked (&writer, cdef_array_size);
for (i = 0; i < cdef_array_size; i++) {
const GValue *item;
gint value;
item = gst_value_array_get_value (cdef_array, i);
value = g_value_get_int (item);
gst_byte_writer_put_uint16_be_unchecked (&writer, i);
if (value > 0) {
gst_byte_writer_put_uint16_be_unchecked (&writer, 0);
gst_byte_writer_put_uint16_be_unchecked (&writer, value);
} else if (value < 0) {
gst_byte_writer_put_uint16_be_unchecked (&writer, -value);
gst_byte_writer_put_uint16_be_unchecked (&writer, 0); /* TODO what here? */
} else {
gst_byte_writer_put_uint16_be_unchecked (&writer, 1);
gst_byte_writer_put_uint16_be_unchecked (&writer, 0);
}
}
}
g_assert (gst_byte_writer_get_remaining (&writer) == 0);
buf = gst_byte_writer_reset_and_get_buffer (&writer);
atom_data = atom_data_new_from_gst_buffer (FOURCC_jp2h, buf);
gst_buffer_unref (buf);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_codec_data_extension (guint32 fourcc, const GstBuffer * codec_data)
{
AtomData *data;
AtomInfo *result = NULL;
if (codec_data) {
data = atom_data_new_from_gst_buffer (fourcc, codec_data);
result = build_atom_info_wrapper ((Atom *) data, atom_data_copy_data,
atom_data_free);
}
return result;
}
AtomInfo *
build_amr_extension (void)
{
guint8 ext[9];
GstBuffer *buf;
AtomInfo *res;
/* vendor */
GST_WRITE_UINT32_LE (ext, 0);
/* decoder version */
GST_WRITE_UINT8 (ext + 4, 0);
/* mode set (all modes) */
GST_WRITE_UINT16_BE (ext + 5, 0x81FF);
/* mode change period (no restriction) */
GST_WRITE_UINT8 (ext + 7, 0);
/* frames per sample */
GST_WRITE_UINT8 (ext + 8, 1);
buf = GST_BUFFER_NEW_READONLY (ext, sizeof (ext));
res = build_codec_data_extension (FOURCC_damr, buf);
gst_buffer_unref (buf);
return res;
}
AtomInfo *
build_h263_extension (void)
{
guint8 ext[7];
GstBuffer *buf;
AtomInfo *res;
/* vendor */
GST_WRITE_UINT32_LE (ext, 0);
/* decoder version */
GST_WRITE_UINT8 (ext + 4, 0);
/* level / profile */
/* FIXME ? maybe ? obtain somewhere; baseline for now */
GST_WRITE_UINT8 (ext + 5, 10);
GST_WRITE_UINT8 (ext + 6, 0);
buf = GST_BUFFER_NEW_READONLY (ext, sizeof (ext));
res = build_codec_data_extension (FOURCC_d263, buf);
gst_buffer_unref (buf);
return res;
}
AtomInfo *
build_gama_atom (gdouble gamma)
{
AtomInfo *res;
guint32 gamma_fp;
GstBuffer *buf;
/* convert to uint32 from fixed point */
gamma_fp = (guint32) 65536 *gamma;
gamma_fp = GUINT32_TO_BE (gamma_fp);
buf = GST_BUFFER_NEW_READONLY (&gamma_fp, 4);
res = build_codec_data_extension (FOURCC_gama, buf);
gst_buffer_unref (buf);
return res;
}
AtomInfo *
build_SMI_atom (const GstBuffer * seqh)
{
AtomInfo *res;
GstBuffer *buf;
gsize size;
guint8 *data;
/* the seqh plus its size and fourcc */
size = gst_buffer_get_size ((GstBuffer *) seqh);
data = g_malloc (size + 8);
GST_WRITE_UINT32_LE (data, FOURCC_SEQH);
GST_WRITE_UINT32_BE (data + 4, size + 8);
gst_buffer_extract ((GstBuffer *) seqh, 0, data + 8, size);
buf = gst_buffer_new_wrapped (data, size + 8);
res = build_codec_data_extension (FOURCC_SMI_, buf);
gst_buffer_unref (buf);
return res;
}
static AtomInfo *
build_ima_adpcm_atom (gint channels, gint rate, gint blocksize)
{
#define IMA_ADPCM_ATOM_SIZE 20
AtomData *atom_data;
guint8 *data;
guint32 fourcc;
gint samplesperblock;
gint bytespersec;
/* The FOURCC for WAV codecs in QT is 'ms' followed by the 16 bit wave codec
identifier. Note that the identifier here is big-endian, but when used
within the WAVE header (below), it's little endian. */
fourcc = MS_WAVE_FOURCC (0x11);
atom_data = atom_data_new (fourcc);
atom_data_alloc_mem (atom_data, IMA_ADPCM_ATOM_SIZE);
data = atom_data->data;
/* This atom's content is a WAVE header, including 2 bytes of extra data.
Note that all of this is little-endian, unlike most stuff in qt. */
/* 4 bytes header per channel (including 1 sample). Then 2 samples per byte
for the rest. Simplifies to this. */
samplesperblock = 2 * blocksize / channels - 7;
bytespersec = rate * blocksize / samplesperblock;
GST_WRITE_UINT16_LE (data, 0x11);
GST_WRITE_UINT16_LE (data + 2, channels);
GST_WRITE_UINT32_LE (data + 4, rate);
GST_WRITE_UINT32_LE (data + 8, bytespersec);
GST_WRITE_UINT16_LE (data + 12, blocksize);
GST_WRITE_UINT16_LE (data + 14, 4);
GST_WRITE_UINT16_LE (data + 16, 2); /* Two extra bytes */
GST_WRITE_UINT16_LE (data + 18, samplesperblock);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_ima_adpcm_extension (gint channels, gint rate, gint blocksize)
{
AtomWAVE *wave;
AtomFRMA *frma;
Atom *ext_atom;
/* Add WAVE atom */
wave = atom_wave_new ();
/* Prepend Terminator atom to the WAVE list first, so it ends up last */
ext_atom = (Atom *) atom_data_new (FOURCC_null);
wave->extension_atoms =
atom_info_list_prepend_atom (wave->extension_atoms, (Atom *) ext_atom,
(AtomCopyDataFunc) atom_data_copy_data, (AtomFreeFunc) atom_data_free);
/* Add wave ima adpcm atom to WAVE */
wave->extension_atoms = g_list_prepend (wave->extension_atoms,
build_ima_adpcm_atom (channels, rate, blocksize));
/* Add FRMA to the WAVE */
frma = atom_frma_new ();
frma->media_type = MS_WAVE_FOURCC (0x11);
wave->extension_atoms =
atom_info_list_prepend_atom (wave->extension_atoms, (Atom *) frma,
(AtomCopyDataFunc) atom_frma_copy_data, (AtomFreeFunc) atom_frma_free);
return build_atom_info_wrapper ((Atom *) wave, atom_wave_copy_data,
atom_wave_free);
}
AtomInfo *
build_ac3_extension (guint8 fscod, guint8 bsid, guint8 bsmod, guint8 acmod,
guint8 lfe_on, guint8 bitrate_code)
{
AtomData *atom_data = atom_data_new (FOURCC_dac3);
guint8 *data;
atom_data_alloc_mem (atom_data, 3);
data = atom_data->data;
/* Bits from the spec
* fscod 2
* bsid 5
* bsmod 3
* acmod 3
* lfeon 1
* bit_rate_code 5
* reserved 5
*/
/* Some bit manipulation magic. Need bitwriter */
data[0] = (fscod << 6) | (bsid << 1) | ((bsmod >> 2) & 1);
data[1] =
((bsmod & 0x3) << 6) | (acmod << 3) | ((lfe_on & 1) << 2) | ((bitrate_code
>> 3) & 0x3);
data[2] = ((bitrate_code & 0x7) << 5);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_opus_extension (guint32 rate, guint8 channels, guint8 mapping_family,
guint8 stream_count, guint8 coupled_count, guint8 channel_mapping[256],
guint16 pre_skip, guint16 output_gain)
{
AtomData *atom_data;
guint8 *data_block;
GstByteWriter bw;
gboolean hdl = TRUE;
guint data_block_len;
gst_byte_writer_init (&bw);
hdl &= gst_byte_writer_put_uint8 (&bw, 0x00); /* version number */
hdl &= gst_byte_writer_put_uint8 (&bw, channels);
hdl &= gst_byte_writer_put_uint16_le (&bw, pre_skip);
hdl &= gst_byte_writer_put_uint32_le (&bw, rate);
hdl &= gst_byte_writer_put_uint16_le (&bw, output_gain);
hdl &= gst_byte_writer_put_uint8 (&bw, mapping_family);
if (mapping_family > 0) {
hdl &= gst_byte_writer_put_uint8 (&bw, stream_count);
hdl &= gst_byte_writer_put_uint8 (&bw, coupled_count);
hdl &= gst_byte_writer_put_data (&bw, channel_mapping, channels);
}
if (!hdl) {
GST_WARNING ("Error creating header");
return NULL;
}
data_block_len = gst_byte_writer_get_size (&bw);
data_block = gst_byte_writer_reset_and_get_data (&bw);
atom_data = atom_data_new_from_data (FOURCC_dops, data_block, data_block_len);
g_free (data_block);
return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
atom_data_free);
}
AtomInfo *
build_uuid_xmp_atom (GstBuffer * xmp_data)
{
AtomUUID *uuid;
gsize size;
static const guint8 xmp_uuid[] = { 0xBE, 0x7A, 0xCF, 0xCB,
0x97, 0xA9, 0x42, 0xE8,
0x9C, 0x71, 0x99, 0x94,
0x91, 0xE3, 0xAF, 0xAC
};
if (xmp_data == NULL)
return NULL;
uuid = atom_uuid_new ();
memcpy (uuid->uuid, xmp_uuid, 16);
size = gst_buffer_get_size (xmp_data);
uuid->data = g_malloc (size);
uuid->datalen = size;
gst_buffer_extract (xmp_data, 0, uuid->data, size);
return build_atom_info_wrapper ((Atom *) uuid, atom_uuid_copy_data,
atom_uuid_free);
}