okular/ui/side_reviews.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005 by Enrico Ros <eros.kde@email.it>                  *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  ***************************************************************************/
   9
  10 #include "side_reviews.h"
  11
  12 // qt/kde includes
  13 #include <QtCore/QStringList>
  14 #include <QtGui/QHeaderView>
  15 #include <QtGui/QLayout>
  16 #include <QtGui/QPaintEvent>
  17 #include <QtGui/QPainter>
  18 #include <QtGui/QSizePolicy>
  19 #include <QtGui/QTextDocument>
  20 #include <QtGui/QToolBar>
  21 #include <QtGui/QTreeView>
  22
  23 #include <kaction.h>
  24 #include <klocale.h>
  25 #include <kiconloader.h>
  26 #include <kicon.h>
  27
  28 // local includes
  29 #include "core/annotations.h"
  30 #include "core/document.h"
  31 #include "core/page.h"
  32 #include "settings.h"
  33 #include "annotationpopup.h"
  34 #include "annotationproxymodels.h"
  35 #include "annotationmodel.h"
  36 #include "ktreeviewsearchline.h"
  37
  38 class TreeView : public QTreeView
  39 {
  40   public:
  41     TreeView( Okular::Document *document, QWidget *parent = 0 )
  42       : QTreeView( parent ), m_document( document )
  43     {
  44     }
  45
  46   protected:
  47     virtual void paintEvent( QPaintEvent *event )
  48     {
  49       bool hasAnnotations = false;
  50       for ( uint i = 0; i < m_document->pages(); ++i )
  51         if ( m_document->page( i )->hasAnnotations() ) {
  52           hasAnnotations = true;
  53           break;
  54         }
  55       if ( !hasAnnotations ) {
  56         QPainter p( viewport() );
  57         p.setRenderHint( QPainter::Antialiasing, true );
  58         p.setClipRect( event->rect() );
  59
  60         QTextDocument document;
  61         document.setHtml( i18n( "<div align=center><h3>No annotations</h3>"
  62                                 "To create new annotations press F6 or select <i>Tools -&gt; Review</i>"
  63                                 " from the menu.</div>" ) );
  64         document.setTextWidth( width() - 50 );
  65
  66         const uint w = document.size().width() + 20;
  67         const uint h = document.size().height() + 20;
  68
  69         p.setBrush( palette().background() );
  70         p.translate( 0.5, 0.5 );
  71         p.drawRoundRect( 15, 15, w, h, (8*200)/w, (8*200)/h );
  72         p.translate( 20, 20 );
  73         document.drawContents( &p );
  74
  75       } else {
  76         QTreeView::paintEvent( event );
  77       }
  78     }
  79
  80   private:
  81     Okular::Document *m_document;
  82 };
  83
  84
  85 Reviews::Reviews( QWidget * parent, Okular::Document * document )
  86     : QWidget( parent ), m_document( document )
  87 {
  88     // create widgets and layout them vertically
  89     QVBoxLayout * vLayout = new QVBoxLayout( this );
  90     vLayout->setMargin( 0 );
  91     vLayout->setSpacing( 6 );
  92
  93     m_view = new TreeView( m_document, this );
  94     m_view->setAlternatingRowColors( true );
  95     m_view->setSelectionMode( QAbstractItemView::ExtendedSelection );
  96     m_view->header()->hide();
  97
  98     QToolBar *toolBar = new QToolBar( this );
  99     toolBar->setObjectName( "reviewOptsBar" );
 100     QSizePolicy sp = toolBar->sizePolicy();
 101     sp.setVerticalPolicy( QSizePolicy::Minimum );
 102     toolBar->setSizePolicy( sp );
 103
 104     m_model = new AnnotationModel( m_document, m_view );
 105
 106     m_filterProxy = new PageFilterProxyModel( m_view );
 107     m_groupProxy = new PageGroupProxyModel( m_view );
 108     m_authorProxy  = new AuthorGroupProxyModel( m_view );
 109
 110     m_filterProxy->setSourceModel( m_model );
 111     m_groupProxy->setSourceModel( m_filterProxy );
 112     m_authorProxy->setSourceModel( m_groupProxy );
 113
 114
 115     m_view->setModel( m_authorProxy );
 116
 117     vLayout->addWidget( new KTreeViewSearchLine( this, m_view ) );
 118     vLayout->addWidget( m_view );
 119     vLayout->addWidget( toolBar );
 120
 121     toolBar->setIconSize( QSize( 16, 16 ) );
 122     toolBar->setMovable( false );
 123     // - add Page button
 124     QAction * groupByPageAction = toolBar->addAction( KIcon( "text-x-generic" ), i18n( "Group by Page" ) );
 125     groupByPageAction->setCheckable( true );
 126     connect( groupByPageAction, SIGNAL( toggled( bool ) ), this, SLOT( slotPageEnabled( bool ) ) );
 127     groupByPageAction->setChecked( Okular::Settings::groupByPage() );
 128     // - add Author button
 129     QAction * groupByAuthorAction = toolBar->addAction( KIcon( "user-identity" ), i18n( "Group by Author" ) );
 130     groupByAuthorAction->setCheckable( true );
 131     connect( groupByAuthorAction, SIGNAL( toggled( bool ) ), this, SLOT( slotAuthorEnabled( bool ) ) );
 132     groupByAuthorAction->setChecked( Okular::Settings::groupByAuthor() );
 133
 134     // - add separator
 135     toolBar->addSeparator();
 136     // - add Current Page Only button
 137     QAction * curPageOnlyAction = toolBar->addAction( KIcon( "arrow-down" ), i18n( "Show reviews for current page only" ) );
 138     curPageOnlyAction->setCheckable( true );
 139     connect( curPageOnlyAction, SIGNAL( toggled( bool ) ), this, SLOT( slotCurrentPageOnly( bool ) ) );
 140     curPageOnlyAction->setChecked( Okular::Settings::currentPageOnly() );
 141
 142     connect( m_view, SIGNAL( activated( const QModelIndex& ) ),
 143              this, SLOT( activated( const QModelIndex& ) ) );
 144
 145     m_view->setContextMenuPolicy( Qt::CustomContextMenu );
 146     connect( m_view, SIGNAL( customContextMenuRequested( const QPoint& ) ),
 147              this, SLOT( contextMenuRequested( const QPoint& ) ) );
 148
 149 }
 150
 151 //BEGIN DocumentObserver Notifies
 152 void Reviews::notifyViewportChanged( bool )
 153 {
 154     m_filterProxy->setCurrentPage( m_document->currentPage() );
 155 }
 156 //END DocumentObserver Notifies
 157
 158 //BEGIN GUI Slots -> requestListViewUpdate
 159 void Reviews::slotPageEnabled( bool on )
 160 {
 161     // store toggle state in Settings and update the listview
 162     Okular::Settings::setGroupByPage( on );
 163     m_groupProxy->groupByPage( on );
 164
 165     m_view->expandAll();
 166 }
 167
 168 void Reviews::slotAuthorEnabled( bool on )
 169 {
 170     // store toggle state in Settings and update the listview
 171     Okular::Settings::setGroupByAuthor( on );
 172     m_authorProxy->groupByAuthor( on );
 173
 174     m_view->expandAll();
 175 }
 176
 177 void Reviews::slotCurrentPageOnly( bool on )
 178 {
 179     // store toggle state in Settings and update the listview
 180     Okular::Settings::setCurrentPageOnly( on );
 181     m_filterProxy->groupByCurrentPage( on );
 182
 183     m_view->expandAll();
 184 }
 185 //END GUI Slots
 186
 187
 188 void Reviews::activated( const QModelIndex &index )
 189 {
 190     const QModelIndex authorIndex = m_authorProxy->mapToSource( index );
 191     const QModelIndex filterIndex = m_groupProxy->mapToSource( authorIndex );
 192     const QModelIndex annotIndex = m_filterProxy->mapToSource( filterIndex );
 193
 194     Okular::Annotation *annotation = m_model->annotationForIndex( annotIndex );
 195     if ( !annotation )
 196       return;
 197
 198     int pageNumber = m_model->data( annotIndex, AnnotationModel::PageRole ).toInt();
 199     const Okular::Page * page = m_document->page( pageNumber );
 200
 201     // calculating the right coordinates to center the view on the annotation
 202     QRect rect = Okular::AnnotationUtils::annotationGeometry( annotation, page->width(), page->height() );
 203     Okular::NormalizedRect nr( rect, (int)page->width(), (int)page->height() );
 204     // set the viewport parameters
 205     Okular::DocumentViewport vp;
 206     vp.pageNumber = pageNumber;
 207     vp.rePos.enabled = true;
 208     vp.rePos.pos = Okular::DocumentViewport::Center;
 209     vp.rePos.normalizedX = ( nr.right + nr.left ) / 2.0;
 210     vp.rePos.normalizedY = ( nr.bottom + nr.top ) / 2.0;
 211     // setting the viewport
 212     m_document->setViewport( vp, -1, true );
 213 }
 214
 215 QModelIndexList Reviews::retrieveAnnotations(const QModelIndex& idx) const
 216 {
 217     QModelIndexList ret;
 218     if ( idx.isValid() )
 219     {
 220         if ( idx.model()->hasChildren( idx ) )
 221         {
 222             int rowCount = idx.model()->rowCount( idx );
 223             for ( int i = 0; i < rowCount; i++ )
 224             {
 225                 ret += retrieveAnnotations( idx.child( i, idx.column() ) );
 226             }
 227         }
 228         else
 229         {
 230             ret += idx;
 231         }
 232     }
 233
 234     return ret;
 235 }
 236
 237 void Reviews::contextMenuRequested( const QPoint &pos )
 238 {
 239     AnnotationPopup popup( m_document, this );
 240     connect( &popup, SIGNAL( setAnnotationWindow( Okular::Annotation* ) ),
 241              this, SIGNAL( setAnnotationWindow( Okular::Annotation* ) ) );
 242     connect( &popup, SIGNAL( removeAnnotationWindow( Okular::Annotation* ) ),
 243              this, SIGNAL( removeAnnotationWindow( Okular::Annotation* ) ) );
 244
 245     QModelIndexList indexes = m_view->selectionModel()->selectedIndexes();
 246     Q_FOREACH ( const QModelIndex &index, indexes )
 247     {
 248         QModelIndexList annotations = retrieveAnnotations(index);
 249         Q_FOREACH ( const QModelIndex &idx, annotations )
 250         {
 251             const QModelIndex authorIndex = m_authorProxy->mapToSource( idx );
 252             const QModelIndex filterIndex = m_groupProxy->mapToSource( authorIndex );
 253             const QModelIndex annotIndex = m_filterProxy->mapToSource( filterIndex );
 254             Okular::Annotation *annotation = m_model->annotationForIndex( annotIndex );
 255             if ( annotation )
 256             {
 257                 const int pageNumber = m_model->data( annotIndex, AnnotationModel::PageRole ).toInt();
 258                 popup.addAnnotation( annotation, pageNumber );
 259             }
 260         }
 261     }
 262
 263     popup.exec( m_view->viewport()->mapToGlobal( pos ) );
 264 }
 265
 266 #include "side_reviews.moc"