/**
 * Copyright 2012 Rafal Lewczuk <rafal.lewczuk@jitlogic.com>
 * <p/>
 * This 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.
 * <p/>
 * This software 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.
 * <p/>
 * You should have received a copy of the GNU General Public License
 * along with this software. If not, see <http://www.gnu.org/licenses/>.
 */

package com.jitlogic.zorka.viewer;

import com.jitlogic.zorka.common.util.ZorkaUtil;

import javax.swing.*;
import javax.swing.table.AbstractTableModel;
import java.util.ArrayList;
import java.util.List;

public class TraceTableModel extends AbstractTableModel {

    private String[] colNames = { "Date", "Time", "Calls", "Err", "Recs", "Label" };
    private int[]    colWidth = { 90, 50, 60, 40, 40, 550 };

    private List<ViewerTraceRecord> traceRecords = new ArrayList<ViewerTraceRecord>();


    public void setDataSet(TraceDataSet dataSet, ViewerRecordFilter filter) {
        if (dataSet != null) {
            traceRecords = new ArrayList<ViewerTraceRecord>(dataSet.numRecords());

            for (ViewerTraceRecord rec : dataSet.getRecords()) {
                if (filter == null || filter.matches(rec)) {
                    traceRecords.add(rec);
                }
            }
        } else {
            traceRecords = new ArrayList<ViewerTraceRecord>();
        }

        fireTableDataChanged();
    }


    public void adjustColumns(JTable table) {
        for (int i = 0; i < colWidth.length; i++) {
            table.getColumnModel().getColumn(i).setPreferredWidth(colWidth[i]);
        }
        table.getColumnModel().getColumn(5).setCellRenderer(new TraceCellRenderer());
    }


    @Override
    public String getColumnName(int idx) {
        return colNames[idx];
    }


    @Override
    public int getRowCount() {
        return traceRecords.size();
    }


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


    @Override
    public Object getValueAt(int rowIndex, int columnIndex) {
        ViewerTraceRecord el = traceRecords.get(rowIndex);

        switch (columnIndex) {
            case 0:
                return el.prettyClock();
            case 1:
                return ZorkaUtil.strTime(el.getTime());
            case 2:
                return el.getCalls();
            case 3:
                return el.getErrors();
            case 4:
                return el.getRecs();
            case 5:
                return el;
        }
        return "?";
    }



    public ViewerTraceRecord get(int i) {
        return traceRecords.get(i);
    }
}