kworship/playlist/KwPlaylistModel.cpp

   1 /**
   2  * @file KwPlaylistModel.cpp
   3  * @brief A Qt model for playlist items.
   4  * @author James Hogan <james@albanarts.com>
   5  */
   6
   7 #include "KwPlaylistModel.h"
   8 #include "KwPlaylistNode.h"
   9
  10 #include <cassert>
  11
  12 /*
  13  * Constructors + destructor.
  14  */
  15
  16 /// Default constructor.
  17 KwPlaylistModel::KwPlaylistModel(QObject* parent)
  18 : QAbstractItemModel(parent)
  19 , m_root(0)
  20 {
  21 }
  22
  23 /// Destructor.
  24 KwPlaylistModel::~KwPlaylistModel()
  25 {
  26   delete m_root;
  27 }
  28
  29 /*
  30  * Main interface
  31  */
  32
  33 /// Set the root node.
  34 void KwPlaylistModel::setRootNode(KwPlaylistNode* root)
  35 {
  36   delete m_root;
  37   m_root = root;
  38   reset();
  39 }
  40
  41 KwPlaylistNode* KwPlaylistModel::itemFromIndex(const QModelIndex& index) const
  42 {
  43   if (index.isValid()) {
  44     return reinterpret_cast<KwPlaylistNode*>(index.internalPointer());
  45   } else {
  46     return m_root;
  47   }
  48 }
  49
  50 QModelIndex KwPlaylistModel::index(int row, int column, const QModelIndex& parent) const
  51 {
  52   if (0 == m_root)
  53   {
  54     return QModelIndex();
  55   }
  56   KwPlaylistNode* parentNode = itemFromIndex(parent);
  57   assert(0 != parentNode);
  58   return createIndex(row, column, parentNode->getChild(row));
  59 }
  60
  61 QModelIndex KwPlaylistModel::parent(const QModelIndex& child) const
  62 {
  63   KwPlaylistNode* node = itemFromIndex(child);
  64   assert(0 != node);
  65   KwPlaylistNode* parentNode = node->getParent();
  66   if (0 == parentNode)
  67   {
  68     return QModelIndex();
  69   }
  70   KwPlaylistNode* grandParentNode = parentNode->getParent();
  71   if (0 == grandParentNode)
  72   {
  73     return QModelIndex();
  74   }
  75   int row = grandParentNode->getChildIndex(parentNode);
  76   assert(row != -1);
  77   return createIndex(row, child.column(), parentNode);
  78 }
  79
  80 int KwPlaylistModel::rowCount(const QModelIndex& parent) const
  81 {
  82   KwPlaylistNode* parentNode = itemFromIndex(parent);
  83   if (0 == parentNode)
  84   {
  85     return 0;
  86   }
  87   return parentNode->getChildCount();
  88 }
  89
  90 int KwPlaylistModel::columnCount(const QModelIndex& parent) const
  91 {
  92   Q_UNUSED(parent)
  93   return 2;
  94 }
  95
  96 QVariant KwPlaylistModel::data(const QModelIndex& index, int role) const
  97 {
  98   KwPlaylistNode* item = itemFromIndex(index);
  99   if (0 == item)
 100   {
 101     return QVariant();
 102   }
 103   return item->getData(role, index.column());
 104 }
 105
 106 QVariant KwPlaylistModel::headerData(int section, Qt::Orientation orientation, int role) const
 107 {
 108   if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
 109   {
 110     if (section == 0)
 111     {
 112       return tr("Node");
 113     }
 114     else if (section == 1)
 115     {
 116       return tr("Value");
 117     }
 118   }
 119   return QVariant();
 120 }
 121