/*

 * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code 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

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/* @test

   @bug 6711682

   @summary  JCheckBox in JTable: checkbox doesn't alaways respond to the first mouse click

   @author Alexander Potochkin

   @run main bug6711682

*/

import sun.awt.SunToolkit;

import javax.swing.*;

import javax.swing.event.CellEditorListener;

import javax.swing.table.TableCellEditor;

import javax.swing.table.TableCellRenderer;

import java.awt.*;

import java.awt.event.InputEvent;

import java.awt.event.KeyEvent;

import java.util.EventObject;

public class bug6711682 {

    private static JCheckBox editorCb;

    private static JCheckBox rendererCb;

    private static JTable table;

    public static void main(String[] args) throws Exception {

        Robot robot = new Robot();

        robot.setAutoDelay(50);

        SunToolkit toolkit = (SunToolkit) Toolkit.getDefaultToolkit();

        SwingUtilities.invokeAndWait(new Runnable() {

            public void run() {

                createAndShowGUI();

            }

        });

        toolkit.realSync();

        Point l = table.getLocationOnScreen();

        int h = table.getRowHeight();

        for (int i = 0; i < 3; i++) {

            robot.mouseMove(l.x + 5, l.y + 5 + i * h);

            robot.mousePress(InputEvent.BUTTON1_MASK);

            robot.mouseRelease(InputEvent.BUTTON1_MASK);

        }

        // Without pressing F2 the last table's cell

        // reported <code>false</code> value

        // note that I can't press it inside the for loop

        // because it doesn't reproduce the bug

        robot.keyPress(KeyEvent.VK_F2);

        robot.keyRelease(KeyEvent.VK_F2);

        for (int i = 0; i < 3; i++) {

            if (!Boolean.TRUE.equals(table.getValueAt(i, 0))) {

                throw new RuntimeException("Row #" + i + " checkbox is not selected");

            }

        }

        for (int i = 2; i >= 0; i--) {

            robot.mouseMove(l.x + 5, l.y + 5 + i * h);

            robot.mousePress(InputEvent.BUTTON1_MASK);

            robot.mouseRelease(InputEvent.BUTTON1_MASK);

        }

        robot.keyPress(KeyEvent.VK_F2);

        robot.keyRelease(KeyEvent.VK_F2);

        for (int i = 0; i < 3; i++) {

            if (Boolean.TRUE.equals(table.getValueAt(i, 0))) {

                throw new RuntimeException("Row #" + i + " checkbox is selected");

            }

        }

    }

    private static void createAndShowGUI() {

        editorCb = new JCheckBox();

        rendererCb = new JCheckBox();

        JFrame f = new JFrame("Table with CheckBox");

        Container p = f.getContentPane();

        p.setLayout(new BorderLayout());

        table = new JTable(new Object[][]{{false}, {false}, {false}}, new Object[]{"CheckBox"});

        TableCellEditor editor = new TableCellEditor() {

            int editedRow;

            public Component getTableCellEditorComponent(JTable table,

                                                         Object value, boolean isSelected, int row, int column) {

                this.editedRow = row;

                editorCb.setSelected(Boolean.TRUE.equals(value));

                editorCb.setBackground(UIManager.getColor("Table.selectionBackground"));

                return editorCb;

            }

            public void addCellEditorListener(CellEditorListener l) {

            }

            public void cancelCellEditing() {

            }

            public Object getCellEditorValue() {

                return editorCb.isSelected();

            }

            public boolean isCellEditable(EventObject anEvent) {

                return true;

            }

            public void removeCellEditorListener(CellEditorListener l) {

            }

            public boolean shouldSelectCell(EventObject anEvent) {

                return true;

            }

            public boolean stopCellEditing() {

                table.getModel().setValueAt(editorCb.isSelected(), editedRow, 0);

                return true;

            }

        };

        table.getColumnModel().getColumn(0).setCellEditor(editor);

        TableCellRenderer renderer = new TableCellRenderer() {

            public Component getTableCellRendererComponent(JTable table,

                                                           Object value, boolean isSelected, boolean hasFocus,

                                                           int row, int column) {

                rendererCb.setSelected(Boolean.TRUE.equals(value));

                return rendererCb;

            }

        };

        table.getColumnModel().getColumn(0).setCellRenderer(renderer);

        p.add(table, BorderLayout.CENTER);

        f.pack();

        f.setVisible(true);

    }

}