dirlistview.cpp

这些源代码

		strDir = OldDir;
//
//	Add a backslash only if the directory name doesn't
//	already end with a backslash (as at the root directory)
		if (strDir.GetAt (strDir.GetLength() - 1) != '\\')
			strDir += '\\';
		strDir += pItem->fdFile.cFileName;
	}
//
//	Save the new directory
	m_DirectoryName = strDir;
//	And change to it
	SetCurrentDirectory ((LPCSTR) m_DirectoryName);
//	Show it in the tree control
	if (m_bInitialized == true)
	{
		CDummyDoc *doc = GetDocument();
		pos = doc->GetFirstViewPosition ();
		while (pos != NULL)
		{
			CView *pView = doc->GetNextView (pos);
			if (pView->IsKindOf(RUNTIME_CLASS(CDirTreeView)))
			{
				((CDirTreeView *)(pView))->SetDirectoryName (m_DirectoryName, true);
			}
		}
	}
//	And show it in the list control
	ListDirectory ();
}

void CDirListView::OnColumnclick(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
	// TODO: Add your control notification handler code here
	GetListCtrl().SortItems (CompareItems, pNMListView->iSubItem);
	
	*pResult = 0;
}

void CDirListView::OnKillFocus(CWnd* pNewWnd) 
{
	CListView::OnKillFocus(pNewWnd);
	
	// TODO: Add your message handler code here
	
}

void CDirListView::OnSetFocus(CWnd* pOldWnd) 
{
	CListView::OnSetFocus(pOldWnd);
	
	// TODO: Add your message handler code here
	
}

CString & CDirListView::GetCurrentDirectory()
{
static CString strCurrentPath;

	strCurrentPath.Empty ();
	int nBytes = ::GetCurrentDirectory (0, NULL);
	char *szDir = new char [nBytes + 1];
	::GetCurrentDirectory (nBytes + 1, szDir);
	strCurrentPath = szDir;
	delete [] szDir;
	return (strCurrentPath);
}

int CALLBACK CDirListView::CompareItems(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
{
int	nResult;
time_t time1, time2;

	ITEMINFO *pItem1 = (ITEMINFO *) lParam1;
	ITEMINFO *pItem2 = (ITEMINFO *) lParam2;
	switch (lParamSort)
	{
		case 0:				// Sort by name
//
//	The following tests force the UP-DIR item to always
//	land on top of the list.
//
			if (!strcmp ((LPCSTR) pItem1->fdFile.cFileName, ".."))
				return (-1);
			if (!strcmp ((LPCSTR) pItem2->fdFile.cFileName, ".."))
				return (1);
//
//	The following tests cause SUB-DIR items to always
//	stay above normal file items.
//
			if ((pItem1->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
				!(pItem2->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
				return (-1);
			if (!(pItem1->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
				(pItem2->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
				return (1);
//
//	The remaining tests compare directory/directory or
//	file/file names.
//
			nResult = strcmpi ((LPCSTR) pItem1->fdFile.cFileName,
							  (LPCSTR) pItem2->fdFile.cFileName);
			break;
		case 1:				// Sort by file type
//
//	If a directory is involved, sort by name.
//
			if ((pItem1->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
				(pItem2->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
				return (CompareItems (lParam1, lParam2, 0));
			nResult = (int) (strcmp ((LPCSTR) pItem1->strFileType, (LPCSTR) pItem2->strFileType));
			break;
		case 2:				// Sort by size
//
//	If a directory is involved, sort by name.
//
			if ((pItem1->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
				(pItem2->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
				return (CompareItems (lParam1, lParam2, 0));
			nResult = (int) (pItem1->fdFile.nFileSizeLow - pItem2->fdFile.nFileSizeLow);
			break;
		case 3:				// Sort by creation time
//
//	If a directory is involved, sort by name.
//
			if ((pItem1->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
				(pItem2->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
				return (CompareItems (lParam1, lParam2, 0));
			time1 = FileTimeToUnixTime (&pItem1->fdFile.ftCreationTime);
			time2 = FileTimeToUnixTime (&pItem2->fdFile.ftCreationTime);
			nResult = (int) (time1 - time2);
			break;
		case 4:				// Sort by modified time
//
//	If a directory is involved, sort by name.
//
			if ((pItem1->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
				(pItem2->fdFile.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
				return (CompareItems (lParam1, lParam2, 0));
			time1 = FileTimeToUnixTime (&pItem1->fdFile.ftLastWriteTime);
			time2 = FileTimeToUnixTime (&pItem2->fdFile.ftLastWriteTime);
			nResult = (int) (time1 - time2);
			break;
	}
	return (nResult);
}

void CDirListView::ListDirectory()
{
	ListDirectory (m_DirectoryName);
}

void CDirListView::ListDirectory(CString &strDir)
{
CListCtrl&	cc = GetListCtrl();
CString		OldDir, FindSpec;
WIN32_FIND_DATA	fData;
HANDLE		hFind;
CString		FileName;
int			sub;

	m_DirectoryName = strDir;
	OldDir = m_DirectoryName;
	if (strDir.IsEmpty())
		strDir = OldDir;

	FindSpec = strDir;
	sub = FindSpec.GetLength();
	if (sub)
	{
		if (FindSpec.GetAt (sub - 1) == '\\')
			FindSpec += "*.*";
		else
			FindSpec += "\\*.*";
	}
	FreeItemMemory();
	hFind = FindFirstFile(FindSpec, &fData);
	if (hFind == INVALID_HANDLE_VALUE)
		return;

	int i = 0;
	do
	{
		if (!strcmp (fData.cFileName, "."))
			continue;
		if (!strcmp (fData.cFileName, ".."))
			continue;
		if (!AddItem(i, &fData))
			break;
		++i;
	} while (FindNextFile(hFind, &fData));
	FindClose (hFind);
	cc.SortItems (CompareItems, 0);
	cc.RedrawItems (0, cc.GetItemCount() - 1);
}

void CDirListView::FreeItemMemory()
{
CListCtrl&	cc = GetListCtrl();

	int nCount = cc.GetItemCount ();
	for (int i = 0; i < nCount; ++i)
		delete (ITEMINFO *) cc.GetItemData (i);
	cc.DeleteAllItems ();
}

bool CDirListView::AddItem(int nIndex, WIN32_FIND_DATA *pfd)
{
ITEMINFO	*pItem;
LV_ITEM		lvitem;

	memset ((char *) &lvitem, '\0', sizeof (LV_ITEM));
	try
	{
		pItem = new ITEMINFO;
	}
	catch (CMemoryException *e)
	{
		e->Delete ();
		return (false);
	}
	memcpy (&pItem->fdFile, pfd, sizeof (WIN32_FIND_DATA));
	GetFileType (pItem->fdFile, pItem->strFileType);
	UINT uiOpen = 0;
	if (!strcmp (pItem->fdFile.cFileName, ".."))
		uiOpen = SHGFI_OPENICON;
	pItem->nIcon = GetIconIndex (pItem->fdFile, uiOpen);

#undef STATEMASK
#define		STATEMASK	0x0100 | LVIS_CUT|LVIS_DROPHILITED|LVIS_FOCUSED|LVIS_SELECTED

	lvitem.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM | LVIF_STATE;
	lvitem.stateMask = STATEMASK; // | INDEXTOSTATEIMAGEMASK(15);
	lvitem.iItem = nIndex;
	lvitem.iSubItem = 0;
	lvitem.iImage = I_IMAGECALLBACK;
	lvitem.pszText = LPSTR_TEXTCALLBACK;
	lvitem.lParam = (LPARAM) pItem;

	if (GetListCtrl ().InsertItem (&lvitem) < 0)
		return (false);
	return (true);
}

int CDirListView::GetIconIndex(WIN32_FIND_DATA &fd, UINT uiOpen)
{
	SHFILEINFO sfi;
	memset(&sfi, 0, sizeof(sfi));
	CString strFilePath = m_DirectoryName;
	if (strFilePath.GetAt (strFilePath.GetLength() -1) != '\\')
		strFilePath += '\\';
	strFilePath += fd.cFileName;
	if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
	{
		SHGetFileInfo (strFilePath, FILE_ATTRIBUTE_DIRECTORY,
			&sfi, sizeof(sfi), 
			SHGFI_SMALLICON | SHGFI_SYSICONINDEX |
            SHGFI_USEFILEATTRIBUTES | uiOpen);
		return (sfi.iIcon);
	}
	else
	{
		SHGetFileInfo (strFilePath, FILE_ATTRIBUTE_NORMAL, 
			&sfi, sizeof(sfi), 
			SHGFI_SMALLICON | SHGFI_SYSICONINDEX |
            SHGFI_USEFILEATTRIBUTES);
		return (sfi.iIcon);
	}
	return (-1);
}

int CDirListView::GetFileType(WIN32_FIND_DATA &fd, CString &strFileType)
{

	SHFILEINFO sfi;
	memset(&sfi, 0, sizeof(sfi));
	CString strFilePath = m_DirectoryName;
	strFilePath += '\\';
	strFilePath += fd.cFileName;
	SHGetFileInfo (
		strFilePath, 
		FILE_ATTRIBUTE_NORMAL,
		&sfi, 
		sizeof(sfi), 
		SHGFI_TYPENAME);
	strFileType = sfi.szTypeName;
	return (0);
}

void CDirListView::SetDirectoryName(char *dir)
{

}