xref: /aosp_15_r20/external/libopus/tests/test_opus_encode.c (revision a58d3d2adb790c104798cd88c8a3aff4fa8b82cc)
1*a58d3d2aSXin Li /* Copyright (c) 2011-2013 Xiph.Org Foundation
2*a58d3d2aSXin Li    Written by Gregory Maxwell */
3*a58d3d2aSXin Li /*
4*a58d3d2aSXin Li    Redistribution and use in source and binary forms, with or without
5*a58d3d2aSXin Li    modification, are permitted provided that the following conditions
6*a58d3d2aSXin Li    are met:
7*a58d3d2aSXin Li 
8*a58d3d2aSXin Li    - Redistributions of source code must retain the above copyright
9*a58d3d2aSXin Li    notice, this list of conditions and the following disclaimer.
10*a58d3d2aSXin Li 
11*a58d3d2aSXin Li    - Redistributions in binary form must reproduce the above copyright
12*a58d3d2aSXin Li    notice, this list of conditions and the following disclaimer in the
13*a58d3d2aSXin Li    documentation and/or other materials provided with the distribution.
14*a58d3d2aSXin Li 
15*a58d3d2aSXin Li    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16*a58d3d2aSXin Li    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17*a58d3d2aSXin Li    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18*a58d3d2aSXin Li    A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
19*a58d3d2aSXin Li    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20*a58d3d2aSXin Li    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21*a58d3d2aSXin Li    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
22*a58d3d2aSXin Li    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
23*a58d3d2aSXin Li    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
24*a58d3d2aSXin Li    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25*a58d3d2aSXin Li    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26*a58d3d2aSXin Li */
27*a58d3d2aSXin Li 
28*a58d3d2aSXin Li #ifdef HAVE_CONFIG_H
29*a58d3d2aSXin Li #include "config.h"
30*a58d3d2aSXin Li #endif
31*a58d3d2aSXin Li 
32*a58d3d2aSXin Li #include <stdio.h>
33*a58d3d2aSXin Li #include <stdlib.h>
34*a58d3d2aSXin Li #include <limits.h>
35*a58d3d2aSXin Li #include <stdint.h>
36*a58d3d2aSXin Li #include <math.h>
37*a58d3d2aSXin Li #include <string.h>
38*a58d3d2aSXin Li #include <time.h>
39*a58d3d2aSXin Li #if (!defined WIN32 && !defined _WIN32) || defined(__MINGW32__)
40*a58d3d2aSXin Li #include <unistd.h>
41*a58d3d2aSXin Li #else
42*a58d3d2aSXin Li #include <process.h>
43*a58d3d2aSXin Li #define getpid _getpid
44*a58d3d2aSXin Li #endif
45*a58d3d2aSXin Li #include "opus_multistream.h"
46*a58d3d2aSXin Li #include "opus.h"
47*a58d3d2aSXin Li #include "../src/opus_private.h"
48*a58d3d2aSXin Li #include "test_opus_common.h"
49*a58d3d2aSXin Li 
50*a58d3d2aSXin Li #define MAX_PACKET (1500)
51*a58d3d2aSXin Li #define SAMPLES (48000*30)
52*a58d3d2aSXin Li #define SSAMPLES (SAMPLES/3)
53*a58d3d2aSXin Li #define MAX_FRAME_SAMP (5760)
54*a58d3d2aSXin Li #define PI (3.141592653589793238462643f)
55*a58d3d2aSXin Li #define RAND_SAMPLE(a) (a[fast_rand() % sizeof(a)/sizeof(a[0])])
56*a58d3d2aSXin Li 
generate_music(short * buf,opus_int32 len)57*a58d3d2aSXin Li void generate_music(short *buf, opus_int32 len)
58*a58d3d2aSXin Li {
59*a58d3d2aSXin Li    opus_int32 a1,b1,a2,b2;
60*a58d3d2aSXin Li    opus_int32 c1,c2,d1,d2;
61*a58d3d2aSXin Li    opus_int32 i,j;
62*a58d3d2aSXin Li    a1=b1=a2=b2=0;
63*a58d3d2aSXin Li    c1=c2=d1=d2=0;
64*a58d3d2aSXin Li    j=0;
65*a58d3d2aSXin Li    /*60ms silence*/
66*a58d3d2aSXin Li    for(i=0;i<2880;i++)buf[i*2]=buf[i*2+1]=0;
67*a58d3d2aSXin Li    for(i=2880;i<len;i++)
68*a58d3d2aSXin Li    {
69*a58d3d2aSXin Li      opus_uint32 r;
70*a58d3d2aSXin Li      opus_int32 v1,v2;
71*a58d3d2aSXin Li      v1=v2=(((j*((j>>12)^((j>>10|j>>12)&26&j>>7)))&128)+128)<<15;
72*a58d3d2aSXin Li      r=fast_rand();v1+=r&65535;v1-=r>>16;
73*a58d3d2aSXin Li      r=fast_rand();v2+=r&65535;v2-=r>>16;
74*a58d3d2aSXin Li      b1=v1-a1+((b1*61+32)>>6);a1=v1;
75*a58d3d2aSXin Li      b2=v2-a2+((b2*61+32)>>6);a2=v2;
76*a58d3d2aSXin Li      c1=(30*(c1+b1+d1)+32)>>6;d1=b1;
77*a58d3d2aSXin Li      c2=(30*(c2+b2+d2)+32)>>6;d2=b2;
78*a58d3d2aSXin Li      v1=(c1+128)>>8;
79*a58d3d2aSXin Li      v2=(c2+128)>>8;
80*a58d3d2aSXin Li      buf[i*2]=v1>32767?32767:(v1<-32768?-32768:v1);
81*a58d3d2aSXin Li      buf[i*2+1]=v2>32767?32767:(v2<-32768?-32768:v2);
82*a58d3d2aSXin Li      if(i%6==0)j++;
83*a58d3d2aSXin Li    }
84*a58d3d2aSXin Li }
85*a58d3d2aSXin Li 
86*a58d3d2aSXin Li #if 0
87*a58d3d2aSXin Li static int save_ctr = 0;
88*a58d3d2aSXin Li static void int_to_char(opus_uint32 i, unsigned char ch[4])
89*a58d3d2aSXin Li {
90*a58d3d2aSXin Li     ch[0] = i>>24;
91*a58d3d2aSXin Li     ch[1] = (i>>16)&0xFF;
92*a58d3d2aSXin Li     ch[2] = (i>>8)&0xFF;
93*a58d3d2aSXin Li     ch[3] = i&0xFF;
94*a58d3d2aSXin Li }
95*a58d3d2aSXin Li 
96*a58d3d2aSXin Li static OPUS_INLINE void save_packet(unsigned char* p, int len, opus_uint32 rng)
97*a58d3d2aSXin Li {
98*a58d3d2aSXin Li    FILE *fout;
99*a58d3d2aSXin Li    unsigned char int_field[4];
100*a58d3d2aSXin Li    char name[256];
101*a58d3d2aSXin Li    snprintf(name,255,"test_opus_encode.%llu.%d.bit",(unsigned long long)iseed,save_ctr);
102*a58d3d2aSXin Li    fprintf(stdout,"writing %d byte packet to %s\n",len,name);
103*a58d3d2aSXin Li    fout=fopen(name, "wb+");
104*a58d3d2aSXin Li    if(fout==NULL)test_failed();
105*a58d3d2aSXin Li    int_to_char(len, int_field);
106*a58d3d2aSXin Li    fwrite(int_field, 1, 4, fout);
107*a58d3d2aSXin Li    int_to_char(rng, int_field);
108*a58d3d2aSXin Li    fwrite(int_field, 1, 4, fout);
109*a58d3d2aSXin Li    fwrite(p, 1, len, fout);
110*a58d3d2aSXin Li    fclose(fout);
111*a58d3d2aSXin Li    save_ctr++;
112*a58d3d2aSXin Li }
113*a58d3d2aSXin Li #endif
114*a58d3d2aSXin Li 
get_frame_size_enum(int frame_size,int sampling_rate)115*a58d3d2aSXin Li int get_frame_size_enum(int frame_size, int sampling_rate)
116*a58d3d2aSXin Li {
117*a58d3d2aSXin Li    int frame_size_enum;
118*a58d3d2aSXin Li 
119*a58d3d2aSXin Li    if(frame_size==sampling_rate/400)
120*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_2_5_MS;
121*a58d3d2aSXin Li    else if(frame_size==sampling_rate/200)
122*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_5_MS;
123*a58d3d2aSXin Li    else if(frame_size==sampling_rate/100)
124*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_10_MS;
125*a58d3d2aSXin Li    else if(frame_size==sampling_rate/50)
126*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_20_MS;
127*a58d3d2aSXin Li    else if(frame_size==sampling_rate/25)
128*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_40_MS;
129*a58d3d2aSXin Li    else if(frame_size==3*sampling_rate/50)
130*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_60_MS;
131*a58d3d2aSXin Li    else if(frame_size==4*sampling_rate/50)
132*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_80_MS;
133*a58d3d2aSXin Li    else if(frame_size==5*sampling_rate/50)
134*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_100_MS;
135*a58d3d2aSXin Li    else if(frame_size==6*sampling_rate/50)
136*a58d3d2aSXin Li       frame_size_enum = OPUS_FRAMESIZE_120_MS;
137*a58d3d2aSXin Li    else
138*a58d3d2aSXin Li       test_failed();
139*a58d3d2aSXin Li 
140*a58d3d2aSXin Li    return frame_size_enum;
141*a58d3d2aSXin Li }
142*a58d3d2aSXin Li 
test_encode(OpusEncoder * enc,int channels,int frame_size,OpusDecoder * dec)143*a58d3d2aSXin Li int test_encode(OpusEncoder *enc, int channels, int frame_size, OpusDecoder *dec)
144*a58d3d2aSXin Li {
145*a58d3d2aSXin Li    int samp_count = 0;
146*a58d3d2aSXin Li    opus_int16 *inbuf;
147*a58d3d2aSXin Li    unsigned char packet[MAX_PACKET+257];
148*a58d3d2aSXin Li    int len;
149*a58d3d2aSXin Li    opus_int16 *outbuf;
150*a58d3d2aSXin Li    int out_samples;
151*a58d3d2aSXin Li    int ret = 0;
152*a58d3d2aSXin Li #ifdef ENABLE_DRED
153*a58d3d2aSXin Li    OpusDREDDecoder *dred_dec;
154*a58d3d2aSXin Li    OpusDRED *dred;
155*a58d3d2aSXin Li    int dred_end;
156*a58d3d2aSXin Li    int dred_amount;
157*a58d3d2aSXin Li    int dred_ret;
158*a58d3d2aSXin Li    dred_dec = opus_dred_decoder_create(&dred_ret);
159*a58d3d2aSXin Li    dred = opus_dred_alloc(&dred_ret);
160*a58d3d2aSXin Li #endif
161*a58d3d2aSXin Li    /* Generate input data */
162*a58d3d2aSXin Li    inbuf = (opus_int16*)malloc(sizeof(*inbuf)*SSAMPLES);
163*a58d3d2aSXin Li    generate_music(inbuf, SSAMPLES/2);
164*a58d3d2aSXin Li 
165*a58d3d2aSXin Li    /* Allocate memory for output data */
166*a58d3d2aSXin Li    outbuf = (opus_int16*)malloc(sizeof(*outbuf)*MAX_FRAME_SAMP*3);
167*a58d3d2aSXin Li 
168*a58d3d2aSXin Li    /* Encode data, then decode for sanity check */
169*a58d3d2aSXin Li    do {
170*a58d3d2aSXin Li       len = opus_encode(enc, &inbuf[samp_count*channels], frame_size, packet, MAX_PACKET);
171*a58d3d2aSXin Li       if(len<0 || len>MAX_PACKET) {
172*a58d3d2aSXin Li          fprintf(stderr,"opus_encode() returned %d\n",len);
173*a58d3d2aSXin Li          ret = -1;
174*a58d3d2aSXin Li          break;
175*a58d3d2aSXin Li       }
176*a58d3d2aSXin Li #ifdef ENABLE_DRED
177*a58d3d2aSXin Li       dred_amount = opus_dred_parse(dred_dec, dred, packet, len, 48000, 48000, &dred_end, 0);
178*a58d3d2aSXin Li       if(dred_amount<0) {
179*a58d3d2aSXin Li          fprintf(stderr,"opus_dred_parse() returned %d\n",dred_amount);
180*a58d3d2aSXin Li          ret = -1;
181*a58d3d2aSXin Li          break;
182*a58d3d2aSXin Li       }
183*a58d3d2aSXin Li       if (dred_amount >= frame_size && (fast_rand()&1)) {
184*a58d3d2aSXin Li          dred_ret = opus_decoder_dred_decode(dec, dred, frame_size, outbuf, frame_size);
185*a58d3d2aSXin Li          if(dred_ret<0) {
186*a58d3d2aSXin Li             fprintf(stderr,"opus_decoder_dred_decode() returned %d\n",dred_ret);
187*a58d3d2aSXin Li             ret = -1;
188*a58d3d2aSXin Li             break;
189*a58d3d2aSXin Li          }
190*a58d3d2aSXin Li       }
191*a58d3d2aSXin Li #endif
192*a58d3d2aSXin Li       out_samples = opus_decode(dec, packet, len, outbuf, MAX_FRAME_SAMP, 0);
193*a58d3d2aSXin Li       if(out_samples!=frame_size) {
194*a58d3d2aSXin Li          fprintf(stderr,"opus_decode() returned %d\n",out_samples);
195*a58d3d2aSXin Li          ret = -1;
196*a58d3d2aSXin Li          break;
197*a58d3d2aSXin Li       }
198*a58d3d2aSXin Li 
199*a58d3d2aSXin Li       samp_count += frame_size;
200*a58d3d2aSXin Li    } while (samp_count < ((SSAMPLES/2)-MAX_FRAME_SAMP));
201*a58d3d2aSXin Li #ifdef ENABLE_DRED
202*a58d3d2aSXin Li    opus_dred_decoder_destroy(dred_dec);
203*a58d3d2aSXin Li    opus_dred_free(dred);
204*a58d3d2aSXin Li #endif
205*a58d3d2aSXin Li    /* Clean up */
206*a58d3d2aSXin Li    free(inbuf);
207*a58d3d2aSXin Li    free(outbuf);
208*a58d3d2aSXin Li    return ret;
209*a58d3d2aSXin Li }
210*a58d3d2aSXin Li 
fuzz_encoder_settings(const int num_encoders,const int num_setting_changes)211*a58d3d2aSXin Li void fuzz_encoder_settings(const int num_encoders, const int num_setting_changes)
212*a58d3d2aSXin Li {
213*a58d3d2aSXin Li    OpusEncoder *enc;
214*a58d3d2aSXin Li    OpusDecoder *dec;
215*a58d3d2aSXin Li    int i,j,err;
216*a58d3d2aSXin Li 
217*a58d3d2aSXin Li    /* Parameters to fuzz. Some values are duplicated to increase their probability of being tested. */
218*a58d3d2aSXin Li    int sampling_rates[5] = {8000, 12000, 16000, 24000, 48000};
219*a58d3d2aSXin Li    int channels[2] = {1, 2};
220*a58d3d2aSXin Li    int applications[3] = {OPUS_APPLICATION_AUDIO, OPUS_APPLICATION_VOIP, OPUS_APPLICATION_RESTRICTED_LOWDELAY};
221*a58d3d2aSXin Li    int bitrates[11] = {6000, 12000, 16000, 24000, 32000, 48000, 64000, 96000, 510000, OPUS_AUTO, OPUS_BITRATE_MAX};
222*a58d3d2aSXin Li    int force_channels[4] = {OPUS_AUTO, OPUS_AUTO, 1, 2};
223*a58d3d2aSXin Li    int use_vbr[3] = {0, 1, 1};
224*a58d3d2aSXin Li    int vbr_constraints[3] = {0, 1, 1};
225*a58d3d2aSXin Li    int complexities[11] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
226*a58d3d2aSXin Li    int max_bandwidths[6] = {OPUS_BANDWIDTH_NARROWBAND, OPUS_BANDWIDTH_MEDIUMBAND,
227*a58d3d2aSXin Li                             OPUS_BANDWIDTH_WIDEBAND, OPUS_BANDWIDTH_SUPERWIDEBAND,
228*a58d3d2aSXin Li                             OPUS_BANDWIDTH_FULLBAND, OPUS_BANDWIDTH_FULLBAND};
229*a58d3d2aSXin Li    int signals[4] = {OPUS_AUTO, OPUS_AUTO, OPUS_SIGNAL_VOICE, OPUS_SIGNAL_MUSIC};
230*a58d3d2aSXin Li    int inband_fecs[3] = {0, 0, 1};
231*a58d3d2aSXin Li    int packet_loss_perc[4] = {0, 1, 2, 5};
232*a58d3d2aSXin Li    int lsb_depths[2] = {8, 24};
233*a58d3d2aSXin Li    int prediction_disabled[3] = {0, 0, 1};
234*a58d3d2aSXin Li    int use_dtx[2] = {0, 1};
235*a58d3d2aSXin Li    int frame_sizes_ms_x2[9] = {5, 10, 20, 40, 80, 120, 160, 200, 240};  /* x2 to avoid 2.5 ms */
236*a58d3d2aSXin Li 
237*a58d3d2aSXin Li    for (i=0; i<num_encoders; i++) {
238*a58d3d2aSXin Li       int sampling_rate = RAND_SAMPLE(sampling_rates);
239*a58d3d2aSXin Li       int num_channels = RAND_SAMPLE(channels);
240*a58d3d2aSXin Li       int application = RAND_SAMPLE(applications);
241*a58d3d2aSXin Li 
242*a58d3d2aSXin Li       dec = opus_decoder_create(sampling_rate, num_channels, &err);
243*a58d3d2aSXin Li       if(err!=OPUS_OK || dec==NULL)test_failed();
244*a58d3d2aSXin Li 
245*a58d3d2aSXin Li       enc = opus_encoder_create(sampling_rate, num_channels, application, &err);
246*a58d3d2aSXin Li       if(err!=OPUS_OK || enc==NULL)test_failed();
247*a58d3d2aSXin Li 
248*a58d3d2aSXin Li       for (j=0; j<num_setting_changes; j++) {
249*a58d3d2aSXin Li          int bitrate = RAND_SAMPLE(bitrates);
250*a58d3d2aSXin Li          int force_channel = RAND_SAMPLE(force_channels);
251*a58d3d2aSXin Li          int vbr = RAND_SAMPLE(use_vbr);
252*a58d3d2aSXin Li          int vbr_constraint = RAND_SAMPLE(vbr_constraints);
253*a58d3d2aSXin Li          int complexity = RAND_SAMPLE(complexities);
254*a58d3d2aSXin Li          int max_bw = RAND_SAMPLE(max_bandwidths);
255*a58d3d2aSXin Li          int sig = RAND_SAMPLE(signals);
256*a58d3d2aSXin Li          int inband_fec = RAND_SAMPLE(inband_fecs);
257*a58d3d2aSXin Li          int pkt_loss = RAND_SAMPLE(packet_loss_perc);
258*a58d3d2aSXin Li          int lsb_depth = RAND_SAMPLE(lsb_depths);
259*a58d3d2aSXin Li          int pred_disabled = RAND_SAMPLE(prediction_disabled);
260*a58d3d2aSXin Li          int dtx = RAND_SAMPLE(use_dtx);
261*a58d3d2aSXin Li          int frame_size_ms_x2 = RAND_SAMPLE(frame_sizes_ms_x2);
262*a58d3d2aSXin Li          int frame_size = frame_size_ms_x2*sampling_rate/2000;
263*a58d3d2aSXin Li          int frame_size_enum = get_frame_size_enum(frame_size, sampling_rate);
264*a58d3d2aSXin Li          force_channel = IMIN(force_channel, num_channels);
265*a58d3d2aSXin Li 
266*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_BITRATE(bitrate)) != OPUS_OK) test_failed();
267*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_FORCE_CHANNELS(force_channel)) != OPUS_OK) test_failed();
268*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_VBR(vbr)) != OPUS_OK) test_failed();
269*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_VBR_CONSTRAINT(vbr_constraint)) != OPUS_OK) test_failed();
270*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_COMPLEXITY(complexity)) != OPUS_OK) test_failed();
271*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_MAX_BANDWIDTH(max_bw)) != OPUS_OK) test_failed();
272*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_SIGNAL(sig)) != OPUS_OK) test_failed();
273*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_INBAND_FEC(inband_fec)) != OPUS_OK) test_failed();
274*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_PACKET_LOSS_PERC(pkt_loss)) != OPUS_OK) test_failed();
275*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_LSB_DEPTH(lsb_depth)) != OPUS_OK) test_failed();
276*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_PREDICTION_DISABLED(pred_disabled)) != OPUS_OK) test_failed();
277*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_DTX(dtx)) != OPUS_OK) test_failed();
278*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_EXPERT_FRAME_DURATION(frame_size_enum)) != OPUS_OK) test_failed();
279*a58d3d2aSXin Li #ifdef ENABLE_DRED
280*a58d3d2aSXin Li          if(opus_encoder_ctl(enc, OPUS_SET_DRED_DURATION(fast_rand()%101)) != OPUS_OK) test_failed();
281*a58d3d2aSXin Li #endif
282*a58d3d2aSXin Li          if(test_encode(enc, num_channels, frame_size, dec)) {
283*a58d3d2aSXin Li             fprintf(stderr,
284*a58d3d2aSXin Li                "fuzz_encoder_settings: %d kHz, %d ch, application: %d, "
285*a58d3d2aSXin Li                "%d bps, force ch: %d, vbr: %d, vbr constraint: %d, complexity: %d, "
286*a58d3d2aSXin Li                "max bw: %d, signal: %d, inband fec: %d, pkt loss: %d%%, lsb depth: %d, "
287*a58d3d2aSXin Li                "pred disabled: %d, dtx: %d, (%d/2) ms\n",
288*a58d3d2aSXin Li                sampling_rate/1000, num_channels, application, bitrate,
289*a58d3d2aSXin Li                force_channel, vbr, vbr_constraint, complexity, max_bw, sig, inband_fec,
290*a58d3d2aSXin Li                pkt_loss, lsb_depth, pred_disabled, dtx, frame_size_ms_x2);
291*a58d3d2aSXin Li             test_failed();
292*a58d3d2aSXin Li          }
293*a58d3d2aSXin Li       }
294*a58d3d2aSXin Li 
295*a58d3d2aSXin Li       opus_encoder_destroy(enc);
296*a58d3d2aSXin Li       opus_decoder_destroy(dec);
297*a58d3d2aSXin Li    }
298*a58d3d2aSXin Li }
299*a58d3d2aSXin Li 
run_test1(int no_fuzz)300*a58d3d2aSXin Li int run_test1(int no_fuzz)
301*a58d3d2aSXin Li {
302*a58d3d2aSXin Li    static const int fsizes[6]={960*3,960*2,120,240,480,960};
303*a58d3d2aSXin Li    static const char *mstrings[3] = {"    LP","Hybrid","  MDCT"};
304*a58d3d2aSXin Li    unsigned char mapping[256] = {0,1,255};
305*a58d3d2aSXin Li    unsigned char db62[36];
306*a58d3d2aSXin Li    opus_int32 i,j;
307*a58d3d2aSXin Li    int rc,err;
308*a58d3d2aSXin Li    OpusEncoder *enc;
309*a58d3d2aSXin Li    OpusMSEncoder *MSenc;
310*a58d3d2aSXin Li    OpusDecoder *dec;
311*a58d3d2aSXin Li    OpusMSDecoder *MSdec;
312*a58d3d2aSXin Li    OpusMSDecoder *MSdec_err;
313*a58d3d2aSXin Li    OpusDecoder *dec_err[10];
314*a58d3d2aSXin Li    short *inbuf;
315*a58d3d2aSXin Li    short *outbuf;
316*a58d3d2aSXin Li    short *out2buf;
317*a58d3d2aSXin Li    opus_int32 bitrate_bps;
318*a58d3d2aSXin Li    unsigned char packet[MAX_PACKET+257];
319*a58d3d2aSXin Li    opus_uint32 enc_final_range;
320*a58d3d2aSXin Li    opus_uint32 dec_final_range;
321*a58d3d2aSXin Li    int fswitch;
322*a58d3d2aSXin Li    int fsize;
323*a58d3d2aSXin Li    int count;
324*a58d3d2aSXin Li 
325*a58d3d2aSXin Li   /*FIXME: encoder api tests, fs!=48k, mono, VBR*/
326*a58d3d2aSXin Li 
327*a58d3d2aSXin Li    fprintf(stdout,"  Encode+Decode tests.\n");
328*a58d3d2aSXin Li    fflush(stdout);
329*a58d3d2aSXin Li 
330*a58d3d2aSXin Li    enc = opus_encoder_create(48000, 2, OPUS_APPLICATION_VOIP, &err);
331*a58d3d2aSXin Li    if(err != OPUS_OK || enc==NULL)test_failed();
332*a58d3d2aSXin Li 
333*a58d3d2aSXin Li    for(i=0;i<2;i++)
334*a58d3d2aSXin Li    {
335*a58d3d2aSXin Li       int *ret_err;
336*a58d3d2aSXin Li       ret_err = i?0:&err;
337*a58d3d2aSXin Li       MSenc = opus_multistream_encoder_create(8000, 2, 2, 0, mapping, OPUS_UNIMPLEMENTED, ret_err);
338*a58d3d2aSXin Li       if((ret_err && *ret_err != OPUS_BAD_ARG) || MSenc!=NULL)test_failed();
339*a58d3d2aSXin Li 
340*a58d3d2aSXin Li       MSenc = opus_multistream_encoder_create(8000, 0, 1, 0, mapping, OPUS_APPLICATION_VOIP, ret_err);
341*a58d3d2aSXin Li       if((ret_err && *ret_err != OPUS_BAD_ARG) || MSenc!=NULL)test_failed();
342*a58d3d2aSXin Li 
343*a58d3d2aSXin Li       MSenc = opus_multistream_encoder_create(44100, 2, 2, 0, mapping, OPUS_APPLICATION_VOIP, ret_err);
344*a58d3d2aSXin Li       if((ret_err && *ret_err != OPUS_BAD_ARG) || MSenc!=NULL)test_failed();
345*a58d3d2aSXin Li 
346*a58d3d2aSXin Li       MSenc = opus_multistream_encoder_create(8000, 2, 2, 3, mapping, OPUS_APPLICATION_VOIP, ret_err);
347*a58d3d2aSXin Li       if((ret_err && *ret_err != OPUS_BAD_ARG) || MSenc!=NULL)test_failed();
348*a58d3d2aSXin Li 
349*a58d3d2aSXin Li       MSenc = opus_multistream_encoder_create(8000, 2, -1, 0, mapping, OPUS_APPLICATION_VOIP, ret_err);
350*a58d3d2aSXin Li       if((ret_err && *ret_err != OPUS_BAD_ARG) || MSenc!=NULL)test_failed();
351*a58d3d2aSXin Li 
352*a58d3d2aSXin Li       MSenc = opus_multistream_encoder_create(8000, 256, 2, 0, mapping, OPUS_APPLICATION_VOIP, ret_err);
353*a58d3d2aSXin Li       if((ret_err && *ret_err != OPUS_BAD_ARG) || MSenc!=NULL)test_failed();
354*a58d3d2aSXin Li    }
355*a58d3d2aSXin Li 
356*a58d3d2aSXin Li    MSenc = opus_multistream_encoder_create(8000, 2, 2, 0, mapping, OPUS_APPLICATION_AUDIO, &err);
357*a58d3d2aSXin Li    if(err != OPUS_OK || MSenc==NULL)test_failed();
358*a58d3d2aSXin Li 
359*a58d3d2aSXin Li    /*Some multistream encoder API tests*/
360*a58d3d2aSXin Li    if(opus_multistream_encoder_ctl(MSenc, OPUS_GET_BITRATE(&i))!=OPUS_OK)test_failed();
361*a58d3d2aSXin Li    if(opus_multistream_encoder_ctl(MSenc, OPUS_GET_LSB_DEPTH(&i))!=OPUS_OK)test_failed();
362*a58d3d2aSXin Li    if(i<16)test_failed();
363*a58d3d2aSXin Li 
364*a58d3d2aSXin Li    {
365*a58d3d2aSXin Li       OpusEncoder *tmp_enc;
366*a58d3d2aSXin Li       if(opus_multistream_encoder_ctl(MSenc,  OPUS_MULTISTREAM_GET_ENCODER_STATE(1,&tmp_enc))!=OPUS_OK)test_failed();
367*a58d3d2aSXin Li       if(opus_encoder_ctl(tmp_enc, OPUS_GET_LSB_DEPTH(&j))!=OPUS_OK)test_failed();
368*a58d3d2aSXin Li       if(i!=j)test_failed();
369*a58d3d2aSXin Li       if(opus_multistream_encoder_ctl(MSenc,  OPUS_MULTISTREAM_GET_ENCODER_STATE(2,&tmp_enc))!=OPUS_BAD_ARG)test_failed();
370*a58d3d2aSXin Li    }
371*a58d3d2aSXin Li 
372*a58d3d2aSXin Li    dec = opus_decoder_create(48000, 2, &err);
373*a58d3d2aSXin Li    if(err != OPUS_OK || dec==NULL)test_failed();
374*a58d3d2aSXin Li 
375*a58d3d2aSXin Li    MSdec = opus_multistream_decoder_create(48000, 2, 2, 0, mapping, &err);
376*a58d3d2aSXin Li    if(err != OPUS_OK || MSdec==NULL)test_failed();
377*a58d3d2aSXin Li 
378*a58d3d2aSXin Li    MSdec_err = opus_multistream_decoder_create(48000, 3, 2, 0, mapping, &err);
379*a58d3d2aSXin Li    if(err != OPUS_OK || MSdec_err==NULL)test_failed();
380*a58d3d2aSXin Li 
381*a58d3d2aSXin Li    dec_err[0]=(OpusDecoder *)malloc(opus_decoder_get_size(2));
382*a58d3d2aSXin Li    memcpy(dec_err[0],dec,opus_decoder_get_size(2));
383*a58d3d2aSXin Li    dec_err[1] = opus_decoder_create(48000, 1, &err);
384*a58d3d2aSXin Li    dec_err[2] = opus_decoder_create(24000, 2, &err);
385*a58d3d2aSXin Li    dec_err[3] = opus_decoder_create(24000, 1, &err);
386*a58d3d2aSXin Li    dec_err[4] = opus_decoder_create(16000, 2, &err);
387*a58d3d2aSXin Li    dec_err[5] = opus_decoder_create(16000, 1, &err);
388*a58d3d2aSXin Li    dec_err[6] = opus_decoder_create(12000, 2, &err);
389*a58d3d2aSXin Li    dec_err[7] = opus_decoder_create(12000, 1, &err);
390*a58d3d2aSXin Li    dec_err[8] = opus_decoder_create(8000, 2, &err);
391*a58d3d2aSXin Li    dec_err[9] = opus_decoder_create(8000, 1, &err);
392*a58d3d2aSXin Li    for(i=0;i<10;i++)if(dec_err[i]==NULL)test_failed();
393*a58d3d2aSXin Li 
394*a58d3d2aSXin Li    {
395*a58d3d2aSXin Li       OpusEncoder *enccpy;
396*a58d3d2aSXin Li       /*The opus state structures contain no pointers and can be freely copied*/
397*a58d3d2aSXin Li       enccpy=(OpusEncoder *)malloc(opus_encoder_get_size(2));
398*a58d3d2aSXin Li       memcpy(enccpy,enc,opus_encoder_get_size(2));
399*a58d3d2aSXin Li       memset(enc,255,opus_encoder_get_size(2));
400*a58d3d2aSXin Li       opus_encoder_destroy(enc);
401*a58d3d2aSXin Li       enc=enccpy;
402*a58d3d2aSXin Li    }
403*a58d3d2aSXin Li 
404*a58d3d2aSXin Li    inbuf=(short *)malloc(sizeof(short)*SAMPLES*2);
405*a58d3d2aSXin Li    outbuf=(short *)malloc(sizeof(short)*SAMPLES*2);
406*a58d3d2aSXin Li    out2buf=(short *)malloc(sizeof(short)*MAX_FRAME_SAMP*3);
407*a58d3d2aSXin Li    if(inbuf==NULL || outbuf==NULL || out2buf==NULL)test_failed();
408*a58d3d2aSXin Li 
409*a58d3d2aSXin Li    generate_music(inbuf,SAMPLES);
410*a58d3d2aSXin Li 
411*a58d3d2aSXin Li /*   FILE *foo;
412*a58d3d2aSXin Li    foo = fopen("foo.sw", "wb+");
413*a58d3d2aSXin Li    fwrite(inbuf, 1, SAMPLES*2*2, foo);
414*a58d3d2aSXin Li    fclose(foo);*/
415*a58d3d2aSXin Li 
416*a58d3d2aSXin Li    if(opus_encoder_ctl(enc, OPUS_SET_BANDWIDTH(OPUS_AUTO))!=OPUS_OK)test_failed();
417*a58d3d2aSXin Li    if(opus_encoder_ctl(enc, OPUS_SET_FORCE_MODE(-2))!=OPUS_BAD_ARG)test_failed();
418*a58d3d2aSXin Li    if(opus_encode(enc, inbuf, 500, packet, MAX_PACKET)!=OPUS_BAD_ARG)test_failed();
419*a58d3d2aSXin Li 
420*a58d3d2aSXin Li    for(rc=0;rc<3;rc++)
421*a58d3d2aSXin Li    {
422*a58d3d2aSXin Li       if(opus_encoder_ctl(enc, OPUS_SET_VBR(rc<2))!=OPUS_OK)test_failed();
423*a58d3d2aSXin Li       if(opus_encoder_ctl(enc, OPUS_SET_VBR_CONSTRAINT(rc==1))!=OPUS_OK)test_failed();
424*a58d3d2aSXin Li       if(opus_encoder_ctl(enc, OPUS_SET_VBR_CONSTRAINT(rc==1))!=OPUS_OK)test_failed();
425*a58d3d2aSXin Li       if(opus_encoder_ctl(enc, OPUS_SET_INBAND_FEC(rc==0))!=OPUS_OK)test_failed();
426*a58d3d2aSXin Li #ifdef ENABLE_DRED
427*a58d3d2aSXin Li       if(opus_encoder_ctl(enc, OPUS_SET_DRED_DURATION(fast_rand()%101)) != OPUS_OK) test_failed();
428*a58d3d2aSXin Li #endif
429*a58d3d2aSXin Li       for(j=0;j<13;j++)
430*a58d3d2aSXin Li       {
431*a58d3d2aSXin Li          int rate;
432*a58d3d2aSXin Li          int modes[13]={0,0,0,1,1,1,1,2,2,2,2,2,2};
433*a58d3d2aSXin Li          int rates[13]={6000,12000,48000,16000,32000,48000,64000,512000,13000,24000,48000,64000,96000};
434*a58d3d2aSXin Li          int frame[13]={960*2,960,480,960,960,960,480,960*3,960*3,960,480,240,120};
435*a58d3d2aSXin Li          rate=rates[j]+fast_rand()%rates[j];
436*a58d3d2aSXin Li          count=i=0;
437*a58d3d2aSXin Li          do {
438*a58d3d2aSXin Li             int bw,len,out_samples,frame_size;
439*a58d3d2aSXin Li             frame_size=frame[j];
440*a58d3d2aSXin Li             if((fast_rand()&255)==0)
441*a58d3d2aSXin Li             {
442*a58d3d2aSXin Li                if(opus_encoder_ctl(enc, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
443*a58d3d2aSXin Li                if(opus_decoder_ctl(dec, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
444*a58d3d2aSXin Li                if((fast_rand()&1)!=0)
445*a58d3d2aSXin Li                {
446*a58d3d2aSXin Li                   if(opus_decoder_ctl(dec_err[fast_rand()&1], OPUS_RESET_STATE)!=OPUS_OK)test_failed();
447*a58d3d2aSXin Li                }
448*a58d3d2aSXin Li             }
449*a58d3d2aSXin Li             if((fast_rand()&127)==0)
450*a58d3d2aSXin Li             {
451*a58d3d2aSXin Li                if(opus_decoder_ctl(dec_err[fast_rand()&1], OPUS_RESET_STATE)!=OPUS_OK)test_failed();
452*a58d3d2aSXin Li             }
453*a58d3d2aSXin Li             if(fast_rand()%10==0){
454*a58d3d2aSXin Li                int complex=fast_rand()%11;
455*a58d3d2aSXin Li                if(opus_encoder_ctl(enc, OPUS_SET_COMPLEXITY(complex))!=OPUS_OK)test_failed();
456*a58d3d2aSXin Li             }
457*a58d3d2aSXin Li             if(fast_rand()%50==0)opus_decoder_ctl(dec, OPUS_RESET_STATE);
458*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_INBAND_FEC(rc==0))!=OPUS_OK)test_failed();
459*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_FORCE_MODE(MODE_SILK_ONLY+modes[j]))!=OPUS_OK)test_failed();
460*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_DTX(fast_rand()&1))!=OPUS_OK)test_failed();
461*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_BITRATE(rate))!=OPUS_OK)test_failed();
462*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_FORCE_CHANNELS((rates[j]>=64000?2:1)))!=OPUS_OK)test_failed();
463*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_COMPLEXITY((count>>2)%11))!=OPUS_OK)test_failed();
464*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_PACKET_LOSS_PERC((fast_rand()&15)&(fast_rand()%15)))!=OPUS_OK)test_failed();
465*a58d3d2aSXin Li             bw=modes[j]==0?OPUS_BANDWIDTH_NARROWBAND+(fast_rand()%3):
466*a58d3d2aSXin Li                modes[j]==1?OPUS_BANDWIDTH_SUPERWIDEBAND+(fast_rand()&1):
467*a58d3d2aSXin Li                            OPUS_BANDWIDTH_NARROWBAND+(fast_rand()%5);
468*a58d3d2aSXin Li             if(modes[j]==2&&bw==OPUS_BANDWIDTH_MEDIUMBAND)bw+=3;
469*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_SET_BANDWIDTH(bw))!=OPUS_OK)test_failed();
470*a58d3d2aSXin Li             len = opus_encode(enc, &inbuf[i<<1], frame_size, packet, MAX_PACKET);
471*a58d3d2aSXin Li             if(len<0 || len>MAX_PACKET)test_failed();
472*a58d3d2aSXin Li             if(opus_encoder_ctl(enc, OPUS_GET_FINAL_RANGE(&enc_final_range))!=OPUS_OK)test_failed();
473*a58d3d2aSXin Li             if((fast_rand()&3)==0)
474*a58d3d2aSXin Li             {
475*a58d3d2aSXin Li                if(opus_packet_pad(packet,len,len+1)!=OPUS_OK)test_failed();
476*a58d3d2aSXin Li                len++;
477*a58d3d2aSXin Li             }
478*a58d3d2aSXin Li             if((fast_rand()&7)==0)
479*a58d3d2aSXin Li             {
480*a58d3d2aSXin Li                if(opus_packet_pad(packet,len,len+256)!=OPUS_OK)test_failed();
481*a58d3d2aSXin Li                len+=256;
482*a58d3d2aSXin Li             }
483*a58d3d2aSXin Li             if((fast_rand()&3)==0)
484*a58d3d2aSXin Li             {
485*a58d3d2aSXin Li                len=opus_packet_unpad(packet,len);
486*a58d3d2aSXin Li                if(len<1)test_failed();
487*a58d3d2aSXin Li             }
488*a58d3d2aSXin Li             out_samples = opus_decode(dec, packet, len, &outbuf[i<<1], MAX_FRAME_SAMP, 0);
489*a58d3d2aSXin Li             if(out_samples!=frame_size)test_failed();
490*a58d3d2aSXin Li             if(opus_decoder_ctl(dec, OPUS_GET_FINAL_RANGE(&dec_final_range))!=OPUS_OK)test_failed();
491*a58d3d2aSXin Li             if(enc_final_range!=dec_final_range)test_failed();
492*a58d3d2aSXin Li             /*LBRR decode*/
493*a58d3d2aSXin Li             out_samples = opus_decode(dec_err[0], packet, len, out2buf, frame_size, (fast_rand()&3)!=0);
494*a58d3d2aSXin Li             if(out_samples!=frame_size)test_failed();
495*a58d3d2aSXin Li             out_samples = opus_decode(dec_err[1], packet, (fast_rand()&3)==0?0:len, out2buf, MAX_FRAME_SAMP, (fast_rand()&7)!=0);
496*a58d3d2aSXin Li             if(out_samples<120)test_failed();
497*a58d3d2aSXin Li             i+=frame_size;
498*a58d3d2aSXin Li             count++;
499*a58d3d2aSXin Li          }while(i<(SSAMPLES-MAX_FRAME_SAMP));
500*a58d3d2aSXin Li          fprintf(stdout,"    Mode %s FB encode %s, %6d bps OK.\n",mstrings[modes[j]],rc==0?" VBR":rc==1?"CVBR":" CBR",rate);
501*a58d3d2aSXin Li          fflush(stdout);
502*a58d3d2aSXin Li       }
503*a58d3d2aSXin Li    }
504*a58d3d2aSXin Li 
505*a58d3d2aSXin Li    if(opus_encoder_ctl(enc, OPUS_SET_FORCE_MODE(OPUS_AUTO))!=OPUS_OK)test_failed();
506*a58d3d2aSXin Li    if(opus_encoder_ctl(enc, OPUS_SET_FORCE_CHANNELS(OPUS_AUTO))!=OPUS_OK)test_failed();
507*a58d3d2aSXin Li    if(opus_encoder_ctl(enc, OPUS_SET_INBAND_FEC(0))!=OPUS_OK)test_failed();
508*a58d3d2aSXin Li    if(opus_encoder_ctl(enc, OPUS_SET_DTX(0))!=OPUS_OK)test_failed();
509*a58d3d2aSXin Li 
510*a58d3d2aSXin Li    for(rc=0;rc<3;rc++)
511*a58d3d2aSXin Li    {
512*a58d3d2aSXin Li       if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_VBR(rc<2))!=OPUS_OK)test_failed();
513*a58d3d2aSXin Li       if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_VBR_CONSTRAINT(rc==1))!=OPUS_OK)test_failed();
514*a58d3d2aSXin Li       if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_VBR_CONSTRAINT(rc==1))!=OPUS_OK)test_failed();
515*a58d3d2aSXin Li       if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_INBAND_FEC(rc==0))!=OPUS_OK)test_failed();
516*a58d3d2aSXin Li       for(j=0;j<16;j++)
517*a58d3d2aSXin Li       {
518*a58d3d2aSXin Li          int rate;
519*a58d3d2aSXin Li          int modes[16]={0,0,0,0,0,0,0,0,2,2,2,2,2,2,2,2};
520*a58d3d2aSXin Li          int rates[16]={4000,12000,32000,8000,16000,32000,48000,88000,4000,12000,32000,8000,16000,32000,48000,88000};
521*a58d3d2aSXin Li          int frame[16]={160*1,160,80,160,160,80,40,20,160*1,160,80,160,160,80,40,20};
522*a58d3d2aSXin Li          if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_INBAND_FEC(rc==0&&j==1))!=OPUS_OK)test_failed();
523*a58d3d2aSXin Li          if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_FORCE_MODE(MODE_SILK_ONLY+modes[j]))!=OPUS_OK)test_failed();
524*a58d3d2aSXin Li          rate=rates[j]+fast_rand()%rates[j];
525*a58d3d2aSXin Li          if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_DTX(fast_rand()&1))!=OPUS_OK)test_failed();
526*a58d3d2aSXin Li          if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_BITRATE(rate))!=OPUS_OK)test_failed();
527*a58d3d2aSXin Li          count=i=0;
528*a58d3d2aSXin Li          do {
529*a58d3d2aSXin Li             int len,out_samples,frame_size,loss;
530*a58d3d2aSXin Li             opus_int32 pred;
531*a58d3d2aSXin Li             if(opus_multistream_encoder_ctl(MSenc, OPUS_GET_PREDICTION_DISABLED(&pred))!=OPUS_OK)test_failed();
532*a58d3d2aSXin Li             if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_PREDICTION_DISABLED((int)(fast_rand()&15)<(pred?11:4)))!=OPUS_OK)test_failed();
533*a58d3d2aSXin Li             frame_size=frame[j];
534*a58d3d2aSXin Li             if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_COMPLEXITY((count>>2)%11))!=OPUS_OK)test_failed();
535*a58d3d2aSXin Li             if(opus_multistream_encoder_ctl(MSenc, OPUS_SET_PACKET_LOSS_PERC((fast_rand()&15)&(fast_rand()%15)))!=OPUS_OK)test_failed();
536*a58d3d2aSXin Li             if((fast_rand()&255)==0)
537*a58d3d2aSXin Li             {
538*a58d3d2aSXin Li                if(opus_multistream_encoder_ctl(MSenc, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
539*a58d3d2aSXin Li                if(opus_multistream_decoder_ctl(MSdec, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
540*a58d3d2aSXin Li                if((fast_rand()&3)!=0)
541*a58d3d2aSXin Li                {
542*a58d3d2aSXin Li                   if(opus_multistream_decoder_ctl(MSdec_err, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
543*a58d3d2aSXin Li                }
544*a58d3d2aSXin Li             }
545*a58d3d2aSXin Li             if((fast_rand()&255)==0)
546*a58d3d2aSXin Li             {
547*a58d3d2aSXin Li                if(opus_multistream_decoder_ctl(MSdec_err, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
548*a58d3d2aSXin Li             }
549*a58d3d2aSXin Li             len = opus_multistream_encode(MSenc, &inbuf[i<<1], frame_size, packet, MAX_PACKET);
550*a58d3d2aSXin Li             if(len<0 || len>MAX_PACKET)test_failed();
551*a58d3d2aSXin Li             if(opus_multistream_encoder_ctl(MSenc, OPUS_GET_FINAL_RANGE(&enc_final_range))!=OPUS_OK)test_failed();
552*a58d3d2aSXin Li             if((fast_rand()&3)==0)
553*a58d3d2aSXin Li             {
554*a58d3d2aSXin Li                if(opus_multistream_packet_pad(packet,len,len+1,2)!=OPUS_OK)test_failed();
555*a58d3d2aSXin Li                len++;
556*a58d3d2aSXin Li             }
557*a58d3d2aSXin Li             if((fast_rand()&7)==0)
558*a58d3d2aSXin Li             {
559*a58d3d2aSXin Li                if(opus_multistream_packet_pad(packet,len,len+256,2)!=OPUS_OK)test_failed();
560*a58d3d2aSXin Li                len+=256;
561*a58d3d2aSXin Li             }
562*a58d3d2aSXin Li             if((fast_rand()&3)==0)
563*a58d3d2aSXin Li             {
564*a58d3d2aSXin Li                len=opus_multistream_packet_unpad(packet,len,2);
565*a58d3d2aSXin Li                if(len<1)test_failed();
566*a58d3d2aSXin Li             }
567*a58d3d2aSXin Li             out_samples = opus_multistream_decode(MSdec, packet, len, out2buf, MAX_FRAME_SAMP, 0);
568*a58d3d2aSXin Li             if(out_samples!=frame_size*6)test_failed();
569*a58d3d2aSXin Li             if(opus_multistream_decoder_ctl(MSdec, OPUS_GET_FINAL_RANGE(&dec_final_range))!=OPUS_OK)test_failed();
570*a58d3d2aSXin Li             if(enc_final_range!=dec_final_range)test_failed();
571*a58d3d2aSXin Li             /*LBRR decode*/
572*a58d3d2aSXin Li             loss=(fast_rand()&63)==0;
573*a58d3d2aSXin Li             out_samples = opus_multistream_decode(MSdec_err, packet, loss?0:len, out2buf, frame_size*6, (fast_rand()&3)!=0);
574*a58d3d2aSXin Li             if(out_samples!=(frame_size*6))test_failed();
575*a58d3d2aSXin Li             i+=frame_size;
576*a58d3d2aSXin Li             count++;
577*a58d3d2aSXin Li          }while(i<(SSAMPLES/12-MAX_FRAME_SAMP));
578*a58d3d2aSXin Li          fprintf(stdout,"    Mode %s NB dual-mono MS encode %s, %6d bps OK.\n",mstrings[modes[j]],rc==0?" VBR":rc==1?"CVBR":" CBR",rate);
579*a58d3d2aSXin Li          fflush(stdout);
580*a58d3d2aSXin Li       }
581*a58d3d2aSXin Li    }
582*a58d3d2aSXin Li 
583*a58d3d2aSXin Li    bitrate_bps=512000;
584*a58d3d2aSXin Li    fsize=fast_rand()%31;
585*a58d3d2aSXin Li    fswitch=100;
586*a58d3d2aSXin Li 
587*a58d3d2aSXin Li    debruijn2(6,db62);
588*a58d3d2aSXin Li    count=i=0;
589*a58d3d2aSXin Li    do {
590*a58d3d2aSXin Li       unsigned char toc;
591*a58d3d2aSXin Li       const unsigned char *frames[48];
592*a58d3d2aSXin Li       short size[48];
593*a58d3d2aSXin Li       int payload_offset;
594*a58d3d2aSXin Li       opus_uint32 dec_final_range2;
595*a58d3d2aSXin Li       int jj,dec2;
596*a58d3d2aSXin Li       int len,out_samples;
597*a58d3d2aSXin Li       int frame_size=fsizes[db62[fsize]];
598*a58d3d2aSXin Li       opus_int32 offset=i%(SAMPLES-MAX_FRAME_SAMP);
599*a58d3d2aSXin Li 
600*a58d3d2aSXin Li       opus_encoder_ctl(enc, OPUS_SET_BITRATE(bitrate_bps));
601*a58d3d2aSXin Li 
602*a58d3d2aSXin Li       len = opus_encode(enc, &inbuf[offset<<1], frame_size, packet, MAX_PACKET);
603*a58d3d2aSXin Li       if(len<0 || len>MAX_PACKET)test_failed();
604*a58d3d2aSXin Li       count++;
605*a58d3d2aSXin Li 
606*a58d3d2aSXin Li       opus_encoder_ctl(enc, OPUS_GET_FINAL_RANGE(&enc_final_range));
607*a58d3d2aSXin Li 
608*a58d3d2aSXin Li       out_samples = opus_decode(dec, packet, len, &outbuf[offset<<1], MAX_FRAME_SAMP, 0);
609*a58d3d2aSXin Li       if(out_samples!=frame_size)test_failed();
610*a58d3d2aSXin Li 
611*a58d3d2aSXin Li       opus_decoder_ctl(dec, OPUS_GET_FINAL_RANGE(&dec_final_range));
612*a58d3d2aSXin Li 
613*a58d3d2aSXin Li       /* compare final range encoder rng values of encoder and decoder */
614*a58d3d2aSXin Li       if(dec_final_range!=enc_final_range)test_failed();
615*a58d3d2aSXin Li 
616*a58d3d2aSXin Li       /* We fuzz the packet, but take care not to only corrupt the payload
617*a58d3d2aSXin Li          Corrupted headers are tested elsewhere and we need to actually run
618*a58d3d2aSXin Li          the decoders in order to compare them. */
619*a58d3d2aSXin Li       if(opus_packet_parse(packet,len,&toc,frames,size,&payload_offset)<=0)test_failed();
620*a58d3d2aSXin Li       if((fast_rand()&1023)==0)len=0;
621*a58d3d2aSXin Li       for(j=(opus_int32)(frames[0]-packet);j<len;j++)for(jj=0;jj<8;jj++)packet[j]^=((!no_fuzz)&&((fast_rand()&1023)==0))<<jj;
622*a58d3d2aSXin Li       out_samples = opus_decode(dec_err[0], len>0?packet:NULL, len, out2buf, MAX_FRAME_SAMP, 0);
623*a58d3d2aSXin Li       if(out_samples<0||out_samples>MAX_FRAME_SAMP)test_failed();
624*a58d3d2aSXin Li       if((len>0&&out_samples!=frame_size))test_failed(); /*FIXME use lastframe*/
625*a58d3d2aSXin Li 
626*a58d3d2aSXin Li       opus_decoder_ctl(dec_err[0], OPUS_GET_FINAL_RANGE(&dec_final_range));
627*a58d3d2aSXin Li 
628*a58d3d2aSXin Li       /*randomly select one of the decoders to compare with*/
629*a58d3d2aSXin Li       dec2=fast_rand()%9+1;
630*a58d3d2aSXin Li       out_samples = opus_decode(dec_err[dec2], len>0?packet:NULL, len, out2buf, MAX_FRAME_SAMP, 0);
631*a58d3d2aSXin Li       if(out_samples<0||out_samples>MAX_FRAME_SAMP)test_failed(); /*FIXME, use factor, lastframe for loss*/
632*a58d3d2aSXin Li 
633*a58d3d2aSXin Li       opus_decoder_ctl(dec_err[dec2], OPUS_GET_FINAL_RANGE(&dec_final_range2));
634*a58d3d2aSXin Li       if(len>0&&dec_final_range!=dec_final_range2)test_failed();
635*a58d3d2aSXin Li 
636*a58d3d2aSXin Li       fswitch--;
637*a58d3d2aSXin Li       if(fswitch<1)
638*a58d3d2aSXin Li       {
639*a58d3d2aSXin Li         int new_size;
640*a58d3d2aSXin Li         fsize=(fsize+1)%36;
641*a58d3d2aSXin Li         new_size=fsizes[db62[fsize]];
642*a58d3d2aSXin Li         if(new_size==960||new_size==480)fswitch=2880/new_size*(fast_rand()%19+1);
643*a58d3d2aSXin Li         else fswitch=(fast_rand()%(2880/new_size))+1;
644*a58d3d2aSXin Li       }
645*a58d3d2aSXin Li       bitrate_bps=((fast_rand()%508000+4000)+bitrate_bps)>>1;
646*a58d3d2aSXin Li       i+=frame_size;
647*a58d3d2aSXin Li    }while(i<SAMPLES*4);
648*a58d3d2aSXin Li    fprintf(stdout,"    All framesize pairs switching encode, %d frames OK.\n",count);
649*a58d3d2aSXin Li    fflush(stdout);
650*a58d3d2aSXin Li 
651*a58d3d2aSXin Li    if(opus_encoder_ctl(enc, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
652*a58d3d2aSXin Li    opus_encoder_destroy(enc);
653*a58d3d2aSXin Li    if(opus_multistream_encoder_ctl(MSenc, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
654*a58d3d2aSXin Li    opus_multistream_encoder_destroy(MSenc);
655*a58d3d2aSXin Li    if(opus_decoder_ctl(dec, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
656*a58d3d2aSXin Li    opus_decoder_destroy(dec);
657*a58d3d2aSXin Li    if(opus_multistream_decoder_ctl(MSdec, OPUS_RESET_STATE)!=OPUS_OK)test_failed();
658*a58d3d2aSXin Li    opus_multistream_decoder_destroy(MSdec);
659*a58d3d2aSXin Li    opus_multistream_decoder_destroy(MSdec_err);
660*a58d3d2aSXin Li    for(i=0;i<10;i++)opus_decoder_destroy(dec_err[i]);
661*a58d3d2aSXin Li    free(inbuf);
662*a58d3d2aSXin Li    free(outbuf);
663*a58d3d2aSXin Li    free(out2buf);
664*a58d3d2aSXin Li    return 0;
665*a58d3d2aSXin Li }
666*a58d3d2aSXin Li 
print_usage(char * _argv[])667*a58d3d2aSXin Li void print_usage(char* _argv[])
668*a58d3d2aSXin Li {
669*a58d3d2aSXin Li    fprintf(stderr,"Usage: %s [<seed>] [-fuzz <num_encoders> <num_settings_per_encoder>]\n",_argv[0]);
670*a58d3d2aSXin Li }
671*a58d3d2aSXin Li 
main(int _argc,char ** _argv)672*a58d3d2aSXin Li int main(int _argc, char **_argv)
673*a58d3d2aSXin Li {
674*a58d3d2aSXin Li    int args=1;
675*a58d3d2aSXin Li    char * strtol_str=NULL;
676*a58d3d2aSXin Li    const char * oversion;
677*a58d3d2aSXin Li    const char * env_seed;
678*a58d3d2aSXin Li    int env_used;
679*a58d3d2aSXin Li    int num_encoders_to_fuzz=5;
680*a58d3d2aSXin Li    int num_setting_changes=40;
681*a58d3d2aSXin Li 
682*a58d3d2aSXin Li    env_used=0;
683*a58d3d2aSXin Li    env_seed=getenv("SEED");
684*a58d3d2aSXin Li    if(_argc>1)
685*a58d3d2aSXin Li       iseed=strtol(_argv[1], &strtol_str, 10);  /* the first input argument might be the seed */
686*a58d3d2aSXin Li    if(strtol_str!=NULL && strtol_str[0]=='\0')   /* iseed is a valid number */
687*a58d3d2aSXin Li       args++;
688*a58d3d2aSXin Li    else if(env_seed) {
689*a58d3d2aSXin Li       iseed=atoi(env_seed);
690*a58d3d2aSXin Li       env_used=1;
691*a58d3d2aSXin Li    }
692*a58d3d2aSXin Li    else iseed=(opus_uint32)time(NULL)^(((opus_uint32)getpid()&65535)<<16);
693*a58d3d2aSXin Li    Rw=Rz=iseed;
694*a58d3d2aSXin Li 
695*a58d3d2aSXin Li    while(args<_argc)
696*a58d3d2aSXin Li    {
697*a58d3d2aSXin Li       if(strcmp(_argv[args], "-fuzz")==0 && _argc==(args+3)) {
698*a58d3d2aSXin Li          num_encoders_to_fuzz=strtol(_argv[args+1], &strtol_str, 10);
699*a58d3d2aSXin Li          if(strtol_str[0]!='\0' || num_encoders_to_fuzz<=0) {
700*a58d3d2aSXin Li             print_usage(_argv);
701*a58d3d2aSXin Li             return EXIT_FAILURE;
702*a58d3d2aSXin Li          }
703*a58d3d2aSXin Li          num_setting_changes=strtol(_argv[args+2], &strtol_str, 10);
704*a58d3d2aSXin Li          if(strtol_str[0]!='\0' || num_setting_changes<=0) {
705*a58d3d2aSXin Li             print_usage(_argv);
706*a58d3d2aSXin Li             return EXIT_FAILURE;
707*a58d3d2aSXin Li          }
708*a58d3d2aSXin Li          args+=3;
709*a58d3d2aSXin Li       }
710*a58d3d2aSXin Li       else {
711*a58d3d2aSXin Li         print_usage(_argv);
712*a58d3d2aSXin Li         return EXIT_FAILURE;
713*a58d3d2aSXin Li       }
714*a58d3d2aSXin Li    }
715*a58d3d2aSXin Li 
716*a58d3d2aSXin Li    oversion=opus_get_version_string();
717*a58d3d2aSXin Li    if(!oversion)test_failed();
718*a58d3d2aSXin Li    fprintf(stderr,"Testing %s encoder. Random seed: %u (%.4X)\n", oversion, iseed, fast_rand() % 65535);
719*a58d3d2aSXin Li    if(env_used)fprintf(stderr,"  Random seed set from the environment (SEED=%s).\n", env_seed);
720*a58d3d2aSXin Li 
721*a58d3d2aSXin Li    regression_test();
722*a58d3d2aSXin Li 
723*a58d3d2aSXin Li    /*Setting TEST_OPUS_NOFUZZ tells the tool not to send garbage data
724*a58d3d2aSXin Li      into the decoders. This is helpful because garbage data
725*a58d3d2aSXin Li      may cause the decoders to clip, which angers CLANG IOC.*/
726*a58d3d2aSXin Li    run_test1(getenv("TEST_OPUS_NOFUZZ")!=NULL);
727*a58d3d2aSXin Li 
728*a58d3d2aSXin Li    /* Fuzz encoder settings online */
729*a58d3d2aSXin Li    if(getenv("TEST_OPUS_NOFUZZ")==NULL) {
730*a58d3d2aSXin Li       fprintf(stderr,"Running fuzz_encoder_settings with %d encoder(s) and %d setting change(s) each.\n",
731*a58d3d2aSXin Li               num_encoders_to_fuzz, num_setting_changes);
732*a58d3d2aSXin Li       fuzz_encoder_settings(num_encoders_to_fuzz, num_setting_changes);
733*a58d3d2aSXin Li    }
734*a58d3d2aSXin Li 
735*a58d3d2aSXin Li    fprintf(stderr,"Tests completed successfully.\n");
736*a58d3d2aSXin Li 
737*a58d3d2aSXin Li    return 0;
738*a58d3d2aSXin Li }
739