aswi2021tri-musketyri-gitlab/Server/ServerApp/Predictor/PredictorConfiguration.cs @ 9a529f10

//
// Author: Roman Kalivoda
//

using System.Collections.Generic;
using System;
using System.IO;
using Newtonsoft.Json;

namespace ServerApp.Predictor
{
    public class PredictorConfiguration
    {
        public static readonly string DEFAULT_CONFIG_PATH = Path.Combine(Path.GetDirectoryName(Environment.CurrentDirectory), "Predictor.config");

        public static readonly string DEFAULT_MODEL_DATA_PATH = Path.Combine(Path.GetDirectoryName(Environment.CurrentDirectory), "model_data");

        public int TimeResolution { get; set; }

        public Dictionary<string, int> BuildingsToAreas { get; set; }

        public int PredictorCount { get; set; }

        public string ModelDataPath { get; set; }

        public static PredictorConfiguration LoadConfig(string filename)
        {
            string json = System.IO.File.ReadAllText(filename);
            PredictorConfiguration configuration = JsonConvert.DeserializeObject<PredictorConfiguration>(json);
            return configuration;
        }

        public static void SaveConfig(string filename, PredictorConfiguration configuration)
        {
            string json = JsonConvert.SerializeObject(configuration);
            System.IO.File.WriteAllText(filename, json);
        }

        public static PredictorConfiguration GetDefaultConfig()
        {
            Dictionary<string, int> dict = new Dictionary<string, int>();
            var locationKeys = Parser.Parsers.TagInfo.buildings;
            foreach (string key in locationKeys)
            {
                dict.Add(key, 0);
            }

            return new PredictorConfiguration
            {
                TimeResolution = 3,
                ModelDataPath = DEFAULT_MODEL_DATA_PATH,
                PredictorCount = 7,
                BuildingsToAreas = new Dictionary<string, int>
                {
                    { "FST+FEK", 0 },
                    { "FDU", 3 },
                    { "FAV", 4 },
                    { "FEL", 5 },
                    { "REK", 6 },
                    { "MENZA", 6 },
                    { "LIB", 6 },
                    { "CIV", 6 },
                    { "UNI14", 0 },
                    { "DOM", 1 },
                    { "HUS", 1 },
                    { "CHOD", 1 },
                    { "JUNG", 1 },
                    { "KLAT", 1 },
                    { "KOLL", 1 },
                    { "RIEG", 1 },
                    { "SADY", 1 },
                    { "SED+VEL", 1 },
                    { "TES", 1 },
                    { "TYL", 1 },
                    { "KARMA", 2 },
                    { "KBORY", 2 },
                    { "KLOCH", 2 },
                    { "KKLAT", 2 }
                }
            };
        }
    }
}