package berack96.sim.util.graph.view;

import java.awt.Color;
import java.awt.Component;
import java.awt.GridLayout;
import java.awt.event.ItemEvent;
import java.io.IOException;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.swing.BorderFactory;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JTextField;
import javax.swing.border.BevelBorder;

import berack96.sim.util.graph.Graph;
import berack96.sim.util.graph.view.edge.EdgeListener;
import berack96.sim.util.graph.view.vertex.VertexListener;
import berack96.sim.util.graph.visit.VisitStrategy;

public class GraphInfo<V, W extends Number> extends JPanel {

	private static final long serialVersionUID = 1L;
	
    private final Map<String, VisitListener<V>> visits;

    public GraphInfo(GraphPanel<V, W> graphPanel, VertexListener<V> vListener, EdgeListener<V, W> eListener, Set<VisitStrategy<V, W>> visits) {
        this.visits = new HashMap<>();

        /* ZERO (DESCRIPTION) */
        JLabel listenerDescription = new JLabel();
        
        JPanel panelDescription = new JPanel();
        panelDescription.setOpaque(false);
        panelDescription.add(listenerDescription);
        
        /* FIRST (GRAPH INFO) */

        JLabel vNumber = new JLabel(String.valueOf(graphPanel.getGraph().numberOfVertices()));
        JLabel eNumber = new JLabel(String.valueOf(graphPanel.getGraph().numberOfEdges()));
        JLabel gCyclic = new JLabel(String.valueOf(graphPanel.getGraph().isCyclic()));

        List<Component> components = new LinkedList<>();
        JLabel selected = new JLabel();
        JComboBox<String> comboBox = new JComboBox<>();
        comboBox.addItemListener(e -> {
            if (e.getStateChange() == ItemEvent.SELECTED) {
                try {
                    String clazz = (String) e.getItem();
                    VisitListener<V> listener = this.visits.get(clazz);
                    
                    selected.setText(listener != null? "visit":"nothing");
                	listenerDescription.setText(listener != null? listener.getDescription():"");
                	
	                graphPanel.getGraph().unMarkAll();
	                graphPanel.repaint();
	                graphPanel.setGraphListener(listener);
                    
                } catch (Exception ignore) {}
            }

        });

        comboBox.addItem("None");
        for(VisitStrategy<V, W> strategy: visits) {
            String clazz = strategy.getClass().getSimpleName();
            VisitListener<V> visit = new VisitListener<>(graphPanel, strategy);
            comboBox.addItem(clazz);
            this.visits.put(clazz, visit);
        }

        components.add(new JLabel("Visit Strategy: "));
        components.add(comboBox);
        components.add(new JLabel("Selected modality: "));
        components.add(selected);
        components.add(new JLabel("Vertex Number: "));
        components.add(vNumber);
        components.add(new JLabel("Edge Number: "));
        components.add(eNumber);
        components.add(new JLabel("Is Cyclic: "));
        components.add(gCyclic);

        JPanel panelInfo = new JPanel();
        panelInfo.setOpaque(false);
        panelInfo.setBorder(BorderFactory.createLineBorder(Color.RED));
        panelInfo.setLayout(new GridLayout(components.size()/2, 2, 2*2, 2*2));
        components.forEach(panelInfo::add);
        components.clear();

        /* SECOND (VERTEX) */
        JLabel vVertex = new JLabel();
        JLabel vEdgesNumber = new JLabel();
        JLabel vEdgesInNumber = new JLabel();
        JLabel vEdgesOutNumber = new JLabel();

        JButton modVertex = new JButton("Modify Vertices");
        modVertex.addActionListener(a -> {
            comboBox.setSelectedIndex(0);
    		listenerDescription.setText(vListener.getDescription());
            graphPanel.setGraphListener(vListener);
            graphPanel.getGraph().unMarkAll();
            graphPanel.repaint();
            selected.setText("vertices");
        });
        
        JButton modEdge = new JButton("Modify Edges");
        modEdge.addActionListener(a -> {
            comboBox.setSelectedIndex(0);
    		listenerDescription.setText(eListener.getDescription());
            graphPanel.setGraphListener(eListener);
            graphPanel.getGraph().unMarkAll();
            graphPanel.repaint();
            selected.setText("edges");
        });

        components.add(modVertex);
        components.add(modEdge);
        components.add(new JLabel("Vertex name: "));
        components.add(vVertex);
        components.add(new JLabel("Edges: "));
        components.add(vEdgesNumber);
        components.add(new JLabel("Edges IN: "));
        components.add(vEdgesInNumber);
        components.add(new JLabel("Edges OUT: "));
        components.add(vEdgesOutNumber);
        
        JPanel panelVertex = new JPanel();
        panelVertex.setOpaque(false);
        panelVertex.setLayout(new GridLayout(components.size()/2, 2, 2*2, 2*2));
        components.forEach(panelVertex::add);
        components.clear();

        /* SAVE/LOAD errors */
        JLabel textResult = new JLabel();
        textResult.setForeground(Color.RED);
        
        JPanel panelErrors = new JPanel();
        panelErrors.setOpaque(false);
        panelErrors.add(textResult);
        
        /* SAVE/LOAD */
        JTextField fileText = new JTextField();
        JButton saveB = new JButton("Save");
        saveB.addActionListener(a -> {
        	try {
				graphPanel.save(fileText.getText());
				textResult.setText("");
			} catch (IOException e1) {
				e1.printStackTrace();
				textResult.setText(e1.getMessage());
			}
        });
        JButton loadB = new JButton("Load");
        loadB.addActionListener(a -> {
        	try {
				graphPanel.load(fileText.getText());
				textResult.setText("");
			} catch (IOException e1) {
				e1.printStackTrace();
				textResult.setText(e1.getMessage());
			}
        });
        components.add(new JLabel("File to save/load: "));
        components.add(fileText);
        components.add(saveB);
        components.add(loadB);
        
        JPanel panelSave = new JPanel();
        panelSave.setOpaque(false);
        panelSave.setBorder(BorderFactory.createLineBorder(Color.RED));
        panelSave.setLayout(new GridLayout(components.size()/2, 2, 2*2, 2*2));
        components.forEach(panelSave::add);
        components.clear();

        /* ADDING COMPONENTS */
        this.setBackground(Color.LIGHT_GRAY);
        this.setLayout(new BoxLayout(this, BoxLayout.Y_AXIS));
        this.setOpaque(true);
        this.setBorder(BorderFactory.createSoftBevelBorder(BevelBorder.RAISED, Color.GRAY, Color.DARK_GRAY));
        this.add(panelDescription);
        this.add(panelInfo);
        this.add(panelVertex);
        this.add(panelErrors);
        this.add(panelSave);
        /*this.add(panelSave);*/

        modVertex.doClick();

        graphPanel.addObserver((o, arg) -> {
            Graph<V, W> graph = graphPanel.getGraph();
            if(arg.equals(graph)) {
                vNumber.setText(String.valueOf(graph.numberOfVertices()));
                eNumber.setText(String.valueOf(graph.numberOfEdges()));
                gCyclic.setText(String.valueOf(graph.isCyclic()));
                
                /* There should be only one */
                for(V v : graph.getMarkedWith("selected")) {
                	int inE = graph.getEdgesIn(v).size();
                	int outE = graph.getEdgesOut(v).size();
                	
                	vEdgesInNumber.setText(String.valueOf(inE));
                	vEdgesOutNumber.setText(String.valueOf(outE));
                	vEdgesNumber.setText(String.valueOf(inE + outE));
                	vVertex.setText(v.toString());
                }
            }
        });
    }
}