Traveling Politician (Metropolis Hastings) - the Clay Davis Edition

claydavis.py

from sys import exit
from scipy import stats
import math
import random


class Engine(object):

    def direction(self):
        if stats.bernoulli.rvs(0.5, size = 1)[0]:
            direction = 'east'
        else:
            direction = 'west'
        return direction

    def next_hop(self, current_point, island_range, populations):
        # continue using the island ranges to determine 'ends'
        # of distributions

        # just make sure to plug in the populations for the
        # proposals... (still need to update east/west)
        next_direction = self.direction()

        # scenarios
        # west & as far west as possible - stay
        # east & as far east as possible - stay
        # else
        # if east: proposed is the pop east
        # else west: proposed is the pop west
        if next_direction == 'west' and (current_point  == island_range[1]):
            current_point = current_point
        elif next_direction == 'east' and (current_point  == island_range[0]):
            current_point = current_point
        else:
            if next_direction == 'east':
                p_proposed = populations[current_point - 1]
            else:
                p_proposed = populations[current_point + 1]
            p_move = p_proposed/populations[current_point]
            spinner = stats.uniform.rvs(size = 1)
            if (spinner < p_move) and next_direction == 'east' :
                current_point -= 1
            elif (spinner < p_move) and next_direction == 'west':
                current_point += 1
            else:
                current_point = current_point

        return current_point

class Politician(object):

    stay_quotes = [
        "Better lay low, Lt. Daniels is out there. We can wait it out here: ",
        """Because if some federal mothafucker comes walkin' through the door,
        I say hey, it's all in the game!
        But a city police, Baltimore City, Hell Naaaw, can't be happenin'
         because I know I done raised too much goddamn money for the Mayor
          and his ticket! Hell Naaaw, aint no soul in the world that fuckin' ungrateful! """,
        """I know you don't wanna know, but Im scratchin' and clawin' to get it
         done for you Clarence, for you and me and the rest of the team and
         who comes through my door but a Baltimore City police lookin' to get
         up in my shit about everything!""",
        """You wanna run a campaign with my money pillowed under your ass, you
         need yo people to back the fuck up Clarence!""",
        """Major crimes? Sheeeeeeeeeeyet. """
    ]

    move_quotes = [
    "Sheeeeeeeeeeeeeeit, warm up the car Day-Day. We need to collect our money from : ",

    """Mm-mmm. Twenty gets you the permits.
    Five is for me for bribing these downtown motherfuckers.
     I mean, I'm the one got to risk walking up to these thieving
     bitches with cash in hand, right?""",

     """Money Launderin' they gonna come talk to me about Money Launderin'
     in West Baltimore, SHIIIIT, Where do you think I'm gonna raise cash
     for the whole damn ticket! From Laundromats and shit, from some tiny ass
      korean groceries, you think I got time to ask a man why he given me
      money or where he gets his money from, I'll take any mothafucker's money
      if he given it away! """,

     """We spend this year dealing with the city, the next doing business with
     the state. However, year three, then we go for the gold. Then we go
      federal. Then we see the man with his hand on the faucet."""
    ]

    def __init__(self, starting_point, island_range):
        self.previous_point = starting_point
        self.current_point = starting_point
        self.island_range = island_range


    def speak(self):
        if(self.previous_point != self.current_point):
            return(Politician.move_quotes[
            random.randint(1,len(Politician.move_quotes)-1)] +
            '\n\n' + Politician.move_quotes[0])
        else:
            return(Politician.stay_quotes[
            random.randint(1,len(Politician.stay_quotes)-1)] +
            '\n\n' + Politician.stay_quotes[0])


class Map(object):



    bmore_names = [
        "The Pit",
        "The Tower",
        "City Hall",
        "Little Johnnies",
        "Marlo's Court"
    ]



    n_islands = len(bmore_names)
    start_point = 3



    def __init__(self):
        self.engine = Engine()
        self.chain = {}
        self.bmore_populations = Map.shuffle_baltimore(Map.bmore_names)
        # self.island_chain = range(1,n_islands)
        self.clay = Politician(Map.start_point, [0,Map.n_islands-1])
        for i in range(0,Map.n_islands):
            self.chain[i] = 0
            if i == Map.start_point:
                self.chain[i] += 1

    def shuffle_baltimore(locations):
        # change bmore to a class instance variable so
        # it can be altered/shuffled
        return random.sample(range(1,len(locations)+1), len(locations))

    def print_histogram(self, iteration_number):
        """
        prints a scaled histogram of the distributions generated
        """
        print("Iteration %d complete. Here is the relative distribution of visits:" % iteration_number)
        for key in self.chain:
            print( "*" * math.floor(100*self.chain[key]/iteration_number),":", self.bmore_names[key]),
        print('\n\n')

    def print_relative_populations(self):
        print("True Relative Money Counts:\n")
        for p in range(0,len(self.bmore_populations)):
            print("Money Count: %r ;location: %r" %
            (self.bmore_populations[p], self.bmore_names[p]))
        print('\n\n')

    def opening_statement(self):
        print(" Welcome to Baltimore. Senator Clay Davis\n",
        "needs to pick up money from his 'constituents'. With\n",
        "the help of his trusty driver, Damien 'Day-Day' Price,\n",
        "he needs to go around and pick up his money, going\n",
        "to the places with more money more frequently.  To\n",
        "accomplish this task he is going to use the Metropolis\n",
        "Hastings algorithm.\nTime to get started....\n\n")

        self.print_relative_populations()
        ("Current Location: " + self.bmore_names[self.clay.current_point] + "\n\n")




    def simulate(self, sim_iterations=500,verbose = 'y'):

        self.opening_statement()
        prompt = '> '

        # make it verbose or not ...
        for i in range(1, sim_iterations):

            self.clay.previous_point = self.clay.current_point
            self.clay.current_point = self.engine.next_hop(self.clay.current_point,
                                                            self.clay.island_range,
                                                            self.bmore_populations)
            self.chain[self.clay.current_point] += 1
            if(verbose == 'y'):
                print("Would you like to continue stepping through simulation?\n",
                "(y - continue, any other key - show results of simulation after %d steps)" %
                sim_iterations)
                verbose = input(prompt)
                if(verbose == 'y'):
                    print("Clay: " + self.clay.speak() +
                    self.bmore_names[self.clay.current_point] + '\n\n')

            if i % 100 == 0:
                self.print_histogram(i)
                self.print_relative_populations()

m = Map()
m.simulate(sim_iterations=5000, verbose = 'y')