src/thirdparty/mfc/afxctrlcontainer.cpp

   1 // This is a part of the Microsoft Foundation Classes C++ library.
   2 // Copyright (C) Microsoft Corporation
   3 // All rights reserved.
   4 //
   5 // This source code is only intended as a supplement to the
   6 // Microsoft Foundation Classes Reference and related
   7 // electronic documentation provided with the library.
   8 // See these sources for detailed information regarding the
   9 // Microsoft Foundation Classes product.
  10 //
  11 // CWnd support for MFC Control containment (Feature Pack controls)
  12 //
  13
  14 #include "stdafx.h"
  15 #undef SubclassWindow
  16 #include "afxctrlcontainer.h"
  17 #if 0
  18 #include "afxtagmanager.h"
  19 #include "afxbutton.h"
  20 #include "afxcolorbutton.h"
  21 #include "afxeditbrowsectrl.h"
  22 #include "afxfontcombobox.h"
  23 #include "afxlinkctrl.h"
  24 #include "afxmaskededit.h"
  25 #include "afxmenubutton.h"
  26 #include "afxpropertygridctrl.h"
  27 #include "afxshelllistctrl.h"
  28 #include "afxshelltreectrl.h"
  29 #include "afxvslistbox.h"
  30 #endif
  31 ////////////////////////////////////////////////////////////////////////////
  32
  33 static void DoRegisterWindowClass(LPCTSTR lpszClassName, LPCTSTR lpszBaseClassName)
  34 {
  35         ASSERT(lpszClassName != NULL);
  36         ASSERT(lpszBaseClassName != NULL);
  37
  38         WNDCLASS wnd = {0};
  39
  40         HINSTANCE hInst = AfxGetInstanceHandle();
  41         if (!AfxCtxGetClassInfo(hInst, lpszBaseClassName, &wnd))
  42         {
  43                 wnd.style = CS_DBLCLKS;
  44                 wnd.hInstance = hInst;
  45                 wnd.lpfnWndProc = ::DefWindowProc;
  46         }
  47
  48         wnd.lpszClassName = lpszClassName;
  49         AfxRegisterClass(&wnd);
  50 }
  51
  52 void AfxRegisterMFCCtrlClasses()
  53 {
  54         DoRegisterWindowClass(_T("MFCButton"), WC_BUTTON);
  55         DoRegisterWindowClass(_T("MFCColorButton"), WC_BUTTON);
  56         DoRegisterWindowClass(_T("MFCEditBrowse"), WC_EDIT);
  57         DoRegisterWindowClass(_T("MFCFontComboBox"), WC_COMBOBOX);
  58         DoRegisterWindowClass(_T("MFCLink"), WC_BUTTON);
  59         DoRegisterWindowClass(_T("MFCMaskedEdit"), WC_EDIT);
  60         DoRegisterWindowClass(_T("MFCMenuButton"), WC_BUTTON);
  61         DoRegisterWindowClass(_T("MFCPropertyGrid"), WC_STATIC);
  62         DoRegisterWindowClass(_T("MFCShellList"), WC_LISTVIEW);
  63         DoRegisterWindowClass(_T("MFCShellTree"), WC_TREEVIEW);
  64         DoRegisterWindowClass(_T("MFCVSListBox"), WC_STATIC);
  65 }
  66
  67 ////////////////////////////////////////////////////////////////////////////
  68 // CMFCControlContainer
  69
  70 CMFCControlContainer::CMFCControlContainer(CWnd* pWnd) : m_pWnd(pWnd)
  71 {
  72 }
  73
  74 CMFCControlContainer::~CMFCControlContainer()
  75 {
  76         FreeSubclassedControls();
  77         ClearControlData();
  78 }
  79
  80 BOOL CMFCControlContainer::SubclassDlgControls()
  81 {
  82         if (m_pWnd->GetSafeHwnd() != NULL)
  83         {
  84                 // Subclass Feature Pack controls:
  85                 CWnd* pWndChild = m_pWnd->GetWindow(GW_CHILD);
  86                 while (pWndChild != NULL)
  87                 {
  88                         ASSERT_VALID(pWndChild);
  89
  90                         TCHAR lpszClassName [MAX_CLASS_NAME + 1];
  91
  92                         ::GetClassName(pWndChild->GetSafeHwnd(), lpszClassName, MAX_CLASS_NAME);
  93                         CWnd* pWndSubclassedCtrl = CreateDlgControl(lpszClassName);
  94
  95                         if (pWndSubclassedCtrl != NULL)
  96                         {
  97                                 m_arSubclassedCtrls.Add((CObject*)pWndSubclassedCtrl);
  98                                 pWndSubclassedCtrl->SubclassWindow(pWndChild->GetSafeHwnd());
  99                         }
 100
 101                         pWndChild = pWndChild->GetNextWindow();
 102                 }
 103
 104                 return TRUE;
 105         }
 106
 107         return FALSE;
 108 }
 109
 110 void CMFCControlContainer::FreeSubclassedControls()
 111 {
 112         // Free subclassed controls:
 113         for (int i = 0; i < m_arSubclassedCtrls.GetCount(); i++)
 114         {
 115                 if (m_arSubclassedCtrls [i] != NULL)
 116                 {
 117                         delete m_arSubclassedCtrls [i];
 118                 }
 119         }
 120         m_arSubclassedCtrls.RemoveAll();
 121 }
 122
 123 CWnd* CMFCControlContainer::CreateDlgControl(LPCTSTR lpszClassName)
 124 {
 125         ASSERT(m_pWnd->GetSafeHwnd() != NULL);
 126
 127         if (lpszClassName != NULL)
 128         {
 129                 CString strClass = lpszClassName;
 130                 CWnd* pWndSubclassedCtrl = NULL;
 131 #if 0
 132                 if (strClass == _T("MFCButton"))
 133                 {
 134                         pWndSubclassedCtrl = new CMFCButton;
 135                 }
 136                 else if (strClass == _T("MFCColorButton"))
 137                 {
 138                         pWndSubclassedCtrl = new CMFCColorButton;
 139                 }
 140                 else if (strClass == _T("MFCEditBrowse"))
 141                 {
 142                         pWndSubclassedCtrl = new CMFCEditBrowseCtrl;
 143                 }
 144                 else if (strClass == _T("MFCFontComboBox"))
 145                 {
 146                         pWndSubclassedCtrl = new CMFCFontComboBox;
 147                 }
 148                 else if (strClass == _T("MFCLink"))
 149                 {
 150                         pWndSubclassedCtrl = new CMFCLinkCtrl;
 151                 }
 152                 else if (strClass == _T("MFCMaskedEdit"))
 153                 {
 154                         pWndSubclassedCtrl = new CMFCMaskedEdit;
 155                 }
 156                 else if (strClass == _T("MFCMenuButton"))
 157                 {
 158                         pWndSubclassedCtrl = new CMFCMenuButton;
 159                 }
 160                 else if (strClass == _T("MFCPropertyGrid"))
 161                 {
 162                         pWndSubclassedCtrl = new CMFCPropertyGridCtrl;
 163                 }
 164                 else if (strClass == _T("MFCShellList"))
 165                 {
 166                         pWndSubclassedCtrl = new CMFCShellListCtrl;
 167                 }
 168                 else if (strClass == _T("MFCShellTree"))
 169                 {
 170                         pWndSubclassedCtrl = new CMFCShellTreeCtrl;
 171                 }
 172                 else if (strClass == _T("MFCVSListBox"))
 173                 {
 174                         pWndSubclassedCtrl = new CVSListBox;
 175                 }
 176 #endif
 177                 return pWndSubclassedCtrl;
 178         }
 179
 180         return NULL;
 181 }
 182
 183 BOOL CMFCControlContainer::IsSubclassedFeaturePackControl(HWND hWndCtrl)
 184 {
 185         if (hWndCtrl == NULL)
 186         {
 187                 return FALSE;
 188         }
 189
 190         for (int i = 0; i < m_arSubclassedCtrls.GetCount(); i++)
 191         {
 192                 CWnd* pWnd = (CWnd*)m_arSubclassedCtrls[i];
 193                 if (pWnd->GetSafeHwnd() == hWndCtrl)
 194                 {
 195                         return TRUE;
 196                 }
 197         }
 198
 199         return FALSE;
 200 }
 201
 202 void CMFCControlContainer::PreUnsubclassControl(CWnd* pControl)
 203 {
 204         UNREFERENCED_PARAMETER(pControl);
 205
 206 //      CMFCShellListCtrl* pListCtrl = DYNAMIC_DOWNCAST(CMFCShellListCtrl, pControl);
 207 //      if (pListCtrl != NULL && pListCtrl->GetHeaderCtrl().GetSafeHwnd() != NULL)
 208 //      {
 209 //              pListCtrl->GetHeaderCtrl().UnsubclassWindow();
 210 //      }
 211 }
 212
 213 BOOL CMFCControlContainer::ReSubclassControl(HWND hWndCtrl, WORD nIDC, CWnd& control)
 214 {
 215         if (hWndCtrl == NULL)
 216         {
 217                 return FALSE;
 218         }
 219
 220         int nIndex = -1;
 221         for (int i = 0; i < m_arSubclassedCtrls.GetCount(); i++)
 222         {
 223                 CWnd* pWnd = (CWnd*)m_arSubclassedCtrls [i];
 224                 if (pWnd->GetSafeHwnd() == hWndCtrl)
 225                 {
 226                         nIndex = i;
 227                         break;
 228                 }
 229         }
 230
 231         if (nIndex != -1)
 232         {
 233                 CWnd* pWnd = DYNAMIC_DOWNCAST(CWnd, m_arSubclassedCtrls [nIndex]);
 234
 235                 if (pWnd->GetSafeHwnd() != NULL)
 236                 {
 237                         ASSERT_VALID(pWnd);
 238
 239                         // get init state
 240                         DWORD dwSize = 0;
 241                         BYTE* pbInitData = NULL;
 242                         GetControlData(nIDC, dwSize, pbInitData);
 243
 244                         // Free old subclassed control:
 245                         m_arSubclassedCtrls [nIndex] = NULL;
 246
 247                         // unsubclass
 248                         PreUnsubclassControl(pWnd);
 249                         VERIFY(pWnd->UnsubclassWindow() == hWndCtrl);
 250                         // destroy
 251                         delete pWnd;
 252
 253                         // subclass
 254                         if (!control.SubclassWindow(hWndCtrl))
 255                         {
 256                                 ASSERT(FALSE);      // possibly trying to subclass twice?
 257                                 AfxThrowNotSupportedException();
 258                         }
 259
 260                         // set init state
 261                         if (dwSize > 0)
 262                         {
 263                                 control.SendMessage(WM_MFC_INITCTRL, (WPARAM)dwSize, (LPARAM)pbInitData);
 264                         }
 265
 266                         return TRUE;
 267                 }
 268         }
 269
 270         return FALSE;
 271 }
 272
 273 void CMFCControlContainer::SetControlData(WORD nIDC, DWORD dwSize, BYTE* pbData)
 274 {
 275         CByteArray* pArray = new CByteArray;
 276         pArray->SetSize(dwSize);
 277
 278         BYTE* pbBuffer = pArray->GetData();
 279         if (memcpy_s(pbBuffer, dwSize, pbData, dwSize) != 0)
 280         {
 281                 delete pArray;
 282                 ASSERT(FALSE);
 283                 return;
 284         }
 285
 286         m_mapControlData.SetAt(nIDC, pArray);
 287 }
 288
 289 BOOL CMFCControlContainer::GetControlData(WORD nIDC, DWORD& dwSize, BYTE*& pbData)
 290 {
 291         CObject* pData = NULL;
 292         if (m_mapControlData.Lookup(nIDC, pData) && pData != NULL)
 293         {
 294                 CByteArray* pArray = (CByteArray*)pData;
 295                 dwSize = (DWORD)pArray->GetSize();
 296                 pbData = pArray->GetData();
 297                 return TRUE;
 298         }
 299
 300         return FALSE;
 301 }
 302
 303 void CMFCControlContainer::ClearControlData()
 304 {
 305         WORD nIDC;
 306         CObject* pData = NULL;
 307         POSITION pos = m_mapControlData.GetStartPosition();
 308         while (pos != NULL)
 309         {
 310                 m_mapControlData.GetNextAssoc(pos, nIDC, pData);
 311                 CByteArray* pArray = (CByteArray*)pData;
 312                 delete pArray;
 313         }
 314
 315         m_mapControlData.RemoveAll();
 316 }
 317
 318 ////////////////////////////////////////////////////////////////////////////
 319 // Accessing dialog DLGINIT helpers
 320
 321 int __stdcall CMFCControlContainer::UTF8ToString(LPCSTR lpSrc, CString& strDst, int nLength)
 322 {
 323         LPTSTR lpDst = NULL;
 324         int count = ::MultiByteToWideChar(CP_UTF8, 0, lpSrc, nLength, NULL, 0);
 325         if (count <= 0)
 326         {
 327                 return 0;
 328         }
 329
 330         LPWSTR lpWide = new WCHAR[count + 1];
 331         memset(lpWide, 0, (count + 1) * sizeof(WCHAR));
 332
 333         ::MultiByteToWideChar(CP_UTF8, 0, lpSrc, nLength, lpWide, count);
 334
 335 #ifdef _UNICODE
 336         lpDst = lpWide;
 337 #else
 338         count = ::WideCharToMultiByte(::GetACP(), 0, lpWide, -1, NULL, 0, NULL, 0);
 339
 340         if (count > 0)
 341         {
 342                 lpDst = new char[count + 1];
 343                 memset(lpDst, 0, count + 1);
 344
 345                 ::WideCharToMultiByte(::GetACP(), 0, lpWide, -1, lpDst, count, NULL, 0);
 346         }
 347
 348         delete [] lpWide;
 349 #endif
 350
 351         strDst = lpDst;
 352         delete[] lpDst;
 353         return count;
 354 }
 355
 356 BOOL __stdcall CMFCControlContainer::ReadBoolProp(CTagManager& /*tagManager*/, LPCTSTR lpszTag, BOOL& bMember)
 357 {
 358         if (lpszTag == NULL)
 359         {
 360                 return FALSE;
 361         }
 362
 363         CString str;
 364 //      tagManager.ExcludeTag(lpszTag, str);
 365
 366         if (str.IsEmpty())
 367         {
 368                 return FALSE;
 369         }
 370
 371         bMember = (str.CompareNoCase(PS_True) == 0);
 372         return TRUE;
 373 }