xyzshop/plot/refill/ChannelCumPainter.py

from .DataStream import EChPosmap as pos_map
from .ChannelReader import ChannelReader
from .ChannelPainter import ChannelPainter, ratio_pathes
from matplotlib.figure import Figure
from matplotlib import ticker
from io import BytesIO
import numpy as np
from .algorithm import calc_chratios
import time as time

import logging
logger = logging.getLogger('ChannelCumPainter')

class ChannelCumPainter(ChannelPainter):
    def __init__(self, start_time: int, end_time: int, chnames: set = None, card_types: set = None, spec: int = None, filter_wave: int = None,
                 only_all: bool = False):
        self._reader = ChannelReader()
        filter_wave = filter_wave or 0
        self._chnames, self._card_types, self._spec, self._filter_wave,self._only_all = chnames, card_types, spec, filter_wave,only_all
        days, self._start_time, self._end_time, self._interval = self.calc_time(self._reader, start_time, end_time)
        pass

    def paint(self):
        reader = ChannelReader()
        days, _start_time, _end_time, _interval = self.calc_time(self._reader, self._start_time - self._filter_wave, self._end_time)

        tuple_pathes = reader.many_tuple_path(days, self._chnames, self._card_types, self._spec)
        gen = ratio_pathes(reader, tuple_pathes, days, self._spec)
        if len(days) == 0:
            return BytesIO()
        day_stamp = days[0]

        fig_create, fig_flush = self._fig_funs()
        ax, fig = fig_create()
        x = np.array([d - self._start_time for d in range(self._start_time, self._end_time)])

        if self._filter_wave is not None and self._filter_wave > 1:
            window = np.ones(self._filter_wave) / self._filter_wave
        else:
            window = None

        left_len = self._start_time - _start_time
        right_len = 0

        stime = lambda t: time.strftime('%d-%H:%M:%S', time.localtime(t))
        logger.debug("start_time %d , %s end_time=%s", self._start_time, stime(self._start_time), stime(self._end_time))

        chname_ratios = []
        for _chname, _card_type, _spec, _data in gen:
            _succ, count, y = calc_chratios(_data, pos_map, _start_time - day_stamp, _end_time - day_stamp, window, left_len, right_len)
            label, ratio = self._label(chname=_chname, succ=_succ, count=count, card_type=_card_type, spec=_spec)
            ax.plot(x, y, ls='-', label=label)
            if _card_type is None and _spec is None and _chname != 'all':
                chname_ratios.append((_chname, ratio, _succ, count))

        ticks = [d - self._start_time for d in range(self._start_time, self._end_time + 1, self._interval)]

        ticks = self.calc_xticks(self._start_time, self._end_time, self._interval)
        xlables = [time.strftime('%d-%H:%M:%S', time.localtime(d + self._start_time)) for d in ticks]
        buf = fig_flush(ax, fig, ticks, xlables)

        chname_ratios = sorted(chname_ratios, key=lambda x: x[1], reverse=True)

        result = {}
        ratios = {}
        for _chname, ratio, _succ, count in chname_ratios:
            ratios[name] = [ratio,_succ,count]
        result['detail'] = ratios

        return buf, result