FreeCalypso > hg > gsm-codec-lib
diff amrtest/encode.c @ 443:526a7f0e027d
amrtest: implement twamr-encode
author | Mychaela Falconia <falcon@freecalypso.org> |
---|---|
date | Thu, 09 May 2024 21:00:33 +0000 |
parents | amrtest/tseq-enc.c@cf90077b753c |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/amrtest/encode.c Thu May 09 21:00:33 2024 +0000 @@ -0,0 +1,97 @@ +/* + * twamr-encode is a speech encoder utility going from WAV input + * to RFC 4867 storage format (.amr) output. It logically replaces + * amrnb-enc from opencore-amr package, but its command line structure + * for specifying AMR modes follows that of twamr-tseq-enc. + */ + +#include <stdio.h> +#include <stdint.h> +#include <stdlib.h> +#include <string.h> +#include <strings.h> +#include <unistd.h> +#include "../libtwamr/tw_amr.h" +#include "../libtest/wavreader.h" +#include "../libtest/wavrdhelp.h" + +main(argc, argv) + char **argv; +{ + char *infname, *outfname, *mode_arg; + void *wav; + FILE *outf; + struct amr_encoder_state *state; + enum Mode mode_val; + int16_t pcm[160]; + struct amr_param_frame frame; + uint8_t out_bytes[AMR_IETF_MAX_PL]; + int opt, dtx = 0, vad2 = 0, use_mode_file = 0, rc; + unsigned nbytes; + extern int optind; + + while ((opt = getopt(argc, argv, "d2")) != EOF) { + switch (opt) { + case 'd': + dtx = 1; + continue; + case '2': + vad2 = 1; + continue; + default: + usage: + fprintf(stderr, + "usage: %s [-d] [-2] input.wav mode output.amr\n", + argv[0]); + exit(1); + } + } + if (argc != optind + 3) + goto usage; + infname = argv[optind]; + mode_arg = argv[optind+1]; + outfname = argv[optind+2]; + + wav = wav_read_open(infname); + if (!wav) { + perror(infname); + exit(1); + } + rc = wavrd_check_header(wav, infname); + if (rc < 0) + exit(1); /* error msg already printed */ + if (!strncmp(mode_arg, "file:", 5)) { + open_mode_file(mode_arg + 5); + use_mode_file = 1; + } else { + rc = grok_mode_name(mode_arg, &mode_val); + if (rc < 0) { + fprintf(stderr, "error: invalid mode argument \"%s\"\n", + mode_arg); + exit(1); + } + } + outf = fopen(outfname, "w"); + if (!outf) { + perror(outfname); + exit(1); + } + state = amr_encoder_create(dtx, vad2); + if (!state) { + perror("amr_encoder_create()"); + exit(1); + } + fwrite(amr_file_header_magic, 1, AMR_IETF_HDR_LEN, outf); + for (;;) { + rc = wavrd_get_pcm_block(wav, pcm); + if (!rc) + break; + if (use_mode_file) + read_mode_file_line(&mode_val); + amr_encode_frame(state, mode_val, pcm, &frame); + nbytes = amr_frame_to_ietf(&frame, out_bytes); + fwrite(out_bytes, 1, nbytes, outf); + } + fclose(outf); + exit(0); +}