com.alee.utils.swing.WebDefaultCellEditor Maven / Gradle / Ivy

Go to download
/*
 * This file is part of WebLookAndFeel library.
 *
 * WebLookAndFeel library is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * WebLookAndFeel library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with WebLookAndFeel library.  If not, see .
 */

package com.alee.utils.swing;

import com.alee.managers.hotkey.Hotkey;
import com.alee.managers.style.StyleId;

import javax.swing.*;
import javax.swing.table.TableCellEditor;
import javax.swing.tree.TreeCellEditor;
import java.awt.*;
import java.awt.event.*;
import java.io.Serializable;
import java.util.EventObject;

/**
 * @author Mikle Garin
 */

public class WebDefaultCellEditor extends AbstractCellEditor implements TableCellEditor, TreeCellEditor
{
    public static final String COMBOBOX_CELL_EDITOR = "JComboBox.isTableCellEditor";

    protected C editorComponent;
    protected EditorDelegate delegate;
    protected int clickCountToStart = 2;

    public WebDefaultCellEditor ()
    {
        super ();
    }

    public WebDefaultCellEditor ( final JTextField textField )
    {
        super ();
        initialize ( ( C ) textField );
    }

    public WebDefaultCellEditor ( final JCheckBox checkBox )
    {
        super ();
        initialize ( ( C ) checkBox );
    }

    public WebDefaultCellEditor ( final JComboBox comboBox )
    {
        super ();
        initialize ( ( C ) comboBox );
    }

    /**
     * Initializes cell editor and its delegate.
     *
     * @param editor actual editor component
     */
    protected void initialize ( final C editor )
    {
        editorComponent = editor;
        if ( editor instanceof JTextField )
        {
            final JTextField textField = ( JTextField ) editor;
            delegate = new EditorDelegate ()
            {
                @Override
                public void setValue ( final Object value )
                {
                    textField.setText ( ( value != null ) ? value.toString () : "" );
                }

                @Override
                public Object getCellEditorValue ()
                {
                    return textField.getText ();
                }
            };
            textField.addActionListener ( delegate );
        }
        else if ( editor instanceof JCheckBox )
        {
            final JCheckBox checkBox = ( JCheckBox ) editor;
            delegate = new EditorDelegate ()
            {
                @Override
                public void setValue ( final Object value )
                {
                    boolean selected = false;
                    if ( value instanceof Boolean )
                    {
                        selected = ( Boolean ) value;
                    }
                    else if ( value instanceof String )
                    {
                        selected = value.equals ( "true" );
                    }
                    checkBox.setSelected ( selected );
                }

                @Override
                public Object getCellEditorValue ()
                {
                    return checkBox.isSelected ();
                }
            };
            checkBox.addActionListener ( delegate );
            checkBox.setRequestFocusEnabled ( false );
        }
        else if ( editor instanceof JComboBox )
        {
            final JComboBox comboBox = ( JComboBox ) editor;
            StyleId.comboboxUndecorated.set ( comboBox );
            comboBox.putClientProperty ( COMBOBOX_CELL_EDITOR, Boolean.TRUE );
            delegate = new EditorDelegate ()
            {
                @Override
                public void setValue ( final Object value )
                {
                    comboBox.setSelectedItem ( value );
                }

                @Override
                public Object getCellEditorValue ()
                {
                    return comboBox.getSelectedItem ();
                }

                @Override
                public boolean shouldSelectCell ( final EventObject anEvent )
                {
                    if ( anEvent instanceof MouseEvent )
                    {
                        final MouseEvent e = ( MouseEvent ) anEvent;
                        return e.getID () != MouseEvent.MOUSE_DRAGGED;
                    }
                    return true;
                }

                @Override
                public boolean stopCellEditing ()
                {
                    if ( comboBox.isEditable () )
                    {
                        comboBox.actionPerformed ( new ActionEvent ( WebDefaultCellEditor.this, 0, "" ) );
                    }
                    return super.stopCellEditing ();
                }
            };
            comboBox.addActionListener ( delegate );
        }
    }

    public Component getComponent ()
    {
        return editorComponent;
    }

    public void setClickCountToStart ( final int count )
    {
        clickCountToStart = count;
    }

    public int getClickCountToStart ()
    {
        return clickCountToStart;
    }

    @Override
    public Object getCellEditorValue ()
    {
        return delegate.getCellEditorValue ();
    }

    @Override
    public boolean isCellEditable ( final EventObject anEvent )
    {
        return delegate.isCellEditable ( anEvent );
    }

    @Override
    public boolean shouldSelectCell ( final EventObject anEvent )
    {
        return delegate.shouldSelectCell ( anEvent );
    }

    @Override
    public boolean stopCellEditing ()
    {
        return delegate.stopCellEditing ();
    }

    @Override
    public void cancelCellEditing ()
    {
        delegate.cancelCellEditing ();
    }

    @Override
    public Component getTreeCellEditorComponent ( final JTree tree, final Object value, final boolean isSelected, final boolean expanded,
                                                  final boolean leaf, final int row )
    {
        //String stringValue = tree.convertValueToText ( value, isSelected, expanded, leaf, row, false );
        delegate.setValue ( value );
        return editorComponent;
    }

    @Override
    public Component getTableCellEditorComponent ( final JTable table, final Object value, final boolean isSelected, final int row,
                                                   final int column )
    {
        delegate.setValue ( value );
        return editorComponent;
    }

    protected class EditorDelegate implements ActionListener, ItemListener, Serializable
    {
        protected T value;

        public T getCellEditorValue ()
        {
            return value;
        }

        public void setValue ( final T value )
        {
            this.value = value;
        }

        public boolean isCellEditable ( final EventObject anEvent )
        {
            if ( anEvent == null || anEvent instanceof ActionEvent )
            {
                return true;
            }
            else if ( anEvent instanceof KeyEvent )
            {
                final KeyEvent keyEvent = ( KeyEvent ) anEvent;
                return Hotkey.F2.isTriggered ( keyEvent );
            }
            else if ( anEvent instanceof MouseEvent )
            {
                final MouseEvent mouseEvent = ( MouseEvent ) anEvent;
                return clickCountToStart >= 0 && mouseEvent.getClickCount () >= clickCountToStart;
            }
            else
            {
                return false;
            }
        }

        public boolean shouldSelectCell ( final EventObject anEvent )
        {
            return true;
        }

        public boolean stopCellEditing ()
        {
            fireEditingStopped ();
            return true;
        }

        public void cancelCellEditing ()
        {
            fireEditingCanceled ();
        }

        @Override
        public void actionPerformed ( final ActionEvent e )
        {
            WebDefaultCellEditor.this.stopCellEditing ();
        }

        @Override
        public void itemStateChanged ( final ItemEvent e )
        {
            WebDefaultCellEditor.this.stopCellEditing ();
        }
    }
}