/*
 * EquationTreeCellEditor.java
 *
 * Created on August 13, 2003, 10:41 PM
 */

package hart2;
import javax.swing.*;
import javax.swing.event.*;
import java.awt.*;

/**
 *
 * @author  James Hart
 */
public class EquationTreeCellEditor extends DefaultCellEditor {
    
    /** This variable keeps track of the original value being edited*/
    private Relation originalRelation;
    
    /** Creates a new instance of EquationTreeCellEditor */
    public EquationTreeCellEditor() {
        super( new JTextField( "" ) );
    }
    
    /** Overrides the basic functionality of the tree editor*/
    public Component getTreeCellEditorComponent( JTree tree,
                                                 Object value,
                                                 boolean isSelected,
                                                 boolean expanded,
                                                 boolean leaf,
                                                 int row) {
        super.getTreeCellEditorComponent(tree, value, isSelected, expanded, leaf, row);
        if (value instanceof Relation) {
            ((JTextField)getComponent()).setText( ((Relation)value).getName() );
            originalRelation = (Relation)value;
        }
        else {
            originalRelation = null;
        }
        Component retVal = getComponent();
        return retVal;
    }
    
    /** Returns the relation that was most recently edited */
    public Relation getEditedRelation() {
        return originalRelation;
    }
}