praktikum-holons/src/algorithms/geneticAlgorithm/holegGA/GenAlgoWindow.java

package algorithms.geneticAlgorithm.holegGA;

import java.awt.EventQueue;

import javax.swing.JFrame;
import javax.swing.JSplitPane;

import java.awt.BorderLayout;

import javax.swing.GroupLayout;
import javax.swing.GroupLayout.Alignment;
import javax.swing.BoxLayout;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.JPanel;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreePath;

import ui.controller.Control;
import ui.model.Model;
import ui.view.MyCanvas;
import ui.view.UnitGraph;

import javax.swing.JButton;

import algorithms.geneticAlgorithm.Components.GAResultListener;
import algorithms.geneticAlgorithm.holegGA.Components.HolegIndividual;

import java.awt.event.ActionListener;
import java.awt.event.ActionEvent;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.util.ArrayList;
import java.util.HashMap;

import javax.swing.JTextField;
import javax.swing.JLabel;
import javax.swing.JCheckBox;
import javax.swing.JTextArea;

public class GenAlgoWindow implements GAResultListener<HolegIndividual> {

	JFrame frame;
	Control controller;
	Model model;
	JTree genTree;
	DefaultTreeModel treeModel;
	DefaultMutableTreeNode treeRoot;
	HashMap<DefaultMutableTreeNode, HolegIndividual> treeIndiHashmap;

	GenAlgoHandler algoHandler;
	private JTextField objectAmountField;
	private JTextField edgeAmountField;
	private JCheckBox chckbxOnlyNodes;
	private JTextArea logArea;
	/**
	 * Create the application.
	 */
	public GenAlgoWindow(Control controller, Model model) {
		this.model = model;
		this.controller = controller;
		algoHandler = new GenAlgoHandler(controller, model);
		algoHandler.addListener(this);
		treeIndiHashmap = new HashMap<DefaultMutableTreeNode, HolegIndividual>();
		initialize();
		
		HolegIndividual originIndividual = new HolegIndividual(model.getObjectsOnCanvas());
		ArrayList<HolegIndividual> originPopulation = new ArrayList<HolegIndividual>();
		originPopulation.add(originIndividual);
		populationCreated(-1, originPopulation);
		
	}

	/**
	 * Initialize the contents of the frame.
	 */
	private void initialize() {
		frame = new JFrame();
		frame.setBounds(100, 100, 450, 409);
		frame.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE);
		frame.getContentPane().setLayout(new BoxLayout(frame.getContentPane(), BoxLayout.X_AXIS));
		
	
		JSplitPane mainSplitPane = new JSplitPane();
		mainSplitPane.setResizeWeight(0.3);
		frame.getContentPane().add(mainSplitPane);
		
		JScrollPane treeScrollPane = new JScrollPane();
		mainSplitPane.setLeftComponent(treeScrollPane);
		
		genTree = new JTree();
		treeScrollPane.setViewportView(genTree);
		
		treeModel = (DefaultTreeModel) genTree.getModel();
		treeRoot = new DefaultMutableTreeNode("Generations");
		treeModel.setRoot(treeRoot);
		genTree.setModel(treeModel);
		
		JSplitPane rightSplitPane = new JSplitPane();
		rightSplitPane.setResizeWeight(0.5);
		rightSplitPane.setOrientation(JSplitPane.VERTICAL_SPLIT);
		mainSplitPane.setRightComponent(rightSplitPane);
		
		JPanel panel = new JPanel();
		rightSplitPane.setLeftComponent(panel);
		
		JButton btnCreategeneration = new JButton("createGeneration");
		btnCreategeneration.setBounds(44, 11, 117, 23);
		btnCreategeneration.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent arg0) {
				int objAmount = Integer.parseInt(objectAmountField.getText());
				int edgeAmount = Integer.parseInt(edgeAmountField.getText());
				algoHandler.onlyNodes = chckbxOnlyNodes.isSelected();
				algoHandler.createGeneration(objAmount, edgeAmount);
			}
		});
		panel.setLayout(null);
		
		objectAmountField = new JTextField();
		objectAmountField.setText("0");
		objectAmountField.setBounds(89, 45, 86, 20);
		panel.add(objectAmountField);
		objectAmountField.setColumns(10);
		panel.add(btnCreategeneration);
		
		edgeAmountField = new JTextField();
		edgeAmountField.setText("0");
		edgeAmountField.setBounds(89, 76, 86, 20);
		panel.add(edgeAmountField);
		edgeAmountField.setColumns(10);
		
		JLabel lblMaxObjects = new JLabel("Max Objects");
		lblMaxObjects.setBounds(10, 48, 69, 14);
		panel.add(lblMaxObjects);
		
		JLabel lblMaxEdges = new JLabel("Max Edges");
		lblMaxEdges.setBounds(10, 79, 69, 14);
		panel.add(lblMaxEdges);
		
		chckbxOnlyNodes = new JCheckBox("\r\n");
		chckbxOnlyNodes.setBounds(89, 103, 25, 23);
		panel.add(chckbxOnlyNodes);
		
		JLabel lblOnlyNodes = new JLabel("Only Nodes?");
		lblOnlyNodes.setBounds(10, 107, 69, 14);
		panel.add(lblOnlyNodes);
		
		logArea = new JTextArea();
		rightSplitPane.setRightComponent(logArea);
		
		 MouseListener ml = new MouseAdapter() {
		     public void mousePressed(MouseEvent e) {
		         int selRow = genTree.getRowForLocation(e.getX(), e.getY());
		         TreePath selPath = genTree.getPathForLocation(e.getX(), e.getY());
		         if(selRow != -1) {
		             if(e.getClickCount() == 1) {
		                 //mySingleClick(selRow, selPath);
		             }
		             else if(e.getClickCount() == 2) {
		                 nodeDoubleClick(selRow, selPath);
		             }
		         }
		     }
		 };
		 genTree.addMouseListener(ml);
		
	}
	
	public void nodeDoubleClick(int selRow, TreePath selPath){
		DefaultMutableTreeNode node = (DefaultMutableTreeNode)selPath.getLastPathComponent();
		if(treeIndiHashmap.get(node) != null){
			algoHandler.drawIndividual(treeIndiHashmap.get(node));
			logArea.setText(treeIndiHashmap.get(node).log);
		}
		
	}
	

	@Override
	public void populationCreated(int generation, ArrayList<HolegIndividual> population) {
		DefaultMutableTreeNode genNode;
		if(! (generation == -1)){
			genNode = new DefaultMutableTreeNode("Generation " + generation);
		}else{
			genNode = new DefaultMutableTreeNode("Origin Network");
		}
		
		for(int i = 0; i < population.size(); i++){
			DefaultMutableTreeNode child = new DefaultMutableTreeNode("Individual " + i);
			population.get(i).setId(generation, i);
			treeIndiHashmap.put(child, population.get(i));
			genNode.add(child);
		}
		treeRoot.insert(genNode, 0);
		treeModel.reload();
		
	}
}