Esempio n. 1
0
 def get(self, evdef, top_dt = None, end_dt = None):
     if top_dt is None:
         top_dt = self._top_dt
     if end_dt is None:
         end_dt = self._end_dt
     l_label = self._dt_label
     l_data, l_score = self._load_data(evdef)
     if len(l_data) == 0:
         return None
     if dtutil.is_sep(top_dt, self._binsize):
         top_index = l_label.index(top_dt)
     else:
         top_index = l_label.index(dtutil.adj_sep(top_dt, self._binsize))
     if dtutil.is_sep(end_dt, self._binsize):
         end_index = l_label.index(end_dt)
     else:
         end_index = l_label.index(dtutil.radj_sep(end_dt, self._binsize))
     return zip(self._dt_label, l_data, l_score)[top_index:end_index]
Esempio n. 2
0
 def get(self, evdef, top_dt=None, end_dt=None):
     if top_dt is None:
         top_dt = self._top_dt
     if end_dt is None:
         end_dt = self._end_dt
     l_label = self._dt_label
     l_data, l_score = self._load_data(evdef)
     if len(l_data) == 0:
         return None
     if dtutil.is_sep(top_dt, self._binsize):
         top_index = l_label.index(top_dt)
     else:
         top_index = l_label.index(dtutil.adj_sep(top_dt, self._binsize))
     if dtutil.is_sep(end_dt, self._binsize):
         end_index = l_label.index(end_dt)
     else:
         end_index = l_label.index(dtutil.radj_sep(end_dt, self._binsize))
     return zip(self._dt_label, l_data, l_score)[top_index:end_index]
Esempio n. 3
0
def dt_ticks(value_term, dt_term, dt_bin, duration = None, recent = False):
    """
    Args:
        value_term (int, int)
        dt_term (datetime.datetime, datetime.datetime):
                The range of datetime for ticks, corresponding to value_term.
        dt_bin (datetime.timedelta): Bin length in datetime.
        duration (datetime.timedelta): Duration of ticks. If None,
                automatically decided in this function.
        recent (bool): In default, ticks are decided from old datetime.
                If recent is true, ticks are decided from recent datetime.
    """
    import dtutil
    top_dt, end_dt = dt_term
    if duration is None:
        whole_term = end_dt - top_dt
        if whole_term <= datetime.timedelta(minutes = 3):
            duration = datetime.timedelta(seconds = 10)
        elif whole_term <= datetime.timedelta(minutes = 10):
            duration = datetime.timedelta(minutes = 1)
        elif whole_term <= datetime.timedelta(minutes = 30):
            duration = datetime.timedelta(minutes = 3)
        elif whole_term <= datetime.timedelta(hours = 1):
            duration = datetime.timedelta(minutes = 5)
        elif whole_term <= datetime.timedelta(hours = 6):
            duration = datetime.timedelta(minutes = 10)
        elif whole_term <= datetime.timedelta(days = 1):
            duration = datetime.timedelta(hours = 6)
        elif whole_term <= datetime.timedelta(days = 7):
            duration = datetime.timedelta(days = 1)
        elif whole_term <= datetime.timedelta(days = 14):
            duration = datetime.timedelta(days = 2)
        elif whole_term <= datetime.timedelta(days = 60):
            duration = datetime.timedelta(days = 7)
        elif whole_term <= datetime.timedelta(days = 300):
            duration = datetime.timedelta(days = 20)
        else:
            duration = datetime.timedelta(days = 50)

    if duration >= datetime.timedelta(days = 1):
        adjsearch_dur = datetime.timedelta(days = 1)
        dtstr = lambda dt: dt.strftime("%Y-%m-%d")
    else:
        adjsearch_dur = duration
        dtstr = lambda dt: dt.strftime("%H:%M:%S")

    ticks_label = []
    if recent is False:
        temp_dt = dtutil.radj_sep(top_dt, adjsearch_dur)
        assert(temp_dt <= end_dt)
        ticks_label.append(temp_dt)
        while 1:
            temp_dt = temp_dt + duration
            if temp_dt > end_dt:
                break
            else:
                ticks_label.append(temp_dt)
    else:
        temp_dt = dtutil.adj_sep(end_dt, adjsearch_dur)
        assert(temp_dt >= top_dt)
        ticks_label.append(temp_dt)
        while 1:
            temp_dt = temp_dt - duration
            if temp_dt < top_dt:
                break
            else:
                ticks_label.append(temp_dt)
        ticks_label.sort()

    val_label = range(value_term[0], value_term[1] + 1, 1)
    dt_label = dtutil.dtrange(top_dt, end_dt, dt_bin, include_end = True)
    ticks_values = []
    ticks_dts = ticks_label[:]
    for val, dt in zip(val_label, dt_label):
        if dt == ticks_dts[0]:
            ticks_dts.pop(0)
            ticks_values.append(val)
            if len(ticks_dts) == 0:
                break
    return ticks_values, [dtstr(l) for l in ticks_label]