src/Utils/MiscUI/DlgTemplate.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2006,2008 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 class CDlgTemplate
  20 {
  21
  22 public:
  23
  24         CDlgTemplate()
  25         {
  26                 usedBufferLength = 0;
  27                 totalBufferLength = 0;
  28                 dialogTemplate = 0;
  29         };
  30     CDlgTemplate(LPCTSTR caption, DWORD style, short x, short y, short w, short h,
  31         LPCTSTR font = NULL, LONG fontSize = 8)
  32     {
  33
  34         usedBufferLength = sizeof(DLGTEMPLATE );
  35         totalBufferLength = usedBufferLength;
  36
  37         dialogTemplate = (DLGTEMPLATE*)malloc(totalBufferLength);
  38
  39                 if (dialogTemplate)
  40                 {
  41                         dialogTemplate->style = style;
  42
  43                         dialogTemplate->style |= DS_SETFONT;
  44
  45                         dialogTemplate->x     = x;
  46                         dialogTemplate->y     = y;
  47                         dialogTemplate->cx    = w;
  48                         dialogTemplate->cy    = h;
  49                         dialogTemplate->cdit  = 0;
  50
  51                         dialogTemplate->dwExtendedStyle = 0;
  52
  53                         //the dialog box doesn't have a menu or a special class
  54
  55                         AppendData(_T("\0"), 2);
  56                         AppendData(_T("\0"), 2);
  57
  58                         //add the dialog's caption to the template
  59
  60                         AppendString(caption);
  61
  62                         AppendData(&fontSize, sizeof(WORD));
  63                         AppendString(font);
  64                 }
  65         }
  66
  67     void AddComponent(LPCTSTR type, LPCTSTR caption, DWORD style, DWORD exStyle,
  68         short x, short y, short w, short h, WORD id)
  69     {
  70
  71         DLGITEMTEMPLATE item;
  72
  73         item.style = style;
  74         item.x     = x;
  75         item.y     = y;
  76         item.cx    = w;
  77         item.cy    = h;
  78         item.id    = id;
  79
  80         item.dwExtendedStyle = exStyle;
  81
  82         AppendData(&item, sizeof(DLGITEMTEMPLATE));
  83
  84         AppendString(type);
  85         AppendString(caption);
  86
  87         WORD creationDataLength = 0;
  88         AppendData(&creationDataLength, sizeof(WORD));
  89
  90         //increment the component count
  91
  92         dialogTemplate->cdit++;
  93
  94     }
  95
  96     void AddButton(LPCTSTR caption, DWORD style, DWORD exStyle, short x, short y,
  97         short w, short h, WORD id)
  98     {
  99
 100         AddStandardComponent(0x0080, caption, style, exStyle, x, y, w, h, id);
 101
 102         WORD creationDataLength = 0;
 103         AppendData(&creationDataLength, sizeof(WORD));
 104
 105     }
 106
 107     void AddEditBox(LPCTSTR caption, DWORD style, DWORD exStyle, short x, short y,
 108         short w, short h, WORD id)
 109     {
 110
 111         AddStandardComponent(0x0081, caption, style, exStyle, x, y, w, h, id);
 112
 113         WORD creationDataLength = 0;
 114         AppendData(&creationDataLength, sizeof(WORD));
 115
 116     }
 117
 118     void AddStatic(LPCTSTR caption, DWORD style, DWORD exStyle, short x, short y,
 119         short w, short h, WORD id)
 120     {
 121
 122         AddStandardComponent(0x0082, caption, style, exStyle, x, y, w, h, id);
 123
 124         WORD creationDataLength = 0;
 125         AppendData(&creationDataLength, sizeof(WORD));
 126
 127     }
 128
 129     void AddListBox(LPCTSTR caption, DWORD style, DWORD exStyle, short x, short y,
 130         short w, short h, WORD id)
 131     {
 132
 133         AddStandardComponent(0x0083, caption, style, exStyle, x, y, w, h, id);
 134
 135         WORD creationDataLength = 0;
 136         AppendData(&creationDataLength, sizeof(WORD));
 137
 138     }
 139
 140     void AddScrollBar(LPCTSTR caption, DWORD style, DWORD exStyle, short x, short y,
 141         short w, short h, WORD id)
 142     {
 143
 144         AddStandardComponent(0x0084, caption, style, exStyle, x, y, w, h, id);
 145
 146         WORD creationDataLength = 0;
 147         AppendData(&creationDataLength, sizeof(WORD));
 148
 149     }
 150
 151     void AddComboBox(LPCTSTR caption, DWORD style, DWORD exStyle, short x, short y,
 152         short w, short h, WORD id)
 153     {
 154
 155         AddStandardComponent(0x0085, caption, style, exStyle, x, y, w, h, id);
 156
 157         WORD creationDataLength = 0;
 158         AppendData(&creationDataLength, sizeof(WORD));
 159
 160     }
 161
 162     /**
 163      * Returns a pointer to the Win32 dialog template which the object
 164      * represents. This pointer may become invalid if additional
 165      * components are added to the template.
 166      */
 167
 168     operator const DLGTEMPLATE*() const
 169     {
 170         return dialogTemplate;
 171     }
 172
 173     virtual ~CDlgTemplate()
 174     {
 175         free(dialogTemplate);
 176     }
 177
 178 protected:
 179
 180     void AddStandardComponent(WORD type, LPCTSTR caption, DWORD style,
 181         DWORD exStyle, short x, short y, short w, short h, WORD id)
 182     {
 183
 184         DLGITEMTEMPLATE item;
 185
 186         // DWORD align the beginning of the component data
 187
 188         AlignData(sizeof(DWORD));
 189
 190         item.style = style;
 191         item.x     = x;
 192         item.y     = y;
 193         item.cx    = w;
 194         item.cy    = h;
 195         item.id    = id;
 196
 197         item.dwExtendedStyle = exStyle;
 198
 199         AppendData(&item, sizeof(DLGITEMTEMPLATE));
 200
 201         WORD preType = 0xFFFF;
 202
 203         AppendData(&preType, sizeof(WORD));
 204         AppendData(&type, sizeof(WORD));
 205
 206         AppendString(caption);
 207
 208         // Increment the component count
 209
 210         dialogTemplate->cdit++;
 211
 212     }
 213
 214     void AlignData(int size)
 215     {
 216
 217         int paddingSize = usedBufferLength % size;
 218
 219         if (paddingSize != 0)
 220         {
 221             EnsureSpace(paddingSize);
 222             usedBufferLength += paddingSize;
 223         }
 224
 225     }
 226
 227     void AppendString(LPCTSTR string)
 228     {
 229 #ifndef _UNICODE
 230         int length = MultiByteToWideChar(CP_ACP, 0, string, -1, NULL, 0);
 231 #else
 232                 int length = (int)wcslen(string)+1;
 233 #endif
 234         WCHAR* wideString = (WCHAR*)malloc(sizeof(WCHAR) * length);
 235                 if (wideString)
 236                 {
 237 #ifndef _UNICODE
 238                         MultiByteToWideChar(CP_ACP, 0, string, -1, wideString, length);
 239 #else
 240                         wcscpy_s(wideString, length, string);
 241 #endif
 242                         AppendData(wideString, length * sizeof(WCHAR));
 243                         free(wideString);
 244                 }
 245         }
 246
 247     void AppendData(void* data, int dataLength)
 248     {
 249
 250         EnsureSpace(dataLength);
 251
 252         memcpy((char*)dialogTemplate + usedBufferLength, data, dataLength);
 253         usedBufferLength += dataLength;
 254
 255     }
 256
 257     void EnsureSpace(int length)
 258     {
 259
 260         if (length + usedBufferLength > totalBufferLength)
 261         {
 262
 263             totalBufferLength += length * 2;
 264
 265             void* newBuffer = malloc(totalBufferLength);
 266             memcpy(newBuffer, dialogTemplate, usedBufferLength);
 267
 268             free(dialogTemplate);
 269             dialogTemplate = (DLGTEMPLATE*)newBuffer;
 270
 271         }
 272
 273     }
 274
 275 private:
 276
 277     DLGTEMPLATE* dialogTemplate;
 278
 279     int totalBufferLength;
 280     int usedBufferLength;
 281
 282 };