kalzium/src/periodictableview.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2003-2005, 2006 by Carsten Niehaus                            *
   3  *   cniehaus@kde.org                                                      *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.          *
  19  ***************************************************************************/
  20
  21 #include "periodictableview.h"
  22 #include "prefs.h"
  23 #include "element.h"
  24 #include "kalziumutils.h"
  25 #include "kalziumdataobject.h"
  26 #include "kalziumpainter.h"
  27 #include "kalziumtabletype.h"
  28
  29 #include <klocale.h>
  30 #include <kdebug.h>
  31
  32 #include <QApplication>
  33 #include <QPixmap>
  34 #include <QPoint>
  35 #include <QCursor>
  36 #include <QMouseEvent>
  37 #include <QPainter>
  38 #include <QRect>
  39 #include <QDrag>
  40 #include <QSvgGenerator>
  41
  42 PeriodicTableView::PeriodicTableView( QWidget *parent )
  43   : QWidget( parent )
  44 {
  45     // No selection
  46     m_currentElement = 0;
  47
  48     setMouseTracking( true );
  49
  50     //eliminates flicker on redraw
  51     setAttribute( Qt::WA_OpaquePaintEvent, true );
  52
  53     table = 0;
  54     table2 = new QPixmap();
  55
  56     //Start with a full draw
  57     doFullDraw = true;
  58
  59     m_startDrag = QPoint();
  60
  61     m_hoverTimer.setSingleShot( true );
  62     connect( &m_hoverTimer, SIGNAL( timeout() ), this, SLOT( slotMouseover() ) );
  63     m_prevHoverElement = 0;
  64
  65     KalziumTableType *tt = KalziumTableTypeFactory::instance()->build( "Short" );
  66     m_painter = new KalziumPainter( tt );
  67
  68     setMinimumSize( tt->size() );
  69
  70     update();
  71 }
  72
  73 PeriodicTableView::~PeriodicTableView()
  74 {
  75         delete table;
  76         delete table2;
  77
  78         delete m_painter;
  79 }
  80
  81 void PeriodicTableView::activateColorScheme( const int nr )
  82 {
  83     m_painter->setScheme( nr );
  84
  85     emit SchemeChanged( m_painter->scheme() );
  86
  87     setFullDraw();
  88     update();
  89 }
  90
  91 KalziumSchemeType* PeriodicTableView::scheme() const
  92 {
  93         return m_painter->scheme();
  94 }
  95
  96 void PeriodicTableView::setMode( KalziumPainter::MODE m )
  97 {
  98         m_painter->setMode( m );
  99
 100     emit ModeChanged( m );
 101
 102         setFullDraw();
 103         update();
 104 }
 105
 106 KalziumPainter::MODE PeriodicTableView::mode() const
 107 {
 108         return m_painter->mode();
 109 }
 110
 111 void PeriodicTableView::generateSvg(const QString& filename)
 112 {
 113         QSvgGenerator *svgGen = new QSvgGenerator();
 114
 115         svgGen->setFileName( filename );
 116
 117         m_painter->begin( svgGen );
 118         m_painter->drawAll();
 119         m_painter->end();
 120 }
 121
 122 void PeriodicTableView::paintEvent( QPaintEvent * e )
 123 {
 124         if ( doFullDraw )
 125         {
 126                 delete table;
 127                 table = new QPixmap( size() );
 128
 129                 m_painter->begin( table );
 130                 m_painter->drawAll();
 131                 m_painter->end();
 132
 133                 doFullDraw = false;
 134         }
 135
 136         *table2 = *table;
 137
 138         if ( m_currentElement > 0 )
 139         {
 140                 m_painter->begin( table2 );
 141                 m_painter->drawElementSelector( m_currentElement );
 142                 m_painter->end();
 143         }
 144
 145         QPainter p( this );
 146         p.drawPixmap( e->rect().topLeft(), *table2, e->rect() );
 147         p.end();
 148 }
 149
 150 void PeriodicTableView::mousePressEvent( QMouseEvent * event )
 151 {
 152         int elementnumber = m_painter->currentTableType()->elementAtCoords( event->pos() );
 153
 154         if ( ( elementnumber > 0 ) && ( event->button() == Qt::LeftButton ) )
 155         {
 156                 m_startDrag = event->pos();
 157         }
 158 }
 159
 160 void PeriodicTableView::mouseMoveEvent( QMouseEvent *event )
 161 {
 162         int num = m_painter->currentTableType()->elementAtCoords( m_startDrag );
 163
 164         if ( ( QApplication::mouseButtons() & Qt::LeftButton ) && ( ( event->pos() - m_startDrag ).manhattanLength() > QApplication::startDragDistance() ) && ( num > 0 ) )
 165         {
 166                 Element* pointedElement = KalziumDataObject::instance()->element( num );
 167
 168                 QDrag *drag = new QDrag( this );
 169                 QMimeData *mimeData = new QMimeData;
 170
 171                 mimeData->setText( pointedElement->dataAsString( ChemicalDataObject::name ) );
 172                 drag->setMimeData( mimeData );
 173
 174                 QRect elrect = m_painter->currentTableType()->elementRect( num );
 175                 elrect.translate( -elrect.topLeft() );
 176                 QPixmap pix( elrect.size() + QSize( 1, 1 ) );
 177                 pix.fill( palette().color( QPalette::Window ) );
 178                 m_painter->begin( &pix );
 179                 m_painter->drawElement( num, elrect );
 180                 m_painter->end();
 181
 182                 drag->setPixmap( pix );
 183
 184                 drag->start( Qt::CopyAction | Qt::MoveAction );
 185
 186                 m_startDrag = QPoint();
 187         }
 188         else
 189         {
 190                 if ( m_hoverTimer.isActive() )
 191                         m_hoverTimer.stop();
 192                 m_hoverTimer.start( 200 );
 193         }
 194 }
 195
 196 void PeriodicTableView::mouseReleaseEvent( QMouseEvent *mouse )
 197 {
 198         int elem = m_painter->currentTableType()->elementAtCoords( mouse->pos() );
 199         if ( elem > 0 )
 200         {
 201                 emit ElementClicked( elem );
 202                 selectElement( elem );
 203         }
 204         else
 205                 unSelect();
 206 }
 207
 208 void PeriodicTableView::unSelect()
 209 {
 210         m_currentElement = 0;
 211         update();
 212 }
 213
 214 void PeriodicTableView::selectPoint( const QPoint& point )
 215 {
 216         m_painter->currentTableType()->elementAtCoords( point );
 217 }
 218
 219 void PeriodicTableView::selectElement( int num )
 220 {
 221         m_currentElement = num;
 222         update();
 223 }
 224
 225 void PeriodicTableView::setValueLimit( int value, ChemicalDataObject::BlueObelisk type )
 226 {
 227         //remove the value of the type if possible
 228         m_sliderValueList.remove( type );
 229
 230         //now add the new type/value
 231         m_sliderValueList.insert( type, value );
 232 }
 233
 234 int PeriodicTableView::sliderValue( ChemicalDataObject::BlueObelisk type )
 235 {
 236         return m_sliderValueList.contains( type ) ? m_sliderValueList.value( type ) : 0;
 237 }
 238
 239 QBrush PeriodicTableView::brushForElement( int element ) const
 240 {
 241         return m_painter->brushForElement( element );
 242 }
 243
 244 KalziumTableType* PeriodicTableView::tableType() const
 245 {
 246         return m_painter->currentTableType();
 247 }
 248
 249 void PeriodicTableView::slotChangeTable(int table)
 250 {
 251     KalziumTableType *tt = KalziumTableTypeFactory::instance()->build( table );
 252     m_painter->setTableType(tt);
 253
 254     emit TableTypeChanged( tt );
 255
 256     setMinimumSize( tt->size() );
 257     setFullDraw();
 258     update();
 259 }
 260
 261 void PeriodicTableView::slotMouseover()
 262 {
 263         int num = m_painter->currentTableType()->elementAtCoords( mapFromGlobal( QCursor::pos() ) );
 264         if ( ( num > 0 ) && ( num != m_prevHoverElement ) )
 265                 emit MouseOver( num );
 266         m_prevHoverElement = num;
 267 }
 268
 269 void PeriodicTableView::setNumeration( int which )
 270 {
 271         m_painter->setNumeration( which );
 272         setFullDraw();
 273         update();
 274 }
 275
 276 KalziumGradientType* PeriodicTableView::gradient() const
 277 {
 278         return m_painter->gradient();
 279 }
 280
 281 void PeriodicTableView::setGradient( int which )
 282 {
 283     m_painter->setGradient( which );
 284
 285     emit GradientTypeChanged( m_painter->gradient() );
 286     setFullDraw();
 287     update();
 288 }
 289
 290 void PeriodicTableView::setTemperature( int temp )
 291 {
 292         m_painter->setTemperature( temp );
 293         setFullDraw();
 294         update();
 295 }
 296
 297 void PeriodicTableView::setTime( int time )
 298 {
 299         m_painter->setTime( time );
 300         setFullDraw();
 301         update();
 302 }
 303
 304 void PeriodicTableView::setFullDraw()
 305 {
 306         doFullDraw = true;
 307 }
 308
 309 #include "periodictableview.moc"