gst/quicktime/atoms.c

   1 /* Quicktime muxer plugin for GStreamer
   2  * Copyright (C) 2008 Thiago Sousa Santos <thiagoss@embedded.ufcg.edu.br>
   3  * Copyright (C) 2008 Mark Nauwelaerts <mnauw@users.sf.net>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20 /*
  21  * Unless otherwise indicated, Source Code is licensed under MIT license.
  22  * See further explanation attached in License Statement (distributed in the file
  23  * LICENSE).
  24  *
  25  * Permission is hereby granted, free of charge, to any person obtaining a copy of
  26  * this software and associated documentation files (the "Software"), to deal in
  27  * the Software without restriction, including without limitation the rights to
  28  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
  29  * of the Software, and to permit persons to whom the Software is furnished to do
  30  * so, subject to the following conditions:
  31  *
  32  * The above copyright notice and this permission notice shall be included in all
  33  * copies or substantial portions of the Software.
  34  *
  35  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  36  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  37  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  38  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  39  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  40  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  41  * SOFTWARE.
  42  */
  43
  44 #include "atoms.h"
  45 #include <string.h>
  46 #include <glib.h>
  47
  48 /* only needed for gst_util_uint64_scale */
  49 #include <gst/gst.h>
  50
  51 /**
  52  * Creates a new AtomsContext for the given flavor.
  53  */
  54 AtomsContext *
  55 atoms_context_new (AtomsTreeFlavor flavor)
  56 {
  57   AtomsContext *context = g_new0 (AtomsContext, 1);
  58   context->flavor = flavor;
  59   return context;
  60 }
  61
  62 /**
  63  * Frees an AtomsContext and all memory associated with it
  64  */
  65 void
  66 atoms_context_free (AtomsContext * context)
  67 {
  68   g_free (context);
  69 }
  70
  71 /* -- creation, initialization, clear and free functions -- */
  72
  73 #define SECS_PER_DAY (24 * 60 * 60)
  74 #define LEAP_YEARS_FROM_1904_TO_1970 17
  75
  76 static guint64
  77 get_current_qt_time ()
  78 {
  79   GTimeVal timeval;
  80
  81   g_get_current_time (&timeval);
  82   /* FIXME this should use UTC coordinated time */
  83   return timeval.tv_sec + (((1970 - 1904) * (guint64) 365) +
  84       LEAP_YEARS_FROM_1904_TO_1970) * SECS_PER_DAY;
  85 }
  86
  87 static void
  88 common_time_info_init (TimeInfo * ti)
  89 {
  90   ti->creation_time = ti->modification_time = get_current_qt_time ();
  91   ti->timescale = 0;
  92   ti->duration = 0;
  93 }
  94
  95 static void
  96 atom_header_set (Atom * header, guint32 fourcc, gint32 size, gint64 ext_size)
  97 {
  98   header->type = fourcc;
  99   header->size = size;
 100   header->extended_size = ext_size;
 101 }
 102
 103 static void
 104 atom_clear (Atom * atom)
 105 {
 106 }
 107
 108 static void
 109 atom_full_init (AtomFull * full, guint32 fourcc, gint32 size, gint64 ext_size,
 110     guint8 version, guint8 flags[3])
 111 {
 112   atom_header_set (&(full->header), fourcc, size, ext_size);
 113   full->version = version;
 114   full->flags[0] = flags[0];
 115   full->flags[1] = flags[1];
 116   full->flags[2] = flags[2];
 117 }
 118
 119 static void
 120 atom_full_clear (AtomFull * full)
 121 {
 122   atom_clear (&full->header);
 123 }
 124
 125 static void
 126 atom_full_free (AtomFull * full)
 127 {
 128   atom_full_clear (full);
 129   g_free (full);
 130 }
 131
 132 static AtomInfo *
 133 build_atom_info_wrapper (Atom * atom, gpointer copy_func, gpointer free_func)
 134 {
 135   AtomInfo *info = NULL;
 136
 137   if (atom) {
 138     info = g_new0 (AtomInfo, 1);
 139
 140     info->atom = atom;
 141     info->copy_data_func = copy_func;
 142     info->free_func = free_func;
 143   }
 144
 145   return info;
 146 }
 147
 148 static GList *
 149 atom_info_list_prepend_atom (GList * ai, Atom * atom,
 150     AtomCopyDataFunc copy_func, AtomFreeFunc free_func)
 151 {
 152   if (atom)
 153     return g_list_prepend (ai,
 154         build_atom_info_wrapper (atom, copy_func, free_func));
 155   else
 156     return ai;
 157 }
 158
 159 static void
 160 atom_info_list_free (GList * ai)
 161 {
 162   while (ai) {
 163     AtomInfo *info = (AtomInfo *) ai->data;
 164
 165     info->free_func (info->atom);
 166     g_free (info);
 167     ai = g_list_delete_link (ai, ai);
 168   }
 169 }
 170
 171 static AtomData *
 172 atom_data_new (guint32 fourcc)
 173 {
 174   AtomData *data = g_new0 (AtomData, 1);
 175
 176   atom_header_set (&data->header, fourcc, 0, 0);
 177   return data;
 178 }
 179
 180 static void
 181 atom_data_alloc_mem (AtomData * data, guint32 size)
 182 {
 183   if (data->data) {
 184     g_free (data->data);
 185   }
 186   data->data = g_new0 (guint8, size);
 187   data->datalen = size;
 188 }
 189
 190 static AtomData *
 191 atom_data_new_from_gst_buffer (guint32 fourcc, const GstBuffer * buf)
 192 {
 193   AtomData *data = atom_data_new (fourcc);
 194
 195   atom_data_alloc_mem (data, GST_BUFFER_SIZE (buf));
 196   g_memmove (data->data, GST_BUFFER_DATA (buf), GST_BUFFER_SIZE (buf));
 197   return data;
 198 }
 199
 200 static void
 201 atom_data_free (AtomData * data)
 202 {
 203   atom_clear (&data->header);
 204   g_free (data->data);
 205   g_free (data);
 206 }
 207
 208 static void
 209 atom_ftyp_init (AtomFTYP * ftyp, guint32 major, guint32 version, GList * brands)
 210 {
 211   gint index;
 212   GList *it = NULL;
 213
 214   atom_header_set (&ftyp->header, FOURCC_ftyp, 16, 0);
 215   ftyp->major_brand = major;
 216   ftyp->version = version;
 217
 218   /* always include major brand as compatible brand */
 219   ftyp->compatible_brands_size = g_list_length (brands) + 1;
 220   ftyp->compatible_brands = g_new (guint32, ftyp->compatible_brands_size);
 221
 222   ftyp->compatible_brands[0] = major;
 223   index = 1;
 224   for (it = brands; it != NULL; it = g_list_next (it)) {
 225     ftyp->compatible_brands[index++] = GPOINTER_TO_UINT (it->data);
 226   }
 227 }
 228
 229 AtomFTYP *
 230 atom_ftyp_new (AtomsContext * context, guint32 major, guint32 version,
 231     GList * brands)
 232 {
 233   AtomFTYP *ftyp = g_new0 (AtomFTYP, 1);
 234
 235   atom_ftyp_init (ftyp, major, version, brands);
 236   return ftyp;
 237 }
 238
 239 void
 240 atom_ftyp_free (AtomFTYP * ftyp)
 241 {
 242   atom_clear (&ftyp->header);
 243   g_free (ftyp->compatible_brands);
 244   ftyp->compatible_brands = NULL;
 245   g_free (ftyp);
 246 }
 247
 248 static void
 249 atom_esds_init (AtomESDS * esds)
 250 {
 251   guint8 flags[3] = { 0, 0, 0 };
 252
 253   atom_full_init (&esds->header, FOURCC_esds, 0, 0, 0, flags);
 254   desc_es_init (&esds->es);
 255 }
 256
 257 static AtomESDS *
 258 atom_esds_new ()
 259 {
 260   AtomESDS *esds = g_new0 (AtomESDS, 1);
 261
 262   atom_esds_init (esds);
 263   return esds;
 264 }
 265
 266 static void
 267 atom_esds_free (AtomESDS * esds)
 268 {
 269   atom_full_clear (&esds->header);
 270   desc_es_descriptor_clear (&esds->es);
 271   g_free (esds);
 272 }
 273
 274 static AtomFRMA *
 275 atom_frma_new ()
 276 {
 277   AtomFRMA *frma = g_new0 (AtomFRMA, 1);
 278
 279   atom_header_set (&frma->header, FOURCC_frma, 0, 0);
 280   return frma;
 281 }
 282
 283 static void
 284 atom_frma_free (AtomFRMA * frma)
 285 {
 286   atom_clear (&frma->header);
 287   g_free (frma);
 288 }
 289
 290 static AtomWAVE *
 291 atom_wave_new ()
 292 {
 293   AtomWAVE *wave = g_new0 (AtomWAVE, 1);
 294
 295   atom_header_set (&wave->header, FOURCC_wave, 0, 0);
 296   return wave;
 297 }
 298
 299 static void
 300 atom_wave_free (AtomWAVE * wave)
 301 {
 302   atom_clear (&wave->header);
 303   atom_info_list_free (wave->extension_atoms);
 304   g_free (wave);
 305 }
 306
 307 static void
 308 atom_elst_init (AtomELST * elst)
 309 {
 310   guint8 flags[3] = { 0, 0, 0 };
 311   atom_full_init (&elst->header, FOURCC_elst, 0, 0, 0, flags);
 312   elst->entries = 0;
 313 }
 314
 315 static void
 316 atom_elst_clear (AtomELST * elst)
 317 {
 318   GSList *walker;
 319
 320   atom_full_clear (&elst->header);
 321   walker = elst->entries;
 322   while (walker) {
 323     g_free ((EditListEntry *) walker->data);
 324     walker = g_slist_next (walker);
 325   }
 326   g_slist_free (elst->entries);
 327 }
 328
 329 static void
 330 atom_edts_init (AtomEDTS * edts)
 331 {
 332   atom_header_set (&edts->header, FOURCC_edts, 0, 0);
 333   atom_elst_init (&edts->elst);
 334 }
 335
 336 static void
 337 atom_edts_clear (AtomEDTS * edts)
 338 {
 339   atom_clear (&edts->header);
 340   atom_elst_clear (&edts->elst);
 341 }
 342
 343 AtomEDTS *
 344 atom_edts_new ()
 345 {
 346   AtomEDTS *edts = g_new0 (AtomEDTS, 1);
 347   atom_edts_init (edts);
 348   return edts;
 349 }
 350
 351 void
 352 atom_edts_free (AtomEDTS * edts)
 353 {
 354   atom_edts_clear (edts);
 355   g_free (edts);
 356 }
 357
 358 static void
 359 atom_sample_entry_init (SampleTableEntry * se, guint32 type)
 360 {
 361   atom_header_set (&se->header, type, 0, 0);
 362
 363   memset (se->reserved, 0, sizeof (guint8) * 6);
 364   se->data_reference_index = 0;
 365 }
 366
 367 static void
 368 atom_sample_entry_free (SampleTableEntry * se)
 369 {
 370   atom_clear (&se->header);
 371 }
 372
 373 static void
 374 sample_entry_mp4a_init (SampleTableEntryMP4A * mp4a)
 375 {
 376   atom_sample_entry_init (&mp4a->se, FOURCC_mp4a);
 377
 378   mp4a->version = 0;
 379   mp4a->revision_level = 0;
 380   mp4a->vendor = 0;
 381   mp4a->channels = 2;
 382   mp4a->sample_size = 16;
 383   mp4a->compression_id = 0;
 384   mp4a->packet_size = 0;
 385   mp4a->sample_rate = 0;
 386   /* following only used if version is 1 */
 387   mp4a->samples_per_packet = 0;
 388   mp4a->bytes_per_packet = 0;
 389   mp4a->bytes_per_frame = 0;
 390   mp4a->bytes_per_sample = 0;
 391
 392   mp4a->extension_atoms = NULL;
 393 }
 394
 395 static SampleTableEntryMP4A *
 396 sample_entry_mp4a_new ()
 397 {
 398   SampleTableEntryMP4A *mp4a = g_new0 (SampleTableEntryMP4A, 1);
 399
 400   sample_entry_mp4a_init (mp4a);
 401   return mp4a;
 402 }
 403
 404 static void
 405 sample_entry_mp4a_free (SampleTableEntryMP4A * mp4a)
 406 {
 407   atom_sample_entry_free (&mp4a->se);
 408   atom_info_list_free (mp4a->extension_atoms);
 409   g_free (mp4a);
 410 }
 411
 412 static void
 413 sample_entry_mp4v_init (SampleTableEntryMP4V * mp4v, AtomsContext * context)
 414 {
 415   atom_sample_entry_init (&mp4v->se, FOURCC_mp4v);
 416
 417   mp4v->version = 0;
 418   mp4v->revision_level = 0;
 419   mp4v->vendor = 0;
 420
 421   mp4v->temporal_quality = 0;
 422   mp4v->spatial_quality = 0;
 423
 424   /* qt and ISO base media do not contradict, and examples agree */
 425   mp4v->horizontal_resolution = 0x00480000;
 426   mp4v->vertical_resolution = 0x00480000;
 427
 428   mp4v->datasize = 0;
 429   mp4v->frame_count = 1;
 430
 431   memset (mp4v->compressor, 0, sizeof (guint8) * 32);
 432
 433   mp4v->depth = 0;
 434   mp4v->color_table_id = 0;
 435
 436   mp4v->extension_atoms = NULL;
 437 }
 438
 439 static void
 440 sample_entry_mp4v_free (SampleTableEntryMP4V * mp4v)
 441 {
 442   atom_sample_entry_free (&mp4v->se);
 443   atom_info_list_free (mp4v->extension_atoms);
 444   g_free (mp4v);
 445 }
 446
 447 static SampleTableEntryMP4V *
 448 sample_entry_mp4v_new (AtomsContext * context)
 449 {
 450   SampleTableEntryMP4V *mp4v = g_new0 (SampleTableEntryMP4V, 1);
 451
 452   sample_entry_mp4v_init (mp4v, context);
 453   return mp4v;
 454 }
 455
 456 static void
 457 atom_stsd_init (AtomSTSD * stsd)
 458 {
 459   guint8 flags[3] = { 0, 0, 0 };
 460
 461   atom_full_init (&stsd->header, FOURCC_stsd, 0, 0, 0, flags);
 462   stsd->entries = NULL;
 463 }
 464
 465 static void
 466 atom_stsd_clear (AtomSTSD * stsd)
 467 {
 468   GList *walker;
 469
 470   atom_full_clear (&stsd->header);
 471   walker = stsd->entries;
 472   while (walker) {
 473     GList *aux = walker;
 474     SampleTableEntry *se = (SampleTableEntry *) aux->data;
 475
 476     walker = g_list_next (walker);
 477     stsd->entries = g_list_remove_link (stsd->entries, aux);
 478
 479     switch (se->kind) {
 480       case AUDIO:
 481         sample_entry_mp4a_free ((SampleTableEntryMP4A *) se);
 482         break;
 483       case VIDEO:
 484         sample_entry_mp4v_free ((SampleTableEntryMP4V *) se);
 485         break;
 486       default:
 487         /* best possible cleanup */
 488         atom_sample_entry_free (se);
 489     }
 490     g_list_free (aux);
 491   }
 492 }
 493
 494 static void
 495 atom_ctts_init (AtomCTTS * ctts)
 496 {
 497   guint8 flags[3] = { 0, 0, 0 };
 498
 499   atom_full_init (&ctts->header, FOURCC_ctts, 0, 0, 0, flags);
 500   ctts->entries = NULL;
 501 }
 502
 503 static AtomCTTS *
 504 atom_ctts_new ()
 505 {
 506   AtomCTTS *ctts = g_new0 (AtomCTTS, 1);
 507
 508   atom_ctts_init (ctts);
 509   return ctts;
 510 }
 511
 512 static void
 513 atom_ctts_free (AtomCTTS * ctts)
 514 {
 515   GList *walker;
 516
 517   atom_full_clear (&ctts->header);
 518   walker = ctts->entries;
 519   while (walker) {
 520     GList *aux = walker;
 521
 522     walker = g_list_next (walker);
 523     ctts->entries = g_list_remove_link (ctts->entries, aux);
 524     g_free ((CTTSEntry *) aux->data);
 525     g_list_free (aux);
 526   }
 527   g_free (ctts);
 528 }
 529
 530 static void
 531 atom_stts_init (AtomSTTS * stts)
 532 {
 533   guint8 flags[3] = { 0, 0, 0 };
 534
 535   atom_full_init (&stts->header, FOURCC_stts, 0, 0, 0, flags);
 536   stts->entries = NULL;
 537 }
 538
 539 static void
 540 atom_stts_clear (AtomSTTS * stts)
 541 {
 542   GList *walker;
 543
 544   atom_full_clear (&stts->header);
 545   walker = stts->entries;
 546   while (walker) {
 547     GList *aux = walker;
 548
 549     walker = g_list_next (walker);
 550     stts->entries = g_list_remove_link (stts->entries, aux);
 551     g_free ((STTSEntry *) aux->data);
 552     g_list_free (aux);
 553   }
 554   stts->n_entries = 0;
 555 }
 556
 557 static void
 558 atom_stsz_init (AtomSTSZ * stsz)
 559 {
 560   guint8 flags[3] = { 0, 0, 0 };
 561
 562   atom_full_init (&stsz->header, FOURCC_stsz, 0, 0, 0, flags);
 563   stsz->sample_size = 0;
 564   stsz->table_size = 0;
 565   stsz->entries = NULL;
 566 }
 567
 568 static void
 569 atom_stsz_clear (AtomSTSZ * stsz)
 570 {
 571   atom_full_clear (&stsz->header);
 572   g_list_free (stsz->entries);
 573   stsz->entries = NULL;
 574   stsz->table_size = 0;
 575 }
 576
 577 static void
 578 atom_stsc_init (AtomSTSC * stsc)
 579 {
 580   guint8 flags[3] = { 0, 0, 0 };
 581
 582   atom_full_init (&stsc->header, FOURCC_stsc, 0, 0, 0, flags);
 583   stsc->entries = NULL;
 584   stsc->n_entries = 0;
 585 }
 586
 587 static void
 588 atom_stsc_clear (AtomSTSC * stsc)
 589 {
 590   GList *walker;
 591
 592   atom_full_clear (&stsc->header);
 593   walker = stsc->entries;
 594   while (walker) {
 595     GList *aux = walker;
 596
 597     walker = g_list_next (walker);
 598     stsc->entries = g_list_remove_link (stsc->entries, aux);
 599     g_free ((STSCEntry *) aux->data);
 600     g_list_free (aux);
 601   }
 602   stsc->n_entries = 0;
 603 }
 604
 605 static void
 606 atom_co64_init (AtomSTCO64 * co64)
 607 {
 608   guint8 flags[3] = { 0, 0, 0 };
 609
 610   atom_full_init (&co64->header, FOURCC_co64, 0, 0, 0, flags);
 611   co64->entries = NULL;
 612   co64->n_entries = 0;
 613 }
 614
 615 static void
 616 atom_stco64_clear (AtomSTCO64 * stco64)
 617 {
 618   GList *walker;
 619
 620   atom_full_clear (&stco64->header);
 621   walker = stco64->entries;
 622   while (walker) {
 623     GList *aux = walker;
 624
 625     walker = g_list_next (walker);
 626     stco64->entries = g_list_remove_link (stco64->entries, aux);
 627     g_free ((guint64 *) aux->data);
 628     g_list_free (aux);
 629   }
 630   stco64->n_entries = 0;
 631 }
 632
 633 static void
 634 atom_stss_init (AtomSTSS * stss)
 635 {
 636   guint8 flags[3] = { 0, 0, 0 };
 637
 638   atom_full_init (&stss->header, FOURCC_stss, 0, 0, 0, flags);
 639   stss->entries = NULL;
 640   stss->n_entries = 0;
 641 }
 642
 643 static void
 644 atom_stss_clear (AtomSTSS * stss)
 645 {
 646   atom_full_clear (&stss->header);
 647   g_list_free (stss->entries);
 648   stss->entries = NULL;
 649   stss->n_entries = 0;
 650 }
 651
 652 static void
 653 atom_stbl_init (AtomSTBL * stbl)
 654 {
 655   atom_header_set (&stbl->header, FOURCC_stbl, 0, 0);
 656
 657   atom_stts_init (&stbl->stts);
 658   atom_stss_init (&stbl->stss);
 659   atom_stsd_init (&stbl->stsd);
 660   atom_stsz_init (&stbl->stsz);
 661   atom_stsc_init (&stbl->stsc);
 662   stbl->ctts = NULL;
 663
 664   atom_co64_init (&stbl->stco64);
 665 }
 666
 667 static void
 668 atom_stbl_clear (AtomSTBL * stbl)
 669 {
 670   atom_clear (&stbl->header);
 671   atom_stsd_clear (&stbl->stsd);
 672   atom_stts_clear (&stbl->stts);
 673   atom_stss_clear (&stbl->stss);
 674   atom_stsc_clear (&stbl->stsc);
 675   atom_stsz_clear (&stbl->stsz);
 676   if (stbl->ctts) {
 677     atom_ctts_free (stbl->ctts);
 678   }
 679   atom_stco64_clear (&stbl->stco64);
 680 }
 681
 682 static void
 683 atom_vmhd_init (AtomVMHD * vmhd, AtomsContext * context)
 684 {
 685   guint8 flags[3] = { 0, 0, 1 };
 686
 687   atom_full_init (&vmhd->header, FOURCC_vmhd, 0, 0, 0, flags);
 688   vmhd->graphics_mode = 0x0;
 689   memset (vmhd->opcolor, 0, sizeof (guint16) * 3);
 690
 691   if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
 692     vmhd->graphics_mode = 0x40;
 693     vmhd->opcolor[0] = 32768;
 694     vmhd->opcolor[1] = 32768;
 695     vmhd->opcolor[2] = 32768;
 696   }
 697 }
 698
 699 static AtomVMHD *
 700 atom_vmhd_new (AtomsContext * context)
 701 {
 702   AtomVMHD *vmhd = g_new0 (AtomVMHD, 1);
 703
 704   atom_vmhd_init (vmhd, context);
 705   return vmhd;
 706 }
 707
 708 static void
 709 atom_vmhd_free (AtomVMHD * vmhd)
 710 {
 711   atom_full_clear (&vmhd->header);
 712   g_free (vmhd);
 713 }
 714
 715 static void
 716 atom_smhd_init (AtomSMHD * smhd)
 717 {
 718   guint8 flags[3] = { 0, 0, 0 };
 719
 720   atom_full_init (&smhd->header, FOURCC_smhd, 0, 0, 0, flags);
 721   smhd->balance = 0;
 722   smhd->reserved = 0;
 723 }
 724
 725 static AtomSMHD *
 726 atom_smhd_new ()
 727 {
 728   AtomSMHD *smhd = g_new0 (AtomSMHD, 1);
 729
 730   atom_smhd_init (smhd);
 731   return smhd;
 732 }
 733
 734 static void
 735 atom_smhd_free (AtomSMHD * smhd)
 736 {
 737   atom_full_clear (&smhd->header);
 738   g_free (smhd);
 739 }
 740
 741 static void
 742 atom_hmhd_free (AtomHMHD * hmhd)
 743 {
 744   atom_full_clear (&hmhd->header);
 745   g_free (hmhd);
 746 }
 747
 748 static void
 749 atom_hdlr_init (AtomHDLR * hdlr)
 750 {
 751   guint8 flags[3] = { 0, 0, 0 };
 752
 753   atom_full_init (&hdlr->header, FOURCC_hdlr, 0, 0, 0, flags);
 754
 755   hdlr->component_type = 0;
 756   hdlr->handler_type = 0;
 757   hdlr->manufacturer = 0;
 758   hdlr->flags = 0;
 759   hdlr->flags_mask = 0;
 760   hdlr->name = g_strdup ("");
 761 }
 762
 763 static AtomHDLR *
 764 atom_hdlr_new ()
 765 {
 766   AtomHDLR *hdlr = g_new0 (AtomHDLR, 1);
 767
 768   atom_hdlr_init (hdlr);
 769   return hdlr;
 770 }
 771
 772 static void
 773 atom_hdlr_clear (AtomHDLR * hdlr)
 774 {
 775   atom_full_clear (&hdlr->header);
 776   if (hdlr->name) {
 777     g_free (hdlr->name);
 778     hdlr->name = NULL;
 779   }
 780 }
 781
 782 static void
 783 atom_hdlr_free (AtomHDLR * hdlr)
 784 {
 785   atom_hdlr_clear (hdlr);
 786   g_free (hdlr);
 787 }
 788
 789 static void
 790 atom_url_init (AtomURL * url)
 791 {
 792   guint8 flags[3] = { 0, 0, 1 };
 793
 794   atom_full_init (&url->header, FOURCC_url_, 0, 0, 0, flags);
 795   url->location = NULL;
 796 }
 797
 798 static void
 799 atom_url_free (AtomURL * url)
 800 {
 801   atom_full_clear (&url->header);
 802   if (url->location) {
 803     g_free (url->location);
 804     url->location = NULL;
 805   }
 806   g_free (url);
 807 }
 808
 809 static AtomURL *
 810 atom_url_new ()
 811 {
 812   AtomURL *url = g_new0 (AtomURL, 1);
 813
 814   atom_url_init (url);
 815   return url;
 816 }
 817
 818 static AtomFull *
 819 atom_alis_new ()
 820 {
 821   guint8 flags[3] = { 0, 0, 1 };
 822   AtomFull *alis = g_new0 (AtomFull, 1);
 823
 824   atom_full_init (alis, FOURCC_alis, 0, 0, 0, flags);
 825   return alis;
 826 }
 827
 828 static void
 829 atom_dref_init (AtomDREF * dref, AtomsContext * context)
 830 {
 831   guint8 flags[3] = { 0, 0, 0 };
 832
 833   atom_full_init (&dref->header, FOURCC_dref, 0, 0, 0, flags);
 834
 835   /* in either case, alis or url init arranges to set self-contained flag */
 836   if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
 837     /* alis dref for qt */
 838     AtomFull *alis = atom_alis_new ();
 839     dref->entries = g_list_append (dref->entries, alis);
 840   } else {
 841     /* url for iso spec, as 'alis' not specified there */
 842     AtomURL *url = atom_url_new ();
 843     dref->entries = g_list_append (dref->entries, url);
 844   }
 845 }
 846
 847 static void
 848 atom_dref_clear (AtomDREF * dref)
 849 {
 850   GList *walker;
 851
 852   atom_full_clear (&dref->header);
 853   walker = dref->entries;
 854   while (walker) {
 855     GList *aux = walker;
 856     Atom *atom = (Atom *) aux->data;
 857
 858     walker = g_list_next (walker);
 859     dref->entries = g_list_remove_link (dref->entries, aux);
 860     switch (atom->type) {
 861       case FOURCC_alis:
 862         atom_full_free ((AtomFull *) atom);
 863         break;
 864       case FOURCC_url_:
 865         atom_url_free ((AtomURL *) atom);
 866         break;
 867       default:
 868         /* we do nothing, better leak than crash */
 869         break;
 870     }
 871     g_list_free (aux);
 872   }
 873 }
 874
 875 static void
 876 atom_dinf_init (AtomDINF * dinf, AtomsContext * context)
 877 {
 878   atom_header_set (&dinf->header, FOURCC_dinf, 0, 0);
 879   atom_dref_init (&dinf->dref, context);
 880 }
 881
 882 static void
 883 atom_dinf_clear (AtomDINF * dinf)
 884 {
 885   atom_clear (&dinf->header);
 886   atom_dref_clear (&dinf->dref);
 887 }
 888
 889 static void
 890 atom_minf_init (AtomMINF * minf, AtomsContext * context)
 891 {
 892   atom_header_set (&minf->header, FOURCC_minf, 0, 0);
 893
 894   minf->vmhd = NULL;
 895   minf->smhd = NULL;
 896   minf->hmhd = NULL;
 897
 898   if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
 899     minf->hdlr = atom_hdlr_new ();
 900     minf->hdlr->component_type = FOURCC_dhlr;
 901     minf->hdlr->handler_type = FOURCC_alis;
 902   } else {
 903     minf->hdlr = NULL;
 904   }
 905   atom_dinf_init (&minf->dinf, context);
 906   atom_stbl_init (&minf->stbl);
 907 }
 908
 909 static void
 910 atom_minf_clear_handlers (AtomMINF * minf)
 911 {
 912   if (minf->vmhd) {
 913     atom_vmhd_free (minf->vmhd);
 914     minf->vmhd = NULL;
 915   }
 916   if (minf->smhd) {
 917     atom_smhd_free (minf->smhd);
 918     minf->smhd = NULL;
 919   }
 920   if (minf->hmhd) {
 921     atom_hmhd_free (minf->hmhd);
 922     minf->hmhd = NULL;
 923   }
 924 }
 925
 926 static void
 927 atom_minf_clear (AtomMINF * minf)
 928 {
 929   atom_clear (&minf->header);
 930   atom_minf_clear_handlers (minf);
 931   if (minf->hdlr) {
 932     atom_hdlr_free (minf->hdlr);
 933   }
 934   atom_dinf_clear (&minf->dinf);
 935   atom_stbl_clear (&minf->stbl);
 936 }
 937
 938 static void
 939 atom_mdhd_init (AtomMDHD * mdhd)
 940 {
 941   guint8 flags[3] = { 0, 0, 0 };
 942
 943   atom_full_init (&mdhd->header, FOURCC_mdhd, 0, 0, 0, flags);
 944   common_time_info_init (&mdhd->time_info);
 945   mdhd->language_code = 0;
 946   mdhd->quality = 0;
 947 }
 948
 949 static void
 950 atom_mdhd_clear (AtomMDHD * mdhd)
 951 {
 952   atom_full_clear (&mdhd->header);
 953 }
 954
 955 static void
 956 atom_mdia_init (AtomMDIA * mdia, AtomsContext * context)
 957 {
 958   atom_header_set (&mdia->header, FOURCC_mdia, 0, 0);
 959
 960   atom_mdhd_init (&mdia->mdhd);
 961   atom_hdlr_init (&mdia->hdlr);
 962   atom_minf_init (&mdia->minf, context);
 963 }
 964
 965 static void
 966 atom_mdia_clear (AtomMDIA * mdia)
 967 {
 968   atom_clear (&mdia->header);
 969   atom_mdhd_clear (&mdia->mdhd);
 970   atom_hdlr_clear (&mdia->hdlr);
 971   atom_minf_clear (&mdia->minf);
 972 }
 973
 974 static void
 975 atom_tkhd_init (AtomTKHD * tkhd, AtomsContext * context)
 976 {
 977   /*
 978    * flags info
 979    * 1 -> track enabled
 980    * 2 -> track in movie
 981    * 4 -> track in preview
 982    */
 983   guint8 flags[3] = { 0, 0, 7 };
 984
 985   atom_full_init (&tkhd->header, FOURCC_tkhd, 0, 0, 0, flags);
 986
 987   tkhd->creation_time = tkhd->modification_time = get_current_qt_time ();
 988   tkhd->duration = 0;
 989   tkhd->track_ID = 0;
 990   tkhd->reserved = 0;
 991
 992   tkhd->reserved2[0] = tkhd->reserved2[1] = 0;
 993   tkhd->layer = 0;
 994   tkhd->alternate_group = 0;
 995   tkhd->volume = 0;
 996   tkhd->reserved3 = 0;
 997   memset (tkhd->matrix, 0, sizeof (guint32) * 9);
 998   tkhd->matrix[0] = 1 << 16;
 999   tkhd->matrix[4] = 1 << 16;
1000   tkhd->matrix[8] = 16384 << 16;
1001   tkhd->width = 0;
1002   tkhd->height = 0;
1003 }
1004
1005 static void
1006 atom_tkhd_clear (AtomTKHD * tkhd)
1007 {
1008   atom_full_clear (&tkhd->header);
1009 }
1010
1011 static void
1012 atom_trak_init (AtomTRAK * trak, AtomsContext * context)
1013 {
1014   atom_header_set (&trak->header, FOURCC_trak, 0, 0);
1015
1016   atom_tkhd_init (&trak->tkhd, context);
1017   trak->edts = NULL;
1018   atom_mdia_init (&trak->mdia, context);
1019 }
1020
1021 AtomTRAK *
1022 atom_trak_new (AtomsContext * context)
1023 {
1024   AtomTRAK *trak = g_new0 (AtomTRAK, 1);
1025
1026   atom_trak_init (trak, context);
1027   return trak;
1028 }
1029
1030 static void
1031 atom_trak_free (AtomTRAK * trak)
1032 {
1033   atom_clear (&trak->header);
1034   atom_tkhd_clear (&trak->tkhd);
1035   if (trak->edts)
1036     atom_edts_free (trak->edts);
1037   atom_mdia_clear (&trak->mdia);
1038   g_free (trak);
1039 }
1040
1041 static void
1042 atom_ilst_init (AtomILST * ilst)
1043 {
1044   atom_header_set (&ilst->header, FOURCC_ilst, 0, 0);
1045   ilst->entries = NULL;
1046 }
1047
1048 static AtomILST *
1049 atom_ilst_new ()
1050 {
1051   AtomILST *ilst = g_new0 (AtomILST, 1);
1052
1053   atom_ilst_init (ilst);
1054   return ilst;
1055 }
1056
1057 static void
1058 atom_ilst_free (AtomILST * ilst)
1059 {
1060   if (ilst->entries)
1061     atom_info_list_free (ilst->entries);
1062   atom_clear (&ilst->header);
1063   g_free (ilst);
1064 }
1065
1066 static void
1067 atom_meta_init (AtomMETA * meta)
1068 {
1069   guint8 flags[3] = { 0, 0, 0 };
1070
1071   atom_full_init (&meta->header, FOURCC_meta, 0, 0, 0, flags);
1072   atom_hdlr_init (&meta->hdlr);
1073   /* FIXME (ISOM says this is always 0) */
1074   meta->hdlr.component_type = FOURCC_mhlr;
1075   meta->hdlr.handler_type = FOURCC_mdir;
1076   meta->ilst = NULL;
1077 }
1078
1079 static AtomMETA *
1080 atom_meta_new ()
1081 {
1082   AtomMETA *meta = g_new0 (AtomMETA, 1);
1083
1084   atom_meta_init (meta);
1085   return meta;
1086 }
1087
1088 static void
1089 atom_meta_free (AtomMETA * meta)
1090 {
1091   atom_full_clear (&meta->header);
1092   atom_hdlr_clear (&meta->hdlr);
1093   if (meta->ilst)
1094     atom_ilst_free (meta->ilst);
1095   meta->ilst = NULL;
1096   g_free (meta);
1097 }
1098
1099 static void
1100 atom_udta_init (AtomUDTA * udta)
1101 {
1102   atom_header_set (&udta->header, FOURCC_udta, 0, 0);
1103   udta->meta = NULL;
1104 }
1105
1106 static AtomUDTA *
1107 atom_udta_new ()
1108 {
1109   AtomUDTA *udta = g_new0 (AtomUDTA, 1);
1110
1111   atom_udta_init (udta);
1112   return udta;
1113 }
1114
1115 static void
1116 atom_udta_free (AtomUDTA * udta)
1117 {
1118   atom_clear (&udta->header);
1119   if (udta->meta)
1120     atom_meta_free (udta->meta);
1121   udta->meta = NULL;
1122   if (udta->entries)
1123     atom_info_list_free (udta->entries);
1124   g_free (udta);
1125 }
1126
1127 static void
1128 atom_tag_data_init (AtomTagData * data)
1129 {
1130   guint8 flags[] = { 0, 0, 0 };
1131
1132   atom_full_init (&data->header, FOURCC_data, 0, 0, 0, flags);
1133 }
1134
1135 static void
1136 atom_tag_data_clear (AtomTagData * data)
1137 {
1138   atom_full_clear (&data->header);
1139   g_free (data->data);
1140   data->datalen = 0;
1141 }
1142
1143 /*
1144  * Fourcc is the tag fourcc
1145  * flags will be truncated to 24bits
1146  */
1147 static AtomTag *
1148 atom_tag_new (guint32 fourcc, guint32 flags_as_uint)
1149 {
1150   AtomTag *tag = g_new0 (AtomTag, 1);
1151
1152   tag->header.type = fourcc;
1153   atom_tag_data_init (&tag->data);
1154   tag->data.header.flags[2] = flags_as_uint & 0xFF;
1155   tag->data.header.flags[1] = (flags_as_uint & 0xFF00) >> 8;
1156   tag->data.header.flags[0] = (flags_as_uint & 0xFF0000) >> 16;
1157   return tag;
1158 }
1159
1160 static void
1161 atom_tag_free (AtomTag * tag)
1162 {
1163   atom_clear (&tag->header);
1164   atom_tag_data_clear (&tag->data);
1165   g_free (tag);
1166 }
1167
1168 static void
1169 atom_mvhd_init (AtomMVHD * mvhd)
1170 {
1171   guint8 flags[3] = { 0, 0, 0 };
1172
1173   atom_full_init (&(mvhd->header), FOURCC_mvhd, sizeof (AtomMVHD), 0, 0, flags);
1174
1175   common_time_info_init (&mvhd->time_info);
1176
1177   mvhd->prefered_rate = 1 << 16;
1178   mvhd->volume = 1 << 8;
1179   mvhd->reserved3 = 0;
1180   memset (mvhd->reserved4, 0, sizeof (guint32[2]));
1181
1182   memset (mvhd->matrix, 0, sizeof (guint32[9]));
1183   mvhd->matrix[0] = 1 << 16;
1184   mvhd->matrix[4] = 1 << 16;
1185   mvhd->matrix[8] = 16384 << 16;
1186
1187   mvhd->preview_time = 0;
1188   mvhd->preview_duration = 0;
1189   mvhd->poster_time = 0;
1190   mvhd->selection_time = 0;
1191   mvhd->selection_duration = 0;
1192   mvhd->current_time = 0;
1193
1194   mvhd->next_track_id = 1;
1195 }
1196
1197 static void
1198 atom_mvhd_clear (AtomMVHD * mvhd)
1199 {
1200   atom_full_clear (&mvhd->header);
1201 }
1202
1203 static void
1204 atom_moov_init (AtomMOOV * moov, AtomsContext * context)
1205 {
1206   atom_header_set (&(moov->header), FOURCC_moov, 0, 0);
1207   atom_mvhd_init (&(moov->mvhd));
1208   moov->udta = NULL;
1209   moov->traks = NULL;
1210   moov->context = *context;
1211 }
1212
1213 AtomMOOV *
1214 atom_moov_new (AtomsContext * context)
1215 {
1216   AtomMOOV *moov = g_new0 (AtomMOOV, 1);
1217
1218   atom_moov_init (moov, context);
1219   return moov;
1220 }
1221
1222 void
1223 atom_moov_free (AtomMOOV * moov)
1224 {
1225   GList *walker;
1226
1227   atom_clear (&moov->header);
1228   atom_mvhd_clear (&moov->mvhd);
1229
1230   walker = moov->traks;
1231   while (walker) {
1232     atom_trak_free ((AtomTRAK *) walker->data);
1233     walker = g_list_next (walker);
1234   }
1235   g_list_free (moov->traks);
1236   moov->traks = NULL;
1237
1238   if (moov->udta) {
1239     atom_udta_free (moov->udta);
1240     moov->udta = NULL;
1241   }
1242
1243   g_free (moov);
1244 }
1245
1246 /* -- end of init / free -- */
1247
1248 /* -- copy data functions -- */
1249
1250 static guint8
1251 atom_full_get_version (AtomFull * full)
1252 {
1253   return full->version;
1254 }
1255
1256 static guint64
1257 common_time_info_copy_data (TimeInfo * ti, gboolean trunc_to_32,
1258     guint8 ** buffer, guint64 * size, guint64 * offset)
1259 {
1260   guint64 original_offset = *offset;
1261
1262   if (trunc_to_32) {
1263     prop_copy_uint32 ((guint32) ti->creation_time, buffer, size, offset);
1264     prop_copy_uint32 ((guint32) ti->modification_time, buffer, size, offset);
1265     prop_copy_uint32 (ti->timescale, buffer, size, offset);
1266     prop_copy_uint32 ((guint32) ti->duration, buffer, size, offset);
1267   } else {
1268     prop_copy_uint64 (ti->creation_time, buffer, size, offset);
1269     prop_copy_uint64 (ti->modification_time, buffer, size, offset);
1270     prop_copy_uint32 (ti->timescale, buffer, size, offset);
1271     prop_copy_uint64 (ti->duration, buffer, size, offset);
1272   }
1273   return *offset - original_offset;
1274 }
1275
1276 static void
1277 atom_write_size (guint8 ** buffer, guint64 * size, guint64 * offset,
1278     guint64 atom_pos)
1279 {
1280   /* this only works for non-extended atom size, which is OK
1281    * (though it could be made to do mem_move, etc and write extended size) */
1282   prop_copy_uint32 (*offset - atom_pos, buffer, size, &atom_pos);
1283 }
1284
1285 guint64
1286 atom_copy_data (Atom * atom, guint8 ** buffer, guint64 * size, guint64 * offset)
1287 {
1288   guint64 original_offset = *offset;
1289
1290   /* copies type and size */
1291   prop_copy_uint32 (atom->size, buffer, size, offset);
1292   prop_copy_fourcc (atom->type, buffer, size, offset);
1293
1294   /* extended size needed */
1295   if (atom->size == 1) {
1296     /* really should not happen other than with mdat atom;
1297      * would be a problem for size (re)write code, not to mention memory */
1298     g_return_val_if_fail (atom->type == FOURCC_mdat, 0);
1299     prop_copy_uint64 (atom->extended_size, buffer, size, offset);
1300   }
1301
1302   return *offset - original_offset;
1303 }
1304
1305 static guint64
1306 atom_full_copy_data (AtomFull * atom, guint8 ** buffer, guint64 * size,
1307     guint64 * offset)
1308 {
1309   guint64 original_offset = *offset;
1310
1311   if (!atom_copy_data (&atom->header, buffer, size, offset)) {
1312     return 0;
1313   }
1314
1315   prop_copy_uint8 (atom->version, buffer, size, offset);
1316   prop_copy_uint8_array (atom->flags, 3, buffer, size, offset);
1317
1318   atom_write_size (buffer, size, offset, original_offset);
1319   return *offset - original_offset;
1320 }
1321
1322 static guint64
1323 atom_info_list_copy_data (GList * ai, guint8 ** buffer, guint64 * size,
1324     guint64 * offset)
1325 {
1326   guint64 original_offset = *offset;
1327
1328   while (ai) {
1329     AtomInfo *info = (AtomInfo *) ai->data;
1330
1331     if (!info->copy_data_func (info->atom, buffer, size, offset)) {
1332       return 0;
1333     }
1334     ai = g_list_next (ai);
1335   }
1336
1337   return *offset - original_offset;
1338 }
1339
1340 static guint64
1341 atom_data_copy_data (AtomData * data, guint8 ** buffer, guint64 * size,
1342     guint64 * offset)
1343 {
1344   guint64 original_offset = *offset;
1345
1346   if (!atom_copy_data (&data->header, buffer, size, offset)) {
1347     return 0;
1348   }
1349   if (data->datalen)
1350     prop_copy_uint8_array (data->data, data->datalen, buffer, size, offset);
1351
1352   atom_write_size (buffer, size, offset, original_offset);
1353   return *offset - original_offset;
1354 }
1355
1356 guint64
1357 atom_ftyp_copy_data (AtomFTYP * ftyp, guint8 ** buffer, guint64 * size,
1358     guint64 * offset)
1359 {
1360   guint64 original_offset = *offset;
1361
1362   if (!atom_copy_data (&ftyp->header, buffer, size, offset)) {
1363     return 0;
1364   }
1365   prop_copy_fourcc (ftyp->major_brand, buffer, size, offset);
1366   prop_copy_uint32 (ftyp->version, buffer, size, offset);
1367
1368   prop_copy_fourcc_array (ftyp->compatible_brands, ftyp->compatible_brands_size,
1369       buffer, size, offset);
1370
1371   atom_write_size (buffer, size, offset, original_offset);
1372   return *offset - original_offset;
1373 }
1374
1375 static guint64
1376 atom_mvhd_copy_data (AtomMVHD * atom, guint8 ** buffer, guint64 * size,
1377     guint64 * offset)
1378 {
1379   guint8 version;
1380   guint64 original_offset = *offset;
1381
1382   if (!atom_full_copy_data (&(atom->header), buffer, size, offset)) {
1383     return 0;
1384   }
1385
1386   version = atom_full_get_version (&(atom->header));
1387   if (version == 0) {
1388     common_time_info_copy_data (&atom->time_info, TRUE, buffer, size, offset);
1389   } else if (version == 1) {
1390     common_time_info_copy_data (&atom->time_info, FALSE, buffer, size, offset);
1391   } else {
1392     *offset = original_offset;
1393     return 0;
1394   }
1395
1396   prop_copy_uint32 (atom->prefered_rate, buffer, size, offset);
1397   prop_copy_uint16 (atom->volume, buffer, size, offset);
1398   prop_copy_uint16 (atom->reserved3, buffer, size, offset);
1399   prop_copy_uint32_array (atom->reserved4, 2, buffer, size, offset);
1400   prop_copy_uint32_array (atom->matrix, 9, buffer, size, offset);
1401   prop_copy_uint32 (atom->preview_time, buffer, size, offset);
1402   prop_copy_uint32 (atom->preview_duration, buffer, size, offset);
1403   prop_copy_uint32 (atom->poster_time, buffer, size, offset);
1404   prop_copy_uint32 (atom->selection_time, buffer, size, offset);
1405   prop_copy_uint32 (atom->selection_duration, buffer, size, offset);
1406   prop_copy_uint32 (atom->current_time, buffer, size, offset);
1407
1408   prop_copy_uint32 (atom->next_track_id, buffer, size, offset);
1409
1410   atom_write_size (buffer, size, offset, original_offset);
1411   return *offset - original_offset;
1412 }
1413
1414 static guint64
1415 atom_tkhd_copy_data (AtomTKHD * tkhd, guint8 ** buffer, guint64 * size,
1416     guint64 * offset)
1417 {
1418   guint64 original_offset = *offset;
1419
1420   if (!atom_full_copy_data (&tkhd->header, buffer, size, offset)) {
1421     return 0;
1422   }
1423
1424   if (atom_full_get_version (&tkhd->header) == 0) {
1425     prop_copy_uint32 ((guint32) tkhd->creation_time, buffer, size, offset);
1426     prop_copy_uint32 ((guint32) tkhd->modification_time, buffer, size, offset);
1427     prop_copy_uint32 (tkhd->track_ID, buffer, size, offset);
1428     prop_copy_uint32 (tkhd->reserved, buffer, size, offset);
1429     prop_copy_uint32 ((guint32) tkhd->duration, buffer, size, offset);
1430   } else {
1431     prop_copy_uint64 (tkhd->creation_time, buffer, size, offset);
1432     prop_copy_uint64 (tkhd->modification_time, buffer, size, offset);
1433     prop_copy_uint32 (tkhd->track_ID, buffer, size, offset);
1434     prop_copy_uint32 (tkhd->reserved, buffer, size, offset);
1435     prop_copy_uint64 (tkhd->duration, buffer, size, offset);
1436   }
1437
1438   prop_copy_uint32_array (tkhd->reserved2, 2, buffer, size, offset);
1439   prop_copy_uint16 (tkhd->layer, buffer, size, offset);
1440   prop_copy_uint16 (tkhd->alternate_group, buffer, size, offset);
1441   prop_copy_uint16 (tkhd->volume, buffer, size, offset);
1442   prop_copy_uint16 (tkhd->reserved3, buffer, size, offset);
1443   prop_copy_uint32_array (tkhd->matrix, 9, buffer, size, offset);
1444
1445   prop_copy_uint32 (tkhd->width, buffer, size, offset);
1446   prop_copy_uint32 (tkhd->height, buffer, size, offset);
1447
1448   atom_write_size (buffer, size, offset, original_offset);
1449   return *offset - original_offset;
1450 }
1451
1452 static guint64
1453 atom_hdlr_copy_data (AtomHDLR * hdlr, guint8 ** buffer, guint64 * size,
1454     guint64 * offset)
1455 {
1456   guint64 original_offset = *offset;
1457
1458   if (!atom_full_copy_data (&hdlr->header, buffer, size, offset)) {
1459     return 0;
1460   }
1461
1462   prop_copy_fourcc (hdlr->component_type, buffer, size, offset);
1463   prop_copy_fourcc (hdlr->handler_type, buffer, size, offset);
1464   prop_copy_fourcc (hdlr->manufacturer, buffer, size, offset);
1465   prop_copy_uint32 (hdlr->flags, buffer, size, offset);
1466   prop_copy_uint32 (hdlr->flags_mask, buffer, size, offset);
1467
1468   prop_copy_null_terminated_string (hdlr->name, buffer, size, offset);
1469
1470   atom_write_size (buffer, size, offset, original_offset);
1471   return *offset - original_offset;
1472 }
1473
1474 static guint64
1475 atom_vmhd_copy_data (AtomVMHD * vmhd, guint8 ** buffer, guint64 * size,
1476     guint64 * offset)
1477 {
1478   guint64 original_offset = *offset;
1479
1480   if (!atom_full_copy_data (&vmhd->header, buffer, size, offset)) {
1481     return 0;
1482   }
1483   prop_copy_uint16 (vmhd->graphics_mode, buffer, size, offset);
1484   prop_copy_uint16_array (vmhd->opcolor, 3, buffer, size, offset);
1485
1486   atom_write_size (buffer, size, offset, original_offset);
1487   return original_offset - *offset;
1488 }
1489
1490 static guint64
1491 atom_smhd_copy_data (AtomSMHD * smhd, guint8 ** buffer, guint64 * size,
1492     guint64 * offset)
1493 {
1494   guint64 original_offset = *offset;
1495
1496   if (!atom_full_copy_data (&smhd->header, buffer, size, offset)) {
1497     return 0;
1498   }
1499   prop_copy_uint16 (smhd->balance, buffer, size, offset);
1500   prop_copy_uint16 (smhd->reserved, buffer, size, offset);
1501
1502   atom_write_size (buffer, size, offset, original_offset);
1503   return original_offset - *offset;
1504 }
1505
1506 static guint64
1507 atom_hmhd_copy_data (AtomHMHD * hmhd, guint8 ** buffer, guint64 * size,
1508     guint64 * offset)
1509 {
1510   guint64 original_offset = *offset;
1511
1512   if (!atom_full_copy_data (&hmhd->header, buffer, size, offset)) {
1513     return 0;
1514   }
1515   prop_copy_uint16 (hmhd->max_pdu_size, buffer, size, offset);
1516   prop_copy_uint16 (hmhd->avg_pdu_size, buffer, size, offset);
1517   prop_copy_uint32 (hmhd->max_bitrate, buffer, size, offset);
1518   prop_copy_uint32 (hmhd->avg_bitrate, buffer, size, offset);
1519   prop_copy_uint32 (hmhd->sliding_avg_bitrate, buffer, size, offset);
1520
1521   atom_write_size (buffer, size, offset, original_offset);
1522   return original_offset - *offset;
1523 }
1524
1525 static gboolean
1526 atom_url_same_file_flag (AtomURL * url)
1527 {
1528   return (url->header.flags[2] & 0x1) == 1;
1529 }
1530
1531 static guint64
1532 atom_url_copy_data (AtomURL * url, guint8 ** buffer, guint64 * size,
1533     guint64 * offset)
1534 {
1535   guint64 original_offset = *offset;
1536
1537   if (!atom_full_copy_data (&url->header, buffer, size, offset)) {
1538     return 0;
1539   }
1540
1541   if (!atom_url_same_file_flag (url)) {
1542     prop_copy_null_terminated_string (url->location, buffer, size, offset);
1543   }
1544
1545   atom_write_size (buffer, size, offset, original_offset);
1546   return original_offset - *offset;
1547 }
1548
1549 static guint64
1550 atom_stts_copy_data (AtomSTTS * stts, guint8 ** buffer, guint64 * size,
1551     guint64 * offset)
1552 {
1553   guint64 original_offset = *offset;
1554   GList *walker;
1555
1556   if (!atom_full_copy_data (&stts->header, buffer, size, offset)) {
1557     return 0;
1558   }
1559
1560   prop_copy_uint32 (stts->n_entries, buffer, size, offset);
1561   /* minimize realloc */
1562   prop_copy_ensure_buffer (buffer, size, offset, 8 * stts->n_entries);
1563   for (walker = g_list_last (stts->entries); walker != NULL;
1564       walker = g_list_previous (walker)) {
1565     STTSEntry *entry = (STTSEntry *) walker->data;
1566
1567     prop_copy_uint32 (entry->sample_count, buffer, size, offset);
1568     prop_copy_int32 (entry->sample_delta, buffer, size, offset);
1569   }
1570
1571   atom_write_size (buffer, size, offset, original_offset);
1572   return *offset - original_offset;
1573 }
1574
1575 static guint64
1576 atom_sample_entry_copy_data (SampleTableEntry * se, guint8 ** buffer,
1577     guint64 * size, guint64 * offset)
1578 {
1579   guint64 original_offset = *offset;
1580
1581   if (!atom_copy_data (&se->header, buffer, size, offset)) {
1582     return 0;
1583   }
1584
1585   prop_copy_uint8_array (se->reserved, 6, buffer, size, offset);
1586   prop_copy_uint16 (se->data_reference_index, buffer, size, offset);
1587
1588   return *offset - original_offset;
1589 }
1590
1591 static guint64
1592 atom_esds_copy_data (AtomESDS * esds, guint8 ** buffer, guint64 * size,
1593     guint64 * offset)
1594 {
1595   guint64 original_offset = *offset;
1596
1597   if (!atom_full_copy_data (&esds->header, buffer, size, offset)) {
1598     return 0;
1599   }
1600   if (!desc_es_descriptor_copy_data (&esds->es, buffer, size, offset)) {
1601     return 0;
1602   }
1603
1604   atom_write_size (buffer, size, offset, original_offset);
1605   return *offset - original_offset;
1606 }
1607
1608 static guint64
1609 atom_frma_copy_data (AtomFRMA * frma, guint8 ** buffer,
1610     guint64 * size, guint64 * offset)
1611 {
1612   guint64 original_offset = *offset;
1613
1614   if (!atom_copy_data (&(frma->header), buffer, size, offset))
1615     return 0;
1616
1617   prop_copy_fourcc (frma->media_type, buffer, size, offset);
1618
1619   atom_write_size (buffer, size, offset, original_offset);
1620   return *offset - original_offset;
1621 }
1622
1623 static guint64
1624 atom_mp4s_copy_data (SampleTableEntryMP4S * mp4s, guint8 ** buffer,
1625     guint64 * size, guint64 * offset)
1626 {
1627   guint64 original_offset = *offset;
1628
1629   if (!atom_sample_entry_copy_data (&mp4s->se, buffer, size, offset)) {
1630     return 0;
1631   }
1632   if (!atom_esds_copy_data (&mp4s->es, buffer, size, offset)) {
1633     return 0;
1634   }
1635
1636   atom_write_size (buffer, size, offset, original_offset);
1637   return *offset - original_offset;
1638 }
1639
1640 static guint64
1641 atom_hint_sample_entry_copy_data (AtomHintSampleEntry * hse, guint8 ** buffer,
1642     guint64 * size, guint64 * offset)
1643 {
1644   guint64 original_offset = *offset;
1645
1646   if (!atom_sample_entry_copy_data (&hse->se, buffer, size, offset)) {
1647     return 0;
1648   }
1649
1650   prop_copy_uint32 (hse->size, buffer, size, offset);
1651   prop_copy_uint8_array (hse->data, hse->size, buffer, size, offset);
1652
1653   atom_write_size (buffer, size, offset, original_offset);
1654   return *offset - original_offset;
1655 }
1656
1657 static guint64
1658 sample_entry_mp4a_copy_data (SampleTableEntryMP4A * mp4a, guint8 ** buffer,
1659     guint64 * size, guint64 * offset)
1660 {
1661   guint64 original_offset = *offset;
1662
1663   if (!atom_sample_entry_copy_data (&mp4a->se, buffer, size, offset)) {
1664     return 0;
1665   }
1666
1667   prop_copy_uint16 (mp4a->version, buffer, size, offset);
1668   prop_copy_uint16 (mp4a->revision_level, buffer, size, offset);
1669   prop_copy_uint32 (mp4a->vendor, buffer, size, offset);
1670   prop_copy_uint16 (mp4a->channels, buffer, size, offset);
1671   prop_copy_uint16 (mp4a->sample_size, buffer, size, offset);
1672   prop_copy_uint16 (mp4a->compression_id, buffer, size, offset);
1673   prop_copy_uint16 (mp4a->packet_size, buffer, size, offset);
1674   prop_copy_uint32 (mp4a->sample_rate, buffer, size, offset);
1675
1676   /* this should always be 0 for mp4 flavor */
1677   if (mp4a->version == 1) {
1678     prop_copy_uint32 (mp4a->samples_per_packet, buffer, size, offset);
1679     prop_copy_uint32 (mp4a->bytes_per_packet, buffer, size, offset);
1680     prop_copy_uint32 (mp4a->bytes_per_frame, buffer, size, offset);
1681     prop_copy_uint32 (mp4a->bytes_per_sample, buffer, size, offset);
1682   }
1683
1684   if (mp4a->extension_atoms) {
1685     if (!atom_info_list_copy_data (mp4a->extension_atoms, buffer, size, offset))
1686       return 0;
1687   }
1688
1689   atom_write_size (buffer, size, offset, original_offset);
1690   return *offset - original_offset;
1691 }
1692
1693 static guint64
1694 sample_entry_mp4v_copy_data (SampleTableEntryMP4V * mp4v, guint8 ** buffer,
1695     guint64 * size, guint64 * offset)
1696 {
1697   guint64 original_offset = *offset;
1698
1699   if (!atom_sample_entry_copy_data (&mp4v->se, buffer, size, offset)) {
1700     return 0;
1701   }
1702
1703   prop_copy_uint16 (mp4v->version, buffer, size, offset);
1704   prop_copy_uint16 (mp4v->revision_level, buffer, size, offset);
1705   prop_copy_fourcc (mp4v->vendor, buffer, size, offset);
1706   prop_copy_uint32 (mp4v->temporal_quality, buffer, size, offset);
1707   prop_copy_uint32 (mp4v->spatial_quality, buffer, size, offset);
1708
1709   prop_copy_uint16 (mp4v->width, buffer, size, offset);
1710   prop_copy_uint16 (mp4v->height, buffer, size, offset);
1711
1712   prop_copy_uint32 (mp4v->horizontal_resolution, buffer, size, offset);
1713   prop_copy_uint32 (mp4v->vertical_resolution, buffer, size, offset);
1714   prop_copy_uint32 (mp4v->datasize, buffer, size, offset);
1715
1716   prop_copy_uint16 (mp4v->frame_count, buffer, size, offset);
1717
1718   prop_copy_fixed_size_string ((guint8 *) mp4v->compressor, 32, buffer, size,
1719       offset);
1720
1721   prop_copy_uint16 (mp4v->depth, buffer, size, offset);
1722   prop_copy_uint16 (mp4v->color_table_id, buffer, size, offset);
1723
1724   /* extra atoms */
1725   if (mp4v->extension_atoms &&
1726       !atom_info_list_copy_data (mp4v->extension_atoms, buffer, size, offset))
1727     return 0;
1728
1729   atom_write_size (buffer, size, offset, original_offset);
1730   return *offset - original_offset;
1731 }
1732
1733 static guint64
1734 atom_stsz_copy_data (AtomSTSZ * stsz, guint8 ** buffer, guint64 * size,
1735     guint64 * offset)
1736 {
1737   guint64 original_offset = *offset;
1738   GList *walker;
1739
1740   if (!atom_full_copy_data (&stsz->header, buffer, size, offset)) {
1741     return 0;
1742   }
1743
1744   prop_copy_uint32 (stsz->sample_size, buffer, size, offset);
1745   prop_copy_uint32 (stsz->table_size, buffer, size, offset);
1746   /* minimize realloc */
1747   prop_copy_ensure_buffer (buffer, size, offset, 4 * stsz->table_size);
1748   if (stsz->sample_size == 0) {
1749     for (walker = g_list_last (stsz->entries); walker != NULL;
1750         walker = g_list_previous (walker)) {
1751       prop_copy_uint32 ((guint32) GPOINTER_TO_UINT (walker->data), buffer, size,
1752           offset);
1753     }
1754   }
1755
1756   atom_write_size (buffer, size, offset, original_offset);
1757   return *offset - original_offset;
1758 }
1759
1760 static guint64
1761 atom_stsc_copy_data (AtomSTSC * stsc, guint8 ** buffer, guint64 * size,
1762     guint64 * offset)
1763 {
1764   guint64 original_offset = *offset;
1765   GList *walker;
1766
1767   if (!atom_full_copy_data (&stsc->header, buffer, size, offset)) {
1768     return 0;
1769   }
1770
1771   prop_copy_uint32 (stsc->n_entries, buffer, size, offset);
1772   /* minimize realloc */
1773   prop_copy_ensure_buffer (buffer, size, offset, 12 * stsc->n_entries);
1774
1775   for (walker = g_list_last (stsc->entries); walker != NULL;
1776       walker = g_list_previous (walker)) {
1777     STSCEntry *entry = (STSCEntry *) walker->data;
1778
1779     prop_copy_uint32 (entry->first_chunk, buffer, size, offset);
1780     prop_copy_uint32 (entry->samples_per_chunk, buffer, size, offset);
1781     prop_copy_uint32 (entry->sample_description_index, buffer, size, offset);
1782   }
1783
1784   atom_write_size (buffer, size, offset, original_offset);
1785   return *offset - original_offset;
1786 }
1787
1788 static guint64
1789 atom_ctts_copy_data (AtomCTTS * ctts, guint8 ** buffer, guint64 * size,
1790     guint64 * offset)
1791 {
1792   guint64 original_offset = *offset;
1793   GList *walker;
1794
1795   if (!atom_full_copy_data (&ctts->header, buffer, size, offset)) {
1796     return 0;
1797   }
1798
1799   prop_copy_uint32 (ctts->n_entries, buffer, size, offset);
1800   /* minimize realloc */
1801   prop_copy_ensure_buffer (buffer, size, offset, 8 * ctts->n_entries);
1802   for (walker = g_list_last (ctts->entries); walker != NULL;
1803       walker = g_list_previous (walker)) {
1804     CTTSEntry *entry = (CTTSEntry *) walker->data;
1805
1806     prop_copy_uint32 (entry->samplecount, buffer, size, offset);
1807     prop_copy_uint32 (entry->sampleoffset, buffer, size, offset);
1808   }
1809
1810   atom_write_size (buffer, size, offset, original_offset);
1811   return *offset - original_offset;
1812 }
1813
1814 static guint64
1815 atom_stco64_copy_data (AtomSTCO64 * stco64, guint8 ** buffer, guint64 * size,
1816     guint64 * offset)
1817 {
1818   guint64 original_offset = *offset;
1819   GList *walker;
1820   gboolean trunc_to_32 = stco64->header.header.type == FOURCC_stco;
1821
1822   if (!atom_full_copy_data (&stco64->header, buffer, size, offset)) {
1823     return 0;
1824   }
1825
1826   prop_copy_uint32 (stco64->n_entries, buffer, size, offset);
1827
1828   /* minimize realloc */
1829   prop_copy_ensure_buffer (buffer, size, offset, 8 * stco64->n_entries);
1830   for (walker = g_list_last (stco64->entries); walker != NULL;
1831       walker = g_list_previous (walker)) {
1832     guint64 *value = (guint64 *) walker->data;
1833
1834     if (trunc_to_32) {
1835       prop_copy_uint32 ((guint32) * value, buffer, size, offset);
1836     } else {
1837       prop_copy_uint64 (*value, buffer, size, offset);
1838     }
1839   }
1840
1841   atom_write_size (buffer, size, offset, original_offset);
1842   return *offset - original_offset;
1843 }
1844
1845 static guint64
1846 atom_stss_copy_data (AtomSTSS * stss, guint8 ** buffer, guint64 * size,
1847     guint64 * offset)
1848 {
1849   guint64 original_offset = *offset;
1850   GList *walker;
1851
1852   if (stss->entries == NULL) {
1853     /* FIXME not needing this atom might be confused with error while copying */
1854     return 0;
1855   }
1856
1857   if (!atom_full_copy_data (&stss->header, buffer, size, offset)) {
1858     return 0;
1859   }
1860
1861   prop_copy_uint32 (stss->n_entries, buffer, size, offset);
1862   /* minimize realloc */
1863   prop_copy_ensure_buffer (buffer, size, offset, 4 * stss->n_entries);
1864   for (walker = g_list_last (stss->entries); walker != NULL;
1865       walker = g_list_previous (walker)) {
1866     prop_copy_uint32 ((guint32) GPOINTER_TO_UINT (walker->data), buffer, size,
1867         offset);
1868   }
1869
1870   atom_write_size (buffer, size, offset, original_offset);
1871   return *offset - original_offset;
1872 }
1873
1874 static guint64
1875 atom_stsd_copy_data (AtomSTSD * stsd, guint8 ** buffer, guint64 * size,
1876     guint64 * offset)
1877 {
1878   guint64 original_offset = *offset;
1879   GList *walker;
1880
1881   if (!atom_full_copy_data (&stsd->header, buffer, size, offset)) {
1882     return 0;
1883   }
1884
1885   prop_copy_uint32 (stsd->n_entries, buffer, size, offset);
1886
1887   for (walker = g_list_last (stsd->entries); walker != NULL;
1888       walker = g_list_previous (walker)) {
1889     SampleTableEntry *se = (SampleTableEntry *) walker->data;
1890
1891     switch (((Atom *) walker->data)->type) {
1892       case FOURCC_mp4a:
1893         if (!sample_entry_mp4a_copy_data ((SampleTableEntryMP4A *) walker->data,
1894                 buffer, size, offset)) {
1895           return 0;
1896         }
1897         break;
1898       case FOURCC_mp4s:
1899         if (!atom_mp4s_copy_data ((SampleTableEntryMP4S *) walker->data,
1900                 buffer, size, offset)) {
1901           return 0;
1902         }
1903         break;
1904       case FOURCC_mp4v:
1905         if (!sample_entry_mp4v_copy_data ((SampleTableEntryMP4V *) walker->data,
1906                 buffer, size, offset)) {
1907           return 0;
1908         }
1909         break;
1910       default:
1911         if (se->kind == VIDEO) {
1912           size += sample_entry_mp4v_copy_data ((SampleTableEntryMP4V *)
1913               walker->data, buffer, size, offset);
1914         } else if (se->kind == AUDIO) {
1915           size += sample_entry_mp4a_copy_data ((SampleTableEntryMP4A *)
1916               walker->data, buffer, size, offset);
1917         } else {
1918           if (!atom_hint_sample_entry_copy_data (
1919                   (AtomHintSampleEntry *) walker->data, buffer, size, offset)) {
1920             return 0;
1921           }
1922         }
1923         break;
1924     }
1925   }
1926
1927   atom_write_size (buffer, size, offset, original_offset);
1928   return *offset - original_offset;
1929 }
1930
1931 static guint64
1932 atom_stbl_copy_data (AtomSTBL * stbl, guint8 ** buffer, guint64 * size,
1933     guint64 * offset)
1934 {
1935   guint64 original_offset = *offset;
1936
1937   if (!atom_copy_data (&stbl->header, buffer, size, offset)) {
1938     return 0;
1939   }
1940
1941   if (!atom_stsd_copy_data (&stbl->stsd, buffer, size, offset)) {
1942     return 0;
1943   }
1944   if (!atom_stts_copy_data (&stbl->stts, buffer, size, offset)) {
1945     return 0;
1946   }
1947   /* this atom is optional, so let's check if we need it
1948    * (to avoid false error) */
1949   if (stbl->stss.entries) {
1950     if (!atom_stss_copy_data (&stbl->stss, buffer, size, offset)) {
1951       return 0;
1952     }
1953   }
1954
1955   if (!atom_stsc_copy_data (&stbl->stsc, buffer, size, offset)) {
1956     return 0;
1957   }
1958   if (!atom_stsz_copy_data (&stbl->stsz, buffer, size, offset)) {
1959     return 0;
1960   }
1961   if (stbl->ctts) {
1962     if (!atom_ctts_copy_data (stbl->ctts, buffer, size, offset)) {
1963       return 0;
1964     }
1965   }
1966   if (!atom_stco64_copy_data (&stbl->stco64, buffer, size, offset)) {
1967     return 0;
1968   }
1969
1970   atom_write_size (buffer, size, offset, original_offset);
1971   return original_offset - *offset;
1972 }
1973
1974
1975 static guint64
1976 atom_dref_copy_data (AtomDREF * dref, guint8 ** buffer, guint64 * size,
1977     guint64 * offset)
1978 {
1979   guint64 original_offset = *offset;
1980   GList *walker;
1981
1982   if (!atom_full_copy_data (&dref->header, buffer, size, offset)) {
1983     return 0;
1984   }
1985
1986   prop_copy_uint32 (g_list_length (dref->entries), buffer, size, offset);
1987
1988   walker = dref->entries;
1989   while (walker != NULL) {
1990     Atom *atom = (Atom *) walker->data;
1991
1992     if (atom->type == FOURCC_url_) {
1993       atom_url_copy_data ((AtomURL *) atom, buffer, size, offset);
1994     } else if (atom->type == FOURCC_alis) {
1995       atom_full_copy_data ((AtomFull *) atom, buffer, size, offset);
1996     } else {
1997       g_error ("Unsupported atom used inside dref atom");
1998     }
1999     walker = g_list_next (walker);
2000   }
2001
2002   atom_write_size (buffer, size, offset, original_offset);
2003   return *offset - original_offset;
2004 }
2005
2006 static guint64
2007 atom_dinf_copy_data (AtomDINF * dinf, guint8 ** buffer, guint64 * size,
2008     guint64 * offset)
2009 {
2010   guint64 original_offset = *offset;
2011
2012   if (!atom_copy_data (&dinf->header, buffer, size, offset)) {
2013     return 0;
2014   }
2015
2016   if (!atom_dref_copy_data (&dinf->dref, buffer, size, offset)) {
2017     return 0;
2018   }
2019
2020   atom_write_size (buffer, size, offset, original_offset);
2021   return original_offset - *offset;
2022 }
2023
2024 static guint64
2025 atom_minf_copy_data (AtomMINF * minf, guint8 ** buffer, guint64 * size,
2026     guint64 * offset)
2027 {
2028   guint64 original_offset = *offset;
2029
2030   if (!atom_copy_data (&minf->header, buffer, size, offset)) {
2031     return 0;
2032   }
2033
2034   if (minf->vmhd) {
2035     if (!atom_vmhd_copy_data (minf->vmhd, buffer, size, offset)) {
2036       return 0;
2037     }
2038   } else if (minf->smhd) {
2039     if (!atom_smhd_copy_data (minf->smhd, buffer, size, offset)) {
2040       return 0;
2041     }
2042   } else if (minf->hmhd) {
2043     if (!atom_hmhd_copy_data (minf->hmhd, buffer, size, offset)) {
2044       return 0;
2045     }
2046   }
2047
2048   if (minf->hdlr) {
2049     if (!atom_hdlr_copy_data (minf->hdlr, buffer, size, offset)) {
2050       return 0;
2051     }
2052   }
2053
2054   if (!atom_dinf_copy_data (&minf->dinf, buffer, size, offset)) {
2055     return 0;
2056   }
2057   if (!atom_stbl_copy_data (&minf->stbl, buffer, size, offset)) {
2058     return 0;
2059   }
2060
2061   atom_write_size (buffer, size, offset, original_offset);
2062   return *offset - original_offset;
2063 }
2064
2065 static guint64
2066 atom_mdhd_copy_data (AtomMDHD * mdhd, guint8 ** buffer, guint64 * size,
2067     guint64 * offset)
2068 {
2069   guint64 original_offset = *offset;
2070
2071   if (!atom_full_copy_data (&mdhd->header, buffer, size, offset)) {
2072     return 0;
2073   }
2074
2075   if (!common_time_info_copy_data (&mdhd->time_info,
2076           atom_full_get_version (&mdhd->header) == 0, buffer, size, offset)) {
2077     return 0;
2078   }
2079
2080   prop_copy_uint16 (mdhd->language_code, buffer, size, offset);
2081   prop_copy_uint16 (mdhd->quality, buffer, size, offset);
2082
2083   atom_write_size (buffer, size, offset, original_offset);
2084   return *offset - original_offset;
2085 }
2086
2087 static guint64
2088 atom_mdia_copy_data (AtomMDIA * mdia, guint8 ** buffer, guint64 * size,
2089     guint64 * offset)
2090 {
2091   guint64 original_offset = *offset;
2092
2093   if (!atom_copy_data (&mdia->header, buffer, size, offset)) {
2094     return 0;
2095   }
2096   if (!atom_mdhd_copy_data (&mdia->mdhd, buffer, size, offset)) {
2097     return 0;
2098   }
2099   if (!atom_hdlr_copy_data (&mdia->hdlr, buffer, size, offset)) {
2100     return 0;
2101   }
2102
2103   if (!atom_minf_copy_data (&mdia->minf, buffer, size, offset)) {
2104     return 0;
2105   }
2106
2107   atom_write_size (buffer, size, offset, original_offset);
2108   return *offset - original_offset;
2109 }
2110
2111 static guint64
2112 atom_elst_copy_data (AtomELST * elst, guint8 ** buffer, guint64 * size,
2113     guint64 * offset)
2114 {
2115   guint64 original_offset = *offset;
2116   GSList *walker;
2117
2118   if (!atom_full_copy_data (&elst->header, buffer, size, offset)) {
2119     return 0;
2120   }
2121
2122   prop_copy_uint32 (g_slist_length (elst->entries), buffer, size, offset);
2123
2124   for (walker = elst->entries; walker != NULL; walker = g_slist_next (walker)) {
2125     EditListEntry *entry = (EditListEntry *) walker->data;
2126     prop_copy_uint32 (entry->duration, buffer, size, offset);
2127     prop_copy_uint32 (entry->media_time, buffer, size, offset);
2128     prop_copy_uint32 (entry->media_rate, buffer, size, offset);
2129   }
2130   atom_write_size (buffer, size, offset, original_offset);
2131   return *offset - original_offset;
2132 }
2133
2134 static guint64
2135 atom_edts_copy_data (AtomEDTS * edts, guint8 ** buffer, guint64 * size,
2136     guint64 * offset)
2137 {
2138   guint64 original_offset = *offset;
2139
2140   if (!atom_copy_data (&(edts->header), buffer, size, offset))
2141     return 0;
2142
2143   if (!atom_elst_copy_data (&(edts->elst), buffer, size, offset))
2144     return 0;
2145
2146   atom_write_size (buffer, size, offset, original_offset);
2147   return *offset - original_offset;
2148 }
2149
2150 static guint64
2151 atom_trak_copy_data (AtomTRAK * trak, guint8 ** buffer, guint64 * size,
2152     guint64 * offset)
2153 {
2154   guint64 original_offset = *offset;
2155
2156   if (!atom_copy_data (&trak->header, buffer, size, offset)) {
2157     return 0;
2158   }
2159   if (!atom_tkhd_copy_data (&trak->tkhd, buffer, size, offset)) {
2160     return 0;
2161   }
2162   if (trak->edts) {
2163     if (!atom_edts_copy_data (trak->edts, buffer, size, offset)) {
2164       return 0;
2165     }
2166   }
2167
2168   if (!atom_mdia_copy_data (&trak->mdia, buffer, size, offset)) {
2169     return 0;
2170   }
2171
2172   atom_write_size (buffer, size, offset, original_offset);
2173   return *offset - original_offset;
2174 }
2175
2176 static guint64
2177 atom_tag_data_copy_data (AtomTagData * data, guint8 ** buffer, guint64 * size,
2178     guint64 * offset)
2179 {
2180   guint64 original_offset = *offset;
2181
2182   if (!atom_full_copy_data (&data->header, buffer, size, offset)) {
2183     return 0;
2184   }
2185
2186   prop_copy_uint32 (data->reserved, buffer, size, offset);
2187   prop_copy_uint8_array (data->data, data->datalen, buffer, size, offset);
2188
2189   atom_write_size (buffer, size, offset, original_offset);
2190   return *offset - original_offset;
2191 }
2192
2193 static guint64
2194 atom_tag_copy_data (AtomTag * tag, guint8 ** buffer, guint64 * size,
2195     guint64 * offset)
2196 {
2197   guint64 original_offset = *offset;
2198
2199   if (!atom_copy_data (&tag->header, buffer, size, offset)) {
2200     return 0;
2201   }
2202
2203   if (!atom_tag_data_copy_data (&tag->data, buffer, size, offset)) {
2204     return 0;
2205   }
2206
2207   atom_write_size (buffer, size, offset, original_offset);
2208   return *offset - original_offset;
2209 }
2210
2211 static guint64
2212 atom_ilst_copy_data (AtomILST * ilst, guint8 ** buffer, guint64 * size,
2213     guint64 * offset)
2214 {
2215   guint64 original_offset = *offset;
2216
2217   if (!atom_copy_data (&ilst->header, buffer, size, offset)) {
2218     return 0;
2219   }
2220   /* extra atoms */
2221   if (ilst->entries &&
2222       !atom_info_list_copy_data (ilst->entries, buffer, size, offset))
2223     return 0;
2224
2225   atom_write_size (buffer, size, offset, original_offset);
2226   return *offset - original_offset;
2227 }
2228
2229 static guint64
2230 atom_meta_copy_data (AtomMETA * meta, guint8 ** buffer, guint64 * size,
2231     guint64 * offset)
2232 {
2233   guint64 original_offset = *offset;
2234
2235   if (!atom_full_copy_data (&meta->header, buffer, size, offset)) {
2236     return 0;
2237   }
2238   if (!atom_hdlr_copy_data (&meta->hdlr, buffer, size, offset)) {
2239     return 0;
2240   }
2241   if (meta->ilst) {
2242     if (!atom_ilst_copy_data (meta->ilst, buffer, size, offset)) {
2243       return 0;
2244     }
2245   }
2246
2247   atom_write_size (buffer, size, offset, original_offset);
2248   return *offset - original_offset;
2249 }
2250
2251 static guint64
2252 atom_udta_copy_data (AtomUDTA * udta, guint8 ** buffer, guint64 * size,
2253     guint64 * offset)
2254 {
2255   guint64 original_offset = *offset;
2256
2257   if (!atom_copy_data (&udta->header, buffer, size, offset)) {
2258     return 0;
2259   }
2260   if (udta->meta) {
2261     if (!atom_meta_copy_data (udta->meta, buffer, size, offset)) {
2262       return 0;
2263     }
2264   } else if (udta->entries) {
2265     /* extra atoms */
2266     if (!atom_info_list_copy_data (udta->entries, buffer, size, offset))
2267       return 0;
2268   }
2269
2270   atom_write_size (buffer, size, offset, original_offset);
2271   return *offset - original_offset;
2272 }
2273
2274 guint64
2275 atom_moov_copy_data (AtomMOOV * atom, guint8 ** buffer, guint64 * size,
2276     guint64 * offset)
2277 {
2278   guint64 original_offset = *offset;
2279   GList *walker;
2280
2281   if (!atom_copy_data (&(atom->header), buffer, size, offset))
2282     return 0;
2283
2284   if (!atom_mvhd_copy_data (&(atom->mvhd), buffer, size, offset))
2285     return 0;
2286
2287   walker = g_list_first (atom->traks);
2288   while (walker != NULL) {
2289     if (!atom_trak_copy_data ((AtomTRAK *) walker->data, buffer, size, offset)) {
2290       return 0;
2291     }
2292     walker = g_list_next (walker);
2293   }
2294
2295   if (atom->udta) {
2296     if (!atom_udta_copy_data (atom->udta, buffer, size, offset)) {
2297       return 0;
2298     }
2299   }
2300
2301   atom_write_size (buffer, size, offset, original_offset);
2302   return *offset - original_offset;
2303 }
2304
2305 static guint64
2306 atom_wave_copy_data (AtomWAVE * wave, guint8 ** buffer,
2307     guint64 * size, guint64 * offset)
2308 {
2309   guint64 original_offset = *offset;
2310
2311   if (!atom_copy_data (&(wave->header), buffer, size, offset))
2312     return 0;
2313
2314   if (wave->extension_atoms) {
2315     if (!atom_info_list_copy_data (wave->extension_atoms, buffer, size, offset))
2316       return 0;
2317   }
2318
2319   atom_write_size (buffer, size, offset, original_offset);
2320   return *offset - original_offset;
2321 }
2322
2323 /* -- end of copy data functions -- */
2324
2325 /* -- general functions, API and support functions */
2326
2327 /* add samples to tables */
2328
2329 static STSCEntry *
2330 stsc_entry_new (guint32 first_chunk, guint32 samples, guint32 desc_index)
2331 {
2332   STSCEntry *entry = g_new0 (STSCEntry, 1);
2333
2334   entry->first_chunk = first_chunk;
2335   entry->samples_per_chunk = samples;
2336   entry->sample_description_index = desc_index;
2337   return entry;
2338 }
2339
2340 static void
2341 atom_stsc_add_new_entry (AtomSTSC * stsc, guint32 first_chunk, guint32 nsamples)
2342 {
2343   if (stsc->entries &&
2344       ((STSCEntry *) stsc->entries->data)->samples_per_chunk == nsamples)
2345     return;
2346
2347   stsc->entries =
2348       g_list_prepend (stsc->entries, stsc_entry_new (first_chunk, nsamples, 1));
2349   stsc->n_entries++;
2350 }
2351
2352 static STTSEntry *
2353 stts_entry_new (guint32 sample_count, gint32 sample_delta)
2354 {
2355   STTSEntry *entry = g_new0 (STTSEntry, 1);
2356
2357   entry->sample_count = sample_count;
2358   entry->sample_delta = sample_delta;
2359   return entry;
2360 }
2361
2362 static void
2363 atom_stts_add_entry (AtomSTTS * stts, guint32 sample_count, gint32 sample_delta)
2364 {
2365   STTSEntry *entry;
2366
2367   if (stts->entries == NULL) {
2368     stts->entries =
2369         g_list_prepend (stts->entries, stts_entry_new (sample_count,
2370             sample_delta));
2371     stts->n_entries++;
2372     return;
2373   }
2374   entry = (STTSEntry *) g_list_first (stts->entries)->data;
2375   if (entry->sample_delta == sample_delta) {
2376     entry->sample_count += sample_count;
2377   } else {
2378     stts->entries =
2379         g_list_prepend (stts->entries, stts_entry_new (sample_count,
2380             sample_delta));
2381     stts->n_entries++;
2382   }
2383 }
2384
2385 static void
2386 atom_stsz_add_entry (AtomSTSZ * stsz, guint32 nsamples, guint32 size)
2387 {
2388   guint32 i;
2389
2390   stsz->table_size += nsamples;
2391   if (stsz->sample_size != 0) {
2392     /* it is constant size, we don't need entries */
2393     return;
2394   }
2395   for (i = 0; i < nsamples; i++) {
2396     stsz->entries = g_list_prepend (stsz->entries, GUINT_TO_POINTER (size));
2397   }
2398 }
2399
2400 static guint32
2401 atom_stco64_get_entry_count (AtomSTCO64 * stco64)
2402 {
2403   return stco64->n_entries;
2404 }
2405
2406 static void
2407 atom_stco64_add_entry (AtomSTCO64 * stco64, guint64 entry)
2408 {
2409   guint64 *pont = g_new (guint64, 1);
2410
2411   *pont = entry;
2412   stco64->entries = g_list_prepend (stco64->entries, pont);
2413   stco64->n_entries++;
2414 }
2415
2416 static void
2417 atom_stss_add_entry (AtomSTSS * stss, guint32 sample)
2418 {
2419   stss->entries = g_list_prepend (stss->entries, GUINT_TO_POINTER (sample));
2420   stss->n_entries++;
2421 }
2422
2423 static void
2424 atom_stbl_add_stss_entry (AtomSTBL * stbl)
2425 {
2426   guint32 sample_index = stbl->stsz.table_size;
2427
2428   atom_stss_add_entry (&stbl->stss, sample_index);
2429 }
2430
2431 static void
2432 atom_ctts_add_entry (AtomCTTS * ctts, guint32 nsamples, guint32 offset)
2433 {
2434   GList *walker;
2435   CTTSEntry *entry;
2436
2437   walker = g_list_first (ctts->entries);
2438   entry = (walker == NULL) ? NULL : (CTTSEntry *) walker->data;
2439
2440   if (entry == NULL || entry->sampleoffset != offset) {
2441     CTTSEntry *entry = g_new0 (CTTSEntry, 1);
2442
2443     entry->samplecount = nsamples;
2444     entry->sampleoffset = offset;
2445     ctts->entries = g_list_prepend (ctts->entries, entry);
2446     ctts->n_entries++;
2447   } else {
2448     entry->samplecount += nsamples;
2449   }
2450 }
2451
2452 static void
2453 atom_stbl_add_ctts_entry (AtomSTBL * stbl, guint32 nsamples, guint32 offset)
2454 {
2455   if (stbl->ctts == NULL) {
2456     stbl->ctts = atom_ctts_new ();
2457   }
2458   atom_ctts_add_entry (stbl->ctts, nsamples, offset);
2459 }
2460
2461 void
2462 atom_trak_add_samples (AtomTRAK * trak, guint32 nsamples, guint32 delta,
2463     guint32 size, guint64 chunk_offset, gboolean sync,
2464     gboolean do_pts, gint64 pts_offset)
2465 {
2466   AtomSTBL *stbl = &trak->mdia.minf.stbl;
2467
2468   atom_stts_add_entry (&stbl->stts, nsamples, delta);
2469   atom_stsz_add_entry (&stbl->stsz, nsamples, size);
2470   atom_stco64_add_entry (&stbl->stco64, chunk_offset);
2471   atom_stsc_add_new_entry (&stbl->stsc,
2472       atom_stco64_get_entry_count (&stbl->stco64), nsamples);
2473   if (sync)
2474     atom_stbl_add_stss_entry (stbl);
2475   if (do_pts)
2476     atom_stbl_add_ctts_entry (stbl, nsamples, pts_offset);
2477 }
2478
2479 /* trak and moov molding */
2480
2481 guint32
2482 atom_trak_get_timescale (AtomTRAK * trak)
2483 {
2484   return trak->mdia.mdhd.time_info.timescale;
2485 }
2486
2487 static void
2488 atom_trak_set_id (AtomTRAK * trak, guint32 id)
2489 {
2490   trak->tkhd.track_ID = id;
2491 }
2492
2493 void
2494 atom_moov_add_trak (AtomMOOV * moov, AtomTRAK * trak)
2495 {
2496   atom_trak_set_id (trak, moov->mvhd.next_track_id++);
2497   moov->traks = g_list_append (moov->traks, trak);
2498 }
2499
2500 static guint64
2501 atom_trak_get_duration (AtomTRAK * trak)
2502 {
2503   return trak->tkhd.duration;
2504 }
2505
2506 static guint64
2507 atom_stts_get_total_duration (AtomSTTS * stts)
2508 {
2509   GList *walker = stts->entries;
2510   guint64 sum = 0;
2511
2512   while (walker) {
2513     STTSEntry *entry = (STTSEntry *) walker->data;
2514
2515     sum += (guint64) (entry->sample_count) * entry->sample_delta;
2516     walker = g_list_next (walker);
2517   }
2518   return sum;
2519 }
2520
2521 static void
2522 atom_trak_update_duration (AtomTRAK * trak, guint64 moov_timescale)
2523 {
2524   trak->mdia.mdhd.time_info.duration =
2525       atom_stts_get_total_duration (&trak->mdia.minf.stbl.stts);
2526   if (trak->mdia.mdhd.time_info.timescale != 0) {
2527     trak->tkhd.duration =
2528         gst_util_uint64_scale (trak->mdia.mdhd.time_info.duration,
2529         moov_timescale, trak->mdia.mdhd.time_info.timescale);
2530   } else {
2531     trak->tkhd.duration = 0;
2532   }
2533 }
2534
2535 static guint32
2536 atom_moov_get_timescale (AtomMOOV * moov)
2537 {
2538   return moov->mvhd.time_info.timescale;
2539 }
2540
2541 void
2542 atom_moov_update_timescale (AtomMOOV * moov, guint32 timescale)
2543 {
2544   moov->mvhd.time_info.timescale = timescale;
2545 }
2546
2547 void
2548 atom_moov_update_duration (AtomMOOV * moov)
2549 {
2550   GList *traks = moov->traks;
2551   guint64 dur, duration = 0;
2552
2553   while (traks) {
2554     AtomTRAK *trak = (AtomTRAK *) traks->data;
2555
2556     atom_trak_update_duration (trak, atom_moov_get_timescale (moov));
2557     dur = atom_trak_get_duration (trak);
2558     if (dur > duration)
2559       duration = dur;
2560     traks = g_list_next (traks);
2561   }
2562   moov->mvhd.time_info.duration = duration;
2563 }
2564
2565 static void
2566 atom_set_type (Atom * atom, guint32 fourcc)
2567 {
2568   atom->type = fourcc;
2569 }
2570
2571 static void
2572 atom_stbl_set_64bits (AtomSTBL * stbl, gboolean use)
2573 {
2574   if (use) {
2575     atom_set_type (&stbl->stco64.header.header, FOURCC_co64);
2576   } else {
2577     atom_set_type (&stbl->stco64.header.header, FOURCC_stco);
2578   }
2579 }
2580
2581 static void
2582 atom_trak_set_64bits (AtomTRAK * trak, gboolean use)
2583 {
2584   atom_stbl_set_64bits (&trak->mdia.minf.stbl, use);
2585 }
2586
2587 void
2588 atom_moov_set_64bits (AtomMOOV * moov, gboolean large_file)
2589 {
2590   GList *traks = moov->traks;
2591
2592   while (traks) {
2593     AtomTRAK *trak = (AtomTRAK *) traks->data;
2594
2595     atom_trak_set_64bits (trak, large_file);
2596     traks = g_list_next (traks);
2597   }
2598 }
2599
2600 static void
2601 atom_stco64_chunks_add_offset (AtomSTCO64 * stco64, guint32 offset)
2602 {
2603   GList *entries = stco64->entries;
2604
2605   while (entries) {
2606     guint64 *value = (guint64 *) entries->data;
2607
2608     *value += offset;
2609     entries = g_list_next (entries);
2610   }
2611 }
2612
2613 void
2614 atom_moov_chunks_add_offset (AtomMOOV * moov, guint32 offset)
2615 {
2616   GList *traks = moov->traks;
2617
2618   while (traks) {
2619     AtomTRAK *trak = (AtomTRAK *) traks->data;
2620
2621     atom_stco64_chunks_add_offset (&trak->mdia.minf.stbl.stco64, offset);
2622     traks = g_list_next (traks);
2623   }
2624 }
2625
2626 /*
2627  * Meta tags functions
2628  */
2629 static void
2630 atom_moov_init_metatags (AtomMOOV * moov, AtomsContext * context)
2631 {
2632   if (!moov->udta) {
2633     moov->udta = atom_udta_new ();
2634   }
2635   if (context->flavor != ATOMS_TREE_FLAVOR_3GP) {
2636     if (!moov->udta->meta) {
2637       moov->udta->meta = atom_meta_new ();
2638     }
2639     if (!moov->udta->meta->ilst) {
2640       moov->udta->meta->ilst = atom_ilst_new ();
2641     }
2642   }
2643 }
2644
2645 static void
2646 atom_tag_data_alloc_data (AtomTagData * data, guint size)
2647 {
2648   if (data->data != NULL) {
2649     g_free (data->data);
2650   }
2651   data->data = g_new0 (guint8, size);
2652   data->datalen = size;
2653 }
2654
2655 static void
2656 atom_moov_append_tag (AtomMOOV * moov, AtomInfo * tag)
2657 {
2658   GList **entries;
2659
2660   atom_moov_init_metatags (moov, &moov->context);
2661   if (moov->udta->meta)
2662     entries = &moov->udta->meta->ilst->entries;
2663   else
2664     entries = &moov->udta->entries;
2665   *entries = g_list_append (*entries, tag);
2666 }
2667
2668 void
2669 atom_moov_add_tag (AtomMOOV * moov, guint32 fourcc, guint32 flags,
2670     const guint8 * data, guint size)
2671 {
2672   AtomTag *tag;
2673   AtomTagData *tdata;
2674
2675   tag = atom_tag_new (fourcc, flags);
2676   tdata = &tag->data;
2677   atom_tag_data_alloc_data (tdata, size);
2678   g_memmove (tdata->data, data, size);
2679
2680   atom_moov_append_tag (moov,
2681       build_atom_info_wrapper ((Atom *) tag, atom_tag_copy_data,
2682           atom_tag_free));
2683 }
2684
2685 void
2686 atom_moov_add_str_tag (AtomMOOV * moov, guint32 fourcc, const gchar * value)
2687 {
2688   gint len = strlen (value);
2689
2690   if (len > 0)
2691     atom_moov_add_tag (moov, fourcc, METADATA_TEXT_FLAG, (guint8 *) value, len);
2692 }
2693
2694 void
2695 atom_moov_add_uint_tag (AtomMOOV * moov, guint32 fourcc, guint32 flags,
2696     guint32 value)
2697 {
2698   guint8 data[8] = { 0, };
2699
2700   if (flags) {
2701     GST_WRITE_UINT16_BE (data, value);
2702     atom_moov_add_tag (moov, fourcc, flags, data, 2);
2703   } else {
2704     GST_WRITE_UINT32_BE (data + 2, value);
2705     atom_moov_add_tag (moov, fourcc, flags, data, 8);
2706   }
2707 }
2708
2709 void
2710 atom_moov_add_blob_tag (AtomMOOV * moov, guint8 * data, guint size)
2711 {
2712   AtomData *data_atom;
2713   GstBuffer *buf;
2714   guint len;
2715   guint32 fourcc;
2716
2717   if (size < 8)
2718     return;
2719
2720   /* blob is unparsed atom;
2721    * extract size and fourcc, and wrap remainder in data atom */
2722   len = GST_READ_UINT32_BE (data);
2723   fourcc = GST_READ_UINT32_LE (data + 4);
2724   if (len > size)
2725     return;
2726
2727   buf = gst_buffer_new ();
2728   GST_BUFFER_SIZE (buf) = len - 8;
2729   GST_BUFFER_DATA (buf) = data + 8;
2730
2731   data_atom = atom_data_new_from_gst_buffer (fourcc, buf);
2732   gst_buffer_unref (buf);
2733
2734   atom_moov_append_tag (moov,
2735       build_atom_info_wrapper ((Atom *) data_atom, atom_data_copy_data,
2736           atom_data_free));
2737 }
2738
2739 void
2740 atom_moov_add_3gp_tag (AtomMOOV * moov, guint32 fourcc, guint8 * data,
2741     guint size)
2742 {
2743   AtomData *data_atom;
2744   GstBuffer *buf;
2745   guint8 *bdata;
2746
2747   /* need full atom */
2748   buf = gst_buffer_new_and_alloc (size + 4);
2749   bdata = GST_BUFFER_DATA (buf);
2750   /* full atom: version and flags */
2751   GST_WRITE_UINT32_BE (bdata, 0);
2752   memcpy (bdata + 4, data, size);
2753
2754   data_atom = atom_data_new_from_gst_buffer (fourcc, buf);
2755   gst_buffer_unref (buf);
2756
2757   atom_moov_append_tag (moov,
2758       build_atom_info_wrapper ((Atom *) data_atom, atom_data_copy_data,
2759           atom_data_free));
2760 }
2761
2762 guint16
2763 language_code (const char *lang)
2764 {
2765   g_return_val_if_fail (lang != NULL, 0);
2766   g_return_val_if_fail (strlen (lang) == 3, 0);
2767
2768   return (((lang[0] - 0x60) & 0x1F) << 10) + (((lang[1] - 0x60) & 0x1F) << 5) +
2769       ((lang[2] - 0x60) & 0x1F);
2770 }
2771
2772 void
2773 atom_moov_add_3gp_str_int_tag (AtomMOOV * moov, guint32 fourcc,
2774     const gchar * value, gint16 ivalue)
2775 {
2776   gint len = 0, size = 0;
2777   guint8 *data;
2778
2779   if (value) {
2780     len = strlen (value);
2781     size = len + 3;
2782   }
2783
2784   if (ivalue >= 0)
2785     size += 2;
2786
2787   data = g_malloc (size + 3);
2788   /* language tag and null-terminated UTF-8 string */
2789   if (value) {
2790     GST_WRITE_UINT16_BE (data, language_code (GST_QT_MUX_DEFAULT_TAG_LANGUAGE));
2791     /* include 0 terminator */
2792     memcpy (data + 2, value, len + 1);
2793   }
2794   /* 16-bit unsigned int if standalone, otherwise 8-bit */
2795   if (ivalue >= 0) {
2796     if (size == 2)
2797       GST_WRITE_UINT16_BE (data + size - 2, ivalue);
2798     else {
2799       GST_WRITE_UINT8 (data + size - 2, ivalue & 0xFF);
2800       size--;
2801     }
2802   }
2803
2804   atom_moov_add_3gp_tag (moov, fourcc, data, size);
2805   g_free (data);
2806 }
2807
2808 void
2809 atom_moov_add_3gp_str_tag (AtomMOOV * moov, guint32 fourcc, const gchar * value)
2810 {
2811   atom_moov_add_3gp_str_int_tag (moov, fourcc, value, -1);
2812 }
2813
2814 void
2815 atom_moov_add_3gp_uint_tag (AtomMOOV * moov, guint32 fourcc, guint16 value)
2816 {
2817   atom_moov_add_3gp_str_int_tag (moov, fourcc, NULL, value);
2818 }
2819
2820 /*
2821  * Functions for specifying media types
2822  */
2823
2824 static void
2825 atom_minf_set_audio (AtomMINF * minf)
2826 {
2827   atom_minf_clear_handlers (minf);
2828   minf->smhd = atom_smhd_new ();
2829 }
2830
2831 static void
2832 atom_minf_set_video (AtomMINF * minf, AtomsContext * context)
2833 {
2834   atom_minf_clear_handlers (minf);
2835   minf->vmhd = atom_vmhd_new (context);
2836 }
2837
2838 static void
2839 atom_hdlr_set_type (AtomHDLR * hdlr, AtomsContext * context, guint32 comp_type,
2840     guint32 hdlr_type)
2841 {
2842   if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
2843     hdlr->component_type = comp_type;
2844   }
2845   hdlr->handler_type = hdlr_type;
2846 }
2847
2848 static void
2849 atom_mdia_set_hdlr_type_audio (AtomMDIA * mdia, AtomsContext * context)
2850 {
2851   atom_hdlr_set_type (&mdia->hdlr, context, FOURCC_mhlr, FOURCC_soun);
2852 }
2853
2854 static void
2855 atom_mdia_set_hdlr_type_video (AtomMDIA * mdia, AtomsContext * context)
2856 {
2857   atom_hdlr_set_type (&mdia->hdlr, context, FOURCC_mhlr, FOURCC_vide);
2858 }
2859
2860 static void
2861 atom_mdia_set_audio (AtomMDIA * mdia, AtomsContext * context)
2862 {
2863   atom_mdia_set_hdlr_type_audio (mdia, context);
2864   atom_minf_set_audio (&mdia->minf);
2865 }
2866
2867 static void
2868 atom_mdia_set_video (AtomMDIA * mdia, AtomsContext * context)
2869 {
2870   atom_mdia_set_hdlr_type_video (mdia, context);
2871   atom_minf_set_video (&mdia->minf, context);
2872 }
2873
2874 static void
2875 atom_tkhd_set_audio (AtomTKHD * tkhd)
2876 {
2877   tkhd->volume = 0x0100;
2878   tkhd->width = tkhd->height = 0;
2879 }
2880
2881 static void
2882 atom_tkhd_set_video (AtomTKHD * tkhd, AtomsContext * context, guint32 width,
2883     guint32 height)
2884 {
2885   tkhd->volume = 0;
2886
2887   /* qt and ISO base media do not contradict, and examples agree */
2888   tkhd->width = width;
2889   tkhd->height = height;
2890 }
2891
2892 static void
2893 atom_edts_add_entry (AtomEDTS * edts, EditListEntry * entry)
2894 {
2895   edts->elst.entries = g_slist_append (edts->elst.entries, entry);
2896 }
2897
2898 /*
2899  * Adds a new entry to this trak edits list
2900  * duration is in the moov's timescale
2901  * media_time is the offset in the media time to start from (media's timescale)
2902  * rate is a 32 bits fixed-point
2903  */
2904 void
2905 atom_trak_add_elst_entry (AtomTRAK * trak, guint32 duration, guint32 media_time,
2906     guint32 rate)
2907 {
2908   EditListEntry *entry = g_new (EditListEntry, 1);
2909
2910   entry->duration = duration;
2911   entry->media_time = media_time;
2912   entry->media_rate = rate;
2913
2914   if (trak->edts == NULL) {
2915     trak->edts = atom_edts_new ();
2916   }
2917   atom_edts_add_entry (trak->edts, entry);
2918 }
2919
2920 /* re-negotiation is prevented at top-level, so only 1 entry expected.
2921  * Quite some more care here and elsewhere may be needed to
2922  * support several entries */
2923 static SampleTableEntryMP4A *
2924 atom_trak_add_audio_entry (AtomTRAK * trak, AtomsContext * context,
2925     guint32 type)
2926 {
2927   AtomSTSD *stsd = &trak->mdia.minf.stbl.stsd;
2928   SampleTableEntryMP4A *mp4a = sample_entry_mp4a_new ();
2929
2930   mp4a->se.header.type = type;
2931   mp4a->se.kind = AUDIO;
2932   mp4a->compression_id = -1;
2933   mp4a->se.data_reference_index = 1;
2934
2935   stsd->entries = g_list_prepend (stsd->entries, mp4a);
2936   stsd->n_entries++;
2937   return mp4a;
2938 }
2939
2940 static SampleTableEntryMP4V *
2941 atom_trak_add_video_entry (AtomTRAK * trak, AtomsContext * context,
2942     guint32 type)
2943 {
2944   SampleTableEntryMP4V *mp4v = sample_entry_mp4v_new (context);
2945   AtomSTSD *stsd = &trak->mdia.minf.stbl.stsd;
2946
2947   mp4v->se.header.type = type;
2948   mp4v->se.kind = VIDEO;
2949   mp4v->se.data_reference_index = 1;
2950   mp4v->horizontal_resolution = 72 << 16;
2951   mp4v->vertical_resolution = 72 << 16;
2952   if (context->flavor == ATOMS_TREE_FLAVOR_MOV) {
2953     mp4v->spatial_quality = 512;
2954     mp4v->temporal_quality = 512;
2955   }
2956
2957   stsd->entries = g_list_prepend (stsd->entries, mp4v);
2958   stsd->n_entries++;
2959   return mp4v;
2960 }
2961
2962 static void
2963 atom_trak_set_constant_size_samples (AtomTRAK * trak, guint32 sample_size)
2964 {
2965   trak->mdia.minf.stbl.stsz.sample_size = sample_size;
2966 }
2967
2968 static void
2969 atom_trak_set_audio (AtomTRAK * trak, AtomsContext * context)
2970 {
2971   atom_tkhd_set_audio (&trak->tkhd);
2972   atom_mdia_set_audio (&trak->mdia, context);
2973 }
2974
2975 static void
2976 atom_trak_set_video (AtomTRAK * trak, AtomsContext * context, guint32 width,
2977     guint32 height)
2978 {
2979   atom_tkhd_set_video (&trak->tkhd, context, width, height);
2980   atom_mdia_set_video (&trak->mdia, context);
2981 }
2982
2983 static void
2984 atom_trak_set_audio_commons (AtomTRAK * trak, AtomsContext * context,
2985     guint32 rate)
2986 {
2987   atom_trak_set_audio (trak, context);
2988   trak->mdia.mdhd.time_info.timescale = rate;
2989 }
2990
2991 static void
2992 atom_trak_set_video_commons (AtomTRAK * trak, AtomsContext * context,
2993     guint32 rate, guint32 width, guint32 height)
2994 {
2995   atom_trak_set_video (trak, context, width, height);
2996   trak->mdia.mdhd.time_info.timescale = rate;
2997   trak->tkhd.width = width << 16;
2998   trak->tkhd.height = height << 16;
2999 }
3000
3001 void
3002 atom_trak_set_audio_type (AtomTRAK * trak, AtomsContext * context,
3003     AudioSampleEntry * entry, guint32 scale, AtomInfo * ext, gint sample_size)
3004 {
3005   SampleTableEntryMP4A *ste;
3006
3007   atom_trak_set_audio_commons (trak, context, scale);
3008   ste = atom_trak_add_audio_entry (trak, context, entry->fourcc);
3009
3010   trak->is_video = FALSE;
3011   trak->is_h264 = FALSE;
3012
3013   ste->version = entry->version;
3014   ste->compression_id = entry->compression_id;
3015   ste->sample_size = entry->sample_size;
3016   ste->sample_rate = entry->sample_rate << 16;
3017   ste->channels = entry->channels;
3018
3019   ste->samples_per_packet = entry->samples_per_packet;
3020   ste->bytes_per_sample = entry->bytes_per_sample;
3021   ste->bytes_per_packet = entry->bytes_per_packet;
3022   ste->bytes_per_frame = entry->bytes_per_frame;
3023
3024   if (ext)
3025     ste->extension_atoms = g_list_prepend (ste->extension_atoms, ext);
3026
3027   /* 0 size means variable size */
3028   atom_trak_set_constant_size_samples (trak, sample_size);
3029 }
3030
3031 AtomInfo *
3032 build_pasp_extension (AtomTRAK * trak, gint par_width, gint par_height)
3033 {
3034   AtomData *atom_data;
3035   GstBuffer *buf;
3036   guint8 *data;
3037
3038   buf = gst_buffer_new_and_alloc (8);
3039   data = GST_BUFFER_DATA (buf);
3040
3041   /* ihdr = image header box */
3042   GST_WRITE_UINT32_BE (data, par_width);
3043   GST_WRITE_UINT32_BE (data + 4, par_height);
3044
3045   atom_data = atom_data_new_from_gst_buffer (FOURCC_pasp, buf);
3046   gst_buffer_unref (buf);
3047
3048   return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
3049       atom_data_free);
3050 }
3051
3052 void
3053 atom_trak_set_video_type (AtomTRAK * trak, AtomsContext * context,
3054     VisualSampleEntry * entry, guint32 scale, AtomInfo * ext)
3055 {
3056   SampleTableEntryMP4V *ste;
3057   gint dwidth, dheight;
3058   gint par_n = 0, par_d = 0;
3059
3060   if ((entry->par_n != 1 || entry->par_d != 1) &&
3061       (entry->par_n != entry->par_d)) {
3062     par_n = entry->par_n;
3063     par_d = entry->par_d;
3064   }
3065
3066   dwidth = entry->width;
3067   dheight = entry->height;
3068   /* ISO file spec says track header w/h indicates track's visual presentation
3069    * (so this together with pixels w/h implicitly defines PAR) */
3070   if (par_n && (context->flavor != ATOMS_TREE_FLAVOR_MOV)) {
3071     if (par_n > par_d) {
3072       dwidth = entry->width * par_n / par_d;
3073       dheight = entry->height;
3074     } else {
3075       dwidth = entry->width * par_n / par_d;
3076       dheight = entry->height;
3077     }
3078   }
3079
3080   atom_trak_set_video_commons (trak, context, scale, dwidth, dheight);
3081   ste = atom_trak_add_video_entry (trak, context, entry->fourcc);
3082
3083   trak->is_video = TRUE;
3084   trak->is_h264 = (entry->fourcc == FOURCC_avc1);
3085
3086   ste->width = entry->width;
3087   ste->height = entry->height;
3088   ste->depth = entry->depth;
3089   ste->color_table_id = entry->color_table_id;
3090   ste->frame_count = entry->frame_count;
3091
3092   if (ext)
3093     ste->extension_atoms = g_list_prepend (ste->extension_atoms, ext);
3094
3095   /* QT spec has a pasp extension atom in stsd that can hold PAR */
3096   if (par_n && (context->flavor == ATOMS_TREE_FLAVOR_MOV)) {
3097     ste->extension_atoms = g_list_append (ste->extension_atoms,
3098         build_pasp_extension (trak, par_n, par_d));
3099   }
3100 }
3101
3102 /* some sample description construction helpers */
3103
3104 AtomInfo *
3105 build_esds_extension (AtomTRAK * trak, guint8 object_type, guint8 stream_type,
3106     const GstBuffer * codec_data)
3107 {
3108   guint32 track_id;
3109   AtomESDS *esds;
3110
3111   track_id = trak->tkhd.track_ID;
3112
3113   esds = atom_esds_new ();
3114   esds->es.id = track_id & 0xFFFF;
3115   esds->es.dec_conf_desc.object_type = object_type;
3116   esds->es.dec_conf_desc.stream_type = stream_type << 2 | 0x01;
3117
3118   /* optional DecoderSpecificInfo */
3119   if (codec_data) {
3120     DecoderSpecificInfoDescriptor *desc;
3121
3122     esds->es.dec_conf_desc.dec_specific_info = desc =
3123         desc_dec_specific_info_new ();
3124     desc_dec_specific_info_alloc_data (desc, GST_BUFFER_SIZE (codec_data));
3125
3126     memcpy (desc->data, GST_BUFFER_DATA (codec_data),
3127         GST_BUFFER_SIZE (codec_data));
3128   }
3129
3130   return build_atom_info_wrapper ((Atom *) esds, atom_esds_copy_data,
3131       atom_esds_free);
3132 }
3133
3134 AtomInfo *
3135 build_mov_aac_extension (AtomTRAK * trak, const GstBuffer * codec_data)
3136 {
3137   AtomWAVE *wave;
3138   AtomFRMA *frma;
3139   Atom *ext_atom;
3140   GstBuffer *buf;
3141
3142   /* Add WAVE atom to the MP4A sample table entry */
3143   wave = atom_wave_new ();
3144
3145   /* Prepend Terminator atom to the WAVE list first, so it ends up last */
3146   ext_atom = (Atom *) atom_data_new (FOURCC_null);
3147   wave->extension_atoms =
3148       atom_info_list_prepend_atom (wave->extension_atoms, (Atom *) ext_atom,
3149       (AtomCopyDataFunc) atom_data_copy_data, (AtomFreeFunc) atom_data_free);
3150
3151   /* Add ESDS atom to WAVE */
3152   wave->extension_atoms = g_list_prepend (wave->extension_atoms,
3153       build_esds_extension (trak, ESDS_OBJECT_TYPE_MPEG4_P3,
3154           ESDS_STREAM_TYPE_AUDIO, codec_data));
3155
3156   /* Add MP4A atom to the WAVE:
3157    * not really in spec, but makes offset based players happy */
3158   buf = gst_buffer_new_and_alloc (4);
3159   *((guint32 *) GST_BUFFER_DATA (buf)) = 0;
3160   ext_atom = (Atom *) atom_data_new_from_gst_buffer (FOURCC_mp4a, buf);
3161   gst_buffer_unref (buf);
3162
3163   wave->extension_atoms =
3164       atom_info_list_prepend_atom (wave->extension_atoms, (Atom *) ext_atom,
3165       (AtomCopyDataFunc) atom_data_copy_data, (AtomFreeFunc) atom_data_free);
3166
3167   /* Add FRMA to the WAVE */
3168   frma = atom_frma_new ();
3169   frma->media_type = FOURCC_mp4a;
3170
3171   wave->extension_atoms =
3172       atom_info_list_prepend_atom (wave->extension_atoms, (Atom *) frma,
3173       (AtomCopyDataFunc) atom_frma_copy_data, (AtomFreeFunc) atom_frma_free);
3174
3175   return build_atom_info_wrapper ((Atom *) wave, atom_wave_copy_data,
3176       atom_wave_free);
3177 }
3178
3179 AtomInfo *
3180 build_jp2h_extension (AtomTRAK * trak, gint width, gint height, guint32 fourcc)
3181 {
3182   AtomData *atom_data;
3183   GstBuffer *buf;
3184   guint8 *data;
3185   guint8 cenum;
3186
3187   if (fourcc == GST_MAKE_FOURCC ('s', 'R', 'G', 'B')) {
3188     cenum = 0x10;
3189   } else if (fourcc == GST_MAKE_FOURCC ('s', 'Y', 'U', 'V')) {
3190     cenum = 0x12;
3191   } else
3192     return FALSE;
3193
3194   buf = gst_buffer_new_and_alloc (22 + 15);
3195   data = GST_BUFFER_DATA (buf);
3196
3197   /* ihdr = image header box */
3198   GST_WRITE_UINT32_BE (data, 22);
3199   GST_WRITE_UINT32_LE (data + 4, GST_MAKE_FOURCC ('i', 'h', 'd', 'r'));
3200   GST_WRITE_UINT32_BE (data + 8, height);
3201   GST_WRITE_UINT32_BE (data + 12, width);
3202   /* FIXME perhaps parse from stream,
3203    * though exactly 3 in any respectable colourspace */
3204   GST_WRITE_UINT16_BE (data + 16, 3);
3205   /* 8 bits per component, unsigned */
3206   GST_WRITE_UINT8 (data + 18, 0x7);
3207   /* compression type; reserved */
3208   GST_WRITE_UINT8 (data + 19, 0x7);
3209   /* colour space (un)known */
3210   GST_WRITE_UINT8 (data + 20, 0x0);
3211   /* intellectual property right (box present) */
3212   GST_WRITE_UINT8 (data + 21, 0x0);
3213
3214   /* colour specification box */
3215   data += 22;
3216   GST_WRITE_UINT32_BE (data, 15);
3217   GST_WRITE_UINT32_LE (data + 4, GST_MAKE_FOURCC ('c', 'o', 'l', 'r'));
3218   /* specification method: enumerated */
3219   GST_WRITE_UINT8 (data + 8, 0x1);
3220   /* precedence; reserved */
3221   GST_WRITE_UINT8 (data + 9, 0x0);
3222   /* approximation; reserved */
3223   GST_WRITE_UINT8 (data + 10, 0x0);
3224   /* enumerated colourspace */
3225   GST_WRITE_UINT32_BE (data + 11, cenum);
3226
3227   atom_data = atom_data_new_from_gst_buffer (FOURCC_jp2h, buf);
3228   gst_buffer_unref (buf);
3229
3230   return build_atom_info_wrapper ((Atom *) atom_data, atom_data_copy_data,
3231       atom_data_free);
3232 }
3233
3234 AtomInfo *
3235 build_codec_data_extension (guint32 fourcc, const GstBuffer * codec_data)
3236 {
3237   AtomData *data;
3238   AtomInfo *result = NULL;
3239
3240   if (codec_data) {
3241     data = atom_data_new_from_gst_buffer (fourcc, codec_data);
3242     result = build_atom_info_wrapper ((Atom *) data, atom_data_copy_data,
3243         atom_data_free);
3244   }
3245
3246   return result;
3247 }
3248
3249 AtomInfo *
3250 build_amr_extension ()
3251 {
3252   guint8 ext[9];
3253   GstBuffer *buf;
3254   AtomInfo *res;
3255
3256   buf = gst_buffer_new ();
3257   GST_BUFFER_DATA (buf) = ext;
3258   GST_BUFFER_SIZE (buf) = sizeof (ext);
3259
3260   /* vendor */
3261   GST_WRITE_UINT32_LE (ext, 0);
3262   /* decoder version */
3263   GST_WRITE_UINT8 (ext + 4, 0);
3264   /* mode set (all modes) */
3265   GST_WRITE_UINT16_BE (ext + 5, 0x81FF);
3266   /* mode change period (no restriction) */
3267   GST_WRITE_UINT8 (ext + 7, 0);
3268   /* frames per sample */
3269   GST_WRITE_UINT8 (ext + 8, 1);
3270
3271   res = build_codec_data_extension (GST_MAKE_FOURCC ('d', 'a', 'm', 'r'), buf);
3272   gst_buffer_unref (buf);
3273   return res;
3274 }
3275
3276 AtomInfo *
3277 build_h263_extension ()
3278 {
3279   guint8 ext[7];
3280   GstBuffer *buf;
3281   AtomInfo *res;
3282
3283   buf = gst_buffer_new ();
3284   GST_BUFFER_DATA (buf) = ext;
3285   GST_BUFFER_SIZE (buf) = sizeof (ext);
3286
3287   /* vendor */
3288   GST_WRITE_UINT32_LE (ext, 0);
3289   /* decoder version */
3290   GST_WRITE_UINT8 (ext + 4, 0);
3291   /* level / profile */
3292   /* FIXME ? maybe ? obtain somewhere; baseline for now */
3293   GST_WRITE_UINT8 (ext + 5, 10);
3294   GST_WRITE_UINT8 (ext + 6, 0);
3295
3296   res = build_codec_data_extension (GST_MAKE_FOURCC ('d', '2', '6', '3'), buf);
3297   gst_buffer_unref (buf);
3298   return res;
3299 }