package fr.ifremer.quadrige3.ui.swing.table;
   
   /*-
    * #%L
    * Quadrige3 Core :: UI Swing Common
    * %%
    * Copyright (C) 2017 - 2019 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 fr.ifremer.quadrige3.core.dao.technical.Assert;
  import org.jdesktop.swingx.table.TableColumnExt;
  
  import javax.swing.event.*;
  import java.awt.event.MouseEvent;
  import java.awt.event.MouseListener;
  import java.beans.PropertyChangeEvent;
  import java.beans.PropertyChangeListener;
  import java.util.Arrays;
  import java.util.HashMap;
  import java.util.Map;
  
  /*
   *	Use a JTable as a renderer for row numbers of a given mainTable table.
   *  This table must be added to the row header of the scrollpane that
   *  contains the main table.
   */
  public class FixedSwingTable extends SwingTable
      implements PropertyChangeListener, TableModelListener, RowSorterListener, MouseListener {
  
      private final SwingTable mainTable;
  
      private Map<Integer, Integer> modelIndexTransposeMap = new HashMap<>();
  
      public FixedSwingTable(SwingTable table) {
          Assert.notNull(table, "A FixedSwingTable must have a main table");
          mainTable = table;
  
          // Set component name
          setName(mainTable.getName() + "_fixed");
  
          // Affect the main model
          setModel(mainTable.getTableModel());
  
          // init table
          setAutoCreateColumnsFromModel(false);
          setSelectionModel(mainTable.getSelectionModel());
          setRowSorter(mainTable.getRowSorter());
          setRowHeight(mainTable.getRowHeight());
          getTableHeader().setReorderingAllowed(false);
          setAutoResizeMode(SwingTable.AUTO_RESIZE_OFF);
  
          // add listeners
          addMouseListener(this);
          mainTable.addPropertyChangeListener(this);
          mainTable.getModel().addTableModelListener(this);
      }
  
      public void addColumnModelTransposition(int fixedColumnModelIndex, int mainColumnModelIndex) {
          modelIndexTransposeMap.put(fixedColumnModelIndex, mainColumnModelIndex);
      }
  
      public void removeColumnModelTransposition(int fixedColumnModelIndex) {
          modelIndexTransposeMap.remove(fixedColumnModelIndex);
      }
  
      public Integer getMainColumnModelIndex(int fixedColumnModelIndex) {
          return modelIndexTransposeMap.getOrDefault(fixedColumnModelIndex, fixedColumnModelIndex);
      }
  
      public Integer getFixedColumnModelIndex(int mainColumnModelIndex) {
          return modelIndexTransposeMap.entrySet().stream().filter(entry -> entry.getValue().equals(mainColumnModelIndex)).findFirst().map(Map.Entry::getKey).orElse(mainColumnModelIndex);
      }
  
      public int getColumnModelIndex(ColumnIdentifier<?> identifier) {
          TableColumnExt column = getColumnExt(identifier);
          Assert.notNull(column);
          return getMainColumnModelIndex(column.getModelIndex());
      }
  
      @Override
      public int convertColumnIndexToModel(int viewColumnIndex) {
          return getMainColumnModelIndex(super.convertColumnIndexToModel(viewColumnIndex));
      }
  
      @Override
     public int convertColumnIndexToView(int modelColumnIndex) {
         return getFixedColumnModelIndex(super.convertColumnIndexToView(modelColumnIndex));
     }
 
     /*
      *  Keep the row table in sync with the mainTable table
      */
     @Override
     public void propertyChange(PropertyChangeEvent e) {
         if ("selectionModel".equals(e.getPropertyName())) {
             setSelectionModel(mainTable.getSelectionModel());
         } else if ("rowSorter".equals(e.getPropertyName())) {
             setRowSorter(mainTable.getRowSorter());
         } else if ("rowHeight".equals(e.getPropertyName())) {
             repaint();
         } else if (SwingTable.PROPERTY_ROW_FILTER.equals(e.getPropertyName())
             || SwingTable.PROPERTY_SORTED.equals(e.getPropertyName())) {
             revalidate();
         }
     }
 
     /*
      *  Implement the TableModelListener
      */
     @Override
     public void tableChanged(TableModelEvent e) {
         revalidate();
     }
 
     @Override
     public void columnAdded(TableColumnModelEvent e) {
         super.columnAdded(e);
         updateViewportSize();
     }
 
     @Override
     public void columnRemoved(TableColumnModelEvent e) {
         super.columnRemoved(e);
         updateViewportSize();
     }
 
     @Override
     public void columnMarginChanged(ChangeEvent e) {
         super.columnMarginChanged(e);
         updateViewportSize();
     }
 
     private void updateViewportSize() {
         setPreferredScrollableViewportSize(getPreferredSize());
     }
 
     @Override
     public void mouseClicked(MouseEvent e) {
         // propagate mouseClicked event to main table (ie. allows popup on right click)
         if (mainTable.getMouseListeners() != null)
             Arrays.stream(mainTable.getMouseListeners()).forEach(mouseListener -> mouseListener.mouseClicked(e));
     }
 
     @Override
     public void mousePressed(MouseEvent e) {
 
     }
 
     @Override
     public void mouseReleased(MouseEvent e) {
 
     }
 
     @Override
     public void mouseEntered(MouseEvent e) {
 
     }
 
     @Override
     public void mouseExited(MouseEvent e) {
 
     }
 }