1 files changed, 1 insertions, 67 deletions
diff --git a/newmoviewizard.cpp b/newmoviewizard.cpp
index 79674dd..bab8e71 100644
--- a/newmoviewizard.cpp
+++ b/newmoviewizard.cpp
@@ -29,7 +29,7 @@
 #include <QDirIterator>
 
 #include "newmoviewizard.h"
-#include "smtreeitem.h"
+#include "wizardtreemodel.h"
 #include "smglobals.h"
 #include "mappingtablemodel.h"
 #include "delegates.h"
@@ -652,69 +652,3 @@ void MovieMetadataPage::initializePage(){
         mWidget->setMetadata(curMetadata);
     }
 }
-
-WizardTreeModel::WizardTreeModel(QStringList &headers, QObject *parent) : SmTreeModel(headers, parent){
-    mFiletypeMap = SmGlobals::instance()->filetypeMap();
-}
-
-Qt::ItemFlags WizardTreeModel::flags(const QModelIndex &index) const{
-    if(index.column() == FileType || index.column() == FilePart){
-        return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;
-    }
-	return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
-}
-
-QList<QVariant> WizardTreeModel::fileData(const QModelIndex &idx) const{
-    SmTreeItem *item = itemAt(idx);
-    QList<QVariant> retval;
-    for(int i = 0; i < item->columnCount(); ++i){
-        retval << item->data(i);
-    }
-    return retval;
-}
-
-void WizardTreeModel::clear(){
-    SmTreeItem *rootItem = new SmTreeItem(NumFields);
-    setRoot(rootItem);
-}
-
-QVariant WizardTreeModel::data(const QModelIndex &index, int role) const{
-	SmTreeItem *item = static_cast<SmTreeItem*>(index.internalPointer());
-	if(role == Qt::TextAlignmentRole){
-		if(index.column() == FileSize){
-			return Qt::AlignRight;
-		}
-		return Qt::AlignLeft;
-	}
-	if(role == FileNameRole){
-		return item->data(FileName);
-	}
-	if(role == FileSizeRole){
-		return item->data(FileSize);
-	}
-	if(role == FileTypeRole){
-		return item->data(FileType);
-	}
-	if(role == FullPathRole){
-		return item->data(FullPath);
-	}
-	if(role == FilePartRole){
-		return item->data(FilePart);
-	}
-    return SmTreeModel::data(index, role);
-}
-
-bool WizardTreeModel::setData(const QModelIndex &index, const QVariant &value, int role){
-    if(role == Qt::EditRole){
-        SmTreeItem *item = itemAt(index);
-        if(index.column() == FileType){
-            QVariant realVal = mFiletypeMap.key(value.toString());
-            item->setData(index.column(), realVal);
-        }else{
-            item->setData(index.column(), value);
-        }
-        emit dataChanged(index, index);
-        return true;
-    }
-    return false;
-}