NDOJ/judge/utils/stats.py

from operator import itemgetter

__all__ = ('chart_colors', 'highlight_colors', 'get_pie_chart', 'get_bar_chart')


chart_colors = [0x3366CC, 0xDC3912, 0xFF9900, 0x109618, 0x990099, 0x3B3EAC, 0x0099C6, 0xDD4477, 0x66AA00, 0xB82E2E,
                0x316395, 0x994499, 0x22AA99, 0xAAAA11, 0x6633CC, 0xE67300, 0x8B0707, 0x329262, 0x5574A6, 0x3B3EAC]


highlight_colors = []


def _highlight_colors():
    for color in chart_colors:
        r, g, b = color >> 16, (color >> 8) & 0xFF, color & 0xFF
        highlight_colors.append('#%02X%02X%02X' % (min(int(r * 1.2), 255),
                                                   min(int(g * 1.2), 255),
                                                   min(int(b * 1.2), 255)))


_highlight_colors()


chart_colors = list(map('#%06X'.__mod__, chart_colors))


def get_pie_chart(data):
    return {
        'labels': list(map(itemgetter(0), data)),
        'datasets': [
            {
                'backgroundColor': chart_colors,
                'highlightBackgroundColor': highlight_colors,
                'data': list(map(itemgetter(1), data)),
            },
        ],
    }


def get_bar_chart(data, **kwargs):
    return {
        'labels': list(map(itemgetter(0), data)),
        'datasets': [
            {
                'backgroundColor': kwargs.get('fillColor', 'rgba(151,187,205,0.5)'),
                'borderColor': kwargs.get('strokeColor', 'rgba(151,187,205,0.8)'),
                'borderWidth': 1,
                'hoverBackgroundColor': kwargs.get('highlightFill', 'rgba(151,187,205,0.75)'),
                'hoverBorderColor': kwargs.get('highlightStroke', 'rgba(151,187,205,1)'),
                'data': list(map(itemgetter(1), data)),
            },
        ],
    }

def get_histogram(data, **kwargs):
    return {
        'labels': [round(i, 1) for i in list(map(itemgetter(0), data))],
        'datasets': [
            {
                'backgroundColor': kwargs.get('fillColor', 'rgba(151,187,205,0.5)'),
                'borderColor': kwargs.get('strokeColor', 'rgba(151,187,205,0.8)'),
                'borderWidth': 1,
                'hoverBackgroundColor': kwargs.get('highlightFill', 'rgba(151,187,205,0.75)'),
                'hoverBorderColor': kwargs.get('highlightStroke', 'rgba(151,187,205,1)'),
                'data': list(map(itemgetter(1), data)),
            },
        ],
    }