import numpy as np
import pandas as pd

import configuration.config_default as cfgd

STEP_pki = 1


def encode_property_change(input_data_path, LOG=None):
    property_change_encoder = {}
    for property_name in cfgd.PROPERTIES:
        if property_name == 'pki':
            # intervals ['(3,4]', ...] 形式
            # start_map_interval <number, interval> 用来找区间
            intervals, start_map_interval = build_intervals(input_data_path, step=STEP_pki, LOG=LOG)
        
        if property_name == 'pki':
            property_change_encoder[property_name] = intervals, start_map_interval
        
    return property_change_encoder


def value_in_interval(value, start_map_interval):
    start_vals = sorted(list(start_map_interval.keys()))
    return start_map_interval[start_vals[np.searchsorted(start_vals, value, side='right') - 1]]


def interval_to_onehot(interval, encoder):
    return encoder.transform([interval]).toarray()[0]


def build_intervals(input_transformations_path, step=STEP_pki, LOG=None):
    df = pd.read_csv(input_transformations_path)
    # df=input_transformations_path
    delta_pki = df['Delta_pki'].tolist()
    min_val, max_val = min(delta_pki), max(delta_pki)
    if LOG:
         LOG.info("pki min and max: {}, {}".format(min_val, max_val))
    
    start_map_interval = {}
    interval_str = '({}, {}]'.format(round(-step/2, 2), round(step/2, 2))
    intervals = [interval_str]
    start_map_interval[-step/2] = interval_str

    smallStep=step
    bigStep=2
    positives = step/2
    while positives < 10:
        if positives>2:
            step=bigStep
        interval_str = '({}, {}]'.format(round(positives, 2), round(positives+step, 2))
        intervals.append(interval_str)
        start_map_interval[positives] = interval_str
        positives += step
        
        
    interval_str = '({}, inf]'.format(round(positives, 2))
    intervals.append(interval_str)
    start_map_interval[positives] = interval_str

    step=smallStep
    negatives = -step/2
    while negatives > min_val:
        interval_str = '({}, {}]'.format(round(negatives-step, 2), round(negatives, 2))
        intervals.append(interval_str)
        negatives -= step
        start_map_interval[negatives] = interval_str
    interval_str = '(-inf, {}]'.format(round(negatives, 2))
    intervals.append(interval_str)
    start_map_interval[float('-inf')] = interval_str

    return intervals, start_map_interval