Revize 870cd163

Re #8953 Implement prediction

Server/ServerApp/Predictor/PredictionController.cs

@@ -7 +7 @@
 using ServerApp.Connection.XMLProtocolHandler;
 using ServerApp.Parser.Parsers;
 using Newtonsoft.Json;
+using ServerApp.WeatherPredictionParser;
+using ServerApp.Parser.OutputInfo;
 
 namespace ServerApp.Predictor
 {
     /// <summary>
     /// Implentation of the <c>IPredicitionController</c> interface.
     /// </summary>
-    class PredictionController : IPredictionController
+    public class PredictionController : IPredictionController
     {
         /// <summary>
         /// Configuration of the <c>Predictor</c>
@@ -32 +34 @@
         /// </summary>
         private FeatureExtractor FeatureExtractor;
 
+        /// <summary>
+        /// A weather prediction parser service
+        /// </summary>
+        private IJsonParser weatherService;
+
         /// <summary>
         /// Instantiates new prediction controller.
         /// </summary>
         /// <param name="dataParser">A data parser used to get training data.</param>
-        public PredictionController(IDataParser dataParser, string pathToConfig = null)
+        public PredictionController(IJsonParser weatherService, IDataParser dataParser, string pathToConfig = null)
         {
+            this.weatherService = weatherService;
             // load config or get the default one
             if (pathToConfig is null)
             {
@@ -82 +90 @@
 
         public Response Predict(Request request)
         {
-            throw new NotImplementedException();
+            DateTime start = new DateTime(year: request.start.year, month: request.start.month, day: request.start.day, hour: request.start.hour, minute: 0, second: 0);
+            List<Prediction> predictions = new List<Prediction>();
+            if (request.useEndDate)
+            {
+                DateTime end = new DateTime(year: request.end.year, month: request.end.month, day: request.end.day, hour: request.end.hour, minute: 0, second: 0);
+                DateTime current = start;
+                while (current < end)
+                {
+                    while (current.Hour < Date.MAX_HOUR)
+                    {
+                        var prediction = PredictSingle(request, current);
+                        predictions.Add(prediction);
+                        current = current.AddHours(this.Configuration.TimeResolution);
+                    }
+                    current = current.AddHours(23 - current.Hour + Date.MIN_HOUR);
+                }
+            } else
+            {
+                if (request.useWeather)
+                {
+                    predictions.Add(PredictSingle(request, start));
+                }
+            }
+            var response = new Response();
+            response.hoursPerSegment = Configuration.TimeResolution;
+            response.predicitons = predictions.ToArray();
+            return response;
         }
 
+        private Prediction PredictSingle(Request request, DateTime current)
+        {
+            double[] predictedValues = new double[this.Configuration.BuildingsToAreas.Count];
+            string[] predictedLabels = new string[this.Predictors.Count];
+            for (int i = 0; i < this.Predictors.Count; i++)
+            {
+                if (request.useWeather)
+                {
+                    predictedLabels[i] = this.Predictors[i].Predict(new ModelInput
+                    {
+                        Rain = (float)request.rain,
+                        Temp = (float)request.temperature,
+                        Wind = (float)request.wind,
+                        Hour = current.Hour,
+                        Time = current
+                    });
+                }
+                else
+                {
+                    List<WeatherInfo> weatherInfos = weatherService.GetPredictionForTime(from: current, to: current.AddHours(this.Configuration.TimeResolution));
+                    predictedLabels[i] = this.Predictors[i].Predict(new ModelInput
+                    {
+                        Rain = weatherInfos[0].rain,
+                        Temp = (float)weatherInfos[0].temp,
+                        Wind = (float)weatherInfos[0].wind,
+                        Hour = current.Hour,
+                        Time = current
+                    });
+                }
+            }
+            for (int i = 0; i < predictedValues.Length; i++)
+            {
+                predictedValues[i] = this.FeatureExtractor.LabelToRatio(predictedLabels[this.Configuration.BuildingsToAreas[TagInfo.buildings[i]]]) * 100;
+            }
+
+            Prediction prediction = new Prediction();
+            prediction.dateTime = new Date
+            {
+                year = current.Year,
+                month = current.Month,
+                day = current.Day,
+                hour = current.Hour
+            };
+            prediction.predictions = predictedValues;
+            return prediction;
+        }
 
         public void Train(string locationKey = null)
         {
             if (locationKey is null)
             // train all predictors
             {
-                // TODO A single predictor is used for all areas, so training is done only once now.
                 for (int i = 0; i < this.Predictors.Count; i++)
                 {
                     // train on all available data
-                    // TODO the train/test split is used just temporarily for demonstration.
                     List<ModelInput> data = FeatureExtractor.PrepareTrainingInput(i, DateTime.MinValue, DateTime.MaxValue);
                     Console.WriteLine("Training predictor with {0} samples.", data.Count);
                     this.Predictors[i].Fit(data);