package jcgp.gui.settings;

import java.util.ArrayList;

import javafx.event.ActionEvent;
import javafx.event.EventHandler;
import javafx.geometry.Insets;
import javafx.scene.control.Button;
import javafx.scene.control.CheckBox;
import javafx.scene.control.ComboBox;
import javafx.scene.control.ScrollPane;
import javafx.scene.layout.AnchorPane;
import javafx.scene.layout.HBox;
import javafx.scene.layout.Priority;
import javafx.scene.layout.VBox;
import javafx.scene.text.Font;
import javafx.scene.text.FontSmoothingType;
import javafx.scene.text.Text;
import jcgp.JCGP;
import jcgp.backend.function.FunctionSet;
import jcgp.backend.modules.ea.EvolutionaryAlgorithm;
import jcgp.backend.modules.fitness.FitnessFunction;
import jcgp.backend.modules.mutator.Mutator;
import jcgp.backend.resources.parameters.Parameter;
import jcgp.gui.GUI;
import jcgp.gui.settings.parameters.GUIParameter;

public class SettingsPane extends AnchorPane {
	
	private VBox mainContainer;
	private VBox baseParameterPane, eaPane, mutatorPane, ffPane, nfPane;
	
	private Button runPause = new Button("Run"), step = new Button("Step"), reset = new Button("Reset");
	
	private ArrayList<GUIParameter> parameters = new ArrayList<GUIParameter>();
		
	public SettingsPane(JCGP cgp, GUI gui) {
		super();
		
		mainContainer = new VBox(8);
		mainContainer.setPadding(new Insets(5, GUI.RESIZE_MARGIN, 0, 2));
		
		setMinWidth(GUI.SETTINGS_WIDTH);
		setPrefWidth(GUI.SETTINGS_WIDTH);
		
		initialiseBaseParameters(cgp);
		
		initialiseEAParameters(cgp);
		
		initialiseMutatorParameters(cgp);
		
		initialiseProblemTypeParameters(cgp, gui);
		
		createControls(cgp, gui);
		
		ScrollPane scroll = new ScrollPane();
		scroll.setFitToWidth(true);
		scroll.setContent(mainContainer);
		scroll.setStyle("-fx-background-color: #FFFFFF");
		
		AnchorPane.setTopAnchor(scroll, 0.0);
		AnchorPane.setBottomAnchor(scroll, 0.0);
		AnchorPane.setRightAnchor(scroll, 0.0);
		AnchorPane.setLeftAnchor(scroll, GUI.RESIZE_MARGIN);
		
		getChildren().add(scroll);
	}

	private void initialiseBaseParameters(JCGP cgp) {
		baseParameterPane = new VBox(2);
		
		Text header = new Text("Base Parameters");
		header.setFont(Font.font("Arial", 14));
		header.setUnderline(true);
		
		baseParameterPane.getChildren().add(header);

		parameters.add(GUIParameter.create(cgp.getResources().getParameter("rows"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("columns"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("inputs"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("outputs"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("levelsBack"), this));
		
		GUIParameter gp = GUIParameter.create(cgp.getResources().getParameter("popSize"), this);
		gp.setPadding(new Insets(0, 0, 10, 0));
		parameters.add(gp);
		
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("currentGen"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("generations"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("currentRun"), this));
		
		gp = GUIParameter.create(cgp.getResources().getParameter("runs"), this);
		gp.setPadding(new Insets(0, 0, 10, 0));
		parameters.add(gp);
		
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("seed"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("report"), this));
		parameters.add(GUIParameter.create(cgp.getResources().getParameter("maxFitness"), this));
		
		baseParameterPane.getChildren().addAll(parameters);
		mainContainer.getChildren().add(baseParameterPane);
	}

	private void initialiseEAParameters(final JCGP cgp) {
		eaPane = new VBox(2);
		
		Text header = new Text("Evolutionary Algorithm");
		header.setFont(Font.font("Arial", 14));
		header.setUnderline(true);
		
		final ComboBox<EvolutionaryAlgorithm> eaCBox = new ComboBox<EvolutionaryAlgorithm>();
		eaCBox.getItems().addAll(cgp.getEvolutionaryAlgorithms());
		eaCBox.getSelectionModel().select(cgp.getEvolutionaryAlgorithm());
		eaCBox.prefWidthProperty().bind(mainContainer.widthProperty());
		
		final VBox eaParameters = new VBox();
		eaParameters.setSpacing(2);
		
		if (cgp.getEvolutionaryAlgorithm().getLocalParameters() != null) {
			refreshParameters(cgp.getEvolutionaryAlgorithm().getLocalParameters(), eaParameters);
		}
		
		eaCBox.setOnAction(new EventHandler<ActionEvent>() {
			@Override
			public void handle(ActionEvent event) {
				cgp.setEvolutionaryAlgorithm(eaCBox.getSelectionModel().getSelectedIndex());
				if (eaCBox.getSelectionModel().getSelectedItem().getLocalParameters() != null) {
					refreshParameters(eaCBox.getSelectionModel().getSelectedItem().getLocalParameters(), eaParameters);
				}
			}
		});
		
		eaPane.getChildren().addAll(header, eaCBox, eaParameters);	
		mainContainer.getChildren().add(eaPane);
	}

	private void initialiseMutatorParameters(final JCGP cgp) {
		mutatorPane = new VBox(2);
		
		Text header = new Text("Mutator");
		header.setFontSmoothingType(FontSmoothingType.LCD);
		header.setFont(Font.font("Arial", 14));
		header.setUnderline(true);
		
		final ComboBox<Mutator> mutatorCBox = new ComboBox<Mutator>();
		mutatorCBox.getItems().addAll(cgp.getMutators());
		mutatorCBox.getSelectionModel().select(cgp.getMutator());
		mutatorCBox.prefWidthProperty().bind(mainContainer.widthProperty());
		
		final VBox mutatorParameters = new VBox();
		mutatorParameters.setSpacing(2);
		if (cgp.getEvolutionaryAlgorithm().getLocalParameters() != null) {
			refreshParameters(cgp.getMutator().getLocalParameters(), mutatorParameters);
		}
		
		mutatorCBox.setOnAction(new EventHandler<ActionEvent>() {
			@Override
			public void handle(ActionEvent event) {
				cgp.setEvolutionaryAlgorithm(mutatorCBox.getSelectionModel().getSelectedIndex());
				if (mutatorCBox.getSelectionModel().getSelectedItem().getLocalParameters() != null) {
					refreshParameters(mutatorCBox.getSelectionModel().getSelectedItem().getLocalParameters(), mutatorParameters);
				}			
			}
		});
		
		mutatorPane.getChildren().addAll(header, mutatorCBox, mutatorParameters);
		mainContainer.getChildren().add(mutatorPane); 
	}
	
	private void initialiseProblemTypeParameters(final JCGP cgp, final GUI gui) {
		ffPane= new VBox(2);
		
		Text header = new Text("Problem type");
		header.setFont(Font.font("Arial", 14));
		header.setUnderline(true);
		
		final ComboBox<FitnessFunction> ffCBox = new ComboBox<FitnessFunction>();
		ffCBox.getItems().addAll(cgp.getFitnessFunctions());
		ffCBox.getSelectionModel().select(cgp.getFitnessFunction());
		ffCBox.prefWidthProperty().bind(mainContainer.widthProperty());
		
		final VBox ffParameters = new VBox();
		ffParameters.setSpacing(2);
		if (cgp.getFitnessFunction().getLocalParameters() != null) {
			refreshParameters(cgp.getFitnessFunction().getLocalParameters(), ffParameters);
		}
		
		ffCBox.setOnAction(new EventHandler<ActionEvent>() {
			@Override
			public void handle(ActionEvent event) {
				cgp.setEvolutionaryAlgorithm(ffCBox.getSelectionModel().getSelectedIndex());
				if (ffCBox.getSelectionModel().getSelectedItem().getLocalParameters() != null) {
					refreshParameters(ffCBox.getSelectionModel().getSelectedItem().getLocalParameters(), ffParameters);
				}		
			}
		});
		
		final VBox nfParameters = new VBox();
		nfParameters.setSpacing(2);
		refreshFunctions(cgp.getResources().getFunctionSet(), nfParameters, gui);
		
		ffPane.getChildren().addAll(header, ffCBox, ffParameters, nfParameters); 
		mainContainer.getChildren().add(ffPane);
		
	}
	
	private void createControls(final JCGP cgp, final GUI gui) {
		final VBox controls = new VBox(2);
		controls.setFillWidth(true);
		
		final HBox flowButtons = new HBox(2);
		runPause.setOnAction(new EventHandler<ActionEvent>() {
			@Override
			public void handle(ActionEvent event) {				
				gui.runPause();
			}
		});
		
		step = new Button("Step");
		step.setOnAction(new EventHandler<ActionEvent>() {
			@Override
			public void handle(ActionEvent event) {
				gui.step();
			}
		});
		
		reset = new Button("Reset");
		reset.setOnAction(new EventHandler<ActionEvent>() {
			@Override
			public void handle(ActionEvent event) {
				gui.reset();
			}
		});
		
		flowButtons.getChildren().addAll(runPause, step, reset);
		
		HBox.setHgrow(runPause, Priority.ALWAYS);
		runPause.setMaxWidth(Double.MAX_VALUE);
		HBox.setHgrow(step, Priority.ALWAYS);
		step.setMaxWidth(Double.MAX_VALUE);
		HBox.setHgrow(reset, Priority.ALWAYS);
		reset.setMaxWidth(Double.MAX_VALUE);
		
		controls.getChildren().add(flowButtons);
		
		mainContainer.getChildren().add(controls);	
	}
	
	/**
	 * @param cgp
	 * @param vb
	 */
	private void refreshParameters(Parameter[] newParameters, VBox vb) {
		parameters.removeAll(vb.getChildren());
		vb.getChildren().clear();
		for (int i = 0; i < newParameters.length; i++) {
	        GUIParameter gp = GUIParameter.create(newParameters[i], this);
	        
	        parameters.add(gp);
        	vb.getChildren().add(gp);
		}
		revalidateParameters();
	}
	
	private void refreshFunctions(final FunctionSet fs, VBox vb, final GUI gui) {
		vb.getChildren().clear();
		CheckBox cb;
		for (int i = 0; i < fs.getTotalFunctionCount(); i++) {
			cb = new CheckBox(fs.getFunction(i).getName());
			cb.setId(String.valueOf(i));
			cb.setSelected(fs.isEnabled(fs.getFunction(i)));
			final int index = i;
			cb.setOnAction(new EventHandler<ActionEvent>() {
				@Override
				public void handle(ActionEvent event) {
					if (((CheckBox) event.getSource()).isSelected()) {
						fs.enableFunction(index);
					} else {
						fs.disableFunction(index);
					}
					GUI.updateFunctionSelector();
				}
			});
			vb.getChildren().add(cb);
			
			GUI.updateFunctionSelector();
		}
	}
	

	
	public void runningMode(boolean value) {
		baseParameterPane.setDisable(value);
		eaPane.setDisable(value);
		mutatorPane.setDisable(value);
		ffPane.setDisable(value);
		nfPane.setDisable(value);
		step.setDisable(value);
		reset.setDisable(value);
		
		runPause.setText(value ? "Pause" : "Run");
	}
	
	/**
	 * 
	 * @return true if the experiment needs to be reset, false otherwise.
	 */
	public boolean isResetRequired() {
		for (GUIParameter parameter : parameters) {
			if (parameter.requiresReset()) {
				return true;
			}
		}
		return false;
	}
	
	public boolean areParametersValid() {
		for (GUIParameter parameter : parameters) {
			if (!parameter.isValid()) {
				return false;
			}
		}
		return true;
	}
	
	public void revalidateParameters() {
		runPause.setDisable(false);
		for (GUIParameter parameter : parameters) {
			parameter.validate();
			if (parameter.requiresReset()) {
				runPause.setDisable(true);
			}
		}
	}
	
	public void applyParameters() {
		for (GUIParameter parameter : parameters) {
			parameter.applyValue();
		}	
	}
}