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.Date;
import java.util.List;

import model.GameModel;
import model.SongHandler;
import model.objects.DifficultyButton;
import model.objects.Highscore;
import model.objects.MenuButton;
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.JoystickEvent;
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 = new SQLConnector();

	private int animationcounter;
	private boolean subscreen, startanimation;
	private VolatileImage mainScreenBackground, subScreenBackground, subScreenForeground;
	
	int yPosDiffButton = 900;
	private int difSelect=0;
	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;

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

	@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;
	    	 
	    	 
	    	buttons2.clear();
	 		int instanceNr = 0;
	 		for(int i = sh.getCurrentSong().getSongs().size(); i>0; i--){
	 			if(sh.getCurrentSong().getSongs().size() == 0)
	 				continue;
	 			SongInstance si = sh.getCurrentSong().getSongs().get(i-1);
	 			buttons2.add(new DifficultyButton(yPosDiffButton-instanceNr,si.getDifficulty(), GameModel.colors[i-1]));
	 			instanceNr += 100;
	 		}
	     }
	     index++;
	     
	}

	@Override
	public void draw(Graphics2D g2) {

		buttons.clear();		
		buttons.add(new MenuButton(400,250,GameModel.colors[kleurButton1],selectedToSong(selected-2)));
		buttons.add(new MenuButton(400,390,GameModel.colors[kleurButton2],selectedToSong(selected-1)));
		buttons.add(new MenuButton(360,525,920,80,GameModel.colors[kleurButton3],selectedToSong(selected)));
		buttons.add(new MenuButton(400,670,GameModel.colors[kleurButton4],selectedToSong(selected+1)));
		buttons.add(new MenuButton(400,810,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.PLAY_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());
				oldselected = 1;
				selected = 0;
			}else if(e.getButton().getButtonID() == 6){
				sh.sort(new SortPLAYED());
				oldselected = 1;
				selected = 0;
			}	
		}
		
		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;
				}
				
//				System.out.println(difSelect);
			}
			sh.set(sh.getCurrentSong().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;
				}
				

				
			}
			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);
		
		
		
		
		List<Highscore> highscores = sql.getHighscore(selectedToSong(selected), sh.getCurrentSong().getSongs().get(difSelect));
		List<Highscore> dailyHighs = new ArrayList<Highscore>();
		int highest = 0;
		int oldHighest = 0;
		
		Date date = new Date();
		int time = date.getDay();
		int dailyHighest = 0;
		int oldDailyHighest = 0;
		
		
		String name = "";
		String dailyName = "";
		
		for(Highscore h:highscores){
			
			oldHighest = h.getScore();
			if(oldHighest > highest){
				highest = oldHighest;
				name = h.getName();
			}
			
			
			if(time == h.getDate().getDay())
				dailyHighs.add(h);
			for(int i = 0; i < dailyHighs.size(); i ++){
				oldDailyHighest = dailyHighs.get(i).getDate().getDay();
				if(oldDailyHighest > dailyHighest){
					dailyHighest = oldDailyHighest;
					dailyName = h.getName();
				}
			}	
		}
		
		
		g2.setColor(Color.WHITE);
		g2.drawString("Author: " + selectedToSong(selected).getAuthor(), 30, 200);
		g2.drawString("Overall Highscore: " + highest , 30, 300);
		g2.drawString("Overall Highscore by: " + name, 30, 400);
		g2.drawString("Daily Highscore: " + dailyHighest, 30, 500);
		g2.drawString("Daily Highscore by: : " + dailyName, 30, 600);
		
		
		
		
		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);
		}
	}
	
}
