src/Utils/SysImageList.cpp

   1 // TortoiseSVN - 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 #include "stdafx.h"
  20 #include "SysImageList.h"
  21 #include "TGitPath.h"
  22
  23
  24 // Singleton constructor and destructor (private)
  25
  26 CSysImageList * CSysImageList::instance = 0;
  27
  28 CSysImageList::CSysImageList()
  29 {
  30         SHFILEINFO ssfi;
  31         TCHAR windir[MAX_PATH];
  32         GetWindowsDirectory(windir, MAX_PATH);  // MAX_PATH ok.
  33         HIMAGELIST hSystemImageList =
  34                 (HIMAGELIST)SHGetFileInfo(
  35                         windir,
  36                         0,
  37                         &ssfi, sizeof ssfi,
  38                         SHGFI_SYSICONINDEX | SHGFI_SMALLICON);
  39         Attach(hSystemImageList);
  40 }
  41
  42 CSysImageList::~CSysImageList()
  43 {
  44         Detach();
  45 }
  46
  47
  48 // Singleton specific operations
  49
  50 CSysImageList& CSysImageList::GetInstance()
  51 {
  52         if (instance == 0)
  53                 instance = new CSysImageList;
  54         return *instance;
  55 }
  56
  57 void CSysImageList::Cleanup()
  58 {
  59         delete instance;
  60         instance = 0;
  61 }
  62
  63
  64 // Operations
  65
  66 int CSysImageList::GetDirIconIndex() const
  67 {
  68         SHFILEINFO sfi;
  69         SecureZeroMemory(&sfi, sizeof sfi);
  70
  71         SHGetFileInfo(
  72                 _T("Doesn't matter"),
  73                 FILE_ATTRIBUTE_DIRECTORY,
  74                 &sfi, sizeof sfi,
  75                 SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_USEFILEATTRIBUTES);
  76
  77         return sfi.iIcon;
  78 }
  79
  80 int CSysImageList::GetDirOpenIconIndex() const
  81 {
  82         SHFILEINFO sfi;
  83         SecureZeroMemory(&sfi, sizeof sfi);
  84
  85         SHGetFileInfo(
  86                 _T("Doesn't matter"),
  87                 FILE_ATTRIBUTE_DIRECTORY,
  88                 &sfi, sizeof sfi,
  89                 SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_USEFILEATTRIBUTES | SHGFI_OPENICON);
  90
  91         return sfi.iIcon;
  92 }
  93
  94 int CSysImageList::GetDefaultIconIndex() const
  95 {
  96         SHFILEINFO sfi;
  97         SecureZeroMemory(&sfi, sizeof sfi);
  98
  99         SHGetFileInfo(
 100                 _T(""),
 101                 FILE_ATTRIBUTE_NORMAL,
 102                 &sfi, sizeof sfi,
 103                 SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_USEFILEATTRIBUTES);
 104
 105         return sfi.iIcon;
 106 }
 107
 108 int CSysImageList::GetFileIconIndex(const CString& file) const
 109 {
 110         SHFILEINFO sfi;
 111         SecureZeroMemory(&sfi, sizeof sfi);
 112
 113         SHGetFileInfo(
 114                 file,
 115                 FILE_ATTRIBUTE_NORMAL,
 116                 &sfi, sizeof sfi,
 117                 SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_USEFILEATTRIBUTES);
 118
 119         return sfi.iIcon;
 120 }
 121
 122 int CSysImageList::GetPathIconIndex(const CTGitPath& filePath) const
 123 {
 124         CString strExtension = filePath.GetFileExtension();
 125         strExtension.MakeUpper();
 126         IconIndexMap::iterator it = m_indexCache.lower_bound(strExtension);
 127         if (it == m_indexCache.end() || strExtension < it->first)
 128         {
 129                 // We don't have this extension in the map
 130                 int iconIndex = GetFileIconIndex(filePath.GetFilename());
 131                 it = m_indexCache.insert(it, std::make_pair(strExtension, iconIndex));
 132         }
 133         // We must have found it
 134         return it->second;
 135 }