#include "ImgPF.h"

static L_BOOL BalanceDlg_OnInitDialog (HWND hWnd, HWND hWndFocus, LPARAM lParam);
static L_VOID BalanceDlg_OnCommand    (HWND hWnd, L_INT nID, HWND hWndCtl, UINT uCodeNotify);
static L_VOID BalanceDlg_OnHScroll    (HWND hWnd, HWND hWndCtl, L_UINT uCode, L_INT nPos);
static L_VOID L_ApplyFilter          (HWND hWnd);
static L_VOID BalanceDlg_OnTimer      (HWND hWnd, L_INT nID);

L_BOOL CALLBACK BalanceDlgProc(HWND hWnd, UINT uMessage, WPARAM wParam, LPARAM lParam)
{
   switch(uMessage)
   {
      HANDLE_MSG(hWnd, WM_INITDIALOG, BalanceDlg_OnInitDialog);
      HANDLE_MSG(hWnd, WM_COMMAND,    BalanceDlg_OnCommand);
      HANDLE_MSG(hWnd, WM_HSCROLL,    BalanceDlg_OnHScroll);
      HANDLE_MSG(hWnd, WM_TIMER,      BalanceDlg_OnTimer);

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

static L_BOOL BalanceDlg_OnInitDialog(HWND hWnd, HWND hWndFocus, LPARAM lParam)
{
   HWND         hListBox;
   LPBALDLGDATA pBalDlgData;
   L_INT        nCounter;
   pCHILDDATA   pData;
   HWND         hWndParent;
   L_INT        nSlider;
   L_INT        nRangeMax[] = { 100,  100,  100};
   L_INT        nRangeMin[] = {-100, -100, -100};
   L_INT        nPosInit [] = { 0,    0,    0};

   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient, SendMessage);
   pData       = LOCKCHILDDATA(hWndParent);
   pBalDlgData = (LPBALDLGDATA)lParam;
   MoveDialogToMiddle(hWnd);

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

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

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

   pBalDlgData->nSldBaseID = IDC_BAL_RED_SLD;
   pBalDlgData->nEdtBaseID = IDC_BAL_RED_EDT;
   pBalDlgData->nSlideNum  = 3;
   pBalDlgData->bLum       = FALSE;

   hListBox   = (HWND) GetDlgItem(hWnd, IDC_BAL_OTPT_COM);

   ComboBox_InsertString (hListBox, 0, TEXT("Shadows"));
   ComboBox_InsertString (hListBox, 1, TEXT("MidTones"));
   ComboBox_InsertString (hListBox, 2, TEXT("HighLights"));
   ComboBox_SetCurSel    (hListBox, 0);

   for(nCounter = 0; nCounter <= 2; nCounter++)
   {
      pBalDlgData->pBalance[nCounter].uStructSize = sizeof(BALANCEDATA);
      pBalDlgData->pBalance[nCounter].nBlue       =
      pBalDlgData->pBalance[nCounter].nGreen      =
      pBalDlgData->pBalance[nCounter].nRed        = 0;
   }

   SetWindowLong(hWnd, GWL_USERDATA, (long)pBalDlgData);  

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

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

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

static L_VOID BalanceDlg_OnCommand(HWND hWnd, L_INT nID, HWND hWndCtl, L_UINT uCodeNotify)
{
   LPBALDLGDATA pBalDlgData;
   L_UINT       uCurrentColor;
   L_INT        nValue;
   pCHILDDATA   pData;
   HWND         hWndParent;
   L_INT        nRangeMax[] = { 100,   100,  100};
   L_INT        nRangeMin[] = { -100, -100, -100};
   L_INT        nIndex;

   pBalDlgData = (LPBALDLGDATA) GetWindowLong(hWnd, GWL_USERDATA);
   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient, SendMessage);
   pData       = LOCKCHILDDATA (hWndParent);

   switch (nID)
   {
   case IDC_BAL_RST_BTN:
      for(nIndex = 0; nIndex <= 2; nIndex++)
      {
         pBalDlgData->pBalance[nIndex].nBlue  =
         pBalDlgData->pBalance[nIndex].nGreen =
         pBalDlgData->pBalance[nIndex].nRed   = 0;
      }
      for (nIndex = pBalDlgData->nEdtBaseID; nIndex < pBalDlgData->nEdtBaseID + pBalDlgData->nSlideNum; nIndex++)
         SetDlgItemInt(hWnd, nIndex, 0, TRUE);
      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;

   case IDC_BAL_PRS_CHK:
      pBalDlgData->bLum = Button_GetCheck(hWndCtl);
      L_ApplyFilter(hWnd);
      break;
      
   case IDC_BAL_OTPT_COM:
      if (uCodeNotify == CBN_SELCHANGE)
      {
         uCurrentColor = ComboBox_GetCurSel(hWndCtl);

         SetDlgItemInt(hWnd, pBalDlgData->nEdtBaseID + 0, pBalDlgData->pBalance[uCurrentColor].nRed, TRUE);
         SetDlgItemInt(hWnd, pBalDlgData->nEdtBaseID + 1, pBalDlgData->pBalance[uCurrentColor].nGreen, TRUE);
         SetDlgItemInt(hWnd, pBalDlgData->nEdtBaseID + 2, pBalDlgData->pBalance[uCurrentColor].nBlue, TRUE);
         InvalidateRect(hWndCtl, NULL, FALSE);
      }
      break;
   }

   if (nID >= pBalDlgData->nEdtBaseID && nID < pBalDlgData->nEdtBaseID + pBalDlgData->nSlideNum)
   {
      if ((uCodeNotify == EN_CHANGE))
      {
         uCurrentColor = ComboBox_GetCurSel(GetDlgItem(hWnd, IDC_BAL_OTPT_COM));
         nValue        = GetDlgItemInt(hWnd, nID, NULL, TRUE);
         nIndex        = nID - pBalDlgData->nEdtBaseID;

         Trackbar_SetPos(GetDlgItem(hWnd, pBalDlgData->nSldBaseID + nIndex), TRUE, nValue);

         switch(nIndex)
         {
         case 0:
            if (pBalDlgData->pBalance[uCurrentColor].nRed == nValue)
               return;
            pBalDlgData->pBalance[uCurrentColor].nRed = nValue;
            break;
         case 1:
            if (pBalDlgData->pBalance[uCurrentColor].nGreen == nValue)
               return;
            pBalDlgData->pBalance[uCurrentColor].nGreen = nValue;
            break;
         case 2:
            if (pBalDlgData->pBalance[uCurrentColor].nBlue == nValue)
               return;
            pBalDlgData->pBalance[uCurrentColor].nBlue = nValue;
            break;
         }
         L_ApplyFilter(hWnd);
      }
   }
}

static L_VOID BalanceDlg_OnHScroll (HWND hWnd, HWND hWndCtl, L_UINT uCode, L_INT nPos)
{
   LPBALDLGDATA pBalDlgData = (LPBALDLGDATA) GetWindowLong(hWnd, GWL_USERDATA);
   SetDlgItemInt(hWnd, GetDlgCtrlID(hWndCtl) - pBalDlgData->nSlideNum, Trackbar_GetPos(hWndCtl), TRUE);
}


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

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

   pBalDlgData = (LPBALDLGDATA) GetWindowLong(hWnd, GWL_USERDATA);
   hWndParent  = FORWARD_WM_MDIGETACTIVE(hWndClient, SendMessage);
   pData       = LOCKCHILDDATA (hWndParent);

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

   nRet = L_ColorIntensityBalance(&pData->LeadBitmap, 
                                  &pBalDlgData->pBalance[0], 
                                  &pBalDlgData->pBalance[1], 
                                  &pBalDlgData->pBalance[2], 
                                  pBalDlgData->bLum);

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