樹莓派raspberry pi3硬件解碼H264 GPU OMX

本程序在樹莓派3B上運行通過，raspberry jessie系統

話不多說，上代碼：

OMXH264Player.h

#ifndef OMXH264PLAYER_H
#define OMXH264PLAYER_H

#include <QObject>
#include "bcm_host.h"
extern "C"{
    #include "ilclient.h"
}

class OMXH264Player: public QObject
{
   Q_OBJECT
public:
    OMXH264Player(QObject *parent = 0);
    ~OMXH264Player();
    void draw( unsigned char * image, int size);
    void clear();
public slots:
    void playbackTest();

public:
    OMX_VIDEO_PARAM_PORTFORMATTYPE format;
    OMX_TIME_CONFIG_CLOCKSTATETYPE cstate;
    COMPONENT_T *video_decode, *video_scheduler, *video_render, *clock;
    COMPONENT_T *list[5];
    TUNNEL_T tunnel[4];
    ILCLIENT_T *client;
    int port_settings_changed;
    int first_packet;
    OMX_BUFFERHEADERTYPE *omxbuf;
};

#endif // OMXH264PLAYER_H

OMXH264Player.cpp

#include "OMXH264Player.h"
#include <QDebug>

OMXH264Player::OMXH264Player(QObject *parent):
    QObject(parent),
    video_decode(NULL),
    video_scheduler(NULL),
    video_render(NULL),
    clock(NULL),
    port_settings_changed(0),
    first_packet(1),
    omxbuf(NULL)
{
    qDebug()<<"OMXH264Player constructor start";
    bcm_host_init();
    int status = 0;
    memset(list, 0, sizeof(list));
    memset(tunnel, 0, sizeof(tunnel));

    if((client = ilclient_init()) == NULL){
        qDebug()<<"OMXH264Player ilclient_init error";
        return;
    }
    OMX_ERRORTYPE error;
    if((error = OMX_Init()) != OMX_ErrorNone){
        ilclient_destroy(client);
        qDebug("OMXH264Player OMX_Init error %x",error);
        return;
    }

    // create video_decode
    if(ilclient_create_component(client, &video_decode, "video_decode", (ILCLIENT_CREATE_FLAGS_T)(ILCLIENT_DISABLE_ALL_PORTS | ILCLIENT_ENABLE_INPUT_BUFFERS)) != 0)
      status = -14;
    list[0] = video_decode;

    // create video_render
    if(status == 0 && ilclient_create_component(client, &video_render, "video_render", ILCLIENT_DISABLE_ALL_PORTS) != 0)
      status = -14;
    list[1] = video_render;

    // create clock
    if(status == 0 && ilclient_create_component(client, &clock, "clock", ILCLIENT_DISABLE_ALL_PORTS) != 0)
      status = -14;
    list[2] = clock;

    memset(&cstate, 0, sizeof(cstate));
    cstate.nSize = sizeof(cstate);
    cstate.nVersion.nVersion = OMX_VERSION;
    cstate.eState = OMX_TIME_ClockStateWaitingForStartTime;
    cstate.nWaitMask = 1;
    if(clock != NULL && OMX_SetParameter(ILC_GET_HANDLE(clock), OMX_IndexConfigTimeClockState, &cstate) != OMX_ErrorNone)
      status = -13;

    // create video_scheduler
    if(status == 0 && ilclient_create_component(client, &video_scheduler, "video_scheduler", ILCLIENT_DISABLE_ALL_PORTS) != 0)
      status = -14;
    list[3] = video_scheduler;

    set_tunnel(tunnel, video_decode, 131, video_scheduler, 10);
    set_tunnel(tunnel+1, video_scheduler, 11, video_render, 90);
    set_tunnel(tunnel+2, clock, 80, video_scheduler, 12);

    // setup clock tunnel first
    if(status == 0 && ilclient_setup_tunnel(tunnel+2, 0, 0) != 0)
      status = -15;
    else
      ilclient_change_component_state(clock, OMX_StateExecuting);

    if(status == 0)
      ilclient_change_component_state(video_decode, OMX_StateIdle);

    memset(&format, 0, sizeof(OMX_VIDEO_PARAM_PORTFORMATTYPE));
    format.nSize = sizeof(OMX_VIDEO_PARAM_PORTFORMATTYPE);
    format.nVersion.nVersion = OMX_VERSION;
    format.nPortIndex = 130;
    format.eCompressionFormat = OMX_VIDEO_CodingAVC;

    if(status == 0 &&
          OMX_SetParameter(ILC_GET_HANDLE(video_decode), OMX_IndexParamVideoPortFormat, &format) == OMX_ErrorNone &&
          ilclient_enable_port_buffers(video_decode, 130, NULL, NULL, NULL) == 0){
        qDebug()<<"OMXH264Player constructor success";
    }else{
        qDebug()<<"OMXH264Player constructor error, status = "<<status;
    }
    ilclient_change_component_state(video_decode, OMX_StateExecuting);
}

OMXH264Player::~OMXH264Player(){
    qDebug()<<"OMXH264Player destructor";
}

void OMXH264Player::draw( unsigned char * image, int size){
//    qDebug()<<"write "<<size;
    if(size > 81920){
        qDebug()<<"write size larger than omxbuf size, largest 81920";
        return;
    }
    unsigned int data_len = 0;
    omxbuf = ilclient_get_input_buffer(video_decode, 130, 1);
    if(omxbuf == NULL){
        return;
    }
    unsigned char *dest = omxbuf->pBuffer;
    memcpy(dest, image, size);
    data_len = size;

    if(port_settings_changed == 0 && (data_len > 0 && ilclient_remove_event(video_decode, OMX_EventPortSettingsChanged, 131, 0, 0, 1) == 0)){
        port_settings_changed = 1;
        if(ilclient_setup_tunnel(tunnel, 0, 0) != 0) return;
        ilclient_change_component_state(video_scheduler, OMX_StateExecuting);
        // now setup tunnel to video_render
        if(ilclient_setup_tunnel(tunnel+1, 0, 1000) != 0) return ;
        ilclient_change_component_state(video_render, OMX_StateExecuting);
    }
    if(!data_len) return ;
    omxbuf->nFilledLen = data_len;
    omxbuf->nOffset = 0;
    if(first_packet)
    {
        omxbuf->nFlags = OMX_BUFFERFLAG_STARTTIME;
        first_packet = 0;
    }else omxbuf->nFlags = OMX_BUFFERFLAG_TIME_UNKNOWN;

    OMX_EmptyThisBuffer(ILC_GET_HANDLE(video_decode), omxbuf);
}

void OMXH264Player::playbackTest(){
    unsigned char* buffer = (unsigned char*)malloc(81920);
    memset(buffer, 0, 81920);
    FILE *in;
    if((in = fopen("/home/pi/middleware/FaceTimeEveryDay.264", "rb")) == NULL)
        return;
    int read = 0;
    while((read = fread(buffer, 1, 2000, in))>0){
        draw(buffer, read);
    }
    clear();
}

void OMXH264Player::clear(){
    if((omxbuf = ilclient_get_input_buffer(video_decode, 130, 1)) != NULL){
        omxbuf->nFilledLen = 0;
        omxbuf->nFlags = OMX_BUFFERFLAG_TIME_UNKNOWN | OMX_BUFFERFLAG_EOS;
        if(OMX_EmptyThisBuffer(ILC_GET_HANDLE(video_decode), omxbuf) != OMX_ErrorNone) qDebug()<<"EmptyThisBuffer error";

        // wait for EOS from render
        ilclient_wait_for_event(video_render, OMX_EventBufferFlag, 90, 0, OMX_BUFFERFLAG_EOS, 0, ILCLIENT_BUFFER_FLAG_EOS, -1);

        // need to flush the renderer to allow video_decode to disable its input port
        ilclient_flush_tunnels(tunnel, 0);
    }

    ilclient_disable_tunnel(tunnel);
    ilclient_disable_tunnel(tunnel+1);
    ilclient_disable_tunnel(tunnel+2);
    ilclient_disable_port_buffers(video_decode, 130, NULL, NULL, NULL);
    ilclient_teardown_tunnels(tunnel);


    ilclient_state_transition(list, OMX_StateIdle);
    ilclient_state_transition(list, OMX_StateLoaded);

    ilclient_cleanup_components(list);

    OMX_Deinit();

    ilclient_destroy(client);
}

參考:

[raspberry jessie] /opt/vc/src/hello_pi/hello_video/video.c