gstreamer/gst-libs/gst/riff/riffencode.c

/* GStreamer
 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#include <string.h>

/*#define DEBUG_ENABLED */
#include "riff.h"

#define GST_RIFF_ENCODER_BUF_SIZE    1024

#define ADD_CHUNK(riffenc, chunkid, chunksize) \
{ \
  gst_riff_chunk *chunk;\
  chunk = (gst_riff_chunk *)(riffenc->dataleft + riffenc->nextlikely);\
  chunk->id = chunkid; \
  chunk->size = chunksize; \
  riffenc->nextlikely += sizeof(gst_riff_chunk); \
}

#define ADD_LIST(riffenc, listsize, listtype) \
{ \
  gst_riff_list *list;\
  list = (gst_riff_list *)(riffenc->dataleft + riffenc->nextlikely); \
  list->id = GST_RIFF_TAG_LIST; \
  list->size = listsize; \
  list->type = listtype; \
  riffenc->nextlikely += sizeof(gst_riff_list); \
}
  

GstRiff *gst_riff_encoder_new(guint32 type) {
  GstRiff *riff;
  gst_riff_list *list;

  GST_DEBUG ("gst_riff_encoder: making %4.4s encoder", (char *)&type);
  riff = (GstRiff *)g_malloc(sizeof(GstRiff));
  g_return_val_if_fail(riff != NULL, NULL);

  riff->form = 0;
  riff->chunks = NULL;
  riff->state = GST_RIFF_STATE_INITIAL;
  riff->curoffset = 0;
  riff->incomplete_chunk = NULL;
  riff->dataleft = g_malloc(GST_RIFF_ENCODER_BUF_SIZE);
  riff->dataleft_size = GST_RIFF_ENCODER_BUF_SIZE;
  riff->nextlikely = 0;

  list = (gst_riff_list *)riff->dataleft;
  list->id = GST_RIFF_TAG_RIFF;
  list->size = 0x00FFFFFF;
  list->type = GST_RIFF_RIFF_AVI;

  riff->nextlikely += sizeof(gst_riff_list);
  
  return riff;
}

gint gst_riff_encoder_avih(GstRiff *riff, gst_riff_avih *head, gulong size) {
  gst_riff_chunk *chunk;

  g_return_val_if_fail(riff->state == GST_RIFF_STATE_INITIAL, GST_RIFF_EINVAL);

  GST_DEBUG ("gst_riff_encoder: add avih");

  ADD_LIST(riff, 0xB8, GST_RIFF_LIST_hdrl);

  ADD_CHUNK(riff, GST_RIFF_TAG_avih, size);
  
  chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
  memcpy(chunk, head, size);
  riff->nextlikely += size;

  riff->state = GST_RIFF_STATE_HASAVIH;
  return GST_RIFF_OK;
}

gint gst_riff_encoder_strh(GstRiff *riff, guint32 fcc_type, gst_riff_strh *head, gulong size) {
  gst_riff_chunk *chunk;

  g_return_val_if_fail(riff->state == GST_RIFF_STATE_HASAVIH ||
		       riff->state == GST_RIFF_STATE_HASSTRF, GST_RIFF_EINVAL);

  GST_DEBUG ("gst_riff_encoder: add strh type %08x (%4.4s)", fcc_type, (char *)&fcc_type);

  ADD_LIST(riff, 108, GST_RIFF_LIST_strl);

  ADD_CHUNK(riff, GST_RIFF_TAG_strh, size);

  chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
  head->type = fcc_type;
  memcpy(chunk, head, size);

  riff->nextlikely += size;

  riff->state = GST_RIFF_STATE_HASSTRH;
  return GST_RIFF_OK;
}

gint gst_riff_encoder_strf(GstRiff *riff, void *format, gulong size) {
  gst_riff_chunk *chunk;

  g_return_val_if_fail(riff->state == GST_RIFF_STATE_HASSTRH, GST_RIFF_EINVAL);

  GST_DEBUG ("gst_riff_encoder: add strf");

  ADD_CHUNK(riff, GST_RIFF_TAG_strf, size);

  chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
  memcpy(chunk, format, size);
  riff->nextlikely += size;

  riff->state = GST_RIFF_STATE_HASSTRF;
  return GST_RIFF_OK;
}

gint gst_riff_encoder_chunk(GstRiff *riff, guint32 chunk_type, void *chunkdata, gulong size) {
  gst_riff_chunk *chunk;

  g_return_val_if_fail(riff->state == GST_RIFF_STATE_HASSTRF ||
		       riff->state == GST_RIFF_STATE_MOVI, GST_RIFF_EINVAL);

  if (riff->state != GST_RIFF_STATE_MOVI) {
    ADD_LIST(riff, 0x00FFFFFF, GST_RIFF_LIST_movi);
    riff->state = GST_RIFF_STATE_MOVI;
  }

  GST_DEBUG ("gst_riff_encoder: add chunk type %08x (%4.4s)", chunk_type, (char *)&chunk_type);
  
  ADD_CHUNK(riff, chunk_type, size);

  if (chunkdata != NULL) {
    chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
    memcpy(chunk, chunkdata, size);
    riff->nextlikely += size + (size&1);
  }

  return GST_RIFF_OK;
}

GstBuffer *gst_riff_encoder_get_buffer(GstRiff *riff) {
  GstBuffer *newbuf;

  newbuf = gst_buffer_new();
  GST_BUFFER_DATA(newbuf) = riff->dataleft;
  GST_BUFFER_SIZE(newbuf) = riff->nextlikely;

  return newbuf;
}

GstBuffer *gst_riff_encoder_get_and_reset_buffer(GstRiff *riff) {
  GstBuffer *newbuf;

  newbuf = gst_riff_encoder_get_buffer(riff);

  riff->dataleft = g_malloc(GST_RIFF_ENCODER_BUF_SIZE);
  riff->dataleft_size = GST_RIFF_ENCODER_BUF_SIZE;
  riff->nextlikely = 0;

  return newbuf;
}