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

#include "ImgPF.h"

static L_BOOL StatsDialog_OnCommand    (HWND hWnd, L_INT nID, HWND hWndCtl, L_UINT uCodeNotify);
static L_BOOL StatsDialog_OnHScroll    (HWND hWnd, HWND hWndCtl, L_UINT nCode, L_INT nPos);
static L_BOOL StatsDialog_OnInitDialog (HWND hWnd, HWND hWndFocus, LPARAM lParam);
static L_VOID StatsDialog_OnTimer      (HWND hWnd, L_INT nID);
static L_BOOL StatsDialog_OnPaint      (HWND hWnd);
static L_VOID L_ApplyFilter            (HWND hWnd);

L_BOOL CALLBACK StatsDlgProc (HWND hWnd, L_UINT uMessage, WPARAM wParam, LPARAM lParam)
{
   switch(uMessage)
   {
      HANDLE_MSG (hWnd, WM_COMMAND,    StatsDialog_OnCommand);
      HANDLE_MSG (hWnd, WM_INITDIALOG, StatsDialog_OnInitDialog);
      HANDLE_MSG (hWnd, WM_HSCROLL,    StatsDialog_OnHScroll);
      HANDLE_MSG (hWnd, WM_PAINT,      StatsDialog_OnPaint);
      HANDLE_MSG (hWnd, WM_TIMER,      StatsDialog_OnTimer);

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

static L_BOOL StatsDialog_OnInitDialog (HWND hWnd, HWND hWndFocus, LPARAM lParam)
{
   LPSTSDLGDATA pStsDlgData;
   HWND         hWndParent;
   pCHILDDATA   pData;
   L_INT        nRangeMax[] = {100, 255, 360};
   L_INT        nRangeMin[] = {  1,   0,   0};
   L_INT        nPosInit [] = {  1, 128,   0};
   L_INT        nSlider;

   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient,SendMessage);
   pData       = LOCKCHILDDATA (hWndParent);
   pStsDlgData = (LPSTSDLGDATA)lParam;
   MoveDialogToMiddle(hWnd);

   if (Is16Bit(&pData->LeadBitmap))
   {
      if (pData->LeadBitmap.BitsPerPixel == 12)
      {
         nRangeMax[1] = 0xFFF;
         nPosInit[1]  = 0x7FF;
      }
      else
      {
         nRangeMax[1] = 0xFFFF;
         nPosInit[1]  = 0x7FFF;
      }

   }
   SetWindowLong(hWnd,GWL_USERDATA,(LONG)pStsDlgData);
   //MEMDISK...//
   MemorizeAndCopyBitmap (hWndParent, &pStsDlgData->Bitmap);

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

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

   Trackbar_SetPos  (GetDlgItem(hWnd, IDC_STS_UDIM_SLD), TRUE, 1);

   pStsDlgData->nSldBaseID   = IDC_STS_UDIM_SLD;
   pStsDlgData->nEdtBaseID   = IDC_STS_UDIM_EDT;
   pStsDlgData->nSlideNum    = 3;
   pStsDlgData->uDim         = nPosInit[0];
   pStsDlgData->uThresh      = nPosInit[1];
   pStsDlgData->nAngle       = nPosInit[2];
   pStsDlgData->crEdgeColor  = RGB(255, 255, 255);
   pStsDlgData->crBkColor    = RGB(0,0,0);

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

   CreateAngleControl(GetDlgItem(hWnd, IDC_STS_ANGL_TXT), IDC_STS_ANGL_EDT, FROM_0_TO_360, 1, APPLY_ON_MOVE);
   ShowWindow(hWnd, SW_SHOW);
   pData->uTimerID = SetTimer(hWnd, 1, 100, NULL);
   L_ApplyFilter (hWnd);
   return ( TRUE );
}

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

static L_BOOL StatsDialog_OnPaint(HWND hWnd)
{
   HDC         hdc;
   PAINTSTRUCT ps;
   RECT        Rect;
   HBRUSH      hBrush;

   LPSTSDLGDATA pStsDlgData;
   pStsDlgData = (LPSTSDLGDATA) GetWindowLong(hWnd, GWL_USERDATA);

   GetClientRect(GetDlgItem(hWnd , IDC_STS_EDG_CLR_TXT), &Rect);
   hdc = BeginPaint(GetDlgItem(hWnd, IDC_STS_EDG_CLR_TXT),&ps);

   hBrush = CreateSolidBrush(pStsDlgData->crEdgeColor);
   FillRect(hdc, &Rect,hBrush);
   DeleteObject (hBrush);
   EndPaint(GetDlgItem(hWnd, IDC_STS_EDG_CLR_TXT), &ps);


   GetClientRect(GetDlgItem(hWnd , IDC_STS_BKG_CLR_TXT), &Rect);
   hdc = BeginPaint(GetDlgItem(hWnd, IDC_STS_BKG_CLR_TXT),&ps);

   hBrush = CreateSolidBrush(pStsDlgData->crBkColor);
   FillRect(hdc, &Rect,hBrush);
   DeleteObject (hBrush);
   EndPaint(GetDlgItem(hWnd, IDC_STS_BKG_CLR_TXT), &ps);

   return FALSE;
}

static L_BOOL StatsDialog_OnCommand (HWND hWnd, L_INT nID, HWND hWndCtl, L_UINT uCodeNotify)
{
   HWND           hWndParent;
   HWND           hCtl;
   pCHILDDATA     pData;
   L_INT          nValue;
   L_INT          nIndex;
   LPSTSDLGDATA   pStsDlgData;
   COLORREF       crColor;

   pStsDlgData = (LPSTSDLGDATA) GetWindowLong(hWnd, GWL_USERDATA);
   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient,SendMessage);
   pData       = LOCKCHILDDATA (hWndParent);

   switch (nID)
   {
   case IDC_STS_ANGL_EDT:
      nValue = GetDlgItemInt(hWnd, nID, NULL, TRUE);
      if (nValue == 360) nValue = 0;
      if ((uCodeNotify == EN_CHANGE) && nValue != pStsDlgData->nAngle)
      {
         pStsDlgData->nAngle = nValue;
         DrawAngleControl(GetDlgItem(hWnd, IDC_STS_ANGL_TXT), nValue);
         L_ApplyFilter(hWnd);
      }
      break;

   case IDC_STS_EDG_CLR_BUT:
      crColor = pStsDlgData->crEdgeColor;

      pStsDlgData->crEdgeColor = ShowColorDialog(hWnd, crColor);

      if (crColor != pStsDlgData->crEdgeColor)
      {
         InvalidateRect(hWnd, NULL, FALSE);
         L_ApplyFilter (hWnd);
      }
      break;

   case IDC_STS_BKG_CLR_BUT:
      crColor = pStsDlgData->crBkColor;

      pStsDlgData->crBkColor = ShowColorDialog(hWnd, crColor);

      if (crColor != pStsDlgData->crBkColor)
      {
         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;
   }

   if (nID >= pStsDlgData->nEdtBaseID && nID < pStsDlgData->nEdtBaseID + pStsDlgData->nSlideNum)
   {
      if(uCodeNotify == EN_CHANGE)
      {
         nValue = GetDlgItemInt(hWnd, nID, NULL,FALSE);
         hCtl = GetDlgItem(hWnd,nID + pStsDlgData->nSlideNum);
         nIndex = nID - pStsDlgData->nEdtBaseID;

         Trackbar_SetPos  (hCtl, TRUE, nValue);

         switch (nIndex)
         {
         case 0:
            if(pStsDlgData->uDim == (L_UINT)nValue)
               return TRUE;
            pStsDlgData->uDim = nValue;
            break;
         case 1:
            if(pStsDlgData->uThresh == (L_UINT)nValue)
               return TRUE;
            pStsDlgData->uThresh = nValue;
            break;
         default:
            return TRUE;
         }
         L_ApplyFilter (hWnd);
      }
   }

   return TRUE;
}

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

static L_VOID L_ApplyFilter (HWND hWnd)
{
   pCHILDDATA     pData;
   L_INT          nRet = 0;
   HWND           hWndParent;
   LPSTSDLGDATA   pStsDlgData;

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

   pStsDlgData = (LPSTSDLGDATA) GetWindowLong(hWnd,GWL_USERDATA);
   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient,SendMessage);
   pData       = LOCKCHILDDATA (hWndParent);

   FreeMemorizedBitmap (hWndParent); //MEMDISK//
   L_CopyBitmap(&pData->LeadBitmap, &pStsDlgData->Bitmap, sizeof(BITMAPHANDLE));
   if (pStsDlgData->nAngle > 180)
      pStsDlgData->nAngle -= 180;

   nRet = L_DirectionEdgeStatisticalBitmap(&pData->LeadBitmap,
                                            pStsDlgData->uDim,
                                            pStsDlgData->uThresh,
                                            abs(pStsDlgData->nAngle * 100),
                                            pStsDlgData->crEdgeColor,
                                            pStsDlgData->crBkColor);

   if(nRet == SUCCESS)
   {
      InvalidateRect(hWndParent, NULL, FALSE);
      StatusProc.ProgressValue = 100;
      Progress_SetPos(StatusProc.ProgWnd , 0);
      if(StatusProc.bEnd)
      {
         StatusProc.bProgress = FALSE;
         StatusProc.bKillProgress = TRUE ;
      }
   }
}