gst/qtmux/atoms.c

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