developer/VC/WTL80/Samples/WTLExplorer/ShellMgr.H

   1 // shellmgr.h
   2
   3 #ifndef __SHELLMGR_H__
   4 #define __SHELLMGR_H__
   5
   6 #pragma once
   7
   8 #include <shlobj.h>
   9 #include <atlctrls.h>
  10
  11
  12 class CShellItemIDList
  13 {
  14 public:
  15         LPITEMIDLIST m_pidl;
  16
  17         CShellItemIDList(LPITEMIDLIST pidl = NULL) : m_pidl(pidl)
  18         { }
  19
  20         ~CShellItemIDList()
  21         {
  22                 ::CoTaskMemFree(m_pidl);
  23         }
  24
  25         void Attach(LPITEMIDLIST pidl)
  26         {
  27                 ::CoTaskMemFree(m_pidl);
  28                 m_pidl = pidl;
  29         }
  30
  31         LPITEMIDLIST Detach()
  32         {
  33                 LPITEMIDLIST pidl = m_pidl;
  34                 m_pidl = NULL;
  35                 return pidl;
  36         }
  37
  38         bool IsNull() const
  39         {
  40                 return (m_pidl == NULL);
  41         }
  42
  43         CShellItemIDList& operator =(LPITEMIDLIST pidl)
  44         {
  45                 Attach(pidl);
  46                 return *this;
  47         }
  48
  49         LPITEMIDLIST* operator &()
  50         {
  51                 return &m_pidl;
  52         }
  53
  54         operator LPITEMIDLIST()
  55         {
  56                 return m_pidl;
  57         }
  58
  59         operator LPCTSTR()
  60         {
  61                 return (LPCTSTR)m_pidl;
  62         }
  63
  64         operator LPTSTR()
  65         {
  66                 return (LPTSTR)m_pidl;
  67         }
  68
  69         void CreateEmpty(UINT cbSize)
  70         {
  71                 ::CoTaskMemFree(m_pidl);
  72                 m_pidl = (LPITEMIDLIST)::CoTaskMemAlloc(cbSize);
  73                 ATLASSERT(m_pidl != NULL);
  74                 if(m_pidl != NULL)
  75                         memset(m_pidl, 0, cbSize);
  76         }
  77 };
  78
  79
  80 typedef struct _LVItemData
  81 {
  82         _LVItemData() : ulAttribs(0)
  83         { }
  84
  85         CComPtr<IShellFolder> spParentFolder;
  86
  87         CShellItemIDList lpi;
  88         ULONG ulAttribs;
  89
  90 } LVITEMDATA, *LPLVITEMDATA;
  91
  92 typedef struct _TVItemData
  93 {
  94         _TVItemData()
  95         { }
  96
  97         CComPtr<IShellFolder> spParentFolder;
  98
  99         CShellItemIDList lpi;
 100         CShellItemIDList lpifq;
 101
 102 } TVITEMDATA, *LPTVITEMDATA;
 103
 104
 105 class CShellMgr
 106 {
 107 public:
 108         int GetIconIndex(LPITEMIDLIST lpi, UINT uFlags);
 109
 110         void GetNormalAndSelectedIcons(LPITEMIDLIST lpifq, LPTVITEM lptvitem);
 111
 112         LPITEMIDLIST ConcatPidls(LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2);
 113
 114         BOOL GetName (LPSHELLFOLDER lpsf, LPITEMIDLIST lpi, DWORD dwFlags, LPTSTR lpFriendlyName);
 115         LPITEMIDLIST Next(LPCITEMIDLIST pidl);
 116         UINT GetSize(LPCITEMIDLIST pidl);
 117         LPITEMIDLIST CopyITEMID(LPITEMIDLIST lpi);
 118
 119         LPITEMIDLIST GetFullyQualPidl(LPSHELLFOLDER lpsf, LPITEMIDLIST lpi);
 120
 121         BOOL DoContextMenu(HWND hwnd, LPSHELLFOLDER lpsfParent, LPITEMIDLIST lpi, POINT point);
 122 };
 123
 124 #endif //__SHELLMGR_H__