287 lines
7.6 KiB
C++
287 lines
7.6 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)
|
||
|
||
|
||
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_IPAddress + 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();
|
||
int code;
|
||
if (code = m_aviStream.Open(m_aviFile, m_BitmapInfor_Full)) {
|
||
MessageBox("创建录像文件失败:"+m_aviFile + "\r\n错误代码: " + CBmpToAvi::GetErrMsg(code).c_str(), "提示");
|
||
m_aviFile.Empty();
|
||
} else {
|
||
pSysMenu->CheckMenuItem(IDM_SAVEAVI, MF_CHECKED);
|
||
}
|
||
}
|
||
|
||
|
||
CVideoDlg::CVideoDlg(CWnd* pParent, Server* IOCPServer, CONTEXT_OBJECT *ContextObject)
|
||
: DialogBase(CVideoDlg::IDD, pParent, IOCPServer, ContextObject, IDI_ICON_CAMERA)
|
||
{
|
||
m_nCount = 0;
|
||
m_aviFile.Empty();
|
||
m_BitmapInfor_Full = NULL;
|
||
m_pVideoCodec = NULL;
|
||
|
||
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_IPAddress, m_BitmapInfor_Full->bmiHeader.biWidth, m_BitmapInfor_Full->bmiHeader.biHeight);
|
||
|
||
SetWindowText(strString);
|
||
|
||
BYTE bToken = COMMAND_NEXT;
|
||
|
||
m_ContextObject->Send2Client(&bToken, sizeof(BYTE));
|
||
}
|
||
|
||
SetIcon(m_hIcon, TRUE);
|
||
SetIcon(m_hIcon, FALSE);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
void CVideoDlg::OnClose()
|
||
{
|
||
CancelIO();
|
||
// 等待数据处理完毕
|
||
if (IsProcessing()) {
|
||
ShowWindow(SW_HIDE);
|
||
return;
|
||
}
|
||
if (!m_aviFile.IsEmpty()) {
|
||
SaveAvi();
|
||
m_aviFile.Empty();
|
||
}
|
||
|
||
DialogBase::OnClose();
|
||
}
|
||
|
||
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)) {
|
||
Mprintf("======> InitCompressor failed \n");
|
||
delete m_pVideoCodec;
|
||
// 置NULL, 发送时判断是否为NULL来判断是否压缩
|
||
m_pVideoCodec = NULL;
|
||
// 通知服务端不启用压缩
|
||
BYTE bToken = COMMAND_WEBCAM_DISABLECOMPRESS;
|
||
m_ContextObject->Send2Client(&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_ContextObject->Send2Client(&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));
|
||
}
|
||
}
|