src/test_opus_custom.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 "opus_custom.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    opus_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    opus_int32 skip;
  62    opus_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 = opus_custom_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 = opus_custom_encoder_create(mode, channels, &err);
 105    if (err != 0)
 106    {
 107       fprintf(stderr, "Failed to create the encoder: %s\n", opus_strerror(err));
 108       return 1;
 109    }
 110    dec = opus_custom_decoder_create(mode, channels, &err);
 111    if (err != 0)
 112    {
 113       fprintf(stderr, "Failed to create the decoder: %s\n", opus_strerror(err));
 114       return 1;
 115    }
 116    opus_custom_decoder_ctl(dec, CELT_GET_LOOKAHEAD(&skip));
 117
 118    if (argc>7)
 119    {
 120       complexity=atoi(argv[5]);
 121       opus_custom_encoder_ctl(enc,CELT_SET_COMPLEXITY(complexity));
 122    }
 123
 124    in = (opus_int16*)malloc(frame_size*channels*sizeof(opus_int16));
 125    out = (opus_int16*)malloc(frame_size*channels*sizeof(opus_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 = opus_custom_encode(enc, in, frame_size, data, bytes_per_packet);
 134       if (len <= 0)
 135          fprintf (stderr, "opus_custom_encode() failed: %s\n", opus_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 = opus_custom_decode(dec, NULL, len, out, frame_size);
 165       else
 166          ret = opus_custom_decode(dec, data, len, out, frame_size);
 167       if (ret < 0)
 168          fprintf(stderr, "opus_custom_decode() failed: %s\n", opus_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    opus_custom_encoder_destroy(enc);
 187    opus_custom_decoder_destroy(dec);
 188    fclose(fin);
 189    fclose(fout);
 190    opus_custom_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