commit | author | age
|
425bf7
|
1 |
#!/usr/bin/python3 |
JK |
2 |
|
|
3 |
import collections |
|
4 |
import functools |
|
5 |
|
|
6 |
from lib.ArgParser import get_config, get_model_from_config |
|
7 |
from lib.Controller import simulate, BruteForceController, CostController, ReplayController, ThermostatController |
|
8 |
from lib.Cost import NegativeCost, NegativeCostExtendedRange |
|
9 |
from lib.Env import CentralHeatingHistoryEnv |
|
10 |
|
|
11 |
# ARGS |
|
12 |
config = get_config(uses_future=True) |
|
13 |
|
|
14 |
# MODEL |
|
15 |
model = get_model_from_config(config) |
|
16 |
|
|
17 |
# PLOT |
|
18 |
config['plot_fields'] = ['time', 'temp_in_calc', 'temp_out', 'temperature', 'mode_calc'] |
|
19 |
config['future_values'] += config['model_future_values'] |
|
20 |
|
|
21 |
# MAIN |
|
22 |
plot = collections.OrderedDict() |
|
23 |
|
|
24 |
for controller_type, controller in ( |
|
25 |
('controler', BruteForceController(model=model, cost_calculator_class=NegativeCost, config=config)), |
|
26 |
('thermostat', ThermostatController(19.6, 20.4)) |
|
27 |
): |
|
28 |
env = CentralHeatingHistoryEnv() |
|
29 |
env.model = model |
|
30 |
env.config = config |
|
31 |
env.cost_class = NegativeCostExtendedRange |
|
32 |
env.reset() |
|
33 |
|
|
34 |
simulate(env, controller, render=False) |
|
35 |
|
|
36 |
plot['time'] = env.window.plot['time'] |
|
37 |
plot['temp_out'] = env.window.plot['temp_out'] |
|
38 |
plot['temp_in_' + controller_type] = env.window.plot['temp_in_calc'] |
|
39 |
plot['mode_' + controller_type] = env.window.plot['mode_calc'] |
|
40 |
|
|
41 |
print(plot.keys()) |
|
42 |
env.window.plot = plot |
|
43 |
|
|
44 |
env.render() |