rosapps/regedit/regtree.cpp

   1 /*
   2  *  ReactOS regedit
   3  *
   4  *  regtree.cpp
   5  *
   6  *  Copyright (C) 2002  Robert Dickenson <robd@reactos.org>
   7  *
   8  *  This program is free software; you can redistribute it and/or modify
   9  *  it under the terms of the GNU General Public License as published by
  10  *  the Free Software Foundation; either version 2 of the License, or
  11  *  (at your option) any later version.
  12  *
  13  *  This program is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *  GNU General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU General Public License
  19  *  along with this program; if not, write to the Free Software
  20  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21  */
  22
  23 #ifdef _MSC_VER
  24 #include "stdafx.h"
  25 #else
  26 #define WIN32_LEAN_AND_MEAN     // Exclude rarely-used stuff from Windows headers
  27 #include <windows.h>
  28 #include <commctrl.h>
  29 #include <stdlib.h>
  30 #include <malloc.h>
  31 #include <memory.h>
  32 #include <tchar.h>
  33 #include <process.h>
  34 #include <stdio.h>
  35 #endif
  36
  37 #include "regedit.h"
  38 #include "regtree.h"
  39
  40 // Global Variables:
  41 extern HINSTANCE hInst;                   // current instance
  42 extern HWND hMainWnd;                     // Main Window
  43
  44 // Global variables and constants
  45 // Image_Open, Image_Closed, and Image_Root - integer variables for
  46 // indexes of the images.
  47 // CX_BITMAP and CY_BITMAP - width and height of an icon.
  48 // NUM_BITMAPS - number of bitmaps to add to the image list.
  49 int Image_Open;
  50 int Image_Closed;
  51 int Image_Root;
  52
  53 #define CX_BITMAP    16
  54 #define CY_BITMAP    16
  55 #define NUM_BITMAPS  3
  56
  57
  58 // AddItemToTree - adds items to a tree view control.
  59 // Returns the handle to the newly added item.
  60 // hwndTV - handle to the tree view control.
  61 // lpszItem - text of the item to add.
  62 // nLevel - level at which to add the item.
  63
  64 HTREEITEM AddItemToTree(HWND hwndTV, LPSTR lpszItem, int nLevel)
  65 {
  66     TVITEM tvi;
  67     TVINSERTSTRUCT tvins;
  68     static HTREEITEM hPrev = (HTREEITEM)TVI_FIRST;
  69     static HTREEITEM hPrevRootItem = NULL;
  70     static HTREEITEM hPrevLev2Item = NULL;
  71     HTREEITEM hti;
  72
  73     tvi.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM;
  74
  75     // Set the text of the item.
  76     tvi.pszText = lpszItem;
  77     tvi.cchTextMax = lstrlen(lpszItem);
  78
  79     // Assume the item is not a parent item, so give it an image.
  80     tvi.iImage = Image_Root;
  81     tvi.iSelectedImage = Image_Root;
  82
  83     tvi.cChildren = 1;
  84
  85
  86     // Save the heading level in the item's application-defined data area.
  87     tvi.lParam = (LPARAM)nLevel;
  88
  89     tvins.item = tvi;
  90     tvins.hInsertAfter = hPrev;
  91
  92     // Set the parent item based on the specified level.
  93     if (nLevel == 1)
  94         tvins.hParent = TVI_ROOT;
  95     else if (nLevel == 2)
  96         tvins.hParent = hPrevRootItem;
  97     else
  98         tvins.hParent = hPrevLev2Item;
  99
 100     // Add the item to the tree view control.
 101     hPrev = (HTREEITEM)SendMessage(hwndTV, TVM_INSERTITEM, 0, (LPARAM)(LPTVINSERTSTRUCT)&tvins);
 102
 103     // Save the handle to the item.
 104     if (nLevel == 1)
 105         hPrevRootItem = hPrev;
 106     else if (nLevel == 2)
 107         hPrevLev2Item = hPrev;
 108
 109     // The new item is a child item. Give the parent item a
 110     // closed folder bitmap to indicate it now has child items.
 111     if (nLevel > 1) {
 112         hti = TreeView_GetParent(hwndTV, hPrev);
 113         tvi.mask = TVIF_IMAGE | TVIF_SELECTEDIMAGE;
 114         tvi.hItem = hti;
 115         tvi.iImage = Image_Closed;
 116         tvi.iSelectedImage = Image_Closed;
 117         TreeView_SetItem(hwndTV, &tvi);
 118     }
 119
 120     return hPrev;
 121 }
 122
 123 // InitTreeViewItems - extracts headings from the specified file and
 124 // passes them to a function that adds them to a tree view control.
 125 // Returns TRUE if successful, or FALSE otherwise.
 126 // hwndTV - handle to the tree view control.
 127 // lpszFileName - name of file with headings.
 128
 129 BOOL InitTreeViewItems(HWND hwndTV, LPSTR lpszFileName)
 130 {
 131     HTREEITEM hItem;
 132
 133     hItem = AddItemToTree(hwndTV, "My Computer", 1);
 134     AddItemToTree(hwndTV, "HKEY_CLASSES_ROOT", 2);
 135     AddItemToTree(hwndTV, "HKEY_CURRENT_USER", 2);
 136     AddItemToTree(hwndTV, "HKEY_LOCAL_MACHINE", 2);
 137     AddItemToTree(hwndTV, "HKEY_USERS", 2);
 138     AddItemToTree(hwndTV, "HKEY_CURRENT_CONFIG", 2);
 139
 140     TreeView_Expand(hwndTV, hItem, TVE_EXPAND);
 141     return TRUE;
 142 }
 143
 144 // InitTreeViewImageLists - creates an image list, adds three bitmaps
 145 // to it, and associates the image list with a tree view control.
 146 // Returns TRUE if successful, or FALSE otherwise.
 147 // hwndTV - handle to the tree view control.
 148
 149 BOOL InitTreeViewImageLists(HWND hwndTV)
 150 {
 151     HIMAGELIST himl;  // handle to image list
 152     HBITMAP hbmp;     // handle to bitmap
 153
 154     // Create the image list.
 155     if ((himl = ImageList_Create(CX_BITMAP, CY_BITMAP,
 156         FALSE, NUM_BITMAPS, 0)) == NULL)
 157         return FALSE;
 158
 159     // Add the open file, closed file, and document bitmaps.
 160     hbmp = LoadBitmap(hInst, MAKEINTRESOURCE(IDB_OPEN_FILE));
 161     Image_Open = ImageList_Add(himl, hbmp, (HBITMAP) NULL);
 162     DeleteObject(hbmp);
 163
 164     hbmp = LoadBitmap(hInst, MAKEINTRESOURCE(IDB_CLOSED_FILE));
 165     Image_Closed = ImageList_Add(himl, hbmp, (HBITMAP) NULL);
 166     DeleteObject(hbmp);
 167
 168     hbmp = LoadBitmap(hInst, MAKEINTRESOURCE(IDB_ROOT));
 169     Image_Root = ImageList_Add(himl, hbmp, (HBITMAP) NULL);
 170     DeleteObject(hbmp);
 171
 172     // Fail if not all of the images were added.
 173     if (ImageList_GetImageCount(himl) < 3)
 174         return FALSE;
 175
 176     // Associate the image list with the tree view control.
 177     TreeView_SetImageList(hwndTV, himl, TVSIL_NORMAL);
 178
 179     return TRUE;
 180 }
 181
 182 // CreateTreeView - creates a tree view control.
 183 // Returns the handle to the new control if successful,
 184 // or NULL otherwise.
 185 // hwndParent - handle to the control's parent window.
 186 // lpszFileName - name of the file to parse for tree view items.
 187
 188 HWND CreateTreeView(HWND hwndParent, LPSTR lpszFileName)
 189 {
 190     RECT rcClient;  // dimensions of client area
 191     HWND hwndTV;    // handle to tree view control
 192
 193     // Get the dimensions of the parent window's client area, and create
 194     // the tree view control.
 195     GetClientRect(hwndParent, &rcClient);
 196     hwndTV = CreateWindowEx(0, WC_TREEVIEW, "Tree View",
 197         WS_VISIBLE | WS_CHILD | WS_BORDER | TVS_HASLINES | TVS_HASBUTTONS | TVS_LINESATROOT,
 198         0, 0, rcClient.right, rcClient.bottom,
 199         hwndParent, (HMENU)TREE_WINDOW, hInst, NULL);
 200
 201     // Initialize the image list, and add items to the control.
 202     if (!InitTreeViewImageLists(hwndTV) || !InitTreeViewItems(hwndTV, lpszFileName)) {
 203         DestroyWindow(hwndTV);
 204         return FALSE;
 205     }
 206     return hwndTV;
 207 }
 208