ContractTableModel.java

/*
 * This file is part of the aidGer project.
 *
 * Copyright (C) 2010-2013 The aidGer Team
 *
 * 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 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 General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package de.aidger.view.models;

import static de.aidger.utils.Translation._;

import java.util.Calendar;
import java.util.Date;

import de.aidger.model.AbstractModel;
import de.aidger.model.models.Assistant;
import de.aidger.model.models.Contract;
import de.aidger.view.forms.ContractEditorForm.ContractType;
import siena.SienaException;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * The class represents the table model for the contracts data.
 * 
 * @author aidGer Team
 */
@SuppressWarnings("serial")
public class ContractTableModel extends TableModel {

    private static Map<Long, UIAssistant> assistantCache = new HashMap<Long, UIAssistant>();

    /**
     * Constructs the table model for contracts.
     */
    public ContractTableModel() {
        super(new String[] { _("ID"), _("Assistant"), _("Completion date"),
                _("Confirmation date"), _("Start date"), _("End date"),
                _("Type"), _("Delegation") });
    }

    /*
     * (non-Javadoc)
     * 
     * @see javax.swing.table.AbstractTableModel#getColumnClass(int)
     */
    @Override
    public Class<?> getColumnClass(int column) {
        // sort specific columns properly
        if (column == 0) {
            return Integer.class;
        } else if (column == 2 || column == 3 || column == 4 || column == 5) {
            return Date.class;
        }

        return super.getColumnClass(column);
    }

    /**
     * (non-Javadoc)
     *
     * @see de.aidger.view.models.TableModel#getRowValue(de.aidger.model.AbstractModel, int)
     */
    @Override
    protected Object getRowValue(AbstractModel model, int row) {
        try {
            Contract contract = (Contract) model;
            switch (row) {
                case 0: return contract.getId();
                case 1:
                    if (assistantCache.containsKey(contract.getAssistantId())) {
                        return assistantCache.get(contract.getAssistantId());
                    } else {
                        UIAssistant assistant = new UIAssistant((new Assistant()).getById(contract.getAssistantId()));
                        assistantCache.put(assistant.getId(), assistant);
                        return assistant;
                    }
                case 2: return contract.getCompletionDate();
                case 3:
                    Date confirmationDate = contract.getConfirmationDate();
                    Calendar cal = Calendar.getInstance();
                    cal.clear();
                    return confirmationDate == null ? cal.getTime() : confirmationDate;
                case 4: return contract.getStartDate();
                case 5: return contract.getEndDate();
                case 6: return ContractType.valueOf(contract.getType());
                case 7: return contract.isDelegation();
            }
        } catch (SienaException ex) {
        }
        return null;
    }

    /**
     * (non-javadoc)
     *
     * @see de.aidger.view.models.TableModel#getModels()
     */
    protected List<AbstractModel> getModels() {
        List<AbstractModel> ret = new ArrayList<AbstractModel>();
        try {
            List<Contract> lst = (new Contract()).getAll();
            for (Contract e : lst) {
                ret.add(new Contract(e));
            }
        } catch (SienaException ex) {
        }

        return ret;
    }

}