package fr.ifremer.quadrige2.ui.swing.common.table;
   
   /*-
    * #%L
    * Quadrige2 Core :: Quadrige2 UI Common
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   * 
   * This program 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 Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  
  import com.google.common.base.Preconditions;
  import org.jdesktop.swingx.JXTable;
  import org.jdesktop.swingx.table.TableColumnExt;
  
  import javax.swing.*;
  import javax.swing.table.TableCellRenderer;
  import javax.swing.table.TableColumnModel;
  import java.awt.Component;
  import java.awt.Container;
  import java.awt.Graphics;
  import java.awt.event.MouseAdapter;
  import java.awt.event.MouseEvent;
  import java.util.List;
  
  /**
   * <p>CheckTableColumn class.</p>
   *
   * @param <R> type of row UI model
   * @param <T> type of table UI model
   * @author Ludovic Pecquot <ludovic.pecquot@e-is.pro>
   */
  public class CheckTableColumn<R extends AbstractRowUIModel, T extends AbstractTableUIModel<?, R, T>> extends TableColumnExt {
  
      /** Constant <code>FIXED_VIEW_POSITION=0</code> */
      public static final int FIXED_VIEW_POSITION = 0;
  
      private final JXTable table;
      private final T tableModel;
      private boolean allSelected;
      private boolean allEnabled;
  
      /**
       * <p>Getter for the field <code>table</code>.</p>
       *
       * @return a {@link JXTable} object.
       */
      public JXTable getTable() {
          return table;
      }
  
      /**
       * <p>Getter for the field <code>tableModel</code>.</p>
       *
       * @return a T object.
       */
      public final T getTableModel() {
          return tableModel;
      }
  
      /**
       * <p>isAllSelected.</p>
       *
       * @return a boolean.
       */
      public boolean isAllSelected() {
          return allSelected;
      }
  
      /**
       * <p>Setter for the field <code>allSelected</code>.</p>
       *
       * @param allSelected a boolean.
       */
      public void setAllSelected(boolean allSelected) {
          this.allSelected = allSelected;
      }
  
      /**
       * <p>isAllEnabled.</p>
       *
       * @return a boolean.
       */
      public boolean isAllEnabled() {
         return allEnabled;
     }
 
     /**
      * <p>Setter for the field <code>allEnabled</code>.</p>
      *
      * @param allEnabled a boolean.
      */
     public void setAllEnabled(boolean allEnabled) {
         this.allEnabled = allEnabled;
     }
 
     /**
      * <p>Constructor for CheckTableColumn.</p>
      *
      * @param table a {@link JXTable} object.
      * @param tableModel a T object.
      */
     public CheckTableColumn(JXTable table, T tableModel) {
         super(table.getColumnCount());
         Preconditions.checkNotNull(table);
         this.table = table;
         this.tableModel = tableModel;
         editable = true;
         hideable = false;
         sortable = false;
         allEnabled = true;
         allSelected = false;
 
         setWidth(41);
         setMinWidth(width);
         setMaxWidth(width);
 
         setCellEditor(table.getDefaultEditor(Boolean.class));
         setCellRenderer(table.getDefaultRenderer(Boolean.class));
 
         // header 
         setHeaderRenderer(new CheckAllHeaderRenderer(this));
 
         // identifier
         setIdentifier(new CheckTableColumnIdentifier<>());
     }
 
     /** {@inheritDoc} */
     @Override
     @SuppressWarnings("unchecked")
     public ColumnIdentifier<R> getIdentifier() {
         return (ColumnIdentifier<R>) super.getIdentifier();
     }
 
     private class CheckAllHeaderRenderer extends JCheckBox implements TableCellRenderer {
 
         private final CheckTableColumn checkColumn;
 
         @SuppressWarnings("unchecked")
         private CheckAllHeaderRenderer(CheckTableColumn targetColumn) {
             super((String) null);
             this.checkColumn = targetColumn;
 
             setOpaque(false);
             setFont(checkColumn.getTable().getTableHeader().getFont());
 
             // add mouse listener if the table is editable
             checkColumn.getTable().getTableHeader().addMouseListener(new MouseAdapter() {
 
                 @Override
                 public void mouseClicked(MouseEvent e) {
                     JXTable table = checkColumn.getTable();
 
                     if (!table.isEnabled() || !table.isEditable()) {
                         return;
                     }
 
                     TableColumnModel columnModel = table.getColumnModel();
                     int vci = columnModel.getColumnIndexAtX(e.getX());
                     int mci = table.convertColumnIndexToModel(vci);
                     if (mci == checkColumn.getModelIndex()) {
                         boolean allChecked = !isSelected();
                         checkColumn.setAllSelected(allChecked);
                         checkColumn.setAllEnabled(true);
                         // propagate is selected to all lines
                         if (checkColumn.getTableModel() != null && checkColumn.getTableModel().getRowCount() > 0) {
                             for (R row : (List<R>) checkColumn.getTableModel().getRows()) {
                                 if (!row.isCalculated() /*&& row.isEditable()*/) {
                                     row.selected = allChecked;
                                 }
                             }
 
                             // recalculate selected rows
                             checkColumn.getTableModel().populateSelectedRows();
                             // redraw immediately
                             table.getTableHeader().repaint();
                             table.repaint();
                         }
 
                     }
                 }
 
             });
         }
 
         @Override
         public Component getTableCellRendererComponent(JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
             TableCellRenderer r = table.getTableHeader().getDefaultRenderer();
             JLabel l = (JLabel) r.getTableCellRendererComponent(table, value, isSelected, hasFocus, row, column);
             if (checkColumn.getModelIndex() == table.convertColumnIndexToModel(column)) {
                 // reset allSelected if table is empty
                 if (!table.isEnabled() || !checkColumn.getTable().isEditable() || table.getRowCount() == 0) checkColumn.setAllSelected(false);
                 setSelected(checkColumn.isAllSelected());
                 setEnabled(checkColumn.isAllEnabled());
                 l.setIcon(new ComponentIcon(this));
                 l.setHorizontalAlignment(SwingConstants.CENTER);
                 l.setVerticalAlignment(SwingConstants.TOP);
             }
             l.setText(null);
             return l;
         }
 
     }
 
     class ComponentIcon implements Icon {
 
         private final JComponent cmp;
 
         public ComponentIcon(JComponent cmp) {
             this.cmp = cmp;
         }
 
         @Override
         public int getIconWidth() {
             return cmp.getPreferredSize().width;
         }
 
         @Override
         public int getIconHeight() {
             return cmp.getPreferredSize().height;
         }
 
         @Override
         public void paintIcon(Component c, Graphics g, int x, int y) {
             SwingUtilities.paintComponent(g, cmp, (Container) c, x, y, getIconWidth(), getIconHeight());
         }
     }
 
 }