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
#!/usr/bin/python3
 
from lib.ArgParser import get_config, get_model_from_config
from lib.Controller import simulate, BruteForceController, CostController, ReplayController, ThermostatController
from lib.Cost import NegativeCost, NegativeCostExtendedRange
from lib.Env import CentralHeatingHistoryEnv
 
# ARGS
config = get_config(uses_future=True)
 
# MODEL
model = get_model_from_config(config)
 
# PLOT
config['plot_fields'] = ['time', 'temp_in_calc', 'temp_out', 'temperature', 'mode_calc']
config['future_values'] += config['model_future_values']
 
# ENV
env = CentralHeatingHistoryEnv()
env.model = model
env.config = config
env.cost_class = NegativeCostExtendedRange
env.reset()
 
# MAIN
# controller = BruteForceController(model=model, cost_calculator_class=NegativeCost, config=config)
controller = ThermostatController(19.6, 20.4)
# controller = ReplayController()
simulate(env, controller)