#!/usr/bin/env python
# -*- coding: utf-8 -*-
# time: 2023/5/11 14:43
# file: cluster_power.py
# author: David
# company: shenyang JY

import os
import numpy as np
import pandas as pd
# 从confinuous_data中拿数据
cluster = []

def


def cluster_power(cluster, out_put):
    """
    把聚类的结果
    """
    if not os.path.exists(out_put):
        os.makedirs(out_put)
    output_filtered_csv_files = '../data/output_filtered_csv_files/'
    files = os.listdir(output_filtered_csv_files)
    files.remove('turbine-144.csv')
    assert len(cluster) == len(files)
    continuous_data_path = '../data/continuous_data/'
    # 遍历整个continuous_data_path文件夹
    continuous_list = []
    for dirpath, dirnames, filenames in os.walk(continuous_data_path):
        dirname = dirpath.split('/')[-1]
        filenames = [file for file in filenames if not file.startswith('turbine-144')]
        x = [os.path.join(dirpath, filename) for filename in filenames]
        continuous_list.append((x, dirname))
    continuous_list.pop(0)
    for i in range(len(continuous_list)):
        cfs = {}
        dfs = [pd.read_csv(path) for path in continuous_list[i][0]]
        for j in range(1, max(cluster)+1):
            arr_j = np.where(cluster == j)[0]   # cluster中聚类j的索引列表
            cfs.setdefault(j, [files[k] for k in arr_j])
            dfs_j = [dfs[k] for k in arr_j]   # cluster
            dfj, time_series = dfs_j[0].loc[:, ['C_TIME','C_ACTIVE_POWER']], dfs_j[0]['C_TIME']
            for df in dfs_j[1:]:
                if df['C_TIME'].equals(time_series) is False:
                    print("风机之间的日期不一致！")
                    raise ValueError
                dfj['C_ACTIVE_POWER'] += df['C_ACTIVE_POWER']
            output_1 = os.path.join(out_put,  continuous_list[i][1])
            if not os.path.exists(output_1):
                os.makedirs(output_1)
            dfj.to_csv(os.path.join(output_1, 'power_' + str(j) + '.csv'), index=False)
        paint_cluster_power(output_1)
        print("------{}的分组信息----".format(continuous_list[i][1]))
        for key, value in cfs.items():
            print("第{}组：{}".format(key, cfs[key]))

def paint_cluster_power(cluster_path):
    dfs = [pd.read_csv(os.path.join(cluster_path, file_path)).rename(columns={'C_ACTIVE_POWER':file_path.split('/')[-1][:-4]}) for file_path in os.listdir(cluster_path)]
    df_cluster = pd.DataFrame({df.columns[-1]: df.iloc[:, -1] for df in dfs})
    df_cluster.insert(loc=0, column='C_TIME', value=dfs[0]['C_TIME'])
    df_cluster.to_csv(os.path.join(cluster_path, 'cluster_data.csv'), index=False)