Esempio n. 1
0
 def get_player_by_hero(self, hero, nameType=HeroIDType.NPC_NAME):
     if nameType != HeroIDType.NPC_NAME:
         convertName(hero, nameType, HeroIDType.NPC_NAME)
     try:
         return next(p for p in self.players if p.hero == hero)
     except StopIteration:
         print("Player with hero {} not found in replay {}.".format(hero, self.replayID))
         return None
Esempio n. 2
0
def x_label_icon(axis, y_pos=-0.15, size=1.0):
    x_axis = axis.get_xaxis()

    x_labels = x_axis.get_majorticklabels()
    x_locations = x_axis.get_major_locator()

    x_min, x_max = axis.get_xlim()
    x_range = x_max - x_min

    extra_artists = []
    for label, x_loc in zip(x_labels, x_locations.locs):
        # This position wont work for some plots
        # x, _ = label.get_position()
        x = x_loc
        hero = label.get_text()
        try:
            # Get and resize the hero icon.
            icon = HeroIconPrefix / convertName(hero, HeroIDType.NPC_NAME,
                                                HeroIDType.ICON_FILENAME)
        except ValueError:
            print("Unable to find hero icon for: " + hero)
            continue
        x_rel = (float(x) - x_min)/x_range
        artist = make_image_annotation(icon, axis, x_rel, y_pos, size)
        extra_artists.append(artist)

    # Remove the old xlabels
    axis.set_xticklabels([])
    return extra_artists
Esempio n. 3
0
def plot_pick_context(picks: DataFrame, team, r_query, fig: Figure, summarise=False, limit=None):
    '''Plots the context (in picks and bans) of a teams top picks.
       Input is a DataFrame of picks, the TeamInfo and r_query
       context.
    '''
    top_picks = picks.sum(axis=1).sort_values(ascending=False)
    fig.set_size_inches(8, 9)
    axes = fig.subplots(4, 5, sharey='row')

    extra_artists = []

    def _do_plot(axis, colour, data):
        data.plot.bar(ax=axis, width=0.9,
                      colormap=colour, grid=True,
                      fontsize=8, sharey=True)
        extra_artists = x_label_icon(axis, y_pos=-0.15, size=0.5)
        axis.yaxis.set_major_locator(MaxNLocator(integer=True))

        return extra_artists

    def _summarise(table, after: int):
        '''Assumes table is already sorted descending.'''
        other = table[after:].sum()
        table['Other'] = other
        table.sort_values(ascending=False, inplace=True)
        return table[:after + 1]

    for i_pick, pick in enumerate(top_picks[:5].index):
        nick = convertName(pick, HeroIDType.NPC_NAME,
                           HeroIDType.NICK_NAME)
        axes[0, i_pick].set_title(nick)

        context = pick_context(pick, team, r_query, limit=limit)
        pick = context['Pick'].sort_values(ascending=False)
        pick = _summarise(pick, 5) if summarise else pick[:5]

        ban = context['Ban'].sort_values(ascending=False)
        ban = _summarise(ban, 5) if summarise else ban[:5]

        opick = context['Opponent Pick'].sort_values(ascending=False)
        opick = _summarise(opick, 5) if summarise else opick[:5]

        oban = context['Opponent Ban'].sort_values(ascending=False)
        oban = _summarise(oban, 5) if summarise else oban[:5]

        _do_plot(axes[0, i_pick], colour_list[0], pick)
        _do_plot(axes[1, i_pick], colour_list[1], ban)
        _do_plot(axes[2, i_pick], colour_list[2], opick)
        a4 = _do_plot(axes[3, i_pick], colour_list[3], oban)
        extra_artists += a4

    axes[0, 0].set_ylabel('Pick', fontsize=14)
    axes[1, 0].set_ylabel('Ban', fontsize=14)
    axes[2, 0].set_ylabel('Opponent Pick', fontsize=14)
    axes[3, 0].set_ylabel('Opponent Ban', fontsize=14)

    return fig, axes, extra_artists
Esempio n. 4
0
def hero_box_image(hero, isPick, isFirst=False, isWinner=False):
    # Get the template box.
    hero_box = pickban_box_image(isPick=isPick, isWinner=isWinner)

    # Get and resize the hero icon.
    icon_location = HeroIconPrefix / convertName(hero, HeroIDType.NPC_NAME,
                                                 HeroIDType.ICON_FILENAME)

    icon = Image.open(icon_location)
    icon = icon.resize((64, 64))

    # Paste the icon into the box
    hero_box.paste(icon, (0, 0), icon)

    # If its the first pick we want to highlight with a triangle.
    if isFirst:
        size = hero_box.size
        canvas = ImageDraw.Draw(hero_box)

        halfPoint = math.floor((size[1]-size[0])/2)
        spacing = 0
        # Draws in the yellow triangle
        canvas.polygon([(1+spacing, size[0]+1+spacing),
                        (halfPoint+spacing, size[0]+halfPoint+spacing),
                        (1+spacing, size[1]-1+spacing)],
                       fill='yellow')

        # Text
        text = "1st"
        font_size = size[1] - size[0]
        font = ImageFont.truetype('arialbd.ttf', font_size)
        outline_colour = (255, 255, 255, 255)
        w, h = font.getsize(text)
        x, y = (0,0)
        canvas.rectangle((x, y, x + w, y + h), fill='black')
        canvas.text((x, y), text, fill='white',
                    font=font)

    return hero_box
Esempio n. 5
0
def plot_pick_pairs(data: Dict[int, DataFrame], fig: Figure, num_heroes=10):
    '''Plots pick pair combinations for the DataFrame data.
       Heroes are replaced by icons.
       num_heroes determines the number of paris to consider.
       Returns axis and extra artists for extending bounding box.
    '''

    nplots = max(data.keys()) + 1
    fig.set_size_inches(8, 4*nplots)
    axes = fig.subplots(nplots)
    # Matplotlib will return a collection if nplots > 1, or a single object if 1
    # So we just always make it a collection... Must be a better way.
    if nplots == 1:
        axes = (axes,)
    y_labels = ["First pair", "Second pair", "Third pair", "Fourth pair"]

    final_icon = []
    for i, axis in zip(range(nplots), axes):
        working = data.get(i, Series(dtype='UInt16'))
        working = working[:num_heroes]
        if working.empty:
            axis.text(0.5, 0.5, "No Data", fontsize=14,
                      horizontalalignment='center',
                      verticalalignment='center')
            axis.yaxis.set_major_locator(MaxNLocator(integer=True))
            continue

        working.plot.bar(ax=axis, width=0.9, grid=True)

        axis.set_ylabel(y_labels[i])
        axis.yaxis.set_major_locator(MaxNLocator(integer=True))

        hero_pairs = (h.split(', ') for h in working.index)

        x_axis = axis.get_xaxis()
        x_locations = x_axis.get_major_locator()
        x_min, x_max = axis.get_xlim()
        x_range = x_max - x_min

        for h_pair, x_loc in zip(hero_pairs, x_locations.locs):
            try:
                i1 = HeroIconPrefix / convertName(h_pair[0], HeroIDType.NPC_NAME,
                                                HeroIDType.ICON_FILENAME)
            except ValueError:
                print("Unable to find hero icon for: " + hero)
                continue
            try:
                i2 = HeroIconPrefix / convertName(h_pair[1], HeroIDType.NPC_NAME,
                                                HeroIDType.ICON_FILENAME)
            except ValueError:
                print("Unable to find hero icon for: " + hero)
                continue

            y_pos1 = -0.1
            y_pos2 = -0.25
            size = 1.0

            x_rel = (float(x_loc) - x_min)/x_range
            make_image_annotation(i1, axis, x_rel, y_pos1, size)
            a2 = make_image_annotation(i2, axis, x_rel, y_pos2, size)
            # Should only need the bottom artist
            # final_icon = [a2, ]
            final_icon.append(a2)

            axis.set_xticklabels([])
            axis.yaxis.set_major_locator(MaxNLocator(integer=True))

    return fig, final_icon