summaryrefslogtreecommitdiffstats
path: root/kalzium/src/spectrumwidget.cpp
diff options
context:
space:
mode:
authortoma <toma@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2009-11-25 17:56:58 +0000
committertoma <toma@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2009-11-25 17:56:58 +0000
commitce599e4f9f94b4eb00c1b5edb85bce5431ab3df2 (patch)
treed3bb9f5d25a2dc09ca81adecf39621d871534297 /kalzium/src/spectrumwidget.cpp
downloadtdeedu-ce599e4f9f94b4eb00c1b5edb85bce5431ab3df2.tar.gz
tdeedu-ce599e4f9f94b4eb00c1b5edb85bce5431ab3df2.zip
Copy the KDE 3.5 branch to branches/trinity for new KDE 3.5 features.
BUG:215923 git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/kdeedu@1054174 283d02a7-25f6-0310-bc7c-ecb5cbfe19da
Diffstat (limited to 'kalzium/src/spectrumwidget.cpp')
-rw-r--r--kalzium/src/spectrumwidget.cpp430
1 files changed, 430 insertions, 0 deletions
diff --git a/kalzium/src/spectrumwidget.cpp b/kalzium/src/spectrumwidget.cpp
new file mode 100644
index 00000000..5531163c
--- /dev/null
+++ b/kalzium/src/spectrumwidget.cpp
@@ -0,0 +1,430 @@
+/***************************************************************************
+ * Copyright (C) 2005 by Carsten Niehaus *
+ * cniehaus@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 "spectrumwidget.h"
+#include "spectrum.h"
+#include "element.h"
+#include "kalziumutils.h"
+
+#include <kdebug.h>
+#include <kglobal.h>
+#include <klocale.h>
+#include <math.h>
+
+#include <qcursor.h>
+#include <qsizepolicy.h>
+#include <qpainter.h>
+#include <qpixmap.h>
+#include <qtooltip.h>
+
+#include <qglobal.h>
+#if defined(Q_OS_SOLARIS)
+#include <ieeefp.h>
+#endif
+
+SpectrumWidget::SpectrumWidget( QWidget *parent, const char* name ) : QWidget( parent, name )
+{
+ startValue = 0;
+ endValue = 0;
+
+ m_LMBPointCurrent.setX( -1 );
+ m_LMBPointPress.setX( -1 );
+
+ m_showtooltip = false;
+
+ m_realHeight = 200;
+
+ Gamma = 0.8;
+ IntensityMax = 255;
+
+ setType( EmissionSpectrum );
+
+ setMinimumSize( 400, 230 );
+ setSizePolicy( QSizePolicy::Expanding, QSizePolicy::Expanding );
+ setBackgroundMode( NoBackground );
+}
+
+void SpectrumWidget::paintEvent( QPaintEvent * /*e*/ )
+{
+ if ( !m_spectrum )
+ return;
+
+ m_pixmap.resize( width(), height() );
+ m_pixmap.fill( this, width(), height() );
+
+ QPainter p;
+ p.begin( &m_pixmap, this );
+ p.fillRect( 0, 0, width(), m_realHeight, Qt::black );
+
+ paintBands( &p );
+
+ drawTickmarks( &p );
+
+ if ( m_showtooltip )
+ drawTooltip( &p );
+
+ if ( m_LMBPointPress.x() != -1 && m_LMBPointCurrent.x() != -1 )
+ drawZoomLine( &p );
+
+ p.end();
+ bitBlt( this, 0, 0, &m_pixmap );
+}
+
+void SpectrumWidget::drawZoomLine( QPainter* p )
+{
+ p->setPen( Qt::white );
+ p->drawLine( m_LMBPointPress.x(), m_LMBPointPress.y(), m_LMBPointCurrent.x(), m_LMBPointPress.y() );
+ p->drawLine( m_LMBPointCurrent.x(), m_LMBPointPress.y()+10, m_LMBPointCurrent.x(), m_LMBPointPress.y()-10 );
+ p->drawLine( m_LMBPointPress.x(), m_LMBPointPress.y()+10, m_LMBPointPress.x(), m_LMBPointPress.y()-10 );
+
+}
+
+void SpectrumWidget::paintBands( QPainter* p )
+{
+ if ( m_type == AbsorptionSpectrum )
+ {
+ for ( double va = startValue; va <= endValue ; va += 0.1 )
+ {
+ int x = xPos( va );
+ p->setPen( linecolor( va ) );
+ p->drawLine( x,0,x, m_realHeight );
+ }
+
+ p->setPen( Qt::black );
+ }
+
+ int i = 0;
+ int x = 0;
+ int temp = 0;
+
+ for ( QValueList<Spectrum::band>::Iterator it = m_spectrum->bandlist()->begin();
+ it != m_spectrum->bandlist()->end();
+ ++it )
+ {
+ if ( ( *it ).wavelength < startValue || ( *it ).wavelength > endValue )
+ continue;
+
+ x = xPos( ( *it ).wavelength );
+
+ temp = 0;
+
+ switch ( m_type )
+ {
+ case EmissionSpectrum:
+ p->setPen( linecolor( ( *it ).wavelength ) );
+ p->drawLine( x,0,x, m_realHeight-1 );
+
+ p->setPen( Qt::black );
+// p->drawLine( x,m_realHeight,x, m_realHeight );
+ break;
+
+ case AbsorptionSpectrum:
+ p->setPen( Qt::black );
+ p->drawLine( x,0,x, m_realHeight-1 );
+ break;
+ }
+
+ i++;
+ }
+}
+
+QColor SpectrumWidget::linecolor( double spectrum )
+{
+ int r,g,b;
+ wavelengthToRGB( spectrum, r,g,b );
+
+ QColor c( r,g,b );
+ return c;
+}
+
+
+void SpectrumWidget::wavelengthToRGB( double wavelength, int& r, int& g, int& b )
+{
+ double blue = 0.0, green = 0.0, red = 0.0, factor = 0.0;
+
+ int wavelength_ = ( int ) floor( wavelength );
+ if ( wavelength_ < 380 || wavelength_ > 780 )
+ {
+ //make everything white
+ r = g = b = 255;
+ return;
+ }
+ else if ( wavelength_ > 380 && wavelength_ < 439 )
+ {
+ red = -( wavelength-440 ) / ( 440-380 );
+ green = 0.0;
+ blue = 1.0;
+
+ }
+ else if ( wavelength_ > 440 && wavelength_ < 489 )
+ {
+ red = 0.0;
+ green = ( wavelength-440 ) / ( 490-440 );
+ blue = 1.0;
+ }
+ else if ( wavelength_ > 490 && wavelength_ < 509 )
+ {
+ red = 0.0;
+ green = 1.0;
+ blue = -( wavelength-510 ) / ( 510-490 );
+ }
+ else if ( wavelength_ > 510 && wavelength_ < 579 )
+ {
+ red = ( wavelength-510 ) / ( 580-510 );
+ green = 1.0;
+ blue = 0.0;
+ }
+ else if ( wavelength_ > 580 && wavelength_ < 644 )
+ {
+ red = 1.0;
+ green = -( wavelength-645 ) / ( 645-580 );
+ blue = 0.0;
+ }
+ else if ( wavelength_ > 645 && wavelength_ < 780 )
+ {
+ red = 1.0;
+ green = 0.0;
+ blue = 0.0;
+ }
+
+ if ( wavelength_ > 380 && wavelength_ < 419 )
+ factor = 0.3 + 0.7*( wavelength - 380 ) / ( 420 - 380 );
+ else if ( wavelength_ > 420 && wavelength_ < 700 )
+ factor = 1.0;
+ else if ( wavelength_ > 701 && wavelength_ < 780 )
+ factor = 0.3 + 0.7*( 780 - wavelength ) / ( 780 - 700 );
+ else
+ factor = 0.0;
+
+ r = Adjust( red, factor );
+ g = Adjust( green, factor );
+ b = Adjust( blue, factor );
+}
+
+int SpectrumWidget::Adjust( double color, double factor )
+{
+ if ( color == 0.0 )
+ return 0;
+ else
+ return qRound( IntensityMax * pow( color*factor, Gamma ) );
+}
+
+void SpectrumWidget::drawTickmarks( QPainter* p )
+{
+ //the size of the text on the tickmarks
+ const int space = 20;
+
+ //the distance between the tickmarks in pixel
+ const int d = 10;
+
+ //the total number of tickmarks to draw (small and long)
+ const int numberOfTickmarks = ( int )floor( width()/d );
+
+ double pos = 0.0;
+
+ for ( int i = 0; i < numberOfTickmarks; i++ )
+ {
+ if( i%5 == 0 )
+ {//long tickmarks plus text
+ p->drawLine( i*d, m_realHeight, i*d, m_realHeight+10 );
+ if ( i%10 == 0 &&
+ i*d > space &&
+ i*d < width()-space )
+ {
+ pos = ( double ) ( i*d )/width();
+ p->fillRect( i*d-space, m_realHeight+12, 2*space, 15, Qt::white );
+ p->drawText( i*d-space, m_realHeight+12, 2*space, 15, Qt::AlignCenter, KalziumUtils::localizedValue( KalziumUtils::strippedValue( Wavelength( pos ) ), 6 ) );
+ }
+ }
+ else {//small tickmarks
+ p->drawLine( i*d, m_realHeight, i*d, m_realHeight+5 );
+ }
+ }
+}
+
+void SpectrumWidget::keyPressEvent( QKeyEvent *e )
+{
+ kdDebug() << "SpectrumWidget::keyPressEvent()" << endl;
+ switch ( e->key() )
+ {
+ case Key_Plus:
+ slotZoomIn();
+ break;
+ case Key_Minus:
+ slotZoomOut();
+ break;
+ }
+}
+
+void SpectrumWidget::slotZoomOut()
+{
+ kdDebug() << "SpectrumWidget::slotZoomOut() "<< startValue << ":: "<< endValue << endl;
+
+ double diff = endValue - startValue;
+
+ double offset = diff * 0.05;
+
+ endValue = endValue + offset;
+ startValue = startValue - offset;
+
+ //check for invalid values
+ if ( startValue < 0.0 )
+ startValue = 0.0;
+
+ if ( endValue > 800.0 )
+ endValue = 800.0;
+
+ setBorders( ( int ) startValue, ( int )endValue );
+
+ kdDebug() << "SpectrumWidget::slotZoomOut() "<< startValue << ":: "<< endValue << endl;
+}
+
+void SpectrumWidget::slotZoomIn()
+{
+ kdDebug() << "SpectrumWidget::slotZoomIn() "<< startValue << ":: "<< endValue << endl;
+
+ double diff = endValue - startValue;
+
+ double offset = diff * 0.05;
+
+ endValue = endValue - offset;
+ startValue = startValue + offset;
+
+ setBorders( ( int ) startValue, ( int )endValue );
+
+ kdDebug() << "SpectrumWidget::slotZoomIn() "<< startValue << ":: "<< endValue << endl;
+}
+
+void SpectrumWidget::mouseMoveEvent( QMouseEvent *e )
+{
+ m_LMBPointCurrent = e->pos();
+ update();
+}
+
+void SpectrumWidget::mousePressEvent( QMouseEvent *e )
+{
+ if ( e->button() == QMouseEvent::LeftButton )
+ m_LMBPointPress = e->pos();
+ if ( e->button() == QMouseEvent::RightButton )
+ slotZoomOut();
+//FIXME
+//the tooltip is not really working. Better to not have it
+//in KDE 3.4 and think of a better solution.
+//PrepareTooltip( Wavelength( ( double )e->pos().x()/width() ) );
+}
+
+void SpectrumWidget::PrepareTooltip( double wavelength )
+{
+ Spectrum::band band;
+
+ QValueList<Spectrum::band>::const_iterator it = m_spectrum->bandlist()->begin();
+ const QValueList<Spectrum::band>::const_iterator itEnd = m_spectrum->bandlist()->end();
+
+ //find the difference in percent (1.0 is 100%, 0.1 is 10%)
+ double dif = 0.0;
+
+ bool foundWavelentgh = false;
+
+ //find the highest intensity
+ for ( ; it != itEnd; ++it )
+ {
+ double thisdif = ( *it ).wavelength / wavelength;
+
+ if ( thisdif < 0.9 || thisdif > 1.1 )
+ continue;
+
+ if ( thisdif > 1.0 ){//convert for example 1.3 to 0.7
+ thisdif = thisdif-1;
+ thisdif = 1-thisdif;
+ }
+
+ if ( thisdif > dif )
+ {
+ dif = thisdif;
+ band = *it;
+ foundWavelentgh = true;
+ }
+ }
+ if ( foundWavelentgh )
+ {
+ m_band = band;
+ m_showtooltip = true;
+ }
+ else
+ m_showtooltip = false;
+
+ kdDebug() << "SpectrumWidget::PrepareTooltip(): "<< m_showtooltip << endl;
+ update();
+}
+
+void SpectrumWidget::drawTooltip( QPainter *p )
+{
+ p->setPen( Qt::white );
+ QPoint pt = mapFromGlobal( QCursor::pos() );
+ p->drawText( pt, i18n("Wavelength: %1").arg(m_band.wavelength) );
+ pt.setY( pt.y() + 15 );
+ p->drawText( pt, i18n("Intensity: %1").arg(m_band.intensity) );
+ pt.setY( pt.y() + 15 );
+ p->drawText( pt, i18n("Energy 1, Energy 2: %1, %2").arg(m_band.energy1).arg( m_band.energy2 ));
+ pt.setY( pt.y() + 15 );
+ p->drawText( pt, i18n("Term 1, Term 2: %1, %2").arg(m_band.term1).arg( m_band.term2 ));
+ pt.setY( pt.y() + 15 );
+ p->drawText( pt, i18n("J 1, J 2: %1, %2").arg(m_band.J1).arg( m_band.J2 ));
+}
+
+void SpectrumWidget::mouseReleaseEvent( QMouseEvent *e )
+{
+ if ( e->button() == QMouseEvent::LeftButton )
+ {
+ int left = (int)Wavelength( ( double )m_LMBPointPress.x()/width() );
+ int right = (int)Wavelength( ( double )e->pos().x()/width() );
+
+ if ( left == right )
+ return;
+
+ if ( left > right )
+ setBorders( right, left );
+ else
+ setBorders( left, right );
+ }
+
+ m_LMBPointPress.setX( -1 );
+ m_LMBPointCurrent.setX( -1 );
+}
+
+QPixmap SpectrumWidget::pixmap()
+{
+ QPixmap tmp( m_pixmap );
+ QString text;
+
+ if( m_type == EmissionSpectrum )
+ text = i18n( "Emission spectrum of %1" ).arg( m_spectrum->parentElement()->elname() );
+ else
+ text = i18n( "Absorption spectrum of %1" ).arg( m_spectrum->parentElement()->elname() );
+
+ QPainter p( &tmp );
+ p.setPen( Qt::black );
+ p.drawText( 30, m_realHeight + 70, text );
+
+ return tmp;
+}
+
+#include "spectrumwidget.moc"
+