kdgantt2/kdganttproxymodel.cpp

   1 /****************************************************************************
   2  ** Copyright (C) 2001-2006 Klarälvdalens Datakonsult AB.  All rights reserved.
   3  **
   4  ** This file is part of the KD Gantt library.
   5  **
   6  ** This file may be distributed and/or modified under the terms of the
   7  ** GNU General Public License version 2 as published by the Free Software
   8  ** Foundation and appearing in the file LICENSE.GPL included in the
   9  ** packaging of this file.
  10  **
  11  ** Licensees holding valid commercial KD Gantt licenses may use this file in
  12  ** accordance with the KD Gantt Commercial License Agreement provided with
  13  ** the Software.
  14  **
  15  ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
  16  ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
  17  **
  18  ** See http://www.kdab.net/kdgantt for
  19  **   information about KD Gantt Commercial License Agreements.
  20  **
  21  ** Contact info@kdab.net if any conditions of this
  22  ** licensing are not clear to you.
  23  **
  24  **********************************************************************/
  25 #include "kdganttproxymodel.h"
  26 #include "kdganttproxymodel_p.h"
  27
  28
  29 using namespace KDGantt;
  30
  31 typedef ForwardingProxyModel BASE;
  32
  33 ProxyModel::Private::Private( ProxyModel* _q )
  34 #if 0
  35     : calendarMode( false )
  36 #endif
  37 {
  38     Q_UNUSED( _q ); // for now
  39
  40     columnMap[Qt::DisplayRole]    = 0;
  41     columnMap[ItemTypeRole]       = 1;
  42     columnMap[StartTimeRole]      = 2;
  43     columnMap[EndTimeRole]        = 3;
  44     columnMap[TaskCompletionRole] = 4;
  45     columnMap[LegendRole]         = 5;
  46
  47     roleMap[Qt::DisplayRole]    = Qt::DisplayRole;
  48     roleMap[ItemTypeRole]       = Qt::DisplayRole;
  49     roleMap[StartTimeRole]      = Qt::DisplayRole;
  50     roleMap[EndTimeRole]        = Qt::DisplayRole;
  51     roleMap[TaskCompletionRole] = Qt::DisplayRole;
  52     roleMap[LegendRole]         = Qt::DisplayRole;
  53 }
  54
  55 ProxyModel::ProxyModel( QObject* parent )
  56     : BASE( parent ), _d( new Private( this ) )
  57 {
  58     init();
  59 }
  60
  61 ProxyModel::~ProxyModel()
  62 {
  63     delete _d; _d = 0;
  64 }
  65
  66 #define d d_func()
  67
  68 void ProxyModel::init()
  69 {
  70 }
  71
  72 QModelIndex ProxyModel::mapFromSource( const QModelIndex& sourceIdx ) const
  73 {
  74 #if 0
  75     if( sourceIdx.isValid() ) {
  76         if ( calendarMode() ) {
  77             const QAbstractItemModel* model = sourceIdx.model();
  78             if ( model->hasChildren( sourceIdx ) ) {
  79                 return BASE::mapFromSource( model->index( sourceIdx.row(),0,sourceIdx.parent()));
  80             } else {
  81                 // Map children to columns
  82                 return BASE::mapFromSource( model->index( sourceIdx.row(),0,sourceIdx.parent()))
  83                     .child( 0, sourceIdx.column() );
  84             }
  85         }
  86         return BASE::mapFromSource( sourceIdx.model()->index( sourceIdx.row(),0,sourceIdx.parent()));
  87     }
  88     else return QModelIndex();
  89 #else
  90     return BASE::mapFromSource( sourceIdx.model()?sourceIdx.model()->index( sourceIdx.row(),0,sourceIdx.parent()):QModelIndex());
  91 #endif
  92 }
  93
  94 QModelIndex ProxyModel::mapToSource( const QModelIndex& proxyIdx ) const
  95 {
  96 #if 0
  97     if( proxyIdx.isValid() ) {
  98         if ( calendarMode() && proxyIdx.column() > 0 ) {
  99             return BASE::mapToSource( proxyIdx.model()->index( proxyIdx.column(), 0, proxyIdx ) );
 100         }
 101         return BASE::mapToSource( proxyIdx );
 102     }
 103     else return QModelIndex();
 104 #else
 105     return BASE::mapToSource( proxyIdx );
 106 #endif
 107 }
 108
 109 void ProxyModel::setColumn( int ganttrole, int col )
 110 {
 111     d->columnMap[ganttrole] = col;
 112 }
 113
 114 int ProxyModel::column( int ganttrole ) const
 115 {
 116     return d->columnMap[ganttrole];
 117 }
 118
 119 void ProxyModel::setRole( int ganttrole, int role )
 120 {
 121     d->roleMap[ganttrole] = role;
 122 }
 123
 124 int ProxyModel::role( int ganttrole ) const
 125 {
 126     return d->roleMap[ganttrole];
 127 }
 128
 129 #if 0
 130 void ProxyModel::setCalendarMode( bool enable )
 131 {
 132     if ( d->calendarMode != enable ) {
 133         d->calendarMode = enable;
 134         reset();
 135     }
 136 }
 137
 138 bool ProxyModel::calendarMode() const
 139 {
 140     return d->calendarMode;
 141 }
 142 #endif
 143
 144 int ProxyModel::rowCount( const QModelIndex& proxyIndex ) const
 145 {
 146     // TODO
 147     return BASE::rowCount( proxyIndex );
 148 }
 149
 150 int ProxyModel::columnCount( const QModelIndex& proxyIndex ) const
 151 {
 152     return qMin( sourceModel()->columnCount( mapToSource( proxyIndex ) ), 1 );
 153 }
 154
 155 QVariant ProxyModel::data( const QModelIndex& proxyIdx, int role ) const
 156 {
 157     int srole = role;
 158     int scol  = proxyIdx.column();
 159     QHash<int, int>::const_iterator it = d->roleMap.find( role );
 160     if ( it != d->roleMap.end() ) srole = *it;
 161     it = d->columnMap.find( role );
 162     if ( it != d->columnMap.end() ) scol = *it;
 163
 164 #if 0
 165     qDebug() << "mapping role"<<static_cast<ItemDataRole>(role)<<"to"<<static_cast<ItemDataRole>(srole);
 166     qDebug() << "mapping column"<<proxyIdx.column()<<"to"<<scol
 167              << "value="<<sourceModel()->data( sourceModel()->index( proxyIdx.row(), scol, mapToSource( proxyIdx.parent() ) ), srole );
 168 #endif
 169
 170     const QAbstractItemModel* model = sourceModel();
 171     return model->data( model->index( proxyIdx.row(), scol, mapToSource( proxyIdx.parent() ) ), srole );
 172 }
 173
 174 bool ProxyModel::setData( const QModelIndex& proxyIdx, const QVariant& value, int role )
 175 {
 176     int srole = role;
 177     int scol  = proxyIdx.column();
 178     QHash<int, int>::const_iterator it = d->roleMap.constFind( role );
 179     if ( it != d->roleMap.constEnd() ) srole = *it;
 180     it = d->columnMap.constFind( role );
 181     if ( it != d->columnMap.constEnd() ) scol = *it;
 182
 183     QAbstractItemModel* model = sourceModel();
 184     return model->setData( model->index( proxyIdx.row(), scol, mapToSource( proxyIdx.parent() ) ), value, srole );
 185 }
 186
 187 #include "moc_kdganttproxymodel.cpp"