ArtistInspectorTracksTableModel.java

/*
* The GPLv3 licence :
* -----------------
* Copyright (c) 2009 Ricardo Dias
*
* This file is part of MuVis.
*
* MuVis 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 3 of the License, or
* (at your option) any later version.
*
* MuVis 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 MuVis.  If not, see <http://www.gnu.org/licenses/>.
 */

package muvis.view;

import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Hashtable;
import javax.swing.table.AbstractTableModel;
import muvis.Environment;
import muvis.database.MusicLibraryDatabaseManager;
import muvis.database.TableRecord;
import muvis.util.Observable;
import muvis.util.Observer;

/**
 * Class that holds the Model for the list visualization of the library.
 * @author Ricardo
 */
public class ArtistInspectorTracksTableModel extends AbstractTableModel implements Observer {

    private String[] columnNames = new String[]{"Nr.", "Track name", "Artist",
        "Album", "Duration", "Genre"/*, "Year", "Beat", "Mood"*/};
    private MusicLibraryDatabaseManager dbManager;
    private Hashtable<Integer, TableRecord> records;

    public ArtistInspectorTracksTableModel(ArrayList<Integer> trackIds) {
        dbManager = Environment.getEnvironmentInstance().getDatabaseManager();
        records = new Hashtable<Integer, TableRecord>(trackIds.size());
        //loading all the tracks to this table
        for (Integer id : trackIds) {
            TableRecord rec = dbManager.getTrackRow(id);
            records.put(id, rec);
        }

    }

    @Override
    public int getColumnCount() {
        return columnNames.length;
    }

    @Override
    public int getRowCount() {
        int rowCount = records.size();
        if (rowCount < 0) {
            return 0;
        }
        return rowCount;
    }

    @Override
    public String getColumnName(int col) {
        return columnNames[col];
    }

    @Override
    public Object getValueAt(int row, int col) {
        int modRow = row, modCol = col;
        if (row == 0) {
            modRow++;
        } else {
            modRow = modRow + 1;
        }
        
        if (col == 0) {
            modCol++;
        } else {
            modCol = modCol + 1;
        }

        for (int i = 1; i <= records.size() ; i++){

            if (i == modRow){
                Enumeration<Integer> keys = records.keys();
                int j = 1;
                while(keys.hasMoreElements()){
                    if (j == i){
                        int id = keys.nextElement();
                        return records.get(id).getValueColumn(modCol);
                    }
                    j++;
                    keys.nextElement();
                }
            }
        }

        return new Object();
    }

    @Override
    public Class getColumnClass(int c) {

        if (c == 0) {
            return Integer.class;
        } else if (c == 1) {
            return String.class;
        } else if (c == 2) {
            return String.class;
        } else if (c == 3) {
            return String.class;
        } else if (c == 4) {
            return Long.class;
        } else if (c == 5) {
            return String.class;
        } else {
            return String.class;
        }
    }

    
    /**
     * This method is called when the database changes
     */
    @Override
    public void update(Observable obs, Object arg) {
        //nothing to do
    }
}