1 /*
2 * Copyright (C) 2014 The Android Open Source Project
3 * All rights reserved.
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 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in
12 * the documentation and/or other materials provided with the
13 * distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <stdio.h>
30 #include <stdint.h>
31 #include <string.h>
32 #include <assert.h>
33 #include <stdlib.h>
34
35 #include "pvamrwbdecoder.h"
36 #include <audio_utils/sndfile.h>
37
38 // Constants for AMR-WB.
39 enum {
40 kInputBufferSize = 64,
41 kSamplesPerFrame = 320,
42 kBitsPerSample = 16,
43 kOutputBufferSize = kSamplesPerFrame * kBitsPerSample/8,
44 kSampleRate = 16000,
45 kChannels = 1,
46 kFileHeaderSize = 9,
47 kMaxSourceDataUnitSize = 477 * sizeof(int16_t)
48 };
49
50 const uint32_t kFrameSizes[] = { 17, 23, 32, 36, 40, 46, 50, 58, 60 };
51
main(int argc,char * argv[])52 int main(int argc, char *argv[]) {
53
54 if (argc != 3) {
55 fprintf(stderr, "Usage %s <input file> <output file>\n", argv[0]);
56 return EXIT_FAILURE;
57 }
58
59 // Open the input file.
60 FILE* fpInput = fopen(argv[1], "rb");
61 if (fpInput == NULL) {
62 fprintf(stderr, "Could not open %s\n", argv[1]);
63 return EXIT_FAILURE;
64 }
65
66 // Validate the input AMR file.
67 char header[kFileHeaderSize];
68 int bytesRead = fread(header, 1, kFileHeaderSize, fpInput);
69 if ((bytesRead != kFileHeaderSize) ||
70 (memcmp(header, "#!AMR-WB\n", kFileHeaderSize) != 0)) {
71 fprintf(stderr, "Invalid AMR-WB file\n");
72 fclose(fpInput);
73 return EXIT_FAILURE;
74 }
75
76 // Open the output file.
77 SF_INFO sfInfo;
78 memset(&sfInfo, 0, sizeof(SF_INFO));
79 sfInfo.channels = kChannels;
80 sfInfo.format = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
81 sfInfo.samplerate = kSampleRate;
82 SNDFILE *handle = sf_open(argv[2], SFM_WRITE, &sfInfo);
83 if (handle == NULL) {
84 fprintf(stderr, "Could not create %s\n", argv[2]);
85 fclose(fpInput);
86 return EXIT_FAILURE;
87 }
88
89 // Allocate the decoder memory.
90 uint32_t memRequirements = pvDecoder_AmrWbMemRequirements();
91 void *decoderBuf = malloc(memRequirements);
92 assert(decoderBuf != NULL);
93
94 // Create AMR-WB decoder instance.
95 void *amrHandle;
96 int16_t *decoderCookie;
97 pvDecoder_AmrWb_Init(&amrHandle, decoderBuf, &decoderCookie);
98
99 // Allocate input buffer.
100 uint8_t *inputBuf = (uint8_t*) malloc(kInputBufferSize);
101 assert(inputBuf != NULL);
102
103 // Allocate input sample buffer.
104 int16_t *inputSampleBuf = (int16_t*) malloc(kMaxSourceDataUnitSize);
105 assert(inputSampleBuf != NULL);
106
107 // Allocate output buffer.
108 int16_t *outputBuf = (int16_t*) malloc(kOutputBufferSize);
109 assert(outputBuf != NULL);
110
111 // Decode loop.
112 int retVal = EXIT_SUCCESS;
113 RX_State_wb rx_state{};
114 while (1) {
115 // Read mode.
116 uint8_t modeByte;
117 bytesRead = fread(&modeByte, 1, 1, fpInput);
118 if (bytesRead != 1) break;
119 int16 mode = ((modeByte >> 3) & 0x0f);
120
121 // AMR-WB file format cannot have mode 10, 11, 12 and 13.
122 if (mode >= 10 && mode <= 13) {
123 fprintf(stderr, "Encountered illegal frame type %d\n", mode);
124 retVal = EXIT_FAILURE;
125 break;
126 }
127
128 if (mode >= 9) {
129 // Produce silence for comfort noise, speech lost and no data.
130 memset(outputBuf, 0, kOutputBufferSize);
131 } else /* if (mode < 9) */ {
132 // Read rest of the frame.
133 int32_t frameSize = kFrameSizes[mode];
134 bytesRead = fread(inputBuf, 1, frameSize, fpInput);
135 if (bytesRead != frameSize) break;
136
137 int16 frameType, frameMode;
138 frameMode = mode;
139 mime_unsorting(
140 (uint8_t *)inputBuf,
141 inputSampleBuf,
142 &frameType, &frameMode, 1, &rx_state);
143
144 int16_t numSamplesOutput;
145 pvDecoder_AmrWb(
146 frameMode, inputSampleBuf,
147 outputBuf,
148 &numSamplesOutput,
149 decoderBuf, frameType, decoderCookie);
150
151 if (numSamplesOutput != kSamplesPerFrame) {
152 fprintf(stderr, "Decoder encountered error\n");
153 retVal = EXIT_FAILURE;
154 break;
155 }
156
157 for (int i = 0; i < kSamplesPerFrame; ++i) {
158 outputBuf[i] &= 0xfffC;
159 }
160 }
161
162 // Write output to wav.
163 sf_writef_short(handle, outputBuf, kSamplesPerFrame / kChannels);
164 }
165
166 // Close input and output file.
167 fclose(fpInput);
168 sf_close(handle);
169
170 // Free allocated memory.
171 free(inputBuf);
172 free(inputSampleBuf);
173 free(outputBuf);
174
175 return retVal;
176 }
177