/* Quicktime muxer plugin for GStreamer * Copyright (C) 2008-2010 Thiago Santos * Copyright (C) 2008 Mark Nauwelaerts * * 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 #include #include #include #include #include /* * Creates a new AtomsContext for the given flavor. */ AtomsContext * atoms_context_new (AtomsTreeFlavor flavor, gboolean force_create_timecode_trak) { AtomsContext *context = g_new0 (AtomsContext, 1); context->flavor = flavor; context->force_create_timecode_trak = force_create_timecode_trak; 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) { gint64 curtime_s = g_get_real_time () / G_USEC_PER_SEC; /* FIXME this should use UTC coordinated time */ return curtime_s + (((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_nmhd_init (AtomNMHD * nmhd) { atom_header_set (&nmhd->header, FOURCC_nmhd, 0, 0); nmhd->flags = 0; } static void atom_nmhd_clear (AtomNMHD * nmhd) { atom_clear (&nmhd->header); } static AtomNMHD * atom_nmhd_new (void) { AtomNMHD *nmhd = g_new0 (AtomNMHD, 1); atom_nmhd_init (nmhd); return nmhd; } static void atom_nmhd_free (AtomNMHD * nmhd) { atom_nmhd_clear (nmhd); g_free (nmhd); } 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); } /* svmi is specified in ISO 23000-11 (Stereoscopic video application format) * MPEG-A */ 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); co64->chunk_offset = 0; co64->max_offset = 0; 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; } if (minf->nmhd) { atom_nmhd_free (minf->nmhd); minf->nmhd = 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 guint64 atom_nmhd_copy_data (AtomNMHD * nmhd, guint8 ** buffer, guint64 * size, guint64 * offset) { guint64 original_offset = *offset; if (!atom_copy_data (&nmhd->header, buffer, size, offset)) { return 0; } prop_copy_uint32 (nmhd->flags, buffer, size, offset); 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) { STSCEntry *prev_entry = &atom_array_index (&stsc->entries, len - 2); STSCEntry *current_entry = &atom_array_index (&stsc->entries, len - 1); if (prev_entry->samples_per_chunk == current_entry->samples_per_chunk && prev_entry->sample_description_index == current_entry->sample_description_index) { 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; /* If any (mdat-relative) offset will by over 32-bits when converted to an * absolute file offset then we need to write a 64-bit co64 atom, otherwise * we can write a smaller stco 32-bit table */ gboolean write_stco64 = (stco64->max_offset + stco64->chunk_offset) > G_MAXUINT32; if (write_stco64) stco64->header.header.type = FOURCC_co64; else 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 (write_stco64) { prop_copy_uint64 (value, buffer, size, offset); } else { prop_copy_uint32 ((guint32) 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; } } else if (minf->nmhd) { if (!atom_nmhd_copy_data (minf->nmhd, 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; } /* only write mehd if we have anything extra to add */ 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, guint32 sample_description_index) { 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 = sample_description_index; } else { STSCEntry nentry; nentry.first_chunk = first_chunk; nentry.samples_per_chunk = nsamples; nentry.sample_description_index = sample_description_index; 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 > stco64->max_offset) stco64->max_offset = entry; 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, stbl->stsd.n_entries); } 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; if (context->flavor != ATOMS_TREE_FLAVOR_MOV && !context->force_create_timecode_trak) { return NULL; } if (context->flavor == ATOMS_TREE_FLAVOR_MOV) { AtomGMHD *gmhd = trak->mdia.minf.gmhd; 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; } else if (context->force_create_timecode_trak) { AtomNMHD *nmhd = trak->mdia.minf.nmhd; /* MOV files use GMHD, other files use NMHD */ nmhd = atom_nmhd_new (); trak->mdia.minf.nmhd = nmhd; } else { return NULL; } ste = atom_trak_add_timecode_entry (trak, context, trak_timescale, tc); 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; primaries = gst_video_color_primaries_to_iso (colorimetry->primaries); transfer_function = gst_video_transfer_function_to_iso (colorimetry->transfer); matrix = gst_video_color_matrix_to_iso (colorimetry->matrix); 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; } if (trak->mdia.minf.stbl.stsd.n_entries < 1) { atom_trak_set_video_commons (trak, context, scale, dwidth, dheight); trak->is_video = TRUE; trak->is_h264 = (entry->fourcc == FOURCC_avc1 || entry->fourcc == FOURCC_avc3); } ste = atom_trak_add_video_entry (trak, context, entry->fourcc); 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; } void atom_moof_set_base_offset (AtomMOOF * moof, guint64 offset) { GList *trafs = moof->trafs; if (offset == moof->traf_offset) return; /* Nothing to do */ while (trafs) { AtomTRAF *traf = (AtomTRAF *) trafs->data; traf->tfhd.header.flags[2] |= TF_BASE_DATA_OFFSET; traf->tfhd.base_data_offset = offset; trafs = g_list_next (trafs); } moof->traf_offset = offset; } 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) { guint64 original_offset = *offset; guint32 flags, i; flags = atom_full_get_flags_as_uint (&trun->header); 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) { 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) { 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)) { 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; 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)) { return 0; } walker = g_list_next (walker); } atom_write_size (buffer, size, offset, original_offset); 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); } void atom_trun_set_offset (AtomTRUN * trun, gint32 offset) { trun->header.flags[2] |= TR_DATA_OFFSET; trun->data_offset = offset; } static gboolean atom_trun_can_append_samples_to_entry (AtomTRUN * trun, TRUNSampleEntry * nentry, guint32 nsamples, guint32 delta, guint32 size, guint32 flags, gint32 data_offset, gint64 pts_offset) { if (pts_offset != 0) return FALSE; if (nentry->sample_flags != flags) return FALSE; if (trun->data_offset + nentry->sample_size != data_offset) return FALSE; if (nentry->sample_size != size) return FALSE; if (nentry->sample_duration != delta) return FALSE; /* FIXME: this should be TRUE but currently fails on demuxing */ return FALSE; } static void atom_trun_append_samples (AtomTRUN * trun, TRUNSampleEntry * nentry, guint32 nsamples, guint32 delta, guint32 size) { trun->sample_count += nsamples; } static void atom_trun_add_samples (AtomTRUN * trun, guint32 nsamples, guint32 delta, guint32 size, guint32 flags, gint64 pts_offset) { int i; if (pts_offset != 0) trun->header.flags[1] |= (TR_COMPOSITION_TIME_OFFSETS >> 8); for (i = 0; i < nsamples; i++) { TRUNSampleEntry nentry; nentry.sample_duration = delta; nentry.sample_size = size; nentry.sample_flags = flags; if (pts_offset != 0) { nentry.sample_composition_time_offset = pts_offset + i * delta; } else { nentry.sample_composition_time_offset = 0; } 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 nsamples, guint32 delta, guint32 size, gint32 data_offset, gboolean sync, gint64 pts_offset, gboolean sdtp_sync) { GList *l = NULL; AtomTRUN *prev_trun, *trun = NULL; TRUNSampleEntry *nentry = NULL; 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 (traf->truns) { trun = g_list_last (traf->truns)->data; nentry = &atom_array_index (&trun->entries, atom_array_get_len (&trun->entries) - 1); if (!atom_trun_can_append_samples_to_entry (trun, nentry, nsamples, delta, size, flags, data_offset, pts_offset)) { /* if we can't add to the previous trun, write a new one */ trun = NULL; nentry = NULL; } } prev_trun = trun; if (!traf->truns) { /* 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; } if (!trun) { trun = atom_trun_new (); atom_traf_add_trun (traf, trun); trun->first_sample_flags = flags; trun->data_offset = data_offset; if (data_offset != 0) trun->header.flags[2] |= TR_DATA_OFFSET; } /* check if still matching defaults, * if not, abandon default and need entry for each sample */ if (traf->tfhd.default_sample_duration != delta || prev_trun == trun) { traf->tfhd.header.flags[2] &= ~TF_DEFAULT_SAMPLE_DURATION; for (l = traf->truns; l; l = g_list_next (l)) { ((AtomTRUN *) l->data)->header.flags[1] |= (TR_SAMPLE_DURATION >> 8); } } if (traf->tfhd.default_sample_size != size || prev_trun == trun) { traf->tfhd.header.flags[2] &= ~TF_DEFAULT_SAMPLE_SIZE; for (l = traf->truns; l; l = g_list_next (l)) { ((AtomTRUN *) l->data)->header.flags[1] |= (TR_SAMPLE_SIZE >> 8); } } if (traf->tfhd.default_sample_flags != flags || prev_trun == trun) { 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; } } if (prev_trun == trun) { atom_trun_append_samples (trun, nentry, nsamples, delta, size); } else { atom_trun_add_samples (trun, nsamples, 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; /* FIXME: only one trun? */ 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); }