/*
 * twamr-tseq-enc is a test program for our AMR encoder: it reads raw
 * 16-bit PCM (matching 3GPP *.inp) as input and writes 3GPP *.cod
 * format as output, allowing libtwamr encoder to be tested with the
 * official test sequences.
 *
 * Both input and output are read/written in the host machine's
 * native byte order.
 */

#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <unistd.h>
#include "../libtwamr/tw_amr.h"

static int
read_input(inf, pcm, filename_for_errs)
	FILE *inf;
	int16_t *pcm;
	char *filename_for_errs;
{
	int cc;

	cc = fread(pcm, 2, 160, inf);
	if (cc == 0)
		return 0;
	if (cc != 160) {
		fprintf(stderr, "error: short read from %s\n",
			filename_for_errs);
		exit(1);
	}
	return 1;
}

main(argc, argv)
	char **argv;
{
	char *infname, *outfname, *mode_arg;
	FILE *inf, *outf;
	struct amr_encoder_state *state;
	enum Mode mode_val;
	int16_t pcm[160];
	struct amr_param_frame frame;
	uint16_t cod_words[AMR_COD_WORDS];
	int opt, dtx = 0, vad2 = 0, use_mode_file = 0, rc;
	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.inp mode output.cod\n",
				argv[0]);
			exit(1);
		}
	}
	if (argc != optind + 3)
		goto usage;
	infname = argv[optind];
	mode_arg = argv[optind+1];
	outfname = argv[optind+2];

	inf = fopen(infname, "r");
	if (!inf) {
		perror(infname);
		exit(1);
	}
	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);
	}
	for (;;) {
		rc = read_input(inf, pcm, infname);
		if (!rc)
			break;
		if (use_mode_file)
			read_mode_file_line(&mode_val);
		amr_encode_frame(state, mode_val, pcm, &frame);
		amr_frame_to_tseq(&frame, cod_words);
		fwrite(cod_words, 2, AMR_COD_WORDS, outf);
	}
	fclose(outf);
	exit(0);
}