mmxshellgenelal.cpp

类似 windows explorer的工具

// MMXShellGenelal.cpp: implementation of the CMMXShellGenelal class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "mmxshell2.h"
#include "MMXShellGenelal.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

#include "MainFrm.h"
#define m_aryShowDriver ((CMainFrame*)AfxGetMainWnd())->m_aryShowDriver


// 初始化静态成员
SDesktopFolderPtr CMMXShellGenelal::m_pDesktopFolder;
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CMMXShellGenelal::CMMXShellGenelal()
{

}

CMMXShellGenelal::~CMMXShellGenelal()
{

}

BOOL CMMXShellGenelal::GetItemContextMenu(TVITEMDATA* pData, 
                    CShellContextMenu& rCtxMenu)
{
	if (!pData->IsValid())
    {
        ASSERT(FALSE); // remove the line if need.
        return FALSE;
    }

	return rCtxMenu.Create(pData->pParentFolder,
		pData->pidlAbs.GetLastChild());
}
// static
// 判断pidl 是否是“我的电脑”
// 判断依据：比较内容
BOOL CMMXShellGenelal::IsMyComputer(LPCITEMIDLIST pidl)
{
    //static CShellPidl pidlDesktop((UINT)CSIDL_DESKTOP);
    static CShellPidl pidlMyComputer((UINT)CSIDL_DRIVES);
    
    IShellFolder* pDesktopFolder = m_pDesktopFolder;
    HRESULT hres = pDesktopFolder->CompareIDs(0,
        pidl, pidlMyComputer);
    
    if ((short)HRESULT_CODE(hres) != 0)
    {
        return FALSE;
    }
    else 
    {
        // the two pidls are equal
        return TRUE;
    }
}

// 得到虚拟对象的显示名称
// static
CString CMMXShellGenelal::GetVDisplayName(SShellFolderPtr& pParentFolder, 
                                        LPCITEMIDLIST pidl)
{
    CString sName;
    CShellString str;
    
    // use a global or contextual displayname
    DWORD uDisplayFlags = SHGDN_INFOLDER;
    
    pParentFolder->GetDisplayNameOf(pidl, uDisplayFlags
        | SHGDN_INCLUDE_NONFILESYS, str.GetPointer(pidl));
    
    sName = str;	// copy to string
    
    return sName;
}

// static
// 是否显示对象
BOOL CMMXShellGenelal::IsShowVirtualObject(BOOL bIsUDisk,
    SShellFolderPtr& pParentFolder, LPCITEMIDLIST pidlParent,
    LPCITEMIDLIST pidl)
{
    int i;
    CString strLast4Char;
    CString strDisplayName;
    
    if (!bIsUDisk) // 不是U盘T
    {
        // 如果父节点是“我的电脑”，不显示优盘
        
        if (m_aryShowDriver.GetSize() == 0)
        {
            return TRUE;
        }
        
        // 判断父节点是否是“我的电脑”
        if (IsMyComputer(pidlParent))
        {
            // 不显示U盘
            strDisplayName = GetVDisplayName(pParentFolder, pidl);
            for(i = 0; i < m_aryShowDriver.GetSize(); i++)
            {
                strLast4Char = "(" + m_aryShowDriver[i] + ":)";
                if (strDisplayName.Right(4) == strLast4Char)
                {
                    return FALSE;
                }
            }
        }
        
        return TRUE;
    }
    else // U盘
    {
        if (m_aryShowDriver.GetSize() == 0)
        {
            return FALSE;
        }
        
        // 如果父节点是“我的电脑”，只显示优盘
        
        // 判断父节点是否是“我的电脑”
        if (IsMyComputer(pidlParent))
        {
            strDisplayName = GetVDisplayName(pParentFolder, pidl);
            
            // 只显示U盘
            for(i = 0; i < m_aryShowDriver.GetSize(); i++)
            {
                strLast4Char = "(" + m_aryShowDriver[i] + ":)";
                if (strDisplayName.Right(4) == strLast4Char)
                {
                    return TRUE;
                }
            }
            
            return FALSE;
        }
        
        return TRUE;
    }
}

// 取得对象属性
ULONG CMMXShellGenelal::GetObjectAttributes(SShellFolderPtr pParentFolder, 
                                          CShellPidl& pidlAbs, ULONG nFlag)
{
    // 如果要判断哪个标志，必须先置这个标志
    ULONG ulAddr = nFlag; // 同时也是返回值
    
    LPCITEMIDLIST pidlRel = pidlAbs.GetLastChild();
    
    if (SUCCEEDED(pParentFolder->GetAttributesOf(1, &pidlRel,&ulAddr)))
    {	
        return ulAddr;
    }
    return 0;
}

CString CMMXShellGenelal::GetDisplayName(SShellFolderPtr pParentFolder, 
                                       CShellPidl& pidlAbs, UINT nFlags)
{
    CString sName;
    CShellString str;
    
    // get a relative pidl
    LPCITEMIDLIST pidlRel = pidlAbs.GetLastChild();
    
    if (nFlags & STCF_SHOWPATH)
    {
        // use an absolute or relative path, if possible
        sName = pidlAbs.GetPath();
        if (!sName.IsEmpty() && !(nFlags & STCF_SHOWFULLNAME))
            sName = sName.Right(sName.ReverseFind(_T('\\')));
    }
    if (sName.IsEmpty())
    {
        // use a global or contextual displayname
        DWORD uDisplayFlags = SHGDN_INFOLDER;
        if (nFlags & STCF_SHOWFULLNAME)
            uDisplayFlags = SHGDN_NORMAL;
        
        pParentFolder->GetDisplayNameOf(pidlRel, uDisplayFlags
            | SHGDN_INCLUDE_NONFILESYS, str.GetPointer(pidlRel));
        
        sName = str;	// copy to string
    }
    return sName;
}