view 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 source

/*
 * 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);
}