package examples.myUtils;

import com.tzld.piaoquan.recommend.similarity.word2vec.Segment;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class TitleConvert {
    public static String getFeature(Map<String, String> record, List<Map<String, String>> videoSeq, int scale) {
        String result = "";
        try {

            Map<String, String> headInfo = ConvertUtils.getRecordCol(record, "v2_feature");
            String title = headInfo.getOrDefault("title", "");
            if (!title.isEmpty()) {
                List<String> words = Segment.getWords(title);
                result = String.join(" ", words);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }

    private static Map<String, Map<String, String>> getUserOriginInfo(Map<String, String> record) {
        Map<String, Map<String, String>> map = new HashMap<>();
        map.put("mid_global_feature_20250212", ConvertUtils.getRecordCol(record, "c1_feature"));
        map.put("alg_mid_feature_return_tags", ConvertUtils.getRecordCol(record, "c5_feature"));
        map.put("alg_mid_feature_share_tags", ConvertUtils.getRecordCol(record, "c6_feature"));
        map.put("alg_mid_feature_sharecf", new HashMap<>());
        map.put("alg_mid_feature_returncf", new HashMap<>());
        map.put("alg_recsys_feature_user_share_return_stat", ConvertUtils.getRecordCol(record, "c9_feature"));
        return map;
    }

    private static Map<String, Map<String, Map<String, String>>> getVideoOriginInfo(Map<String, String> record) {
        Map<String, Map<String, String>> map = new HashMap<>();
        map.put("alg_recsys_feature_video_clean_stat", ConvertUtils.getRecordCol(record, "b0_feature"));
        map.put("alg_vid_global_feature_20250212", ConvertUtils.getRecordCol(record, "b1_feature"));
        map.put("alg_vid_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b2_feature"));
        map.put("alg_vid_recommend_flowpool_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b3_feature"));
        map.put("alg_vid_apptype_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b4_feature"));
        map.put("alg_vid_province_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b5_feature"));
        map.put("alg_vid_brand_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b6_feature"));
        map.put("alg_vid_hotsencetype_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b7_feature"));
        map.put("alg_merge_cate1_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b8_feature"));
        map.put("alg_merge_cate2_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b9_feature"));
        map.put("alg_channel_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b10_feature"));
        map.put("alg_festive_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b11_feature"));
        map.put("alg_video_unionid_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b13_feature"));
        map.put("scene_type_vid_cf_feature_20250212", ConvertUtils.getRecordCol(record, "d1_feature"));
        map.put("vid_click_cf_feature_20250212", ConvertUtils.getRecordCol(record, "d2_feature"));
        map.put("alg_recsys_feature_cf_i2i_v2", ConvertUtils.getRecordCol(record, "d3_feature"));
        map.put("alg_recsys_feature_video_recommend_channel_layer", ConvertUtils.getRecordCol(record, "b14_feature"));
        map.put("alg_recsys_feature_video_recommend_channel_layer_head", ConvertUtils.getRecordCol(record, "b15_feature"));

        Map<String, Map<String, Map<String, String>>> allMap = new HashMap<>();
        String vid = record.get("vid");
        allMap.put(vid, map);
        return allMap;
    }
}