diff -r 2a40e88564c8 -r 4e84c994a771 mpviewplugins/mpcollectionviewplugin/src/mpcollectiondatamodel.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mpviewplugins/mpcollectionviewplugin/src/mpcollectiondatamodel.cpp	Fri Mar 19 09:28:13 2010 +0200
@@ -0,0 +1,242 @@
+/*
+* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
+* All rights reserved.
+* This component and the accompanying materials are made available
+* under the terms of "Eclipse Public License v1.0"
+* which accompanies this distribution, and is available
+* at the URL "http://www.eclipse.org/legal/epl-v10.html".
+*
+* Initial Contributors:
+* Nokia Corporation - initial contribution.
+*
+* Contributors:
+*
+* Description: Music Player collection data model.
+*
+*/
+
+#include <QBrush>
+#include <QColor>
+
+#include <hbicon.h>
+
+#include "mpcollectiondatamodel.h"
+#include "mpcollectionalbumartmanager.h"
+#include "mpmpxcollectiondata.h"
+#include "mptrace.h"
+
+const int KInitCacheSize = 11;
+
+/*!
+    \class MpCollectionDataModel
+    \brief Music Player collection data model.
+
+    Collection data model implements the interface specified by HbAbstractDataModel,
+    which defines the standard interface that item models must use to be able to
+    interoperate with other components in the model/view architecture.
+
+    Every item of data that can be accessed via a model has an associated model
+    index.
+
+    Each item has a number of data elements associated with it and they can be
+    retrieved by specifying a role (see Qt::ItemDataRole) to the model's data
+    returned by itemData() function.
+
+    \sa HbAbstractDataModel
+*/
+
+/*!
+ Constructs the collection data model.
+ */
+MpCollectionDataModel::MpCollectionDataModel( MpMpxCollectionData *data, QObject *parent )
+    : QAbstractListModel(parent),
+      mCollectionData(data),
+      mRowCount(0),
+      mCachingInProgress(false)
+{
+    TX_ENTRY
+    mAlbumArtManager = new MpCollectionAlbumArtManager();
+    connect( mAlbumArtManager, SIGNAL(albumArtReady(int)), this, SLOT(updateAlbumArt(int)) );
+    connect( mAlbumArtManager, SIGNAL(albumCacheReady()), this, SLOT(albumCacheReady()) );
+    TX_EXIT
+}
+
+/*!
+ Destructs the collection data model.
+ */
+MpCollectionDataModel::~MpCollectionDataModel()
+{
+    TX_ENTRY
+    delete mAlbumArtManager;
+    TX_EXIT
+}
+
+/*!
+ Returns the number of rows under the given \a parent.
+
+ View will request for the row count immediately after a model is set.
+ To prevent it from reading data while caching the album art for the first
+ screen, return row count as zero.
+
+ \reimp
+ */
+int MpCollectionDataModel::rowCount( const QModelIndex &parent ) const
+{
+    TX_LOG
+    Q_UNUSED(parent);
+    if ( mCachingInProgress ) {
+        return 0;
+    }
+    else {
+        return mRowCount;
+    }
+}
+
+/*!
+ Returns the data stored for the item referred to by the \a index.
+
+ \reimp
+ */
+QVariant MpCollectionDataModel::data(const QModelIndex &index, int role) const
+{
+    TX_ENTRY
+    QVariant returnValue = QVariant();
+    if ( !index.isValid() ) {
+        return returnValue;
+    }
+
+    int row = index.row();
+    TX_LOG_ARGS("index=" << row << ", role=" << role);
+    TCollectionContext context = mCollectionData->context();
+    if ( role == Qt::DisplayRole ) {
+        QStringList display;
+        // Fetch the primary text, which is the title, if available.
+        QString primaryText;
+        primaryText = mCollectionData->itemData(row, MpMpxCollectionData::Title);
+        if ( !primaryText.isEmpty() ) {
+            display << primaryText;
+        }
+        else {
+            display << QString( tr("Unknown") );
+        }
+
+        // Fetch the secondary text, which depends on the current context, if available.
+        QString secondaryText;
+        switch ( context ) {
+            case ECollectionContextAllSongs:
+            case ECollectionContextArtistAlbums:
+            case ECollectionContextPlaylistSongs:
+                secondaryText = mCollectionData->itemData(row, MpMpxCollectionData::Artist);
+                if ( !secondaryText.isEmpty() ) {
+                    display << secondaryText;
+                }
+                else {
+                    display << QString( tr("Unknown") );
+                }
+                break;
+            case ECollectionContextPlaylists:
+                secondaryText = mCollectionData->itemData(row, MpMpxCollectionData::Count);
+                if ( !secondaryText.isEmpty() ) {
+                    display << secondaryText;
+                }
+                break;
+        }
+        returnValue = display;
+    }
+    else if ( role == Qt::DecorationRole ) {
+        switch ( context ) {
+            case ECollectionContextArtistAlbums:
+                bool defaultArt = true;
+                QString albumArtUri = mCollectionData->itemData(row, MpMpxCollectionData::AlbumArtUri);
+                if ( !albumArtUri.isEmpty() ) {
+                     HbIcon icon = mAlbumArtManager->albumArt(albumArtUri, row);
+                     if ( !icon.isNull() ) {
+                         returnValue = icon;
+                         defaultArt = false;
+                     }
+                }
+                if ( defaultArt ) {
+                    // No album art, use default album art
+                    HbIcon icon(QString(":/icons/default_album.png"));
+                    returnValue = icon;
+                }
+                break;
+        }
+    }
+    TX_EXIT
+    return returnValue;
+}
+
+/*!
+ Must be called when data has changed and model needs to be refreshed
+ to reflect the new data.
+ */
+void MpCollectionDataModel::refreshModel()
+{
+    TX_ENTRY
+    // Cancel all outstanding album art request first, then reset the model.
+    mAlbumArtManager->cancel();
+    mRowCount = mCollectionData->count();
+
+    TCollectionContext context = mCollectionData->context();
+    if ( context == ECollectionContextArtistAlbums ) {
+        // Before providing the new data to the view (list, grid, etc.), we want
+        // to make sure that we have enough album arts for the first screen.
+        if ( mRowCount > 0 ) {
+            int initCount = ( mRowCount > KInitCacheSize ) ? KInitCacheSize : mRowCount;
+            QStringList albumArtList;
+            QString albumArtUri;
+            for ( int i = 0; i < initCount; i++ ) {
+                albumArtUri = mCollectionData->itemData(i, MpMpxCollectionData::AlbumArtUri);
+                if ( !albumArtUri.isEmpty() ) {
+                     albumArtList << albumArtUri;
+                }
+            }
+            mCachingInProgress = mAlbumArtManager->cacheAlbumArt(albumArtList);
+            if ( !mCachingInProgress ) {
+                reset();
+            }
+        }
+        else {
+            reset();
+        }
+    }
+    else {
+        reset();
+    }
+    TX_EXIT
+}
+
+/*!
+ Returns pointer to MpMpxCollectionData, which is the collection data.
+ */
+MpMpxCollectionData *MpCollectionDataModel::collectionData()
+{
+    return mCollectionData;
+}
+
+/*!
+ Slot to be called when album art for the \a index needs to be updated.
+ */
+void MpCollectionDataModel::updateAlbumArt( int index )
+{
+    TX_ENTRY_ARGS("index=" << index);
+    if ( index > 0 && index < mRowCount ) {
+        QModelIndex modelIndex = QAbstractItemModel::createIndex(index, 0);
+        emit dataChanged(modelIndex, modelIndex);
+    }
+    TX_EXIT
+}
+
+/*!
+ Slot to be called when album art cache is ready.
+ */
+void MpCollectionDataModel::albumCacheReady()
+{
+    TX_ENTRY
+    if ( mCachingInProgress ) {
+        mCachingInProgress = false;
+        reset();
+    }
+    TX_EXIT
+}