/*
 *   Copyright 2008 Marco Martin <notmart@gmail.com>
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU Library General Public License as
 *   published by the Free Software Foundation; either version 2, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details
 *
 *   You should have received a copy of the GNU Library General Public
 *   License along with this program; if not, write to the
 *   Free Software Foundation, Inc.,
 *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#include "treeview.h"

#include <QTreeView>
#include <QHeaderView>
#include <QScrollBar>

#include <kiconloader.h>

#include "private/style_p.h"

namespace Plasma
{

class TreeViewPrivate
{
public:
    Plasma::Style *style;
};

TreeView::TreeView(QGraphicsWidget *parent)
    : QGraphicsProxyWidget(parent),
      d(new TreeViewPrivate)
{
    QTreeView *native = new QTreeView;
    setWidget(native);
    native->setAttribute(Qt::WA_NoSystemBackground);
    native->setFrameStyle(QFrame::NoFrame);

    d->style = new Plasma::Style();
    native->verticalScrollBar()->setStyle(d->style);
    native->horizontalScrollBar()->setStyle(d->style);
}

TreeView::~TreeView()
{
    widget()->setStyle(0);
    delete d->style;
    delete d;
}

void TreeView::setModel(QAbstractItemModel *model)
{
    nativeWidget()->setModel(model);
}

QAbstractItemModel *TreeView::model()
{
    return nativeWidget()->model();
}

void TreeView::setStyleSheet(const QString &stylesheet)
{
    widget()->setStyleSheet(stylesheet);
}

QString TreeView::styleSheet()
{
    return widget()->styleSheet();
}

QTreeView *TreeView::nativeWidget() const
{
    return static_cast<QTreeView*>(widget());
}

}

#include <treeview.moc>