BOOL CDirTreeCtrl::DisplayTree(LPCTSTR strRoot, BOOL bFiles)
{
DWORD dwStyle = GetStyle(); // read the windowstyle
if ( dwStyle & TVS_EDITLABELS )
{
// Don't allow the user to edit ItemLabels
ModifyStyle( TVS_EDITLABELS , 0 );
}
// Display the DirTree with the Rootname e.g. C:\
// if Rootname == NULL then Display all Drives on this PC
// First, we need the system-ImageList
DeleteAllItems();
if ( !GetSysImgList() )
return FALSE;
m_bFiles = bFiles; // if TRUE, Display Path- and Filenames
if ( strRoot == NULL || strRoot[0] == '\0' )
{
if ( !DisplayDrives() )
return FALSE;
m_strRoot = _T("");
}
else
{
m_strRoot = strRoot;
if ( m_strRoot.Right(1) != '\\' )
m_strRoot += _T("\\");
HTREEITEM hParent = AddItem( TVI_ROOT, m_strRoot );
DisplayPath( hParent, strRoot );
}
return TRUE;
}
void CFileAndFolder::OnViewRefresh()
{
// HTREEITEM hSelItem = GetSelectedItem();
// CString sItem = ItemToPath(hSelItem);
// BOOL bExpanded = (GetChildItem(hSelItem) != NULL);
DisplayDrives(TVI_ROOT, FALSE);
// if (sItem.GetLength())
// hSelItem = SetSelectedPath(sItem, bExpanded);
}
void FolderTree::DoExpand(HTREEITEM hItem)
{
FolderTreeItemInfo* pItem = (FolderTreeItemInfo*) GetItemData(hItem);
//Reset the drive node if the drive is empty or the media has changed
if (IsMediaValid(pItem->m_sFQPath))
{
//Delete the item if the path is no longer valid
if (IsFolder(pItem->m_sFQPath))
{
//Add the new items to the tree if it does not have any child items
//already
if (!GetChildItem(hItem))
DisplayPath(pItem->m_sFQPath, hItem);
}
else if (hItem == m_hMyComputerRoot)
{
//Display an hour glass as this may take some time
//CWaitCursor wait;
//Enumerate the local drive letters
auto shared = sp->getViewItems(sp->curProfile);
DisplayDrives(m_hMyComputerRoot, FALSE, shared);
}
else if ((hItem == m_hNetworkRoot) || (pItem->m_pNetResource))
{
//Display an hour glass as this may take some time
//CWaitCursor wait;
//Enumerate the network resources
EnumNetwork(hItem);
}
else
{
//Before we delete it see if we are the only child item
HTREEITEM hParent = GetParentItem(hItem);
//Delete the item
DeleteItem(hItem);
//Remove all the child items from the parent
SetHasPlusButton(hParent, false);
}
}
else
{
//Display an hour glass as this may take some time
//CWaitCursor wait;
//Collapse the drive node and remove all the child items from it
Expand(hItem, TVE_COLLAPSE);
DeleteChildren(hItem, true);
}
}
BOOL CDirTreeCtrl::DisplayTree(LPCTSTR strRoot)
{
SetItemHeight(20);
SetTextColor(RGB(0X0,0X0,0XFF));
DeleteAllItems();
SetDirTreeStyle();
//if ( !AttachImgList() )
// return FALSE;
DisplayDrives();
return TRUE;
}
void CFileAndFolder::OnViewCachedDrives() {
CWaitCursor c;
if (m_CachedDrives.GetCount() == 0) {
DisplayDrives(TVI_ROOT, FALSE);
return;
}
POSITION pos = m_CachedDrives.GetHeadPosition();
for (pos = m_CachedDrives.GetHeadPosition(); pos != NULL; ) {
CString string = m_CachedDrives.GetAt(pos);
CString drive,desc;
drive = string.Left(3);
desc = string.Mid(4);
InsertFileItem(desc,drive,TVI_ROOT);
m_CachedDrives.GetNext(pos);
}
}
void CTreeFileCtrl::OnViewRefresh()
{
SetRedraw(FALSE);
//Get the item which is currently selected
HTREEITEM hSelItem = GetSelectedItem();
CString sItem = ItemToPath(hSelItem);
BOOL bExpanded = (GetChildItem(hSelItem) != NULL);
//Display the folder items in the tree
if (m_sRootFolder.IsEmpty())
DisplayDrives(TVI_ROOT, FALSE);
else
DisplayPath(m_sRootFolder, TVI_ROOT, FALSE);
//Reselect the initially selected item
if (sItem.GetLength())
hSelItem = SetSelectedPath(sItem, bExpanded);
SetRedraw(TRUE);
}
bool CLocalTreeView::CreateRoot()
{
int iconIndex, openIconIndex;
wxString name = GetSpecialFolder(CSIDL_DESKTOP, iconIndex, openIconIndex);
if (name == _T(""))
{
name = _("Desktop");
iconIndex = openIconIndex = -1;
}
m_desktop = AddRoot(name, iconIndex, openIconIndex);
name = GetSpecialFolder(CSIDL_PERSONAL, iconIndex, openIconIndex);
if (name == _T(""))
{
name = _("My Documents");
iconIndex = openIconIndex = -1;
}
m_documents = AppendItem(m_desktop, name, iconIndex, openIconIndex);
name = GetSpecialFolder(CSIDL_DRIVES, iconIndex, openIconIndex);
if (name == _T(""))
{
name = _("My Computer");
iconIndex = openIconIndex = -1;
}
m_drives = AppendItem(m_desktop, name, iconIndex, openIconIndex);
DisplayDrives(m_drives);
Expand(m_desktop);
Expand(m_drives);
return true;
}
bool CLocalListView::DisplayDir(wxString dirname)
{
wxString focused;
std::list<wxString> selectedNames;
if (m_dir != dirname)
{
// Clear selection
int item = -1;
while (true)
{
item = GetNextItem(item, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED | wxLIST_STATE_FOCUSED);
if (item == -1)
break;
SetItemState(item, 0, wxLIST_STATE_SELECTED | wxLIST_STATE_FOCUSED);
}
focused = _T("..");
if (GetItemCount())
EnsureVisible(0);
m_dir = dirname;
}
else
{
// Remember which items were selected
selectedNames = RememberSelectedItems(focused);
}
const int oldItemCount = m_indexMapping.size();
m_fileData.clear();
m_indexMapping.clear();
m_hasParent = CState::LocalDirHasParent(dirname);
if (m_hasParent)
{
t_fileData data;
data.dir = true;
data.icon = -2;
data.name = _T("..");
data.size = -1;
data.hasTime = 0;
m_fileData.push_back(data);
m_indexMapping.push_back(0);
}
#ifdef __WXMSW__
if (dirname == _T("\\"))
{
DisplayDrives();
}
else if (dirname.Left(2) == _T("\\\\"))
{
int pos = dirname.Mid(2).Find('\\');
if (pos != -1 && pos + 3 != (int)dirname.Len())
goto regular_dir;
// UNC path without shares
DisplayShares(dirname);
}
else
#endif
{
#ifdef __WXMSW__
regular_dir:
#endif
CFilterDialog filter;
wxDir dir(dirname);
if (!dir.IsOpened())
{
SetItemCount(1);
return false;
}
wxString file;
bool found = dir.GetFirst(&file);
int num = m_fileData.size();
while (found)
{
if (file == _T(""))
{
wxGetApp().DisplayEncodingWarning();
found = dir.GetNext(&file);
continue;
}
t_fileData data;
data.dir = wxFileName::DirExists(dirname + file);
data.icon = -2;
data.name = file;
wxStructStat buf;
int result;
result = wxStat(dirname + file, &buf);
if (!result)
{
data.hasTime = true;
data.lastModified = wxDateTime(buf.st_mtime);
}
else
data.hasTime = false;
if (data.dir)
data.size = -1;
else
data.size = result ? -1 : buf.st_size;
m_fileData.push_back(data);
if (!filter.FilenameFiltered(data.name, data.dir, data.size, true))
m_indexMapping.push_back(num);
num++;
found = dir.GetNext(&file);
}
}
if (m_dropTarget != -1)
{
t_fileData* data = GetData(m_dropTarget);
if (!data || !data->dir)
{
SetItemState(m_dropTarget, 0, wxLIST_STATE_DROPHILITED);
m_dropTarget = -1;
}
}
const int count = m_indexMapping.size();
if (oldItemCount != count)
SetItemCount(count);
SortList();
ReselectItems(selectedNames, focused);
Refresh();
return true;
}
bool CLocalListView::DisplayDir(wxString dirname)
{
CancelLabelEdit();
wxString focused;
std::list<wxString> selectedNames;
bool ensureVisible = false;
if (m_dir != dirname)
{
ResetSearchPrefix();
if (IsComparing())
ExitComparisonMode();
ClearSelection();
focused = m_pState->GetPreviouslyVisitedLocalSubdir();
ensureVisible = !focused.IsEmpty();
if (focused.IsEmpty())
focused = _T("..");
if (GetItemCount())
EnsureVisible(0);
m_dir = dirname;
}
else
{
// Remember which items were selected
selectedNames = RememberSelectedItems(focused);
}
if (m_pFilelistStatusBar)
m_pFilelistStatusBar->UnselectAll();
const int oldItemCount = m_indexMapping.size();
m_fileData.clear();
m_indexMapping.clear();
m_hasParent = CLocalPath(dirname).HasLogicalParent();
if (m_hasParent)
{
CLocalFileData data;
data.flags = normal;
data.dir = true;
data.icon = -2;
data.name = _T("..");
#ifdef __WXMSW__
data.label = _T("..");
#endif
data.size = -1;
m_fileData.push_back(data);
m_indexMapping.push_back(0);
}
#ifdef __WXMSW__
if (dirname == _T("\\"))
{
DisplayDrives();
}
else if (dirname.Left(2) == _T("\\\\"))
{
int pos = dirname.Mid(2).Find('\\');
if (pos != -1 && pos + 3 != (int)dirname.Len())
goto regular_dir;
// UNC path without shares
DisplayShares(dirname);
}
else
#endif
{
#ifdef __WXMSW__
regular_dir:
#endif
CFilterManager filter;
CLocalFileSystem local_filesystem;
if (!local_filesystem.BeginFindFiles(dirname, false))
{
SetItemCount(1);
return false;
}
wxLongLong totalSize;
int unknown_sizes = 0;
int totalFileCount = 0;
int totalDirCount = 0;
int hidden = 0;
int num = m_fileData.size();
CLocalFileData data;
data.flags = normal;
data.icon = -2;
bool wasLink;
while (local_filesystem.GetNextFile(data.name, wasLink, data.dir, &data.size, &data.lastModified, &data.attributes))
{
if (data.name.IsEmpty())
{
wxGetApp().DisplayEncodingWarning();
continue;
}
#ifdef __WXMSW__
data.label = data.name;
#endif
m_fileData.push_back(data);
if (!filter.FilenameFiltered(data.name, dirname, data.dir, data.size, true, data.attributes, data.lastModified))
{
if (data.dir)
totalDirCount++;
else
{
if (data.size != -1)
totalSize += data.size;
else
unknown_sizes++;
totalFileCount++;
}
m_indexMapping.push_back(num);
}
else
hidden++;
num++;
}
if (m_pFilelistStatusBar)
m_pFilelistStatusBar->SetDirectoryContents(totalFileCount, totalDirCount, totalSize, unknown_sizes, hidden);
}
if (m_dropTarget != -1)
{
CLocalFileData* data = GetData(m_dropTarget);
if (!data || !data->dir)
{
SetItemState(m_dropTarget, 0, wxLIST_STATE_DROPHILITED);
m_dropTarget = -1;
}
}
const int count = m_indexMapping.size();
if (oldItemCount != count)
SetItemCount(count);
SortList(-1, -1, false);
if (IsComparing())
{
m_originalIndexMapping.clear();
RefreshComparison();
}
ReselectItems(selectedNames, focused, ensureVisible);
RefreshListOnly();
return true;
}
void FolderTree::Refresh() {
//Just in case this will take some time
//CWaitCursor wait;
SetRedraw(FALSE);
//Get the item which is currently selected
HTREEITEM hSelItem = GetSelectedItem();
tstring sItem;
bool bExpanded = false;
if (hSelItem)
{
sItem = ItemToPath(hSelItem);
bExpanded = IsExpanded(hSelItem);
}
theSharedEnumerator.Refresh();
//Remove all nodes that currently exist
Clear();
//Display the folder items in the tree
if (m_sRootFolder.empty()) {
//Should we insert a "My Computer" node
if (m_bShowMyComputer) {
FolderTreeItemInfo* pItem = new FolderTreeItemInfo;
pItem->m_bNetworkNode = false;
int nIcon = 0;
int nSelIcon = 0;
//Get the localized name and correct icons for "My Computer"
LPITEMIDLIST lpMCPidl;
if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_DRIVES, &lpMCPidl))) {
SHFILEINFO sfi;
if (SHGetFileInfo((LPCTSTR)lpMCPidl, 0, &sfi, sizeof(sfi), SHGFI_PIDL | SHGFI_DISPLAYNAME))
pItem->m_sRelativePath = sfi.szDisplayName;
nIcon = GetIconIndex(lpMCPidl);
nSelIcon = GetSelIconIndex(lpMCPidl);
//Free up the pidl now that we are finished with it
//ASSERT(m_pMalloc);
m_pMalloc->Free(lpMCPidl);
m_pMalloc->Release();
}
//Add it to the tree control
m_hMyComputerRoot = InsertFileItem(TVI_ROOT, pItem, false, nIcon, nSelIcon, false);
SetHasSharedChildren(m_hMyComputerRoot);
}
//Display all the drives
if (!m_bShowMyComputer)
DisplayDrives(TVI_ROOT, false);
//Display homegroup
/*if (m_bDisplayNetwork) {
FolderTreeItemInfo* pItem = new FolderTreeItemInfo;
//pItem-> = true;
int nIcon = 0;
int nSelIcon = 0;
//Get the localized name and correct icons for "Network Neighborhood"
LPITEMIDLIST lpNNPidl;
//SHGetKnownFolderIDList(FOLDERID_HomeGroup, 0, NULL, &lpNNPidl);
if (SUCCEEDED(SHGetKnownFolderIDList(FOLDERID_HomeGroup, 0, NULL, &lpNNPidl))) {
SHFILEINFO sfi;
if (SHGetFileInfo((LPCTSTR)lpNNPidl, 0, &sfi, sizeof(sfi), SHGFI_PIDL | SHGFI_DISPLAYNAME))
pItem->m_sRelativePath = sfi.szDisplayName;
nIcon = GetIconIndex(lpNNPidl);
nSelIcon = GetSelIconIndex(lpNNPidl);
//Free up the pidl now that we are finished with it
//ASSERT(m_pMalloc);
m_pMalloc->Free(lpNNPidl);
m_pMalloc->Release();
}
//Add it to the tree control
m_hHomegroupRoot = InsertFileItem(TVI_ROOT, pItem, false, nIcon, nSelIcon, false, shared);
SetHasSharedChildren(m_hHomegroupRoot, shared);
//checkRemovedDirs(Util::emptyStringT, TVI_ROOT, shared);
}*/
//Also add network neighborhood if requested to do so
if (m_bDisplayNetwork) {
FolderTreeItemInfo* pItem = new FolderTreeItemInfo;
pItem->m_bNetworkNode = true;
int nIcon = 0;
int nSelIcon = 0;
//Get the localized name and correct icons for "Network Neighborhood"
LPITEMIDLIST lpNNPidl;
if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_NETWORK, &lpNNPidl))) {
SHFILEINFO sfi;
if (SHGetFileInfo((LPCTSTR)lpNNPidl, 0, &sfi, sizeof(sfi), SHGFI_PIDL | SHGFI_DISPLAYNAME))
pItem->m_sRelativePath = sfi.szDisplayName;
nIcon = GetIconIndex(lpNNPidl);
nSelIcon = GetSelIconIndex(lpNNPidl);
//Free up the pidl now that we are finished with it
//ASSERT(m_pMalloc);
m_pMalloc->Free(lpNNPidl);
m_pMalloc->Release();
}
//Add it to the tree control
m_hNetworkRoot = InsertFileItem(TVI_ROOT, pItem, false, nIcon, nSelIcon, false);
SetHasSharedChildren(m_hNetworkRoot);
//checkRemovedDirs(Util::emptyStringT, TVI_ROOT, shared);
}
} else {
DisplayPath(m_sRootFolder, TVI_ROOT, false);
}
//Reselect the initially selected item
if (hSelItem)
SetSelectedPath(sItem, bExpanded);
//Turn back on the redraw flag
SetRedraw(true);
}
