COSC-3P91-Assignments/Assignment 2/src/ca/cosc3p91/a2/gameobjects/IronMine.java

package ca.cosc3p91.a2.gameobjects;

public class IronMine extends ResourceBuilding {

    public static String resource = "iron";

    public IronMine(ResourceStage baseStage) {
        upgrade(baseStage);
    }

    @Override
    protected ResourceHarvestHandler getHarvestHandler() {
        return hall -> hall.addIron(getHarvestRate());
    }

    @Override
    public Stage getUpgradeStage() {
        return ResourceStages.ironStages[getLevel()+1];
    }
}