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);
+}