#include "global.h"

#include "ActorUtil.h"
#include "BeginnerHelper.h"
#include "GameState.h"
#include "PrefsManager.h"
#include "RageLog.h"
#include "RageDisplay.h"
#include "ThemeManager.h"

// "PLAYER_X" offsets are relative to the pad.. ex: Setting this to 10, and the HELPER to 300, will put the dancer at 310
#define PLAYER_X( px )		THEME->GetMetricF("BeginnerHelper",ssprintf("Player%d_X",px+1))
#define PLAYER_ANGLE		THEME->GetMetricF("BeginnerHelper","PlayerAngle")
#define DANCEPAD_ANGLE		THEME->GetMetricF("BeginnerHelper","DancePadAngle")

// STEPCIRCLE commands are relative to the HELPER position
#define STEPCIRCLE_LEFT_COMMAND( pz )	THEME->GetMetric ("BeginnerHelper",ssprintf("StepCircleP%d_LEFT_COMMAND",pz+1))
#define STEPCIRCLE_DOWN_COMMAND( pz )	THEME->GetMetric ("BeginnerHelper",ssprintf("StepCircleP%d_DOWN_COMMAND",pz+1))
#define STEPCIRCLE_UP_COMMAND( pz )		THEME->GetMetric ("BeginnerHelper",ssprintf("StepCircleP%d_UP_COMMAND",pz+1))
#define STEPCIRCLE_RIGHT_COMMAND( pz )	THEME->GetMetric ("BeginnerHelper",ssprintf("StepCircleP%d_RIGHT_COMMAND",pz+1))

// "HELPER" offsets effect the pad/dancer as a whole.. Their relative Y cooridinates are hard-coded for eachother.
#define HELPER_X			THEME->GetMetricF("BeginnerHelper","HelperX")
#define HELPER_Y			THEME->GetMetricF("BeginnerHelper","HelperY")


BeginnerHelper::BeginnerHelper()
{
	LOG->Trace("BeginnerHelper::BeginnerHelper()");
	m_bFlashEnabled = false;
	m_bInitialized = false;
	this->AddChild(&m_sBackground);
	this->AddChild(&m_sFlash);
}

BeginnerHelper::~BeginnerHelper()
{
	LOG->Trace("BeginnerHelper::~BeginnerHelper()");
}

void BeginnerHelper::ShowStepCircle( int CSTEP )
{
	/*for( int ps=0; ps<NUM_PLAYERS; ps++ )
	{
		m_sStepCircle[ps].SetXY( (HELPER_X + PLAYER_X(ps)), (HELPER_Y + 10) );
		switch( CSTEP )
		{
			case ST_LEFT: m_sStepCircle[ps].SetRotationZ(270); break;
			case ST_DOWN: m_sStepCircle[ps].SetRotationZ(180); break;
			case ST_UP: break;
			case ST_RIGHT: m_sStepCircle[ps].SetRotationZ(90); break;
			default: break;
		}
		
		m_sStepCircle[ps].SetZoom(1);
		m_sStepCircle[ps].SetEffectNone();
		m_sStepCircle[ps].StopTweening();
		m_sStepCircle[ps].BeginTweening((GAMESTATE->m_fCurBPS/4), TWEEN_LINEAR);
		m_sStepCircle[ps].SetZoom(0);
	}*/
}

void BeginnerHelper::SetFlash( CString sFilename, float fX, float fY )
{
	m_sFlash.Load( sFilename );
	m_sFlash.SetX(fX);
	m_sFlash.SetY(fY);
}

void BeginnerHelper::Initialize( int iDancePadType, NoteData *pNotes )
{
	// Copy the note data
	ASSERT(pNotes != NULL);
	m_NoteData.CopyAll(pNotes);

	// Load the StepCircle, Background, and flash animation
	m_sBackground.Load( THEME->GetPathToG("BeginnerHelper background") );
	m_sBackground.SetXY( CENTER_X, CENTER_Y);
	
	m_sFlash.Load( THEME->GetPathToG("BeginnerHelper flash") );
	m_sFlash.SetXY( CENTER_X, CENTER_Y );
	m_sFlash.SetDiffuseAlpha(0);

	/*for( int sc=0; sc<NUM_PLAYERS*2; sc++ )
	{
		m_sStepCircle[sc].Load( THEME->GetPathToG("BeginnerHelper stepcircle") );
		m_sStepCircle[sc].SetZoom(0);	// Hide until needed.
		this->AddChild(&m_sStepCircle[sc]);
	}*/

	// Load the DancePad
	m_mDancePad.SetHorizAlign( align_left );
	m_mDancePad.SetRotationX( DANCEPAD_ANGLE );
	m_mDancePad.SetX(HELPER_X);
	m_mDancePad.SetY(HELPER_Y);
	m_mDancePad.SetZoom( 23 );	// Pad should always be 3 units bigger in zoom than the dancer.
	switch(iDancePadType)
	{
		case 0: break; // No pad
		case 1:m_mDancePad.LoadMilkshapeAscii( "Characters" SLASH "DancePad-DDR.txt" ); break;
		case 2:m_mDancePad.LoadMilkshapeAscii( "Characters" SLASH "DancePads-DDR.txt" ); break;
	}
	this->AddChild( &m_mDancePad );

	for( int pl=0; pl<NUM_PLAYERS; pl++ )	// Load players
	{
		// if there is no character set, try loading a random one.
		if( GAMESTATE->m_pCurCharacters[pl] == NULL )
		{
			GAMESTATE->m_pCurCharacters[pl] = GAMESTATE->GetRandomCharacter();
		}

		if( GAMESTATE->m_PreferredDifficulty[pl] == DIFFICULTY_BEGINNER && GAMESTATE->m_pCurCharacters[pl] != NULL )
		{
			// Load textures
			m_mDancer[pl].SetHorizAlign( align_left );
			m_mDancer[pl].LoadMilkshapeAscii( GAMESTATE->m_pCurCharacters[pl]->GetModelPath() );

			// Load needed animations
			m_mDancer[pl].LoadMilkshapeAsciiBones( "Step-LEFT","Characters" SLASH "BeginnerHelper_step-left.bones.txt" );
			m_mDancer[pl].LoadMilkshapeAsciiBones( "Step-DOWN","Characters" SLASH "BeginnerHelper_step-down.bones.txt" );
			m_mDancer[pl].LoadMilkshapeAsciiBones( "Step-UP","Characters" SLASH "BeginnerHelper_step-up.bones.txt" );
			m_mDancer[pl].LoadMilkshapeAsciiBones( "Step-RIGHT","Characters" SLASH "BeginnerHelper_step-right.bones.txt" );
			m_mDancer[pl].LoadMilkshapeAsciiBones( "Step-JUMPLR","Characters" SLASH "BeginnerHelper_step-jumplr.bones.txt" );
			/*m_mDancer[pl].LoadMilkshapeAsciiBones( "Step-JUMPUD","Characters\\BeginnerHelper_step-jumpud.bones.txt" );*/
			m_mDancer[pl].LoadMilkshapeAsciiBones( "rest",GAMESTATE->m_pCurCharacters[0]->GetRestAnimationPath() );
			m_mDancer[pl].SetDefaultAnimation( "rest" );
			m_mDancer[pl].PlayAnimation( "rest" );
			m_mDancer[pl].SetRotationX( PLAYER_ANGLE );
			m_mDancer[pl].SetX( HELPER_X + PLAYER_X(pl) );
			m_mDancer[pl].SetY( HELPER_Y + 10 );
			m_mDancer[pl].SetZoom( 20 );
			m_mDancer[pl].m_bRevertToDefaultAnimation = true;		// Stay bouncing after a step has finished animating.

			this->AddChild( &m_mDancer[pl] );
		}
	}

	m_bInitialized = true;
}

void BeginnerHelper::FlashOnce()
{
	m_sFlash.SetDiffuseAlpha(1);
	m_sFlash.SetEffectNone();
	m_sFlash.StopTweening();
	m_sFlash.BeginTweening((GAMESTATE->m_fCurBPS/4));
	m_sFlash.SetDiffuseAlpha(0);
}

void BeginnerHelper::DrawPrimitives()
{
	if(!m_bInitialized)
		return;

	m_sBackground.Draw();
	m_sFlash.Draw();

	DISPLAY->SetLighting( true );
	DISPLAY->SetLightDirectional( 
		0, 
		RageColor(0.5,0.5,0.5,1), 
		RageColor(1,1,1,1),
		RageColor(0,0,0,1),
		RageVector3(0, 0, 1) );

	m_mDancePad.Draw();
	for( int scd=0; scd<NUM_PLAYERS*2; scd++ )
		m_sStepCircle[scd].Draw();		// Should be drawn before the dancer, but after the pad, so it is drawn over the pad and under the dancer.
	
	for( int pn=0; pn<NUM_PLAYERS; pn++ )	// Draw each dancer
		if( GAMESTATE->IsPlayerEnabled(pn) )
			m_mDancer[pn].Draw();

	DISPLAY->SetLightOff( 0 );
	DISPLAY->SetLighting( false );
}

void BeginnerHelper::Step( int CSTEP )
{
	LOG->Trace( "BeginnerHelper::Step()" );
	for( int p=0; p<NUM_PLAYERS; p++ )
	{
		// First make sure this player is on beginner mode and enabled... The difficulty check may be redundant, tho.
		if( (GAMESTATE->IsPlayerEnabled(p)) && (GAMESTATE->m_PreferredDifficulty[p] == DIFFICULTY_BEGINNER) )
		{
			ShowStepCircle(CSTEP);
			m_mDancer[p].StopTweening();
			m_mDancer[p].SetRotationY(0);	// Make sure we're not still inside of a JUMPUD tween.
			switch( CSTEP )
			{
			case ST_LEFT:	m_mDancer[p].PlayAnimation( "Step-LEFT" ); break;
			case ST_RIGHT:	m_mDancer[p].PlayAnimation( "Step-RIGHT" ); break;
			case ST_UP:	m_mDancer[p].PlayAnimation( "Step-UP" ); break;
			case ST_DOWN:	m_mDancer[p].PlayAnimation( "Step-DOWN" ); break;
			case ST_JUMPLR: m_mDancer[p].PlayAnimation( "Step-JUMPLR" ); break;
			case ST_JUMPUD:
				// Until I can get an UP+DOWN jump animation, this will have to do.
				m_mDancer[p].PlayAnimation( "Step-JUMPLR" );
				
				m_mDancer[p].StopTweening();
				m_mDancer[p].BeginTweening( GAMESTATE->m_fCurBPS/8, TWEEN_LINEAR );
				m_mDancer[p].SetRotationY( 90 );
				m_mDancer[p].BeginTweening( GAMESTATE->m_fCurBPS/2 ); //sleep between jump-frames
				m_mDancer[p].BeginTweening( GAMESTATE->m_fCurBPS/6, TWEEN_LINEAR );
				m_mDancer[p].SetRotationY( 0 );
				break;
			}
		}
	}
}

void BeginnerHelper::Update( float fDeltaTime )
{
	if(!m_bInitialized)
		return;

	float beat = fDeltaTime * GAMESTATE->m_fCurBPS;

	m_sFlash.Update( beat );
	for( int scu=0; scu<NUM_PLAYERS*2; scu++ )
		m_sStepCircle[scu].Update( beat );

	for( int pu=0; pu<NUM_PLAYERS; pu++ )
		m_mDancer[pu].Update( beat );	//Update dancers

	int iStep = 0;
	if( (m_NoteData.IsThereATapAtRow( BeatToNoteRowNotRounded((GAMESTATE->m_fSongBeat+0.1f)) ) ) )
		FlashOnce();
	if((m_NoteData.IsThereATapAtRow( BeatToNoteRowNotRounded((GAMESTATE->m_fSongBeat+0.45f)))))
	{
		for( int k=0; k<m_NoteData.GetNumTracks(); k++ )
			if( m_NoteData.GetTapNote(k, BeatToNoteRowNotRounded((GAMESTATE->m_fSongBeat+0.45f)) ) == TAP_TAP )
			{
				switch(k)
				{
				case 0: iStep += 6; break;
				case 1: iStep += 3; break;
				case 2: iStep += 8; break;
				case 3: iStep += 4; break;
				};
			}
		Step( iStep );
	}
}