/***************************************************************************
    begin                : Jun 23 2003
    copyright            : (C) 2003, 2004, 2005, 2006 by Carsten Niehaus
    email                : 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.                                   *
 *                                                                         *
 ***************************************************************************/

#include "orbitswidget.h"

//QT-Includes
#include <QLatin1String>
#include <QPainter>

#include <KLocale>

#include <math.h>

static QStringList hulllist;

/**
 * @return the delta of the x-coordinate
 * @param r is the radius of the circle
 * @param angle is the n'st circle out of num
 * @param num is the number of circles
 */
inline static double translateToDX( double r, double angle, int num )
{
    const double t = 2 * M_PI * angle / num;
    return r * sin( t );
}

/**
 * @return the delta of the y-coordinate
 * @param r is the radius of the circle
 * @param angle is the n'st circle out of num
 * @param num is the number of circles
 */
inline static double translateToDY( double r, double angle, int num )
{
    const double t = 2 * M_PI * angle / num;
    return r * cos( t );
}

OrbitsWidget::OrbitsWidget( QWidget *parent ) : QWidget( parent )
{
	if ( hulllist.count() == 0 )
	{
		hulllist.append( QLatin1String("1") );
		hulllist.append( QLatin1String("2") );     //Helium
		hulllist.append( QLatin1String("2 1") );
		hulllist.append( QLatin1String("2 2") );
		hulllist.append( QLatin1String("2 3") );
		hulllist.append( QLatin1String("2 4") );
		hulllist.append( QLatin1String("2 5") );
		hulllist.append( QLatin1String("2 6") );
		hulllist.append( QLatin1String("2 7") );
		hulllist.append( QLatin1String("2 8") );   //Neon
		hulllist.append( QLatin1String("2 8 1") );
		hulllist.append( QLatin1String("2 8 2") );
		hulllist.append( QLatin1String("2 8 3") );
		hulllist.append( QLatin1String("2 8 4") );
		hulllist.append( QLatin1String("2 8 5") );
		hulllist.append( QLatin1String("2 8 6") );
		hulllist.append( QLatin1String("2 8 7") );
		hulllist.append( QLatin1String("2 8 8") );  //Argon
		hulllist.append( QLatin1String("2 8 8 1") );
		hulllist.append( QLatin1String("2 8 8 2") );//Calcium
		hulllist.append( QLatin1String("2 8 9 2") );
		hulllist.append( QLatin1String("2 8 10 2") );
		hulllist.append( QLatin1String("2 8 11 2") );
		hulllist.append( QLatin1String("2 8 13 1") );
		hulllist.append( QLatin1String("2 8 13 2") );//Manganese
		hulllist.append( QLatin1String("2 8 14 2") );
		hulllist.append( QLatin1String("2 8 15 2") );
		hulllist.append( QLatin1String("2 8 16 2") );
		hulllist.append( QLatin1String("2 8 18 1") );//Copper
		hulllist.append( QLatin1String("2 8 18 2") );
		hulllist.append( QLatin1String("2 8 18 3") );
		hulllist.append( QLatin1String("2 8 18 4") );
		hulllist.append( QLatin1String("2 8 18 5") );
		hulllist.append( QLatin1String("2 8 18 6") );
		hulllist.append( QLatin1String("2 8 18 7") );
		hulllist.append( QLatin1String("2 8 18 8") );//Krypton
		hulllist.append( QLatin1String("2 8 18 8 1") );
		hulllist.append( QLatin1String("2 8 18 8 2") );//Rubidium
		hulllist.append( QLatin1String("2 8 18 9 2") );
		hulllist.append( QLatin1String("2 8 18 10 2") );//Zirconium
		hulllist.append( QLatin1String("2 8 18 12 1") );
		hulllist.append( QLatin1String("2 8 18 13 1") );
		hulllist.append( QLatin1String("2 8 18 14 1") );//Techneticum
		hulllist.append( QLatin1String("2 8 18 15 1") );
		hulllist.append( QLatin1String("2 8 18 16 1") );
		hulllist.append( QLatin1String("2 8 18 18") );  //Palladium
		hulllist.append( QLatin1String("2 8 18 18 1") );
		hulllist.append( QLatin1String("2 8 18 18 2") );
		hulllist.append( QLatin1String("2 8 18 18 3") );//Indium
		hulllist.append( QLatin1String("2 8 18 18 4") );
		hulllist.append( QLatin1String("2 8 18 18 5") );
		hulllist.append( QLatin1String("2 8 18 18 6") );
		hulllist.append( QLatin1String("2 8 18 18 7") );
		hulllist.append( QLatin1String("2 8 18 18 8") );//Xenon
		hulllist.append( QLatin1String("2 8 18 18 8 1") );//Caesium
		hulllist.append( QLatin1String("2 8 18 18 8 2") );//Barium
		hulllist.append( QLatin1String("2 8 18 18 9 2") );
		hulllist.append( QLatin1String("2 8 18 20 8 2") );//Cerium
		hulllist.append( QLatin1String("2 8 18 21 8 2") );
		hulllist.append( QLatin1String("2 8 18 22 8 2") );
		hulllist.append( QLatin1String("2 8 18 23 8 2") );
		hulllist.append( QLatin1String("2 8 18 24 8 2") );
		hulllist.append( QLatin1String("2 8 18 25 8 2") );
		hulllist.append( QLatin1String("2 8 18 25 9 2") );//Gadolinium
		hulllist.append( QLatin1String("2 8 18 27 8 2") );//Terbium
		hulllist.append( QLatin1String("2 8 18 28 8 2") );
		hulllist.append( QLatin1String("2 8 18 29 8 2") );
		hulllist.append( QLatin1String("2 8 18 30 8 2") );
		hulllist.append( QLatin1String("2 8 18 31 8 2") );
		hulllist.append( QLatin1String("2 8 18 32 8 2") );//Ytterbium
		hulllist.append( QLatin1String("2 8 18 32 9 2") );//Lutetium
		hulllist.append( QLatin1String("2 8 18 32 10 2") );//Hafnium
		hulllist.append( QLatin1String("2 8 18 32 11 2") );
		hulllist.append( QLatin1String("2 8 18 32 12 2") );
		hulllist.append( QLatin1String("2 8 18 32 13 2") );
		hulllist.append( QLatin1String("2 8 18 32 14 2") );
		hulllist.append( QLatin1String("2 8 18 32 15 2") );//Irdium
		hulllist.append( QLatin1String("2 8 18 32 17 1") );
		hulllist.append( QLatin1String("2 8 18 32 18 1") );
		hulllist.append( QLatin1String("2 8 18 32 18 2") );//Mercury
		hulllist.append( QLatin1String("2 8 18 32 18 3") );
		hulllist.append( QLatin1String("2 8 18 32 18 4") );
		hulllist.append( QLatin1String("2 8 18 32 18 5") );
		hulllist.append( QLatin1String("2 8 18 32 18 6") );
		hulllist.append( QLatin1String("2 8 18 32 18 7") );
		hulllist.append( QLatin1String("2 8 18 32 18 8") );//Radon
		hulllist.append( QLatin1String("2 8 18 32 18 8 1") );//Francium
		hulllist.append( QLatin1String("2 8 18 32 18 8 2") );//Radium
		hulllist.append( QLatin1String("2 8 18 32 18 9 2") );//Actinium
		hulllist.append( QLatin1String("2 8 18 32 20 8 2") );//Thorium
		hulllist.append( QLatin1String("2 8 18 32 21 8 2") );
		hulllist.append( QLatin1String("2 8 18 32 22 8 2") );//Uran
		hulllist.append( QLatin1String("2 8 18 32 23 8 2") );
		hulllist.append( QLatin1String("2 8 18 32 24 8 2") );//Plutonium
		hulllist.append( QLatin1String("2 8 18 32 25 8 2") );
		hulllist.append( QLatin1String("2 8 18 32 26 8 2") );//Cm
		hulllist.append( QLatin1String("2 8 18 32 27 8 2") );
		hulllist.append( QLatin1String("2 8 18 32 28 8 2") );//Cf
		hulllist.append( QLatin1String("2 8 18 32 29 8 2") );
		hulllist.append( QLatin1String("2 8 18 32 30 8 2") );//Fm
		hulllist.append( QLatin1String("2 8 18 32 31 8 2") );
		hulllist.append( QLatin1String("2 8 18 32 32 8 2") );//Nobelium
		hulllist.append( QLatin1String("2 8 18 32 32 9 2") );//Lawrencium	
		hulllist.append( QLatin1String("2 8 18 32 32 10 2") );	
		hulllist.append( QLatin1String("2 8 18 32 32 11 2") );//Dubnium (105)
		hulllist.append( QLatin1String("2 8 18 32 32 12 2") );	
		hulllist.append( QLatin1String("2 8 18 32 32 13 2") );//Bohrium
		hulllist.append( QLatin1String("2 8 18 32 32 14 2") );	
		hulllist.append( QLatin1String("2 8 18 32 32 15 2") );//Mt
		hulllist.append( QLatin1String("2 8 18 32 32 16 2") );//Darmstadtium
		hulllist.append( QLatin1String("2 8 18 32 32 17 2") );//Roentgenium
	}
}

void OrbitsWidget::setElementNumber( int num )
{
	Elemno = num;
	
	numOfElectrons.clear();
	QString o;
	if ( ( Elemno > 0 ) && ( Elemno <= hulllist.count() ) )
		o = hulllist[ Elemno-1 ];

	foreach( const QString& str, o.split( ' ', QString::SkipEmptyParts ) )
		numOfElectrons.append( str.toInt() );

	update();
}

void OrbitsWidget::paintEvent(  QPaintEvent* )
{
	QPainter DC;
	DC.begin( this );
	DC.setRenderHint(  QPainter::Antialiasing, true );
	DC.setPen( Qt::black );

	int min_size = qMin( width(), height() );
	int min_delta = min_size / 10;
	
	const int num = numOfElectrons.count();
	if ( num == 0 ){
		DC.drawText( QPoint( width()/3, height()/3 ), i18n( "Unknown Electron Distribution" ) );
		return; // no orbits, do nothing
	}

	//make sure the biggest orbit fits in the widget
	//diameter
	int d = min_size - 2 * min_delta;

	//the radius of the current orbit
	int r = d / 2;

	//the radius of an 'electron'
	int r_electron = r/20;

	//difference to the previous circle
	int ddx = r / num;

	QList<int>::Iterator it = numOfElectrons.end();
	--it;

	for ( int i = 0 ; i < num ; ++i )
	{
		int mx = min_delta + ddx * i; //the x-coordinate for the current circle
		int my = min_delta + ddx * i; //the y-coordinate for the current circle

		DC.setBrush( Qt::NoBrush );
		
		//draw the big ellipses in concentric circles
		DC.drawEllipse( mx , my , d , d);

		DC.setBrush( Qt::yellow );
		
		for ( int e = 0 ; e < *it ; ++e )
		{
			int x = (int)translateToDX( d/2.0, (double)e, *it );
			int y = (int)translateToDY( d/2.0, (double)e, *it );

			DC.drawEllipse( x + mx + d/2 - r_electron, 
					y + my + d/2 - r_electron,
					2*r_electron ,
					2*r_electron );
			
		}
		--it;
		d -= 2 * ddx;
	}
}


#include "orbitswidget.moc"