Version update 1.2.0 -> 1.3.0 (fix crash on 64bit)

[platform/upstream/opencore-amr.git] / test / amrnb-enc.c

diff --git a/test/amrnb-enc.c b/test/amrnb-enc.c
index 9a16098..6fc5d40 100644 (file)
--- a/test/amrnb-enc.c
+++ b/test/amrnb-enc.c
@@ -18,32 +18,127 @@
 
 #include <stdio.h>
 #include <stdint.h>
-#include <math.h>
 #include <interf_enc.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include "wavreader.h"
+
+void usage(const char* name) {
+       fprintf(stderr, "%s [-r bitrate] [-d] in.wav out.amr\n", name);
+}
+
+enum Mode findMode(const char* str) {
+       struct {
+               enum Mode mode;
+               int rate;
+       } modes[] = {
+               { MR475,  4750 },
+               { MR515,  5150 },
+               { MR59,   5900 },
+               { MR67,   6700 },
+               { MR74,   7400 },
+               { MR795,  7950 },
+               { MR102, 10200 },
+               { MR122, 12200 }
+       };
+       int rate = atoi(str);
+       int closest = -1;
+       int closestdiff = 0;
+       unsigned int i;
+       for (i = 0; i < sizeof(modes)/sizeof(modes[0]); i++) {
+               if (modes[i].rate == rate)
+                       return modes[i].mode;
+               if (closest < 0 || closestdiff > abs(modes[i].rate - rate)) {
+                       closest = i;
+                       closestdiff = abs(modes[i].rate - rate);
+               }
+       }
+       fprintf(stderr, "Using bitrate %d\n", modes[closest].rate);
+       return modes[closest].mode;
+}
 
 int main(int argc, char *argv[]) {
-       int i, j;
-       void* amr;
-       FILE* out;
-       int sample_pos = 0;
+       enum Mode mode = MR122;
+       int ch, dtx = 0;
+       const char *infile, *outfile;
+       FILE *out;
+       void *wav, *amr;
+       int format, sampleRate, channels, bitsPerSample;
+       int inputSize;
+       uint8_t* inputBuf;
+       while ((ch = getopt(argc, argv, "r:d")) != -1) {
+               switch (ch) {
+               case 'r':
+                       mode = findMode(optarg);
+                       break;
+               case 'd':
+                       dtx = 1;
+                       break;
+               case '?':
+               default:
+                       usage(argv[0]);
+                       return 1;
+               }
+       }
+       if (argc - optind < 2) {
+               usage(argv[0]);
+               return 1;
+       }
+       infile = argv[optind];
+       outfile = argv[optind + 1];
 
-       amr = Encoder_Interface_init(0);
-       out = fopen("out.amr", "wb");
+       wav = wav_read_open(infile);
+       if (!wav) {
+               fprintf(stderr, "Unable to open wav file %s\n", infile);
+               return 1;
+       }
+       if (!wav_get_header(wav, &format, &channels, &sampleRate, &bitsPerSample, NULL)) {
+               fprintf(stderr, "Bad wav file %s\n", infile);
+               return 1;
+       }
+       if (format != 1) {
+               fprintf(stderr, "Unsupported WAV format %d\n", format);
+               return 1;
+       }
+       if (bitsPerSample != 16) {
+               fprintf(stderr, "Unsupported WAV sample depth %d\n", bitsPerSample);
+               return 1;
+       }
+       if (channels != 1)
+               fprintf(stderr, "Warning, only compressing one audio channel\n");
+       if (sampleRate != 8000)
+               fprintf(stderr, "Warning, AMR-NB uses 8000 Hz sample rate (WAV file has %d Hz)\n", sampleRate);
+       inputSize = channels*2*160;
+       inputBuf = (uint8_t*) malloc(inputSize);
+
+       amr = Encoder_Interface_init(dtx);
+       out = fopen(outfile, "wb");
+       if (!out) {
+               perror(outfile);
+               return 1;
+       }
 
        fwrite("#!AMR\n", 1, 6, out);
-       for (i = 0; i < 1000; i++) {
+       while (1) {
                short buf[160];
                uint8_t outbuf[500];
-               int n;
-               for (j = 0; j < 160; j++) {
-                       buf[j] = 32767*sin(440*2*3.141592654*sample_pos/8000);
-                       sample_pos++;
+               int read, i, n;
+               read = wav_read_data(wav, inputBuf, inputSize);
+               read /= channels;
+               read /= 2;
+               if (read < 160)
+                       break;
+               for (i = 0; i < 160; i++) {
+                       const uint8_t* in = &inputBuf[2*channels*i];
+                       buf[i] = in[0] | (in[1] << 8);
                }
-               n = Encoder_Interface_Encode(amr, MR475, buf, outbuf, 0);
+               n = Encoder_Interface_Encode(amr, mode, buf, outbuf, 0);
                fwrite(outbuf, 1, n, out);
        }
+       free(inputBuf);
        fclose(out);
        Encoder_Interface_exit(amr);
+       wav_read_close(wav);
 
        return 0;
 }