mytreectrl.cpp

Visual C++网络通信编程实用案例精逊配套源码 光盘中存放的是书中涉及的所有实例的源代码和经过编译后的应用程序。所有程序均经过测试

#include "stdafx.h"
#include "MyTreeCtrl.h"
#include "InternetDownload.h"

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

/////////////////////////////////////////////////////////////////////////////
// CMyTreeCtrl

// Constructor
CMyTreeCtrl::CMyTreeCtrl()
{
}

// Destructor
CMyTreeCtrl::~CMyTreeCtrl()
{
}

BEGIN_MESSAGE_MAP(CMyTreeCtrl, CTreeCtrl)
	//{{AFX_MSG_MAP(CMyTreeCtrl)
		// NOTE - the ClassWizard will add and remove mapping macros here.
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

// Adds an item to the root of the tree
HTREEITEM CMyTreeCtrl::AddRootItem(LPCSTR lpszTitle, int nImage)
{
	return(InsertItem(lpszTitle,nImage,nImage,TVI_ROOT,TVI_LAST));
}

// Adds a child item under the specified parent node
HTREEITEM CMyTreeCtrl::AddChildItem(LPCTSTR lpszTitle, HTREEITEM htreeParent, int nImage)
{
	return(InsertItem(lpszTitle,nImage,nImage,htreeParent,TVI_LAST));
}

// Returns the proper multimedia node type based on the provide filename
CMyTreeCtrl::NODE_TYPE CMyTreeCtrl::GetMediaType(CString& strFileName)
{
	NODE_TYPE nodeType = CMyTreeCtrl::IMAGE_NODE;

	// Get the extension of the file
	CString strExt = CInternetDownload::SplitFileName(strFileName,CInternetDownload::EXT);
	strExt.MakeLower();

	CString strTest = strFileName;

	// Is it mail link?
 	if(strTest.Left(7) == "mailto:")      
 		nodeType = CMyTreeCtrl::EMAIL_NODE;
	else 
	{
		// Sound file?
		if(strExt == ".wav" || strExt == ".au" || strExt == ".mid" || strExt == ".midi")
			nodeType = CMyTreeCtrl::WAVE_NODE;
		else if(strExt == ".txt" || strExt == ".pdf")
			nodeType = CMyTreeCtrl::DOC_NODE;
		else if(strExt == ".class")
			nodeType = CMyTreeCtrl::JAVA_NODE;
	}

	// Return the type
	return(nodeType);
}

/////////////////////////////////////////////////////////////////////////////
// CMyTreeCtrl message handlers