WorkProductionPanel.java

/**
 *  Copyright (C) 2002-2012   The FreeCol Team
 *
 *  This file is part of FreeCol.
 *
 *  FreeCol 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.
 *
 *  FreeCol 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 FreeCol.  If not, see <http://www.gnu.org/licenses/>.
 */

package net.sf.freecol.client.gui.panel;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;

import javax.swing.BorderFactory;
import javax.swing.ImageIcon;
import javax.swing.JLabel;
import javax.swing.JSeparator;
import javax.swing.border.Border;

import net.miginfocom.swing.MigLayout;
import net.sf.freecol.client.FreeColClient;
import net.sf.freecol.client.gui.GUI;
import net.sf.freecol.client.gui.i18n.Messages;
import net.sf.freecol.common.model.Building;
import net.sf.freecol.common.model.Colony;
import net.sf.freecol.common.model.ColonyTile;
import net.sf.freecol.common.model.FeatureContainer;
import net.sf.freecol.common.model.FreeColObject;
import net.sf.freecol.common.model.GoodsType;
import net.sf.freecol.common.model.Modifier;
import net.sf.freecol.common.model.Scope;
import net.sf.freecol.common.model.TileType;
import net.sf.freecol.common.model.Turn;
import net.sf.freecol.common.model.Unit;
import net.sf.freecol.common.model.UnitType;
import net.sf.freecol.common.resources.ResourceManager;

public class WorkProductionPanel extends FreeColPanel {

    private final Turn turn = getGame().getTurn();

    private static final Border border = BorderFactory
        .createCompoundBorder(BorderFactory.createMatteBorder(1, 0, 0, 0, Color.BLACK),
                              BorderFactory.createEmptyBorder(2, 2, 2, 2));

    public WorkProductionPanel(FreeColClient freeColClient, GUI gui, Unit unit) {
        super(freeColClient, gui);

        setLayout(new MigLayout("wrap 3, insets 10 10 10 10", "[]30:push[right][]", ""));

        Colony colony = unit.getColony();
        UnitType unitType = unit.getType();

        List<Modifier> modifiers = new ArrayList<Modifier>();
        if (unit.getLocation() instanceof ColonyTile) {
            ColonyTile colonyTile = (ColonyTile) unit.getLocation();
            GoodsType goodsType = unit.getWorkType();
            List<Modifier> tileModifiers
                = colonyTile.getProductionModifiers(goodsType, unitType);
            if (FeatureContainer.applyModifiers(0f, turn, tileModifiers) > 0) {
                modifiers.addAll(tileModifiers);
            }

            add(localizedLabel(colonyTile.getLabel()),
                               "span, align center, wrap 30");
            TileType tileType = colonyTile.getWorkTile().getType();
            int width = getGUI().getImageLibrary()
                .getTerrainImageWidth(tileType);
            int height = getGUI().getImageLibrary()
                .getTerrainImageHeight(tileType);
            BufferedImage image = new BufferedImage(width, height,
                BufferedImage.TYPE_INT_ARGB);
            getGUI().getMapViewer()
                .displayColonyTile((Graphics2D) image.getGraphics(),
                                   colonyTile.getWorkTile(), colony);
            add(new JLabel(new ImageIcon(image)));

        } else if (unit.getLocation() instanceof Building) {
            Building building = (Building) unit.getLocation();
            GoodsType goodsType = building.getGoodsOutputType();
            modifiers.addAll(building.getProductionModifiers(goodsType,
                                                             unitType));

            add(localizedLabel(building.getNameKey()),
                               "span, align center, wrap 30");
            add(new JLabel(ResourceManager.getImageIcon(building.getType()
                        .getId() + ".image")));
        }
        add(new UnitLabel(getFreeColClient(), unit, getGUI(), false, false),
            "wrap");

        Collections.sort(modifiers);
        float result = 0.0f;
        for (Modifier modifier : modifiers) {
            result = addModifier(modifier, unitType, result);
        }

        if (result < 0.0f) {
            add(new JLabel(Messages.message("model.source.zeroThreshold.name")),
                "newline");
            add(new JLabel(getModifierFormat().format(-result)), "wrap 30");
            result = 0.0f;
        }
        Font bigFont = getFont().deriveFont(Font.BOLD, 16);
        JLabel finalLabel = new JLabel(Messages.message("model.source.finalResult.name"));
        finalLabel.setFont(bigFont);
        add(finalLabel, "newline");

        JLabel finalResult = new JLabel(getModifierFormat().format(result));
        finalResult.setFont(bigFont);
        finalResult.setBorder(border);
        add(finalResult, "wrap 30");

        add(okButton, "span, tag ok");

        setSize(getPreferredSize());
    }


    private float addModifier(Modifier modifier, UnitType unitType, float result) {
        FreeColObject source = modifier.getSource();
        String sourceName;
        if (source == null) {
            sourceName = "???";
        } else {
            sourceName = Messages.getName(source);
            if (unitType != null && modifier.hasScope()) {
                for (Scope scope : modifier.getScopes()) {
                    if (scope.appliesTo(unitType)) {
                        sourceName += " (" + Messages.message(unitType.getNameKey()) + ")";
                    }
                }
            }
        }
        float value = modifier.getValue(turn);
        if (value == 0) {
            return result;
        }
        String[] bonus = getModifierStrings(value, modifier.getType());
        add(new JLabel(sourceName), "newline");
        add(new JLabel(bonus[0] + bonus[1]));
        if (bonus[2] != null) {
            add(new JLabel(bonus[2]));
        }
        return modifier.applyTo(result, turn);
    }



}