Implements CELT_SET_LOSS_PERC
[opus.git] / libcelt / testcelt.c
1 /* Copyright (c) 2007-2008 CSIRO
2    Copyright (c) 2007-2009 Xiph.Org Foundation
3    Written by Jean-Marc Valin */
4 /*
5    Redistribution and use in source and binary forms, with or without
6    modification, are permitted provided that the following conditions
7    are met:
8    
9    - Redistributions of source code must retain the above copyright
10    notice, this list of conditions and the following disclaimer.
11    
12    - Redistributions in binary form must reproduce the above copyright
13    notice, this list of conditions and the following disclaimer in the
14    documentation and/or other materials provided with the distribution.
15    
16    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19    A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
20    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
21    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
22    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
23    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
24    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
25    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
26    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 #ifdef HAVE_CONFIG_H
30 #include "config.h"
31 #endif
32
33 #include "celt.h"
34 #include "arch.h"
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <math.h>
38 #include <string.h>
39
40 #define MAX_PACKET 1275
41
42 int main(int argc, char *argv[])
43 {
44    int err;
45    char *inFile, *outFile;
46    FILE *fin, *fout;
47    CELTMode *mode=NULL;
48    CELTEncoder *enc;
49    CELTDecoder *dec;
50    int len;
51    celt_int32 frame_size, channels;
52    int bytes_per_packet;
53    unsigned char data[MAX_PACKET];
54    int rate;
55    int complexity;
56 #if !(defined (FIXED_POINT) && !defined(CUSTOM_MODES)) && defined(RESYNTH)
57    int i;
58    double rmsd = 0;
59 #endif
60    int count = 0;
61    celt_int32 skip;
62    celt_int16 *in, *out;
63    if (argc != 9 && argc != 8 && argc != 7)
64    {
65       fprintf (stderr, "Usage: testcelt <rate> <channels> <frame size> "
66                " <bytes per packet> [<complexity> [packet loss rate]] "
67                "<input> <output>\n");
68       return 1;
69    }
70    
71    rate = atoi(argv[1]);
72    channels = atoi(argv[2]);
73    frame_size = atoi(argv[3]);
74    mode = celt_mode_create(rate, frame_size, NULL);
75    if (mode == NULL)
76    {
77       fprintf(stderr, "failed to create a mode\n");
78       return 1;
79    }
80
81    bytes_per_packet = atoi(argv[4]);
82    if (bytes_per_packet < 0 || bytes_per_packet > MAX_PACKET)
83    {
84       fprintf (stderr, "bytes per packet must be between 0 and %d\n",
85                         MAX_PACKET);
86       return 1;
87    }
88
89    inFile = argv[argc-2];
90    fin = fopen(inFile, "rb");
91    if (!fin)
92    {
93       fprintf (stderr, "Could not open input file %s\n", argv[argc-2]);
94       return 1;
95    }
96    outFile = argv[argc-1];
97    fout = fopen(outFile, "wb+");
98    if (!fout)
99    {
100       fprintf (stderr, "Could not open output file %s\n", argv[argc-1]);
101       return 1;
102    }
103    
104    enc = celt_encoder_create_custom(mode, channels, &err);
105    if (err != 0)
106    {
107       fprintf(stderr, "Failed to create the encoder: %s\n", celt_strerror(err));
108       return 1;
109    }
110    dec = celt_decoder_create_custom(mode, channels, &err);
111    if (err != 0)
112    {
113       fprintf(stderr, "Failed to create the decoder: %s\n", celt_strerror(err));
114       return 1;
115    }
116    celt_decoder_ctl(dec, CELT_GET_LOOKAHEAD(&skip));
117
118    if (argc>7)
119    {
120       complexity=atoi(argv[5]);
121       celt_encoder_ctl(enc,CELT_SET_COMPLEXITY(complexity));
122    }
123    
124    in = (celt_int16*)malloc(frame_size*channels*sizeof(celt_int16));
125    out = (celt_int16*)malloc(frame_size*channels*sizeof(celt_int16));
126
127    while (!feof(fin))
128    {
129       int ret;
130       err = fread(in, sizeof(short), frame_size*channels, fin);
131       if (feof(fin))
132          break;
133       len = celt_encode(enc, in, frame_size, data, bytes_per_packet);
134       if (len <= 0)
135          fprintf (stderr, "celt_encode() failed: %s\n", celt_strerror(len));
136
137       /* This is for simulating bit errors */
138 #if 0
139       int errors = 0;
140       int eid = 0;
141       /* This simulates random bit error */
142       for (i=0;i<len*8;i++)
143       {
144          if (rand()%atoi(argv[8])==0)
145          {
146             if (i<64)
147             {
148                errors++;
149                eid = i;
150             }
151             data[i/8] ^= 1<<(7-(i%8));
152          }
153       }
154       if (errors == 1)
155          data[eid/8] ^= 1<<(7-(eid%8));
156       else if (errors%2 == 1)
157          data[rand()%8] ^= 1<<rand()%8;
158 #endif
159
160 #if 1 /* Set to zero to use the encoder's output instead */
161       /* This is to simulate packet loss */
162       if (argc==9 && rand()%1000<atoi(argv[argc-3]))
163       /*if (errors && (errors%2==0))*/
164          ret = celt_decode(dec, NULL, len, out, frame_size);
165       else
166          ret = celt_decode(dec, data, len, out, frame_size);
167       if (ret < 0)
168          fprintf(stderr, "celt_decode() failed: %s\n", celt_strerror(ret));
169 #else
170       for (i=0;i<ret*channels;i++)
171          out[i] = in[i];
172 #endif
173 #if !(defined (FIXED_POINT) && !defined(CUSTOM_MODES)) && defined(RESYNTH)
174       for (i=0;i<ret*channels;i++)
175       {
176          rmsd += (in[i]-out[i])*1.0*(in[i]-out[i]);
177          /*out[i] -= in[i];*/
178       }
179 #endif
180       count++;
181       fwrite(out+skip*channels, sizeof(short), (ret-skip)*channels, fout);
182       skip = 0;
183    }
184    PRINT_MIPS(stderr);
185    
186    celt_encoder_destroy(enc);
187    celt_decoder_destroy(dec);
188    fclose(fin);
189    fclose(fout);
190    celt_mode_destroy(mode);
191    free(in);
192    free(out);
193 #if !(defined (FIXED_POINT) && !defined(CUSTOM_MODES)) && defined(RESYNTH)
194    if (rmsd > 0)
195    {
196       rmsd = sqrt(rmsd/(1.0*frame_size*channels*count));
197       fprintf (stderr, "Error: encoder doesn't match decoder\n");
198       fprintf (stderr, "RMS mismatch is %f\n", rmsd);
199       return 1;
200    } else {
201       fprintf (stderr, "Encoder matches decoder!!\n");
202    }
203 #endif
204    return 0;
205 }
206