package fr.ifremer.quadrige3.ui.swing.table.renderer;
   
   /*-
    * #%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.enumeration.EnumValue;
  import fr.ifremer.quadrige3.core.dao.technical.enumeration.Enums;
  
  import javax.swing.JTable;
  import javax.swing.table.DefaultTableCellRenderer;
  import java.awt.Component;
  
  /**
   * @author peck7 on 14/06/2019.
   */
  public class EnumValueCellRenderer extends DefaultTableCellRenderer {
  
      private final Class enumClass;
  
      public <E extends Enum<E>> EnumValueCellRenderer(Class<E> enumClass) {
          this.enumClass = enumClass;
      }
  
      @Override
      @SuppressWarnings("unchecked")
      public Component getTableCellRendererComponent(JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
  
          if (value instanceof EnumValue) {
              value = ((EnumValue) value).getLabel();
          } else if (value instanceof String) {
              EnumValue enumValue = Enums.fromCode(enumClass, (String) value);
              if (enumValue != null) value = enumValue.getLabel();
          }
  
          return super.getTableCellRendererComponent(table, value, isSelected, hasFocus, row, column);
      }
  }