src/Model_FileList.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
   4 //
   5 // This program is free software; you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation; either version 2 of the License, or
   8 // (at your option) any later version.
   9 //
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14 //
  15 // You should have received a copy of the GNU General Public License along
  16 // with this program; if not, write to the Free Software Foundation, Inc.,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 // http://www.gnu.org/licenses/gpl-2.0.txt
  20 ///////////////////////////////////////////////////////////////////////////////
  21
  22 #include "Model_FileList.h"
  23
  24 #include <QFileInfo>
  25 #include <QDir>
  26
  27 ////////////////////////////////////////////////////////////
  28 // Constructor & Destructor
  29 ////////////////////////////////////////////////////////////
  30
  31 FileListModel::FileListModel(void)
  32 :
  33         m_fileIcon(":/icons/page_white_cd.png")
  34 {
  35 }
  36
  37 FileListModel::~FileListModel(void)
  38 {
  39 }
  40
  41 ////////////////////////////////////////////////////////////
  42 // Public Functions
  43 ////////////////////////////////////////////////////////////
  44
  45 int FileListModel::columnCount(const QModelIndex &parent) const
  46 {
  47         return 2;
  48 }
  49
  50 int FileListModel::rowCount(const QModelIndex &parent) const
  51 {
  52         return m_fileList.count();
  53 }
  54
  55 QVariant FileListModel::data(const QModelIndex &index, int role) const
  56 {
  57         if((role == Qt::DisplayRole || role == Qt::ToolTipRole) && index.row() < m_fileList.count() && index.row() >= 0)
  58         {
  59                 switch(index.column())
  60                 {
  61                 case 0:
  62                         return m_fileList.at(index.row()).fileName();
  63                         break;
  64                 case 1:
  65                         return QDir::toNativeSeparators(m_fileList.at(index.row()).filePath());
  66                         break;
  67                 default:
  68                         return QVariant();
  69                         break;
  70                 }
  71         }
  72         else if(role == Qt::DecorationRole && index.column() == 0)
  73         {
  74                 return m_fileIcon;
  75         }
  76         else
  77         {
  78                 return QVariant();
  79         }
  80 }
  81
  82 QVariant FileListModel::headerData(int section, Qt::Orientation orientation, int role) const
  83 {
  84         if(role == Qt::DisplayRole)
  85         {
  86                 if(orientation == Qt::Horizontal)
  87                 {
  88                         switch(section)
  89                         {
  90                         case 0:
  91                                 return QVariant(tr("Title"));
  92                                 break;
  93                         case 1:
  94                                 return QVariant(tr("Full Path"));
  95                                 break;
  96                         default:
  97                                 return QVariant();
  98                                 break;
  99                         }
 100                 }
 101                 else
 102                 {
 103                         if(m_fileList.count() < 10)
 104                         {
 105                                 return QVariant(QString().sprintf("%d", section + 1));
 106                         }
 107                         else if(m_fileList.count() < 100)
 108                         {
 109                                 return QVariant(QString().sprintf("%02d", section + 1));
 110                         }
 111                         else if(m_fileList.count() < 1000)
 112                         {
 113                                 return QVariant(QString().sprintf("%03d", section + 1));
 114                         }
 115                         else
 116                         {
 117                                 return QVariant(QString().sprintf("%04d", section + 1));
 118                         }
 119                 }
 120         }
 121         else
 122         {
 123                 return QVariant();
 124         }
 125 }
 126
 127 void FileListModel::addFile(const QString &filePath)
 128 {
 129         QFileInfo fileInfo(filePath);
 130
 131         for(int i = 0; i < m_fileList.count(); i++)
 132         {
 133                 if(m_fileList.at(i).filePath().compare(fileInfo.canonicalFilePath(), Qt::CaseInsensitive) == 0)
 134                 {
 135                         return;
 136                 }
 137         }
 138
 139         beginResetModel();
 140         m_fileList.append(AudioFileModel(fileInfo.canonicalFilePath(), fileInfo.baseName()));
 141         endResetModel();
 142 }
 143
 144 void FileListModel::addFile(const AudioFileModel &file)
 145 {
 146         for(int i = 0; i < m_fileList.count(); i++)
 147         {
 148                 if(m_fileList.at(i).filePath().compare(file.filePath(), Qt::CaseInsensitive) == 0)
 149                 {
 150                         return;
 151                 }
 152         }
 153
 154         beginResetModel();
 155         m_fileList.append(file);
 156         endResetModel();
 157 }
 158
 159 bool FileListModel::removeFile(const QModelIndex &index)
 160 {
 161         if(index.row() >= 0 && index.row() < m_fileList.count())
 162         {
 163                 beginResetModel();
 164                 m_fileList.removeAt(index.row());
 165                 endResetModel();
 166                 return true;
 167         }
 168         else
 169         {
 170                 return false;
 171         }
 172 }
 173
 174 void FileListModel::clearFiles(void)
 175 {
 176         beginResetModel();
 177         m_fileList.clear();
 178         endResetModel();
 179 }
 180
 181 bool FileListModel::moveFile(const QModelIndex &index, int delta)
 182 {
 183         if(delta != 0 && index.row() >= 0 && index.row() < m_fileList.count() && index.row() + delta >= 0 && index.row() + delta < m_fileList.count())
 184         {
 185                 beginResetModel();
 186                 m_fileList.move(index.row(), index.row() + delta);
 187                 endResetModel();
 188                 return true;
 189         }
 190         else
 191         {
 192                 return false;
 193         }
 194 }
 195
 196 AudioFileModel FileListModel::getFile(const QModelIndex &index)
 197 {
 198         if(index.row() >= 0 && index.row() < m_fileList.count())
 199         {
 200                 return m_fileList.at(index.row());
 201         }
 202         else
 203         {
 204                 return AudioFileModel();
 205         }
 206 }
 207
 208 AudioFileModel &FileListModel::operator[] (const QModelIndex &index)
 209 {
 210         return m_fileList[index.row()];
 211 }
 212
 213 bool FileListModel::setFile(const QModelIndex &index, const AudioFileModel &audioFile)
 214 {
 215         if(index.row() >= 0 && index.row() < m_fileList.count())
 216         {
 217                 beginResetModel();
 218                 m_fileList.replace(index.row(), audioFile);
 219                 endResetModel();
 220                 return true;
 221         }
 222         else
 223         {
 224                 return false;
 225         }
 226 }