myvtm/Other/libmediaplayer/packet.cpp

#include "packet.h"

int packet_queue_init(packet_queue_t *q, CMediaHostApi* hostapi)
{
    memset(q, 0, sizeof(packet_queue_t));
    q->mutex = SDL_CreateMutex();
    if (!q->mutex)
    {
		hostapi->Debug("SDL_CreateMutex(): %s.", SDL_GetError());
        return AVERROR(ENOMEM);
    }
    q->cond = SDL_CreateCond();
    if (!q->cond)
    {
		hostapi->Debug("SDL_CreateCond(): %s.", SDL_GetError());
        return AVERROR(ENOMEM);
    }
    q->abort_request = 0;
    return 0;
}


// 写队列尾部。pkt是一包还未解码的音频数据
int packet_queue_put(packet_queue_t *q, AVPacket *pkt, CMediaHostApi* hostapi)
{
	AVPacketList* pkt_list = NULL;

    if (av_packet_make_refcounted(pkt) < 0)
    {
		hostapi->Debug("[pkt] is not refrence counted.");
        return -1;
    }
    pkt_list = (AVPacketList*)av_malloc(sizeof(AVPacketList));
    if (!pkt_list)
    {
        return -1;
    }
    
    pkt_list->pkt = *pkt;
    pkt_list->next = NULL;

    SDL_LockMutex(q->mutex);

    if (!q->last_pkt)   // 队列为空
    {
        q->first_pkt = pkt_list;
    }
    else
    {
        q->last_pkt->next = pkt_list;
    }
    q->last_pkt = pkt_list;
    q->nb_packets++;
    q->size += pkt_list->pkt.size;
    // 发个条件变量的信号：重启等待q->cond条件变量的一个线程
	//printf("packet queue put new pkt, CondSignal cond.pkt == 0x%0x, pkt_list->pkt.data == 0x%0x,(pkt->data == 0x%0x) pkt_list->pkt.size = %d, (pkt->size == %d)\n\n\n",pkt, pkt_list->pkt.data, pkt->data, pkt_list->pkt.size, pkt->size);
	//printf("packet queue put new pkt, CondSignal cond.\n");
	SDL_CondSignal(q->cond);

    SDL_UnlockMutex(q->mutex);
    return 0;
}

// 读队列头部。
int packet_queue_get(packet_queue_t *q, AVPacket *pkt, int block)
{
    AVPacketList *p_pkt_node = NULL;
    int ret = -1;

    SDL_LockMutex(q->mutex);

    while (0 == q->abort_request)
    {
        p_pkt_node = q->first_pkt;
        if (p_pkt_node)             // 队列非空，取一个出来
        {
			//printf("packet queue is not empty, get one.\n");
            q->first_pkt = p_pkt_node->next;
            if (!q->first_pkt)
            {
                q->last_pkt = NULL;
            }
            q->nb_packets--;
            q->size -= p_pkt_node->pkt.size;
            *pkt = p_pkt_node->pkt;
            av_free(p_pkt_node);
            ret = 1;
            break;
        }
		else if (!block)            // 队列空且阻塞标志无效，则立即退出
		{
			//printf("packet queue is empty, and block flag is invalid, break.\n");
			ret = 0;
			break;
		}
		else                        // 队列空且阻塞标志有效，则等待
		{
			//printf("packet queue is empty, and block flag is valid, continue wait.\n");
			SDL_CondWait(q->cond, q->mutex);
		}
    }
    SDL_UnlockMutex(q->mutex);

    return ret;
}

int packet_queue_put_nullpacket(packet_queue_t *q, int stream_index, CMediaHostApi* hostapi)
{
    AVPacket pkt1, *pkt = &pkt1;
    av_init_packet(pkt);
    pkt->data = NULL;
    pkt->size = 0;
    pkt->stream_index = stream_index;
    return packet_queue_put(q, pkt, hostapi);
}

void packet_queue_flush(packet_queue_t *q)
{
    AVPacketList *pkt, *pkt1;

    SDL_LockMutex(q->mutex);
    for (pkt = q->first_pkt; pkt; pkt = pkt1) {
        pkt1 = pkt->next;
        //av_packet_unref(&pkt->pkt);
        //av_freep(&pkt);
    }
    q->last_pkt = NULL;
    q->first_pkt = NULL;
    q->nb_packets = 0;
    q->size = 0;
    q->duration = 0;
    SDL_UnlockMutex(q->mutex);
}

void packet_queue_destroy(packet_queue_t *q, CMediaHostApi* m_hostapi)
{
    packet_queue_flush(q);
    SDL_DestroyMutex(q->mutex);
    SDL_DestroyCond(q->cond);
	m_hostapi->Debug("packet queue destroy.");
}

void packet_queue_abort(packet_queue_t *q, CMediaHostApi* m_hostapi)
{
    SDL_LockMutex(q->mutex);
    q->abort_request = 1;
	//m_hostapi->Debug("packet_queue_abort, CondSignal cond.");
    SDL_CondSignal(q->cond);
    SDL_UnlockMutex(q->mutex);
}