aswi2021tri-musketyri-gitlab/Server/ServerApp/Predictor/FeatureExtractor.cs @ 662b2404

//
// Author: Roman Kalivoda
//

using System.Collections.Generic;
using System;
using ServerApp.Parser.Parsers;

namespace ServerApp.Predictor
{
    /// <summary>
    /// A class responsible for preparation of features for classifiers.
    /// </summary>
    public class FeatureExtractor
    {
        /// <summary>
        /// A DataParser instance used to access info objects.
        /// </summary>
        private readonly IDataParser dataParser;

        private Dictionary<string, int> buildingsToAreas;

        /// <summary>
        /// Instantiates new FeatureExtractor class.
        /// </summary>
        /// <param name="dataParser">Data parser used to access training data.</param>
        public FeatureExtractor(IDataParser dataParser, Dictionary<string, int> buildingsToAreas)
        {
            this.dataParser = dataParser;
            this.buildingsToAreas = buildingsToAreas;
        }


        /// <summary>
        /// Extracts list of feature vectors from parsed info objects.
        /// </summary>
        /// <param name="weatherInfos">List of weather info objects.</param>
        /// <param name="activityInfos">List of info objects about activities at the site.</param>
        /// <returns>A list of feature vectors for model training.</returns>
        public IEnumerable<ModelInput> PrepareModelInput(int area, DateTime startDate, DateTime endDate, int interval = 1, bool wholeDay = true)
        {
            dataParser.Parse(startDate, endDate, interval, wholeDay);
            List<string> buildings = new List<string>();

            // find all buildings in area
            foreach (KeyValuePair<string, int> kvp in buildingsToAreas)
            {
                if (kvp.Value == area)
                {
                    buildings.Add(kvp.Key);
                }
            }

            var res =  new List<ModelInput>();
            // use enumerator to access records about attendance and weather together
            IEnumerator<Parser.OutputInfo.WeatherInfo> weatherEnumerator = dataParser.WeatherList.GetEnumerator();
            weatherEnumerator.MoveNext();
            foreach (Parser.OutputInfo.ActivityInfo val in dataParser.AttendanceList)
            {
                if (!weatherEnumerator.Current.startTime.Equals(val.startTime))
                {
                    weatherEnumerator.MoveNext();
                }
                if (buildings.Contains(val.building))
                {
                    Parser.OutputInfo.WeatherInfo weatherInfo = weatherEnumerator.Current;
                    res.Add(new ModelInput
                    {
                        Temp = (float)weatherInfo.temp,
                        Time = val.startTime,
                        Wind = (float)weatherInfo.wind,
                        Rain = (float)weatherInfo.rain,
                        // TODO convert val.amount to label
                    });
                }
            }

            return res;
        }
    }
}