package model.gameState;

import image.Images;
import image.Images.ImageType;

import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.GradientPaint;
import java.awt.Graphics2D;
import java.awt.Polygon;
import java.awt.RenderingHints;
import java.awt.Transparency;
import java.awt.image.BufferedImage;
import java.awt.image.VolatileImage;
import java.util.ArrayList;
import java.util.List;

import model.GameModel;
import model.SongHandler;
import model.objects.DifficultyButton;
import model.objects.MenuButton;
import model.objects.highscore.Highscore;
import audio.Song;
import audio.SongInstance;
import audio.sorting.SortALPHA;
import audio.sorting.SortPLAYED;
import control.GameStateManager;
import control.GameStateManager.State;
import control.button.ButtonEvent;
import control.button.ButtonHandler;
import control.joystick.Joystick;
import control.joystick.Joystick.Position;
import control.joystick.JoystickEvent;
import control.joystick.JoystickHandler;
import data.io.SQLConnector;

public class MenuState extends GameState {
	private ArrayList<MenuButton> buttons;
	private ArrayList<DifficultyButton> buttons2;
	private int selected, oldselected;
	private List<Song> songs;
	SQLConnector sql;

	private int animationcounter;
	private boolean subscreen, startanimation;
	private VolatileImage mainScreenBackground, subScreenBackground, subScreenForeground;
	
	int yPosDiffButton = 900;
	private int difSelect=0, oldDifSelect = -1;
	Font textFont = new Font("OCR A Extended", Font.BOLD, 50);
	Font textFontSmall = new Font("OCR A Extended", Font.BOLD, 15);
	BufferedImage aanwijzers = Images.getImage(ImageType.aanwijzers);
	int index = 0;
	
	
	
	int kleurButton1 = 0;
	int kleurButton2 = 1;
	int kleurButton3 = 2;
	int kleurButton4 = 3;
	int kleurButton5 = 4;
	
	
	
	public MenuState(GameStateManager gsm, SongHandler sh, SQLConnector sql) {	
		super(gsm, sh, sql);
		buttons = new ArrayList<MenuButton>();
		buttons2 = new ArrayList<DifficultyButton>();
		this.songs = sh.getSongs();
		startanimation = true;
		subscreen = false;
		this.sql = sql;
		
		selected = 0;
		oldselected = -1;
		
		buttons2.clear();
 		int instanceNr = 0;
 		for(int i = selectedToSong(selected).getSongs().size(); i>0; i--){
 			if(sh.getCurrentSong().getSongs().size() == 0)
 				continue;
 			SongInstance si = selectedToSong(selected).getSongs().get(i-1);
 			buttons2.add(new DifficultyButton(yPosDiffButton-instanceNr,si.getDifficulty(), GameModel.colors[i-1]));
 			instanceNr += 100;
 		}
		difSelect = 0;

		generateMainScreenBackground();	
		generateSubScreenBackground();
	}
	@Override
	public void init() {
		if(subscreen)
		{
			ButtonHandler.getButton(1).setColor(GameModel.colors[0]);
			ButtonHandler.getButton(2).setColor(GameModel.colors[2]);
			generateSubScreenForeground();
		}
		else
		{
			ButtonHandler.getButton(1).setColor(GameModel.colors[0]);
			
			ButtonHandler.getButton(5).setColor(GameModel.colors[1]);
			ButtonHandler.getButton(6).setColor(GameModel.colors[4]);
			JoystickHandler.REPEAT = true;
		}
		
		
		
	}

	@Override
	public void update(float factor) {
		if(startanimation){
			buttonInAnimation(animationcounter);
			if(animationcounter == 5){
				startanimation = false;
			}
		}else if(subscreen){
			//nextScreen();
		}else{
			//previousScreen();
		}
	     if(selected != oldselected){
	    	 for(int i = 0; i < buttons.size(); i++){
		    	 buttons.get(i).setSong(selectedToSong(selected+(i-2)));
		    	 
	    	 }
	    	 oldselected = selected;
	     }
	     if(difSelect != oldDifSelect)
	     {
	    	 oldDifSelect = difSelect;
	    	 generateSubScreenForeground();
	     }
	     index++;
	     
	}

	@Override
	public void draw(Graphics2D g2) {

		buttons.clear();		
		buttons.add(new MenuButton(300,260,GameModel.colors[kleurButton1],selectedToSong(selected-2)));
		buttons.add(new MenuButton(300,380,GameModel.colors[kleurButton2],selectedToSong(selected-1)));
		buttons.add(new MenuButton(260,520,1020,80,GameModel.colors[kleurButton3],selectedToSong(selected)));
		buttons.add(new MenuButton(300,680,GameModel.colors[kleurButton4],selectedToSong(selected+1)));
		buttons.add(new MenuButton(300,800,GameModel.colors[kleurButton5],selectedToSong(selected+2)));
		buttons.get(2).setSelected(true);
		
		
		g2.setColor(Color.BLACK);
		Font textFont2 = new Font("OCR A Extended", Font.BOLD, 50);
		g2.setFont(textFont2);
		if(!subscreen) {
			g2.drawImage(mainScreenBackground, 0, 0, 1280,1024,null);
			for(MenuButton b:buttons){
		    	 b.draw(g2);
		     }
		}
		if(subscreen) {
			g2.drawImage(subScreenBackground,0,0,1280,1024,null);
			g2.drawImage(subScreenForeground,0,0,1280,1024,null);
			index%=25;
			int y = (index/5)*75;
			int x = (index%5)*75;
			g2.drawImage(aanwijzers.getSubimage(x, y, 75, 75), 825,900 - difSelect*100,75,75,null);		
		}
	}
	
	@Override
	public void buttonPressed(ButtonEvent e) {
		if(subscreen){								//Screen for Song details
			if(e.getButton().getButtonID() == 2){
				subscreen = false;
				gsm.init();
			}
			if(e.getButton().getButtonID() == 1){
				sh.close();
				gsm.setState(control.GameStateManager.State.PRE_GAME_STATE);
			}
		}else{										//Screen for selecting song
			if(e.getButton().getButtonID() == 1){
				subscreen = true;
				gsm.init();
				generateSubScreenForeground();
			}else if(e.getButton().getButtonID() == 5){
				sh.sort(new SortALPHA());
				selected = 0;
				sh.set(songs.indexOf(selectedToSong(selected)));
				sh.play();	
			}else if(e.getButton().getButtonID() == 6){
				sh.sort(new SortPLAYED());
				selected = 0;
				sh.set(songs.indexOf(selectedToSong(selected)));
				sh.play();	
			}	
		}
		
		if(e.getButton().getButtonID() == 0)
		{
			gsm.setState(State.TITLE_STATE);
		}
	}
	@Override
	public void buttonReleased(ButtonEvent e) {
		
	}
	@Override
	public void onJoystickMoved(JoystickEvent e) {
		
		if(subscreen){		
			if(e.getJoystick().getPos() == Joystick.Position.DOWN){
				difSelect--;
				if(difSelect < 0){
					difSelect += buttons2.size();
				}
			}else if(e.getJoystick().getPos() == Joystick.Position.UP){
				difSelect++;
				if(difSelect > buttons2.size()-1){
					difSelect = 0;
				}
				
			}
			else if(e.getJoystick().getPos() == Joystick.Position.CENTER)	
			{
				sh.set(selectedToSong(selected).getSongs().get(difSelect));
			}
			//generateSubScreenForeground();			
		}else{										//Screen for selecting song
			if(e.getJoystick().getPos() == Joystick.Position.DOWN){
				selected++;
				
				kleurButton1++;
				kleurButton1 %= 6;
				
				kleurButton2++;
				kleurButton2 %= 6;
				
				kleurButton3++;
				kleurButton3 %= 6;
				
				kleurButton4++;
				kleurButton4 %= 6;
				
				kleurButton5++;
				kleurButton5 %= 6;
				
				

			}else if(e.getJoystick().getPos() == Joystick.Position.UP){
				selected--;
				kleurButton1--;
				if(kleurButton1 < 0){
					kleurButton1 = 5;
				}
				
				kleurButton2--;
				if(kleurButton2 < 0){
					kleurButton2 = 5;
				}
				
				kleurButton3--;
				if(kleurButton3 < 0){
					kleurButton3 = 5;
				}
					
				kleurButton4--;
				if(kleurButton4 < 0){
					kleurButton4 = 5;
				}
				
				kleurButton5--;
				if(kleurButton5 < 0){
					kleurButton5 = 5;
				}
				

				
			}
			
			else if(e.getJoystick().getPos() == Position.CENTER)
			{
				buttons2.clear();
		 		int instanceNr = 0;
		 		for(int i = selectedToSong(selected).getSongs().size(); i>0; i--){
		 			if(sh.getCurrentSong().getSongs().size() == 0)
		 				continue;
		 			SongInstance si = selectedToSong(selected).getSongs().get(i-1);
		 			buttons2.add(new DifficultyButton(yPosDiffButton-instanceNr,si.getDifficulty(), GameModel.colors[i-1]));
		 			instanceNr += 100;
		 		}
				difSelect = 0;
		 		
				sh.set(songs.indexOf(selectedToSong(selected)));
				sh.play();	
			}
		}
	}
	
	public void generateMainScreenBackground(){
		mainScreenBackground = Images.initVolatileImage(1280, 1024, Transparency.OPAQUE);
		Graphics2D g2 = mainScreenBackground.createGraphics();
		RenderingHints rh = new RenderingHints(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
	    g2.setRenderingHints(rh);
		g2.setFont(textFont);
		g2.setPaint(new GradientPaint(0, 0, new Color(0,0,1, 0.6f),1280,1024 ,new Color(0,0,1, 0.2f)));
		g2.fillRect(0, 0, 1280, 1024);	
		g2.setPaint(new GradientPaint(0, 0, new Color(1,1,0, 0.6f),1280,1024 ,new Color(0,0,1, 0.2f)));
		
	 
		Polygon triangle2 = new Polygon(); 
		triangle2.addPoint(0, 0);
		triangle2.addPoint(0, 1024/4+50);
		triangle2.addPoint(1280/2+50, 0);
		g2.fillPolygon(triangle2);
		
		g2.setColor(Color.ORANGE);
		Polygon triangle = new Polygon();
	    triangle.addPoint(0, 0);
	    triangle.addPoint(0, 1024/4);
	    triangle.addPoint(1280/2, 0);
		g2.fillPolygon(triangle);
		
		g2.setColor(Color.BLACK);
		g2.drawString("Main Menu", 32, 60);
		
		g2.setFont(textFontSmall);
	
		//select
		g2.setColor(GameModel.colors[0]);
		g2.fillOval(20, 900, 30, 30);
		g2.drawString("Play", 55, 920);
		
		//letters
		g2.setColor(GameModel.colors[1]);
		g2.fillOval(20, 940, 30, 30);
		g2.drawString("A-Z", 55, 960);
		
		//most played
		g2.setColor(GameModel.colors[4]);
		g2.fillOval(20, 980, 30, 30);
		g2.drawString("Most Played", 55, 1000);		
		g2.dispose();
	
		mainScreenBackground.createGraphics();
		
		
	}
	
	public void generateSubScreenBackground(){
		subScreenBackground = Images.initVolatileImage(1280, 1024, Transparency.OPAQUE);
		Graphics2D g2 = subScreenBackground.createGraphics();
		RenderingHints rh = new RenderingHints(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
	    g2.setRenderingHints(rh);
	    g2.setColor(Color.BLACK);
		GradientPaint gp3 = new GradientPaint(640, 1024/8,Color.BLUE,640,1024 ,Color.WHITE);
		g2.setPaint(gp3);
		g2.fillRect(0, 0, 1280, 1024);
		GradientPaint gp4 = new GradientPaint(0, 0,new Color(1,1,0,0.6f),1280,1024,new Color(1,1,1,0.2f));
		g2.setPaint(gp4);
		g2.fillRect(0,128,1280,25);
		g2.setColor(Color.ORANGE);
		g2.fillRect(0, 0, 1280, 1024/8);
		g2.setColor(Color.BLACK);
		g2.dispose();
		subScreenBackground.createGraphics();
	}
	
	public void generateSubScreenForeground(){
		subScreenForeground = Images.initVolatileImage(1280, 1024, Transparency.TRANSLUCENT);
		Graphics2D g2 = subScreenForeground.createGraphics();
		RenderingHints rh = new RenderingHints(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
		g2.setComposite(AlphaComposite.DstOut);
		g2.fillRect(0, 0, subScreenForeground.getWidth(), subScreenForeground.getHeight());
		g2.setComposite(AlphaComposite.SrcOver);
	    g2.setRenderingHints(rh);
	    g2.setColor(Color.BLACK);
 		g2.setFont(textFont);
		g2.drawString(selectedToSong(selected).getTitle(), 30, 60);
		
		g2.setColor(Color.WHITE);
		g2.drawString("Author: " + selectedToSong(selected).getAuthor(), 30, 200);
		
		boolean highscoresFound = true;
		int HIGHSCORES_TO_DISPLAY = 5;
		
		List<Highscore> highscores = sql.getHighscoresToday(selectedToSong(selected), sh.getCurrentSong().getSongs().get(difSelect));
		if(highscores.isEmpty())
		{
			highscores = sql.getHighscores(selectedToSong(selected), sh.getCurrentSong().getSongs().get(difSelect));
			if(highscores.isEmpty())
			{
				highscoresFound = false;
			}
			else
				g2.drawString("All Time Highscores", 30, 300);
		}
		else
			g2.drawString("Daily Highscore", 30, 300);
		
		if(highscoresFound)
		{
			HIGHSCORES_TO_DISPLAY = HIGHSCORES_TO_DISPLAY > highscores.size() ? highscores.size() : HIGHSCORES_TO_DISPLAY;
			
			for(int i = 0; i<HIGHSCORES_TO_DISPLAY; i++)
			{
				Highscore hi = highscores.get(i);
				
				g2.drawString(hi.getName() + " - " + hi.getScore(), 30, 400 + i*100);
			}
		}
		else
		{
			g2.drawString("No Highscores found", 30, 400);
		}
		
		
		for(DifficultyButton b : buttons2){
			b.draw(g2);
		}
		
		g2.setFont(textFontSmall);
		
		//play song
		g2.setColor(GameModel.colors[0]);
		g2.fillOval(20, 860, 30, 30);
		g2.drawString("Play", 55, 880);
		
		//back
		g2.setColor(GameModel.colors[2]);
		g2.fillOval(20, 900, 30, 30);
		g2.drawString("Back", 55, 920);
		
		//up
		g2.setColor(Color.BLACK);		
		Polygon p  = new Polygon();
		p.addPoint(35, 940);
		p.addPoint(20, 960);
		p.addPoint(50, 960);	
		g2.fillPolygon(p);
		g2.drawString("Up", 55, 960);
		
		//down
		Polygon p2  = new Polygon();
		p2.addPoint(35, 1000);
		p2.addPoint(20, 980);
		p2.addPoint(50, 980);	
		g2.fillPolygon(p2);

		g2.drawString("Down", 55, 1000);	
		
		
		
	    g2.dispose();
	    subScreenForeground.createGraphics();
	}
	
	public void buttonInAnimation(int button){
		if(button >= 0 && button < buttons.size() ){
			buttons.get(button).setX(buttons.get(button).getX()+40);
			if(buttons.get(button).getX() >= 320){
				animationcounter++;
			}
			if(buttons.get(button).getX() >= -200){
				buttonInAnimation(button+1);
			}
		}
	}

	/*
	public void nextScreen(){
		if(buttons.get(0).getX() > -700){
			for(MenuButton b:buttons){
				b.setX(b.getX()-100);
			}
		}
	}
	
	public void previousScreen(){
		if(buttons.get(0).getX() < 300){
			for(MenuButton b:buttons){
				b.setX(b.getX()+100);
			}
		}
	}
	*/
	
	public Song selectedToSong(int s){
		s %= songs.size();
		if(s < 0){
			return songs.get(songs.size()+s);
		}else{
			return songs.get(s);
		}
	}
	
}