Code/Sandbox/Plugins/MFCToolsPlugin/Dialogs/CustomColorDialog.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "stdafx.h"
   4 #include "CustomColorDialog.h"
   5
   6 class CScreenWnd;
   7
   8 static std::vector<std::unique_ptr<CScreenWnd>> g_screenWnds;
   9 static BOOL CALLBACK CreateScreenWndCB(HMONITOR monitor, HDC dc, LPRECT rect, LPARAM data);
  10
  11 /////////////////////////////////////////////////////////////////////////////
  12 // CScreenWnd window
  13
  14 class CScreenWnd : public CWnd
  15 {
  16         // Construction
  17 public:
  18         CScreenWnd();
  19
  20         // Overrides
  21 public:
  22         virtual BOOL Create(CMFCColorDialog* pColorDlg, CRect rect);
  23
  24         // Implementation
  25 public:
  26         virtual ~CScreenWnd();
  27
  28         // Generated message map functions
  29 protected:
  30         //{{AFX_MSG(CScreenWnd)
  31         afx_msg void OnMouseMove(UINT nFlags, CPoint point);
  32         afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
  33         afx_msg BOOL OnEraseBkgnd(CDC* pDC);
  34         //}}AFX_MSG
  35         DECLARE_MESSAGE_MAP()
  36
  37         CMFCColorDialog * m_pColorDlg;
  38 };
  39
  40 /////////////////////////////////////////////////////////////////////////////
  41 // CCustomColorDialog
  42
  43 IMPLEMENT_DYNAMIC(CCustomColorDialog, CMFCColorDialog)
  44
  45 //////////////////////////////////////////////////////////////////////////
  46 CCustomColorDialog::CCustomColorDialog(COLORREF clrInit, DWORD dwFlags, CWnd* pParentWnd) :
  47         CMFCColorDialog(clrInit, dwFlags, pParentWnd), m_callback(0), m_colorPrev(0)
  48 {
  49 }
  50
  51 //////////////////////////////////////////////////////////////////////////
  52 BEGIN_MESSAGE_MAP(CCustomColorDialog, CMFCColorDialog)
  53 ON_WM_MOUSEMOVE()
  54 ON_MESSAGE(WM_KICKIDLE, OnKickIdle)
  55 ON_BN_CLICKED(IDC_AFXBARRES_COLOR_SELECT, OnColorSelect)
  56 END_MESSAGE_MAP()
  57
  58 //////////////////////////////////////////////////////////////////////////
  59 BOOL CCustomColorDialog::OnInitDialog()
  60 {
  61         CMFCColorDialog::OnInitDialog();
  62
  63         m_colorPrev = m_CurrentColor;
  64         m_pPropSheet->SetActivePage(m_pColourSheetTwo);
  65
  66         return TRUE;  // return TRUE unless you set the focus to a control
  67                       // EXCEPTION: OCX Property Pages should return FALSE
  68 }
  69
  70 //////////////////////////////////////////////////////////////////////////
  71 void CCustomColorDialog::OnMouseMove(UINT nFlags, CPoint point)
  72 {
  73         CMFCColorDialog::OnMouseMove(nFlags, point);
  74
  75         if (m_colorPrev != m_NewColor)
  76                 OnColorChange(m_NewColor);
  77 }
  78
  79 //////////////////////////////////////////////////////////////////////////
  80 LRESULT CCustomColorDialog::OnKickIdle(WPARAM, LPARAM lCount)
  81 {
  82         if (m_colorPrev != m_NewColor)
  83                 OnColorChange(m_NewColor);
  84
  85         return FALSE;
  86 }
  87
  88 //////////////////////////////////////////////////////////////////////////
  89 void CCustomColorDialog::OnColorChange(COLORREF col)
  90 {
  91         m_colorPrev = col;
  92         if (m_callback)
  93                 m_callback(col);
  94 }
  95
  96 //////////////////////////////////////////////////////////////////////////
  97 void CCustomColorDialog::OnColorSelect()
  98 {
  99         if (m_bPickerMode)
 100         {
 101                 return;
 102         }
 103
 104         CWinThread* pCurrThread = ::AfxGetThread();
 105         if (pCurrThread == NULL)
 106         {
 107                 ASSERT(FALSE);
 108                 return;
 109         }
 110
 111         MSG msg;
 112         m_bPickerMode = TRUE;
 113         ;
 114         ::SetCursor(m_hcurPicker);
 115
 116         LONG_PTR thisPtr = reinterpret_cast<LONG_PTR>(this);
 117         ::EnumDisplayMonitors(NULL, NULL, CreateScreenWndCB, thisPtr);
 118
 119         SetForegroundWindow();
 120         BringWindowToTop();
 121
 122         SetCapture();
 123
 124         COLORREF colorSaved = m_NewColor;
 125
 126         while (m_bPickerMode)
 127         {
 128                 while (m_bPickerMode && ::PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
 129                 {
 130                         if (msg.message == WM_KEYDOWN)
 131                         {
 132                                 switch (msg.wParam)
 133                                 {
 134                                 case VK_RETURN:
 135                                         m_bPickerMode = FALSE;
 136                                         break;
 137
 138                                 case VK_ESCAPE:
 139                                         SetNewColor(colorSaved);
 140                                         m_bPickerMode = FALSE;
 141                                         break;
 142                                 }
 143                         }
 144                         else if (msg.message == WM_RBUTTONDOWN || msg.message == WM_MBUTTONDOWN)
 145                         {
 146                                 m_bPickerMode = FALSE;
 147                         }
 148                         else
 149                         {
 150                                 if (!pCurrThread->PreTranslateMessage(&msg))
 151                                 {
 152                                         ::TranslateMessage(&msg);
 153                                         ::DispatchMessage(&msg);
 154                                 }
 155
 156                                 pCurrThread->OnIdle(0);
 157                         }
 158                 }
 159
 160                 WaitMessage();
 161         }
 162
 163         ReleaseCapture();
 164         std::for_each(g_screenWnds.begin(), g_screenWnds.end(), [](const std::unique_ptr<CScreenWnd>& pScreenWnd)
 165         {
 166                 pScreenWnd->DestroyWindow();
 167         });
 168         g_screenWnds.clear();
 169
 170         m_bPickerMode = FALSE;
 171 }
 172
 173 //////////////////////////////////////////////////////////////////////////
 174 BOOL CALLBACK CreateScreenWndCB(HMONITOR monitor, HDC dc, LPRECT rect, LPARAM data)
 175 {
 176         MONITORINFOEX monitorInfo;
 177         monitorInfo.cbSize = sizeof(MONITORINFOEX);
 178         ::GetMonitorInfo(monitor, &monitorInfo);
 179
 180         CCustomColorDialog* self = reinterpret_cast<CCustomColorDialog*>(data);
 181
 182         std::unique_ptr<CScreenWnd> pScreenWnd(new CScreenWnd());
 183         pScreenWnd->Create(self, monitorInfo.rcMonitor);
 184
 185         g_screenWnds.push_back(std::move(pScreenWnd));
 186
 187         return TRUE;
 188 }
 189
 190 /////////////////////////////////////////////////////////////////////////////
 191 // CScreenWnd
 192
 193 CScreenWnd::CScreenWnd()
 194 {
 195 }
 196
 197 CScreenWnd::~CScreenWnd()
 198 {
 199 }
 200
 201 BEGIN_MESSAGE_MAP(CScreenWnd, CWnd)
 202 //{{AFX_MSG_MAP(CScreenWnd)
 203 ON_WM_MOUSEMOVE()
 204 ON_WM_LBUTTONDOWN()
 205 ON_WM_SETCURSOR()
 206 ON_WM_ERASEBKGND()
 207 //}}AFX_MSG_MAP
 208 END_MESSAGE_MAP()
 209
 210 /////////////////////////////////////////////////////////////////////////////
 211 // CScreenWnd message handlers
 212
 213 BOOL CScreenWnd::Create(CMFCColorDialog* pColorDlg, CRect rect)
 214 {
 215         m_pColorDlg = pColorDlg;
 216
 217         string strClassName = ::AfxRegisterWndClass(CS_SAVEBITS, AfxGetApp()->LoadCursor(IDC_AFXBARRES_COLOR), (HBRUSH)(COLOR_BTNFACE + 1), NULL);
 218
 219         return CWnd::CreateEx(WS_EX_TOOLWINDOW | WS_EX_TRANSPARENT, strClassName, _T(""), WS_VISIBLE | WS_POPUP, rect, NULL, 0);
 220 }
 221
 222 void CScreenWnd::OnMouseMove(UINT nFlags, CPoint point)
 223 {
 224         MapWindowPoints(m_pColorDlg, &point, 1);
 225         m_pColorDlg->SendMessage(WM_MOUSEMOVE, nFlags, MAKELPARAM(point.x, point.y));
 226
 227         CWnd::OnMouseMove(nFlags, point);
 228 }
 229
 230 void CScreenWnd::OnLButtonDown(UINT nFlags, CPoint point)
 231 {
 232         MapWindowPoints(m_pColorDlg, &point, 1);
 233         m_pColorDlg->SendMessage(WM_LBUTTONDOWN, nFlags, MAKELPARAM(point.x, point.y));
 234 }
 235
 236 BOOL CScreenWnd::OnEraseBkgnd(CDC* /*pDC*/)
 237 {
 238         return TRUE;
 239 }
 240
 241 BOOL CCustomColorDialog::PreTranslateMessage(MSG* pMsg)
 242 {
 243 #ifdef _UNICODE
 244         #define AFX_TCF_TEXT CF_UNICODETEXT
 245 #else
 246         #define AFX_TCF_TEXT CF_TEXT
 247 #endif
 248
 249         if (m_colorPrev != m_NewColor)
 250         {
 251                 OnColorChange(m_NewColor);
 252         }
 253
 254         if (pMsg->message == WM_KEYDOWN)
 255         {
 256                 UINT nChar = (UINT) pMsg->wParam;
 257                 BOOL bIsCtrl = (::GetAsyncKeyState(VK_CONTROL) & 0x8000);
 258
 259                 if (bIsCtrl && (nChar == _T('C') || nChar == VK_INSERT))
 260                 {
 261                         if (OpenClipboard())
 262                         {
 263                                 EmptyClipboard();
 264
 265                                 string strText;
 266                                 strText.Format(_T("RGB(%d, %d, %d)"), GetRValue(m_NewColor), GetGValue(m_NewColor), GetBValue(m_NewColor));
 267
 268                                 HGLOBAL hClipbuffer = ::GlobalAlloc(GMEM_DDESHARE, (strText.GetLength() + 1) * sizeof(TCHAR));
 269                                 LPTSTR lpszBuffer = (LPTSTR) GlobalLock(hClipbuffer);
 270
 271                                 lstrcpy(lpszBuffer, (LPCTSTR) strText);
 272
 273                                 ::GlobalUnlock(hClipbuffer);
 274                                 ::SetClipboardData(AFX_TCF_TEXT, hClipbuffer);
 275
 276                                 CloseClipboard();
 277                         }
 278                 }
 279         }
 280
 281         return CMFCColorDialog::PreTranslateMessage(pMsg);
 282 }