基于FFmpeg的Swscale示例代码,Java代码库,德仔网

代码语言
代码分类
【Java】基于FFmpeg的Swscale示例
作者:雷霄骅 / 发布于2015/1/12/ 543

	/**
	* 最简单的基于FFmpeg的Swscale示例
	* Simplest FFmpeg Swscale
	*
	* 雷霄骅 Lei Xiaohua
	* leixiaohua1020@126.com
	* 中国传媒大学/数字电视技术
	* Communication University of China / Digital TV Technology
	* http://blog.csdn.net/leixiaohua1020
	*
	* 本程序使用libswscale对像素数据进行缩放转换等处理。
	* 是最简单的libswscale的教程。
	*
	* This software uses libswscale to scale / convert pixels.
	* It the simplest tutorial about libswscale.
	*/
	
	#include <stdio.h>
	
	
	extern "C"
	{
	#include "libswscale/swscale.h"
	#include "libavutil/opt.h"
	#include "libavutil/imgutils.h"
	};
	
	
	int main(int argc, char* argv[])
	{
	 //Parameters
	 FILE *src_file =fopen("sintel_480x272_yuv420p.yuv", "rb");
	 const int src_w=480,src_h=272;
	 AVPixelFormat src_pixfmt=AV_PIX_FMT_YUV420P;
	
	 int src_bpp=av_get_bits_per_pixel(av_pix_fmt_desc_get(src_pixfmt));
	
	 FILE *dst_file = fopen("sintel_1280x720_rgb24.rgb", "wb");
	 const int dst_w=1280,dst_h=720;
	 AVPixelFormat dst_pixfmt=AV_PIX_FMT_RGB24;
	 int dst_bpp=av_get_bits_per_pixel(av_pix_fmt_desc_get(dst_pixfmt));
	
	 //Structures
	 uint8_t *src_data[4];
	 int src_linesize[4];
	
	 uint8_t *dst_data[4];
	 int dst_linesize[4];
	
	 int rescale_method=SWS_BICUBIC;
	 struct SwsContext *img_convert_ctx;
	 uint8_t *temp_buffer=(uint8_t *)malloc(src_w*src_h*src_bpp/8);
	
	 int frame_idx=0;
	 int ret=0;
	 ret= av_image_alloc(src_data, src_linesize,src_w, src_h, src_pixfmt, 1);
	 if (ret< 0) {
	 printf( "Could not allocate source image\n");
	 return -1;
	 }
	 ret = av_image_alloc(dst_data, dst_linesize,dst_w, dst_h, dst_pixfmt, 1);
	 if (ret< 0) {
	 printf( "Could not allocate destination image\n");
	 return -1;
	 }
	 //-----------------------------
	 //Init Method 1
	 img_convert_ctx =sws_alloc_context();
	 //Show AVOption
	 av_opt_show2(img_convert_ctx,stdout,AV_OPT_FLAG_VIDEO_PARAM,NULL);
	 //Set Value
	 av_opt_set_int(img_convert_ctx,"sws_flags",SWS_BICUBIC|SWS_PRINT_INFO,NULL);
	 av_opt_set_int(img_convert_ctx,"srcw",src_w,NULL);
	 av_opt_set_int(img_convert_ctx,"srch",src_h,NULL);
	 av_opt_set_int(img_convert_ctx,"src_format",src_pixfmt,NULL);
	 //'0' for MPEG (Y:0-235);'1' for JPEG (Y:0-255)
	 av_opt_set_int(img_convert_ctx,"src_range",1,NULL);
	 av_opt_set_int(img_convert_ctx,"dstw",dst_w,NULL);
	 av_opt_set_int(img_convert_ctx,"dsth",dst_h,NULL);
	 av_opt_set_int(img_convert_ctx,"dst_format",dst_pixfmt,NULL);
	 av_opt_set_int(img_convert_ctx,"dst_range",1,NULL);
	 sws_init_context(img_convert_ctx,NULL,NULL);
	
	 //Init Method 2
	 //img_convert_ctx = sws_getContext(src_w, src_h,src_pixfmt, dst_w, dst_h, dst_pixfmt,
	 // rescale_method, NULL, NULL, NULL);
	 //-----------------------------
	 /*
	 //Colorspace
	 ret=sws_setColorspaceDetails(img_convert_ctx,sws_getCoefficients(SWS_CS_ITU601),0,
	 sws_getCoefficients(SWS_CS_ITU709),0,
	 0, 1 << 16, 1 << 16);
	 if (ret==-1) {
	 printf( "Colorspace not support.\n");
	 return -1;
	 }
	 */
	 while(1)
	 {
	 if (fread(temp_buffer, 1, src_w*src_h*src_bpp/8, src_file) != src_w*src_h*src_bpp/8){
	 break;
	 }
	
	 switch(src_pixfmt){
	 case AV_PIX_FMT_GRAY8:{
	 memcpy(src_data[0],temp_buffer,src_w*src_h);
	 break;
	 }
	 case AV_PIX_FMT_YUV420P:{
	 memcpy(src_data[0],temp_buffer,src_w*src_h); //Y
	 memcpy(src_data[1],temp_buffer+src_w*src_h,src_w*src_h/4); //U
	 memcpy(src_data[2],temp_buffer+src_w*src_h*5/4,src_w*src_h/4); //V
	 break;
	 }
	 case AV_PIX_FMT_YUV422P:{
	 memcpy(src_data[0],temp_buffer,src_w*src_h); //Y
	 memcpy(src_data[1],temp_buffer+src_w*src_h,src_w*src_h/2); //U
	 memcpy(src_data[2],temp_buffer+src_w*src_h*3/2,src_w*src_h/2); //V
	 break;
	 }
	 case AV_PIX_FMT_YUV444P:{
	 memcpy(src_data[0],temp_buffer,src_w*src_h); //Y
	 memcpy(src_data[1],temp_buffer+src_w*src_h,src_w*src_h); //U
	 memcpy(src_data[2],temp_buffer+src_w*src_h*2,src_w*src_h); //V
	 break;
	 }
	 case AV_PIX_FMT_YUYV422:{
	 memcpy(src_data[0],temp_buffer,src_w*src_h*2); //Packed
	 break;
	 }
	 case AV_PIX_FMT_RGB24:{
	 memcpy(src_data[0],temp_buffer,src_w*src_h*3); //Packed
	 break;
	 }
	 default:{
	 printf("Not Support Input Pixel Format.\n");
	 break;
	 }
	 }
	
	 sws_scale(img_convert_ctx, src_data, src_linesize, 0, src_h, dst_data, dst_linesize);
	 printf("Finish process frame %5d\n",frame_idx);
	 frame_idx++;
	
	 switch(dst_pixfmt){
	 case AV_PIX_FMT_GRAY8:{
	 fwrite(dst_data[0],1,dst_w*dst_h,dst_file);
	 break;
	 }
	 case AV_PIX_FMT_YUV420P:{
	 fwrite(dst_data[0],1,dst_w*dst_h,dst_file); //Y
	 fwrite(dst_data[1],1,dst_w*dst_h/4,dst_file); //U
	 fwrite(dst_data[2],1,dst_w*dst_h/4,dst_file); //V
	 break;
	 }
	 case AV_PIX_FMT_YUV422P:{
	 fwrite(dst_data[0],1,dst_w*dst_h,dst_file); //Y
	 fwrite(dst_data[1],1,dst_w*dst_h/2,dst_file); //U
	 fwrite(dst_data[2],1,dst_w*dst_h/2,dst_file); //V
	 break;
	 }
	 case AV_PIX_FMT_YUV444P:{
	 fwrite(dst_data[0],1,dst_w*dst_h,dst_file); //Y
	 fwrite(dst_data[1],1,dst_w*dst_h,dst_file); //U
	 fwrite(dst_data[2],1,dst_w*dst_h,dst_file); //V
	 break;
	 }
	 case AV_PIX_FMT_YUYV422:{
	 fwrite(dst_data[0],1,dst_w*dst_h*2,dst_file); //Packed
	 break;
	 }
	 case AV_PIX_FMT_RGB24:{
	 fwrite(dst_data[0],1,dst_w*dst_h*3,dst_file); //Packed
	 break;
	 }
	 default:{
	 printf("Not Support Output Pixel Format.\n");
	 break;
	 }
	 }
	 }
	
	 sws_freeContext(img_convert_ctx);
	
	 free(temp_buffer);
	 fclose(dst_file);
	 av_freep(&src_data[0]);
	 av_freep(&dst_data[0]);
	
	 return 0;
	}
试试其它关键字
　FFmpeg　　Swscale　
同语言下
可能有用的
雷霄骅贡献的其它代码(1)
.基于FFmpeg的Swscale示例