1 // 2 // This file is part of the aMule Project. 3 // 4 // Copyright (c) 2003-2011 aMule Team ( admin@amule.org / http://www.amule.org ) 5 // Copyright (c) 2002-2011 Robert Rostek ( tecxx@rrs.at ) 6 // 7 // Any parts of this program derived from the xMule, lMule or eMule project, 8 // or contributed by third-party developers are copyrighted by their 9 // respective authors. 10 // 11 // This program is free software; you can redistribute it and/or modify 12 // it under the terms of the GNU General Public License as published by 13 // the Free Software Foundation; either version 2 of the License, or 14 // (at your option) any later version. 15 // 16 // This program is distributed in the hope that it will be useful, 17 // but WITHOUT ANY WARRANTY; without even the implied warranty of 18 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 19 // GNU General Public License for more details. 20 // 21 // You should have received a copy of the GNU General Public License 22 // along with this program; if not, write to the Free Software 23 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA 24 // 25 26 #ifndef DIRECTORYTREECTRL_H 27 #define DIRECTORYTREECTRL_H 28 29 #include <wx/treectrl.h> 30 #include <vector> 31 #include <map> 32 33 #include <common/Path.h> 34 35 36 class CDirectoryTreeCtrl : public wxTreeCtrl 37 { 38 public: 39 typedef std::vector<CPath> PathList; 40 41 CDirectoryTreeCtrl(wxWindow* parent, int id, const wxPoint& pos, wxSize siz, int flags); 42 virtual ~CDirectoryTreeCtrl(); 43 44 // get all shared directories 45 void GetSharedDirectories(PathList* list); 46 // set list of shared directories 47 void SetSharedDirectories(PathList* list); 48 49 // User made any changes to list? 50 bool HasChanged; 51 52 // initialize control 53 void Init(); 54 55 private: 56 // add a new item 57 void AddChildItem(wxTreeItemId hBranch, const CPath& item); 58 // add subdirectory items 59 void AddSubdirectories(wxTreeItemId hBranch, const CPath& path); 60 // returns true if folder has at least one subdirectory 61 bool HasSubdirectories(const CPath& path); 62 // return the full path of an item (like C:\abc\somewhere\inheaven\) 63 CPath GetFullPath(wxTreeItemId hItem); 64 // check status of an item has changed 65 void CheckChanged(wxTreeItemId hItem, bool bChecked, bool recursed); 66 // returns true if a subdirectory of strDir is shared 67 bool HasSharedSubdirectory(const CPath& path); 68 // set shared directories according to list 69 void UpdateSharedDirectories(); 70 // when sharing a directory, make all parent directories red 71 void UpdateParentItems(wxTreeItemId hChild, bool add); 72 // set color red if there's a shared subdirectory 73 void SetHasSharedSubdirectory(wxTreeItemId hItem, bool add); 74 75 // share list access 76 bool IsShared(const CPath& path); 77 void AddShare(const CPath& path); 78 void DelShare(const CPath& path); 79 void MarkChildren(wxTreeItemId hChild, bool mark, bool recursed); 80 81 void OnItemExpanding(wxTreeEvent& evt); 82 void OnRButtonDown(wxTreeEvent& evt); 83 void OnItemActivated(wxTreeEvent& evt); 84 85 typedef std::pair<wxString, CPath> SharedMapItem; 86 typedef std::map<wxString, CPath> SharedMap; 87 SharedMap m_lstShared; 88 // get map key from path (normalized path) 89 wxString GetKey(const CPath& path); 90 91 bool m_IsInit; 92 // Are we running the remote GUI, and from a remote location? 93 bool m_IsRemote; 94 95 wxTreeItemId m_root; 96 97 DECLARE_EVENT_TABLE() 98 }; 99 100 #undef wxTreeItemId 101 102 #endif // DIRECTORYTREECTRL_H 103 // File_checked_for_headers 104