# coding=utf-8
import math

import matplotlib.pyplot as plt

from process.main_process import get_limit_taobao_comments
from utils.belief_propagation_utils.node_vec_utils.fea_partition import divide_fea_rates
from utils.node_vec_utils.vec_building_utils import SentenceNode

__author__ = 'jayvee'

if __name__ == '__main__':
    json_comments = get_limit_taobao_comments(1000,
                                              '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/北通阿修罗SE版USB双震动XBOX360架构PS3安卓PC电脑游戏手柄-1410934375464.txt',
                                              is_datetime=True)
    json_comments.extend(get_limit_taobao_comments(1000,
                                                   '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/包邮大学生宿舍床头置物架寝室置物架上铺必备挂篮收纳神器-1411148210639.txt',
                                                   is_datetime=True))
    json_comments.extend(get_limit_taobao_comments(10000,
                                                   '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/特价Apple苹果iPhone5c苹果手机港版美V版三网电信移动4G无锁-1411017234827.txt',
                                                   is_datetime=True))
    json_comments.extend(get_limit_taobao_comments(10000,
                                                   '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/飞利浦核心技术佰益莱空气净化器家用除甲醛pm2.5离子杀菌除烟尘-1411211340771.txt',
                                                   is_datetime=True))
    # open('../data/trainset/shenqi.json', 'w').write(json.dumps(json_comments, ensure_ascii=False))
    # for i in json_comments:
    #     print i
    # sm = cal_similarity_matrix(json_comments,
    #                            output_path='/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/visualization/flask_app/'
    #                                        'static/trainset-taobao-shenqi-0.5-0.3-100-soft.json',
    #                            sim_rate=0.5,
            self.var_node_list[node_index].px_tuple = [1 - px, px]
        for i in xrange(10):
            # iter 10 times
            for node in self.var_node_list:
                for nei_node in node.neighbors:
                    node.cal_m_ij(nei_node, self.sim_matrix[node.node_id][nei_node.node_id])
            # update
            for node in self.var_node_list:
                node.on_update()
                node.cal_bx()
        return self.var_node_list


if __name__ == '__main__':
    json_comments = get_limit_taobao_comments(50,
                                              '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/北通阿修罗SE版USB双震动XBOX360架构PS3安卓PC电脑游戏手柄-1410934375464.txt',
                                              is_datetime=True)
    # json_comments.extend(get_limit_taobao_comments(1000,
    #                                                '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/包邮大学生宿舍床头置物架寝室置物架上铺必备挂篮收纳神器-1411148210639.txt',
    #                                                is_datetime=True))
    # json_comments.extend(get_limit_taobao_comments(10000,
    #                                                '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/特价Apple苹果iPhone5c苹果手机港版美V版三网电信移动4G无锁-1411017234827.txt',
    #                                                is_datetime=True))
    # json_comments.extend(get_limit_taobao_comments(10000,
    #                                                '/Users/jayvee/github_project/shcolarship/OpinionRankProject/python/data/飞利浦核心技术佰益莱空气净化器家用除甲醛pm2.5离子杀菌除烟尘-1411211340771.txt',
    #                                                is_datetime=True))
    sent_node_fea_list = []
    sent_node_vec_list = []
    sent_node_belief_list = []
    sent_list = []
    verb_list = []