package fr.ifremer.quadrige3.ui.swing.table;
   
   /*-
    * #%L
    * Quadrige3 Core :: Quadrige3 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 org.jdesktop.swingx.table.DefaultTableColumnModelExt;
  import org.jdesktop.swingx.table.TableColumnExt;
  
  import javax.swing.table.TableColumn;
  import java.util.HashSet;
  import java.util.Objects;
  import java.util.Set;
  import java.util.function.Predicate;
  
  /**
   * The default table column model used for application
   * <p/>
   * The CheckTableColumn is not draggable
   * <p/>
   * Created by Ludovic on 13/05/2015.
   */
  public class SwingTableColumnModel extends DefaultTableColumnModelExt {
  
      /*
          Map holding model and view indexes of fixed column
          key id model index, value is view index
       */
      private Set<TableColumn> fixedColumns = new HashSet<>();
  
      @Override
      public TableColumnExt getColumnExt(Object identifier) {
  
          TableColumnExt column = super.getColumnExt(identifier);
  
          if (column == null && identifier instanceof ColumnIdentifier) {
              ColumnIdentifier<?> columnIdentifier = (ColumnIdentifier<?>) identifier;
  
              // try to
              column = (TableColumnExt) getColumns(true).stream()
                  .filter(tableColumn -> tableColumn.getIdentifier() instanceof ColumnIdentifier)
                  .filter(tableColumn -> Objects.equals(((ColumnIdentifier<?>) tableColumn.getIdentifier()).getPropertyName(), columnIdentifier.getPropertyName()))
                  .filter(tableColumn -> {
                      if (identifier instanceof PmfmColumnIdentifier && tableColumn.getIdentifier() instanceof PmfmColumnIdentifier) {
                          return ((PmfmColumnIdentifier) identifier).getPmfmId() == ((PmfmColumnIdentifier) tableColumn.getIdentifier()).getPmfmId();
                      }
                      return false;
                  })
                  .findFirst()
                  .orElse(null);
          }
  
          return column;
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      public void moveColumn(int columnIndex, int newIndex) {
  
          // prevent a fixed column to be moved or a column to be moved before fixed columns
          if (fixedColumns.contains(getColumn(columnIndex)) || fixedColumns.contains(getColumn(newIndex))) {
              return;
          }
  
          super.moveColumn(columnIndex, newIndex);
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      public void addColumn(TableColumn column) {
          super.addColumn(column);
  
          if (column instanceof FixedColumn) {
              Integer fixedPosition = ((FixedColumn) column).getFixedPosition();
              if (fixedPosition != null) {
                  setFixedColumn(column, fixedPosition);
             }
         }
     }
 
     @Override
     public void removeColumn(TableColumn column) {
         super.removeColumn(column);
 
         fixedColumns.remove(column);
     }
 
     private void setFixedColumn(TableColumn column, int fixedViewIndex) {
 
         // adjust fixedViewIndex if > to fixed column count
         fixedViewIndex = Math.min(fixedViewIndex, fixedColumns.size());
 
         fixedColumns.add(column);
 
         // move column to target view index
         super.moveColumn(getColumnIndex(column.getIdentifier()), fixedViewIndex);
     }
 
     public boolean isFixedColumn(TableColumn column) {
         return column != null && fixedColumns.contains(column);
     }
 
     public boolean hasFixedColumn(Predicate<TableColumn> predicate) {
         return fixedColumns.stream().anyMatch(predicate);
     }
 }