您的位置:首页 > 其它

ffmpegturtor06.c做了修改,使用sdl2.0

2016-06-06 15:28 225 查看
// tutorial05.c

// A pedagogical video player that really works!

//

// Code based on FFplay, Copyright (c) 2003 Fabrice Bellard,

// and a tutorial by Martin Bohme (boehme@inb.uni-luebeckREMOVETHIS.de)

// Tested on Gentoo, CVS version 5/01/07 compiled with GCC 4.1.1

// With updates from https://github.com/chelyaev/ffmpeg-tutorial
// Updates tested on:

// LAVC 54.59.100, LAVF 54.29.104, LSWS 2.1.101, SDL 1.2.15

// on GCC 4.7.2 in Debian February 2015

// Use

//

// gcc -o tutorial05 tutorial05.c -lavformat -lavcodec -lswscale -lz -lm `sdl-config --cflags --libs`

// to build (assuming libavformat and libavcodec are correctly installed,

// and assuming you have sdl-config. Please refer to SDL docs for your installation.)

//

// Run using

// tutorial04 myvideofile.mpg

//

// to play the video stream on your screen.

#include <libavcodec/avcodec.h>

#include <libavformat/avformat.h>

//#include <libswscale/swscale.h>

#include <libswresample/swresample.h>

#include <SDL.h>

#include <SDL_thread.h>

#ifdef __MINGW32__

#undef main /* Prevents SDL from overriding main() */

#endif

#include <stdio.h>

#include <assert.h>

#include <math.h>

// compatibility with newer API

#if LIBAVCODEC_VERSION_INT < AV_VERSION_INT(55,28,1)

#define av_frame_alloc avcodec_alloc_frame

#define av_frame_free avcodec_free_frame

#endif

#define SDL_AUDIO_BUFFER_SIZE 1024

#define MAX_AUDIO_FRAME_SIZE 192000

#define MAX_AUDIOQ_SIZE (5 *
16 * 1024)

#define MAX_VIDEOQ_SIZE (5 *
256 * 1024)

#define AV_SYNC_THRESHOLD 0.01

#define AV_NOSYNC_THRESHOLD 10.0

#define SAMPLE_CORRECTION_PERCENT_MAX 10

#define AUDIO_DIFF_AVG_NB 20

#define FF_REFRESH_EVENT (SDL_USEREVENT)

#define FF_QUIT_EVENT (SDL_USEREVENT + 1)

#define VIDEO_PICTURE_QUEUE_SIZE 1

#define DEFAULT_AV_SYNC_TYPE AV_SYNC_VIDEO_MASTER

struct SwrContext *au_convert_ctx;

typedef struct PacketQueue {

AVPacketList *first_pkt, *last_pkt;

int nb_packets;

int size;

SDL_mutex *mutex;

SDL_cond *cond;

} PacketQueue;

typedef struct VideoPicture {

SDL_Texture *bmp;

int width, height;
/* source height & width */

int allocated;

double pts;

} VideoPicture;

#define AVCODEC_MAX_AUDIO_FRAME_SIZE 8192

typedef struct VideoState {

AVFormatContext *pFormatCtx;

int videoStream, audioStream;

int av_sync_type;

double external_clock;
/* external clock base */

int64_t external_clock_time;

double audio_clock;

AVStream *audio_st;

AVCodecContext *audio_ctx;

PacketQueue audioq;

uint8_t audio_buf[(AVCODEC_MAX_AUDIO_FRAME_SIZE *
3) / 2];

unsigned
int audio_buf_size;

unsigned
int audio_buf_index;

AVFrame audio_frame;

AVPacket audio_pkt;

uint8_t *audio_pkt_data;

int audio_pkt_size;

int audio_hw_buf_size;

double audio_diff_cum;
/* used for AV difference average computation */

double audio_diff_avg_coef;

double audio_diff_threshold;

int audio_diff_avg_count;

double frame_timer;

double frame_last_pts;

double frame_last_delay;

double video_clock;
///<pts of last decoded frame / predicted pts of next decoded frame

double video_current_pts;
///<current displayed pts (different from video_clock if frame fifos are used)

int64_t video_current_pts_time; ///<time (av_gettime) at which we updated video_current_pts - used to have running video pts

AVStream *video_st;

AVCodecContext *video_ctx;

PacketQueue videoq;

struct SwsContext *sws_ctx;

VideoPicture pictq[VIDEO_PICTURE_QUEUE_SIZE];

int pictq_size, pictq_rindex, pictq_windex;

SDL_mutex *pictq_mutex;

SDL_cond *pictq_cond;

SDL_Thread *parse_tid;

SDL_Thread *video_tid;

char filename[1024];

int quit;

} VideoState;

enum {

AV_SYNC_AUDIO_MASTER,

AV_SYNC_VIDEO_MASTER,

AV_SYNC_EXTERNAL_MASTER,

};

//SDL_Surface *screen;

SDL_Window *screen;

SDL_Renderer* sdlRenderer;

SDL_Texture* sdlTexture;

SDL_mutex *screen_mutex;

int screen_w = 1920/2;

int screen_h = 1080/2;

SDL_Rect sdlRect;

/* Since we only have one decoding thread, the Big Struct

can be global in case we need it. */

VideoState *global_video_state;

void packet_queue_init(PacketQueue *q) {

memset(q, 0,
sizeof(PacketQueue));

q->mutex = SDL_CreateMutex();

q->cond = SDL_CreateCond();

}

int packet_queue_put(PacketQueue *q, AVPacket *pkt) {

AVPacketList *pkt1;

if(av_dup_packet(pkt) <
0) {

return -1;

}

// if(q->nb_packets>=5) return -1;

pkt1 = av_malloc(sizeof(AVPacketList));

if (!pkt1)

return -1;

pkt1->pkt = *pkt;

pkt1->next = NULL;

SDL_LockMutex(q->mutex);

if (!q->last_pkt)

q->first_pkt = pkt1;

else

q->last_pkt->next = pkt1;

q->last_pkt = pkt1;

q->nb_packets++;

q->size += pkt1->pkt.size;

SDL_CondSignal(q->cond);

SDL_UnlockMutex(q->mutex);

// printf(" PUT q->nb_packets=%d\n", q->nb_packets);

return
0;

}

static int packet_queue_get(PacketQueue *q, AVPacket *pkt,
int block)

{

AVPacketList *pkt1;

int ret;

SDL_LockMutex(q->mutex);

for(;;) {

if(global_video_state->quit) {

ret = -1;

break;

}

pkt1 = q->first_pkt;

if (pkt1) {

q->first_pkt = pkt1->next;

if (!q->first_pkt)

q->last_pkt = NULL;

q->nb_packets--;

q->size -= pkt1->pkt.size;

*pkt = pkt1->pkt;

av_free(pkt1);

ret = 1;

break;

} else
if (!block) {

ret = 0;

break;

} else {

SDL_CondWait(q->cond, q->mutex);

}

}

SDL_UnlockMutex(q->mutex);

// printf("GETa q->nb_packets=%d\n", q->nb_packets);

return ret;

}

double get_audio_clock(VideoState *is) {

double pts;

int hw_buf_size, bytes_per_sec, n;

pts = is->audio_clock; /* maintained in the audio thread */

hw_buf_size = is->audio_buf_size - is->audio_buf_index;

bytes_per_sec = 0;

n = is->audio_ctx->channels * 2;

if(is->audio_st) {

bytes_per_sec = is->audio_ctx->sample_rate * n;

}

if(bytes_per_sec) {

pts -= (double)hw_buf_size / bytes_per_sec;

}

return pts;

}

double get_video_clock(VideoState *is) {

double delta;

delta = (av_gettime() - is->video_current_pts_time) / 1000000.0;

return is->video_current_pts + delta;

}

double get_external_clock(VideoState *is) {

return av_gettime() /
1000000.0;

}

double get_master_clock(VideoState *is) {

if(is->av_sync_type == AV_SYNC_VIDEO_MASTER) {

return get_video_clock(is);

} else
if(is->av_sync_type == AV_SYNC_AUDIO_MASTER) {

return get_audio_clock(is);

} else {

return get_external_clock(is);

}

}

/* Add or subtract samples to get a better sync, return new

audio buffer size */

int synchronize_audio(VideoState *is,
short *samples,

int samples_size,
double pts) {

int n;

double ref_clock;

n = 2 * is->audio_ctx->channels;

if(is->av_sync_type != AV_SYNC_AUDIO_MASTER) {

double diff, avg_diff;

int wanted_size, min_size, max_size
/*, nb_samples */;

ref_clock = get_master_clock(is);

diff = get_audio_clock(is) - ref_clock;

if(diff < AV_NOSYNC_THRESHOLD) {

// accumulate the diffs

is->audio_diff_cum = diff + is->audio_diff_avg_coef

* is->audio_diff_cum;

if(is->audio_diff_avg_count < AUDIO_DIFF_AVG_NB) {

is->audio_diff_avg_count++;

} else {

avg_diff = is->audio_diff_cum * (1.0 - is->audio_diff_avg_coef);

if(fabs(avg_diff) >= is->audio_diff_threshold) {

wanted_size = samples_size + ((int)(diff * is->audio_ctx->sample_rate) * n);

min_size = samples_size * ((100 - SAMPLE_CORRECTION_PERCENT_MAX) /
100);

max_size = samples_size * ((100 + SAMPLE_CORRECTION_PERCENT_MAX) /
100);

if(wanted_size < min_size) {

wanted_size = min_size;

} else
if (wanted_size > max_size) {

wanted_size = max_size;

}

if(wanted_size < samples_size) {

/* remove samples */

samples_size = wanted_size;

} else
if(wanted_size > samples_size) {

uint8_t *samples_end, *q;

int nb;

/* add samples by copying final sample*/

nb = (samples_size - wanted_size);

samples_end = (uint8_t *)samples + samples_size - n;

q = samples_end + n;

while(nb >
0) {

memcpy(q, samples_end, n);

q += n;

nb -= n;

}

samples_size = wanted_size;

}

}

}

} else {

/* difference is TOO big; reset diff stuff */

is->audio_diff_avg_count = 0;

is->audio_diff_cum = 0;

}

}

return samples_size;

}

void convertFmt(AVFrame *audioFrame,short *outputBuffer,int fmt){

// Convert from AV_SAMPLE_FMT_FLTP to AV_SAMPLE_FMT_S16

int in_samples = audioFrame->nb_samples;

int in_linesize = audioFrame->linesize[0];

int i=0;

if(fmt == AV_SAMPLE_FMT_S16P){

// Mono

if (audioFrame->channels==1) {

short* inputChannel0 = (short*)audioFrame->extended_data[0];

for (i=0 ; i<in_samples ; i++) {

outputBuffer[i] = (int16_t) ((*inputChannel0++) );

}

}

// Stereo

else {

short* inputChannel0 = (short*)audioFrame->extended_data[0];

short* inputChannel1 = (short*)audioFrame->extended_data[1];

for (i=0 ; i<in_samples ; i++) {

outputBuffer[i*2] = (int16_t) ((*inputChannel0++) );

outputBuffer[i*2+1] = (int16_t) ((*inputChannel1++) );

}

}

}else
if(fmt == AV_SAMPLE_FMT_FLTP){

if (audioFrame->channels==1) {

float* inputChannel0 = (float*)audioFrame->extended_data[0];

for (i=0 ; i<in_samples ; i++) {

outputBuffer[i] = (int16_t) ((*inputChannel0++) *
32767.0f);

}

}else{

float* inputChannel0 = (float*)audioFrame->extended_data[0];

float* inputChannel1 = (float*)audioFrame->extended_data[1];

for (i=0 ; i<in_samples ; i++) {

outputBuffer[i*2] = (int16_t) ((*inputChannel0++) *
32767.0f);

outputBuffer[i*2+1] = (int16_t) ((*inputChannel1++) *
32767.0f);

}

}

}

// outputBuffer now contains 16-bit PCM!

}

int audio_decode_frame(VideoState *is, uint8_t *audio_buf,
int buf_size, double *pts_ptr) {

int len1, data_size =
0;

AVPacket *pkt = &is->audio_pkt;

double pts;

int n;

int out_buffer_size;

for(;;) {

while(is->audio_pkt_size >
0) {

int got_frame =
0;

len1 = avcodec_decode_audio4(is->audio_ctx, &is->audio_frame, &got_frame, pkt);

if(len1 <
0) {

/* if error, skip frame */

is->audio_pkt_size = 0;

break;

}

data_size = 0;

if(got_frame) {

data_size = av_samples_get_buffer_size(NULL,

is->audio_ctx->channels,

is->audio_frame.nb_samples,

is->audio_ctx->sample_fmt,

1);

// printf("data_size=%d,is->audio_ctx->sample_fmt=%d\n",data_size,is->audio_ctx->sample_fmt);

assert(data_size <= buf_size);

out_buffer_size = is->audio_frame.nb_samples*is->audio_ctx->channels*2;

char *out_buffer=malloc(out_buffer_size);

// convertFmt(&is->audio_frame,out_buffer,is->audio_ctx->sample_fmt);//s16p to s16

int tmp = swr_convert(au_convert_ctx,&out_buffer, is->audio_frame.nb_samples,(const uint8_t **)is->audio_frame.extended_data
, is->audio_frame.nb_samples);

memcpy(audio_buf, out_buffer, out_buffer_size);

free(out_buffer);

}

is->audio_pkt_data += len1;

is->audio_pkt_size -= len1;

if(data_size <=
0) {

/* No data yet, get more frames */

continue;

}

pts = is->audio_clock;

*pts_ptr = pts;

n = 2 * is->audio_ctx->channels;

is->audio_clock += (double)data_size /

(double)(n * is->audio_ctx->sample_rate);

/* We have data, return it and come back for more later */

return out_buffer_size;

}

if(pkt->data)

av_free_packet(pkt);

if(is->quit) {

return -1;

}

/* next packet */

if(packet_queue_get(&is->audioq, pkt,
1) < 0) {

return -1;

}

is->audio_pkt_data = pkt->data;

is->audio_pkt_size = pkt->size;

/* if update, update the audio clock w/pts */

if(pkt->pts != AV_NOPTS_VALUE) {

is->audio_clock = av_q2d(is->audio_st->time_base)*pkt->pts;

}

}

}

void audio_callback(void *userdata, Uint8 *stream,
int len) {

VideoState *is = (VideoState *)userdata;

int len1, audio_size;

double pts;

while(len >
0) {

if(is->audio_buf_index >= is->audio_buf_size) {

/* We have already sent all our data; get more */

audio_size = audio_decode_frame(is, is->audio_buf, sizeof(is->audio_buf), &pts);

if(audio_size <
0) {

/* If error, output silence */

is->audio_buf_size = 1024;

memset(is->audio_buf,
0, is->audio_buf_size);

} else {

audio_size = synchronize_audio(is, (int16_t *)is->audio_buf,

audio_size, pts);

is->audio_buf_size = audio_size;

}

is->audio_buf_index = 0;

}

len1 = is->audio_buf_size - is->audio_buf_index;

if(len1 > len)

len1 = len;

memcpy(stream, (uint8_t *)is->audio_buf + is->audio_buf_index, len1);

len -= len1;

stream += len1;

is->audio_buf_index += len1;

}

}

static Uint32 sdl_refresh_timer_cb(Uint32 interval,
void *opaque) {

SDL_Event event;

event.type = FF_REFRESH_EVENT;

event.user.data1 = opaque;

SDL_PushEvent(&event);

return
0; /* 0 means stop timer */

}

/* schedule a video refresh in 'delay' ms */

static void schedule_refresh(VideoState *is,
int delay) {

SDL_AddTimer(delay, sdl_refresh_timer_cb, is);

}

void video_display(VideoState *is) {

SDL_Rect rect;

VideoPicture *vp;

float aspect_ratio;

int w, h, x, y;

int i;

// printf("pictq_rindex=%d\n",is->pictq_rindex);

vp = &is->pictq[is->pictq_rindex];

if(vp->bmp) {

// printf("333\n");

// SDL_LockMutex(screen_mutex);

SDL_RenderClear( sdlRenderer );

SDL_RenderCopy( sdlRenderer, vp->bmp, NULL, &sdlRect);

SDL_RenderPresent( sdlRenderer );

// printf("444\n");

// SDL_UnlockMutex(screen_mutex);

// if(is->video_ctx->sample_aspect_ratio.num == 0) {

// aspect_ratio = 0;

// } else {

// aspect_ratio = av_q2d(is->video_ctx->sample_aspect_ratio) *

// is->video_ctx->width / is->video_ctx->height;

// }

// if(aspect_ratio <= 0.0) {

// aspect_ratio = (float)is->video_ctx->width /

// (float)is->video_ctx->height;

// }

// h = screen->h;

// w = ((int)rint(h * aspect_ratio)) & -3;

// if(w > screen->w) {

// w = screen->w;

// h = ((int)rint(w / aspect_ratio)) & -3;

// }

// x = (screen->w - w) / 2;

// y = (screen->h - h) / 2;

//

// rect.x = x;

// rect.y = y;

// rect.w = w;

// rect.h = h;

// SDL_LockMutex(screen_mutex);

// SDL_DisplayYUVOverlay(vp->bmp, &rect);

// SDL_UnlockMutex(screen_mutex);

}

}

void video_refresh_timer(void *userdata) {

VideoState *is = (VideoState *)userdata;

VideoPicture *vp;

double actual_delay, delay, sync_threshold, ref_clock, diff;

static
int first = 1;

if(is->video_st) {

if(is->pictq_size ==
0) {

schedule_refresh(is, 1);

} else {

vp = &is->pictq[is->pictq_rindex];

is->video_current_pts = vp->pts;

// printf("delta=%d\n",av_gettime()-is->video_current_pts_time);

is->video_current_pts_time = av_gettime();

delay = vp->pts - is->frame_last_pts; /* the pts from last time */

printf("before vp->pts=%f,is->frame_last_pts=%f,delay=%f\n",vp->pts,is->frame_last_pts,delay);

if(delay <=
0 || delay >= 1.0) {

/* if incorrect delay, use previous one */

delay = is->frame_last_delay;

}

/* save for next time */

is->frame_last_delay = delay;

is->frame_last_pts = vp->pts;

if(first){

is->frame_timer= (av_gettime() / 1000000.0);

delay = is->frame_last_delay;

schedule_refresh(is, (int)(delay *
1000 + 0.5));

video_display(is);

first =0;

if(++is->pictq_rindex == VIDEO_PICTURE_QUEUE_SIZE) {

is->pictq_rindex = 0;

}

SDL_LockMutex(is->pictq_mutex);

is->pictq_size--;

SDL_CondSignal(is->pictq_cond);

SDL_UnlockMutex(is->pictq_mutex);

return;

}

/* update delay to sync to audio if not master source */

if(is->av_sync_type != AV_SYNC_VIDEO_MASTER) {

ref_clock = get_master_clock(is);

diff = vp->pts - ref_clock;

/* Skip or repeat the frame. Take delay into account

FFPlay still doesn't "know if this is the best guess." */

sync_threshold = (delay > AV_SYNC_THRESHOLD) ? delay : AV_SYNC_THRESHOLD;

if(fabs(diff) < AV_NOSYNC_THRESHOLD) {

if(diff <= -sync_threshold) {

delay = 0;

} else
if(diff >= sync_threshold) {

delay = 2 * delay;

}

}

}

is->frame_timer += delay;

/* computer the REAL delay */

actual_delay = is->frame_timer - (av_gettime() / 1000000.0);

// printf("is->frame_timer=%f,actual=%f,delay=%f ,av_gettime()=%f\n",is->frame_timer,actual_delay,delay,av_gettime() / 1000000.0);

if(actual_delay <
0.010) {

/* Really it should skip the picture instead */

actual_delay = 0.010;

}

// actual_delay = 0.010;

// printf("====================================delay=%f,actual=%f\n",delay,actual_delay);

schedule_refresh(is, (int)(actual_delay *
1000 + is->frame_last_delay));

/* show the picture! */

video_display(is);

/* update queue for next picture! */

if(++is->pictq_rindex == VIDEO_PICTURE_QUEUE_SIZE) {

is->pictq_rindex = 0;

}

SDL_LockMutex(is->pictq_mutex);

is->pictq_size--;

SDL_CondSignal(is->pictq_cond);

SDL_UnlockMutex(is->pictq_mutex);

}

} else {

schedule_refresh(is, 20);

}

}

void alloc_picture(void *userdata) {

VideoState *is = (VideoState *)userdata;

VideoPicture *vp;

vp = &is->pictq[is->pictq_windex];

if(vp->bmp) {

// we already have one make another, bigger/smaller

// SDL_FreeYUVOverlay(vp->bmp);

}else{

// printf("alloc_picture\n");

vp->bmp = SDL_CreateTexture(sdlRenderer, SDL_PIXELFORMAT_IYUV, SDL_TEXTUREACCESS_STREAMING,1920,1080);

}

// Allocate a place to put our YUV image on that screen

// SDL_LockMutex(screen_mutex);

// vp->bmp = SDL_CreateYUVOverlay(is->video_ctx->width,

// is->video_ctx->height,

// SDL_YV12_OVERLAY,

// screen);

// SDL_UnlockMutex(screen_mutex);

vp->width = is->video_ctx->width;

vp->height = is->video_ctx->height;

vp->allocated = 1;

}

int queue_picture(VideoState *is, AVFrame *pFrame,
double pts) {

VideoPicture *vp;

int dst_pix_fmt;

AVPicture pict;

/* wait until we have space for a new pic */

SDL_LockMutex(is->pictq_mutex);

while(is->pictq_size >= VIDEO_PICTURE_QUEUE_SIZE &&

!is->quit) {

SDL_CondWait(is->pictq_cond, is->pictq_mutex);

}

SDL_UnlockMutex(is->pictq_mutex);

if(is->quit)

return -1;

// printf("pictq_windex=%d\n",is->pictq_windex);

// windex is set to 0 initially

vp = &is->pictq[is->pictq_windex];

/* allocate or resize the buffer! */

if(!vp->bmp ||

vp->width != is->video_ctx->width ||

vp->height != is->video_ctx->height) {

SDL_Event event;

vp->allocated = 0;

alloc_picture(is);

if(is->quit) {

return -1;

}

}

/* We have a place to put our picture on the queue */

if(vp->bmp) {

// printf("111\n");

// SDL_LockMutex(screen_mutex);

SDL_UpdateYUVTexture(vp->bmp,NULL,

pFrame->data[0], pFrame->linesize[0],

pFrame->data[1], pFrame->linesize[1],

pFrame->data[2], pFrame->linesize[2]);

// printf("222\n");

vp->pts = pts;

// SDL_UnlockMutex(screen_mutex);

// printf("queue pts=%f\n",pts);

// SDL_LockYUVOverlay(vp->bmp);

// vp->pts = pts;

//

// dst_pix_fmt = PIX_FMT_YUV420P;

// /* point pict at the queue */

//

// pict.data[0] = vp->bmp->pixels[0];

// pict.data[1] = vp->bmp->pixels[2];

// pict.data[2] = vp->bmp->pixels[1];

//

// pict.linesize[0] = vp->bmp->pitches[0];

// pict.linesize[1] = vp->bmp->pitches[2];

// pict.linesize[2] = vp->bmp->pitches[1];

//

// // Convert the image into YUV format that SDL uses

// sws_scale(is->sws_ctx, (uint8_t const * const *)pFrame->data,

// pFrame->linesize, 0, is->video_ctx->height,

// pict.data, pict.linesize);

//

// SDL_UnlockYUVOverlay(vp->bmp);

// /* now we inform our display thread that we have a pic ready */

if(++is->pictq_windex == VIDEO_PICTURE_QUEUE_SIZE) {

is->pictq_windex = 0;

}

SDL_LockMutex(is->pictq_mutex);

is->pictq_size++;

SDL_UnlockMutex(is->pictq_mutex);

}

return
0;

}

double synchronize_video(VideoState *is, AVFrame *src_frame,
double pts) {

double frame_delay;

if(pts !=
0) {

/* if we have pts, set video clock to it */

is->video_clock = pts;

} else {

/* if we aren't given a pts, set it to the clock */

pts = is->video_clock;

}

/* update the video clock */

frame_delay = av_q2d(is->video_ctx->time_base);

/* if we are repeating a frame, adjust clock accordingly */

frame_delay += src_frame->repeat_pict * (frame_delay * 0.5);

is->video_clock += frame_delay;

return pts;

}

int video_thread(void *arg) {

VideoState *is = (VideoState *)arg;

AVPacket pkt1, *packet = &pkt1;

int frameFinished;

AVFrame *pFrame;

double pts;

pFrame = av_frame_alloc();

for(;;) {

if(packet_queue_get(&is->videoq, packet,
1) < 0) {

// means we quit getting packets

break;

}

pts = 0;

// Decode video frame

avcodec_decode_video2(is->video_ctx, pFrame, &frameFinished, packet);

// printf("frameFinished=%d\n",frameFinished);

if((pts = av_frame_get_best_effort_timestamp(pFrame)) == AV_NOPTS_VALUE) {

} else {

pts = 0;

// pts = pFrame->pkt_pts;

}

pts = pFrame->pkt_pts;

if(frameFinished){

static
int first = 1;

if(first<=1){

first++;

continue;

}

}

// printf("pts1=%f\n",pts);

// printf("pts=%f,pkt_pts=%d\n",pFrame->pts,pFrame->pkt_pts);

// printf("timebase=%f,is->video_clock=%f\n",av_q2d(is->video_st->time_base),is->video_clock);

pts *= av_q2d(is->video_st->time_base);

// Did we get a video frame?

if(frameFinished) {

pts = synchronize_video(is, pFrame, pts);

// printf("pts2=%f\n",pts);

if(queue_picture(is, pFrame, pts) <
0) {

break;

}

}

av_free_packet(packet);

}

av_frame_free(&pFrame);

return
0;

}

int stream_component_open(VideoState *is,
int stream_index) {

AVFormatContext *pFormatCtx = is->pFormatCtx;

AVCodecContext *codecCtx = NULL;

AVCodec *codec = NULL;

SDL_AudioSpec wanted_spec, spec;

if(stream_index <
0 || stream_index >= pFormatCtx->nb_streams) {

return -1;

}

codec = avcodec_find_decoder(pFormatCtx->streams[stream_index]->codec->codec_id);

if(!codec) {

fprintf(stderr, "Unsupported codec!\n");

return -1;

}

codecCtx = avcodec_alloc_context3(codec);

if(avcodec_copy_context(codecCtx, pFormatCtx->streams[stream_index]->codec) !=
0) {

fprintf(stderr, "Couldn't copy codec context");

return -1;
// Error copying codec context

}

if(codecCtx->codec_type == AVMEDIA_TYPE_AUDIO) {

// Set audio settings from codec info

wanted_spec.freq = codecCtx->sample_rate;

wanted_spec.format = AUDIO_S16SYS;

wanted_spec.channels = codecCtx->channels;

wanted_spec.silence = 0;

wanted_spec.samples = SDL_AUDIO_BUFFER_SIZE;

wanted_spec.callback = audio_callback;

wanted_spec.userdata = is;

if(SDL_OpenAudio(&wanted_spec, &spec) <
0) {

fprintf(stderr, "SDL_OpenAudio: %s\n", SDL_GetError());

return -1;

}

is->audio_hw_buf_size = spec.size;

}

if(avcodec_open2(codecCtx, codec,
NULL) < 0) {

fprintf(stderr, "Unsupported codec!\n");

return -1;

}

switch(codecCtx->codec_type) {

case AVMEDIA_TYPE_AUDIO:

is->audioStream = stream_index;

is->audio_st = pFormatCtx->streams[stream_index];

is->audio_ctx = codecCtx;

is->audio_buf_size = 0;

is->audio_buf_index = 0;

memset(&is->audio_pkt, 0,
sizeof(is->audio_pkt));

packet_queue_init(&is->audioq);

au_convert_ctx=swr_alloc_set_opts(0,codecCtx->channel_layout, AV_SAMPLE_FMT_S16, codecCtx->sample_rate,codecCtx->channel_layout,codecCtx->sample_fmt , codecCtx->sample_rate,0,
NULL);

if(au_convert_ctx==NULL){

perror("alloc swr");

}else

swr_init(au_convert_ctx);

SDL_PauseAudio(0);

break;

case AVMEDIA_TYPE_VIDEO:

is->videoStream = stream_index;

is->video_st = pFormatCtx->streams[stream_index];

is->video_ctx = codecCtx;

is->frame_timer = (double)av_gettime() /
1000000.0;

is->frame_last_delay = 40e-3;

is->video_current_pts_time = av_gettime();

packet_queue_init(&is->videoq);

is->video_tid = SDL_CreateThread(video_thread, "video_thread",is);

// is->sws_ctx = sws_getContext(is->video_ctx->width, is->video_ctx->height,

// is->video_ctx->pix_fmt, is->video_ctx->width,

// is->video_ctx->height, PIX_FMT_YUV420P,

// SWS_BILINEAR, NULL, NULL, NULL

// );

break;

default:

break;

}

}

int decode_thread(void *arg) {

VideoState *is = (VideoState *)arg;

AVFormatContext *pFormatCtx;

AVPacket pkt1, *packet = &pkt1;

int video_index = -1;

int audio_index = -1;

int i;

is->videoStream=-1;

is->audioStream=-1;

global_video_state = is;

pFormatCtx = avformat_alloc_context();

AVDictionary* options = NULL;

av_dict_set(&options, "rtsp_transport",
"udp", 0);

// Open video file

if(avformat_open_input(&pFormatCtx, is->filename,
NULL, &options)!=0)

return -1;
// Couldn't open file

is->pFormatCtx = pFormatCtx;

// pFormatCtx->flags |= AVFMT_FLAG_GENPTS;

// Retrieve stream information

if(avformat_find_stream_info(pFormatCtx,
NULL)<0)

return -1;
// Couldn't find stream information

// Dump information about file onto standard error

av_dump_format(pFormatCtx, 0, is->filename,
0);

// Find the first video stream

for(i=0; i<pFormatCtx->nb_streams; i++) {

if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO &&

video_index < 0) {

video_index=i;

}

if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_AUDIO &&

audio_index < 0) {

audio_index=i;

}

}

if(audio_index >=
0) {

stream_component_open(is, audio_index);

}

if(video_index >=
0) {

stream_component_open(is, video_index);

}

if(is->videoStream <
0 || is->audioStream <
0) {

fprintf(stderr, "%s: could not open codecs\n", is->filename);

goto fail;

}

// main decode loop

for(;;) {

if(is->quit) {

break;

}

// seek stuff goes here

if(is->audioq.size > MAX_AUDIOQ_SIZE ||

is->videoq.size > MAX_VIDEOQ_SIZE) {

SDL_Delay(10);

continue;

}

// static int time1=0;

// if( av_gettime()-time1>40000){

// printf(">>>>>>>>>>>>>>>>>>>>>>>>>>=%d\n",av_gettime()-time1);

// }

//

// time1=av_gettime();

if(av_read_frame(is->pFormatCtx, packet) <
0) {

// if(is->pFormatCtx->pb->error == 0) {

// SDL_Delay(100); /* no error; wait for user input */

// continue;

// } else {

break;

// }

}

// static int cnt=10;

// if(cnt<=30){

// cnt++;

// continue;

// }

// Is this a packet from the video stream?

if(packet->stream_index == is->videoStream) {

packet_queue_put(&is->videoq, packet);

} else
if(packet->stream_index == is->audioStream) {

packet_queue_put(&is->audioq, packet);

} else {

av_free_packet(packet);

}

}

/* all done - wait for it */

while(!is->quit) {

SDL_Delay(100);

}

fail:

if(1){

SDL_Event event;

event.type = FF_QUIT_EVENT;

event.user.data1 = is;

SDL_PushEvent(&event);

}

return
0;

}

int main(int argc,
char *argv[]) {

SDL_Event event;

VideoState *is;

is = av_mallocz(sizeof(VideoState));

if(argc <
2) {

fprintf(stderr, "Usage: test <file>\n");

exit(1);

}

// Register all formats and codecs

av_register_all();

avformat_network_init();

if(SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO | SDL_INIT_TIMER)) {

fprintf(stderr, "Could not initialize SDL - %s\n", SDL_GetError());

exit(1);

}

sdlRect.x=0;

sdlRect.y=0;

sdlRect.w=screen_w;

sdlRect.h=screen_h;

// Make a screen to put our video

//#ifndef __DARWIN__

// screen = SDL_SetVideoMode(640, 480, 0, 0);

//#else

// screen = SDL_SetVideoMode(640, 480, 24, 0);

//#endif

screen = SDL_CreateWindow("Tutorial Window", SDL_WINDOWPOS_UNDEFINED, SDL_WINDOWPOS_UNDEFINED,

screen_w, screen_h,

SDL_WINDOW_OPENGL);

if(!screen) {

fprintf(stderr, "SDL: could not set video mode - exiting\n");

exit(1);

}

sdlRenderer = SDL_CreateRenderer(screen, -1,
0);

screen_mutex = SDL_CreateMutex();

// av_strlcpy(is->filename, "rtsp://admin:admin@10.61.1.53", sizeof(is->filename));

av_strlcpy(is->filename, "/Users/gangzhang/Downloads/colette.15.02.22.ivy.feeling.playful.mp4",
sizeof(is->filename));

// printf("filename=%s",is->filename);

is->pictq_mutex = SDL_CreateMutex();

is->pictq_cond = SDL_CreateCond();

schedule_refresh(is, 40);

is->av_sync_type = DEFAULT_AV_SYNC_TYPE;

is->parse_tid = SDL_CreateThread(decode_thread,"decode_thread", is);

if(!is->parse_tid) {

av_free(is);

return -1;

}

for(;;) {

SDL_WaitEvent(&event);

switch(event.type) {

case FF_QUIT_EVENT:

case SDL_QUIT:

is->quit = 1;

SDL_Quit();

return
0;

break;

case FF_REFRESH_EVENT:

video_refresh_timer(event.user.data1);

break;

default:

break;

}

}

return
0;

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: