def getFood(agent, gameState):
    """
    Returns the food you're meant to eat. This is in the form of a matrix
    where m[x][y]=true if there is food you can eat (based on your team) in that square.
    """
    if agent.red:
        return gameState.getBlueFood()
    else:
        return gameState.getRedFood()