Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 gvSIG Association

 *

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

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * 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 General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.swing.legend.impl.colortable.selector;

import java.io.File;

import java.util.List;

import javax.swing.table.AbstractTableModel;

import org.apache.commons.lang3.tuple.Pair;

import org.gvsig.raster.lib.legend.api.colortable.ColorTable;

import org.gvsig.raster.swing.legend.colortable.selector.ColorTablesTableModel;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

/**

 * @author fdiaz

 *

 */

public class DefaultColorTablesTableModel extends AbstractTableModel implements ColorTablesTableModel{

    /**

     *

     *

     */

    private static final long serialVersionUID = -6739781555636046945L;

    private List<Pair<File,ColorTable>> colorTables;

    private static final int COLUMNS = 4;

    public static final int COLUMN_PREVIEW = 0;

    public static final int COLUMN_NAME = 1;

    public static final int COLUMN_CLASSES = 2;

    public static final int COLUMN_INTERPOLATED = 3;

    /**

     * @param bandsInfo

     */

    public DefaultColorTablesTableModel(List<Pair<File,ColorTable>> colorTables) {

        this.colorTables = colorTables;

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getRowCount()

     */

    @Override

    public int getRowCount() {

        return colorTables.size();

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getColumnCount()

     */

    @Override

    public int getColumnCount() {

        return COLUMNS;

    }

    /* (non-Javadoc)

     * @see javax.swing.table.TableModel#getValueAt(int, int)

     */

    @Override

    public Object getValueAt(int rowIndex, int columnIndex) {

        switch (columnIndex) {

        case COLUMN_PREVIEW:

            return colorTables.get(rowIndex).getValue();

        case COLUMN_NAME:

            return colorTables.get(rowIndex).getValue().getName();

        case COLUMN_CLASSES:

            return colorTables.get(rowIndex).getValue().getClasses().size();

        case COLUMN_INTERPOLATED:

            return colorTables.get(rowIndex).getValue().isInterpolated();

        }

        return null;

    }

    @Override

    public String getColumnName(int column) {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        switch (column) {

        case COLUMN_PREVIEW:

            return i18nManager.getTranslation("_preview");

        case COLUMN_NAME:

            return i18nManager.getTranslation("_name");

        case COLUMN_CLASSES:

            return i18nManager.getTranslation("_classes");

        case COLUMN_INTERPOLATED:

            return i18nManager.getTranslation("_interpolated");

        }

        return null;

    }

    public boolean isCellEditable(int row, int col) {

        return false;

    }

    @Override

    public Class<?> getColumnClass(int col) {

        if (col == COLUMN_PREVIEW) {

            return ColorTable.class;

        } else if (col == COLUMN_INTERPOLATED) {

            return Boolean.class;

        }

        return super.getColumnClass(col);

    }

    @Override

    public void addColorTable(Pair<File,ColorTable> colorTable) {

        colorTables.add(colorTable);

        this.fireTableDataChanged();

    }

    @Override

    public void removeColorTable(int index) {

        colorTables.remove(index);

        this.fireTableDataChanged();

    }

    @Override

    public Pair<File, ColorTable> getColorTable(int index) {

        return colorTables.get(index);

    }

    @Override

    public void setColorTable(int index, ColorTable colorTableEdited) {

        colorTables.get(index).getValue().copyFrom(colorTableEdited);

        this.fireTableDataChanged();

    }

    @Override

    public List<Pair<File, ColorTable>> getColorTables() {

        return colorTables;

    }

}