1 files changed, 7 insertions, 0 deletions
diff --git a/smtreemodel.cpp b/smtreemodel.cpp
index a39f718..2712cef 100644
--- a/smtreemodel.cpp
+++ b/smtreemodel.cpp
@@ -12,6 +12,9 @@
 
 SmTreeModel::SmTreeModel(const QStringList &headers, QObject *parent) : QAbstractItemModel(parent), mRootItem(0){
 	mHeaders = headers;
+	for(int i = 0; i < mHeaders.size(); ++i){
+		mHeaderData.insert(mHeaders.at(i), i);
+	}
 	mRootItem = new SmTreeItem(headers.count());
 }
 
@@ -69,6 +72,10 @@ QVariant SmTreeModel::headerData(int section, Qt::Orientation orientation, int r
 	return QVariant();
 }
 
+const QHash<QString, int> SmTreeModel::headerData() const{
+	return mHeaderData;
+}
+
 bool SmTreeModel::setHeaderData(int section, Qt::Orientation orientation, const QVariant &value, int role){
 	if((orientation == Qt::Horizontal) && (role == Qt::DisplayRole)){
 		mRootItem->setData(section, value);