module fundit::bfiMatcher

use fundit::sqlUtilities;
use fundit::operationDataPuller;
use fundit::performanceDataPuller;
use fundit::dataSaver;


/*
 *   返回预设的指标最小值
 *   
 *   NOTE: 对数据量的要求， Java 计算coe相关性表时用48，但计算bfi时用10，这里统一用10
 * 
 */  
def get_min_threshold(data_name) {

    d = dict(STRING, FLOAT);

    d['correlation'] = 0.64;
    d['data_count'] = 10; 
    d['beta'] = 0.64;
    d['t_value'] = 2.58;
    d['r2'] = 0.64;
    d['r2_neutral'] = 0.04;

    return d[data_name];
}

/*
 *  T-value 的聚合函数版
 * 
 */
defg regressionT(y, x) {

	r = SELECT beta, tstat FROM ols(y, x, true, 1) WHERE rowNo(beta) = 1;

	return r[0]['tstat'];
}


/*
 *   计算 correlation & bfi-matching 所需要的数据指标（周数据计算，返回月度结果）
 *   
 *   NOTE: 与 Java BFI 不同，这里为了与RBSA保持统一，用收益率来计算相关性；转化成月度时用了各周平均值
 */
def cal_monthly_closity0(nav1, nav2, win) {

    n1 = nav1;
    n1.sortBy!(['entity_id', 'price_date'], [1, 1]);
    n2 = nav2;
    n2.sortBy!(['benchmark_id', 'price_date'], [1, 1]);

    t0 = SELECT entity_id, end_date, n1.price_date, n1.nav AS nav1, n1.nav.ratios()-1 AS ret1,
                n2.nav AS nav2, n2.nav.ratios()-1 AS ret2, tmoving(count, end_date, end_date, win) AS data_count
         FROM n1
         INNER JOIN n2 ON n1.end_date = n2.end_date
         ORDER BY end_date;

    t = SELECT entity_id, end_date, price_date,
               tmcorr(t0.end_date, ret1, ret2, win) AS corr,
               iif(tmstd(end_date, ret1-ret2, win) == 0, null, tmavg(end_date, ret1-ret2, win)\tmstd(end_date, ret1-ret2, win)) AS info, // 貌似没用
               iif(data_count >= get_min_threshold('data_count'), tmoving(regressionT, end_date, [ret1, ret2], win), double(NULL)) AS t_value,
               iif(data_count >= get_min_threshold('data_count'), tmbeta(end_date, ret1, ret2, win), double(NULL)) AS beta // 用 ols() 算的值和这个一样
        FROM t0
        ORDER BY price_date;

	// 将每月各周的数据平均值作为月度数据返回
    return SELECT entity_id, price_date.month().last() AS end_date, price_date.last() AS price_date,
                  corr.avg() AS corr,
                  info.avg() AS info,
                  t_value.avg() AS t_value,
                  beta.avg() AS beta
           FROM t
           GROUP BY entity_id, price_date.month();
}


/*
 *   计算 correlation & bfi-matching 所需要的数据指标（周数据计算，返回月度结果）
 *   
 *   NOTE: 与 Java BFI 不同，这里为了与RBSA保持统一，用收益率来计算相关性；转化成月度时用了各周平均值
 */
def cal_monthly_closity(entity, nav1, nav2, win) {

    n1 = nav1;
    n1.sortBy!(['entity_id', 'price_date'], [1, 1]);
    n2 = nav2;
    n2.sortBy!(['benchmark_id', 'price_date'], [1, 1]);

    t_dates = SELECT entity_id, end_date FROM nav1 WHERE end_date >= entity.price_date.weekEnd();

    t0 = SELECT entity_id, end_date, n1.price_date, n1.nav AS nav1, n1.nav.ratios()-1 AS ret1,
                n2.nav AS nav2, n2.nav.ratios()-1 AS ret2, tmoving(count, end_date, end_date, win) AS data_count
         FROM n1
         INNER JOIN n2 ON n1.end_date = n2.end_date
         ORDER BY end_date;

	t_rt = table(100:0, ['entity_id', 'end_date', 'price_date', 'corr', 'info', 't_value', 'beta'], 
				[entity.entity_id.type(), MONTH, DATE, DOUBLE, DOUBLE, DOUBLE, DOUBLE]);
				
	for(dt in t_dates.end_date) {

		if( (EXEC data_count FROM t0 WHERE end_date = dt)[0] >= get_min_threshold('data_count') ){

			rets = EXEC ret1, ret2 FROM t0 WHERE end_date BETWEEN(dt.temporalAdd(duration('-' + win$STRING)):dt);
		
			cor = corr(rets.ret1, rets.ret2);
			info = mean(rets.ret1 - rets.ret2) \ std(rets.ret1 - rets.ret2); // 貌似没用
		    t_value = regressionT(rets.ret1, rets.ret2);
	    	bta = beta(rets.ret1, rets.ret2); // 用 ols() 算的值和这个一样
	
			INSERT INTO t_rt VALUES (entity.entity_id, dt, entity.price_date, cor, info, t_value, bta);
		}
	}

	// 将每月各周的数据平均值作为月度数据返回
    return SELECT entity_id, price_date.month().last() AS end_date, price_date.last() AS price_date,
                  corr.avg() AS corr,
                  info.avg() AS info,
                  t_value.avg() AS t_value,
                  beta.avg() AS beta
           FROM t_rt
           GROUP BY entity_id, price_date.month();
}


/*
 *   计算目标和各资产类别指数及BFI因子的累计净值相关系数
 *   
 *   @param entity_info <TABLE>: [COLUMNS] entity_id, price_date
 *   
 *   NOTE: 整合 Java中 TampCalcCorrelationServiceImpl 和 BestFitIndexServiceImpl 做的计算, 但只支持周数据计算
 * 
 *   Example: cal_entity_index_coe('MF', get_fund_info(['MF00003PW1', 'MF00003PW2', 'MF00003RZI']).join(take(2024.09.30, 3) AS price_date).rename!('fund_id', 'entity_id'));
 *   		  cal_entity_index_coe('PF', get_portfolio_info([166002]).join(take(2024.09.30, 1) AS price_date).rename!('portfolio_id', 'entity_id'));
 *   
 */
def cal_entity_index_coe(entity_type, entity_info) {

// entity_type = 'PF'
// entity_info = get_portfolio_info([166002]).join(take(2024.09.30, 1) AS price_date).rename!('portfolio_id', 'entity_id');

// entity_type = 'MF'
// entity_info =  get_fund_info(['MF00003PW1', 'MF00003PW2', 'MF00003RZI']).join(take(2024.09.30, 3) AS price_date).rename!('fund_id', 'entity_id');
// entity_info = tb_cal_entity[i : min(size, i+batch_size)]

// entity_type = 'HF'
// entity_info =  get_fund_info(['HF0000A134', 'HF0000A12R', 'HF00017JVX']).join(take(2024.05.30, 3) AS price_date).rename!('fund_id', 'entity_id');


    if(entity_info.isVoid() || entity_info.size() == 0) return null;

    // 取数据集中最早日期作为因子的起始日期
    start_day = entity_info.price_date.min();

    // 取数据集每个基金组合指定日期之前5年至今的周净值
    s_json = (SELECT entity_id AS sec_id, price_date.temporalAdd(-5y) AS price_date FROM entity_info).toStdJson();

	nav_entity = get_nav_for_return_calculation(entity_type, 'w', s_json);

	if(nav_entity.isVoid() || nav_entity.size() == 0) return null;

    // 取相关性计算及BFI用得到的指数/因子列表
    // 只有基金需要单独做相关性计算，目的是为基金推荐做数据准备
    if(entity_type in ('MF', 'HF'))
	    v_indexes = (get_bfi_index_list().factor_id <- get_correlation_index_list().entity_id).distinct();
	else {
		v_indexes = get_bfi_index_list().factor_id;
		// Portfolio_id 改回整型
		v_port_id = nav_entity.sec_id$INT;
		nav_entity.replaceColumn!('sec_id', v_port_id);
	}

    s_json2 = table(v_indexes AS sec_id, take(start_day.temporalAdd(-5y), v_indexes.size()) AS price_date).toStdJson();

    // 取指数及因子周点位
    nav_index = get_nav_for_return_calculation('FA', 'w', s_json2).unionAll(get_nav_for_return_calculation('MI', 'w', s_json2));

	if(nav_index.isVoid() || nav_index.size() == 0) return null;

	// 按照SQL 建表
    entity_coe = create_entity_index_coe();

    // 两次循环遍历所有entity和指数
    for(entity in entity_info) {
//entity= entity_info[0]
        nav1 = SELECT sec_id AS entity_id, weekEnd(price_date) AS end_date, price_date, nav
               FROM nav_entity WHERE sec_id = entity.entity_id;

    	for(index in v_indexes) {
//index=v_indexes[0]
            nav2 = SELECT sec_id AS benchmark_id, weekEnd(price_date) AS end_date, price_date, nav 
                   FROM nav_index WHERE sec_id = index;

            if(nav2.isVoid() || nav2.size() == 0) continue; // 忽略已经停止更新的指数，或者是特殊的无风险利率 IN0000000M

            closity_1y = cal_monthly_closity(entity, nav1, nav2, 1y);
            closity_3y = cal_monthly_closity(entity, nav1, nav2, 3y);
            closity_5y = cal_monthly_closity(entity, nav1, nav2, 5y);

    		INSERT INTO entity_coe
        	    SELECT c1.entity_id, c1.end_date, index,
        	    	   c1.corr AS coe_1y, c3.corr AS coe_3y, c5.corr AS coe_5y, 
        	    	   //c1.corr2 AS coe_1y_2, c3.corr2 AS coe_3y_2, c5.corr2 AS coe_5y_2, 
        	           c1.info AS info_ratio_1y, c3.info AS info_ratio_3y, c5.info AS info_ratio_5y,
        	           c1.t_value AS t_value_1y, c3.t_value AS t_value_3y, c5.t_value AS t_value_5y,
        	           c1.beta AS beta_1y, c3.beta AS beta_3y, c5.beta AS beta_5y
                FROM closity_1y c1
                LEFT JOIN closity_3y  c3 ON c1.end_date = c3.end_date
                LEFT JOIN closity_5y c5 ON c1.end_date = c5.end_date;

    	}
    }

	return entity_coe;
}

/*
 *   匹配BFI, 逻辑和 Java BestFitIndexServiceImpl 类似
 * 
 *   @param entity_info <TABLE>: [COLUMNS] entity_id, strategy
 *   @param entity_coe <TABLE>: [COLUMNS] entity_id, end_date, index_id, coe_1y, t_value_1y, beta_1y
 * 
 *   NOTE: Java 中的 rule2 还包括 FA00000VN7 (100%中证全指 IN0000007N) 是不对的，而且漏掉了CTA和FOF。已将DEV数据库中此因子划入category_group 74；另外找r2最小的因子也离谱
 *                  rule3 FA00000VMX （100%中证转债 中证转债）漏掉了公募债券（FOF, 相对价值（套利），多策略，公募混合是否要加？怕会和股票打架，待研究）
 *                  
          UPDATE pfdb.`cm_factor_information` 
            SET category_group_id = 74, category='全市场', factor_name='全市场', category_group='规模', strategy=',101,102,103,107,', maximum_num=1, updaterid=123, updatetime='2024-11-25' 
          WHERE factor_id = 'FA00000VN7';
          
          UPDATE pfdb.`cm_factor_information` 
            SET category_group_id = 78, category_group='配置', maximum_num=1, updaterid=123, updatetime='2024-11-25' 
          WHERE factor_id = 'FA00000VNB' AND category_group_id = 80;
          
          UPDATE pfdb.`cm_factor_information` 
            SET category_group_id = 78, category_group='配置', maximum_num=1, updaterid=123, updatetime='2024-11-25' 
          WHERE factor_id = 'FA00000VND' AND category_group_id = 76;
 *         
 */
def match_entity_bfi(entity_type, entity_info, entity_coe) {

	// 特殊因子：现金，可被应用于所有策略
    v_factor_cash = ['FA000000MJ'];

	//有一些特殊的因子只会被部分策略所用, 否则会引起歧义
	v_factor_1 = ['FA00000VMY', 'FA00000VMZ', 'FA00000VN0', 'FA00000VN1', 'FA00000VN2', 'FA00000VN3', 'FA00000VN4', 'FA00000VN5', 'FA00000VN6'];
	v_strategy_1 = [3, 7, 8, 105]; // 私募CTA, 私募FOF, 私募多策略, 公募商品

	v_factor_2 = ['FA00000SMB', 'FA00000VMG'];
	v_strategy_2 = [1, 3, 5, 7, 8]; // 私募股票（多空），CTA, 相对价值，私募FOF, 私募多策略

	v_factor_3 = ['FA00000VMX'];
	v_strategy_3 = [6, 103]; // 私募固收，公募债券

	// 只需要BFI因子的相关性数据
	coe = SELECT *
	      FROM ej(entity_info, ej(entity_coe, get_bfi_index_list(), 'index_id', 'factor_id'), 'entity_id')
		  ORDER BY entity_id, end_date, category_group_id, coe_1y DESC, order_id;

	t_bfi_raw = table(1000:0, 
					  ['entity_id', 'end_date', 'category_group_id', 'factor_id', 'rank', 'coe_1y', 'r2',
					  //'rank2', 'coe_1y_2', 'r2_2',
					   'performance_flag', 't_value_1y', 'beta_1y', 'maximum_num', 'order_id', 'factor_name'],
					  [iif(entity_type=='PF', INT, SYMBOL), MONTH, SHORT, SYMBOL, SHORT, DOUBLE, DOUBLE,
					  //SHORT, DOUBLE, DOUBLE,
					   STRING, DOUBLE, DOUBLE, SHORT, SHORT, STRING]);				  

	// 首先处理特殊情况 TODO: java treats rule2 differently by finding min R2 without checking t_value & corr
	v_special_rule = [v_factor_1, v_factor_2, v_factor_3];
	v_special_strategy = [v_strategy_1, v_strategy_2, v_strategy_3];
	
	for(i in 0..v_special_rule.size()-1) {
		INSERT INTO t_bfi_raw
			SELECT * FROM (
				SELECT entity_id, end_date, category_group_id, index_id AS factor_id,
					   coe_1y.rank(false) AS rank, coe_1y, square(coe_1y) AS r2, 
					   'w', t_value_1y, beta_1y,
					   maximum_num, order_id, factor_name
				FROM entity_info ei
				INNER JOIN coe ON ei.entity_id = coe.entity_id
				WHERE ei.strategy IN v_special_strategy[i]
				  AND coe.index_id IN v_special_rule[i].join(v_factor_cash)
				  AND t_value_1y >= get_min_threshold('t_value') 
	 			  AND coe_1y >= get_min_threshold('correlation')
				  AND order_id IS NOT NULL
				CONTEXT BY entity_id, end_date, category_group_id )
			WHERE rank < maximum_num;

		DELETE FROM coe WHERE index_id IN v_special_rule[i];
	}
	
	INSERT INTO t_bfi_raw
		SELECT * FROM (
			SELECT entity_id, end_date, category_group_id, index_id AS factor_id,
				   coe_1y.rank(false) AS rank, coe_1y, square(coe_1y) AS r2, 
				   'w', t_value_1y, beta_1y,
				   maximum_num, order_id, factor_name
			FROM entity_info ei
			INNER JOIN coe ON ei.entity_id = coe.entity_id
			WHERE t_value_1y >= get_min_threshold('t_value')

			  AND coe_1y >= get_min_threshold('correlation')
			  AND order_id IS NOT NULL
			CONTEXT BY entity_id, end_date, category_group_id )
		WHERE rank < maximum_num;

	return SELECT * FROM t_bfi_raw ORDER BY entity_id, end_date, category_group_id;

}