/* * ReactOS regedit * * regtree.cpp * * Copyright (C) 2002 Robert Dickenson * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ #ifdef _MSC_VER #include "stdafx.h" #else #define WIN32_LEAN_AND_MEAN // Exclude rarely-used stuff from Windows headers #include #include #include #include #include #include #include #include #endif #include "regedit.h" #include "regtree.h" // Global Variables: extern HINSTANCE hInst; // current instance extern HWND hMainWnd; // Main Window // Global variables and constants // Image_Open, Image_Closed, and Image_Root - integer variables for // indexes of the images. // CX_BITMAP and CY_BITMAP - width and height of an icon. // NUM_BITMAPS - number of bitmaps to add to the image list. int Image_Open; int Image_Closed; int Image_Root; #define CX_BITMAP 16 #define CY_BITMAP 16 #define NUM_BITMAPS 3 // AddItemToTree - adds items to a tree view control. // Returns the handle to the newly added item. // hwndTV - handle to the tree view control. // lpszItem - text of the item to add. // nLevel - level at which to add the item. HTREEITEM AddItemToTree(HWND hwndTV, LPSTR lpszItem, int nLevel) { TVITEM tvi; TVINSERTSTRUCT tvins; static HTREEITEM hPrev = (HTREEITEM)TVI_FIRST; static HTREEITEM hPrevRootItem = NULL; static HTREEITEM hPrevLev2Item = NULL; HTREEITEM hti; tvi.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM; // Set the text of the item. tvi.pszText = lpszItem; tvi.cchTextMax = lstrlen(lpszItem); // Assume the item is not a parent item, so give it an image. tvi.iImage = Image_Root; tvi.iSelectedImage = Image_Root; tvi.cChildren = 1; // Save the heading level in the item's application-defined data area. tvi.lParam = (LPARAM)nLevel; tvins.item = tvi; tvins.hInsertAfter = hPrev; // Set the parent item based on the specified level. if (nLevel == 1) tvins.hParent = TVI_ROOT; else if (nLevel == 2) tvins.hParent = hPrevRootItem; else tvins.hParent = hPrevLev2Item; // Add the item to the tree view control. hPrev = (HTREEITEM)SendMessage(hwndTV, TVM_INSERTITEM, 0, (LPARAM)(LPTVINSERTSTRUCT)&tvins); // Save the handle to the item. if (nLevel == 1) hPrevRootItem = hPrev; else if (nLevel == 2) hPrevLev2Item = hPrev; // The new item is a child item. Give the parent item a // closed folder bitmap to indicate it now has child items. if (nLevel > 1) { hti = TreeView_GetParent(hwndTV, hPrev); tvi.mask = TVIF_IMAGE | TVIF_SELECTEDIMAGE; tvi.hItem = hti; tvi.iImage = Image_Closed; tvi.iSelectedImage = Image_Closed; TreeView_SetItem(hwndTV, &tvi); } return hPrev; } // InitTreeViewItems - extracts headings from the specified file and // passes them to a function that adds them to a tree view control. // Returns TRUE if successful, or FALSE otherwise. // hwndTV - handle to the tree view control. // lpszFileName - name of file with headings. BOOL InitTreeViewItems(HWND hwndTV, LPSTR lpszFileName) { HTREEITEM hItem; hItem = AddItemToTree(hwndTV, "My Computer", 1); AddItemToTree(hwndTV, "HKEY_CLASSES_ROOT", 2); AddItemToTree(hwndTV, "HKEY_CURRENT_USER", 2); AddItemToTree(hwndTV, "HKEY_LOCAL_MACHINE", 2); AddItemToTree(hwndTV, "HKEY_USERS", 2); AddItemToTree(hwndTV, "HKEY_CURRENT_CONFIG", 2); TreeView_Expand(hwndTV, hItem, TVE_EXPAND); return TRUE; } // InitTreeViewImageLists - creates an image list, adds three bitmaps // to it, and associates the image list with a tree view control. // Returns TRUE if successful, or FALSE otherwise. // hwndTV - handle to the tree view control. BOOL InitTreeViewImageLists(HWND hwndTV) { HIMAGELIST himl; // handle to image list HBITMAP hbmp; // handle to bitmap // Create the image list. if ((himl = ImageList_Create(CX_BITMAP, CY_BITMAP, FALSE, NUM_BITMAPS, 0)) == NULL) return FALSE; // Add the open file, closed file, and document bitmaps. hbmp = LoadBitmap(hInst, MAKEINTRESOURCE(IDB_OPEN_FILE)); Image_Open = ImageList_Add(himl, hbmp, (HBITMAP) NULL); DeleteObject(hbmp); hbmp = LoadBitmap(hInst, MAKEINTRESOURCE(IDB_CLOSED_FILE)); Image_Closed = ImageList_Add(himl, hbmp, (HBITMAP) NULL); DeleteObject(hbmp); hbmp = LoadBitmap(hInst, MAKEINTRESOURCE(IDB_ROOT)); Image_Root = ImageList_Add(himl, hbmp, (HBITMAP) NULL); DeleteObject(hbmp); // Fail if not all of the images were added. if (ImageList_GetImageCount(himl) < 3) return FALSE; // Associate the image list with the tree view control. TreeView_SetImageList(hwndTV, himl, TVSIL_NORMAL); return TRUE; } // CreateTreeView - creates a tree view control. // Returns the handle to the new control if successful, // or NULL otherwise. // hwndParent - handle to the control's parent window. // lpszFileName - name of the file to parse for tree view items. HWND CreateTreeView(HWND hwndParent, LPSTR lpszFileName) { RECT rcClient; // dimensions of client area HWND hwndTV; // handle to tree view control // Get the dimensions of the parent window's client area, and create // the tree view control. GetClientRect(hwndParent, &rcClient); hwndTV = CreateWindowEx(0, WC_TREEVIEW, "Tree View", WS_VISIBLE | WS_CHILD | WS_BORDER | TVS_HASLINES | TVS_HASBUTTONS | TVS_LINESATROOT, 0, 0, rcClient.right, rcClient.bottom, hwndParent, (HMENU)TREE_WINDOW, hInst, NULL); // Initialize the image list, and add items to the control. if (!InitTreeViewImageLists(hwndTV) || !InitTreeViewItems(hwndTV, lpszFileName)) { DestroyWindow(hwndTV); return FALSE; } return hwndTV; }