1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
from json import JSONDecodeError, load
UNIT_FACTORS: dict[int, str] = {
-9: 'n',
-6: 'μ',
-3: 'm',
0: '',
1: 'k',
3: 'M',
6: 'G',
9: 'T'
}
SIMULATION_ANALYSIS_FOLDER_NAME = 'simulation-analysis'
EMISSIONS_ANALYSIS_FOLDER_NAME = 'carbon_emission'
ENERGY_ANALYSIS_FOLDER_NAME = 'power_draw'
"""
SIMULATION_DATA_FILE (str): The name of the file containing the simulation data. Enter only the name, not the path, not
the extension. The data file must be parquet format.
✅ Good: "host", "simulation_data", "cats_predictions"
❌ Wrong: "host.json", "opendc/folder_x/folder_y/data"
"""
SIMULATION_DATA_FILE = "host" # opendc outputs in file host.parquet
def adjust_unit(target_unit: str, magnitude: int) -> tuple[str, int]:
"""
Adjusts the unit based on the magnitude provided.
Example:
adjust_unit('W', 3) -> ('kW', 1000)
Args:
target_unit: The target unit to adjust.
magnitude: The magnitude to adjust the unit by.
Returns:
A tuple containing the adjusted unit and magnitude.
"""
result_unit = UNIT_FACTORS.get(magnitude, '') + target_unit
result_magnitude = (10 ** magnitude) if magnitude in UNIT_FACTORS else 1
return result_unit, result_magnitude
def clean_analysis_file(metric: str) -> None:
analysis_file_path = SIMULATION_ANALYSIS_FOLDER_NAME + "/"
if metric == "power_draw":
analysis_file_path += ENERGY_ANALYSIS_FOLDER_NAME
else:
analysis_file_path += EMISSIONS_ANALYSIS_FOLDER_NAME
analysis_file_path += "/analysis.txt"
with open(analysis_file_path, "w") as f:
f.write("")
def parse_json(json_path: str) -> dict[str, any]:
"""
Parses a JSON file and returns the dictionary representation.
Args:
json_path: The path to the JSON file.
Returns:
A dictionary containing the JSON data.
"""
try:
with open(json_path, 'r') as raw_json:
return load(raw_json)
except JSONDecodeError:
print(f"Error decoding JSON in file: {json_path}")
exit(1)
except IOError:
print(f"Error reading file: {json_path}")
exit(1)
|
