src/DirectoryTreeCtrl.h

   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