/*[]=====================================================================[]*/
/*[]     LeadTools Run Time Library - Version 14                         []*/
/*[]                                                                     []*/
/*[]                                                                     []*/
/*[]     Copyright (c) 1991-2004 LEAD Technologies, Inc.                 []*/
/*[]     All Rights Reserved.                                            []*/
/*[]=====================================================================[]*/


#include "ImgPF.h"

static L_BOOL DeskewDlg_OnInitDialog (HWND hWnd, HWND hWndFocus, LPARAM lParam);
static L_BOOL DeskewDlg_OnCommand    (HWND hWnd, L_INT nID, HWND hWndCtl, L_UINT uCodeNotify);
static L_BOOL DeskewDlg_OnHScroll    (HWND hWnd, HWND hWndCtl, L_UINT nCode, L_INT nPos);
static L_VOID DeskewDlg_OnTimer      (HWND hWnd, L_INT nID);
static L_VOID DeskewDlg_OnPaint      (HWND hWnd);
static L_VOID L_ApplyFilter          (HWND hWnd);

L_BOOL CALLBACK DeskewDlgProc (HWND hWnd, L_UINT uMessage, WPARAM wParam, LPARAM lParam)
{
   switch(uMessage)
   {
      HANDLE_MSG (hWnd, WM_COMMAND,     DeskewDlg_OnCommand);
      HANDLE_MSG (hWnd, WM_INITDIALOG,  DeskewDlg_OnInitDialog);
      HANDLE_MSG (hWnd, WM_HSCROLL,     DeskewDlg_OnHScroll);
      HANDLE_MSG (hWnd, WM_PAINT,       DeskewDlg_OnPaint);
      HANDLE_MSG (hWnd, WM_TIMER,       DeskewDlg_OnTimer);

   case WM_USER_UPDATEVIEW:
      StatusProc.ProgressValue = 100;
      L_ApplyFilter (hWnd);
      break;
   }
   return FALSE; 
}

static L_BOOL DeskewDlg_OnInitDialog (HWND hWnd, HWND hWndFocus, LPARAM lParam)
{
   LPDSKWDLGDATA  pDskDlgData;
   HWND         hWndParent;
   pCHILDDATA   pData;
   L_INT        nRangeMax[] = {4500 ,100};
   L_INT        nRangeMin[] = {0,      0};
   L_INT        nPosInit [] = {2000 , 20};
   L_INT        nSlider;

   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient,SendMessage);
   pData       = LOCKCHILDDATA (hWndParent);
   pDskDlgData = (LPDSKWDLGDATA)lParam;

   SetWindowLong(hWnd,GWL_USERDATA,(LONG)pDskDlgData);

   MoveDialogToMiddle(hWnd);

   StatusProc.bProgress       = TRUE;
   StatusProc.bInvalidate     = FALSE;
   StatusProc.bEnd            = FALSE;
   StatusProc.DlgWnd          = hWnd;
   StatusProc.ProgressValue   = 100;
   StatusProc.ProgWnd         = GetDlgItem(hWnd, IDC_DSK_PROG);

   Progress_SetRange(StatusProc.ProgWnd, 0, 100);
   Progress_SetPos(StatusProc.ProgWnd, 0);

   //MEMDISK...//
   MemorizeAndCopyBitmap (hWndParent, &pDskDlgData->LeadBitmap);

   pDskDlgData->nSlideNum        = 2;
   pDskDlgData->nSldBaseID       = IDC_DSK_RNG_SLD;
   pDskDlgData->nEdtBaseID       = IDC_DSK_RNG_EDT;
   pDskDlgData->uAngleResolution = 20;
   pDskDlgData->uAngleRange      = 2000;
   pDskDlgData->crColor          = RGB(0, 0, 0);
   pDskDlgData->uImageType       = 0;
   pDskDlgData->uNoProcess       = 0;
   pDskDlgData->uInterpolation   = 2;
   pDskDlgData->uNoFill          = 0;
   pDskDlgData->uThreshold       = 0;


   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_PRS_COM), "Process");
   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_PRS_COM), "No Process");
   ComboBox_SetCurSel(GetDlgItem(hWnd, IDC_DSK_PRS_COM), 0);

   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_FIL_COM), "Fill");
   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_FIL_COM), "No Fill");
   ComboBox_SetCurSel(GetDlgItem(hWnd, IDC_DSK_FIL_COM), 0);

   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_THS_COM), "No Threshold");
   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_THS_COM), "Threshold");
   ComboBox_SetCurSel(GetDlgItem(hWnd, IDC_DSK_THS_COM), 0);

   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_INP_COM), "Linear");
   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_INP_COM), "Resample");
   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_INP_COM), "Bicubic");
   ComboBox_SetCurSel(GetDlgItem(hWnd, IDC_DSK_INP_COM), 2);

   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_TYP_COM), "Document");
   ComboBox_AddString(GetDlgItem(hWnd, IDC_DSK_TYP_COM), "Document & Picture");
   ComboBox_SetCurSel(GetDlgItem(hWnd, IDC_DSK_TYP_COM), 0);


   for (nSlider = 0; nSlider < pDskDlgData->nSlideNum; nSlider++)
   {
      Trackbar_SetRange(GetDlgItem(hWnd, pDskDlgData->nSldBaseID + nSlider), TRUE, nRangeMin[nSlider], nRangeMax[nSlider]);
      SetNumericEditBox(GetDlgItem(hWnd, pDskDlgData->nEdtBaseID + nSlider), nRangeMin[nSlider], nRangeMax[nSlider], 1, 0, FALSE);
      SetDlgItemInt(hWnd, pDskDlgData->nEdtBaseID + nSlider, nPosInit[nSlider], TRUE);
   }

   ShowWindow(hWnd, SW_SHOW);
   UpdateWindow(hWnd);
   L_ApplyFilter (hWnd);
   pData->uTimerID = SetTimer(hWnd, 1, 100, NULL);
   return ( TRUE );
}

static L_VOID DeskewDlg_OnTimer(HWND hWnd, L_INT nID)
{
   Dialog_Timer(nID);
}

static L_BOOL DeskewDlg_OnCommand (HWND hWnd, L_INT nID, HWND hWndCtl, L_UINT uCodeNotify)
{
   LPDSKWDLGDATA pDskDlgData;
   HWND          hWndParent;
   L_INT         nValue;
   pCHILDDATA    pData;
   L_INT         nIndex;
   COLORREF      crColor;

   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient,SendMessage);
   pData       = LOCKCHILDDATA (hWndParent);
   pDskDlgData = (LPDSKWDLGDATA) GetWindowLong(hWnd, GWL_USERDATA);

   if (nID >= pDskDlgData->nEdtBaseID && nID < pDskDlgData->nEdtBaseID + pDskDlgData->nSlideNum)
   {
      if (uCodeNotify == EN_CHANGE)
      {
         nIndex = nID - pDskDlgData->nEdtBaseID;
         nValue = GetDlgItemInt(hWnd, nID, NULL, TRUE);
         Trackbar_SetPos(GetDlgItem(hWnd, nID + pDskDlgData->nSlideNum), TRUE, nValue);
      
         switch (nIndex)
         {
         case 0:
            if (pDskDlgData->uAngleRange == (L_UINT)nValue)
               return FALSE;
            pDskDlgData->uAngleRange = (L_UINT)nValue;
            nValue /= 10;
            nValue = min(nValue, 100);

            Trackbar_SetTheRange(GetDlgItem(hWnd, IDC_DSK_RSL_SLD),TRUE, 0, nValue);
            pDskDlgData->uAngleResolution = min(pDskDlgData->uAngleResolution, (L_UINT)nValue);
            SetDlgItemInt(hWnd, IDC_DSK_RSL_EDT, min(pDskDlgData->uAngleResolution, (L_UINT)nValue), TRUE);
            break;

         case 1:
            if (pDskDlgData->uAngleResolution == (L_UINT)nValue)
               return FALSE;
            pDskDlgData->uAngleResolution = (L_UINT)nValue;
            break;
         }
         L_ApplyFilter (hWnd);
      }
   }

   switch(nID)
   {
   case IDC_DSK_TYP_COM:
     if (uCodeNotify == CBN_SELCHANGE)
     {
         nValue = ComboBox_GetCurSel(hWndCtl);
         if (pDskDlgData->uImageType == (L_UINT)nValue || uCodeNotify != CBN_SELCHANGE)
            return FALSE;
         pDskDlgData->uImageType = nValue;
         InvalidateRect(hWndCtl, NULL, FALSE);
         L_ApplyFilter(hWnd);
     }
     break;

   case IDC_DSK_FIL_COM:
     if (uCodeNotify == CBN_SELCHANGE)
     {
         nValue = ComboBox_GetCurSel(hWndCtl);
         if (pDskDlgData->uNoFill == (L_UINT)nValue || uCodeNotify != CBN_SELCHANGE)
            return FALSE;
         pDskDlgData->uNoFill = nValue;
         Button_Enable(GetDlgItem(hWnd, IDC_DSK_CLR_BTN), !pDskDlgData->uNoFill);
         InvalidateRect(hWnd, NULL, FALSE);
         InvalidateRect(hWndCtl, NULL, FALSE);
         L_ApplyFilter(hWnd);
     }
      break;

   case IDC_DSK_INP_COM:
     if (uCodeNotify == CBN_SELCHANGE)
     {
         nValue = ComboBox_GetCurSel(hWndCtl);
         if (pDskDlgData->uInterpolation == (L_UINT)nValue || uCodeNotify != CBN_SELCHANGE)
            return FALSE;
         pDskDlgData->uInterpolation = nValue;
         InvalidateRect(hWndCtl, NULL, FALSE);
         L_ApplyFilter(hWnd);
     }
      break;

   case IDC_DSK_PRS_COM:
     if (uCodeNotify == CBN_SELCHANGE)
     {
         nValue = ComboBox_GetCurSel(hWndCtl);
         if (pDskDlgData->uNoProcess == (L_UINT)nValue || uCodeNotify != CBN_SELCHANGE)
            return FALSE;
         pDskDlgData->uNoProcess = nValue;
         InvalidateRect(hWndCtl, NULL, FALSE);
         L_ApplyFilter(hWnd);
     }
     break;

   case IDC_DSK_THS_COM:
     if (uCodeNotify == CBN_SELCHANGE)
     {
         nValue = ComboBox_GetCurSel(hWndCtl);
         if (pDskDlgData->uThreshold == (L_UINT)nValue || uCodeNotify != CBN_SELCHANGE)
            return FALSE;
         pDskDlgData->uThreshold = nValue;
         InvalidateRect(hWndCtl, NULL, FALSE);
         L_ApplyFilter(hWnd);
     }
     break;

   case IDC_DSK_CLR_BTN:
      crColor = pDskDlgData->crColor;
      pDskDlgData->crColor = ShowColorDialog(hWnd, crColor);
      if (pDskDlgData->crColor != crColor)
      {
         InvalidateRect(hWnd, NULL, FALSE);
         L_ApplyFilter(hWnd);
      }
      break;

   case IDCANCEL:
      StatusProc.bProgress     = FALSE;
      StatusProc.bKillProgress = TRUE;
      StatusProc.ProgressValue = 100;
      EndDialog(hWnd, FALSE);
      break;

   case IDOK:
      ShowStatusWindow(FALSE);
      StatusProc.bStillWork  = TRUE;
      StatusProc.bInvalidate = TRUE;
      StatusProc.bEnd        = TRUE;
      EndDialog(hWnd, TRUE);
      break;
   }
   return TRUE;
}

static L_BOOL DeskewDlg_OnHScroll (HWND hWnd, HWND hWndCtl, L_UINT nCode, L_INT nPos)
{
   LPDSKWDLGDATA pDskDlgData = (LPDSKWDLGDATA) GetWindowLong(hWnd,GWL_USERDATA);
   return SetDlgItemInt(hWnd, GetDlgCtrlID(hWndCtl) - pDskDlgData->nSlideNum, Trackbar_GetPos(hWndCtl) , TRUE);
}

static L_VOID DeskewDlg_OnPaint(HWND hWnd)
{
   LPDSKWDLGDATA pDskDlgData = (LPDSKWDLGDATA) GetWindowLong(hWnd, GWL_USERDATA);
   RECT          rcRect;
   HDC           hDC;
   HBRUSH        hBrush;
   PAINTSTRUCT   ps;

   hDC = BeginPaint(GetDlgItem(hWnd, IDC_DSK_CLR_TXT), &ps);
   GetClientRect(GetDlgItem(hWnd, IDC_DSK_CLR_TXT), &rcRect);
   hBrush = CreateSolidBrush(pDskDlgData->crColor);
   FillRect(hDC, &rcRect, pDskDlgData->uNoFill ? (HBRUSH)COLOR_WINDOW : hBrush);
   DeleteBrush(hBrush);
   EndPaint(GetDlgItem(hWnd, IDC_DSK_CLR_TXT), &ps);
}

static L_VOID L_ApplyFilter (HWND hWnd)
{
   pCHILDDATA     pData;
   L_INT          nRet = 0;
   HWND           hWndParent;
   LPDSKWDLGDATA  pDskDlgData;
   L_INT          nAngle;

   if ( 100 != StatusProc.ProgressValue )
   {
      StatusProc.bKillProgress = TRUE ;
      return ;
   }

   SetDlgItemText(hWnd, IDC_DSK_ANG_TXT, TEXT(""));
   pDskDlgData = (LPDSKWDLGDATA)GetWindowLong(hWnd,GWL_USERDATA);
   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient,SendMessage);
   pData       = LOCKCHILDDATA(hWndParent);

   FreeMemorizedBitmap (hWndParent); //MEMDISK//
   L_CopyBitmap(&pData->LeadBitmap, &pDskDlgData->LeadBitmap, sizeof(BITMAPHANDLE));

   nRet = L_DeskewBitmapExt(&pData->LeadBitmap,
                             &nAngle,
                             pDskDlgData->uAngleRange,
                             pDskDlgData->uAngleResolution,
                             pDskDlgData->crColor,
                             (pDskDlgData->uImageType << 16) | 
                             (pDskDlgData->uInterpolation << 12) | 
                             (pDskDlgData->uThreshold << 8) |
                             (pDskDlgData->uNoFill << 4) | 
                              pDskDlgData->uNoProcess);

   if(nRet == SUCCESS)
   {
      L_TCHAR szText[1024];
      wsprintf(szText, TEXT("Angle = %d"), nAngle);
      SetDlgItemText(hWnd, IDC_DSK_ANG_TXT, szText);
      nAngle = 0;
      InvalidateRect(hWndParent, NULL, FALSE);
      StatusProc.ProgressValue = 100;
      Progress_SetPos(StatusProc.ProgWnd , 0);
      if(StatusProc.bEnd)
      {
         StatusProc.bProgress = FALSE;
         StatusProc.bKillProgress = TRUE ;
      }
   }
}