/********************************************************************
filename:   CRTSPPlayer.h
created:    2013-03-25
author:     firehood
purpose:    ffmpeg库实现的RTSP视频播放器
*********************************************************************/
#pragma once
#include "windows.h"  extern "C"
{
#include "libavformat\avformat.h"
#include "libavcodec\avcodec.h"
#include "libswscale\swscale.h"
};  // 播放状态
enum RTSP_PLAYSTATUS
{  RTSP_PLAYSTATUS_NONE,       // 未知状态（未播放）  RTSP_PLAYSTATUS_PLAYING,    // 正在播放  RTSP_PLAYSTATUS_PAUSE,      // 已暂停  RTSP_PLAYSTATUS_STOP,       // 已停止
};  class CRTSPPlayer
{
public:  CRTSPPlayer(HWND hWnd, LPRECT lpRect);  ~CRTSPPlayer(void);
public:  // 打开媒体文件  BOOL OpenMedia(LPCTSTR pFileName);  // 播放  void Play();  // 暂停  void Pause();  // 停止  void Stop();  // 获取播放状态  RTSP_PLAYSTATUS GetPlayStatus(void);
private:  // 解码初始化  int DecodeInit(LPCTSTR pFileName);  // 卸载  void DecodeUninit();  // 开始解码线程  BOOL StartDecodeThread();  // 停止解码线程  void StopDecodeThread();  // 解码线程  static int WINAPI ThreadDecodeVideo(LPVOID lpParam);  // 开始解码任务  int BeginDecode();  // 显示  void Display();  // 图像转换  int ImgConvert(AVPicture * dst, PixelFormat dstFormt, const AVPicture * src, PixelFormat srcFormt, int src_width, int src_height);  // 设置播放状态  void SetPlayStatus(RTSP_PLAYSTATUS playStatus);
private:  HANDLE  m_hDecodeThread;  BOOL    m_bExitDecodeThread;  TCHAR   m_strFilePath[MAX_PATH];  AVFormatContext* m_pFormatContext;  AVCodecContext*  m_pCodecContext;  AVCodec* m_pCodec;  AVPacket m_struPacket;  int m_nStreamIndex;  AVFrame* m_pFrameYUV;  AVFrame* m_pFrameRGB;  int     m_nFrameWidth;   int     m_nFrameHeight;  BYTE*   m_pBufRGB;        // 解码后的RGB数据  RTSP_PLAYSTATUS  m_nPlayStatus;  HWND    m_hWnd;  RECT    m_rcWnd;
};  

/********************************************************************
filename:   CRTSPPlayer.cpp
created:    2013-03-25
author:     firehood
purpose:    ffmpeg库实现的RTSP视频播放器
*********************************************************************/
#include "StdAfx.h"
#include "RTSPPlayer.h"  #pragma comment(lib, "avformat.lib")
#pragma comment(lib, "avcodec.lib")
#pragma comment(lib, "swscale.lib")
#pragma comment(lib, "avutil.lib")  #define SHOW_TITLE  const char* WcharToUtf8(const wchar_t *pwStr)
{    if (pwStr == NULL)    {    return NULL;    }    int len = WideCharToMultiByte(CP_UTF8, 0, pwStr, -1, NULL, 0, NULL, NULL);    if (len <= 0)    {    return NULL;    }    char *pStr = new char[len];    WideCharToMultiByte(CP_UTF8, 0, pwStr, -1, pStr, len, NULL, NULL);    return pStr;
}    CRTSPPlayer::CRTSPPlayer(HWND hWnd, LPRECT lpRect):
m_hWnd(hWnd),
m_rcWnd(*lpRect),
m_hDecodeThread(NULL),
m_bExitDecodeThread(FALSE),
m_nFrameWidth(0),
m_nFrameHeight(0),
m_pFormatContext(NULL),
m_pCodecContext(NULL),
m_pCodec(NULL),
m_nStreamIndex(-1),
m_pFrameYUV(NULL),
m_pFrameRGB(NULL),
m_pBufRGB(NULL),
m_nPlayStatus(RTSP_PLAYSTATUS_NONE)
{  memset(m_strFilePath,0,sizeof(m_strFilePath));
}  CRTSPPlayer::~CRTSPPlayer(void)
{  DecodeUninit();
}  // 打开媒体文件
BOOL CRTSPPlayer::OpenMedia(LPCTSTR pFileName)
{  if(pFileName == NULL)  return FALSE;  DecodeUninit();  memcpy(m_strFilePath,pFileName,sizeof(m_strFilePath));  DecodeInit(m_strFilePath);  return TRUE;
}  // 播放
void CRTSPPlayer::Play()
{   if(GetPlayStatus() == RTSP_PLAYSTATUS_STOP)  {  DecodeInit(m_strFilePath);  }  BOOL bRet = StartDecodeThread();  if(bRet)  {  SetPlayStatus(RTSP_PLAYSTATUS_PLAYING);  }
}  // 暂停
void CRTSPPlayer::Pause()
{  StopDecodeThread();  SetPlayStatus(RTSP_PLAYSTATUS_PAUSE);
}  // 停止
void CRTSPPlayer::Stop()
{  StopDecodeThread();  DecodeUninit();  SetPlayStatus(RTSP_PLAYSTATUS_STOP);
}  BOOL CRTSPPlayer::StartDecodeThread()
{  if(m_hDecodeThread == NULL)  {  m_hDecodeThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)ThreadDecodeVideo, this, 0, NULL);  }  return m_hDecodeThread ? TRUE : FALSE;
}  void CRTSPPlayer::StopDecodeThread()
{  if(m_hDecodeThread)  {  m_bExitDecodeThread = TRUE;  WaitForSingleObject(m_hDecodeThread,INFINITE);  CloseHandle(m_hDecodeThread);  m_hDecodeThread = NULL;  }
}  int CRTSPPlayer::ImgConvert(AVPicture * dst, PixelFormat dst_pix_fmt, const AVPicture * src, PixelFormat src_pix_fmt, int src_width, int src_height)
{  unsigned char * srcSlice[4];  int srcStride[4] = {0};  unsigned char * dstSlice[4];  int dstStride[4] = {0};  for (int i=0; i<4; i++)  {  srcSlice[i] = src->data[i];  srcStride[i] = src->linesize[i];  dstSlice[i] = dst->data[i];  dstStride[i] = dst->linesize[i];  }  SwsContext *pSwsContext = sws_getContext(src_width, src_height, src_pix_fmt, src_width, src_height, dst_pix_fmt, SWS_BICUBIC, NULL, NULL, NULL);  int nRet = sws_scale(pSwsContext, srcSlice, srcStride, 0, src_height, dstSlice, dstStride);  if (pSwsContext != NULL)  {  sws_freeContext(pSwsContext);  }  return nRet;
}  int WINAPI CRTSPPlayer::ThreadDecodeVideo(LPVOID lpParam)
{  CRTSPPlayer *pPlayer = (CRTSPPlayer*)lpParam;  pPlayer->BeginDecode();  return 0;
}  int CRTSPPlayer::DecodeInit(LPCTSTR pFileName)
{  if(pFileName == NULL)  {  return -1;  }  av_register_all();  #ifdef  UNICODE     const char *filePath = WcharToUtf8(pFileName);   // Open video  if (av_open_input_file(&m_pFormatContext, filePath, NULL, 0, NULL) != 0)  {  return -2; // Couldn't open file  }  delete[] filePath;
#else  // Open video  if (av_open_input_file(&m_pFormatContext, pFileName, NULL, 0, NULL) != 0)  {  return -2; // Couldn't open file  }
#endif  // Retrieve stream information  if (av_find_stream_info(m_pFormatContext) < 0)  {  return -3; // Couldn't find stream information  }  // Find the first video stream  for (UINT i=0; i<m_pFormatContext->nb_streams; i++)  {  if (m_pFormatContext->streams[i]->codec->codec_type == CODEC_TYPE_VIDEO)  {  m_nStreamIndex = i;  break;  }  }  if (m_nStreamIndex == -1)  {  return -4; // Didn't find a video stream  }  // Get a pointer to the codec context for the video stream  m_pCodecContext = m_pFormatContext->streams[m_nStreamIndex]->codec;  // Find the decoder for the video stream  m_pCodec = avcodec_find_decoder(m_pCodecContext->codec_id);  if (m_pCodec == NULL)  {  return -5 ; // Codec not found  }  // Inform the codec that we can handle truncated bitstreams -- i.e.,  // bitstreams where frame boundaries can fall in the middle of packets  if (m_pCodec->capabilities & CODEC_CAP_TRUNCATED)  {  m_pCodecContext->flags |= CODEC_FLAG_TRUNCATED;  // we do not send complete frames  }  // Open codec  if (avcodec_open(m_pCodecContext, m_pCodec) < 0)  {  return -6; // Could not open codec  }  // Allocate video frame  m_pFrameYUV = avcodec_alloc_frame();  // Allocate an AVFrame structure  m_pFrameRGB = avcodec_alloc_frame();  // Determine required buffer size and allocate buffer  int numBytes = avpicture_get_size(PIX_FMT_BGR24, m_pCodecContext->width, m_pCodecContext->height);  m_pBufRGB = new BYTE [numBytes];  memset(m_pBufRGB,0,numBytes);  // Assign appropriate parts of buffer to image planes in m_pFrameRGB  avpicture_fill((AVPicture *)m_pFrameRGB, m_pBufRGB, PIX_FMT_BGR24, m_pCodecContext->width, m_pCodecContext->height);  m_nFrameWidth  = m_pCodecContext->width;  m_nFrameHeight = m_pCodecContext->height;  return 0;
}  void CRTSPPlayer::DecodeUninit()
{  // Close the codec  if (m_pCodecContext)  {  avcodec_close(m_pCodecContext);  //av_free(m_pCodec);  m_pCodecContext = NULL;  m_pCodec = NULL;  }  // Close the video file  if (m_pFormatContext)  {  av_close_input_file(m_pFormatContext);  m_pFormatContext = NULL;  }  if (m_pFrameYUV)  {  av_free(m_pFrameYUV);  m_pFrameYUV = NULL;  }  if (m_pFrameRGB)  {  av_free(m_pFrameRGB);  m_pFrameRGB = NULL;  }  if (m_pBufRGB)  {  delete [] m_pBufRGB;  m_pBufRGB = NULL;  }
}  int CRTSPPlayer::BeginDecode()
{  int bytesRemaining = 0, bytesDecoded;  BYTE * rawData = NULL;  int frameFinished = 0;  m_struPacket.data = NULL;  m_struPacket.size = 0;  m_bExitDecodeThread = FALSE;  while (!m_bExitDecodeThread && m_pFormatContext)  {  // Read the next packet, skipping all packets that aren't for this stream  do  {  // Read new packet  if (av_read_frame(m_pFormatContext, &m_struPacket) < 0)  {  return -2;  }  } while (m_struPacket.stream_index != m_nStreamIndex);  bytesRemaining = m_struPacket.size;  rawData = m_struPacket.data;  // Work on the current packet until we have decoded all of it  while (bytesRemaining > 0)  {  // Decode the next chunk of data  bytesDecoded = avcodec_decode_video(m_pCodecContext, m_pFrameYUV, &frameFinished, rawData, bytesRemaining);  // Was there an error?  if (bytesDecoded < 0)  {  return -1;  }  bytesRemaining -= bytesDecoded;  rawData += bytesDecoded;  // Did we finish the current frame? Then we can return  if (frameFinished)  {  ImgConvert(  (AVPicture *)m_pFrameRGB,  PIX_FMT_BGR24,  (AVPicture *)m_pFrameYUV,  m_pCodecContext->pix_fmt,  m_pCodecContext->width,  m_pCodecContext->height);  Display();  }  }  }  m_hDecodeThread = NULL;  return 0;
}  void CRTSPPlayer::Display()
{  HDC hdc = GetDC(m_hWnd);  // 创建内存DC  HDC hMemDc = CreateCompatibleDC(hdc);       // 创建位图  BITMAPINFOHEADER bmpHdr = {0};    bmpHdr.biSize = sizeof (BITMAPINFOHEADER);    bmpHdr.biWidth = m_nFrameWidth;    bmpHdr.biHeight = -m_nFrameHeight;    bmpHdr.biPlanes = 1;    bmpHdr.biBitCount = 24;    bmpHdr.biCompression = BI_RGB;    BYTE *pData = NULL;     HBITMAP hBitmap = CreateDIBSection (NULL, (BITMAPINFO *)&bmpHdr, DIB_RGB_COLORS, (void**)&pData, NULL, 0);    try  {  memcpy(pData, m_pBufRGB, m_nFrameWidth * m_nFrameHeight * 3);  }  catch (CMemoryException* e)  {  }  HBITMAP hOldBitmap = (HBITMAP)SelectObject(hMemDc, hBitmap);  #ifdef SHOW_TITLE  // 设置字体参数  LOGFONT logfont;  memset(&logfont, 0, sizeof(LOGFONT));  logfont.lfHeight = 40;  logfont.lfWidth = 0;      logfont.lfEscapement = 0;  logfont.lfOrientation = 0;  logfont.lfWeight = 30;  logfont.lfItalic = 0;  logfont.lfUnderline = 0;  logfont.lfStrikeOut = 0;  logfont.lfCharSet = DEFAULT_CHARSET;     logfont.lfOutPrecision= OUT_DEFAULT_PRECIS;     logfont.lfClipPrecision= OUT_DEFAULT_PRECIS;     logfont.lfQuality = DEFAULT_QUALITY;     logfont.lfPitchAndFamily= DEFAULT_PITCH;    // 创建字体并选入环境  HFONT hFont = CreateFontIndirect(&logfont);  HFONT hOldFont = (HFONT)SelectObject(hMemDc, hFont);  // 设置绘图环境  SetBkMode(hMemDc, TRANSPARENT);    SetTextColor(hMemDc, RGB(255, 255, 0));  // 绘制文字  TextOut(hMemDc,0,0,m_strFilePath,_tcslen(m_strFilePath));  // 恢复环境释放字体  SelectObject(hMemDc, hOldFont);
#endif  StretchBlt(    hdc,    m_rcWnd.left,     m_rcWnd.top,     m_rcWnd.right-m_rcWnd.left,     m_rcWnd.bottom-m_rcWnd.top,     hMemDc,    0,     0,     m_nFrameWidth,     m_nFrameHeight,     SRCCOPY);    // 恢复并释放环境      SelectObject(hMemDc,hOldBitmap);    DeleteObject(hBitmap);    DeleteDC(hMemDc);
}  // 获取播放状态
RTSP_PLAYSTATUS CRTSPPlayer::GetPlayStatus(void)
{  return m_nPlayStatus;
}  // 设置播放状态
void CRTSPPlayer::SetPlayStatus(RTSP_PLAYSTATUS playStatus)
{  m_nPlayStatus = playStatus;
}