gst/quicktime/atoms.c

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