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

package ca.cosc3p91.a2.gameobjects;

public class LumberMine extends ResourceBuilding {

    public static String resource = "wood";

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

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

    @Override
    public String getResource() {
        return resource;
    }

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