20141008 영상처리 박영주

#include <windows.h>
#include "vfw.h"

#define SETVFW_X 500
#define SETVFW_Y 600
#define BITMAP_MAXSIZE         (3*1024*768)+100

#pragma comment(lib, "vfw32.lib")

LRESULT CALLBACK  WndProc(HWND,UINT,WPARAM,LPARAM);
LRESULT CALLBACK  FramInfo(HWND, LPVIDEOHDR);
HINSTANCE      g_hInst;
HWND        HWndMain;
LPCTSTR        lpszClass=TEXT("WiseCat");
HWND        vfw;
BITMAPINFO      Bm;

int APIENTRY WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance
           ,LPSTR lpszCmdParam,int nCmdShow)
{
  HWND hWnd;
  MSG Message;
  WNDCLASS WndClass;
  g_hInst = hInstance;

  WndClass.cbClsExtra=0;
  WndClass.cbWndExtra=0;
  WndClass.hbrBackground=(HBRUSH)(COLOR_WINDOW+1);
  WndClass.hCursor=LoadCursor(NULL,IDC_ARROW);
  WndClass.hIcon=LoadIcon(NULL,IDI_APPLICATION);
  WndClass.hInstance=hInstance;
  WndClass.lpfnWndProc=WndProc;
  WndClass.lpszClassName=lpszClass;
  WndClass.lpszMenuName=NULL;
  WndClass.style=CS_HREDRAW | CS_VREDRAW;
  RegisterClass(&WndClass);

  hWnd=CreateWindow(  lpszClass
            ,lpszClass
            ,WS_OVERLAPPEDWINDOW
            ,CW_USEDEFAULT
            ,CW_USEDEFAULT
            ,CW_USEDEFAULT,CW_USEDEFAULT
            ,NULL
            ,(HMENU)NULL
            ,hInstance
            ,NULL);       
  ShowWindow(hWnd,nCmdShow);

  while(GetMessage(&Message,NULL,0,0))
  {
    TranslateMessage(&Message);
    DispatchMessage(&Message);
  }

  return (int)Message.wParam;
}

HDC hdc;
unsigned char * ucbitBuffer;
unsigned int uix;
unsigned int uiy;
unsigned int uiPad;
HBITMAP hbit;

LRESULT CALLBACK WndProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam)
{
  PAINTSTRUCT ps;
  HANDLE hFile;
  DWORD dwReadSize;

  switch(iMessage)
  {
    case WM_CREATE:
      HWndMain = hWnd;
   ucbitBuffer=(unsigned char *)malloc(BITMAP_MAXSIZE);
   hFile=CreateFile(TEXT("image.bmp"),GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
      vfw = capCreateCaptureWindow(  TEXT("CAM")
                      ,WS_CHILD | WS_VISIBLE
                      ,0
                      ,0
                      ,400
                      ,300
                      ,hWnd
                      ,NULL);

      capDriverConnect(vfw,0);
      capGetVideoFormat(vfw,&Bm,sizeof(Bm));
      Bm.bmiHeader.biWidth  = 320;
      Bm.bmiHeader.biHeight = 240;
      capSetVideoFormat(vfw,&Bm,sizeof(Bm));
      //capDlgVideoFormat(m_capwnd);
      capSetCallbackOnFrame(vfw, FramInfo);
      capPreviewRate(vfw, 1);
      capPreview(vfw, FALSE);

   hbit = CreateCompatibleBitmap(hdc, Bm.bmiHeader.biWidth, Bm.bmiHeader.biHeight);

   if(hFile == INVALID_HANDLE_VALUE) // 파일 생성 에러 처리
   {
   MessageBox(hWnd,TEXT("니 잘못으로 종료합니다."), TEXT("Error"), MB_OK);
   DestroyWindow(hWnd);
   }

 

  ReadFile(hFile,ucbitBuffer,BITMAP_MAXSIZE,&dwReadSize,NULL); // hFile의 데이터를 ucbitBuffer에 적재
  CloseHandle(hFile); // 파일 닫기
      return 0;

    case WM_PAINT:
      hdc = BeginPaint(hWnd,&ps);
      EndPaint(hWnd,&ps);
      return 0;

    case WM_DESTROY:
      PostQuitMessage(0);
      return 0;
  }
  return(DefWindowProc(hWnd,iMessage,wParam,lParam));
}

static BITMAPFILEHEADER * stpBFH;
static BITMAPINFOHEADER * stpBIH;
unsigned char * ucpPixel;

LRESULT CALLBACK FramInfo(HWND hWnd, LPVIDEOHDR lpData)
{
 HDC hMemDC;
  static int iCntX;
  static int iCntY;
  static int Jump;
  hMemDC = CreateCompatibleDC(hdc);
  hdc = GetDC(HWndMain);

   stpBFH = (BITMAPFILEHEADER*)ucbitBuffer;
 stpBIH = (BITMAPINFOHEADER*)(ucbitBuffer+sizeof(BITMAPFILEHEADER));

 
 uix = (unsigned int)stpBIH->biWidth;
 uiy = (unsigned int)stpBIH->biHeight;

  StretchDIBits(hdc  , 0
            , 0
            , Bm.bmiHeader.biWidth
            , Bm.bmiHeader.biHeight
            , 0
            , 0
            , Bm.bmiHeader.biWidth
            , Bm.bmiHeader.biHeight
            , lpData->lpData
            , &Bm
            , DIB_RGB_COLORS
            , SRCCOPY);

  uiPad = uix%4;
 ucpPixel= ucbitBuffer + stpBFH->bfOffBits ;

  Jump= 0;
  for(iCntY = 0; iCntY < Bm.bmiHeader.biHeight ; ++iCntY) 
  {       
    for(iCntX = 0; iCntX  < Bm.bmiHeader.biWidth  ; ++iCntX, Jump += 3,ucpPixel=ucpPixel+3 )
    {

      if(lpData->lpData[Jump + 2] > 100)
      {
        continue;
      }
      if(lpData->lpData[Jump + 1] > 100)
      {
        continue;
      }
      if(lpData->lpData[Jump + 0] > 50)
      {
        lpData->lpData[Jump]   = *(ucpPixel+0);  // Blue
        lpData->lpData[Jump + 2]  = *(ucpPixel+2);  // Red
        lpData->lpData[Jump + 1]  = *(ucpPixel+1);  // Green
      }

    }
 ucpPixel =ucpPixel+uiPad;
  }

  StretchDIBits(hdc  , Bm.bmiHeader.biWidth +20
            , 0
            , Bm.bmiHeader.biWidth
            , Bm.bmiHeader.biHeight
            , 0
            , 0
            , Bm.bmiHeader.biWidth
            , Bm.bmiHeader.biHeight
            , lpData->lpData
            , &Bm
            , DIB_RGB_COLORS
            , SRCCOPY);

  ReleaseDC(HWndMain, hdc);

  return 0;
}