/**************************************************************************** ** ** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies). ** All rights reserved. ** Contact: Nokia Corporation (qt-info@nokia.com) ** ** This file is part of the QtGui module of the Qt Toolkit. ** ** $QT_BEGIN_LICENSE:LGPL$ ** No Commercial Usage ** This file contains pre-release code and may not be distributed. ** You may use this file in accordance with the terms and conditions ** contained in the Technology Preview License Agreement accompanying ** this package. ** ** GNU Lesser General Public License Usage ** Alternatively, this file may be used under the terms of the GNU Lesser ** General Public License version 2.1 as published by the Free Software ** Foundation and appearing in the file LICENSE.LGPL included in the ** packaging of this file. Please review the following information to ** ensure the GNU Lesser General Public License version 2.1 requirements ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html. ** ** In addition, as a special exception, Nokia gives you certain additional ** rights. These rights are described in the Nokia Qt LGPL Exception ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package. ** ** If you have questions regarding the use of this file, please contact ** Nokia at qt-info@nokia.com. ** ** ** ** ** ** ** ** ** $QT_END_LICENSE$ ** ****************************************************************************/ #include "qfileiconprovider.h" #ifndef QT_NO_FILEICONPROVIDER #include #include #include #include #if defined(Q_WS_WIN) #define _WIN32_IE 0x0500 #include #include #elif defined(Q_WS_MAC) #include #endif #include #ifdef Q_OS_WINCE #include #endif #ifndef SHGFI_ADDOVERLAYS #define SHGFI_ADDOVERLAYS 0x000000020 #endif QT_BEGIN_NAMESPACE /*! \class QFileIconProvider \brief The QFileIconProvider class provides file icons for the QDirModel class. */ /*! \enum QFileIconProvider::IconType \value Computer \value Desktop \value Trashcan \value Network \value Drive \value Folder \value File */ class QFileIconProviderPrivate { Q_DECLARE_PUBLIC(QFileIconProvider) public: QFileIconProviderPrivate(); QIcon getIcon(QStyle::StandardPixmap name) const; #ifdef Q_WS_WIN QIcon getWinIcon(const QFileInfo &fi) const; #elif defined(Q_WS_MAC) QIcon getMacIcon(const QFileInfo &fi) const; #endif QFileIconProvider *q_ptr; QString homePath; private: QIcon file; QIcon fileLink; QIcon directory; QIcon directoryLink; QIcon harddisk; QIcon floppy; QIcon cdrom; QIcon ram; QIcon network; QIcon computer; QIcon desktop; QIcon trashcan; QIcon generic; QIcon home; }; QFileIconProviderPrivate::QFileIconProviderPrivate() { QStyle *style = QApplication::style(); file = style->standardIcon(QStyle::SP_FileIcon); directory = style->standardIcon(QStyle::SP_DirIcon); fileLink = style->standardIcon(QStyle::SP_FileLinkIcon); directoryLink = style->standardIcon(QStyle::SP_DirLinkIcon); harddisk = style->standardIcon(QStyle::SP_DriveHDIcon); floppy = style->standardIcon(QStyle::SP_DriveFDIcon); cdrom = style->standardIcon(QStyle::SP_DriveCDIcon); network = style->standardIcon(QStyle::SP_DriveNetIcon); computer = style->standardIcon(QStyle::SP_ComputerIcon); desktop = style->standardIcon(QStyle::SP_DesktopIcon); trashcan = style->standardIcon(QStyle::SP_TrashIcon); home = style->standardIcon(QStyle::SP_DirHomeIcon); homePath = QDir::home().absolutePath(); } QIcon QFileIconProviderPrivate::getIcon(QStyle::StandardPixmap name) const { switch (name) { case QStyle::SP_FileIcon: return file; case QStyle::SP_FileLinkIcon: return fileLink; case QStyle::SP_DirIcon: return directory; case QStyle::SP_DirLinkIcon: return directoryLink; case QStyle::SP_DriveHDIcon: return harddisk; case QStyle::SP_DriveFDIcon: return floppy; case QStyle::SP_DriveCDIcon: return cdrom; case QStyle::SP_DriveNetIcon: return network; case QStyle::SP_ComputerIcon: return computer; case QStyle::SP_DesktopIcon: return desktop; case QStyle::SP_TrashIcon: return trashcan; case QStyle::SP_DirHomeIcon: return home; default: return QIcon(); } return QIcon(); } /*! Constructs a file icon provider. */ QFileIconProvider::QFileIconProvider() : d_ptr(new QFileIconProviderPrivate) { } /*! Destroys the file icon provider. */ QFileIconProvider::~QFileIconProvider() { delete d_ptr; } /*! Returns an icon set for the given \a type. */ QIcon QFileIconProvider::icon(IconType type) const { Q_D(const QFileIconProvider); switch (type) { case Computer: return d->getIcon(QStyle::SP_ComputerIcon); case Desktop: return d->getIcon(QStyle::SP_DesktopIcon); case Trashcan: return d->getIcon(QStyle::SP_TrashIcon); case Network: return d->getIcon(QStyle::SP_DriveNetIcon); case Drive: return d->getIcon(QStyle::SP_DriveHDIcon); case Folder: return d->getIcon(QStyle::SP_DirIcon); case File: return d->getIcon(QStyle::SP_FileIcon); default: break; }; return QIcon(); } #ifdef Q_WS_WIN QIcon QFileIconProviderPrivate::getWinIcon(const QFileInfo &fileInfo) const { QIcon retIcon; QString fileExtension = fileInfo.suffix().toUpper(); fileExtension.prepend( QLatin1String(".") ); QString key; if (fileInfo.isFile() && !fileInfo.isExecutable() && !fileInfo.isSymLink()) key = QLatin1String("qt_") + fileExtension; QPixmap pixmap; if (!key.isEmpty()) { QPixmapCache::find(key, pixmap); } if (!pixmap.isNull()) { retIcon.addPixmap(pixmap); if (QPixmapCache::find(key + QLatin1Char('l'), pixmap)) retIcon.addPixmap(pixmap); return retIcon; } /* We don't use the variable, but by storing it statically, we * ensure CoInitialize is only called once. */ static HRESULT comInit = CoInitialize(NULL); Q_UNUSED(comInit); SHFILEINFO info; unsigned long val = 0; //Get the small icon #ifndef Q_OS_WINCE val = SHGetFileInfo((const WCHAR *)QDir::toNativeSeparators(fileInfo.filePath()).utf16(), 0, &info, sizeof(SHFILEINFO), SHGFI_ICON|SHGFI_SMALLICON|SHGFI_SYSICONINDEX|SHGFI_ADDOVERLAYS); #else val = SHGetFileInfo((const WCHAR *)QDir::toNativeSeparators(fileInfo.filePath()).utf16(), 0, &info, sizeof(SHFILEINFO), SHGFI_SMALLICON|SHGFI_SYSICONINDEX); #endif if (val) { if (fileInfo.isDir() && !fileInfo.isRoot()) { //using the unique icon index provided by windows save us from duplicate keys key = QString::fromLatin1("qt_dir_%1").arg(info.iIcon); QPixmapCache::find(key, pixmap); if (!pixmap.isNull()) { retIcon.addPixmap(pixmap); if (QPixmapCache::find(key + QLatin1Char('l'), pixmap)) retIcon.addPixmap(pixmap); DestroyIcon(info.hIcon); return retIcon; } } if (pixmap.isNull()) { #ifndef Q_OS_WINCE pixmap = convertHIconToPixmap(info.hIcon); #else pixmap = convertHIconToPixmap(ImageList_GetIcon((HIMAGELIST) val, info.iIcon, ILD_NORMAL)); #endif if (!pixmap.isNull()) { retIcon.addPixmap(pixmap); if (!key.isEmpty()) QPixmapCache::insert(key, pixmap); } else { qWarning("QFileIconProviderPrivate::getWinIcon() no small icon found"); } } DestroyIcon(info.hIcon); } //Get the big icon #ifndef Q_OS_WINCE val = SHGetFileInfo((const WCHAR *)QDir::toNativeSeparators(fileInfo.filePath()).utf16(), 0, &info, sizeof(SHFILEINFO), SHGFI_ICON|SHGFI_LARGEICON|SHGFI_SYSICONINDEX|SHGFI_ADDOVERLAYS); #else val = SHGetFileInfo((const WCHAR *)QDir::toNativeSeparators(fileInfo.filePath()).utf16(), 0, &info, sizeof(SHFILEINFO), SHGFI_LARGEICON|SHGFI_SYSICONINDEX); #endif if (val) { if (fileInfo.isDir() && !fileInfo.isRoot()) { //using the unique icon index provided by windows save us from duplicate keys key = QString::fromLatin1("qt_dir_%1").arg(info.iIcon); } #ifndef Q_OS_WINCE pixmap = convertHIconToPixmap(info.hIcon); #else pixmap = convertHIconToPixmap(ImageList_GetIcon((HIMAGELIST) val, info.iIcon, ILD_NORMAL), true); #endif if (!pixmap.isNull()) { retIcon.addPixmap(pixmap); if (!key.isEmpty()) QPixmapCache::insert(key + QLatin1Char('l'), pixmap); } else { qWarning("QFileIconProviderPrivate::getWinIcon() no large icon found"); } DestroyIcon(info.hIcon); } return retIcon; } #elif defined(Q_WS_MAC) QIcon QFileIconProviderPrivate::getMacIcon(const QFileInfo &fi) const { QIcon retIcon; QString fileExtension = fi.suffix().toUpper(); fileExtension.prepend(QLatin1String(".")); const QString keyBase = QLatin1String("qt_") + fileExtension; QPixmap pixmap; if (fi.isFile() && !fi.isExecutable() && !fi.isSymLink()) { QPixmapCache::find(keyBase + QLatin1String("16"), pixmap); } if (!pixmap.isNull()) { retIcon.addPixmap(pixmap); if (QPixmapCache::find(keyBase + QLatin1String("32"), pixmap)) { retIcon.addPixmap(pixmap); if (QPixmapCache::find(keyBase + QLatin1String("64"), pixmap)) { retIcon.addPixmap(pixmap); if (QPixmapCache::find(keyBase + QLatin1String("128"), pixmap)) { retIcon.addPixmap(pixmap); return retIcon; } } } } FSRef macRef; OSStatus status = FSPathMakeRef(reinterpret_cast(fi.canonicalFilePath().toUtf8().constData()), &macRef, 0); if (status != noErr) return retIcon; FSCatalogInfo info; HFSUniStr255 macName; status = FSGetCatalogInfo(&macRef, kIconServicesCatalogInfoMask, &info, &macName, 0, 0); if (status != noErr) return retIcon; IconRef iconRef; SInt16 iconLabel; status = GetIconRefFromFileInfo(&macRef, macName.length, macName.unicode, kIconServicesCatalogInfoMask, &info, kIconServicesNormalUsageFlag, &iconRef, &iconLabel); if (status != noErr) return retIcon; extern void qt_mac_constructQIconFromIconRef(const IconRef, const IconRef, QIcon*, QStyle::StandardPixmap = QStyle::SP_CustomBase); // qmacstyle_mac.cpp qt_mac_constructQIconFromIconRef(iconRef, 0, &retIcon); ReleaseIconRef(iconRef); pixmap = retIcon.pixmap(16); QPixmapCache::insert(keyBase + QLatin1String("16"), pixmap); pixmap = retIcon.pixmap(32); QPixmapCache::insert(keyBase + QLatin1String("32"), pixmap); pixmap = retIcon.pixmap(64); QPixmapCache::insert(keyBase + QLatin1String("64"), pixmap); pixmap = retIcon.pixmap(128); QPixmapCache::insert(keyBase + QLatin1String("128"), pixmap); return retIcon; } #endif /*! Returns an icon for the file described by \a info. */ QIcon QFileIconProvider::icon(const QFileInfo &info) const { Q_D(const QFileIconProvider); #ifdef Q_WS_MAC QIcon retIcon = d->getMacIcon(info); if (!retIcon.isNull()) return retIcon; #elif defined Q_WS_WIN QIcon icon = d->getWinIcon(info); if (!icon.isNull()) return icon; #endif if (info.isRoot()) #if defined (Q_WS_WIN) && !defined(Q_OS_WINCE) { uint type = DRIVE_UNKNOWN; QT_WA({ type = GetDriveTypeW((wchar_t *)info.absoluteFilePath().utf16()); }, { type = GetDriveTypeA(info.absoluteFilePath().toLocal8Bit()); }); switch (type) { case DRIVE_REMOVABLE: return d->getIcon(QStyle::SP_DriveFDIcon); case DRIVE_FIXED: return d->getIcon(QStyle::SP_DriveHDIcon); case DRIVE_REMOTE: return d->getIcon(QStyle::SP_DriveNetIcon); case DRIVE_CDROM: return d->getIcon(QStyle::SP_DriveCDIcon); case DRIVE_RAMDISK: case DRIVE_UNKNOWN: case DRIVE_NO_ROOT_DIR: default: return d->getIcon(QStyle::SP_DriveHDIcon); } } #else return d->getIcon(QStyle::SP_DriveHDIcon); #endif if (info.isFile()) { if (info.isSymLink()) return d->getIcon(QStyle::SP_FileLinkIcon); else return d->getIcon(QStyle::SP_FileIcon); } if (info.isDir()) { if (info.isSymLink()) { return d->getIcon(QStyle::SP_DirLinkIcon); } else { if (info.absoluteFilePath() == d->homePath) { return d->getIcon(QStyle::SP_DirHomeIcon); } else { return d->getIcon(QStyle::SP_DirIcon); } } } return QIcon(); } /*! Returns the type of the file described by \a info. */ QString QFileIconProvider::type(const QFileInfo &info) const { if (info.isRoot()) return QApplication::translate("QFileDialog", "Drive"); if (info.isFile()) { if (!info.suffix().isEmpty()) return info.suffix() + QLatin1Char(' ') + QApplication::translate("QFileDialog", "File"); return QApplication::translate("QFileDialog", "File"); } if (info.isDir()) return QApplication::translate("QFileDialog", #ifdef Q_WS_WIN "File Folder", "Match Windows Explorer" #else "Folder", "All other platforms" #endif ); // Windows - "File Folder" // OS X - "Folder" // Konqueror - "Folder" // Nautilus - "folder" if (info.isSymLink()) return QApplication::translate("QFileDialog", #ifdef Q_OS_MAC "Alias", "Mac OS X Finder" #else "Shortcut", "All other platforms" #endif ); // OS X - "Alias" // Windows - "Shortcut" // Konqueror - "Folder" or "TXT File" i.e. what it is pointing to // Nautilus - "link to folder" or "link to object file", same as Konqueror return QApplication::translate("QFileDialog", "Unknown"); } QT_END_NAMESPACE #endif