ffplay原始碼分析02 ---- 資料讀取執行緒
阿新 • • 發佈:2020-12-05
1.stream_open()
static VideoState *stream_open(const char *filename, AVInputFormat *iformat) { // 資料初始化 VideoState *is; is = av_mallocz(sizeof(VideoState)); /* 分配結構體並初始化 */ if (!is) return NULL; is->filename = av_strdup(filename); if (!is->filename)goto fail; is->iformat = iformat; is->ytop = 0; is->xleft = 0; /* 初始化幀佇列 */ if (frame_queue_init(&is->pictq, &is->videoq, VIDEO_PICTURE_QUEUE_SIZE, 1) < 0) goto fail; if (frame_queue_init(&is->subpq, &is->subtitleq, SUBPICTURE_QUEUE_SIZE, 0) < 0) goto fail; if (frame_queue_init(&is->sampq, &is->audioq, SAMPLE_QUEUE_SIZE, 1) < 0) goto fail; if (packet_queue_init(&is->videoq) < 0 || packet_queue_init(&is->audioq) < 0 || packet_queue_init(&is->subtitleq) < 0) goto fail; /* 建立 */ if (!(is->continue_read_thread = SDL_CreateCond())) { av_log(NULL, AV_LOG_FATAL, "SDL_CreateCond(): %s\n", SDL_GetError()); goto fail; } /* * 初始化時鐘 * 時鐘序列->queue_serial,實際上指向的是is->videoq.serial */ init_clock(&is->vidclk, &is->videoq.serial); init_clock(&is->audclk, &is->audioq.serial); init_clock(&is->extclk, &is->extclk.serial); is->audio_clock_serial = -1; /* 初始化音量 */ if (startup_volume < 0) av_log(NULL, AV_LOG_WARNING, "-volume=%d < 0, setting to 0\n", startup_volume); if (startup_volume > 100) av_log(NULL, AV_LOG_WARNING, "-volume=%d > 100, setting to 100\n", startup_volume); startup_volume = av_clip(startup_volume, 0, 100); startup_volume = av_clip(SDL_MIX_MAXVOLUME * startup_volume / 100, 0, SDL_MIX_MAXVOLUME); is->audio_volume = startup_volume; is->muted = 0; // 同步型別:預設以音訊為基準 is->av_sync_type = av_sync_type; /* 建立讀執行緒 */ is->read_tid = SDL_CreateThread(read_thread, "read_thread", is); if (!is->read_tid) { av_log(NULL, AV_LOG_FATAL, "SDL_CreateThread(): %s\n", SDL_GetError()); fail: stream_close(is); return NULL; } return is; }
這個比較簡單,就是一些資料初始化
2. read_thread()
/* this thread gets the stream from the disk or the network */ /* * 資料都由這裡讀取 * 主要功能是做解複用,從碼流中分離音視訊packet,並插入快取佇列 */ static int read_thread(void *arg) { VideoState *is = arg; AVFormatContext *ic = NULL; int err, i, ret; int st_index[AVMEDIA_TYPE_NB]; AVPacket pkt1, *pkt = &pkt1; int64_t stream_start_time; int pkt_in_play_range = 0; AVDictionaryEntry *t; SDL_mutex *wait_mutex = SDL_CreateMutex(); int scan_all_pmts_set = 0; int64_t pkt_ts; // 一、準備流程 if (!wait_mutex) { av_log(NULL, AV_LOG_FATAL, "SDL_CreateMutex(): %s\n", SDL_GetError()); ret = AVERROR(ENOMEM); goto fail; } memset(st_index, -1, sizeof(st_index)); // 初始化為-1,如果一直為-1說明沒相應steam is->last_video_stream = is->video_stream = -1; is->last_audio_stream = is->audio_stream = -1; is->last_subtitle_stream = is->subtitle_stream = -1; is->eof = 0; // =1是表明資料讀取完畢 // 1. 建立上下文結構體,這個結構體是最上層的結構體,表示輸入上下文 ic = avformat_alloc_context(); if (!ic) { av_log(NULL, AV_LOG_FATAL, "Could not allocate context.\n"); ret = AVERROR(ENOMEM); goto fail; } /* 2.設定中斷回撥函式,如果出錯或者退出,就根據目前程式設定的狀態選擇繼續check或者直接退出 */ /* 當執行耗時操作時(一般是在執行while或者for迴圈的資料讀取時),會呼叫interrupt_callback.callback * 回撥函式中返回1則代表ffmpeg結束耗時操作退出當前函式的呼叫 * 回撥函式中返回0則代表ffmpeg內部繼續執行耗時操作,直到完成既定的任務(比如讀取到既定的資料包) */ ic->interrupt_callback.callback = decode_interrupt_cb; ic->interrupt_callback.opaque = is; //特定選項處理 if (!av_dict_get(format_opts, "scan_all_pmts", NULL, AV_DICT_MATCH_CASE)) { av_dict_set(&format_opts, "scan_all_pmts", "1", AV_DICT_DONT_OVERWRITE); scan_all_pmts_set = 1; } /* 3.開啟檔案,主要是探測協議型別,如果是網路檔案則建立網路連結等 */ err = avformat_open_input(&ic, is->filename, is->iformat, &format_opts); if (err < 0) { print_error(is->filename, err); ret = -1; goto fail; } if (scan_all_pmts_set) av_dict_set(&format_opts, "scan_all_pmts", NULL, AV_DICT_MATCH_CASE); if ((t = av_dict_get(format_opts, "", NULL, AV_DICT_IGNORE_SUFFIX))) { av_log(NULL, AV_LOG_ERROR, "Option %s not found.\n", t->key); ret = AVERROR_OPTION_NOT_FOUND; goto fail; } is->ic = ic; // videoState的ic指向分配的ic if (genpts) ic->flags |= AVFMT_FLAG_GENPTS; av_format_inject_global_side_data(ic); if (find_stream_info) { AVDictionary **opts = setup_find_stream_info_opts(ic, codec_opts); int orig_nb_streams = ic->nb_streams; /* * 4.探測媒體型別,可得到當前檔案的封裝格式,音視訊編碼引數等資訊 * 呼叫該函式後得多的引數資訊會比只調用avformat_open_input更為詳細, * 其本質上是去做了decdoe packet獲取資訊的工作 * codecpar, filled by libavformat on stream creation or * in avformat_find_stream_info() */ err = avformat_find_stream_info(ic, opts); for (i = 0; i < orig_nb_streams; i++) av_dict_free(&opts[i]); av_freep(&opts); if (err < 0) { av_log(NULL, AV_LOG_WARNING, "%s: could not find codec parameters\n", is->filename); ret = -1; goto fail; } } if (ic->pb) ic->pb->eof_reached = 0; // FIXME hack, ffplay maybe should not use avio_feof() to test for the end if (seek_by_bytes < 0) seek_by_bytes = !!(ic->iformat->flags & AVFMT_TS_DISCONT) && strcmp("ogg", ic->iformat->name); is->max_frame_duration = (ic->iformat->flags & AVFMT_TS_DISCONT) ? 10.0 : 3600.0; if (!window_title && (t = av_dict_get(ic->metadata, "title", NULL, 0))) window_title = av_asprintf("%s - %s", t->value, input_filename); /* if seeking requested, we execute it */ /* 5. 檢測是否指定播放起始時間 */ if (start_time != AV_NOPTS_VALUE) { int64_t timestamp; timestamp = start_time; /* add the stream start time */ if (ic->start_time != AV_NOPTS_VALUE) timestamp += ic->start_time; // seek的指定的位置開始播放 ret = avformat_seek_file(ic, -1, INT64_MIN, timestamp, INT64_MAX, 0); if (ret < 0) { av_log(NULL, AV_LOG_WARNING, "%s: could not seek to position %0.3f\n", is->filename, (double)timestamp / AV_TIME_BASE); } } /* 是否為實時流媒體 */ is->realtime = is_realtime(ic); if (show_status) av_dump_format(ic, 0, is->filename, 0); // 6. 查詢AVStream // 6.1 根據使用者指定來查詢流, for (i = 0; i < ic->nb_streams; i++) { AVStream *st = ic->streams[i]; enum AVMediaType type = st->codecpar->codec_type; st->discard = AVDISCARD_ALL; if (type >= 0 && wanted_stream_spec[type] && st_index[type] == -1) if (avformat_match_stream_specifier(ic, st, wanted_stream_spec[type]) > 0) st_index[type] = i; } for (i = 0; i < AVMEDIA_TYPE_NB; i++) { if (wanted_stream_spec[i] && st_index[i] == -1) { av_log(NULL, AV_LOG_ERROR, "Stream specifier %s does not match any %s stream\n", wanted_stream_spec[i], av_get_media_type_string(i)); // st_index[i] = INT_MAX; st_index[i] = -1; } } // 6.2 利用av_find_best_stream選擇流, if (!video_disable) st_index[AVMEDIA_TYPE_VIDEO] = av_find_best_stream(ic, AVMEDIA_TYPE_VIDEO, st_index[AVMEDIA_TYPE_VIDEO], -1, NULL, 0); if (!audio_disable) st_index[AVMEDIA_TYPE_AUDIO] = av_find_best_stream(ic, AVMEDIA_TYPE_AUDIO, st_index[AVMEDIA_TYPE_AUDIO], st_index[AVMEDIA_TYPE_VIDEO], NULL, 0); if (!video_disable && !subtitle_disable) st_index[AVMEDIA_TYPE_SUBTITLE] = av_find_best_stream(ic, AVMEDIA_TYPE_SUBTITLE, st_index[AVMEDIA_TYPE_SUBTITLE], (st_index[AVMEDIA_TYPE_AUDIO] >= 0 ? st_index[AVMEDIA_TYPE_AUDIO] : st_index[AVMEDIA_TYPE_VIDEO]), NULL, 0); is->show_mode = show_mode; //7 從待處理流中獲取相關引數,設定顯示視窗的寬度、高度及寬高比 if (st_index[AVMEDIA_TYPE_VIDEO] >= 0) { AVStream *st = ic->streams[st_index[AVMEDIA_TYPE_VIDEO]]; AVCodecParameters *codecpar = st->codecpar; /*根據流和幀寬高比猜測幀的樣本寬高比。該值只是一個參考 */ AVRational sar = av_guess_sample_aspect_ratio(ic, st, NULL); if (codecpar->width) { // 設定顯示視窗的大小和寬高比 set_default_window_size(codecpar->width, codecpar->height, sar); } } /* open the streams */ /* 8. 開啟視訊、音訊解碼器。在此會開啟相應解碼器,並建立相應的解碼執行緒。 */ if (st_index[AVMEDIA_TYPE_AUDIO] >= 0) {// 如果有音訊流則開啟音訊流 stream_component_open(is, st_index[AVMEDIA_TYPE_AUDIO]); } ret = -1; if (st_index[AVMEDIA_TYPE_VIDEO] >= 0) { // 如果有視訊流則開啟視訊流 ret = stream_component_open(is, st_index[AVMEDIA_TYPE_VIDEO]); } if (is->show_mode == SHOW_MODE_NONE) { //選擇怎麼顯示,如果視訊開啟成功,就顯示視訊畫面,否則,顯示音訊對應的頻譜圖 is->show_mode = ret >= 0 ? SHOW_MODE_VIDEO : SHOW_MODE_RDFT; } if (st_index[AVMEDIA_TYPE_SUBTITLE] >= 0) { // 如果有字幕流則開啟字幕流 stream_component_open(is, st_index[AVMEDIA_TYPE_SUBTITLE]); } if (is->video_stream < 0 && is->audio_stream < 0) { av_log(NULL, AV_LOG_FATAL, "Failed to open file '%s' or configure filtergraph\n", is->filename); ret = -1; goto fail; } if (infinite_buffer < 0 && is->realtime) infinite_buffer = 1; // 如果是實時流 /* * 二、For迴圈流程 */ for (;;) { // 1 檢測是否退出 if (is->abort_request) break; // 2 檢測是否暫停/繼續 if (is->paused != is->last_paused) { is->last_paused = is->paused; if (is->paused) is->read_pause_return = av_read_pause(ic); // 網路流的時候有用 else av_read_play(ic); } #if CONFIG_RTSP_DEMUXER || CONFIG_MMSH_PROTOCOL if (is->paused && (!strcmp(ic->iformat->name, "rtsp") || (ic->pb && !strncmp(input_filename, "mmsh:", 5)))) { /* wait 10 ms to avoid trying to get another packet */ /* XXX: horrible */ // 等待10ms,避免立馬嘗試下一個Packet SDL_Delay(10); continue; } #endif // 3 檢測是否seek if (is->seek_req) { // 是否有seek請求 int64_t seek_target = is->seek_pos; int64_t seek_min = is->seek_rel > 0 ? seek_target - is->seek_rel + 2: INT64_MIN; int64_t seek_max = is->seek_rel < 0 ? seek_target - is->seek_rel - 2: INT64_MAX; // FIXME the +-2 is due to rounding being not done in the correct direction in generation // of the seek_pos/seek_rel variables // 修復由於四捨五入,沒有再seek_pos/seek_rel變數的正確方向上進行 ret = avformat_seek_file(is->ic, -1, seek_min, seek_target, seek_max, is->seek_flags); if (ret < 0) { av_log(NULL, AV_LOG_ERROR, "%s: error while seeking\n", is->ic->url); } else { /* seek的時候,要把原先的資料情況,並重啟解碼器,put flush_pkt的目的是告知解碼執行緒需要 * reset decoder */ if (is->audio_stream >= 0) { // 如果有音訊流 packet_queue_flush(&is->audioq); // 清空packet佇列資料 // 放入flush pkt, 用來開起新的一個播放序列, 解碼器讀取到flush_pkt也清空解碼器 packet_queue_put(&is->audioq, &flush_pkt); } if (is->subtitle_stream >= 0) { // 如果有字幕流 packet_queue_flush(&is->subtitleq); // 和上同理 packet_queue_put(&is->subtitleq, &flush_pkt); } if (is->video_stream >= 0) { // 如果有視訊流 packet_queue_flush(&is->videoq); // 和上同理 packet_queue_put(&is->videoq, &flush_pkt); } if (is->seek_flags & AVSEEK_FLAG_BYTE) { set_clock(&is->extclk, NAN, 0); } else { set_clock(&is->extclk, seek_target / (double)AV_TIME_BASE, 0); } } is->seek_req = 0; is->queue_attachments_req = 1; is->eof = 0; if (is->paused) step_to_next_frame(is); } // 4 檢測video是否為attached_pic if (is->queue_attachments_req) { // attached_pic 附帶的圖片。比如說一些MP3,AAC音訊檔案附帶的專輯封面,所以需要注意的是音訊檔案不一定只存在音訊流本身 if (is->video_st && is->video_st->disposition & AV_DISPOSITION_ATTACHED_PIC) { AVPacket copy = { 0 }; if ((ret = av_packet_ref(©, &is->video_st->attached_pic)) < 0) goto fail; packet_queue_put(&is->videoq, ©); packet_queue_put_nullpacket(&is->videoq, is->video_stream); } is->queue_attachments_req = 0; } // 5 檢測佇列是否已經有足夠資料 /* if the queue are full, no need to read more */ /* 快取佇列有足夠的包,不需要繼續讀取資料 15M * 8 / 3(Mbps) = 40秒*/ if (infinite_buffer<1 && (is->audioq.size + is->videoq.size + is->subtitleq.size > MAX_QUEUE_SIZE || (stream_has_enough_packets(is->audio_st, is->audio_stream, &is->audioq) && stream_has_enough_packets(is->video_st, is->video_stream, &is->videoq) && stream_has_enough_packets(is->subtitle_st, is->subtitle_stream, &is->subtitleq)))) { /* wait 10 ms */ SDL_LockMutex(wait_mutex); // 如果沒有喚醒則超時10ms退出,比如在seek操作時這裡會被喚醒 SDL_CondWaitTimeout(is->continue_read_thread, wait_mutex, 10); SDL_UnlockMutex(wait_mutex); continue; } // 6 檢測碼流是否已經播放結束 if (!is->paused // 非暫停 && // 這裡的執行是因為碼流讀取完畢後 插入空包所致 (!is->audio_st // 沒有音訊流 || (is->auddec.finished == is->audioq.serial // 或者音訊播放完畢 && frame_queue_nb_remaining(&is->sampq) == 0)) && (!is->video_st // 沒有視訊流 || (is->viddec.finished == is->videoq.serial // 或者視訊播放完畢 && frame_queue_nb_remaining(&is->pictq) == 0))) { if (loop != 1 // a 是否迴圈播放 && (!loop || --loop)) { // stream_seek不是ffmpeg的函式,是ffplay封裝的,每次seek的時候會呼叫 stream_seek(is, start_time != AV_NOPTS_VALUE ? start_time : 0, 0, 0); } else if (autoexit) { // b 是否自動退出 ret = AVERROR_EOF; goto fail; } } // 7.讀取媒體資料,得到的是音視訊分離後、解碼前的資料 ret = av_read_frame(ic, pkt); // 呼叫不會釋放pkt的資料,需要我們自己去釋放packet的資料 // 8 檢測資料是否讀取完畢 if (ret < 0) { if ((ret == AVERROR_EOF || avio_feof(ic->pb)) && !is->eof) { // 插入空包說明碼流資料讀取完畢了,之前講解碼的時候說過刷空包是為了從解碼器把所有幀都讀出來 if (is->video_stream >= 0) packet_queue_put_nullpacket(&is->videoq, is->video_stream); if (is->audio_stream >= 0) packet_queue_put_nullpacket(&is->audioq, is->audio_stream); if (is->subtitle_stream >= 0) packet_queue_put_nullpacket(&is->subtitleq, is->subtitle_stream); is->eof = 1; // 檔案讀取完畢 } if (ic->pb && ic->pb->error) break; SDL_LockMutex(wait_mutex); SDL_CondWaitTimeout(is->continue_read_thread, wait_mutex, 10); SDL_UnlockMutex(wait_mutex); continue; // 繼續迴圈 } else { is->eof = 0; } // 9 檢測是否在播放範圍內 /* check if packet is in play range specified by user, then queue, otherwise discard */ stream_start_time = ic->streams[pkt->stream_index]->start_time; // 獲取流的起始時間 pkt_ts = pkt->pts == AV_NOPTS_VALUE ? pkt->dts : pkt->pts; // 獲取packet的時間戳 // 這裡的duration是在命令列時用來指定播放長度 pkt_in_play_range = duration == AV_NOPTS_VALUE || (pkt_ts - (stream_start_time != AV_NOPTS_VALUE ? stream_start_time : 0)) * av_q2d(ic->streams[pkt->stream_index]->time_base) - (double)(start_time != AV_NOPTS_VALUE ? start_time : 0) / 1000000 <= ((double)duration / 1000000); // 10 將音視訊資料分別送入相應的queue中 if (pkt->stream_index == is->audio_stream && pkt_in_play_range) { packet_queue_put(&is->audioq, pkt); } else if (pkt->stream_index == is->video_stream && pkt_in_play_range && !(is->video_st->disposition & AV_DISPOSITION_ATTACHED_PIC)) { //printf("pkt pts:%ld, dts:%ld\n", pkt->pts, pkt->dts); packet_queue_put(&is->videoq, pkt); } else if (pkt->stream_index == is->subtitle_stream && pkt_in_play_range) { packet_queue_put(&is->subtitleq, pkt); } else { av_packet_unref(pkt);// // 不入佇列則直接釋放資料 } } // 三 退出執行緒處理 ret = 0; fail: if (ic && !is->ic) avformat_close_input(&ic); if (ret != 0) { SDL_Event event; event.type = FF_QUIT_EVENT; event.user.data1 = is; SDL_PushEvent(&event); } SDL_DestroyMutex(wait_mutex); return 0; }