/*
 * @(#)CSetBuilderUI.java       0.0.0 99/07/02
 *
 * Copyright (c) 1999 by Willie Wheeler.  All rights reserved.
 */
package stats.plaf.concrete;

import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import javax.swing.border.EmptyBorder;
import javax.swing.plaf.ComponentUI;
import stats.SetBuilder;
import stats.plaf.SetBuilderUI;

/**
 * Notes: Willie needs to go through the code and replace references to
 * samples, sample elements, etc., with references to data sets, data,
 * etc.
 *
 * @version     0.0.0 07/02/99
 * @author      Willie Wheeler
 */
public class CSetBuilderUI extends SetBuilderUI implements ActionListener {

    /**
     * <tt>SetBuilder</tt> for which we are providing a PLAF.
     */
    protected SetBuilder                  builder;

    protected ImageIcon[]                 atoms;
    protected Image[]                     atomImages;
    protected JPanel                      samplePanel;

    // Action commands generated by atom and sample element buttons.
    protected static final String         ADD = "add";
    protected static final String         REMOVE = "remove";

    /**
     * Creates a new <tt>CSetBuilderUI</tt> instance.
     */
    public static ComponentUI createUI(JComponent c) {
	return new CSetBuilderUI();
    }

    /**
     * Installs <tt>this</tt> UI as the PLAF for the passed
     * <tt>SetBuilder</tt>.
     */
    public void installUI(JComponent c) {
	System.err.println("CSetBuilderUI.installUI() entered.");
	builder = (SetBuilder)c;
	setAtoms(builder.getAtoms());
	installDefaults();
	installComponents();
    }

    protected void installDefaults() {
	builder.setOpaque(true);
	builder.setLayout(new BorderLayout());
    }

    protected void installComponents() {
	JSplitPane splitPane = new JSplitPane();
	splitPane.setOrientation(JSplitPane.VERTICAL_SPLIT);
	builder.add(splitPane, BorderLayout.CENTER);

	// Build top panel for atom label and atom panel.
	JPanel topPanel = new JPanel();
	topPanel.setLayout(new BorderLayout());

	// Build atom label.
	JLabel atomLabel = new JLabel("Atoms", JLabel.CENTER);
	topPanel.add(atomLabel, BorderLayout.NORTH);

	// Build atom panel.
	JPanel atomPanel = new JPanel();
	atomPanel.setBackground(Color.white);
	atomPanel.setLayout(new GridLayout(1, atoms.length));
	for(int i=0; i<atoms.length; i++) {

	    // Build button.
	    JButton button = new JButton(atoms[i]);
	    button.setVerticalAlignment(JButton.TOP);
	    button.setOpaque(false);
	    button.setBorder(new EmptyBorder(0, 0, 0, 0));
	    button.setActionCommand(ADD);
	    button.addActionListener(this);
	    atomPanel.add(button);
	}
	topPanel.add(atomPanel, BorderLayout.CENTER);
	splitPane.setTopComponent(topPanel);

	// Build bottom panel for sample label and sample.
	JPanel bottomPanel = new JPanel();
	bottomPanel.setLayout(new BorderLayout());

	// Build sample label.
	JLabel sampleLabel = new JLabel("Set", JLabel.CENTER);
	bottomPanel.add(sampleLabel, BorderLayout.NORTH);

	// Build sample panel.
	samplePanel = new JPanel();
	samplePanel.setBackground(Color.pink);
	samplePanel.setLayout(new GridLayout(0, 10));
	bottomPanel.add(samplePanel, BorderLayout.CENTER);
	splitPane.setBottomComponent(bottomPanel);
    }

    /**
     * Removes <tt>this</tt> UI as the PLAF for the passed <tt>SetBuilder</tt>.
     */
    public void uninstallUI(JComponent c) {
	uninstallComponents();
	uninstallDefaults();
	builder = null;
    }

    protected void uninstallComponents() { builder.removeAll(); }

    protected void uninstallDefaults() { builder.setLayout(null); }

    public void setAtoms(ImageIcon[] atoms) {
	this.atoms = atoms;

	// Build scaled images from the atom icons.  These scaled images will
	// be used for the sample.
	Image image = null;
	atomImages = new Image[atoms.length];
	for(int i=0; i<atoms.length; i++) {
	    image = atoms[i].getImage();
	    atomImages[i] =
		image.getScaledInstance(3*image.getWidth(null)/5,
					3*image.getHeight(null)/5,
					Image.SCALE_SMOOTH);
	}
    }

    /**
     * Updates sample model in response to mouse presses on the atoms and
     * sample elements.
     *
     * @param     e       action event
     */
    public void actionPerformed(ActionEvent e) {
	String cmd = e.getActionCommand();
	JButton src = (JButton)e.getSource();

	if (cmd == ADD) {

	    // We search for the atom that was just pressed, since there
	    // doesn't seem to be a method in JComponent to tell us.
	    for (int i=0; i<atoms.length; i++) {
		if (src.getIcon() == atoms[i]) {

		    // Found the atom.  First we add a corresponding datum to
		    // the data panel.
		    ImageIcon elemIcon =
			new ImageIcon(atomImages[i],
				      atoms[i].getDescription());
		    JButton elem = new JButton(elemIcon);
		    elem.setVerticalAlignment(JButton.TOP);
		    elem.setOpaque(false);
		    elem.setBorder(new EmptyBorder(0, 0, 0, 0));
		    elem.setActionCommand(REMOVE);
		    elem.addActionListener(this);
		    samplePanel.add(elem);
		    samplePanel.invalidate();
		    samplePanel.validate();

		    // Now we tell the builder to update the model.
		    builder.addAtom(i);           // i = atom index

		    break;
		}
	    }

	} else if (cmd == REMOVE) {

	    // We search for the sample element that was just pressed, since
	    // there doesn't seem to be a method in JComponent to tell us.
	    Component[] sampleElements = samplePanel.getComponents();
	    for (int i=0; i<sampleElements.length; i++) {
		if (src == sampleElements[i]) {

		    // Found the sample element.  First we remove it from the
		    // sample panel.
		    samplePanel.remove(src);
		    samplePanel.invalidate();
		    samplePanel.validate();

		    // Now we tell the builder to update the model.
		    builder.removeDatum(i);       // i = datum index

		    break;
		}
	    }
	}
    }
}