extracticon.cpp

Drive.dll is a dll that creates a drive to system directoryor drive similar to existing drives such

#include <windows.h>
#include <shlobj.h>

#include "EnumIDL.h"
#include "ExtractIcon.h"
#include "ShellView.h"
#include "resource.h"

extern UINT			g_uiRefThisDll;
extern HINSTANCE	g_hInstance;
extern HIMAGELIST	g_hImageListSmall;
extern HIMAGELIST	g_hImageListLarge;

CExtractIcon::CExtractIcon(LPCITEMIDLIST pidl)
{
	m_pPidlMgr = new CPidlMgr();

	m_pidl = m_pPidlMgr->Copy(pidl);

	m_uiRefCount = 1;

	g_uiRefThisDll++;
}

CExtractIcon::~CExtractIcon()
{
	if(m_pidl)
	{
		m_pPidlMgr->Delete(m_pidl);
		m_pidl = NULL;
	}

	if(m_pPidlMgr)
	{
		delete m_pPidlMgr;
	}

	g_uiRefThisDll--;
}

///////////////////////////////////////////////////////////
// IUnknown Implementation
//
STDMETHODIMP CExtractIcon::QueryInterface(REFIID riid, LPVOID *ppReturn)
{
	*ppReturn = NULL;

	//IUnknown
	if(IsEqualIID(riid, IID_IUnknown))
	{
		*ppReturn = this;
	}
	//IExtractIcon
	else if(IsEqualIID(riid, IID_IExtractIcon))
	{
		*ppReturn = (IExtractIcon*)this;
	}

	if(*ppReturn)
	{
		(*(LPUNKNOWN*)ppReturn)->AddRef();
		return S_OK;
	}

	return E_NOINTERFACE;
}                                             

STDMETHODIMP_(DWORD) CExtractIcon::AddRef()
{
	return ++m_uiRefCount;
}

STDMETHODIMP_(DWORD) CExtractIcon::Release()
{
	if(--m_uiRefCount == 0)
	{
		delete this;
		return 0;
	}

	return m_uiRefCount;
}

///////////////////////////////////////////////////////////
// IExtractIcon Implementation
// TODO : Modify to suit your needs
STDMETHODIMP CExtractIcon::GetIconLocation(  UINT uFlags, 
                                             LPTSTR szIconFile, 
                                             UINT cchMax, 
                                             LPINT piIndex, 
                                             LPUINT puFlags)
{
	return E_NOTIMPL;
}

// TODO : Modify to suit your needs
STDMETHODIMP CExtractIcon::Extract( LPCTSTR pszFile, UINT nIconIndex, 
									HICON *phiconLarge, HICON *phiconSmall, 
									UINT nIconSize)
{
	return E_NOTIMPL;
}

//EOF