Adds a _ctl() control API to the encoder and decoder
[opus.git] / src / hybrid_encoder.c
1 /* Copyright (c) 2010 Xiph.Org Foundation, Skype Limited
2    Written by Jean-Marc Valin and Koen Vos */
3 /*
4    Redistribution and use in source and binary forms, with or without
5    modification, are permitted provided that the following conditions
6    are met:
7
8    - Redistributions of source code must retain the above copyright
9    notice, this list of conditions and the following disclaimer.
10
11    - Redistributions in binary form must reproduce the above copyright
12    notice, this list of conditions and the following disclaimer in the
13    documentation and/or other materials provided with the distribution.
14
15    - Neither the name of the Xiph.org Foundation nor the names of its
16    contributors may be used to endorse or promote products derived from
17    this software without specific prior written permission.
18
19    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22    A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
23    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
26    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
27    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
28    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
29    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 #ifdef HAVE_CONFIG_H
33 #include "config.h"
34 #endif
35
36 #include <stdlib.h>
37 #include <stdio.h>
38 #include <stdarg.h>
39 #include "hybrid_encoder.h"
40 #include "celt/libcelt/entenc.h"
41 #include "celt/libcelt/modes.h"
42 #include "SKP_Silk_SDK_API.h"
43
44 HybridEncoder *hybrid_encoder_create()
45 {
46         HybridEncoder *st;
47         int ret, encSizeBytes;
48
49         st = malloc(sizeof(HybridEncoder));
50
51     /* Create SILK encoder */
52     ret = SKP_Silk_SDK_Get_Encoder_Size( &encSizeBytes );
53     if( ret ) {
54         /* Handle error */
55     }
56         st->silk_enc = malloc(encSizeBytes);
57
58     ret = SKP_Silk_SDK_InitEncoder( st->silk_enc, &st->encControl );
59     if( ret ) {
60         /* Handle error */
61     }
62     /* Set Encoder parameters */
63     st->encControl.API_sampleRate        = 48000;
64     st->encControl.maxInternalSampleRate = 16000;
65     st->encControl.packetSize            = 960;
66     st->encControl.packetLossPercentage  = 0;
67     st->encControl.useInBandFEC          = 0;
68     st->encControl.useDTX                = 0;
69     st->encControl.complexity            = 2;
70     st->encControl.bitRate               = 18000;
71
72     /* Create CELT encoder */
73         /* We should not have to create a CELT mode for each encoder state */
74         st->celt_mode = celt_mode_create(48000, 960, NULL);
75         /* Initialize CELT encoder */
76         st->celt_enc = celt_encoder_create(st->celt_mode, 1, NULL);
77
78         st->mode = MODE_HYBRID;
79         st->bandwidth = BANDWIDTH_FULLBAND;
80
81         return st;
82 }
83
84 int hybrid_encode(HybridEncoder *st, const short *pcm, int frame_size,
85                 unsigned char *data, int bytes_per_packet)
86 {
87         int ret=0;
88         SKP_int16 nBytes;
89         ec_enc enc;
90         ec_byte_buffer buf;
91
92         ec_byte_writeinit_buffer(&buf, data, bytes_per_packet);
93         ec_enc_init(&enc,&buf);
94
95         if (st->mode != MODE_CELT_ONLY)
96         {
97             st->encControl.bitRate               = (bytes_per_packet*50*8+4000)/2;
98             /* Call SILK encoder for the low band */
99             nBytes = bytes_per_packet;
100             ret = SKP_Silk_SDK_Encode( st->silk_enc, &st->encControl, pcm, 960, &enc, &nBytes );
101             if( ret ) {
102                 fprintf (stderr, "SILK encode error\n");
103                 /* Handle error */
104             }
105             ret = (ec_enc_tell(&enc, 0)+7)>>3;
106         }
107
108         if (st->mode == MODE_HYBRID)
109         {
110             /* This should be adjusted based on the SILK bandwidth */
111             celt_encoder_ctl(st->celt_enc, CELT_SET_START_BAND(13));
112         } else {
113         celt_encoder_ctl(st->celt_enc, CELT_SET_START_BAND(0));
114         }
115
116         if (st->mode != MODE_SILK_ONLY)
117         {
118             /* Encode high band with CELT */
119             /* FIXME: Do some delay compensation here */
120             ret = celt_encode_with_ec(st->celt_enc, pcm, NULL, frame_size, data, bytes_per_packet, &enc);
121         } else {
122             ec_enc_done(&enc);
123         }
124
125         return ret;
126 }
127
128 void hybrid_encoder_ctl(HybridEncoder *st, int request, ...)
129 {
130     va_list ap;
131
132     va_start(ap, request);
133
134     switch (request)
135     {
136         case HYBRID_SET_MODE_REQUEST:
137         {
138             int value = va_arg(ap, int);
139             st->mode = value;
140         }
141         break;
142         case HYBRID_GET_MODE_REQUEST:
143         {
144             int *value = va_arg(ap, int*);
145             *value = st->mode;
146         }
147         break;
148         case HYBRID_SET_BANDWIDTH_REQUEST:
149         {
150             int value = va_arg(ap, int);
151             st->bandwidth = value;
152         }
153         break;
154         case HYBRID_GET_BANDWIDTH_REQUEST:
155         {
156             int *value = va_arg(ap, int*);
157             *value = st->bandwidth;
158         }
159         break;
160         case HYBRID_SET_VBR_RATE_REQUEST:
161         {
162             int value = va_arg(ap, int);
163             st->vbr_rate = value;
164         }
165         break;
166         case HYBRID_GET_VBR_RATE_REQUEST:
167         {
168             int *value = va_arg(ap, int*);
169             *value = st->vbr_rate;
170         }
171         break;
172         default:
173             fprintf(stderr, "unknown hybrid_encoder_ctl() request: %d", request);
174             break;
175     }
176 }
177
178 void hybrid_encoder_destroy(HybridEncoder *st)
179 {
180         free(st->silk_enc);
181
182         celt_encoder_destroy(st->celt_enc);
183         celt_mode_destroy(st->celt_mode);
184
185         free(st);
186 }
187