[vlmc-devel] Add a Media Library model

[Hugo Beauzée-Luyssen]

vlmc | branch: medialibrary | Hugo Beauzée-Luyssen <hugo at beauzee.fr> | Wed Jun 29 15:33:40 2016 +0200| [7585b6603b24fa78a0c9f84072e35ac1e1107b78] | committer: Hugo Beauzée-Luyssen

Add a Media Library model

> https://code.videolan.org/videolan/vlmc/commit/7585b6603b24fa78a0c9f84072e35ac1e1107b78
---

 src/CMakeLists.txt                |   1 +
 src/Library/MediaLibraryModel.cpp | 133 ++++++++++++++++++++++++++++++++++++++
 src/Library/MediaLibraryModel.h   |  65 +++++++++++++++++++
 3 files changed, 199 insertions(+)

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 3c9f4df..e611efd 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -86,6 +86,7 @@ SET(VLMC_SRCS
     Library/Library.cpp
     Library/MediaContainer.cpp
     Library/MediaLibrary.cpp
+    Library/MediaLibraryModel.cpp
     Main/Core.cpp
     Main/main.cpp
     Media/Clip.cpp
diff --git a/src/Library/MediaLibraryModel.cpp b/src/Library/MediaLibraryModel.cpp
new file mode 100644
index 0000000..f0f8617
--- /dev/null
+++ b/src/Library/MediaLibraryModel.cpp
@@ -0,0 +1,133 @@
+/*****************************************************************************
+ * MediaLibraryModel.cpp
+ *****************************************************************************
+ * Copyright (C) 2008-2016 VideoLAN
+ *
+ * Authors: Hugo Beauzée-Luyssen <hugo at beauzee.fr>
+ *
+ * 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, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
+ *****************************************************************************/
+
+#include "config.h"
+
+#include "MediaLibraryModel.h"
+
+#ifdef WITH_GUI
+#include <QPixmap>
+#endif
+
+MediaLibraryModel::MediaLibraryModel( medialibrary::IMediaLibrary& ml, medialibrary::IMedia::Type type, QObject *parent )
+    : QAbstractListModel(parent)
+    , m_ml( ml )
+    , m_mediaType( type )
+{
+    switch (type)
+    {
+    case medialibrary::IMedia::Type::AudioType:
+        m_media = m_ml.audioFiles();
+        break;
+    case medialibrary::IMedia::Type::VideoType:
+        m_media = m_ml.videoFiles();
+        break;
+    default:
+        Q_UNREACHABLE();
+    }
+    if ( m_media.size() == 0 )
+        return;
+    beginInsertRows( QModelIndex(), 0, m_media.size() );
+    m_rowCount = m_media.size();
+    insertRows( 0, m_media.size() );
+    endInsertRows();
+}
+
+void MediaLibraryModel::addMedia( medialibrary::MediaPtr media )
+{
+    if ( media->type() != m_mediaType )
+        return;
+    std::lock_guard<std::mutex> lock( m_mediaMutex );
+    auto size = m_media.size();
+    beginInsertRows( QModelIndex(), size, size );
+    m_media.push_back( media );
+    m_rowCount.fetch_add( 1, std::memory_order_relaxed );
+    insertRow( size );
+    endInsertRows();
+}
+
+void MediaLibraryModel::removeMedia( medialibrary::MediaPtr media )
+{
+    if ( media->type() != m_mediaType )
+        return;
+    std::lock_guard<std::mutex> lock( m_mediaMutex );
+    auto it = std::find_if( begin( m_media ), end( m_media ), [media](medialibrary::MediaPtr m) {
+        return m->id() == media->id();
+    });
+    if ( it == end( m_media ) )
+        return;
+    auto idx = it - begin( m_media );
+    beginRemoveRows(QModelIndex(), idx, idx );
+    m_media.erase( it );
+    m_rowCount.fetch_sub( 1, std::memory_order_relaxed );
+    removeRow( idx );
+    endRemoveRows();
+}
+
+int MediaLibraryModel::rowCount( const QModelIndex& ) const
+{
+    return m_rowCount.load( std::memory_order_relaxed );
+}
+
+QVariant MediaLibraryModel::data( const QModelIndex &index, int role ) const
+{
+    medialibrary::MediaPtr m;
+    {
+        std::lock_guard<std::mutex> lock( m_mediaMutex );
+
+        if ( !index.isValid() || index.row() < 0 ||
+             static_cast<size_t>( index.row() ) >= m_media.size() )
+            return QVariant();
+        m = m_media.at( static_cast<size_t>( index.row() ) );
+    }
+
+    switch ( role )
+    {
+    case Qt::DisplayRole:
+    case Roles::Title:
+        return QVariant( QString::fromStdString( m->title() ) );
+#ifdef WITH_GUI
+    case Qt::DecorationRole:
+        return QPixmap( QString::fromStdString( m->thumbnail() ) );
+#endif
+    case Roles::ThumbnailPath:
+        return QVariant( QString::fromStdString( m->thumbnail() ) );
+    case Roles::Duration:
+        return QVariant::fromValue( m->duration() );
+    case Qt::UserRole:
+        return QVariant::fromValue( m );
+    default:
+        break;
+    }
+
+    return QVariant();
+}
+
+QHash<int, QByteArray>
+MediaLibraryModel::roleNames() const
+{
+    return {
+        { Roles::Title, "title" },
+        { Roles::ThumbnailPath, "thumbnailPath" },
+        { Roles::Duration, "duration" }
+    };
+}
diff --git a/src/Library/MediaLibraryModel.h b/src/Library/MediaLibraryModel.h
new file mode 100644
index 0000000..12eff3f
--- /dev/null
+++ b/src/Library/MediaLibraryModel.h
@@ -0,0 +1,65 @@
+/*****************************************************************************
+ * MediaLibraryModel.h
+ *****************************************************************************
+ * Copyright (C) 2008-2016 VideoLAN
+ *
+ * Authors: Hugo Beauzée-Luyssen <hugo at beauzee.fr>
+ *
+ * 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, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
+ *****************************************************************************/
+
+#ifndef MEDIALIBRARYMODEL_H
+#define MEDIALIBRARYMODEL_H
+
+#include <QAbstractListModel>
+
+#include <atomic>
+#include <mutex>
+
+#include <medialibrary/IMediaLibrary.h>
+#include <medialibrary/IMedia.h>
+
+class MediaLibraryModel : public QAbstractListModel
+{
+public:
+    enum Roles
+    {
+        Title = Qt::UserRole + 1,
+        ThumbnailPath,
+        Duration,
+    };
+
+    explicit MediaLibraryModel( medialibrary::IMediaLibrary& ml, medialibrary::IMedia::Type type,
+                                QObject *parent = 0 );
+
+    void addMedia( medialibrary::MediaPtr media );
+    void removeMedia( medialibrary::MediaPtr media );
+    int rowCount( const QModelIndex &parent = QModelIndex() ) const override;
+    QVariant data( const QModelIndex &index, int role = Qt::DisplayRole ) const override;
+    QHash<int, QByteArray> roleNames() const;
+
+private:
+    medialibrary::IMediaLibrary& m_ml;
+    const medialibrary::IMedia::Type m_mediaType;
+
+    // Use an atomic int to avoid locking m_media within rowCount()
+    mutable std::mutex m_mediaMutex;
+    std::atomic_int m_rowCount;
+    std::vector<medialibrary::MediaPtr> m_media;
+};
+
+Q_DECLARE_METATYPE( medialibrary::MediaPtr );
+
+#endif // MEDIALIBRARYMODEL_H