gst-libs/gst/riff/riffencode.c

   1 /* Gnome-Streamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20 #include <string.h>
  21
  22 //#define DEBUG_ENABLED
  23 #include "riff.h"
  24
  25 #define GST_RIFF_ENCODER_BUF_SIZE    1024
  26
  27 #define ADD_CHUNK(riffenc, chunkid, chunksize) \
  28 { \
  29   gst_riff_chunk *chunk;\
  30   chunk = (gst_riff_chunk *)(riffenc->dataleft + riffenc->nextlikely);\
  31   chunk->id = chunkid; \
  32   chunk->size = chunksize; \
  33   riffenc->nextlikely += sizeof(gst_riff_chunk); \
  34 }
  35
  36 #define ADD_LIST(riffenc, listsize, listtype) \
  37 { \
  38   gst_riff_list *list;\
  39   list = (gst_riff_list *)(riffenc->dataleft + riffenc->nextlikely); \
  40   list->id = GST_RIFF_TAG_LIST; \
  41   list->size = listsize; \
  42   list->type = listtype; \
  43   riffenc->nextlikely += sizeof(gst_riff_list); \
  44 }
  45
  46
  47 GstRiff *gst_riff_encoder_new(guint32 type) {
  48   GstRiff *riff;
  49   gst_riff_list *list;
  50
  51   GST_DEBUG (0,"gst_riff_encoder: making %4.4s encoder\n", (char *)&type);
  52   riff = (GstRiff *)g_malloc(sizeof(GstRiff));
  53   g_return_val_if_fail(riff != NULL, NULL);
  54
  55   riff->form = 0;
  56   riff->chunks = NULL;
  57   riff->state = GST_RIFF_STATE_INITIAL;
  58   riff->curoffset = 0;
  59   riff->incomplete_chunk = NULL;
  60   riff->dataleft = g_malloc(GST_RIFF_ENCODER_BUF_SIZE);
  61   riff->dataleft_size = GST_RIFF_ENCODER_BUF_SIZE;
  62   riff->nextlikely = 0;
  63
  64   list = (gst_riff_list *)riff->dataleft;
  65   list->id = GST_RIFF_TAG_RIFF;
  66   list->size = 0x00FFFFFF;
  67   list->type = GST_RIFF_RIFF_AVI;
  68
  69   riff->nextlikely += sizeof(gst_riff_list);
  70
  71   return riff;
  72 }
  73
  74 gint gst_riff_encoder_avih(GstRiff *riff, gst_riff_avih *head, gulong size) {
  75   gst_riff_chunk *chunk;
  76
  77   g_return_val_if_fail(riff->state == GST_RIFF_STATE_INITIAL, GST_RIFF_EINVAL);
  78
  79   GST_DEBUG (0,"gst_riff_encoder: add avih\n");
  80
  81   ADD_LIST(riff, 0xB8, GST_RIFF_LIST_hdrl);
  82
  83   ADD_CHUNK(riff, GST_RIFF_TAG_avih, size);
  84
  85   chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
  86   memcpy(chunk, head, size);
  87   riff->nextlikely += size;
  88
  89   riff->state = GST_RIFF_STATE_HASAVIH;
  90   return GST_RIFF_OK;
  91 }
  92
  93 gint gst_riff_encoder_strh(GstRiff *riff, guint32 fcc_type, gst_riff_strh *head, gulong size) {
  94   gst_riff_chunk *chunk;
  95
  96   g_return_val_if_fail(riff->state == GST_RIFF_STATE_HASAVIH ||
  97                        riff->state == GST_RIFF_STATE_HASSTRF, GST_RIFF_EINVAL);
  98
  99   GST_DEBUG (0,"gst_riff_encoder: add strh type %08x (%4.4s)\n", fcc_type, (char *)&fcc_type);
 100
 101   ADD_LIST(riff, 108, GST_RIFF_LIST_strl);
 102
 103   ADD_CHUNK(riff, GST_RIFF_TAG_strh, size);
 104
 105   chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
 106   head->type = fcc_type;
 107   memcpy(chunk, head, size);
 108
 109   riff->nextlikely += size;
 110
 111   riff->state = GST_RIFF_STATE_HASSTRH;
 112   return GST_RIFF_OK;
 113 }
 114
 115 gint gst_riff_encoder_strf(GstRiff *riff, void *format, gulong size) {
 116   gst_riff_chunk *chunk;
 117
 118   g_return_val_if_fail(riff->state == GST_RIFF_STATE_HASSTRH, GST_RIFF_EINVAL);
 119
 120   GST_DEBUG (0,"gst_riff_encoder: add strf\n");
 121
 122   ADD_CHUNK(riff, GST_RIFF_TAG_strf, size);
 123
 124   chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
 125   memcpy(chunk, format, size);
 126   riff->nextlikely += size;
 127
 128   riff->state = GST_RIFF_STATE_HASSTRF;
 129   return GST_RIFF_OK;
 130 }
 131
 132 gint gst_riff_encoder_chunk(GstRiff *riff, guint32 chunk_type, void *chunkdata, gulong size) {
 133   gst_riff_chunk *chunk;
 134
 135   g_return_val_if_fail(riff->state == GST_RIFF_STATE_HASSTRF ||
 136                        riff->state == GST_RIFF_STATE_MOVI, GST_RIFF_EINVAL);
 137
 138   if (riff->state != GST_RIFF_STATE_MOVI) {
 139     ADD_LIST(riff, 0x00FFFFFF, GST_RIFF_LIST_movi);
 140     riff->state = GST_RIFF_STATE_MOVI;
 141   }
 142
 143   GST_DEBUG (0,"gst_riff_encoder: add chunk type %08x (%4.4s)\n", chunk_type, (char *)&chunk_type);
 144
 145   ADD_CHUNK(riff, chunk_type, size);
 146
 147   if (chunkdata != NULL) {
 148     chunk = (gst_riff_chunk *)(riff->dataleft + riff->nextlikely);
 149     memcpy(chunk, chunkdata, size);
 150     riff->nextlikely += size + (size&1);
 151   }
 152
 153   return GST_RIFF_OK;
 154 }
 155
 156 GstBuffer *gst_riff_encoder_get_buffer(GstRiff *riff) {
 157   GstBuffer *newbuf;
 158
 159   newbuf = gst_buffer_new();
 160   GST_BUFFER_DATA(newbuf) = riff->dataleft;
 161   GST_BUFFER_SIZE(newbuf) = riff->nextlikely;
 162
 163   return newbuf;
 164 }
 165
 166 GstBuffer *gst_riff_encoder_get_and_reset_buffer(GstRiff *riff) {
 167   GstBuffer *newbuf;
 168
 169   newbuf = gst_riff_encoder_get_buffer(riff);
 170
 171   riff->dataleft = g_malloc(GST_RIFF_ENCODER_BUF_SIZE);
 172   riff->dataleft_size = GST_RIFF_ENCODER_BUF_SIZE;
 173   riff->nextlikely = 0;
 174
 175   return newbuf;
 176 }
 177