/*** * 張暉 Hui Zhang* zhanghuicuc@gmail.com* 中國(guó)傳媒大學(xué)/數(shù)字電視技術(shù)* Communication University of China / Digital TV Technology* * 本程序?qū)崿F(xiàn)了讀取PC端攝像頭數(shù)據(jù)并進(jìn)行編碼和流媒體傳輸。**/#include <stdio.h> #include <windows.h> extern "C" { #include "libavutil\opt.h" #include "libavcodec\avcodec.h" #include "libavformat\avformat.h" #include "libavutil\time.h" #include "libavdevice\avdevice.h" #include "libswscale/swscale.h" #include "libavutil/mathematics.h" };//Show Device void show_dshow_device(){AVFormatContext *pFmtCtx = avformat_alloc_context();AVDictionary* options = NULL;av_dict_set(&options, "list_devices", "true", 0);AVInputFormat *iformat = av_find_input_format("dshow");printf("Device Info=============\n");avformat_open_input(&pFmtCtx, "video=dummy", iformat, &options);printf("========================\n"); }int flush_encoder(AVFormatContext *ifmt_ctx,AVFormatContext *ofmt_ctx, unsigned int stream_index, int framecnt);int exit_thread = 0; DWORD WINAPI MyThreadFunction(LPVOID lpParam) {while ((getchar()) != '\n');exit_thread = 1;return 0; }int main(int argc, char* argv[]) {AVFormatContext *ifmt_ctx=NULL;AVFormatContext *ofmt_ctx;AVInputFormat* ifmt;AVStream* video_st;AVCodecContext* pCodecCtx;AVCodec* pCodec;AVPacket *dec_pkt, enc_pkt;AVFrame *pframe, *pFrameYUV;struct SwsContext *img_convert_ctx;char capture_name[80] = {0};char device_name[80] = {0};int framecnt=0;int videoindex;int i;int ret;HANDLE hThread;const char* out_path = "rtmp://localhost/live/livestream"; int dec_got_frame,enc_got_frame;av_register_all();//Register Deviceavdevice_register_all();avformat_network_init();//Show Dshow Device show_dshow_device();printf("\nChoose capture device: ");if (gets(capture_name) == 0){printf("Error in gets()\n");return -1;}sprintf(device_name, "video=USB2.0 Camera");//sprintf(device_name, "video=screen-capture-recorder");ifmt=av_find_input_format("dshow");//Set own video device's nameif (avformat_open_input(&ifmt_ctx, device_name, ifmt, NULL) != 0){printf("Couldn't open input stream.（無(wú)法打開輸入流）\n");return -1;}//input initializeif (avformat_find_stream_info(ifmt_ctx, NULL)<0){printf("Couldn't find stream information.（無(wú)法獲取流信息）\n");return -1;}videoindex = -1;for (i = 0; i<ifmt_ctx->nb_streams; i++)if (ifmt_ctx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO){videoindex = i;break;}if (videoindex == -1){printf("Couldn't find a video stream.（沒(méi)有找到視頻流）\n");return -1;}if (avcodec_open2(ifmt_ctx->streams[videoindex]->codec, avcodec_find_decoder(ifmt_ctx->streams[videoindex]->codec->codec_id), NULL)<0){printf("Could not open codec.（無(wú)法打開解碼器）\n");return -1;}//output initializeavformat_alloc_output_context2(&ofmt_ctx, NULL, "flv", out_path);//output encoder initializepCodec = avcodec_find_encoder(AV_CODEC_ID_H264);if (!pCodec){printf("Can not find encoder! (沒(méi)有找到合適的編碼器！)\n");return -1;}pCodecCtx=avcodec_alloc_context3(pCodec);pCodecCtx->pix_fmt = PIX_FMT_YUV420P;pCodecCtx->width = ifmt_ctx->streams[videoindex]->codec->width;pCodecCtx->height = ifmt_ctx->streams[videoindex]->codec->height;pCodecCtx->time_base.num = 1;pCodecCtx->time_base.den = 25;pCodecCtx->bit_rate = 400000;pCodecCtx->gop_size = 250;/* Some formats want stream headers to be separate. */if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)pCodecCtx->flags |= CODEC_FLAG_GLOBAL_HEADER;//H264 codec param//pCodecCtx->me_range = 16;//pCodecCtx->max_qdiff = 4;//pCodecCtx->qcompress = 0.6;pCodecCtx->qmin = 10;pCodecCtx->qmax = 51;//Optional ParampCodecCtx->max_b_frames = 3;// Set H264 preset and tuneAVDictionary *param = 0;av_dict_set(?m, "preset", "fast", 0);av_dict_set(?m, "tune", "zerolatency", 0);if (avcodec_open2(pCodecCtx, pCodec,?m) < 0){printf("Failed to open encoder! (編碼器打開失敗！)\n");return -1;}//Add a new stream to output,should be called by the user before avformat_write_header() for muxingvideo_st = avformat_new_stream(ofmt_ctx, pCodec);if (video_st == NULL){return -1;}video_st->time_base.num = 1;video_st->time_base.den = 25;video_st->codec = pCodecCtx;//Open output URL,set before avformat_write_header() for muxingif (avio_open(&ofmt_ctx->pb,out_path, AVIO_FLAG_READ_WRITE) < 0){printf("Failed to open output file! (輸出文件打開失敗！)\n");return -1;}//Show some Informationav_dump_format(ofmt_ctx, 0, out_path, 1);//Write File Headeravformat_write_header(ofmt_ctx,NULL);//prepare before decode and encodedec_pkt = (AVPacket *)av_malloc(sizeof(AVPacket));//enc_pkt = (AVPacket *)av_malloc(sizeof(AVPacket));//camera data has a pix fmt of RGB,convert it to YUV420img_convert_ctx = sws_getContext(ifmt_ctx->streams[videoindex]->codec->width, ifmt_ctx->streams[videoindex]->codec->height, ifmt_ctx->streams[videoindex]->codec->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);pFrameYUV = av_frame_alloc();uint8_t *out_buffer = (uint8_t *)av_malloc(avpicture_get_size(PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));avpicture_fill((AVPicture *)pFrameYUV, out_buffer, PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);printf("\n --------call started----------\n\n");printf("Press enter to stop...");hThread = CreateThread(NULL, // default security attributes0, // use default stack size MyThreadFunction, // thread function nameNULL, // argument to thread function 0, // use default creation flags NULL); // returns the thread identifier //start decode and encodeint64_t start_time=av_gettime();while (av_read_frame(ifmt_ctx, dec_pkt) >= 0){ if (exit_thread)break;av_log(NULL, AV_LOG_DEBUG, "Going to reencode the frame\n");pframe = av_frame_alloc();if (!pframe) {ret = AVERROR(ENOMEM);return -1;}//av_packet_rescale_ts(dec_pkt, ifmt_ctx->streams[dec_pkt->stream_index]->time_base,// ifmt_ctx->streams[dec_pkt->stream_index]->codec->time_base);ret = avcodec_decode_video2(ifmt_ctx->streams[dec_pkt->stream_index]->codec, pframe,&dec_got_frame, dec_pkt);if (ret < 0) {av_frame_free(&pframe);av_log(NULL, AV_LOG_ERROR, "Decoding failed\n");break;}if (dec_got_frame){sws_scale(img_convert_ctx, (const uint8_t* const*)pframe->data, pframe->linesize, 0, pCodecCtx->height, pFrameYUV->data, pFrameYUV->linesize); enc_pkt.data = NULL;enc_pkt.size = 0;av_init_packet(&enc_pkt);ret = avcodec_encode_video2(pCodecCtx, &enc_pkt, pFrameYUV, &enc_got_frame);av_frame_free(&pframe);if (enc_got_frame == 1){//printf("Succeed to encode frame: %5d\tsize:%5d\n", framecnt, enc_pkt.size);framecnt++; enc_pkt.stream_index = video_st->index;//Write PTSAVRational time_base = ofmt_ctx->streams[videoindex]->time_base;//{ 1, 1000 };AVRational r_framerate1 = ifmt_ctx->streams[videoindex]->r_frame_rate;// { 50, 2 };AVRational time_base_q = { 1, AV_TIME_BASE };//Duration between 2 frames (us)int64_t calc_duration = (double)(AV_TIME_BASE)*(1 / av_q2d(r_framerate1)); //內(nèi)部時(shí)間戳//Parameters//enc_pkt.pts = (double)(framecnt*calc_duration)*(double)(av_q2d(time_base_q)) / (double)(av_q2d(time_base));enc_pkt.pts = av_rescale_q(framecnt*calc_duration, time_base_q, time_base);enc_pkt.dts = enc_pkt.pts;enc_pkt.duration = av_rescale_q(calc_duration, time_base_q, time_base); //(double)(calc_duration)*(double)(av_q2d(time_base_q)) / (double)(av_q2d(time_base));enc_pkt.pos = -1;//Delayint64_t pts_time = av_rescale_q(enc_pkt.dts, time_base, time_base_q);int64_t now_time = av_gettime() - start_time;if (pts_time > now_time)av_usleep(pts_time - now_time);ret = av_interleaved_write_frame(ofmt_ctx, &enc_pkt);av_free_packet(&enc_pkt);}}else {av_frame_free(&pframe);}av_free_packet(dec_pkt);}//Flush Encoderret = flush_encoder(ifmt_ctx,ofmt_ctx,0,framecnt);if (ret < 0) {printf("Flushing encoder failed\n");return -1;}//Write file trailerav_write_trailer(ofmt_ctx);//Cleanif (video_st)avcodec_close(video_st->codec);av_free(out_buffer);avio_close(ofmt_ctx->pb);avformat_free_context(ifmt_ctx);avformat_free_context(ofmt_ctx);CloseHandle(hThread);return 0; }int flush_encoder(AVFormatContext *ifmt_ctx, AVFormatContext *ofmt_ctx, unsigned int stream_index, int framecnt){int ret;int got_frame;AVPacket enc_pkt;if (!(ofmt_ctx->streams[stream_index]->codec->codec->capabilities &CODEC_CAP_DELAY))return 0;while (1) {enc_pkt.data = NULL;enc_pkt.size = 0;av_init_packet(&enc_pkt);ret = avcodec_encode_video2 (ofmt_ctx->streams[stream_index]->codec, &enc_pkt,NULL, &got_frame);av_frame_free(NULL);if (ret < 0)break;if (!got_frame){ret=0;break;}printf("Flush Encoder: Succeed to encode 1 frame!\tsize:%5d\n",enc_pkt.size);//Write PTSAVRational time_base = ofmt_ctx->streams[stream_index]->time_base;//{ 1, 1000 };AVRational r_framerate1 = ifmt_ctx->streams[stream_index]->r_frame_rate;// { 50, 2 };AVRational time_base_q = { 1, AV_TIME_BASE };//Duration between 2 frames (us)int64_t calc_duration = (double)(AV_TIME_BASE)*(1 / av_q2d(r_framerate1)); //內(nèi)部時(shí)間戳//Parametersenc_pkt.pts = av_rescale_q(framecnt*calc_duration, time_base_q, time_base);enc_pkt.dts = enc_pkt.pts;enc_pkt.duration = av_rescale_q(calc_duration, time_base_q, time_base);/* copy packet*///轉(zhuǎn)換PTS/DTS（Convert PTS/DTS）enc_pkt.pos = -1;framecnt++;ofmt_ctx->duration=enc_pkt.duration * framecnt;/* mux encoded frame */ret = av_interleaved_write_frame(ofmt_ctx, &enc_pkt);if (ret < 0)break;}return ret; }
網(wǎng)上找的代碼，親測(cè)有效，服務(wù)器推薦使用Adobe Media Server 5，不會(huì)搭建服務(wù)器環(huán)境請(qǐng)看 Adobe Media Server 5（AMS）的安裝及使用

代碼工程下載：

https://download.csdn.net/download/liujiayu2/10513179

附注：

將上述代碼中rtmp改寫成rtsp流的時(shí)候就會(huì)出錯(cuò)，錯(cuò)誤碼?AVERROR_PROTOCOL_NOT_FOUND

但是使用ffmpeg命令推攝像頭rtsp流的時(shí)候是沒(méi)有問(wèn)題的。不知道什么原因

總結(jié)

以上是生活随笔為你收集整理的ffmpeg推送摄像头rtmp流的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。