from aoe.types import *



class Gather():


    def __init__(self):
        self.gather_rates = {
                Tasks.BERRIES: {Resources.FOOD: 18.6/60},
                Tasks.SHEEP: {Resources.FOOD: 19.8/60},
                Tasks.BOAR: {Resources.FOOD: 24.6/60},
                Tasks.HUNT: {Resources.FOOD: 24.6/60},
                Tasks.FARM: {Resources.FOOD: 20/60},
                Tasks.FISH_VILLAGER: {Resources.FOOD: 25.8/60},
                Tasks.FISH_SHIP_SHORE: {Resources.FOOD: 16.8/60},
                Tasks.FISH_SHIP_DEEP: {Resources.FOOD: 29.4/60},
                Tasks.FISHTRAP: {Resources.FOOD: 21/60},
                Tasks.WOOD: {Resources.WOOD: 23.4/60},
                Tasks.GOLD: {Resources.GOLD: 22.8/60},
                Tasks.TRADECART: {Resources.GOLD: 25.9/60},
                Tasks.TRADECOG: {Resources.GOLD: 25.9/60},
                Tasks.RELIC: {Resources.GOLD: 30/60},
                Tasks.STONE: {Resources.STONE: 21.6/60},
                Tasks.BUILD: {},
                Tasks.IDLE: {},
               }

        self.required_gather_buildings = {
                Tasks.BERRIES: [set([Buildings.MILL]),
                                set([Buildings.TOWNCENTER])],
                Tasks.SHEEP: [set([Buildings.MILL]),
                              set([Buildings.TOWNCENTER])],
                Tasks.BOAR: [set([Buildings.MILL]),
                             set([Buildings.TOWNCENTER])],
                Tasks.HUNT: [set([Buildings.MILL]),
                             set([Buildings.TOWNCENTER])],
                Tasks.FARM: [set([Buildings.FARM, Buildings.MILL]),
                             set([Buildings.FARM, Buildings.TOWNCENTER])],
                Tasks.FISH_VILLAGER: [set([Buildings.DOCK]),
                                      set([Buildings.MILL]),
                                      set([Buildings.TOWNCENTER])],
                Tasks.FISH_SHIP_SHORE: [set([Buildings.DOCK])],
                Tasks.FISH_SHIP_DEEP: [set([Buildings.DOCK])],
                Tasks.FISHTRAP: [set([Buildings.DOCK, Buildings.FISHTRAP])],
                Tasks.WOOD: [set([Buildings.LUMBERCAMP]),
                             set([Buildings.TOWNCENTER])],
                Tasks.GOLD: [set([Buildings.MININGCAMP]),
                             set([Buildings.TOWNCENTER])],
                Tasks.TRADECART: [set([Buildings.MARKET])],
                Tasks.TRADECOG: [set([Buildings.DOCK])],
                #Tasks.RELIC: [set([Buildings.MONASTERY])],
                Tasks.STONE: [set([Buildings.MININGCAMP]),
                              set([Buildings.TOWNCENTER])],
                Tasks.BUILD: [],
                Tasks.IDLE: [],
               }

        self.improved_gather_buildings = {}