// Camera.cpp: implementation of the CCamera class.
// 對Camera.h的實現
// 
// 編寫人：鄧福丹：2009年9月24日
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Camera.h"


LRESULT PASCAL FrameCallbackProc( HWND hWnd, VIDEOHDR* hdr ) 
{ 
	BITMAPINFO vfmt;
	CCamera* camera = (CCamera*)capGetUserData(hWnd);
	int sz;

	if (!hWnd) return FALSE;
	if (camera && !camera->IsRunning() ) return FALSE;
	
	sz = capGetVideoFormat( hWnd, &vfmt, sizeof(vfmt));
    
	if( hdr && hdr->lpData && sz != 0 && camera )
		camera->PutCompressedFrame( hdr, &vfmt );

	return (LRESULT)TRUE; 
}

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CCamera::CCamera()
{
	m_hCapWnd = 0;
	m_bIsRunning = false;
	m_hic = 0;
	m_dwFourcc = -1;
	m_bIsFlip = true;
}

CCamera::~CCamera()
{
	Uninitialize();
}

CvvImage& CCamera::GetImage()
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	return m_img;
}

void CCamera::GetImage( CvvImage &a_img )
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	a_img.CopyOf( m_img );
}

bool  CCamera::Initialize( int a_iWidth, int a_iHeight, int a_iFormat, HWND a_hParent, int a_iCamIndex/*=-1*/ )
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
    char szDeviceName[80];
    char szDeviceVersion[80];
    HWND hWndC = 0;
    
    if( (unsigned)a_iCamIndex < 10 )
        goto init_camera;

    for( a_iCamIndex = 0; a_iCamIndex < 10; a_iCamIndex++ ) 
    {
init_camera:
        if( capGetDriverDescription( a_iCamIndex, szDeviceName, 
            sizeof (szDeviceName), szDeviceVersion, 
            sizeof (szDeviceVersion))) 
        {
            hWndC = capCreateCaptureWindow ( "My Own Capture Window", 
                  WS_CHILD | WS_VISIBLE , 0, 0, 160, 120, a_hParent, 0);

            if( capDriverConnect (hWndC, a_iCamIndex))
            {
                break;
            }
            DestroyWindow( hWndC );
            hWndC = 0;
        }
    }
    
    if( hWndC )
    {
        m_hCapWnd = hWndC;
        
        memset( &m_caps, 0, sizeof(m_caps));
        capDriverGetCaps( hWndC, &m_caps, sizeof(m_caps));
        capSetUserData( hWndC, (long)this );
        capSetCallbackOnFrame( m_hCapWnd, FrameCallbackProc ); 
        ::MoveWindow( m_hCapWnd, 0, 0, 1, 1, TRUE );
        CAPTUREPARMS p;
        capCaptureGetSetup(hWndC,&p,sizeof(CAPTUREPARMS));
        p.dwRequestMicroSecPerFrame = 1;
        capCaptureSetSetup(hWndC,&p,sizeof(CAPTUREPARMS));
        capPreviewScale(hWndC,FALSE);
        capPreviewRate(hWndC,1);
    }
    return m_hCapWnd != 0;
}

void CCamera::Uninitialize()
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	Stop();
	capSetCallbackOnFrame( m_hCapWnd, NULL ); 
	capDriverDisconnect( m_hCapWnd );
	DestroyWindow( m_hCapWnd );

	if( m_hic )
	{
		ICDecompressEnd( m_hic );
		ICClose( m_hic );
		m_hic = 0;
	}
}

void CCamera::Start()
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	if( m_hCapWnd )
	{
		m_bIsRunning = true;
	}
}

void CCamera::Stop()
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	if( m_hCapWnd )
	{
		m_bIsRunning = false;
	}
}

bool CCamera::IsRunning()
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	return m_bIsRunning;
}

bool CCamera::Flip()
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	m_bIsFlip =!m_bIsFlip;
	return m_bIsFlip;
}

void CCamera::PutCompressedFrame( VIDEOHDR* a_hdr, BITMAPINFO* a_vfmt )
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
	long code = ICERR_OK;
	char* frame_data = (char*)a_hdr->lpData;//dib中的bits的數組。 	
        
	if( a_vfmt->bmiHeader.biCompression != BI_RGB ||
		a_vfmt->bmiHeader.biBitCount != 24 )//判斷是否為真彩色圖
	{
		BITMAPINFOHEADER& vfmt0 = a_vfmt->bmiHeader;
		BITMAPINFOHEADER vfmt1;
		code = ICERR_ERROR;

		memset( &vfmt1, 0, sizeof(vfmt1));
		vfmt1.biSize = sizeof(vfmt1);
		vfmt1.biWidth = vfmt0.biWidth;
		vfmt1.biHeight = vfmt0.biHeight;
		vfmt1.biBitCount = 24;
		vfmt1.biCompression = BI_RGB;
		vfmt1.biPlanes = 1;

		if( m_hic == 0 || m_dwFourcc != vfmt0.biCompression || m_img.GetImage() == 0 ||
			vfmt0.biWidth != m_img.GetImage()->width ||
			vfmt0.biHeight != m_img.GetImage()->height )
		{
			if( m_hic )
			{
				ICDecompressEnd( m_hic );
				ICClose( m_hic );
			}
			m_hic = ICOpen( MAKEFOURCC('V','I','D','C'),
				vfmt0.biCompression, ICMODE_DECOMPRESS );

			if( m_hic && ICDecompressBegin( m_hic, &vfmt0, &vfmt1 ) == ICERR_OK )
			{
				m_img.Create( vfmt0.biWidth, vfmt0.biHeight, 24 );
				m_img.GetImage()->origin = IPL_ORIGIN_BL;

				code = ICDecompress( m_hic, 0, &vfmt0, a_hdr->lpData,
					&vfmt1, m_img.GetImage()->imageData );
				if( m_bIsFlip )
					cvFlip( m_img.GetImage(), m_img.GetImage(), 0 );
			}
		}
	}
	else//是彩色圖
	{
		m_img.Create( a_vfmt->bmiHeader.biWidth, a_vfmt->bmiHeader.biHeight, 24 );//建立圖片
		memcpy( m_img.GetImage()->imageData, a_hdr->lpData, m_img.GetImage()->imageSize );
		if( m_bIsFlip )
			cvFlip( m_img.GetImage(), m_img.GetImage(), 0 );
	}
	OnFrame();//當前楨
}

void CCamera::OnFrame()
{
	// 鎖定臨界區
	CSingleLock lock(&m_cs);
	lock.Lock();
	// 
    HWND parent = GetParent( m_hCapWnd );
    RECT r = { 0, 0, 0, 0 };

    IplImage* img = m_img.GetImage();

    if( img )
    {
        r.right = img->width;
        r.bottom = img->height;
    }
    
    InvalidateRect( parent, &r, 0 );
    UpdateWindow( parent );
}