rails-commits

[Stefan F. <ste...@us...>]

Update of /cvsroot/rails/18xx/rails/game/specific/_1851
In directory sfp-cvsdas-4.v30.ch3.sourceforge.com:/tmp/cvs-serv16169/rails/game/specific/_1851

Added Files:
	OffBoardRevenueModifier.java 
Log Message:
1851 revenue calculation support and removed configuration interface from offboard modifiers

--- NEW FILE: OffBoardRevenueModifier.java ---
package rails.game.specific._1851;

import java.util.HashSet;
import java.util.Set;

import rails.algorithms.NetworkVertex;
import rails.algorithms.RevenueAdapter;
import rails.algorithms.RevenueBonus;
import rails.algorithms.RevenueStaticModifier;
import rails.game.Station;

public class OffBoardRevenueModifier implements RevenueStaticModifier {

    private static final int BONUS_VALUE = 10;
    
    public void modifyCalculator(RevenueAdapter revenueAdapter) {
        // 1. get all off-board type stations and all other stations
        Set<NetworkVertex> offBoard = new HashSet<NetworkVertex>();
        Set<NetworkVertex> otherStations = new HashSet<NetworkVertex>();
        for (NetworkVertex vertex:revenueAdapter.getVertices()) {
            if (vertex.isStation()) {
                if (vertex.getStation().getType().equals(Station.OFF_MAP_AREA)){
                    offBoard.add(vertex);
                } else {
                    otherStations.add(vertex);
                }
            }
        }
        // 2. combine those to revenueBonuses
        // always two offboard areas and one other
        Set<NetworkVertex> destOffBoard = new HashSet<NetworkVertex>(offBoard);
        for (NetworkVertex offA:offBoard) {
            destOffBoard.remove(offA);
            for (NetworkVertex offB:destOffBoard) {
                for (NetworkVertex station:otherStations) {
                    RevenueBonus bonus = new RevenueBonus(BONUS_VALUE, "OB/" + station.toString());
                    bonus.addVertex(offA); bonus.addVertex(offB); bonus.addVertex(station);
                    revenueAdapter.addRevenueBonus(bonus);
                }
            }
        }
    }
}