/***************************************************************************
                          obslistwizard.cpp  -  Display overhead view of the solar system
                             -------------------
    begin                : Thu 23 Jun 2005
    copyright            : (C) 2005 by Jason Harris
    email                : jharris@30doradus.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 <tqcheckbox.h>
#include <tqlabel.h>
#include <tqlayout.h>
#include <tqwidgetstack.h>
#include <knuminput.h>
#include <klistbox.h>
#include <klistview.h>
#include <kpushbutton.h>

#include "dmsbox.h"
#include "kstars.h"
#include "kstarsdata.h"
#include "locationdialog.h"
#include "magnitudespinbox.h"
#include "skyobject.h"
#include "starobject.h"
//#include "libtdeedu/extdate/extdatetimeedit.h"

#include "obslistwizard.h"

ObsListWizard::ObsListWizard( TQWidget *parent, const char *name ) 
  : KDialogBase( KDialogBase::Plain, i18n("Observing List Wizard"), Ok|Cancel, Ok, parent, name )
{
	ksw = (KStars*)parent;
	TQFrame *page = plainPage();
	TQVBoxLayout *vlay = new TQVBoxLayout( page, 0, 0 );

	olw = new ObsListWizardUI( page );
	vlay->addWidget( olw );

	connect( olw->AllButton, TQT_SIGNAL( clicked() ), this, TQT_SLOT( slotAllButton() ) );
	connect( olw->NoneButton, TQT_SIGNAL( clicked() ), this, TQT_SLOT( slotNoneButton() ) );
	connect( olw->DeepSkyButton, TQT_SIGNAL( clicked() ), this, TQT_SLOT( slotDeepSkyButton() ) );
	connect( olw->SolarSystemButton, TQT_SIGNAL( clicked() ), this, TQT_SLOT( slotSolarSystemButton() ) );
	connect( olw->LocationButton, TQT_SIGNAL( clicked() ), this, TQT_SLOT( slotChangeLocation() ) );

	connect( olw->FilterList, TQT_SIGNAL( clicked(TQListViewItem*) ), this, TQT_SLOT( slotShowStackWidget(TQListViewItem*) ) );

	//Update the selected observing list when certain UI elements are modified
	connect( olw->TypeList, TQT_SIGNAL( selectionChanged() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->ConstellationList, TQT_SIGNAL( selectionChanged() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->RAMin, TQT_SIGNAL( lostFocus() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->RAMax, TQT_SIGNAL( lostFocus() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->DecMin, TQT_SIGNAL( lostFocus() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->DecMax, TQT_SIGNAL( lostFocus() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->RA, TQT_SIGNAL( lostFocus() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->Dec, TQT_SIGNAL( lostFocus() ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->Radius, TQT_SIGNAL( valueChanged(double) ), this, TQT_SLOT( slotUpdateObjectCount() ) );
//	connect( olw->Date, TQT_SIGNAL( valueChanged( const ExtDate& ) ), this, TQT_SLOT( slotApplyFilters() ) );
	connect( olw->Mag, TQT_SIGNAL( valueChanged( double ) ), this, TQT_SLOT( slotUpdateObjectCount() ) );
	connect( olw->ExcludeNoMag, TQT_SIGNAL( clicked() ), this, TQT_SLOT( slotUpdateObjectCount() ) );

	connect( olw->SelectByConstellation, TQT_SIGNAL( toggled(bool) ), this, TQT_SLOT( slotEnableConstellationPage(bool) ) );
	connect( olw->SelectByRect, TQT_SIGNAL( toggled(bool) ), this, TQT_SLOT( slotEnableRectPage(bool) ) );
	connect( olw->SelectByCirc, TQT_SIGNAL( toggled(bool) ), this, TQT_SLOT( slotEnableCircPage(bool) ) );
//	connect( olw->SelectByDate, TQT_SIGNAL( toggled(bool) ), this, TQT_SLOT( slotEnableDatePage(bool) ) );
	connect( olw->SelectByMag,  TQT_SIGNAL( toggled(bool) ), this, TQT_SLOT( slotEnableMagPage(bool) ) );

	connect( this, TQT_SIGNAL( okClicked() ), this, TQT_SLOT( slotApplyFilters() ) );

	initialize();
}

ObsListWizard::~ObsListWizard()
{
}

void ObsListWizard::initialize()
{
	//Populate the list of constellations
	for ( SkyObject *p = ksw->data()->cnameList.first(); p; p = ksw->data()->cnameList.next() ) {
		olw->ConstellationList->insertItem( p->name() );
	}

//	//Initialize date
//	olw->Date->setDate( ksw->data()->lt().date() );

	//unSelect all object types
	olw->TypeList->selectAll( false );

	olw->Mag->setMinValue( -5.0 );
	olw->Mag->setMaxValue( 20.0 );
	olw->Mag->setValue( 6.0 );

	olw->FilterList->setSelected( olw->FilterList->firstChild(), true );

	olw->RA->setDegType( false );
	olw->RAMin->setDegType( false );
	olw->RAMax->setDegType( false );

	//Initialize object counts
	ObjectCount = 0; //number of objects in observing list
	StarCount = ksw->data()->starList.count(); //total number of stars
	PlanetCount = 10; //Sun, Moon, 8 planets
	AsteroidCount = ksw->data()->asteroidList.count(); //total number of asteroids
	CometCount = ksw->data()->cometList.count(); //total number of comets
	//DeepSkyObjects
	OpenClusterCount = 0;
	GlobClusterCount = 0;
	GasNebCount = 0;
	PlanNebCount = 0;
	GalaxyCount = 0;
	for ( SkyObject *o = (SkyObject*)(ksw->data()->deepSkyList.first()); o; o = (SkyObject*)(ksw->data()->deepSkyList.next()) ) {
		if ( o->type() == SkyObject::GALAXY ) ++GalaxyCount; //most deepsky obj are galaxies, so check them first
		else if ( o->type() == SkyObject::STAR || o->type() == SkyObject::CATALOG_STAR ) ++StarCount;
		else if ( o->type() == SkyObject::OPEN_CLUSTER ) ++OpenClusterCount;
		else if ( o->type() == SkyObject::GLOBULAR_CLUSTER ) ++GlobClusterCount;
		else if ( o->type() == SkyObject::GASEOUS_NEBULA || o->type() == SkyObject::SUPERNOVA_REMNANT ) ++GasNebCount;
		else if ( o->type() == SkyObject::PLANETARY_NEBULA ) ++PlanNebCount;
	}

// 	//DEBUG
// 	kdDebug() << "StarCount: " << StarCount << endl;
// 	kdDebug() << "OpenClusterCount: " << OpenClusterCount << endl;
// 	kdDebug() << "GlobClusterCount: " << GlobClusterCount << endl;
// 	kdDebug() << "GasNebCount: " << GasNebCount << endl;
// 	kdDebug() << "PlanNebCount: " << PlanNebCount << endl;
// 	kdDebug() << "GalaxyCount: " << GalaxyCount << endl;

}

void ObsListWizard::slotAllButton() { olw->TypeList->selectAll( true ); }
void ObsListWizard::slotNoneButton() { olw->TypeList->selectAll( false ); }

void ObsListWizard::slotEnableConstellationPage( bool t ) {
	olw->ConstellationList->setEnabled(t);

	//disable the other two region options
	if ( t ) {
		olw->SelectByRect->setChecked( false );
		olw->SelectByCirc->setChecked( false );
	}
	
	slotUpdateObjectCount();
}

void ObsListWizard::slotEnableRectPage( bool t ) {
	olw->RAMin->setEnabled(t);
	olw->RAMax->setEnabled(t);
	olw->DecMin->setEnabled(t);
	olw->DecMax->setEnabled(t);

	//disable the other two region options
	if ( t ) {
		olw->SelectByConstellation->setChecked( false );
		olw->SelectByCirc->setChecked( false );
	}
	
	slotUpdateObjectCount();
}

void ObsListWizard::slotEnableCircPage( bool t ) {
	olw->RA->setEnabled(t);
	olw->Dec->setEnabled(t);
	olw->Radius->setEnabled(t);

	//disable the other two region options
	if ( t ) {
		olw->SelectByConstellation->setChecked( false );
		olw->SelectByRect->setChecked( false );
	}
	
	slotUpdateObjectCount();
}

// void ObsListWizard::slotEnableDatePage( bool t ) {
// 	olw->Date->setEnabled(t);
// 	olw->LocationButton->setEnabled(t);
// }

void ObsListWizard::slotEnableMagPage( bool t ) {
	olw->Mag->setEnabled(t);
	olw->ExcludeNoMag->setEnabled(t);
	slotUpdateObjectCount();
}

void ObsListWizard::slotShowStackWidget( TQListViewItem *i )
{
	if ( i ) {
		TQString t = i->text(0);
	
		if ( t.contains( i18n( "Object type(s)" ) ) )      olw->FilterStack->raiseWidget( olw->ObjTypePage );
		if ( t.contains( i18n( "Region" ) ) )              olw->FilterStack->raiseWidget( olw->RegionPage );
		if ( t.contains( i18n( "In constellation(s)" ) ) ) olw->FilterStack->raiseWidget( olw->ConstellationPage );
		if ( t.contains( i18n( "Circular" ) ) )            olw->FilterStack->raiseWidget( olw->CircRegionPage );
		if ( t.contains( i18n( "Rectangular" ) ) )         olw->FilterStack->raiseWidget( olw->RectRegionPage );
//		if ( t.contains( i18n( "Observable on date" ) ) )  olw->FilterStack->raiseWidget( olw->ObsDatePage );
		if ( t.contains( i18n( "Magnitude limit" ) ) )     olw->FilterStack->raiseWidget( olw->MagLimitPage );
	}
}

void ObsListWizard::slotDeepSkyButton()
{
	olw->TypeList->selectAll( false );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Open Clusters" ) ), true );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Globular Clusters" ) ), true );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Gaseous Nebulae" ) ), true );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Planetary Nebulae" ) ), true );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Galaxies" ) ), true );
}

void ObsListWizard::slotSolarSystemButton()
{
	olw->TypeList->selectAll( false );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Sun, Moon, Planets" ) ), true );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Comets" ) ), true );
	olw->TypeList->setSelected( olw->TypeList->findItem( i18n( "Asteroids" ) ), true );
}

void ObsListWizard::slotChangeLocation()
{
	LocationDialog ld( ksw );

	if ( ld.exec() == TQDialog::Accepted ) {
		//set geographic location
	}
}

void ObsListWizard::slotUpdateObjectCount() 
{
	ObjectCount = 0;
	if ( olw->TypeList->findItem( i18n( "Stars" ) )->isSelected() )
		ObjectCount += StarCount;
	if ( olw->TypeList->findItem( i18n( "Sun, Moon, Planets" ) )->isSelected() ) 
		ObjectCount += PlanetCount;
	if ( olw->TypeList->findItem( i18n( "Comets" ) )->isSelected() )
		ObjectCount += CometCount;
	if ( olw->TypeList->findItem( i18n( "Asteroids" ) )->isSelected() )
		ObjectCount += AsteroidCount;
	if ( olw->TypeList->findItem( i18n( "Galaxies" ) )->isSelected() )
		ObjectCount += GalaxyCount;
	if ( olw->TypeList->findItem( i18n( "Open Clusters" ) )->isSelected() )
		ObjectCount += OpenClusterCount;
	if ( olw->TypeList->findItem( i18n( "Globular Clusters" ) )->isSelected() )
		ObjectCount += GlobClusterCount;
	if ( olw->TypeList->findItem( i18n( "Gaseous Nebulae" ) )->isSelected() )
		ObjectCount += GasNebCount;
	if ( olw->TypeList->findItem( i18n( "Planetary Nebulae" ) )->isSelected() )
		ObjectCount += PlanNebCount;

	applyFilters( false ); //false = only adjust counts, do not build list
}

void ObsListWizard::applyFilters( bool doBuildList )
{
	if ( doBuildList )
		obsList().clear();

	//make sure rect region data are valid
	rectOk = false;
	if ( olw->SelectByRect->isChecked() ) {
		ra1 = olw->RAMin->createDms( false, &rectOk ).Hours();
		if ( rectOk ) ra2 = olw->RAMax->createDms( false, &rectOk ).Hours();
		if ( rectOk ) dc1 = olw->DecMin->createDms( true, &rectOk ).Degrees();
		if ( rectOk ) dc2 = olw->DecMax->createDms( true, &rectOk ).Degrees();
		if ( ra2 == 0.0 ) ra2 = 24.0;

		//Make sure dc1 < dc2.  
		if ( dc1 > dc2 ) {
			double temp = dc2;
			dc2 = dc1;
			dc1 = temp;
		}

		//If ra1 > ra2, we may need to swap the two values, or subtract 24h from ra1.
		if ( ra1 > ra2 ) {
			if ( ra1 - ra2 > 12.0 ) { //the user probably wants a region that straddles 0h
				ra1 -= 24.0;
			} else { //the user probably wants ra2 to be the lower limit
				double temp = ra2;
				ra2 = ra1;
				ra1 = temp;
			}
		}
	}

	//make sure circ region data are valid
	circOk = false;
	if ( olw->SelectByCirc->isChecked() ) {
		double ra = olw->RA->createDms( false, &circOk ).Hours();
		double dc(0.0);
		if ( circOk ) dc = olw->Dec->createDms( true, &circOk ).Degrees();
		if ( circOk ) {
			pCirc.set( ra, dc );
			rCirc = olw->Radius->value();
		}
	}

	double maglimit = 100.;
	if ( olw->SelectByMag->isChecked() ) maglimit = olw->Mag->value();

	//Stars
	int starIndex(ksw->data()->starList.count());
	if ( olw->TypeList->findItem( i18n( "Stars" ) )->isSelected() ) {
		if ( maglimit < 100. ) {
			//Stars are sorted by mag, so use binary search algo to find index of faintest mag
			int low(0), high(ksw->data()->starList.count()-1), middle;
			while ( low < high ) {
				middle = (low + high)/2;
				if ( maglimit == ksw->data()->starList.at(middle)->mag() ) break;
				if ( maglimit <  ksw->data()->starList.at(middle)->mag() ) high = middle - 1;
				if ( maglimit >  ksw->data()->starList.at(middle)->mag() ) low = middle + 1;
			}
			//now, the star at "middle" has the right mag, but we want the *last* star that has this mag.
			for ( starIndex=middle+1; starIndex<ksw->data()->starList.count(); ++starIndex ) {
				if ( ksw->data()->starList.at(starIndex)->mag() > maglimit ) break;
			}
		}

		if ( doBuildList ) {
			for ( uint i=0; i < starIndex; ++i ) {
				SkyObject *o = (SkyObject*)(ksw->data()->starList.at(i));
				applyRegionFilter( o, doBuildList, false ); //false = don't adjust ObjectCount 
			}
		} else {
			ObjectCount -= (ksw->data()->starList.count() - starIndex); //reduce StarCount by appropriate amount
			for ( uint i=0; i < starIndex; ++i ) {
				SkyObject *o = (SkyObject*)(ksw->data()->starList.at(i));
				applyRegionFilter( o, doBuildList );
			}
		}
	}

	//Sun, Moon, Planets
	if ( olw->TypeList->findItem( i18n( "Sun, Moon, Planets" ) )->isSelected() ) {
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->planetSun(), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->Moon, doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Mercury"), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Venus"), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Mars"), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Jupiter"), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Saturn"), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Uranus"), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Neptune"), doBuildList );
		applyRegionFilter( (SkyObject*)ksw->data()->PCat->findByName("Pluto"), doBuildList );
	}

	//Deep sky objects
	for ( SkyObject *o = (SkyObject*)(ksw->data()->deepSkyList.first()); o; o = (SkyObject*)(ksw->data()->deepSkyList.next()) ) {
		//Skip unselected object types
		if ( (o->type() == SkyObject::STAR || o->type() == SkyObject::CATALOG_STAR) && ! olw->TypeList->findItem( i18n( "Stars" ) )->isSelected() ) continue;
		if ( o->type() == SkyObject::OPEN_CLUSTER && ! olw->TypeList->findItem( i18n( "Open Clusters" ) )->isSelected() ) continue;
		if ( o->type() == SkyObject::GLOBULAR_CLUSTER && ! olw->TypeList->findItem( i18n( "Globular Clusters" ) )->isSelected() ) continue;
		if ( (o->type() == SkyObject::GASEOUS_NEBULA || o->type() == SkyObject::SUPERNOVA_REMNANT) && ! olw->TypeList->findItem( i18n( "Gaseous Nebulae" ) )->isSelected() ) continue;
		if ( o->type() == SkyObject::PLANETARY_NEBULA && ! olw->TypeList->findItem( i18n( "Planetary Nebulae" ) )->isSelected() ) continue;
		if ( o->type() == SkyObject::GALAXY && ! olw->TypeList->findItem( i18n( "Galaxies" ) )->isSelected() ) continue;
		if ( o->type() == SkyObject::TYPE_UNKNOWN ) continue;

		if ( olw->SelectByMag->isChecked() ) {
			if ( o->mag() > 90. ) {
				if ( ! olw->ExcludeNoMag->isChecked() ) 
					applyRegionFilter( o, doBuildList );
				else if ( ! doBuildList )
					--ObjectCount;
			} else {
				if ( o->mag() <= maglimit ) 
					applyRegionFilter( o, doBuildList );
				else if ( ! doBuildList )
					--ObjectCount;
			}
		} else {
			applyRegionFilter( o, doBuildList );
		}
	}

	//Comets
	if ( olw->TypeList->findItem( i18n( "Comets" ) )->isSelected() ) {
		for ( SkyObject *o = (SkyObject*)(ksw->data()->cometList.first()); o; o = (SkyObject*)(ksw->data()->cometList.next()) ) {
			//comets don't have magnitudes at this point, so skip mag check
			applyRegionFilter( o, doBuildList );
		}
	}

	//Asteroids
	if ( olw->TypeList->findItem( i18n( "Asteroids" ) )->isSelected() ) {
		for ( SkyObject *o = (SkyObject*)(ksw->data()->asteroidList.first()); o; o = (SkyObject*)(ksw->data()->asteroidList.next()) ) {
	
			if ( olw->SelectByMag->isChecked() ) {
				if ( o->mag() > 90. ) {
					if ( ! olw->ExcludeNoMag->isChecked() ) 
						applyRegionFilter( o, doBuildList );
					else if ( ! doBuildList )
						--ObjectCount;
				} else {
					if ( o->mag() <= maglimit ) 
						applyRegionFilter( o, doBuildList );
					else if ( ! doBuildList )
						--ObjectCount;
				}
			} else {
				applyRegionFilter( o, doBuildList );
			}
		}
	}

	//Update the object count label
	if ( doBuildList ) ObjectCount = obsList().count();
	olw->CountLabel->setText( i18n("Current selection: %1 objects").tqarg( ObjectCount ) );
}
	
void ObsListWizard::applyRegionFilter( SkyObject *o, bool doBuildList, bool doAdjustCount ) {
	//select by constellation
	if ( olw->SelectByConstellation->isChecked() ) {
		TQString c( o->constellation( ksw->data()->csegmentList, ksw->data()->cnameList ) );
		TQListBoxItem *it = olw->ConstellationList->findItem( c );

		if ( it && it->isSelected() ) { 
			if ( doBuildList ) obsList().append ( o );
		} else if ( doAdjustCount ) --ObjectCount;
	}

	//select by rectangular region
	else if ( rectOk ) {
		double ra = o->ra()->Hours();
		double dec = o->dec()->Degrees();
		bool addObject = false;
		if ( dec >= dc1 && dec <= dc2 ) {
			if ( ra1 < 0.0 ) {
				if (ra >= ra1 + 24.0 || ra <= ra2 ) { addObject = true; }
			} else {
				if ( ra >= ra1 && ra <= ra2 ) { addObject = true; }
			}
		}

		if ( addObject && doBuildList ) obsList().append( o );
		if ( ! addObject && doAdjustCount ) --ObjectCount;
	}

	//select by circular region
	else if ( circOk ) {
		if ( o->angularDistanceTo( &pCirc ).Degrees() < rCirc ) { 
			if ( doBuildList ) obsList().append( o );
		} else if ( doAdjustCount ) --ObjectCount;
	}

	//No region filter, just add the object
	else if ( doBuildList ) {
		obsList().append( o );
	}
}

#include "obslistwizard.moc"