SeriesTreeModel::findMovies

Implemented function to return a QFileInfoList of only the movies for a
given QModelIndex.

1 files changed, 19 insertions, 0 deletions

diff --git a/seriestreemodel.cpp b/seriestreemodel.cpp
index 846f00c..122c163 100644
--- a/seriestreemodel.cpp
+++ b/seriestreemodel.cpp
@@ -180,6 +180,25 @@ QFileInfoList SeriesTreeModel::findFiles(const QModelIndex &where) const{
 	return retval;
 }
 
+QFileInfoList SeriesTreeModel::findMovies(const QModelIndexList &from) const{
+	if(from.isEmpty()){
+		return QFileInfoList();
+	}
+	QFileInfoList retval;
+	foreach(QModelIndex idx, from){
+		QFileInfoList files = findFiles(idx);
+		foreach(QFileInfo fi, files){
+			QString mime = Helper::mimeType(fi.absoluteFilePath());
+			if(mime.startsWith("video")){
+				if(!retval.contains(fi)){
+					retval << fi;
+				}
+			}
+		}
+	}
+	return retval;
+}
+
 bool SeriesTreeModel::deleteFromSeries(const QModelIndex &what){
 	int nodeType = what.data(TypeRole).toInt();
 	QSqlQuery *query = 0;