VPU硬编码

  平台是RK3066(福州瑞芯微公司),android 4.2.0,其实时VP8硬编码,与软件编码是ffpmeg,x264,xvid等软编码是有区别的。硬编码主要是依赖于
硬件。

  硬编码:通过调用Android系统自带的Camera录制视频,实际上是调用了底层的高清编码硬件模块,也即显卡,不使用CPU,速度快
  软编码:使用CPU进行编码,如常见C/C++代码,一般编译生成的二进制都是的,速度相对较慢。例如使用Android NDK编译H264生成so库,编写jni接口,再使用java调用so库
VPUCoder.h

/*
 * VPUCoder.h
 *
 * Current, Only Support YUV420sp encoder and decoder
 *
 *  Created on: Dec 16, 2013
 *      Author: henry
 *
 *  Example:
 *
 *  int main()
 *  {
 *  	int ret = InitCodec();
 *  	//===========encode video
 *  	ret = StartEnc("/sdcard/test.mkv", 1280, 720, 30);
 *
 *  	while(1)
 *  	{
 *  		//get data and length, //unsigned char* data; int length
 *  		ret = ProcessEnc(data, length);
 *  	}
 *  	ret = StopEnc();
 *
 *  	//===========decode video
 *  	ret = ProcessDec("/sdcard/test.mkv", 1280, 720);
 *  	return 0;
 *  }
 */

#ifndef VPUCODER_H_
#define VPUCODER_H_

/**
 * Init encoder and decoder handle, only call once, must first call
 *
 * @return 0 is successful, another fail
 */
int InitCodec();

/**
 * setup encoder configure
 * @param filePath   : save file path
 * @param enc_width  : video width
 * @param enc_height : video height
 * @param enc_fps    : video fps
 *
 * @return 0 is successful, another fail
 */
int StartEnc(const char* filePath, uint32_t enc_width, uint32_t enc_height, uint32_t enc_fps);

/**
 * stop encode video
 */
void StopEnc();

/**
 * @params data   : frame data
 * @params length : frame length
 *
 * @return 0 is successful, another fail
 */
int ProcessEnc(const unsigned char* data, uint32_t length);

/**
 * setup decoder configure
 * @param filePath   : source file path
 * @param enc_width  : video width
 * @param enc_height : video height
 *
 * @return 0 is successful, another fail
 */
int ProcessDec(const char* filePath, uint32_t dec_width, uint32_t dec_height);

#endif /* VPUCODER_H_ */

  VPUCoder.cpp

#include 
#include 
#include 

#include 
#include 
#include 

#include 
#include 
#include 
#include 
#include 
#include 
#include 

using namespace android;

#ifdef LOGI
	#undef LOGI
	#undef LOGW
	#undef LOGE
	#undef LOGD
#endif

#define  LOG_TAG_SERVICE    "TeliDHVPU8Codec"
#define  LOGW(...)  __android_log_print(ANDROID_LOG_WARN,  LOG_TAG_SERVICE, __VA_ARGS__)
#define  LOGE(...)  __android_log_print(ANDROID_LOG_ERROR, LOG_TAG_SERVICE, __VA_ARGS__)
#define  LOGD(...)  __android_log_print(ANDROID_LOG_DEBUG, LOG_TAG_SERVICE, __VA_ARGS__)
#define  LOGI(...)  __android_log_print(ANDROID_LOG_INFO,  LOG_TAG_SERVICE, __VA_ARGS__)

#define MAX_STREM_LENGTH        (1280 * 400)

static uint32_t g_lenght_enc        = MAX_STREM_LENGTH;
static FILE*    g_file_enc          = NULL;
static FILE*    g_file_dec          = NULL;
static VPU_API  g_vpu_api;
static void*    g_vp8encHandle      = NULL;
static void*    g_vp8decHandle      = NULL;
static unsigned int g_bufferLen     = 0;
static unsigned char* g_frameBuffer = NULL;

static EncParams1     g_encParams;
static int            g_SyncFlag = 0;
static unsigned int   g_dataSize;
static unsigned int   g_InputTimestamp;
static int            g_nal = 0;
static int            g_ret = 0;

int InitCodec()
{
	vpu_api_init(&g_vpu_api, OMX_ON2_VIDEO_CodingVPX);
	return 0;
}

int StartEnc(const char* filePath, uint32_t enc_width, uint32_t enc_height, uint32_t enc_fps)
{
	LOGI("StartEnc Begin ====================");
	if(!filePath)
	{
		LOGD("Please make sure file path not null!");
		return -1;
	}

	if(g_file_enc)
	{
		LOGD("Another file is being encoded!");
		return -1;
	}

	g_file_enc = fopen(filePath, "wb+");
	if(!g_file_enc)
	{
		LOGD("file open fail!");
		return -1;
	}

	g_lenght_enc = enc_width * enc_height * 3 >> 1;

	memset(&g_encParams, 0, sizeof(EncParams1));
	g_encParams.width = enc_width;
	g_encParams.height = enc_height;
	g_encParams.framerate = 25;
	g_encParams.bitRate = 10000;

	g_frameBuffer = (unsigned char*)malloc(MAX_STREM_LENGTH);

	g_vp8encHandle = g_vpu_api.get_class_On2Encoder();
	g_vpu_api.init_class_On2Encoder(g_vp8encHandle, &g_encParams, g_frameBuffer, &g_bufferLen);

	g_nal = g_bufferLen;
	fwrite(&g_nal, 1, sizeof(g_nal), g_file_enc);
	fwrite(g_frameBuffer, 1, g_bufferLen, g_file_enc);

	LOGI("StartEnc End g_bufferLen = %d   ====================", g_bufferLen);

	return 0;
}

void StopEnc()
{
	LOGI("StartEnc Begin ====================");
	g_lenght_enc = 0;
	if(g_frameBuffer)
	{
		free(g_frameBuffer);
		g_frameBuffer = NULL;
	}

	if(g_file_enc)
	{
		fclose(g_file_enc);
		g_file_enc = NULL;
	}

	if(g_vp8encHandle)
	{
		g_vpu_api.deinit_class_On2Encoder(g_vp8encHandle);
		g_vpu_api.destroy_class_On2Encoder(g_vp8encHandle);
		g_vp8encHandle = NULL;
	}
	LOGI("StartEnc End ====================");
}

int ProcessEnc(const unsigned char* data, uint32_t length)
{
	if(length != g_lenght_enc)
	{
		LOGD("encode frame length is error!");
		StopEnc();
		return -1;
	}

	if(!g_file_enc)
	{
		LOGD("file not open!");
		return -1;
	}

	LOGD("enc_oneframe_class_On2AvcEncoder");

	g_SyncFlag = 0;
	g_ret = g_vpu_api.enc_oneframe_class_On2Encoder(g_vp8encHandle, g_frameBuffer, &g_bufferLen
			, (unsigned char*)data, 0, &g_dataSize, &g_InputTimestamp, &g_SyncFlag);

	if(g_bufferLen > 0 && g_ret >=0)
	{
		LOGI("ProcessEnc encode one frame ====================");
		g_nal = g_bufferLen;
		fwrite(&g_nal, 1, sizeof(g_nal), g_file_enc);
		fwrite(g_frameBuffer, 1, g_bufferLen, g_file_enc);
		fflush(g_file_enc);
	}
	else
	{
		LOGD("ProcessEnc encode frame fail!");
		return -1;
	}
	g_bufferLen = 0;

	return 0;
}

int32_t FillFrameHead(unsigned char* dst, unsigned char* src, uint32_t size
		, uint32_t bitsreamLen, int64_t time, uint32_t type, uint32_t num)
{
	VPU_BITSTREAM h;
	uint32_t TimeLow = (uint32_t)(time/1000);
	h.StartCode = BSWAP(VPU_BITSTREAM_START_CODE);
	h.SliceLength= BSWAP(size);
	h.SliceTime.TimeLow = BSWAP(TimeLow);
	h.SliceTime.TimeHigh= 0;
	h.SliceType= BSWAP(type);
	h.SliceNum= BSWAP(num);
	h.Res[0] = 0;
	h.Res[1] = 0;
	memcpy(dst, &h, bitsreamLen);

	return 0;
}

int ProcessDec(const char* filePath, uint32_t dec_width, uint32_t dec_height)
{
	LOGI("ProcessDec Begin dec_width = %d dec_height = %d====================", dec_width, dec_height);
	if(!filePath)
	{
		LOGD("src file path is null");
		return -1;
	}

	FILE *fp_in = fopen(filePath, "rb");
	if(!fp_in)
	{
		LOGD("File not exist!");
		return -1;
	}

	char filePathOut[64];
	memset(filePathOut, 0, 64);
	sprintf(filePathOut, "%s.yuv", filePath);
	g_file_dec = fopen(filePathOut, "wb+");

	if(!g_file_dec)
	{
		LOGD("file open fail!");
		fclose(fp_in);
		return -1;
	}

	unsigned char* aOutBuffer;
	unsigned char* aInputBuf  = NULL;
	unsigned char* aInputData = NULL;
	unsigned int   aOutputLength;
	unsigned int   aInBufSize = MAX_STREM_LENGTH;

	g_vp8decHandle = g_vpu_api.get_class_On2Decoder();
	g_vpu_api.init_class_On2Decoder(g_vp8decHandle);

	int tmp_length = (dec_width * dec_height * 3) >> 1;

	LOGI(" init_class_On2AvcDecoder OK! tmp_length=%d", tmp_length);

	aOutBuffer = (unsigned char*)malloc(tmp_length);
	aInputBuf = (unsigned char*)malloc(MAX_STREM_LENGTH);

	fread(&g_nal, 1, sizeof(g_nal), fp_in);

	if(g_nal > 0)
	{
		fread(aInputBuf, 1, g_nal, fp_in);
	}
	else
	{
		LOGD("Can't decoder the video!");
		goto fail;
	}

	int ret;
	uint32_t bitsLen;
	uint32_t vpu_stream_lenc;

	vpu_stream_lenc = sizeof(VPU_BITSTREAM);
	aInputData = aInputBuf + vpu_stream_lenc;

	while(1)
	{
		ret = fread(&g_nal, 1, sizeof(g_nal), fp_in);
		if(ret <= 0)
			break;

		fread(aInputData, 1, g_nal, fp_in);
		aInBufSize = g_nal;
		bitsLen    = vpu_stream_lenc + aInBufSize;
		FillFrameHead(aInputBuf, aInputData, aInBufSize, vpu_stream_lenc, 0, 0, 0);

		ret = g_vpu_api.dec_oneframe_class_On2Decoder(g_vp8decHandle, aOutBuffer, &aOutputLength, aInputBuf, &bitsLen);
		LOGI("out_length=%d in_length=%d ret=%d", aOutputLength, bitsLen, ret);

		if(aOutputLength > 0)
		{
			VPU_FRAME *frame = (VPU_FRAME *)aOutBuffer;

			if(frame->vpumem.phy_addr)
			{
				VPUMemLink(&frame->vpumem);

				if(frame->vpumem.vir_addr)
				{
					LOGI("VPUMemLinear_t size=%d", frame->vpumem.size);
					fwrite(frame->vpumem.vir_addr, 1, tmp_length, g_file_dec);
					fflush(g_file_dec);
				}
				else
				{
					LOGI("vir_addr is null");
				}
				VPUFreeLinear(&frame->vpumem);
			}
			else
			{
				LOGI("phy_addr is null");
			}
		}
		fflush(fp_in);
	}

fail:
	if(aOutBuffer)
	{
		free(aOutBuffer);
		aOutBuffer = NULL;
	}

	if(aInputBuf)
	{
		free(aInputBuf);
		aInputBuf = NULL;
	}

	if(g_file_dec)
	{
	   fclose(g_file_dec);
	   g_file_dec = NULL;
	}

	if(fp_in)
	{
		fclose(fp_in);
		fp_in = NULL;
	}

	if(g_vp8decHandle)
	{
	   g_vpu_api.deinit_class_On2Decoder(g_vp8decHandle);
	   g_vpu_api.destroy_class_On2Decoder(g_vp8decHandle);

	   g_vp8decHandle = NULL;
	}
	LOGI("ProcessDec End ====================");

	return 0;
}

  android.mk

LOCAL_PATH := $(call my-dir)
include $(CLEAR_VARS)

VISION=1.0.1

LOCAL_MODULE := VPU8Coder_${VISION}
LOCAL_SRC_FILES := VPUCoder.cpp

LOCAL_LDLIBS := -L$(LOCAL_PATH)/libs \
-lvpu \
-lcutils \
-lutils \
-lgui \
-lrk_on2 \
-lstagefright

LOCAL_C_INCLUDES := \
$(JNI_H_INCLUDE) \
$(LOCAL_PATH)/include \
$(LOCAL_PATH)/include/media/openmax

LOCAL_CFLAGS := -DOSCL_IMPORT_REF= -DOSCL_UNUSED_ARG= -DOSCL_EXPORT_REF=

include $(BUILD_SHARED_LIBRARY)

你可能感兴趣的:(Linux)