import pandas as pd
from average import average
import ray

ray.init()

ray.RAY_DISABLE_MEMORY_MONITOR=1
filepath = "D:/python_project_data/1.csv"
origin_data = pd.read_csv(filepath)
row_len = origin_data.shape[0]
cow_len = origin_data.shape[1]
average_data = origin_data
algorithm_average = average.remote()

cow_name = "G1.TTXD1_3"
contrast_data = pd.DataFrame()
contrast_data[cow_name+'_origin'] = origin_data[cow_name]
algorithm_average.set_config.remote('{"AVERAGE_NUMBER": 5 }')
# algorithm_step.set_config('{"CYCLE_TIME_BASE": 5 }')
for i in range(0,row_len):
    futures=algorithm_average.eval.remote(origin_data.loc[i,cow_name])
    average_data.loc[i, cow_name]=ray.get(futures)

print(average_data.loc[:, cow_name])
contrast_data[cow_name+'_average'] = average_data[cow_name]
# average_data.to_csv("D:/python_project_data/1_disturb.csv", index=False)
contrast_data.to_csv("D:/python_project_data/average_data.csv", index=False)


# 以下均为测试性能用
# print(algorithm_step.config_)