src/collectionbrowser/CollectionTreeItem.h

   1 /******************************************************************************
   2  * copyright: (c) 2007 Alexandre Pereira de Oliveira <aleprj@gmail.com>       *
   3  *            (c) 2007 Maximilian Kossick <maximilian.kossick@googlemail.com> *
   4  *   This program is free software; you can redistribute it and/or modify     *
   5  *   it under the terms of the GNU General Public License version 2           *
   6  *   as published by the Free Software Foundation.                            *
   7  ******************************************************************************/
   8
   9 #ifndef COLLECTIONTREEITEM_H
  10 #define COLLECTIONTREEITEM_H
  11
  12 #include "meta/Meta.h"
  13 #include "Collection.h"
  14
  15 #include <QList>
  16
  17 namespace CustomRoles
  18 {
  19     enum CustomRolesId {
  20     SortRole = Qt::UserRole + 1,
  21     FilterRole = Qt::UserRole + 2
  22     };
  23 }
  24
  25 class CollectionTreeItem {
  26     public:
  27         CollectionTreeItem( Meta::DataPtr data, CollectionTreeItem *parent );
  28         CollectionTreeItem( Collection *parentCollection, CollectionTreeItem *parent );
  29         //this ctor creates a "Various Artists" node. do not use it for anything else
  30         CollectionTreeItem( const Meta::DataList &data, CollectionTreeItem *parent );
  31
  32         ~CollectionTreeItem();
  33
  34         CollectionTreeItem* parent()  { return m_parent; }
  35
  36         void appendChild(CollectionTreeItem *child);
  37         void removeChild( int index );
  38
  39         CollectionTreeItem *child(int row);
  40
  41         int childCount() const { return m_childItems.count(); }
  42         int columnCount() const { return 1; }
  43
  44         QVariant data(int role) const;
  45
  46         int row() const;
  47
  48         int level() const;
  49
  50         bool isDataItem() const;
  51
  52         QueryMaker* queryMaker() const;
  53
  54         bool operator<( const CollectionTreeItem& other ) const;
  55         bool childrenLoaded() const { return  m_childrenLoaded; }
  56         void setChildrenLoaded( bool childrenLoaded );
  57
  58         const Meta::DataPtr data() const { return m_data; }
  59         Collection* parentCollection() const { return m_parentCollection; }
  60
  61         KUrl::List urls() const;
  62         QList<Meta::TrackPtr> descendentTracks();
  63
  64         bool allDescendentTracksLoaded() const;
  65
  66     private:
  67         Meta::DataPtr m_data;
  68         CollectionTreeItem *m_parent;
  69         Collection* m_parentCollection;
  70
  71         QList<CollectionTreeItem*> m_childItems;
  72         bool m_childrenLoaded;
  73         bool m_isVariousArtistsNode;
  74 };
  75
  76 #endif