Example #1
0
 def __init__(self, *paths, **kwargs):
     self.data_objects = []
     self.randomised = kwargs.get('randomised', {})
     for i, path in enumerate(paths):
         randomise = self.randomised.get(i + 1, False)
         self.data_objects.append(FileData(path,
                                           level=i + 1,
                                           randomise=randomise))
     self.sequence = DataSequence(*self.data_objects)
     self.number_of_levels = len(self.sequence)
     self.graph = self.sequence.get_graph()
Example #2
0
class Allocation(object):
    def __init__(self, *paths, **kwargs):
        self.data_objects = []
        self.randomised = kwargs.get('randomised', {})
        for i, path in enumerate(paths):
            randomise = self.randomised.get(i + 1, False)
            self.data_objects.append(FileData(path,
                                              level=i + 1,
                                              randomise=randomise))
        self.sequence = DataSequence(*self.data_objects)
        self.number_of_levels = len(self.sequence)
        self.graph = self.sequence.get_graph()

    def intro_string(self):
        print '''
################################################################
#                                                              #
# This is alloa by Mante Zelvyte, Uli Kraehmer and Paul Slevin #
#                                                              #
#    [email protected]                             #
#    [email protected]                                    #
#                                                              #
# Runtime with 100 agents at each hierarchy is a few ms        #
#                                                              #
################################################################\n
                '''
        for i in xrange(self.number_of_levels):
            print "{} agents of hierarchy {}".format(self.graph.hierarchies[
                                                         i].number_of_agents,
                                                     i + 1)

    def setup_allocation(self, *costs):
        assert len(costs) == len(self.data_objects) - 1
        self.graph.setup_graph(*costs)
        self.graph.set_flow()
        self.graph.simplify_flow()
        self.graph.allocate()

    def write_allocations(self):
        allocation = open(settings.ALLOCATION_PATH, "wb")
        writer = csv.writer(allocation, delimiter=",")
        for row in self.allocation:
            writer.writerow(row)
        allocation.close()

    def write_profile(self):
        profile = open(settings.ALLOCATION_PROFILE_PATH, "wb")
        writer = csv.writer(profile, delimiter="\n")
        writer.writerow(["Total number of assigned level 1 agents is " +
                         str(self.graph.max_flow),
                         "Total cost of assignment is " + str(
                             self.graph.flow_cost)])
        for i in xrange(self.number_of_levels - 1):
            writer.writerow(["", "Level {} Preference Count".format(i + 1)])
            for j in xrange(self.graph.hierarchies[i].max_preferences_length):
                split_point = self.number_of_levels
                column = [row[split_point + i] for row in self.allocation]
                count = str(column.count(j + 1))
                writer.writerow(["Number of level {} agents that were "
                                 "choice #".format(i + 2) + str(j + 1) + ": " +
                                 count])
        profile.close()

    @property
    def allocation(self):
        unnamed = self.graph.allocation
        split_point = self.number_of_levels
        rows = [[agent.name for agent in row[:split_point]] + row[split_point:]
                for row in unnamed]
        rows = sorted(rows,
                      key=lambda r: r[0][len(r[0]) - r[0][::-1].index(" "):])
        return rows