allocate buffer
[libopusenc.git] / src / opusenc.c
1 /* Copyright (C)2002-2017 Jean-Marc Valin
2    Copyright (C)2007-2013 Xiph.Org Foundation
3    Copyright (C)2008-2013 Gregory Maxwell
4    File: opusenc.c
5
6    Redistribution and use in source and binary forms, with or without
7    modification, are permitted provided that the following conditions
8    are met:
9
10    - Redistributions of source code must retain the above copyright
11    notice, this list of conditions and the following disclaimer.
12
13    - Redistributions in binary form must reproduce the above copyright
14    notice, this list of conditions and the following disclaimer in the
15    documentation and/or other materials provided with the distribution.
16
17    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20    A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
21    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
22    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
23    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
24    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
25    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
26    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
27    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30 #ifdef HAVE_CONFIG_H
31 #include "config.h"
32 #endif
33
34 #include <stdlib.h>
35 #include <stdio.h>
36 #include <opus_multistream.h>
37 #include "opusenc.h"
38 #include "opus_header.h"
39
40 #define BUFFER_SAMPLES 96000
41
42 struct StdioObject {
43   FILE *file;
44 };
45
46 struct OggOpusEnc {
47   OpusMSEncoder *st;
48   float *buffer;
49   OpusEncCallbacks callbacks;
50   void *user_data;
51 };
52
53 int stdio_write(void *user_data, const unsigned char *ptr, int len) {
54   struct StdioObject *obj = (struct StdioObject*)user_data;
55   return fwrite(ptr, 1, len, obj->file) != (size_t)len;
56 }
57
58 int stdio_close(void *user_data) {
59   struct StdioObject *obj = (struct StdioObject*)user_data;
60   int ret = fclose(obj->file);
61   free(obj);
62   return ret;
63 }
64
65 static const OpusEncCallbacks stdio_callbacks = {
66   stdio_write,
67   stdio_close
68 };
69
70 /* Create a new OggOpus file. */
71 OggOpusEnc *ope_create_file(const char *path, const OggOpusComments *comments,
72     int rate, int channels, int family, int *error) {
73   OggOpusEnc *enc;
74   struct StdioObject *obj;
75   obj = malloc(sizeof(*obj));
76   enc = ope_create_callbacks(&stdio_callbacks, obj, comments, rate, channels, family, error);
77   if (enc == NULL || (error && *error)) {
78     return NULL;
79   }
80   obj->file = fopen(path, "wb");
81   if (!obj->file) {
82     if (error) *error = OPE_ERROR_CANNOT_OPEN;
83     /* FIXME: Destroy the encoder properly. */
84     free(obj);
85     return NULL;
86   }
87   return enc;
88 }
89
90 /* Create a new OggOpus file (callback-based). */
91 OggOpusEnc *ope_create_callbacks(const OpusEncCallbacks *callbacks, void *user_data,
92     const OggOpusComments *comments, int rate, int channels, int family, int *error) {
93   OpusMSEncoder *st=NULL;
94   OggOpusEnc *enc=NULL;
95   OpusHeader header;
96   int ret;
97   if (family != 0 && family != 1 && family != 255) {
98     if (error) *error = OPE_ERROR_UNIMPLEMENTED;
99     return NULL;
100   }
101   if (channels <= 0 || channels > 255) {
102     if (error) *error = OPE_BAD_ARG;
103     return NULL;
104   }
105   header.channels=channels;
106   header.channel_mapping=family;
107   header.input_sample_rate=rate;
108   header.gain=0;
109   st=opus_multistream_surround_encoder_create(48000, channels, header.channel_mapping, &header.nb_streams, &header.nb_coupled,
110      header.stream_map, OPUS_APPLICATION_AUDIO, &ret);
111   if (! (ret == OPUS_OK && st != NULL) ) {
112     goto fail;
113   }
114   if ( (enc = malloc(sizeof(*enc))) == NULL) goto fail;
115   if ( (enc->buffer = malloc(sizeof(*enc->buffer)*BUFFER_SAMPLES*channels)) == NULL) goto fail;
116   enc->st = st;
117   enc->callbacks = *callbacks;
118   enc->user_data = user_data;
119   (void)comments;
120   return enc;
121 fail:
122   if (enc) {
123     free(enc);
124     if (enc->buffer) free(enc->buffer);
125   }
126   if (st) {
127     opus_multistream_encoder_destroy(st);
128   }
129   return NULL;
130 }
131
132 /* Add/encode any number of float samples to the file. */
133 int ope_write_float(OggOpusEnc *enc, float *pcm, int samples_per_channel) {
134   (void)enc;
135   (void)pcm;
136   (void)samples_per_channel;
137   return 0;
138 }
139
140 /* Add/encode any number of int16 samples to the file. */
141 int ope_write(OggOpusEnc *enc, opus_int16 *pcm, int samples_per_channel) {
142   (void)enc;
143   (void)pcm;
144   (void)samples_per_channel;
145   return 0;
146 }
147
148 static void finalize_stream(OggOpusEnc *enc) {
149   (void)enc;
150 }
151
152 /* Close/finalize the stream. */
153 int ope_close_and_free(OggOpusEnc *enc) {
154   finalize_stream(enc);
155   free(enc->buffer);
156   opus_multistream_encoder_destroy(enc->st);
157   return OPE_OK;
158 }
159
160 /* Ends the stream and create a new stream within the same file. */
161 int ope_chain_current(OggOpusEnc *enc, const OggOpusComments *comments) {
162   (void)enc;
163   (void)comments;
164   return 0;
165 }
166
167 /* Ends the stream and create a new file. */
168 int ope_continue_new_file(OggOpusEnc *enc, const OggOpusComments *comments, const char *path) {
169   (void)enc;
170   (void)comments;
171   (void)path;
172   return 0;
173 }
174
175 /* Ends the stream and create a new file (callback-based). */
176 int ope_continue_new_callbacks(OggOpusEnc *enc, const OggOpusComments *comments, void *user_data) {
177   (void)enc;
178   (void)comments;
179   (void)user_data;
180   return 0;
181 }
182
183 /* Goes straight to the libopus ctl() functions. */
184 int ope_encoder_ctl(OggOpusEnc *enc, int request, ...) {
185   (void)enc;
186   (void)request;
187   return 0;
188 }
189
190 /* ctl()-type call for the OggOpus layer. */
191 int ope_set_params(OggOpusEnc *enc, int request, ...) {
192   (void)enc;
193   (void)request;
194   return 0;
195 }