www.pudn.com > sobelqc.zip > SobelQCDlg.cpp
// SobelQCDlg.cpp : implementation file
//
#include "stdafx.h"
#include "SobelQC.h"
#include "SobelQCDlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#include "qchelpers.h"
/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// Dialog Data
//{{AFX_DATA(CAboutDlg)
enum { IDD = IDD_ABOUTBOX };
//}}AFX_DATA
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CAboutDlg)
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
//}}AFX_VIRTUAL
// Implementation
protected:
//{{AFX_MSG(CAboutDlg)
//}}AFX_MSG
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
//{{AFX_DATA_INIT(CAboutDlg)
//}}AFX_DATA_INIT
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CAboutDlg)
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
//{{AFX_MSG_MAP(CAboutDlg)
// No message handlers
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CSobelQCDlg dialog
CSobelQCDlg::CSobelQCDlg(CWnd* pParent /*=NULL*/)
: CDialog(CSobelQCDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CSobelQCDlg)
// NOTE: the ClassWizard will add member initialization here
//}}AFX_DATA_INIT
// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
m_pbi = NULL;
}
void CSobelQCDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CSobelQCDlg)
DDX_Control(pDX, IDC_VIDEOPORTAL1, m_cVideo);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CSobelQCDlg, CDialog)
//{{AFX_MSG_MAP(CSobelQCDlg)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CSobelQCDlg message handlers
BOOL CSobelQCDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// Add "About..." menu item to system menu.
// IDM_ABOUTBOX must be in the system command range.
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
CString strAboutMenu;
strAboutMenu.LoadString(IDS_ABOUTBOX);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// Set the icon for this dialog. The framework does this automatically
// when the application's main window is not a dialog
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
// Prepare the control and registry settings
if (!m_cVideo.PrepareControl("PivotVC",
"HKEY_CURRENT_USER\\Software\\Generation5 Software\\SobelQC", 0 )) {
MessageBox("Couldn't initialize video camera. Exiting...");
EndDialog(-1);
}
AllocateDib(CSize(320,240));
m_cVideo.ConnectCamera2(); // Restore camera settings
m_cVideo.SetPreviewMaxWidth(320); // Set maximum width
m_cVideo.SetPreviewMaxHeight(240); // Set maximum height
m_cVideo.SetEnablePreview(FALSE); // Turn on the preview
m_cVideo.EnableUIElements(4, 0, 1); // Turn on UI.
m_cVideo.StartVideoHook(0); // Turn on the video hook.
m_cVideo.GetWindowRect(m_cVideoRect);
ScreenToClient(m_cVideoRect);
return TRUE; // return TRUE unless you set the focus to a control
}
void CSobelQCDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.
void CSobelQCDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
// Center icon in client rectangle
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
// The system calls this to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CSobelQCDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;
}
BEGIN_EVENTSINK_MAP(CSobelQCDlg, CDialog)
//{{AFX_EVENTSINK_MAP(CSobelQCDlg)
ON_EVENT(CSobelQCDlg, IDC_VIDEOPORTAL1, 1 /* PortalNotification */, OnPortalNotificationVideoportal1, VTS_I4 VTS_I4 VTS_I4 VTS_I4)
//}}AFX_EVENTSINK_MAP
END_EVENTSINK_MAP()
void CSobelQCDlg::OnPortalNotificationVideoportal1(long lMsg, long lParam1, long lParam2, long lParam3)
{
switch( lMsg ) {
case 10: {
LPBITMAPINFOHEADER lpbi = (LPBITMAPINFOHEADER) lParam1;
LPBYTE lpBytes = (LPBYTE) lParam2;
unsigned long lTimeStamp = (unsigned long)lParam3;
EdgeDetect(lpbi, lpBytes);
} break;
default:
break;
}
}
inline BYTE *getPixel(BYTE *s, int x, int y, int rs)
{
return s + x * 3 + y * rs;
}
void CSobelQCDlg::EdgeDetect(LPBITMAPINFOHEADER lpbi, LPBYTE lpBits)
{
int W = lpbi->biWidth;
int H = lpbi->biHeight;
int nPels = W * H;
int nPadBytes = DibRowPadding(lpbi);
int nRowStep = DibRowSize(m_pbi);
BYTE *dst = m_pbits;
BYTE *src, *prv;
short gc;
BYTE *a, *b, *c, *d, *e, *f, *g, *h, *z;
float sobscale = 1;
float offsetval = 0;
int i = 0, j = 0;
int final_i = 320 - 2;
int final_j = 240 - 2;
for (j = 0; j < final_j; j++) {
a = getPixel(lpBits, i, j, nRowStep);
b = getPixel(lpBits, i+1, j, nRowStep);
d = getPixel(lpBits, i, j+1, nRowStep);
f = getPixel(lpBits, i, j+2, nRowStep);
g = getPixel(lpBits, i+1, j+2, nRowStep);
z = getPixel(lpBits, i, j, nRowStep);
for (i=0; i 255) ? 0 : 255 - gc; // Truncate and inverse
gc = (gc > 255) ? 255 : gc;
gc = (gc < 0) ? 0 : gc;
dst = getPixel(m_pbits, i, j, nRowStep);
dst[0] = gc;
dst[1] = gc;
dst[2] = gc;
a = b; // Move the necessary pixels
b = c; // across into their new positions.
d = z; // This saves a lot of time since we
f = g; // don't need to access the pixel
g = h; // data again.
z = e; //
}
i = 0;
}
CDC* pDC = GetDC();
int nOldMode = SetStretchBltMode(pDC->GetSafeHdc(),COLORONCOLOR);
StretchDIBits( pDC->GetSafeHdc(),
m_cVideoRect.left,
m_cVideoRect.top,
m_cVideoRect.Width(),
m_cVideoRect.Height(),
0,
0,
lpbi->biWidth,
lpbi->biHeight,
m_pbits,
(BITMAPINFO*)m_pbi,
DIB_RGB_COLORS,
SRCCOPY);
SetStretchBltMode(pDC->GetSafeHdc(),nOldMode);
ReleaseDC(pDC);
}
void CSobelQCDlg::AllocateDib(CSize sz)
{
if (m_pbi) {
free(m_pbi); m_pbi = NULL;
}
if (sz.cx | sz.cy) {
m_pbi = (LPBITMAPINFOHEADER)malloc(DibSize(sz.cx, sz.cy, 24));
ASSERT(m_pbi);
m_pbi->biSize = sizeof (BITMAPINFOHEADER);
m_pbi->biWidth = sz.cx;
m_pbi->biHeight = sz.cy;
m_pbi->biPlanes = 1;
m_pbi->biBitCount = 24;
m_pbi->biCompression = 0;
m_pbi->biSizeImage = DibImageSize(sz.cx, sz.cy, 24);
m_pbi->biXPelsPerMeter = m_pbi->biYPelsPerMeter = 0;
m_pbi->biClrImportant = m_pbi->biClrUsed = 0;
m_pbits = (BYTE*)m_pbi + m_pbi->biSize;
}
}