diff amrtest/tseq-enc.c @ 422:1ceda5586d01

implement twamr-tseq-enc test program
author Mychaela Falconia <falcon@freecalypso.org>
date Tue, 07 May 2024 06:27:20 +0000
parents efrtest/etsi-enc.c@a13b1605142b
children cf90077b753c
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/amrtest/tseq-enc.c	Tue May 07 06:27:20 2024 +0000
@@ -0,0 +1,111 @@
+/*
+ * 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);
+}