#!/usr/bin/python3

from lib.ArgParser import get_config, get_model_from_config
from lib.Controller import simulate, BruteForceController, CostController, ReplayController
from lib.Cost import NegativeCost
from lib.Env import CentralHeatingInfiniteEnv

# ARGS
config = get_config()

# MODEL
model = get_model_from_config(config)

# PLOT
config['plot_fields'] = ['time', 'temp_in_calc', 'temp_out', 'temperature', 'mode_calc']

# ENV
env = CentralHeatingInfiniteEnv()
env.model = model
env.config = config
env.reset()

# MAIN
#controller = CostController(model, NegativeCost())
controller = ReplayController()
simulate(env, controller)