| 12 | |
| 13 | |
| 14 | class Log(object): |
| 15 | def __init__(self, worker_id=None, num_of_step=None, root_path=None, time=None): |
| 16 | self.time = time |
| 17 | self.num_of_step = num_of_step |
| 18 | self.full_path = os.path.join(root_path, self.time) |
| 19 | self.worker_id = worker_id |
| 20 | if worker_id is not None: |
| 21 | self.full_path = os.path.join(self.full_path, str(worker_id)) |
| 22 | os.makedirs(self.full_path) |
| 23 | |
| 24 | self.file_path = self.full_path + '/REPORT.txt' |
| 25 | file = open(self.file_path, 'w') |
| 26 | file.close() |
| 27 | self.result_path = self.full_path + '/' + 'result.npz' |
| 28 | self.color_list = ["red", "purple", "green", "grey", "blue"] |
| 29 | |
| 30 | if worker_id == 0: |
| 31 | self.reward_list = [] |
| 32 | self.data_collection_list = [] |
| 33 | self.energy_comsumption_list = [] |
| 34 | self.eficiency_list = [] |
| 35 | self.aoi_list = [] |
| 36 | self.uav_aoi_list = [] |
| 37 | self.move_aoi_list = [] |
| 38 | self.collect_aoi_list = [] |
| 39 | self.send_back_aoi_list = [] |
| 40 | self.uav_reward_list = [] |
| 41 | self.uav_penalty_list = [] |
| 42 | self.jain_fairness_list = [] |
| 43 | |
| 44 | self.time_usage_ratio_list = [] |
| 45 | self.time_usage_ratio_min_list = [] |
| 46 | self.time_usage_ratio_max_list = [] |
| 47 | self.time_usage_ratio_std_list = [] |
| 48 | |
| 49 | self.collection_effort_ratio_list = [] |
| 50 | self.collection_effort_ratio_min_list = [] |
| 51 | self.collection_effort_ratio_max_list = [] |
| 52 | self.collection_effort_ratio_std_list = [] |
| 53 | |
| 54 | self.data_completion_ratio_list = [] |
| 55 | self.data_completion_ratio_min_list = [] |
| 56 | self.data_completion_ratio_max_list = [] |
| 57 | self.data_completion_ratio_std_list = [] |
| 58 | |
| 59 | def log(self, values): |
| 60 | if isinstance(values, dict): |
| 61 | with open(self.file_path, 'a') as file: |
| 62 | for key, value in values.items(): |
| 63 | file.write(str([key, value]) + '\n') |
| 64 | # print key, value, "file",file |
| 65 | elif isinstance(values, list): |
| 66 | with open(self.file_path, 'a') as file: |
| 67 | for value in values: |
| 68 | file.write(str(value) + '\n') |
| 69 | # print value,"file",file |
| 70 | else: |
| 71 | with open(self.file_path, 'a') as file: |
no outgoing calls
no test coverage detected