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 <stdio.h>#include <unistd.h>#include <stdlib.h>#include <vpu_global.h>#include <SoftwareRenderer.h>#include <vpu_api_interface.h>#include <utils/Log.h>#include <ui/DisplayInfo.h>#include <gui/ISurfaceTexture.h>#include <gui/ISurfaceComposer.h>#include <gui/SurfaceTextureClient.h>#include <gui/SurfaceComposerClient.h>#include <media/stagefright/MetaData.h>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)