directoryloader.cpp 3.56 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/**
 * Copyright (C) 2018 Michael Pyne <mpyne@kde.org>
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option) any later
 * version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE. See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "directoryloader.h"

#include <QFileInfo>
20 21
#include <QMutex>
#include <QMutexLocker>
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113

#include "mediafiles.h"

// Classifies files into types for potential loading purposes.
enum class MediaFileType {
    UnusableFile,
    MediaFile,
    Playlist,
    Directory
};

static MediaFileType classifyFile(const QFileInfo &fileInfo);
static FileHandle loadMediaFile(const QString &fileName);

DirectoryLoader::DirectoryLoader(const QString &dir, QObject *parent)
    : QObject(parent)
    , m_dirIterator(
        dir,
        QDir::AllEntries | QDir::NoDotAndDotDot,
        QDirIterator::Subdirectories | QDirIterator::FollowSymlinks)
{
}

void DirectoryLoader::startLoading()
{
    static const int BATCH_SIZE = 256;
    FileHandleList files;

    while(m_dirIterator.hasNext()) {
        const auto fileName = m_dirIterator.next();
        const QFileInfo fileInfo(fileName);
        const auto type = classifyFile(fileInfo);

        switch(type) {
            case MediaFileType::Playlist:
                emit loadedPlaylist(fileName);
                break;

            case MediaFileType::MediaFile:
                {
                    const auto loadedMetadata = loadMediaFile(fileInfo.canonicalFilePath());
                    files << loadedMetadata;

                    if(files.count() >= BATCH_SIZE) {
                        emit loadedFiles(files);
                        files.clear();
                    }
                }
                break;

            case MediaFileType::Directory:
                // this should be impossible based on the
                // QDirIterator settings
                continue;
            case MediaFileType::UnusableFile:
                continue;
            default:
                continue;
        }
    }

    if(!files.isEmpty()) {
        emit loadedFiles(files);
    }
}

MediaFileType classifyFile(const QFileInfo &fileInfo)
{
    const QString path = fileInfo.canonicalFilePath();

    if(fileInfo.isFile() && fileInfo.isReadable() &&
        MediaFiles::isMediaFile(path))
    {
        return MediaFileType::MediaFile;
    }

    // These are all the files we care about, anything remaining needs to be a
    // directory or playlist or it must be unusable

    if(MediaFiles::isPlaylistFile(path)) {
        return MediaFileType::Playlist;
    }

    if(fileInfo.isDir()) {
        return MediaFileType::Directory;
    }

    return MediaFileType::UnusableFile;
}

FileHandle loadMediaFile(const QString &fileName)
{
114 115 116 117 118 119 120
    // Just because our GUI thread accesses are serialized by signal/slot magic
    // doesn't mean that our non-GUI threads don't require deconfliction
    // Neither FileHandle nor TagLib are not thread-safe so synchronize access
    // to this function.
    static QMutex fhLock;
    QMutexLocker locker(&fhLock);

121 122 123 124 125
    FileHandle loadedMetadata(fileName);
    (void) loadedMetadata.tag(); // Ensure tag is read

    return loadedMetadata;
}