228 lines
6.2 KiB
C++
Executable File
228 lines
6.2 KiB
C++
Executable File
/**
|
|
* 本程序实现对内存中数据的读写并播放
|
|
* 2016-11-03
|
|
*
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
|
|
#define __STDC_CONSTANT_MACROS
|
|
|
|
#ifdef _WIN32
|
|
//Windows
|
|
extern "C"
|
|
{
|
|
#include "libavcodec/avcodec.h"
|
|
#include "libavformat/avformat.h"
|
|
#include "libswscale/swscale.h"
|
|
#include "SDL/SDL.h"
|
|
};
|
|
#else
|
|
//Linux...
|
|
#ifdef __cplusplus
|
|
extern "C"
|
|
{
|
|
#endif
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavformat/avformat.h>
|
|
#include <libswscale/swscale.h>
|
|
#include <SDL/SDL.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <sys/socket.h>
|
|
#include <resolv.h>
|
|
#include <stdlib.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
#ifdef __cplusplus
|
|
};
|
|
#endif
|
|
#endif
|
|
#define MAXBUF 1024
|
|
//Output YUV420P
|
|
#define OUTPUT_YUV420P 0
|
|
//FILE *fp_open=NULL;
|
|
int sockfd;
|
|
|
|
//Callback
|
|
int read_buffer(void *opaque, uint8_t *buf, int buf_size){
|
|
/* 接收对方发过来的文件*/
|
|
int true_size;
|
|
if (true_size=recv(sockfd,buf,buf_size,0)){
|
|
return true_size;
|
|
}else{
|
|
return -1;
|
|
}
|
|
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
//socket
|
|
struct sockaddr_in6 dest; // IPv6
|
|
|
|
if (argc != 3) {
|
|
printf
|
|
("参数格式错误!正确用法如下:\n\t\t%s IP地址 端口\n\t比如:\t%s 127.0.0.1 80\n此程序用来从某个 IP 地址的服务器某个端口接收最多 MAXBUF 个字节的消息",
|
|
argv[0], argv[0]);
|
|
exit(0);
|
|
}
|
|
if ((sockfd = socket(AF_INET6, SOCK_STREAM, 0)) < 0) { // IPv6
|
|
perror("Socket");
|
|
exit(errno);
|
|
}
|
|
printf("socket created\n");
|
|
|
|
/* 初始化服务器端(对方)的地址和端口信息 */
|
|
bzero(&dest, sizeof(dest));
|
|
dest.sin6_family = AF_INET6; // IPv6
|
|
dest.sin6_port = htons(atoi(argv[2])); // IPv6
|
|
if ( inet_pton(AF_INET6, argv[1], &dest.sin6_addr) < 0 ) { // IPv6
|
|
perror(argv[1]);
|
|
exit(errno);
|
|
}
|
|
printf("address created\n");
|
|
|
|
/* 连接服务器 */
|
|
if (connect(sockfd, (struct sockaddr *) &dest, sizeof(dest)) != 0) {
|
|
perror("Connect ");
|
|
exit(errno);
|
|
}
|
|
printf("server connected\n");
|
|
AVFormatContext *pFormatCtx;
|
|
int i, videoindex;
|
|
AVCodecContext *pCodecCtx;
|
|
AVCodec *pCodec;
|
|
//char filepath[]="1.mp4";
|
|
|
|
av_register_all();
|
|
avformat_network_init();
|
|
pFormatCtx = avformat_alloc_context();
|
|
|
|
//fp_open=fopen(filepath,"rb+");
|
|
//Init AVIOContext
|
|
unsigned char *aviobuffer=(unsigned char *)av_malloc(32768);
|
|
AVIOContext *avio =avio_alloc_context(aviobuffer, 32768,0,NULL,read_buffer,NULL,NULL);
|
|
pFormatCtx->pb=avio;
|
|
|
|
if(avformat_open_input(&pFormatCtx,NULL,NULL,NULL)!=0){
|
|
printf("Couldn't open input stream.\n");
|
|
return -1;
|
|
}
|
|
if(avformat_find_stream_info(pFormatCtx,NULL)<0){
|
|
printf("Couldn't find stream information.\n");
|
|
return -1;
|
|
}
|
|
videoindex=-1;
|
|
for(i=0; i<pFormatCtx->nb_streams; i++)
|
|
if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO){
|
|
videoindex=i;
|
|
break;
|
|
}
|
|
if(videoindex==-1){
|
|
printf("Didn't find a video stream.\n");
|
|
return -1;
|
|
}
|
|
pCodecCtx=pFormatCtx->streams[videoindex]->codec;
|
|
pCodec=avcodec_find_decoder(pCodecCtx->codec_id);
|
|
if(pCodec==NULL){
|
|
printf("Codec not found.\n");
|
|
return -1;
|
|
}
|
|
if(avcodec_open2(pCodecCtx, pCodec,NULL)<0){
|
|
printf("Could not open codec.\n");
|
|
return -1;
|
|
}
|
|
AVFrame *pFrame,*pFrameYUV;
|
|
pFrame=av_frame_alloc();
|
|
pFrameYUV=av_frame_alloc();
|
|
//uint8_t *out_buffer=(uint8_t *)av_malloc(avpicture_get_size(AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));
|
|
//avpicture_fill((AVPicture *)pFrameYUV, out_buffer, AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);
|
|
//SDL----------------------------
|
|
if(SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO | SDL_INIT_TIMER)) {
|
|
printf( "Could not initialize SDL - %s\n", SDL_GetError());
|
|
return -1;
|
|
}
|
|
|
|
int screen_w=0,screen_h=0;
|
|
SDL_Surface *screen;
|
|
screen_w = pCodecCtx->width;
|
|
screen_h = pCodecCtx->height;
|
|
screen = SDL_SetVideoMode(screen_w, screen_h, 0,0);
|
|
|
|
if(!screen) {
|
|
printf("SDL: could not set video mode - exiting:%s\n",SDL_GetError());
|
|
return -1;
|
|
}
|
|
SDL_Overlay *bmp;
|
|
bmp = SDL_CreateYUVOverlay(pCodecCtx->width, pCodecCtx->height,SDL_YV12_OVERLAY, screen);
|
|
SDL_Rect rect;
|
|
rect.x = 0;
|
|
rect.y = 0;
|
|
rect.w = screen_w;
|
|
rect.h = screen_h;
|
|
//SDL End------------------------
|
|
int ret, got_picture;
|
|
|
|
AVPacket *packet=(AVPacket *)av_malloc(sizeof(AVPacket));
|
|
|
|
#if OUTPUT_YUV420P
|
|
FILE *fp_yuv=fopen("output.yuv","wb+");
|
|
#endif
|
|
SDL_WM_SetCaption("Simplest FFmpeg Mem Player",NULL);
|
|
|
|
struct SwsContext *img_convert_ctx;
|
|
img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, AV_PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);
|
|
//------------------------------
|
|
while(av_read_frame(pFormatCtx, packet)>=0){
|
|
if(packet->stream_index==videoindex){
|
|
ret = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);
|
|
if(ret < 0){
|
|
printf("Decode Error.\n");
|
|
return -1;
|
|
}
|
|
if(got_picture){
|
|
SDL_LockYUVOverlay(bmp);
|
|
pFrameYUV->data[0]=bmp->pixels[0];
|
|
pFrameYUV->data[1]=bmp->pixels[2];
|
|
pFrameYUV->data[2]=bmp->pixels[1];
|
|
pFrameYUV->linesize[0]=bmp->pitches[0];
|
|
pFrameYUV->linesize[1]=bmp->pitches[2];
|
|
pFrameYUV->linesize[2]=bmp->pitches[1];
|
|
sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize, 0, pCodecCtx->height, pFrameYUV->data, pFrameYUV->linesize);
|
|
#if OUTPUT_YUV420P
|
|
int y_size=pCodecCtx->width*pCodecCtx->height;
|
|
fwrite(pFrameYUV->data[0],1,y_size,fp_yuv); //Y
|
|
fwrite(pFrameYUV->data[1],1,y_size/4,fp_yuv); //U
|
|
fwrite(pFrameYUV->data[2],1,y_size/4,fp_yuv); //V
|
|
#endif
|
|
SDL_UnlockYUVOverlay(bmp);
|
|
|
|
SDL_DisplayYUVOverlay(bmp, &rect);
|
|
//Delay 40ms
|
|
SDL_Delay(40);
|
|
}
|
|
}
|
|
av_free_packet(packet);
|
|
}
|
|
sws_freeContext(img_convert_ctx);
|
|
|
|
#if OUTPUT_YUV420P
|
|
fclose(fp_yuv);
|
|
#endif
|
|
|
|
//fclose(fp_open);
|
|
|
|
SDL_Quit();
|
|
|
|
//av_free(out_buffer);
|
|
av_free(pFrameYUV);
|
|
avcodec_close(pCodecCtx);
|
|
avformat_close_input(&pFormatCtx);
|
|
|
|
return 0;
|
|
}
|
|
|