am 5cc5fd61: Merge "AMR-NB encoder: Add test" (3192cbda) · Commits · e / devices / smdk4412 / android_frameworks_av

media/libstagefright/codecs/amrnb/enc/Android.mk

+22 −0

Original line number	Original line	Diff line number	Diff line
	@@ -103,3 +103,25 @@ LOCAL_MODULE := libstagefright_soft_amrnbenc
	LOCAL_MODULE_TAGS := optional		LOCAL_MODULE_TAGS := optional

	include $(BUILD_SHARED_LIBRARY)		include $(BUILD_SHARED_LIBRARY)

			################################################################################

			include $(CLEAR_VARS)
			LOCAL_SRC_FILES := \
			test/amrnb_enc_test.cpp

			LOCAL_C_INCLUDES := \
			$(LOCAL_PATH)/src \
			$(LOCAL_PATH)/../common/include


			LOCAL_STATIC_LIBRARIES := \
			libstagefright_amrnbenc

			LOCAL_SHARED_LIBRARIES := \
			libstagefright_amrnb_common

			LOCAL_MODULE := libstagefright_amrnbenc_test
			LOCAL_MODULE_TAGS := tests

			include $(BUILD_EXECUTABLE)

media/libstagefright/codecs/amrnb/enc/test/amrnb_enc_test.cpp

0 → 100644

+237 −0

Original line number	Original line	Diff line number	Diff line
			/*
			* Copyright (C) 2014 The Android Open Source Project
			* All rights reserved.
			*
			* Redistribution and use in source and binary forms, with or without
			* modification, are permitted provided that the following conditions
			* are met:
			* * Redistributions of source code must retain the above copyright
			* notice, this list of conditions and the following disclaimer.
			* * Redistributions in binary form must reproduce the above copyright
			* notice, this list of conditions and the following disclaimer in
			* the documentation and/or other materials provided with the
			* distribution.
			*
			* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
			* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
			* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
			* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
			* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
			* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
			* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
			* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
			* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
			* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
			* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
			* SUCH DAMAGE.
			*/
			#include <stdio.h>
			#include <stdlib.h>
			#include <time.h>
			#include <stdint.h>
			#include <assert.h>
			#include "gsmamr_enc.h"

			enum {
			kInputSize = 320, // 160 samples * 16-bit per sample.
			kOutputSize = 1024
			};

			struct AmrNbEncState {
			void *encCtx;
			void *pidSyncCtx;
			};

			void usage(void) {
			printf("Usage:\n");
			printf("AMRNBEnc [options] <input file> <output file>\n");
			printf("\n");
			printf("Options +M* for setting compression bitrate mode, default is 4.75 kbps\n");
			printf(" +M0 = 4.75 kbps\n");
			printf(" +M1 = 5.15 kbps\n");
			printf(" +M2 = 5.90 kbps\n");
			printf(" +M3 = 6.70 kbps\n");
			printf(" +M4 = 7.40 kbps\n");
			printf(" +M5 = 7.95 kbps\n");
			printf(" +M6 = 10.2 kbps\n");
			printf(" +M7 = 12.2 kbps\n");
			printf("\n");
			}

			int encode(int mode, const char srcFile, const char dstFile) {
			int retVal = EXIT_SUCCESS;
			FILE *fSrc = NULL;
			FILE *fDst = NULL;
			int frameNum = 0;
			bool eofReached = false;
			uint16_t *inputBuf = NULL;
			uint8_t *outputBuf = NULL;
			AmrNbEncState *amr = NULL;

			clock_t start, finish;
			double duration = 0.0;

			// Open input file.
			fSrc = fopen(srcFile, "rb");
			if (fSrc == NULL) {
			fprintf(stderr, "Error opening input file\n");
			retVal = EXIT_FAILURE;
			goto safe_exit;
			}

			// Open output file.
			fDst = fopen(dstFile, "wb");
			if (fDst == NULL) {
			fprintf(stderr, "Error opening output file\n");
			retVal = EXIT_FAILURE;
			goto safe_exit;
			}

			// Allocate input buffer.
			inputBuf = (uint16_t*) malloc(kInputSize);
			assert(inputBuf != NULL);

			// Allocate output buffer.
			outputBuf = (uint8_t*) malloc(kOutputSize);
			assert(outputBuf != NULL);

			// Initialize encoder.
			amr = (AmrNbEncState*) malloc(sizeof(AmrNbEncState));
			AMREncodeInit(&amr->encCtx, &amr->pidSyncCtx, 0);

			// Write file header.
			fwrite("#!AMR\n", 1, 6, fDst);

			while (1) {
			// Read next input frame.
			int bytesRead;
			bytesRead = fread(inputBuf, 1, kInputSize, fSrc);
			if (bytesRead != kInputSize && !feof(fSrc)) {
			retVal = EXIT_FAILURE; // Invalid magic number.
			fprintf(stderr, "Error reading input file\n");
			goto safe_exit;
			} else if (feof(fSrc) && bytesRead == 0) {
			eofReached = true;
			break;
			}

			start = clock();

			// Encode the frame.
			Frame_Type_3GPP frame_type = (Frame_Type_3GPP) mode;
			int bytesGenerated;
			bytesGenerated = AMREncode(amr->encCtx, amr->pidSyncCtx, (Mode)mode,
			(Word16*)inputBuf, outputBuf, &frame_type,
			AMR_TX_WMF);

			// Convert from WMF to RFC 3267 format.
			if (bytesGenerated > 0) {
			outputBuf[0] = ((outputBuf[0] << 3) \| 4) & 0x7c;
			}

			finish = clock();
			duration += finish - start;

			if (bytesGenerated < 0) {
			retVal = EXIT_FAILURE;
			fprintf(stderr, "Encoding error\n");
			goto safe_exit;
			}

			frameNum++;
			printf(" Frames processed: %d\n", frameNum);

			// Write the output.
			fwrite(outputBuf, 1, bytesGenerated, fDst);
			}

			// Dump the time taken by encode.
			printf("\n%2.5lf seconds\n", (double)duration/CLOCKS_PER_SEC);

			safe_exit:

			// Free the encoder instance.
			if (amr) {
			AMREncodeExit(&amr->encCtx, &amr->pidSyncCtx);
			free(amr);
			}

			// Free input and output buffer.
			free(inputBuf);
			free(outputBuf);

			// Close the input and output files.
			if (fSrc) {
			fclose(fSrc);
			}
			if (fDst) {
			fclose(fDst);
			}

			return retVal;
			}

			int main(int argc, char *argv[]) {
			Mode mode = MR475;
			int retVal;
			char *inFileName = NULL;
			char *outFileName = NULL;
			int arg, filename = 0;

			if (argc < 3) {
			usage();
			return EXIT_FAILURE;
			} else {
			for (arg = 1; arg < argc; arg++) {
			if (argv[arg][0] == '+') {
			if (argv[arg][1] == 'M') {
			switch (argv[arg][2]) {
			case '0': mode = MR475;
			break;
			case '1': mode = MR515;
			break;
			case '2': mode = MR59;
			break;
			case '3': mode = MR67;
			break;
			case '4': mode = MR74;
			break;
			case '5': mode = MR795;
			break;
			case '6': mode = MR102;
			break;
			case '7': mode = MR122;
			break;
			default:
			usage();
			fprintf(stderr, "Invalid parameter '%s'.\n", argv[arg]);
			return EXIT_FAILURE;
			break;
			}
			} else {
			usage();
			fprintf(stderr, "Invalid parameter '%s'.\n", argv[arg]);
			return EXIT_FAILURE;
			}
			} else {
			switch (filename) {
			case 0:
			inFileName = argv[arg];
			break;
			case 1:
			outFileName = argv[arg];
			break;
			default:
			usage();
			fprintf(stderr, "Invalid parameter '%s'.\n", argv[arg]);
			return EXIT_FAILURE;
			}
			filename++;
			}
			}
			}

			retVal = encode(mode, inFileName, outFileName);
			return retVal;
			}