void CShellMgr::GetNormalAndSelectedIcons(LPITEMIDLIST lpifq, LPTVITEM lptvitem)
{
int nRet = lptvitem->iImage = GetIconIndex(lpifq, SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON);
ATLASSERT(nRet >= 0);
nRet = lptvitem->iSelectedImage = GetIconIndex(lpifq, SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_OPENICON);
ATLASSERT(nRet >= 0);
}
CLocalTreeView::CLocalTreeView(wxWindow* parent, wxWindowID id, CState *pState, CQueueView *pQueueView)
: wxTreeCtrlEx(parent, id, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL | wxTR_EDIT_LABELS | wxTR_LINES_AT_ROOT | wxTR_HAS_BUTTONS | wxNO_BORDER),
CSystemImageList(16),
CStateEventHandler(pState),
m_pQueueView(pQueueView)
{
wxGetApp().AddStartupProfileRecord(_T("CLocalTreeView::CLocalTreeView"));
#ifdef __WXMAC__
SetFont(wxSystemSettings::GetFont(wxSYS_DEFAULT_GUI_FONT));
#endif
pState->RegisterHandler(this, STATECHANGE_LOCAL_DIR);
pState->RegisterHandler(this, STATECHANGE_APPLYFILTER);
SetImageList(GetSystemImageList());
#ifdef __WXMSW__
m_pVolumeEnumeratorThread = 0;
CreateRoot();
#else
wxTreeItemId root = AddRoot(_T("/"));
SetItemImage(root, GetIconIndex(dir), wxTreeItemIcon_Normal);
SetItemImage(root, GetIconIndex(opened_dir), wxTreeItemIcon_Selected);
SetItemImage(root, GetIconIndex(dir), wxTreeItemIcon_Expanded);
SetItemImage(root, GetIconIndex(opened_dir), wxTreeItemIcon_SelectedExpanded);
SetDir(_T("/"));
#endif
SetDropTarget(new CLocalTreeViewDropTarget(this));
}
BOOL CXTPPopupItem::SetIcons(HBITMAP hBitmap, COLORREF clrTransparent, int itemIcon, BOOL bAlphaBitmap)
{
BITMAP bmpInfo;
ZeroMemory(&bmpInfo, sizeof(BITMAP));
::GetObject(hBitmap, sizeof(BITMAP), &bmpInfo);
int nCount = (itemIcon & xtpPopupItemIconNormal ? 1 : 0) +
(itemIcon & xtpPopupItemIconSelected ? 1 : 0) +
(itemIcon & xtpPopupItemIconPressed ? 1 : 0);
if (nCount == 0)
return FALSE;
int nIndex = 0;
CBitmap bmp;
bmp.Attach(hBitmap);
COLORREF clr = m_pControl->GetImageManager()->GetMaskColor();
m_pControl->GetImageManager()->SetMaskColor(clrTransparent);
if ((itemIcon & xtpPopupItemIconNormal) != 0)
{
UINT nIDs[] = {IMAGE_PLACEHOLDER, IMAGE_PLACEHOLDER, IMAGE_PLACEHOLDER};
nIDs[nIndex++] = GetIconIndex();
m_pControl->GetImageManager()->SetIcons(bmp, nIDs, nCount, CSize(bmpInfo.bmWidth / nCount, bmpInfo.bmHeight), xtpImageNormal, bAlphaBitmap);
}
if ((itemIcon & xtpPopupItemIconSelected) != 0)
{
UINT nIDs[] = {IMAGE_PLACEHOLDER, IMAGE_PLACEHOLDER, IMAGE_PLACEHOLDER};
nIDs[nIndex++] = GetIconIndex();
m_pControl->GetImageManager()->SetIcons(bmp, nIDs, nCount, CSize(bmpInfo.bmWidth / nCount, bmpInfo.bmHeight), xtpImageHot, bAlphaBitmap);
}
if ((itemIcon & xtpPopupItemIconPressed) != 0)
{
UINT nIDs[] = {IMAGE_PLACEHOLDER, IMAGE_PLACEHOLDER, IMAGE_PLACEHOLDER};
nIDs[nIndex++] = GetIconIndex();
m_pControl->GetImageManager()->SetIcons(bmp, nIDs, nCount, CSize(bmpInfo.bmWidth / nCount, bmpInfo.bmHeight), xtpImageChecked, bAlphaBitmap);
}
m_pControl->GetImageManager()->SetMaskColor(clr);
bmp.Detach();
return TRUE;
}
HTREEITEM CFileAndFolder::InsertFileItem(const CString& sFile,
const CString& sPath, HTREEITEM hParent)
{
//Retreive the icon indexes for the specified file/folder
int nIconIndex = GetIconIndex(sPath);
int nSelIconIndex = GetSelIconIndex(sPath);
if (nIconIndex == -1 || nSelIconIndex == -1)
{
TRACE(_T("Failed in call to SHGetFileInfo for %s, GetLastError:%d\n"), sPath, ::GetLastError());
return NULL;
}
//Add the actual item
CString sTemp(sFile);
TV_INSERTSTRUCT tvis;
ZeroMemory(&tvis, sizeof(TV_INSERTSTRUCT));
tvis.hParent = hParent;
tvis.hInsertAfter = TVI_LAST;
tvis.item.mask = TVIF_CHILDREN | TVIF_IMAGE | TVIF_SELECTEDIMAGE
| TVIF_TEXT;
tvis.item.pszText = sTemp.GetBuffer(sTemp.GetLength());
tvis.item.cchTextMax = sTemp.GetLength();
tvis.item.iImage = nIconIndex;
tvis.item.iSelectedImage = nSelIconIndex;
tvis.item.cChildren = HasGotSubEntries(sPath);
HTREEITEM hItem = m_Tree.InsertItem(&tvis);
sTemp.ReleaseBuffer();
return hItem;
}
CSearchDialogFileList::CSearchDialogFileList(CSearchDialog* pParent, CState* pState, CQueueView* pQueue)
: CFileListCtrl<CSearchFileData>(pParent, pState, pQueue, true),
m_searchDialog(pParent)
{
m_hasParent = false;
SetImageList(GetSystemImageList(), wxIMAGE_LIST_SMALL);
m_dirIcon = GetIconIndex(dir);
InitSort(OPTION_SEARCH_SORTORDER);
InitHeaderSortImageList();
const unsigned long widths[7] = { 130, 130, 75, 80, 120, 80, 80 };
AddColumn(_("Filename"), wxLIST_FORMAT_LEFT, widths[0]);
AddColumn(_("Path"), wxLIST_FORMAT_LEFT, widths[1]);
AddColumn(_("Filesize"), wxLIST_FORMAT_RIGHT, widths[2]);
AddColumn(_("Filetype"), wxLIST_FORMAT_LEFT, widths[3]);
AddColumn(_("Last modified"), wxLIST_FORMAT_LEFT, widths[4]);
AddColumn(_("Permissions"), wxLIST_FORMAT_LEFT, widths[5]);
AddColumn(_("Owner/Group"), wxLIST_FORMAT_LEFT, widths[6]);
LoadColumnSettings(OPTION_SEARCH_COLUMN_WIDTHS, OPTION_SEARCH_COLUMN_SHOWN, OPTION_SEARCH_COLUMN_ORDER);
}
bool CLocalTreeView::DisplayDrives(wxTreeItemId parent)
{
wxGetApp().AddStartupProfileRecord(_T("CLocalTreeView::DisplayDrives"));
std::list<wxString> drives = CVolumeDescriptionEnumeratorThread::GetDrives();
m_pVolumeEnumeratorThread = new CVolumeDescriptionEnumeratorThread(this);
if (m_pVolumeEnumeratorThread->Failed()) {
delete m_pVolumeEnumeratorThread;
m_pVolumeEnumeratorThread = 0;
}
for( std::list<wxString>::const_iterator it = drives.begin(); it != drives.end(); ++it ) {
wxString drive = *it;
if (drive.Right(1) == _T("\\"))
drive = drive.RemoveLast();
wxGetApp().AddStartupProfileRecord(wxString::Format(_T("CLocalTreeView::DisplayDrives adding drive %s"), drive.c_str()));
wxTreeItemId item = AppendItem(parent, drive, GetIconIndex(iconType::dir, drive));
AppendItem(item, _T(""));
SortChildren(parent);
}
wxGetApp().AddStartupProfileRecord(_T("CLocalTreeView::DisplayDrives adding drives done"));
return true;
}
void CRemoteTreeView::CreateImageList()
{
wxSize s = CThemeProvider::GetIconSize(iconSizeSmall);
m_pImageList = new wxImageList(s.x, s.y, true, 4);
// Normal directory
int index = GetIconIndex(iconType::dir, _T("{78013B9C-3532-4fe1-A418-5CD1955127CC}"), false);
m_pImageList->Add(CreateIcon(index));
m_pImageList->Add(CreateIcon(index, _T("ART_UNKNOWN")));
// Opened directory
index = GetIconIndex(iconType::opened_dir, _T("{78013B9C-3532-4fe1-A418-5CD1955127CC}"), false);
m_pImageList->Add(CreateIcon(index));
m_pImageList->Add(CreateIcon(index, _T("ART_UNKNOWN")));
SetImageList(m_pImageList);
}
BOOL CXTPPopupItem::SetIcon(HICON hIcon, XTPPopupItemIcon itemIcon)
{
if (!hIcon)
return FALSE;
return m_pControl->GetImageManager()->SetIcon(hIcon, GetIconIndex(), 0,
itemIcon == xtpPopupItemIconNormal ? xtpImageNormal: itemIcon == xtpPopupItemIconSelected ? xtpImageHot : xtpImageChecked);
}
void CreateInitialPages(wxBookCtrlBase *parent)
{
// Create and add some panels to the notebook
wxPanel *panel = CreateRadioButtonsPage(parent);
parent->AddPage( panel, RADIOBUTTONS_PAGE_NAME, false, GetIconIndex(parent) );
panel = CreateVetoPage(parent);
parent->AddPage( panel, VETO_PAGE_NAME, false, GetIconIndex(parent) );
panel = CreateBigButtonPage(parent);
parent->AddPage( panel, MAXIMIZED_BUTTON_PAGE_NAME, false, GetIconIndex(parent) );
panel = CreateInsertPage(parent);
parent->InsertPage( 0, panel, I_WAS_INSERTED_PAGE_NAME, false, GetIconIndex(parent) );
parent->SetSelection(1);
}
void CLocalTreeView::OnSelectionChanging(wxTreeEvent& event)
{
// On-demand open icon for selected items
wxTreeItemId item = event.GetItem();
if (GetItemImage(item, wxTreeItemIcon_Selected) == -1)
{
int icon = GetIconIndex(opened_dir, GetDirFromItem(item));
SetItemImage(item, icon, wxTreeItemIcon_Selected);
SetItemImage(item, icon, wxTreeItemIcon_SelectedExpanded);
}
}
// CFileOpenDlg::EnumDirs
//
// Enumerates all subdirectories. Recurses depth-first
//
int CFileOpenDlg::EnumDirs(LPCTSTR pszPath, LPCTSTR pszFilter, HTREEITEM hItemParent)
{
WIN32_FIND_DATA fd;
HANDLE hFind;
BOOL bFind;
HTREEITEM hItem = TVI_LAST;
CString strSearch(pszPath),
strBase(pszPath);
int nCount = 0;
strSearch += pszFilter;
hFind = FindFirstFile(strSearch, &fd);
bFind = (hFind != INVALID_HANDLE_VALUE);
while(bFind)
{
if((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
!(fd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN))
{
TVINSERTSTRUCT tvi;
CString strSub(strBase);
int iIcon;
strSub += fd.cFileName;
iIcon = GetIconIndex(strSub);
tvi.hParent = hItemParent;
tvi.hInsertAfter = TVI_LAST;
tvi.item.mask = TVIF_CHILDREN | TVIF_IMAGE | TVIF_SELECTEDIMAGE |
TVIF_TEXT | TVIF_PARAM;
tvi.item.pszText = fd.cFileName;
tvi.item.iImage = iIcon;
tvi.item.iSelectedImage = iIcon;
tvi.item.cChildren = I_CHILDRENCALLBACK;
tvi.item.lParam = -1;
hItem = m_treeFolder.InsertItem(&tvi);
++nCount;
/*
strSub += _T("\\");
if(hItem)
{
m_treeFolder.SetItemData(hItem, 1);
EnumDirs(strSub, pszFilter, hItem);
}*/
}
bFind = FindNextFile(hFind, &fd);
}
return nCount;
}
CLocalTreeView::CLocalTreeView(wxWindow* parent, wxWindowID id, CState *pState, CQueueView *pQueueView)
: wxTreeCtrl(parent, id, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL | wxTR_EDIT_LABELS | wxTR_LINES_AT_ROOT | wxTR_HAS_BUTTONS | wxNO_BORDER),
CSystemImageList(16),
CStateEventHandler(pState, STATECHANGE_LOCAL_DIR | STATECHANGE_APPLYFILTER),
m_pQueueView(pQueueView)
{
m_setSelection = false;
SetImageList(GetSystemImageList());
#ifdef __WXMSW__
m_pVolumeEnumeratorThread = 0;
CreateRoot();
#else
AddRoot(_T("/"), GetIconIndex(dir), GetIconIndex(opened_dir));
SetDir(_T("/"));
#endif
SetDropTarget(new CLocalTreeViewDropTarget(this));
}
void CTreeFileCtrl::OnEndlabeledit(NMHDR* pNMHDR, LRESULT* pResult)
{
TV_DISPINFO* pDispInfo = (TV_DISPINFO*)pNMHDR;
if (pDispInfo->item.pszText)
{
SHFILEOPSTRUCT shfo;
ZeroMemory(&shfo, sizeof(SHFILEOPSTRUCT));
shfo.hwnd = AfxGetMainWnd()->GetSafeHwnd();
shfo.wFunc = FO_RENAME;
shfo.fFlags = FOF_ALLOWUNDO;
CString sFrom = ItemToPath(pDispInfo->item.hItem);
int nFromLength = sFrom.GetLength();
TCHAR* pszFrom = new TCHAR[nFromLength + 2];
_tcscpy(pszFrom, sFrom);
pszFrom[nFromLength+1] = _T('\0');
shfo.pFrom = pszFrom;
HTREEITEM hParent = GetParentItem(pDispInfo->item.hItem);
CString sParent = ItemToPath(hParent);
CString sTo;
if (IsDrive(sParent))
sTo = sParent + pDispInfo->item.pszText;
else
sTo = sParent + _T("\\") + pDispInfo->item.pszText;
int nToLength = _tcslen(sTo);
TCHAR* pszTo = new TCHAR[nToLength + 2];
_tcscpy(pszTo, sTo);
pszTo[nToLength+1] = _T('\0');
shfo.pTo = pszTo;
//Let the shell perform the actual rename
if (SHFileOperation(&shfo) == 0 && shfo.fAnyOperationsAborted == FALSE)
{
*pResult = TRUE;
//Update its text
SetItemText(pDispInfo->item.hItem, pDispInfo->item.pszText);
//Also update the icons for it
CString sPath = ItemToPath(pDispInfo->item.hItem);
SetItemImage(pDispInfo->item.hItem, GetIconIndex(sPath), GetSelIconIndex(sPath));
}
//Don't forget to free up the memory we allocated
delete [] pszFrom;
delete [] pszTo;
}
*pResult = 0;
}
void MyFrame::OnInsertPage(wxCommandEvent& WXUNUSED(event))
{
static unsigned s_pageIns = 0;
wxBookCtrlBase *currBook = GetCurrentBook();
if ( currBook )
{
wxPanel *panel = CreateUserCreatedPage( currBook );
currBook->InsertPage( 0, panel,
wxString::Format(INSERTED_PAGE_NAME wxT("%u"), ++s_pageIns), false,
GetIconIndex(currBook) );
currBook->SetSelection(0);
}
}
void MyFrame::OnAddPage(wxCommandEvent& WXUNUSED(event))
{
static unsigned s_pageAdded = 0;
wxBookCtrlBase *currBook = GetCurrentBook();
if ( currBook )
{
wxPanel *panel = new wxPanel( currBook, wxID_ANY );
(void) new wxButton( panel, wxID_ANY, wxT("First button"),
wxPoint(10, 10), wxDefaultSize );
(void) new wxButton( panel, wxID_ANY, wxT("Second button"),
wxPoint(50, 100), wxDefaultSize );
currBook->AddPage(panel, wxString::Format(ADDED_PAGE_NAME wxT("%u"),
++s_pageAdded), true, GetIconIndex(currBook) );
}
}
bool CFileAssoc::ReAssocIcons(const CAtlList<CString>& exts)
{
if (!LoadIconLib()) {
return false;
}
SaveIconLibVersion();
const CString progPath = GetProgramPath(true);
CRegKey key;
POSITION pos = exts.GetHeadPosition();
while (pos) {
const CString ext = exts.GetNext(pos);
const CString strProgID = PROGID + ext;
CString appIcon;
int iconIndex = GetIconIndex(ext);
/* icon_index value -1 means no icon was found in the iconlib for the file extension */
if (iconIndex >= 0 && ExtractIcon(AfxGetApp()->m_hInstance, m_iconLibPath, iconIndex)) {
appIcon.Format(_T("\"%s\",%d"), m_iconLibPath, iconIndex);
}
/* no icon was found for the file extension, so use MPC's icon */
if (appIcon.IsEmpty()) {
appIcon = "\"" + progPath + "\",0";
}
if (ERROR_SUCCESS != key.Create(HKEY_CLASSES_ROOT, strProgID + _T("\\DefaultIcon"))
|| ERROR_SUCCESS != key.SetStringValue(NULL, appIcon)) {
return false;
}
key.Close();
}
FreeIconLib();
return true;
}
unsigned int PointMan::GetIconTexture( const wxBitmap *pbm, int &glw, int &glh )
{
#ifdef ocpnUSE_GL
int index = GetIconIndex( pbm );
ODMarkIcon *pmi = (ODMarkIcon *) m_pIconArray->Item( index );
assert(pmi != 0);
unsigned int *IconTexture;
switch (m_ColourScheme) {
case PI_GLOBAL_COLOR_SCHEME_RGB:
IconTexture = &pmi->icon_texture_RGB;
break;
case PI_GLOBAL_COLOR_SCHEME_DAY:
IconTexture = &pmi->icon_texture_Day;
break;
case PI_GLOBAL_COLOR_SCHEME_DUSK:
IconTexture = &pmi->icon_texture_Dusk;
break;
case PI_GLOBAL_COLOR_SCHEME_NIGHT:
IconTexture = &pmi->icon_texture_Night;
break;
default:
IconTexture = &pmi->icon_texture_Day;
break;
}
if(*IconTexture == 0) {
/* make rgba texture */
glGenTextures(1, IconTexture);
glBindTexture(GL_TEXTURE_2D, *IconTexture);
glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST );
glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST );
wxImage image = pbm->ConvertToImage();
int w = image.GetWidth(), h = image.GetHeight();
pmi->tex_w = NextPow2(w);
pmi->tex_h = NextPow2(h);
unsigned char *d = image.GetData();
unsigned char *a = image.GetAlpha();
unsigned char mr, mg, mb;
if(!a)
image.GetOrFindMaskColour( &mr, &mg, &mb );
unsigned char *e = new unsigned char[4 * w * h];
if(d && e){
for( int y = 0; y < h; y++ )
for( int x = 0; x < w; x++ ) {
unsigned char r, g, b;
int off = ( y * image.GetWidth() + x );
r = d[off * 3 + 0];
g = d[off * 3 + 1];
b = d[off * 3 + 2];
e[off * 4 + 0] = r;
e[off * 4 + 1] = g;
e[off * 4 + 2] = b;
e[off * 4 + 3] = a ? a[off] : ( ( r == mr ) && ( g == mg ) && ( b == mb ) ? 0 : 255 );
}
}
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, pmi->tex_w, pmi->tex_h,
0, GL_RGBA, GL_UNSIGNED_BYTE, NULL);
glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, w, h,
GL_RGBA, GL_UNSIGNED_BYTE, e);
delete [] e;
}
glw = pmi->tex_w;
glh = pmi->tex_h;
return *IconTexture;
#else
return 0;
#endif
}
UINT CNBLogicalDevice::GetSelectIconIndex(UINT *anIconIDs, int nCount)
{
ATLASSERT(m_mapUnitDevices.size());
return GetIconIndex(anIconIDs, nCount);
}
UINT CNBUnitDevice::GetSelectIconIndex(UINT *anIconIDs, int nCount)
{
return GetIconIndex(anIconIDs, nCount);
}
void CLocalTreeView::DisplayDir(wxTreeItemId parent, const wxString& dirname, const wxString& knownSubdir /*=_T("")*/)
{
fz::local_filesys local_filesys;
{
wxLogNull log;
if (!local_filesys.begin_find_files(fz::to_native(dirname), true)) {
if (!knownSubdir.empty()) {
wxTreeItemId item = GetSubdir(parent, knownSubdir);
if (item != wxTreeItemId())
return;
const wxString fullName = dirname + knownSubdir;
item = AppendItem(parent, knownSubdir, GetIconIndex(iconType::dir, fullName),
#ifdef __WXMSW__
-1
#else
GetIconIndex(iconType::opened_dir, fullName)
#endif
);
CheckSubdirStatus(item, fullName);
}
else
{
m_setSelection = true;
DeleteChildren(parent);
m_setSelection = false;
}
return;
}
}
wxASSERT(parent);
m_setSelection = true;
DeleteChildren(parent);
m_setSelection = false;
CFilterManager filter;
bool matchedKnown = false;
fz::native_string file;
bool wasLink;
int attributes;
bool is_dir;
static int64_t const size(-1);
fz::datetime date;
while (local_filesys.get_next_file(file, wasLink, is_dir, 0, &date, &attributes)) {
wxASSERT(is_dir);
if (file.empty()) {
wxGetApp().DisplayEncodingWarning();
continue;
}
wxString file_wx = file;
wxString fullName = dirname + file_wx;
#ifdef __WXMSW__
if (file_wx.CmpNoCase(knownSubdir))
#else
if (file_wx != knownSubdir)
#endif
{
if (filter.FilenameFiltered(file_wx, dirname, true, size, true, attributes, date))
continue;
}
else
matchedKnown = true;
wxTreeItemId item = AppendItem(parent, file_wx, GetIconIndex(iconType::dir, fullName),
#ifdef __WXMSW__
-1
#else
GetIconIndex(iconType::opened_dir, fullName)
#endif
);
CheckSubdirStatus(item, fullName);
}
if (!matchedKnown && !knownSubdir.empty()) {
const wxString fullName = dirname + knownSubdir;
wxTreeItemId item = AppendItem(parent, knownSubdir, GetIconIndex(iconType::dir, fullName),
#ifdef __WXMSW__
-1
#else
GetIconIndex(iconType::opened_dir, fullName)
#endif
);
CheckSubdirStatus(item, fullName);
}
SortChildren(parent);
}
void FolderTree::DisplayPath(const tstring &sPath, HTREEITEM hParent, bool bUseSetRedraw /* = true */)
{
//CWaitCursor c;
//Speed up the job by turning off redraw
if (bUseSetRedraw)
SetRedraw(false);
//Remove all the items currently under hParent
HTREEITEM hChild = GetChildItem(hParent);
while (hChild)
{
DeleteItem(hChild);
hChild = GetChildItem(hParent);
}
//Should we display the root folder
if (m_bShowRootedFolder && (hParent == TVI_ROOT))
{
FolderTreeItemInfo* pItem = new FolderTreeItemInfo;
pItem->m_sFQPath = m_sRootFolder;
pItem->m_sRelativePath = m_sRootFolder;
m_hRootedFolder = InsertFileItem(TVI_ROOT, pItem, false, GetIconIndex(m_sRootFolder), GetSelIconIndex(m_sRootFolder), true);
Expand(m_hRootedFolder, TVE_EXPAND);
return;
}
//find all the directories underneath sPath
int nDirectories = 0;
tstring sFile;
if (sPath[sPath.size()-1] != _T('\\'))
sFile = sPath + _T("\\");
else
sFile = sPath;
WIN32_FIND_DATA fData;
HANDLE hFind;
hFind = FindFirstFile((sFile + _T("*")).c_str(), &fData);
if(hFind != INVALID_HANDLE_VALUE)
{
do
{
tstring filename = fData.cFileName;
if((fData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
(filename != _T(".")) && (filename != _T("..")))
{
++nDirectories;
tstring path = sFile + filename;
TCHAR szPath[_MAX_PATH];
TCHAR szFname[_MAX_FNAME];
TCHAR szExt[_MAX_EXT];
_tsplitpath(path.c_str(), NULL, NULL, szFname, szExt);
_tmakepath(szPath, NULL, NULL, szFname, szExt);
FolderTreeItemInfo* pItem = new FolderTreeItemInfo;
pItem->m_sFQPath = path;
pItem->m_sRelativePath = szPath;
InsertFileItem(hParent, pItem, m_bShowSharedUsingDifferentIcon && IsShared(path), GetIconIndex(path), GetSelIconIndex(path), true);
}
} while (FindNextFile(hFind, &fData));
}
FindClose(hFind);
//Now sort the items we just added
TVSORTCB tvsortcb;
tvsortcb.hParent = hParent;
tvsortcb.lpfnCompare = CompareByFilenameNoCase;
tvsortcb.lParam = 0;
SortChildrenCB(&tvsortcb);
//We want to add them before sorting
checkRemovedDirs(sFile, hParent);
//If no items were added then remove the "+" indicator from hParent
if(nDirectories == 0)
SetHasPlusButton(hParent, FALSE);
//Turn back on the redraw flag
if(bUseSetRedraw)
SetRedraw(true);
}
void FolderTree::DisplayDrives(HTREEITEM hParent, bool bUseSetRedraw)
{
//CWaitCursor c;
//Speed up the job by turning off redraw
if (bUseSetRedraw)
SetRedraw(false);
//Enumerate the drive letters and add them to the tree control
DWORD dwDrives = GetLogicalDrives();
DWORD dwMask = 1;
for (int i=0; i<32; i++)
{
if (dwDrives & dwMask)
{
tstring sDrive;
sDrive = (wchar_t)('A' + i);
sDrive += _T(":\\");
//check if this drive is one of the types to hide
if (CanDisplayDrive(sDrive))
{
FolderTreeItemInfo* pItem = new FolderTreeItemInfo;
pItem->m_sFQPath = sDrive;
pItem->m_sRelativePath = sDrive;
//Insert the item into the view
InsertFileItem(hParent, pItem, m_bShowSharedUsingDifferentIcon && IsShared(sDrive), GetIconIndex(sDrive), GetSelIconIndex(sDrive), true);
}
}
dwMask <<= 1;
}
if (bUseSetRedraw)
SetRedraw(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);
}
void CLocalTreeView::Refresh()
{
wxLogNull nullLog;
const wxString separator = wxFileName::GetPathSeparator();
std::list<t_dir> dirsToCheck;
#ifdef __WXMSW__
int prevErrorMode = SetErrorMode(SEM_FAILCRITICALERRORS);
wxTreeItemIdValue tmp;
wxTreeItemId child = GetFirstChild(m_drives, tmp);
while (child)
{
if (IsExpanded(child))
{
wxString drive = GetItemText(child);
int pos = drive.Find(_T(" "));
if (pos != -1)
drive = drive.Left(pos);
t_dir dir;
dir.dir = drive + separator;
dir.item = child;
dirsToCheck.push_back(dir);
}
child = GetNextSibling(child);
}
#else
t_dir dir;
dir.dir = separator;
dir.item = GetRootItem();
dirsToCheck.push_back(dir);
#endif
CFilterManager filter;
while (!dirsToCheck.empty())
{
t_dir dir = dirsToCheck.front();
dirsToCheck.pop_front();
// Step 1: Check if directory exists
CLocalFileSystem local_filesystem;
if (!local_filesystem.BeginFindFiles(dir.dir, true))
{
// Dir does exist (listed in parent) but may not be accessible.
// Recurse into children anyhow, they might be accessible again.
wxTreeItemIdValue value;
wxTreeItemId child = GetFirstChild(dir.item, value);
while (child)
{
t_dir subdir;
subdir.dir = dir.dir + GetItemText(child) + separator;
subdir.item = child;
dirsToCheck.push_back(subdir);
child = GetNextSibling(child);
}
continue;
}
// Step 2: Enumerate subdirectories on disk and sort them
std::list<wxString> dirs;
wxString file;
const wxLongLong size(-1);
bool was_link;
bool is_dir;
int attributes;
wxDateTime date;
while (local_filesystem.GetNextFile(file, was_link, is_dir, 0, &date, &attributes))
{
if (file == _T(""))
{
wxGetApp().DisplayEncodingWarning();
continue;
}
if (filter.FilenameFiltered(file, dir.dir, true, size, true, attributes, date.IsValid() ? &date : 0))
continue;
dirs.push_back(file);
}
dirs.sort(sortfunc);
bool inserted = false;
// Step 3: Merge list of subdirectories with subtree.
wxTreeItemId child = GetLastChild(dir.item);
std::list<wxString>::reverse_iterator iter = dirs.rbegin();
while (child || iter != dirs.rend())
{
int cmp;
if (child && iter != dirs.rend())
#ifdef __WXMSW__
cmp = GetItemText(child).CmpNoCase(*iter);
#else
cmp = GetItemText(child).Cmp(*iter);
#endif
else if (child)
cmp = 1;
else
cmp = -1;
if (!cmp)
{
// Found item with same name. Mark it for further processing
if (!IsExpanded(child))
{
wxString path = dir.dir + *iter + separator;
if (!CheckSubdirStatus(child, path))
{
t_dir subdir;
subdir.dir = path;
subdir.item = child;
dirsToCheck.push_front(subdir);
}
}
else
{
t_dir subdir;
subdir.dir = dir.dir + *iter + separator;
subdir.item = child;
dirsToCheck.push_front(subdir);
}
child = GetPrevSibling(child);
++iter;
}
else if (cmp > 0)
{
// Subdirectory currently in tree no longer exists.
// Delete child from tree, unless current selection
// is in the subtree.
wxTreeItemId sel = GetSelection();
while (sel && sel != child)
sel = GetItemParent(sel);
wxTreeItemId prev = GetPrevSibling(child);
if (!sel)
Delete(child);
child = prev;
}
else if (cmp < 0)
{
// New subdirectory, add treeitem
wxString fullname = dir.dir + *iter + separator;
wxTreeItemId newItem = AppendItem(dir.item, *iter, GetIconIndex(::dir, fullname),
#ifdef __WXMSW__
-1
#else
GetIconIndex(opened_dir, fullname)
#endif
);
CheckSubdirStatus(newItem, fullname);
++iter;
inserted = true;
}
}
if (inserted)
SortChildren(dir.item);
}
bool CLocalTreeView::DisplayDrives(wxTreeItemId parent)
{
wxGetApp().AddStartupProfileRecord(_T("CLocalTreeView::DisplayDrives"));
long drivesToHide = 0;
// Adhere to the NODRIVES group policy
wxRegKey key(_T("HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer"));
if (key.Exists())
{
if (!key.HasValue(_T("NoDrives")) || !key.QueryValue(_T("NoDrives"), &drivesToHide))
drivesToHide = 0;
}
int len = GetLogicalDriveStrings(0, 0);
if (!len)
return false;
wxChar* drives = new wxChar[len + 1];
if (!GetLogicalDriveStrings(len, drives))
{
delete [] drives;
return false;
}
m_pVolumeEnumeratorThread = new CVolumeDescriptionEnumeratorThread(this);
if (m_pVolumeEnumeratorThread->Failed())
{
delete m_pVolumeEnumeratorThread;
m_pVolumeEnumeratorThread = 0;
}
const wxChar* pDrive = drives;
while (*pDrive)
{
// Check if drive should be hidden by default
if (pDrive[0] != 0 && pDrive[1] == ':')
{
int bit = 0;
char letter = pDrive[0];
if (letter >= 'A' && letter <= 'Z')
bit = 1 << (letter - 'A');
if (letter >= 'a' && letter <= 'z')
bit = 1 << (letter - 'a');
if (drivesToHide & bit)
{
pDrive += wxStrlen(pDrive) + 1;
continue;
}
}
wxString drive = pDrive;
if (drive.Right(1) == _T("\\"))
drive = drive.RemoveLast();
wxGetApp().AddStartupProfileRecord(wxString::Format(_T("CLocalTreeView::DisplayDrives adding drive %s"), drive.c_str()));
wxTreeItemId item = AppendItem(parent, drive, GetIconIndex(dir, pDrive));
AppendItem(item, _T(""));
SortChildren(parent);
pDrive += wxStrlen(pDrive) + 1;
}
wxGetApp().AddStartupProfileRecord(_T("CLocalTreeView::DisplayDrives adding drives done"));
delete [] drives;
return true;
}
CXTPImageManagerIcon* CXTPPopupItem::GetImage() const
{
return m_pControl->GetImageManager()->GetImage(GetIconIndex(), m_rcItem.Width());
}
void CLocalTreeView::DisplayDir(wxTreeItemId parent, const wxString& dirname, const wxString& knownSubdir /*=_T("")*/)
{
CLocalFileSystem local_filesystem;
{
wxLogNull log;
if (!local_filesystem.BeginFindFiles(dirname, true))
{
if (knownSubdir != _T(""))
{
wxTreeItemId item = GetSubdir(parent, knownSubdir);
if (item != wxTreeItemId())
return;
const wxString fullName = dirname + knownSubdir;
item = AppendItem(parent, knownSubdir, GetIconIndex(::dir, fullName),
#ifdef __WXMSW__
-1
#else
GetIconIndex(opened_dir, fullName)
#endif
);
CheckSubdirStatus(item, fullName);
}
else
{
m_setSelection = true;
DeleteChildren(parent);
m_setSelection = false;
}
return;
}
}
wxASSERT(parent);
m_setSelection = true;
DeleteChildren(parent);
m_setSelection = false;
CFilterManager filter;
bool matchedKnown = false;
wxString file;
bool wasLink;
int attributes;
bool is_dir;
const wxLongLong size(-1);
wxDateTime date;
while (local_filesystem.GetNextFile(file, wasLink, is_dir, 0, &date, &attributes))
{
wxASSERT(is_dir);
if (file == _T(""))
{
wxGetApp().DisplayEncodingWarning();
continue;
}
wxString fullName = dirname + file;
#ifdef __WXMSW__
if (file.CmpNoCase(knownSubdir))
#else
if (file != knownSubdir)
#endif
{
if (filter.FilenameFiltered(file, dirname, true, size, true, attributes, date.IsValid() ? &date : 0))
continue;
}
else
matchedKnown = true;
wxTreeItemId item = AppendItem(parent, file, GetIconIndex(::dir, fullName),
#ifdef __WXMSW__
-1
#else
GetIconIndex(opened_dir, fullName)
#endif
);
CheckSubdirStatus(item, fullName);
}
if (!matchedKnown && knownSubdir != _T(""))
{
const wxString fullName = dirname + knownSubdir;
wxTreeItemId item = AppendItem(parent, knownSubdir, GetIconIndex(::dir, fullName),
#ifdef __WXMSW__
-1
#else
GetIconIndex(opened_dir, fullName)
#endif
);
CheckSubdirStatus(item, fullName);
}
SortChildren(parent);
}
int CDocStructRecordItem::Draw(XTP_REPORTRECORDITEM_DRAWARGS* pDrawArgs) {
AFX_MANAGE_STATE(AfxGetStaticModuleState());
if (pDrawArgs->pDC == NULL || pDrawArgs->pControl == NULL || pDrawArgs->pRow == NULL || pDrawArgs->pItem == NULL) {
ASSERT(FALSE);
return 0;
}
int offSet = 0;
CDC* pDC = pDrawArgs->pDC;
int nTreeDepth = pDrawArgs->pRow->GetTreeDepth();
int nLeftIndent = pDrawArgs->pControl->GetIndent(nTreeDepth);
CRect rcRow = pDrawArgs->pRow->GetRect();
nTreeDepth == 0 ? offSet = 0 : offSet = 0;
CXTPOffice2007Image *pImage = XTPOffice2007Images()->LoadFile(_T("PUSHBUTTON"));
ASSERT(pImage);
int nState = 0;
if (pDrawArgs->pRow->IsSelected() == TRUE) {
nState = 2;
}
CDocStructRecord* pRecord = (CDocStructRecord*)GetRecord();
if (pRecord) {
int nCurrentLayout = pRecord->m_nCurrentLayout;
if (nCurrentLayout == DocumentStructureLayout(Thumbnail)) {
rcRow.left += nTreeDepth * TREE_DEPTH_OFFSET;
rcRow.top += 2;
rcRow.bottom -= 2;
rcRow.right -= nTreeDepth * TREE_DEPTH_OFFSET;
} else {
rcRow.left += 0; //nLeftIndent + offSet;
}
if (pImage) {
// Draw a blue rectangle that surrounds all child items to show that an item is part of a group.
if (nCurrentLayout == DocumentStructureLayout(Thumbnail) && pDrawArgs->pRow->GetParentRow() != NULL) {
int nTopRow = pDrawArgs->pControl->GetTopRowIndex();
int nVisibleRows = GetReportAreaRows(pDrawArgs->pControl, pDrawArgs->pControl->GetTopRowIndex(), TRUE);
CXTPReportRow *pTopRow = pDrawArgs->pControl->GetRows()->GetAt(nTopRow);
if (pTopRow != NULL) {
if (pTopRow->GetTreeDepth() > 1 && pTopRow == pDrawArgs->pRow) {
for (int i = 1; i < pTopRow->GetTreeDepth(); i++) {
int iExpandedChildsHeight = GetExpandedChildsHeight(pDrawArgs->pControl, pDrawArgs->pRow->GetParentRow()->GetRecord());
CRect rcParentRect(pTopRow->GetRect());
bool bReset = false;
for (int j = nTopRow; j < nTopRow + nVisibleRows ; j++) {
int iRowLevel = pDrawArgs->pControl->GetRows()->GetAt(j)->GetTreeDepth();
if (iRowLevel >= i+1) {
if (bReset == false) {
iExpandedChildsHeight = 0;
bReset = true;
}
iExpandedChildsHeight += pDrawArgs->pControl->GetPaintManager()->GetRowHeight(NULL, pDrawArgs->pControl->GetRows()->GetAt(j));
} else {
break;
}
}
rcParentRect.bottom = iExpandedChildsHeight < pDrawArgs->pControl->GetReportRectangle().bottom ? iExpandedChildsHeight : pDrawArgs->pControl->GetReportRectangle().bottom + 2;
rcParentRect.left += i * TREE_DEPTH_OFFSET;
rcParentRect.right -= i * TREE_DEPTH_OFFSET;
pImage->DrawImage(pDC, rcParentRect, pImage->GetSource(0, 5), CRect(4, 4, 4, 4), COLORREF_NULL);
}
}
}
}
// Draw an orange/blue image over item.
if ((nCurrentLayout == DocumentStructureLayout(Thumbnail)) ||((nCurrentLayout == DocumentStructureLayout(Compact)) && nState == 2)) {
if (pRecord->IsPage()) {
pImage->DrawImage(pDC, rcRow, pImage->GetSource(nState, 5), CRect(4, 4, 4, 4), COLORREF_NULL);
} else {
// take tree depth and draw button images of tree depth size.
CRect newRowOffset(rcRow);
if (pDrawArgs->pRow->GetTreeDepth() != 0 && nCurrentLayout == DocumentStructureLayout(Thumbnail)) {
newRowOffset.bottom += GetExpandedChildsHeight(pDrawArgs->pControl, pDrawArgs->pRow->GetRecord());
if ( pDrawArgs->pRow->IsExpanded() == TRUE ) {
newRowOffset.bottom += 1;
}
pImage->DrawImage(pDC, newRowOffset, pImage->GetSource(0, 5), CRect(4, 4, 4, 4), COLORREF_NULL);
}
if (pDrawArgs->pRow->GetTreeDepth() == 0) {
pImage->DrawImage(pDC, rcRow, pImage->GetSource(nState, 5), CRect(4, 4, 4, 4), COLORREF_NULL);
}
if (nState == 2) {
pImage->DrawImage(pDC, rcRow, pImage->GetSource(nState, 5), CRect(4, 4, 4, 4), COLORREF_NULL);
}
}
}
}
pDC->SetBkMode(TRANSPARENT);
if (nCurrentLayout == DocumentStructureLayout(Thumbnail) && pRecord->IsPage()) {
int x = rcRow.left + RECORD_ITEM_OFFSET;
int y = rcRow.top + RECORD_ITEM_OFFSET;
int nWidth = rcRow.Width() - 2*RECORD_ITEM_OFFSET;
int nHeight = nWidth * 3/4;
if (nWidth > rcRow.Width()) {
nWidth = rcRow.Width();
}
CRect rcThumb;
rcThumb.SetRect(x, y, x + nWidth, y + nHeight);
if (pRecord->m_pContainer != NULL) {
pRecord->m_pContainer->DrawThumbWhiteboard(pRecord->m_pvData, rcThumb, pDC);
}
}
}
// Code taken from CXTPReportControlItem::Draw - but a bit modified
m_pControl = pDrawArgs->pControl;
CRect& rcItem = pDrawArgs->rcItem;
CXTPReportPaintManager* pPaintManager = pDrawArgs->pControl->GetPaintManager();
CRgn rgn;
rgn.CreateRectRgn(rcItem.left, rcItem.top - 1, rcItem.right, rcItem.bottom);
if (!pDC->IsPrinting()) {
pDC->SelectClipRgn(&rgn);
}
XTP_REPORTRECORDITEM_METRICS* pMetrics = new XTP_REPORTRECORDITEM_METRICS();
pMetrics->strText = GetCaption(pDrawArgs->pColumn);
pDrawArgs->pRow->GetItemMetrics(pDrawArgs, pMetrics);
ASSERT(pMetrics->pFont);
ASSERT(pMetrics->clrForeground != XTP_REPORT_COLOR_DEFAULT);
int nItemGlyphs = rcItem.left;
// draw tree inside item rect (see also HitTest function)
if (pDrawArgs->pColumn && pDrawArgs->pColumn->IsTreeColumn()) {
int nTreeDepth = pDrawArgs->pRow->GetTreeDepth() - pDrawArgs->pRow->GetGroupLevel();
if (nTreeDepth > 0) {
nTreeDepth++;
}
rcItem.left += pDrawArgs->pControl->GetIndent(nTreeDepth);
BOOL bHasChildren = pDrawArgs->pControl->IsVirtualMode() ?
pDrawArgs->pRow->GetTreeDepth() == 0 && (pMetrics->nVirtRowFlags & xtpVirtRowHasChildren) :
pDrawArgs->pRow->HasChildren();
CRect rcBitmap(rcItem);
CSize sizeBitmap = pPaintManager->DrawCollapsedBitmap(NULL, pDrawArgs->pRow, rcBitmap);
int nIconAlign = pDrawArgs->nTextAlign & xtpColumnIconMask;
// horizontal alignment
switch (nIconAlign) {
case xtpColumnIconRight:
rcBitmap.left = rcBitmap.right - sizeBitmap.cx - 2;
break;
case xtpColumnIconLeft:
rcBitmap.right = rcBitmap.left + sizeBitmap.cx + 2;
break;
}
// vertical alignment
switch (nIconAlign) {
case xtpColumnIconTop:
rcBitmap.bottom = rcBitmap.top + sizeBitmap.cy + 2;
break;
case xtpColumnIconBottom:
rcBitmap.top = rcBitmap.bottom - sizeBitmap.cy - 2;
break;
}
bool bDrawArrow = false;
if (bHasChildren || (pRecord->IsDocument() && !pRecord->IsEmptyDocument())) {
bDrawArrow = true;
}
if (pRecord->IsDocument() && !pRecord->IsLoadedDocument()) {
pDrawArgs->pRow->SetExpanded(FALSE);
}
sizeBitmap = pPaintManager->DrawCollapsedBitmap(bDrawArrow ? pDC : NULL, pDrawArgs->pRow, rcBitmap);
if (!pDC->IsPrinting() && bHasChildren) {
pDrawArgs->pRow->SetCollapseRect(rcBitmap);
}
// Draw a custom image at the left side of the record item
if (pPaintManager->GetRuntimeClass() == RUNTIME_CLASS(CXTPDocumentStructurePaintManager)) {
if (pRecord->IsChapter() || pRecord->IsDocument()) {
CRect rcBitmap(rcItem);
if (((CDocStructRecord*) pDrawArgs->pRow->GetRecord())->m_nCurrentLayout == DocumentStructureLayout(Thumbnail)) {
rcBitmap.left += 15;
rcBitmap.top -= 5;
CSize csBitmap = ((CXTPDocumentStructurePaintManager*)pPaintManager)->DrawCustomBitmap(pDC, pDrawArgs->pRow, rcBitmap, m_nImageIndex);
rcItem.left += sizeBitmap.cx + 35;
} else {
rcBitmap.left += 18;
CSize csBitmap = ((CXTPDocumentStructurePaintManager*)pPaintManager)->DrawCustomBitmap(pDC, pDrawArgs->pRow, rcBitmap, m_nImageIndex);
rcItem.left += sizeBitmap.cx + 25;
}
} else {
if (((CDocStructRecord*) pDrawArgs->pRow->GetRecord())->m_nCurrentLayout == DocumentStructureLayout(Compact)) {
rcBitmap.left += 15;
CSize csBitmap = ((CXTPDocumentStructurePaintManager*)pPaintManager)->DrawCustomBitmap(pDC, pDrawArgs->pRow, rcBitmap, m_nImageIndex);
rcItem.left += sizeBitmap.cx + 10;
}
rcItem.left += sizeBitmap.cx + 2;
}
}
}
COLORREF clrText = XTPOffice2007Images()->GetImageColor(_T("LISTBOX"), _T("NormalText"));
pDC->SetTextColor(clrText);
CFont* pOldFont = (CFont*)pDC->SelectObject(pMetrics->pFont);
if (pMetrics->clrBackground != XTP_REPORT_COLOR_DEFAULT) {
pDC->SetBkColor(pMetrics->clrBackground);
} else {
pDC->SetBkColor(pPaintManager->m_clrControlBack);
}
if (m_bHasCheckbox) {
DrawCheckBox(pDrawArgs, rcItem);
}
// Do the draw bitmap pDC, rcItem, GetIconIndex()
if (pMetrics->nItemIcon != XTP_REPORT_NOICON || GetIconIndex() != XTP_REPORT_NOICON) {
pPaintManager->DrawItemBitmap(pDrawArgs, rcItem, pMetrics->nItemIcon != XTP_REPORT_NOICON ? pMetrics->nItemIcon : GetIconIndex());
}
nItemGlyphs = (rcItem.left - nItemGlyphs);
OnDrawControls(pDrawArgs, rcItem);
if (rcItem.Width()) {
OnDrawCaption(pDrawArgs, pMetrics);
}
int nItemTextWidth = nItemGlyphs + 7;
if (m_pMarkupUIElement) {
nItemTextWidth += XTPMarkupMeasureElement(m_pMarkupUIElement, rcItem.Width(), INT_MAX).cx;
} else {
nItemTextWidth += pDC->GetTextExtent(pMetrics->strText).cx;
}
pDC->SelectObject(pOldFont);
pMetrics->InternalRelease();
if (!pDC->IsPrinting()) {
pDC->SelectClipRgn(NULL);
}
return nItemTextWidth;
}
bool CPPageFormats::RegisterExt(CString ext, CString strLabel, bool fRegister)
{
CRegKey key;
bool bSetValue;
CString strProgID = _T("mplayerc") + ext;
if(!fRegister)
{
if(fRegister != IsRegistered(ext))
SetFileAssociation (ext, strProgID, fRegister);
key.Attach(HKEY_CLASSES_ROOT);
key.RecurseDeleteKey(strProgID);
return(true);
}
bSetValue = fRegister || (ERROR_SUCCESS != key.Open(HKEY_CLASSES_ROOT, strProgID + _T("\\shell\\open\\command"), KEY_READ));
// Create ProgID for this file type
if(ERROR_SUCCESS != key.Create(HKEY_CLASSES_ROOT, strProgID)) return(false);
if(ERROR_SUCCESS != key.SetStringValue(NULL, strLabel)) return(false);
// Add to playlist option
if(f_setContextFiles)
{
if(ERROR_SUCCESS != key.Create(HKEY_CLASSES_ROOT, strProgID + _T("\\shell\\enqueue"))) return(false);
if(ERROR_SUCCESS != key.SetStringValue(NULL, ResStr(IDS_ADD_TO_PLAYLIST))) return(false);
if(ERROR_SUCCESS != key.Create(HKEY_CLASSES_ROOT, strProgID + _T("\\shell\\enqueue\\command"))) return(false);
if(bSetValue && (ERROR_SUCCESS != key.SetStringValue(NULL, GetEnqueueCommand()))) return(false);
}
else
{
key.Close();
key.Attach(HKEY_CLASSES_ROOT);
key.RecurseDeleteKey(strProgID + _T("\\shell\\enqueue"));
}
// Play option
if(ERROR_SUCCESS != key.Create(HKEY_CLASSES_ROOT, strProgID + _T("\\shell\\open"))) return(false);
if(f_setContextFiles)
{
if(ERROR_SUCCESS != key.SetStringValue(NULL, ResStr(IDS_OPEN_WITH_MPC))) return(false);
}
else
{
if(ERROR_SUCCESS != key.SetStringValue(NULL, _T(""))) return(false);
}
if(ERROR_SUCCESS != key.Create(HKEY_CLASSES_ROOT, strProgID + _T("\\shell\\open\\command"))) return(false);
if(bSetValue && (ERROR_SUCCESS != key.SetStringValue(NULL, GetOpenCommand()))) return(false);
if(ERROR_SUCCESS != key.Create(HKEY_LOCAL_MACHINE, g_strRegisteredKey + _T("\\FileAssociations"))) return(false);
if(ERROR_SUCCESS != key.SetStringValue(ext, strProgID)) return(false);
if(f_setAssociatedWithIcon)
{
CString AppIcon = _T("");
TCHAR buff[_MAX_PATH];
CString mpciconlib = GetProgramDir() + _T("\\mpciconlib.dll");
if(FileExists(mpciconlib))
{
int icon_index = GetIconIndex(ext);
CString m_typeicon = mpciconlib;
/* icon_index value -1 means no icon was found in the iconlib for the file extension */
if((icon_index >= 0) && ExtractIcon(AfxGetApp()->m_hInstance,(LPCWSTR)m_typeicon, icon_index))
{
m_typeicon = "\""+mpciconlib+"\"";
AppIcon.Format(_T("%s,%d"), m_typeicon, icon_index);
}
}
/* no icon was found for the file extension, so use MPC's icon */
if((AppIcon.IsEmpty()) && (::GetModuleFileName(AfxGetInstanceHandle(), buff, MAX_PATH)))
{
AppIcon = buff;
AppIcon = "\""+AppIcon+"\"";
AppIcon += _T(",0");
}
if(ERROR_SUCCESS != key.Create(HKEY_CLASSES_ROOT, strProgID + _T("\\DefaultIcon"))) return(false);
if(bSetValue && (ERROR_SUCCESS != key.SetStringValue(NULL, AppIcon))) return(false);
}
else
{
key.Attach(HKEY_CLASSES_ROOT);
key.RecurseDeleteKey(strProgID + _T("\\DefaultIcon"));
}
if(fRegister != IsRegistered(ext))
SetFileAssociation (ext, strProgID, fRegister);
return(true);
}
bool FolderTree::EnumNetwork(HTREEITEM hParent)
{
//What will be the return value from this function
bool bGotChildren = false;
//Check if the item already has a network resource and use it.
FolderTreeItemInfo* pItem = (FolderTreeItemInfo*) GetItemData(hParent);
NETRESOURCE* pNetResource = pItem->m_pNetResource;
//Setup for the network enumeration
HANDLE hEnum;
DWORD dwResult = WNetOpenEnum(pNetResource ? RESOURCE_GLOBALNET : RESOURCE_CONTEXT, m_dwNetworkItemTypes,
0, pNetResource ? pNetResource : NULL, &hEnum);
//Was the read sucessful
if (dwResult != NO_ERROR)
{
//TRACE(_T("Cannot enumerate network drives, Error:%d\n"), dwResult);
return FALSE;
}
//Do the network enumeration
DWORD cbBuffer = 16384;
bool bNeedMoreMemory = true;
bool bSuccess = false;
LPNETRESOURCE lpnrDrv = NULL;
DWORD cEntries = 0;
while (bNeedMoreMemory && !bSuccess)
{
//Allocate the memory and enumerate
lpnrDrv = (LPNETRESOURCE) new BYTE[cbBuffer];
cEntries = 0xFFFFFFFF;
dwResult = WNetEnumResource(hEnum, &cEntries, lpnrDrv, &cbBuffer);
if (dwResult == ERROR_MORE_DATA)
{
//Free up the heap memory we have used
delete [] lpnrDrv;
cbBuffer *= 2;
}
else if (dwResult == NO_ERROR)
bSuccess = true;
else
bNeedMoreMemory = false;
}
//Enumeration successful?
if (bSuccess)
{
//Scan through the results
for (DWORD i=0; i<cEntries; i++)
{
tstring sNameRemote;
if(lpnrDrv[i].lpRemoteName != NULL)
sNameRemote = lpnrDrv[i].lpRemoteName;
else
sNameRemote = lpnrDrv[i].lpComment;
//Remove leading back slashes
if (sNameRemote.size() > 0 && sNameRemote[0] == _T('\\'))
sNameRemote = sNameRemote.substr(1);
if (sNameRemote.size() > 0 && sNameRemote[0] == _T('\\'))
sNameRemote = sNameRemote.substr(1);
//Setup the item data for the new item
pItem = new FolderTreeItemInfo;
pItem->m_pNetResource = new NETRESOURCE;
memzero(pItem->m_pNetResource, sizeof(NETRESOURCE));
*pItem->m_pNetResource = lpnrDrv[i];
if (lpnrDrv[i].lpLocalName)
pItem->m_pNetResource->lpLocalName = _tcsdup(lpnrDrv[i].lpLocalName);
if (lpnrDrv[i].lpRemoteName)
pItem->m_pNetResource->lpRemoteName = _tcsdup(lpnrDrv[i].lpRemoteName);
if (lpnrDrv[i].lpComment)
pItem->m_pNetResource->lpComment = _tcsdup(lpnrDrv[i].lpComment);
if (lpnrDrv[i].lpProvider)
pItem->m_pNetResource->lpProvider = _tcsdup(lpnrDrv[i].lpProvider);
if (lpnrDrv[i].lpRemoteName)
pItem->m_sFQPath = lpnrDrv[i].lpRemoteName;
else
pItem->m_sFQPath = sNameRemote;
pItem->m_sRelativePath = sNameRemote;
pItem->m_bNetworkNode = true;
//Display a share and the appropiate icon
if (lpnrDrv[i].dwDisplayType == RESOURCEDISPLAYTYPE_SHARE)
{
//Display only the share name
int nPos = pItem->m_sRelativePath.find(_T('\\'));
if (nPos >= 0)
pItem->m_sRelativePath = pItem->m_sRelativePath.substr(nPos+1);
//Now add the item into the control
InsertFileItem(hParent, pItem, m_bShowSharedUsingDifferentIcon, GetIconIndex(pItem->m_sFQPath), GetSelIconIndex(pItem->m_sFQPath), false);
}
else if (lpnrDrv[i].dwDisplayType == RESOURCEDISPLAYTYPE_SERVER)
{
//Now add the item into the control
tstring sServer = _T("\\\\");
sServer += pItem->m_sRelativePath;
InsertFileItem(hParent, pItem, false, GetIconIndex(sServer), GetSelIconIndex(sServer), false);
}
else
{
//Now add the item into the control
//Just use the generic Network Neighborhood icons for everything else
LPITEMIDLIST lpNNPidl;
int nIcon = 0xFFFF;
int nSelIcon = nIcon;
if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_NETWORK, &lpNNPidl)))
{
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();
}
InsertFileItem(hParent, pItem, false, nIcon, nSelIcon, false);
}
bGotChildren = true;
}
}
/* else
TRACE(_T("Cannot complete network drive enumeration, Error:%d\n"), dwResult);
*/
//Clean up the enumeration handle
WNetCloseEnum(hEnum);
//Free up the heap memory we have used
delete [] lpnrDrv;
//Return whether or not we added any items
return bGotChildren;
}