diff amrtest/encode-r.c @ 445:30cff6b60890

amrtest: implement twamr-encode-r
author Mychaela Falconia <falcon@freecalypso.org>
date Fri, 10 May 2024 00:07:07 +0000
parents amrtest/encode.c@526a7f0e027d
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/amrtest/encode-r.c	Fri May 10 00:07:07 2024 +0000
@@ -0,0 +1,90 @@
+/*
+ * twamr-encode-r is a speech encoder utility going from "robe" (raw BE)
+ * input to RFC 4867 storage format (.amr) output.
+ */
+
+#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/roberead.h"
+
+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;
+	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.robe 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];
+
+	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);
+	}
+	fwrite(amr_file_header_magic, 1, AMR_IETF_HDR_LEN, outf);
+	for (;;) {
+		rc = robe_get_pcm_block(inf, 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);
+}