More init code, silencing warnings
[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 struct StdioObject {
41   FILE *file;
42 };
43
44 struct OggOpusEnc {
45   OpusMSEncoder *st;
46   float *buffer;
47   OpusEncCallbacks callbacks;
48   void *user_data;
49 };
50
51 int stdio_write(void *user_data, const unsigned char *ptr, int len) {
52   struct StdioObject *obj = (struct StdioObject*)user_data;
53   return fwrite(ptr, 1, len, obj->file) != (size_t)len;
54 }
55
56 int stdio_close(void *user_data) {
57   struct StdioObject *obj = (struct StdioObject*)user_data;
58   int ret = fclose(obj->file);
59   free(obj);
60   return ret;
61 }
62
63 static const OpusEncCallbacks stdio_callbacks = {
64   stdio_write,
65   stdio_close
66 };
67
68 /* Create a new OggOpus file. */
69 OggOpusEnc *ope_create_file(const char *path, const OggOpusComments *comments,
70     int rate, int channels, int family, int *error) {
71   OggOpusEnc *enc;
72   struct StdioObject *obj;
73   obj = malloc(sizeof(*obj));
74   enc = ope_create_callbacks(&stdio_callbacks, obj, comments, rate, channels, family, error);
75   if (enc == NULL || (error && *error)) {
76     return NULL;
77   }
78   obj->file = fopen(path, "wb");
79   if (!obj->file) {
80     if (error) *error = OPE_ERROR_CANNOT_OPEN;
81     /* FIXME: Destroy the encoder properly. */
82     free(obj);
83     return NULL;
84   }
85   return enc;
86 }
87
88 /* Create a new OggOpus file (callback-based). */
89 OggOpusEnc *ope_create_callbacks(const OpusEncCallbacks *callbacks, void *user_data,
90     const OggOpusComments *comments, int rate, int channels, int family, int *error) {
91   OpusMSEncoder *st=NULL;
92   OggOpusEnc *enc=NULL;
93   OpusHeader header;
94   int ret;
95   if (family != 0 && family != 1 && family != 255) {
96     if (error) *error = OPE_ERROR_UNIMPLEMENTED;
97     return NULL;
98   }
99   header.channels=channels;
100   header.channel_mapping=family;
101   header.input_sample_rate=rate;
102   header.gain=0;
103   st=opus_multistream_surround_encoder_create(48000, channels, header.channel_mapping, &header.nb_streams, &header.nb_coupled,
104      header.stream_map, OPUS_APPLICATION_AUDIO, &ret);
105   if (! (ret == OPUS_OK && st != NULL) ) {
106     goto fail;
107   }
108   if ( (enc = malloc(sizeof(*enc))) == NULL) goto fail;
109   enc->st = st;
110   enc->callbacks = *callbacks;
111   enc->user_data = user_data;
112   (void)comments;
113   return enc;
114 fail:
115   if (enc) {
116     free(enc);
117   }
118   if (st) {
119     opus_multistream_encoder_destroy(st);
120   }
121   return NULL;
122 }
123
124 /* Add/encode any number of float samples to the file. */
125 int ope_write_float(OggOpusEnc *enc, float *pcm, int samples_per_channel) {
126   (void)enc;
127   (void)pcm;
128   (void)samples_per_channel;
129   return 0;
130 }
131
132 /* Add/encode any number of int16 samples to the file. */
133 int ope_write(OggOpusEnc *enc, opus_int16 *pcm, int samples_per_channel) {
134   (void)enc;
135   (void)pcm;
136   (void)samples_per_channel;
137   return 0;
138 }
139
140 static void finalize_stream(OggOpusEnc *enc) {
141   (void)enc;
142 }
143
144 /* Close/finalize the stream. */
145 int ope_close_and_free(OggOpusEnc *enc) {
146   finalize_stream(enc);
147   free(enc->buffer);
148   opus_multistream_encoder_destroy(enc->st);
149   return OPE_OK;
150 }
151
152 /* Ends the stream and create a new stream within the same file. */
153 int ope_chain_current(OggOpusEnc *enc, const OggOpusComments *comments) {
154   (void)enc;
155   (void)comments;
156   return 0;
157 }
158
159 /* Ends the stream and create a new file. */
160 int ope_continue_new_file(OggOpusEnc *enc, const OggOpusComments *comments, const char *path) {
161   (void)enc;
162   (void)comments;
163   (void)path;
164   return 0;
165 }
166
167 /* Ends the stream and create a new file (callback-based). */
168 int ope_continue_new_callbacks(OggOpusEnc *enc, const OggOpusComments *comments, void *user_data) {
169   (void)enc;
170   (void)comments;
171   (void)user_data;
172   return 0;
173 }
174
175 /* Goes straight to the libopus ctl() functions. */
176 int ope_encoder_ctl(OggOpusEnc *enc, int request, ...) {
177   (void)enc;
178   (void)request;
179   return 0;
180 }
181
182 /* ctl()-type call for the OggOpus layer. */
183 int ope_set_params(OggOpusEnc *enc, int request, ...) {
184   (void)enc;
185   (void)request;
186   return 0;
187 }