kde-playground/kdepim/akonadiconsole/jobtrackerwidget.cpp
2015-04-14 21:49:29 +00:00

135 lines
3.9 KiB
C++

/*
This file is part of Akonadi.
Copyright (c) 2009 Till Adam <adam@kde.org>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, 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 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 "jobtrackerwidget.h"
#include <QCheckBox>
#include "jobtrackermodel.h"
#include <akonadi/control.h>
#include <KLocalizedString>
#include <KFileDialog>
#include <QTreeView>
#include <QHeaderView>
#include <QVBoxLayout>
#include <QMenu>
#include <QPushButton>
#include <QFile>
class JobTrackerWidget::Private
{
public:
JobTrackerModel* model;
};
JobTrackerWidget::JobTrackerWidget( const char *name, QWidget *parent, const QString& checkboxText )
: QWidget( parent ),
d( new Private )
{
d->model = new JobTrackerModel( name, this );
QVBoxLayout *layout = new QVBoxLayout( this );
QCheckBox* enableCB = new QCheckBox( this );
enableCB->setText(checkboxText);
connect( enableCB, SIGNAL(toggled(bool)), d->model, SLOT(setEnabled(bool)) );
layout->addWidget(enableCB);
QTreeView *tv = new QTreeView( this );
tv->setModel( d->model );
tv->expandAll();
tv->setAlternatingRowColors( true );
tv->setContextMenuPolicy( Qt::CustomContextMenu );
// too slow with many jobs:
// tv->header()->setResizeMode( QHeaderView::ResizeToContents );
connect( d->model, SIGNAL(modelReset()), tv, SLOT(expandAll()) );
connect( tv, SIGNAL(customContextMenuRequested(QPoint)), SLOT(contextMenu(QPoint)) );
layout->addWidget( tv );
d->model->setEnabled( false ); // since it can be slow, default to off
QHBoxLayout *layout2 = new QHBoxLayout( this );
QPushButton *button = new QPushButton( QLatin1String( "Save to file..." ), this );
connect( button, SIGNAL(clicked(bool)),
this, SLOT(slotSaveToFile()) );
layout2->addWidget( button );
layout2->addStretch( 1 );
layout->addLayout( layout2 );
Akonadi::Control::widgetNeedsAkonadi( this );
}
JobTrackerWidget::~JobTrackerWidget()
{
delete d;
}
void JobTrackerWidget::contextMenu( const QPoint &pos )
{
QMenu menu;
menu.addAction( i18n( "Clear View" ), d->model, SLOT(resetTracker()) );
menu.exec( mapToGlobal( pos ) );
}
void JobTrackerWidget::slotSaveToFile()
{
const QString fileName = KFileDialog::getSaveFileName( KUrl(), QString(), 0, QString(), KFileDialog::ConfirmOverwrite );
if ( fileName.isEmpty() ) {
return;
}
QFile file( fileName );
if ( !file.open( QIODevice::WriteOnly | QIODevice::Truncate ) ) {
return;
}
file.write("Job ID\t\tCreated\t\tWait Time\tJob Duration\tJob Type\t\tState\tInfo\n");
writeRows( QModelIndex(), file, 0 );
file.close();
}
void JobTrackerWidget::writeRows( const QModelIndex &parent, QFile &file, int indentLevel )
{
for ( int row = 0; row < d->model->rowCount( parent ); ++row ) {
QByteArray data;
for ( int tabs = 0; tabs < indentLevel; ++tabs ) {
data += '\t';
}
const int columnCount = d->model->columnCount( parent );
for ( int column = 0; column < columnCount; ++column ) {
const QModelIndex index = d->model->index( row, column, parent );
data += index.data().toByteArray();
if ( column < columnCount - 1 ) {
data += '\t';
}
}
data += '\n';
file.write( data );
const QModelIndex index = d->model->index( row, 0, parent );
writeRows( index, file, indentLevel + 1 );
}
}