#include "stdafx.h"
/*
-----------------------------------------------------------------------------
 Class: SongSelector

 Desc: See header.

 Copyright (c) 2001-2002 by the person(s) listed below.  All rights reserved.
	Chris Danford
-----------------------------------------------------------------------------
*/

#include "SongSelector.h"
#include "RageLog.h"
#include "SongManager.h"
#include "GameState.h"

//
// Defines specific to SongSelector
//
const float GROUP_X				=	CENTER_X;
const float GROUP_Y				=	CENTER_Y - 160;

const float SONG_BANNER_X		=	CENTER_X;
const float SONG_BANNER_Y		=	CENTER_Y - 80;

const float SONG_BANNER_WIDTH	=	286;
const float SONG_BANNER_HEIGHT	=	92;

const float ARROWS_X[2]			=	{ SONG_BANNER_X - 200, SONG_BANNER_X + 200 };
const float ARROWS_Y[2]			=	{ SONG_BANNER_Y,       SONG_BANNER_Y };

const float SONG_TEXT_BANNER_X	=	CENTER_X;
const float SONG_TEXT_BANNER_Y	=	CENTER_Y - 10;

const float GAME_STYLE_X		=	CENTER_X;
const float GAME_STYLE_Y		=	CENTER_Y + 40;

const float STEPS_X				=	CENTER_X;
const float STEPS_Y				=	CENTER_Y + 90;

SongSelector::SongSelector()
{
	LOG->Trace( "ScreenEditMenu::ScreenEditMenu()" );

	NewNotes = false;

	m_textGroup.LoadFromFont( THEME->GetPathTo("Fonts","header1") );
	m_textGroup.SetXY( GROUP_X, GROUP_Y );
	m_textGroup.SetDiffuseColor( D3DXCOLOR(0.7f,0.7f,0.7f,1) );
	m_textGroup.SetText( "blah" );
	this->AddSubActor( &m_textGroup );

	m_Banner.SetXY( SONG_BANNER_X, SONG_BANNER_Y );
	m_Banner.SetCroppedSize( SONG_BANNER_WIDTH, SONG_BANNER_HEIGHT );
	this->AddSubActor( &m_Banner );

	m_TextBanner.SetXY( SONG_TEXT_BANNER_X, SONG_TEXT_BANNER_Y );
	this->AddSubActor( &m_TextBanner );
	
	m_sprArrowLeft.Load( THEME->GetPathTo("Graphics","edit menu left") );
	m_sprArrowLeft.SetXY( ARROWS_X[0], ARROWS_Y[0] );
	m_sprArrowLeft.SetDiffuseColor( D3DXCOLOR(1,1,1,0) );
	this->AddSubActor( &m_sprArrowLeft );

	m_sprArrowRight.Load( THEME->GetPathTo("Graphics","edit menu right") );
	m_sprArrowRight.SetXY( ARROWS_X[1], ARROWS_Y[1] );
	m_sprArrowRight.SetDiffuseColor( D3DXCOLOR(1,1,1,0) );
	this->AddSubActor( &m_sprArrowRight );

	m_textNotesType.LoadFromFont( THEME->GetPathTo("Fonts","header1") );
	m_textNotesType.SetXY( GAME_STYLE_X, GAME_STYLE_Y );
	m_textNotesType.SetDiffuseColor( D3DXCOLOR(0.7f,0.7f,0.7f,1) );
	m_textNotesType.SetText( "blah" );
	this->AddSubActor( &m_textNotesType );

	m_textNotes.LoadFromFont( THEME->GetPathTo("Fonts","header1") );
	m_textNotes.SetXY( STEPS_X, STEPS_Y );
	m_textNotes.SetDiffuseColor( D3DXCOLOR(0.7f,0.7f,0.7f,1) );
	m_textNotes.SetText( "blah" );
	this->AddSubActor( &m_textNotes );

	// data structures
	ChangeSelectedRow(ROW_GROUP);

	SONGMAN->GetGroupNames( m_sGroups );
	GAMEMAN->GetNotesTypesForGame( GAMESTATE->m_CurGame, m_NotesTypes );
	m_iSelectedGroup = m_iSelectedSong = m_iSelectedNotesType = m_iSelectedNotes = 0;

	if( GAMESTATE->m_pCurSong )
	{
		int i;

		for( i=0; i<m_sGroups.GetSize(); i++ )
			if( GAMESTATE->m_pCurSong->m_sGroupName == m_sGroups[i] )
				m_iSelectedGroup = i;
		OnGroupChange();

		for( i=0; i<m_pSongs.GetSize(); i++ )
			if( GAMESTATE->m_pCurSong == m_pSongs[i] )
				m_iSelectedSong = i;
		OnSongChange();

		for( i=0; i<m_NotesTypes.GetSize(); i++ )
			if( GAMESTATE->GetCurrentStyleDef()->m_NotesType == m_NotesTypes[i] )
				m_iSelectedNotesType = i;
		OnNotesTypeChange();

		for( i=0; i<m_pNotess.GetSize(); i++ )
			if( GAMESTATE->m_pCurNotes[PLAYER_1] == m_pNotess[i] )
				m_iSelectedNotes = i;
		OnNotesChange();
	}
	else
		OnGroupChange();

	m_Menu.Load( 
		THEME->GetPathTo("Graphics","edit background"), 
		THEME->GetPathTo("Graphics","edit top edge"),
		ssprintf("%c %c change line    %c %c change value    START to continue", char(3), char(4), char(1), char(2) ),
		false, false, 40 
		);
	this->AddSubActor( &m_Menu );

	m_soundChangeMusic.Load( THEME->GetPathTo("Sounds","select music change music") );

	m_Menu.TweenOnScreenFromBlack( SM_None );
}

SongSelector::~SongSelector()
{

}

void SongSelector::DrawPrimitives()
{
	m_Menu.DrawBottomLayer();
	ActorFrame::DrawPrimitives();
	m_Menu.DrawTopLayer();
}

void SongSelector::Up()
{
	if( m_SelectedRow == 0 )	// can't go up any further
		return;	

	ChangeSelectedRow(SelectedRow(m_SelectedRow-1));
}

void SongSelector::Down()
{
	if( m_SelectedRow == NUM_ROWS-1 )	// can't go down any further
		return;	

	ChangeSelectedRow(SelectedRow(m_SelectedRow+1));
}

void SongSelector::Left()
{
	switch( m_SelectedRow )
	{
	case ROW_GROUP:
		if( m_iSelectedGroup == 0 )	// can't go left any further
			return;
		m_iSelectedGroup--;
		OnGroupChange();
		break;
	case ROW_SONG:
		if( m_iSelectedSong == 0 )	// can't go left any further
			return;
		m_iSelectedSong--;
		OnSongChange();
		break;
	case ROW_NOTES_TYPE:
		if( m_iSelectedNotesType == 0 )	// can't go left any further
			return;
		m_iSelectedNotesType--;
		OnNotesTypeChange();
		break;
	case ROW_STEPS:
		if( m_iSelectedNotes == 0 )	// can't go left any further
			return;
		m_iSelectedNotes--;
		OnNotesChange();
		break;
	default:
		ASSERT(false);
	}
}

void SongSelector::Right()
{
	switch( m_SelectedRow )
	{
	case ROW_GROUP:
		if( m_iSelectedGroup == m_sGroups.GetSize()-1 )	// can't go right any further
			return;
		m_iSelectedGroup++;
		OnGroupChange();
		break;
	case ROW_SONG:
		if( m_iSelectedSong == m_pSongs.GetSize()-1 )	// can't go right any further
			return;
		m_iSelectedSong++;
		OnSongChange();
		break;
	case ROW_NOTES_TYPE:
		if( m_iSelectedNotesType == m_NotesTypes.GetSize()-1 )	// can't go right any further
			return;
		m_iSelectedNotesType++;
		OnNotesTypeChange();
		break;
	case ROW_STEPS:
		if( m_iSelectedNotes == m_pNotess.GetSize()-1 )	// can't go right any further
			return;
		m_iSelectedNotes++;
		OnNotesChange();
		break;
	default:
		ASSERT(false);
	}
}


void SongSelector::ChangeSelectedRow( SelectedRow row )
{
	m_textGroup.SetEffectNone();
	m_sprArrowLeft.SetDiffuseColor( D3DXCOLOR(1,1,1,0) );
	m_sprArrowRight.SetDiffuseColor( D3DXCOLOR(1,1,1,0) );
	m_textNotesType.SetEffectNone();
	m_textNotes.SetEffectNone();

	m_SelectedRow = row;
	
	switch( m_SelectedRow )
	{
	case ROW_GROUP:			m_textGroup.SetEffectGlowing();			break;
	case ROW_SONG:
		m_sprArrowLeft.SetDiffuseColor( D3DXCOLOR(1,1,1,1) );
		m_sprArrowRight.SetDiffuseColor( D3DXCOLOR(1,1,1,1) );
		break;
	case ROW_NOTES_TYPE:	m_textNotesType.SetEffectGlowing();		break;
	case ROW_STEPS:			m_textNotes.SetEffectGlowing();			break;
	default:		ASSERT(false);
	}
}

void SongSelector::OnGroupChange()
{
	m_iSelectedGroup = clamp( m_iSelectedGroup, 0, m_sGroups.GetSize()-1 );

	m_textGroup.SetText( SONGMAN->ShortenGroupName(GetSelectedGroup()) );

	// reload songs
	m_pSongs.RemoveAll();
	SONGMAN->GetSongsInGroup( GetSelectedGroup(), m_pSongs );

	OnSongChange();
}

void SongSelector::OnSongChange()
{
	m_iSelectedSong = clamp( m_iSelectedSong, 0, m_pSongs.GetSize()-1 );

	m_Banner.LoadFromSong( GetSelectedSong() );
	m_TextBanner.LoadFromSong( GetSelectedSong() );

	OnNotesTypeChange();
}

void SongSelector::OnNotesTypeChange()
{
	m_iSelectedNotesType = clamp( m_iSelectedNotesType, 0, m_NotesTypes.GetSize()-1 );

	m_textNotesType.SetText( GameManager::NotesTypeToString( GetSelectedNotesType() ) );

	m_pNotess.RemoveAll();
	GetSelectedSong()->GetNotesThatMatch( GetSelectedNotesType(), m_pNotess );
	SortNotesArrayByDifficulty( m_pNotess );
	m_pNotess.Add( NULL );		// marker for "(NEW)"
	m_iSelectedNotes = 0;

	OnNotesChange();
}

void SongSelector::OnNotesChange()
{
	m_iSelectedNotes = clamp( m_iSelectedNotes, 0, m_pNotess.GetSize()-1 );

	if( GetSelectedNotes() == NULL )
	{
		m_textNotes.SetText( "(NEW)" );
	}
	else
		m_textNotes.SetText( GetSelectedNotes()->m_sDescription!="" ? GetSelectedNotes()->m_sDescription : "[no name]" );
}

void SongSelector::TweenOffScreenToBlack( ScreenMessage smSendWhenDone, bool bPlayBackSound )
{
	m_Menu.TweenOffScreenToBlack( smSendWhenDone, bPlayBackSound  );
}