package cytoscape.view;

import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.swing.Icon;
import javax.swing.tree.DefaultMutableTreeNode;

import cytoscape.CyNetwork;
import cytoscape.Cytoscape;
import cytoscape.util.swing.AbstractTreeTableModel;

/**
 * Inner class that extends the AbstractTreeTableModel
 */
public final class NetworkTreeTableModel extends AbstractTreeTableModel {

	private List<ColumnTypes> columnNames;
	private final Map<String, Icon> networkIcons;

	public NetworkTreeTableModel(Object root) {
		super(root);
		columnNames = new ArrayList<ColumnTypes>();
		columnNames.add(ColumnTypes.NETWORK);
		columnNames.add(ColumnTypes.NODES);
		columnNames.add(ColumnTypes.EDGES);

		networkIcons = new HashMap<String, Icon>();
	}

	public void addColumn(final ColumnTypes model, final int idx) {
		columnNames.add(idx, model);
	}

	public void removeColumn(final int idx) {
		columnNames.remove(idx);
	}

	public Object getChild(Object parent, int index) {
		final Enumeration<?> tree_node_enum = ((DefaultMutableTreeNode) getRoot())
				.breadthFirstEnumeration();

		while (tree_node_enum.hasMoreElements()) {
			DefaultMutableTreeNode node = (DefaultMutableTreeNode) tree_node_enum
					.nextElement();

			if (node == parent) {
				return node.getChildAt(index);
			}
		}

		return null;
	}

	public int getChildCount(Object parent) {
		Enumeration<?> tree_node_enum = ((DefaultMutableTreeNode) getRoot())
				.breadthFirstEnumeration();

		while (tree_node_enum.hasMoreElements()) {
			DefaultMutableTreeNode node = (DefaultMutableTreeNode) tree_node_enum
					.nextElement();

			if (node == parent) {
				return node.getChildCount();
			}
		}

		return 0;
	}

	public int getColumnCount() {
		return columnNames.size();
	}

	public String getColumnName(int column) {
		return columnNames.get(column).getDisplayName();
	}

	public Class<?> getColumnClass(int column) {
		return columnNames.get(column).getType();
	}

	public Object getValueAt(Object node, int column) {
		if (columnNames.get(column).equals(ColumnTypes.NETWORK))
			return ((DefaultMutableTreeNode) node).getUserObject();
		else if (columnNames.get(column).equals(ColumnTypes.NODES)) {
			CyNetwork cyNetwork = Cytoscape.getNetwork(((NetworkTreeNode) node)
					.getNetworkID());

			return "" + cyNetwork.getNodeCount() + "("
					+ cyNetwork.getSelectedNodes().size() + ")";
		} else if (columnNames.get(column).equals(ColumnTypes.EDGES)) {
			CyNetwork cyNetwork = Cytoscape.getNetwork(((NetworkTreeNode) node)
					.getNetworkID());

			return "" + cyNetwork.getEdgeCount() + "("
					+ cyNetwork.getSelectedEdges().size() + ")";
		} else if (columnNames.get(column).equals(ColumnTypes.NETWORK_ICONS)) {

			return networkIcons.get(((NetworkTreeNode) node).getNetworkID());
		}

		return "";
	}

	public void setValueAt(Object aValue, Object node, int column) {
		if (columnNames.get(column).equals(ColumnTypes.NETWORK)) {
			((DefaultMutableTreeNode) node).setUserObject(aValue);
		} else if (columnNames.get(column).equals(ColumnTypes.NETWORK_ICONS)) {
			if(aValue == null)
				networkIcons.put(((NetworkTreeNode) node).getNetworkID(), null);
			else
				networkIcons.put(((NetworkTreeNode) node).getNetworkID(),
					(Icon) aValue);
		}
	}
}