import com.googlecode.javacpp.IntPointer;

 import com.googlecode.javacpp.Pointer;

 import com.googlecode.javacv.cpp.avcodec;

 import com.googlecode.javacv.cpp.avcodec.AVPacket;

 import com.googlecode.javacv.cpp.avcodec.AVPicture;

 import com.googlecode.javacv.cpp.avformat;

 import com.googlecode.javacv.cpp.avutil;

 import com.googlecode.javacv.cpp.avutil.AVDictionary;

 import com.googlecode.javacv.cpp.opencv_core;

 import static com.googlecode.javacv.cpp.opencv_core.IPL_DEPTH_8U;

 import com.googlecode.javacv.cpp.opencv_core.IplImage;

 import static com.googlecode.javacv.cpp.opencv_core.cvSize;

 import com.googlecode.javacv.cpp.opencv_highgui;

 import static com.googlecode.javacv.cpp.opencv_highgui.cvDestroyWindow;

 import static com.googlecode.javacv.cpp.opencv_highgui.cvNamedWindow;

 import com.googlecode.javacv.cpp.swscale;

 import com.googlecode.javacv.cpp.swscale.SwsContext;

 /**

  *

  * @author cuizhenfu@gmail.com

  */

 public class DecodingH264 {

     public static void main(String[] args) {

         String filePath = "Images/butterfly.h264";

         // Register all formats and codes

         avformat.av_register_all();

         avcodec.avcodec_register_all();

         // Open video file

         avformat.AVFormatContext formatCtx = avformat.avformat_alloc_context();

         if (avformat.avformat_open_input(formatCtx, filePath, null, null) != 0) {

             System.out.println("无法打开文件\n");

             return;

         }

         if (avformat.avformat_find_stream_info(formatCtx, (AVDictionary) null) < 0) {

             System.out.println("Couldn't find stream information.\n");

             return;

         }

         int videoIndex = -1;

         for (int i = 0; i < formatCtx.nb_streams(); i++) {

             if (formatCtx.streams(i).codec().codec_type() == avutil.AVMEDIA_TYPE_VIDEO) {

                 videoIndex = i;

                 break;

             }

         }

         if (videoIndex == -1) {

             System.out.println("Didn't find a video stream.\n");

             return;

         }

         avcodec.AVCodecContext codecCtx = formatCtx.streams(videoIndex).codec();

         avcodec.AVCodec codec = avcodec.avcodec_find_decoder(codecCtx.codec_id());

         if (codec == null) {

             System.out.println("Codec not found.\n");

             return;

         }

         if (avcodec.avcodec_open2(codecCtx, codec, (AVDictionary) null) < 0) {

             System.out.println("Could not open codec.\n");

         }

         avutil.AVFrame frame = avcodec.avcodec_alloc_frame();

         avutil.AVFrame frameRGB = avcodec.avcodec_alloc_frame();

         // 注意AVPicture的构造方法，参考http://blog.csdn.net/wahrlr/article/details/21970755

         // 注意avutil.av_malloc的返回值（用Pointer代替uint8_t*）

         int nunBytes = avcodec.avpicture_get_size(avutil.AV_PIX_FMT_RGB24, codecCtx.width(), codecCtx.height());

         Pointer outBuffer = avutil.av_malloc(nunBytes);

         avcodec.avpicture_fill(new AVPicture(frameRGB), outBuffer.asByteBuffer(), avutil.AV_PIX_FMT_RGB24, codecCtx.width(), codecCtx.height());

         avformat.av_dump_format(formatCtx, 0, filePath, 0);

         SwsContext img_convert_ctx = swscale.sws_getContext(codecCtx.width(), codecCtx.height(), codecCtx.pix_fmt(), codecCtx.width(), codecCtx.height(), avutil.AV_PIX_FMT_RGB24, swscale.SWS_BICUBIC, null, null, (double[])null);

         AVPacket packet = new AVPacket();

         int y_size = codecCtx.width() * codecCtx.height();

         avcodec.av_new_packet(packet, y_size);

         cvNamedWindow("butterfly.h264");

         IplImage showImage = opencv_core.cvCreateImage(cvSize(codecCtx.width(),codecCtx.height()), IPL_DEPTH_8U, 3);

         while(avformat.av_read_frame(formatCtx, packet) >= 0) {

             if(packet.stream_index() == videoIndex) {

                 IntPointer ip = new IntPointer();

                 int ret = avcodec.avcodec_decode_video2(codecCtx, frame, ip, packet);

                 if(ret < 0) {

                     System.out.println("解码错误");

                     return;

                 }

                 if(ip.get() != 0) {

                     swscale.sws_scale(img_convert_ctx, frame.data(), frame.linesize(), 0, codecCtx.height(), frameRGB.data(), frameRGB.linesize());

                     showImage.imageData(frameRGB.data(0));

                     showImage.widthStep(frameRGB.linesize().get(0));

                     opencv_highgui.cvShowImage("butterfly.h264", showImage);

                     opencv_highgui.cvWaitKey(25);

                 }

             }

             avcodec.av_free_packet(packet);

         }

         showImage.release();

         cvDestroyWindow("butterfly.h264");

         avutil.av_free(frameRGB);

         avcodec.avcodec_close(codecCtx);

         avformat.avformat_close_input(formatCtx);

     }

 }