// Copyright (C) Stichting Deltares 2016. All rights reserved.
//
// This file is part of Ringtoets.
//
// Ringtoets is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// All names, logos, and references to "Deltares" are registered trademarks of
// Stichting Deltares and remain full property of Stichting Deltares at all times.
// All rights reserved.

using System;
using System.Collections.Generic;
using System.Linq;
using Core.Common.Base.Data;
using Core.Common.Base.IO;
using Core.Common.IO.Exceptions;
using Core.Common.IO.Readers;
using log4net;
using Ringtoets.Common.Data.Calculation;
using Ringtoets.Common.Data.Hydraulics;
using Ringtoets.Common.Data.Probabilistics;
using Ringtoets.Common.IO.Exceptions;
using Ringtoets.Piping.Data;
using Ringtoets.Piping.IO.Properties;
using Ringtoets.Piping.IO.Readers;
using Ringtoets.Piping.IO.Schema;
using Ringtoets.Piping.Primitives;
using RingtoetsCommonIOResources = Ringtoets.Common.IO.Properties.Resources;

namespace Ringtoets.Piping.IO.Importers
{
    /// <summary>
    /// Imports a piping configuration and stores it on a <see cref="CalculationGroup"/>.
    /// </summary>
    public class PipingConfigurationImporter : FileImporterBase<CalculationGroup>
    {
        private static readonly ILog log = LogManager.GetLogger(typeof(PipingConfigurationImporter));

        private readonly IEnumerable<HydraulicBoundaryLocation> hydraulicBoundaryLocations;
        private readonly PipingFailureMechanism failureMechanism;

        /// <summary>
        /// Creates a new instance of <see cref="PipingConfigurationImporter"/>.
        /// </summary>
        /// <param name="filePath">The path to the file to import from.</param>
        /// <param name="importTarget">The calculation group to update.</param>
        /// <param name="hydraulicBoundaryLocations">The hydraulic boundary locations
        /// used to check if the imported objects contain the right location.</param>
        /// <param name="failureMechanism">The piping failure mechanism used to check
        /// if the imported objects contain the right data.</param>
        /// <exception cref="ArgumentNullException">Thrown when any parameter is
        /// <c>null</c>.</exception>
        public PipingConfigurationImporter(string filePath,
                                           CalculationGroup importTarget,
                                           IEnumerable<HydraulicBoundaryLocation> hydraulicBoundaryLocations,
                                           PipingFailureMechanism failureMechanism)
            : base(filePath, importTarget)
        {
            if (hydraulicBoundaryLocations == null)
            {
                throw new ArgumentNullException(nameof(hydraulicBoundaryLocations));
            }
            if (failureMechanism == null)
            {
                throw new ArgumentNullException(nameof(failureMechanism));
            }
            this.hydraulicBoundaryLocations = hydraulicBoundaryLocations;
            this.failureMechanism = failureMechanism;
        }

        protected override void LogImportCanceledMessage()
        {
            log.Info(Resources.PipingConfigurationImporter_LogImportCanceledMessage_Import_canceled_no_data_read);
        }

        protected override bool OnImport()
        {
            NotifyProgress(Resources.PipingConfigurationImporter_ProgressText_Reading_configuration, 1, 3);

            ReadResult<IReadPipingCalculationItem> readResult = ReadConfiguration();
            if (readResult.CriticalErrorOccurred || Canceled)
            {
                return false;
            }

            NotifyProgress(Resources.PipingConfigurationImporter_ProgressText_Validating_imported_data, 2, 3);

            var validCalculationItems = new List<ICalculationBase>();

            foreach (IReadPipingCalculationItem readItem in readResult.Items)
            {
                if (Canceled)
                {
                    return false;
                }

                ICalculationBase processedItem = ProcessReadItem(readItem);
                if (processedItem != null)
                {
                    validCalculationItems.Add(processedItem);
                }
            }

            NotifyProgress(RingtoetsCommonIOResources.Importer_ProgressText_Adding_imported_data_to_data_model, 3, 3);
            AddItemsToModel(validCalculationItems);

            return true;
        }

        private ReadResult<IReadPipingCalculationItem> ReadConfiguration()
        {
            try
            {
                return new ReadResult<IReadPipingCalculationItem>(false)
                {
                    Items = new PipingConfigurationReader(FilePath).Read().ToList()
                };
            }
            catch (Exception exception) when (exception is ArgumentException
                                              || exception is CriticalFileReadException)
            {
                string errorMessage = string.Format(Resources.PipingConfigurationImporter_HandleCriticalFileReadError_Error_0_no_configuration_imported,
                                                    exception.Message);
                log.Error(errorMessage, exception);
                return new ReadResult<IReadPipingCalculationItem>(true);
            }
        }

        private ICalculationBase ProcessReadItem(IReadPipingCalculationItem readItem)
        {
            var readCalculationGroup = readItem as ReadPipingCalculationGroup;
            if (readCalculationGroup != null)
            {
                return ProcessCalculationGroup(readCalculationGroup);
            }

            var readCalculation = readItem as ReadPipingCalculation;
            if (readCalculation != null)
            {
                return ProcessCalculation(readCalculation);
            }

            return null;
        }

        private CalculationGroup ProcessCalculationGroup(ReadPipingCalculationGroup readCalculationGroup)
        {
            var group = new CalculationGroup(readCalculationGroup.Name, true);

            foreach (IReadPipingCalculationItem item in readCalculationGroup.Items)
            {
                ICalculationBase processedItem = ProcessReadItem(item);
                if (processedItem != null)
                {
                    group.Children.Add(processedItem);
                }
            }

            return group;
        }

        private PipingCalculationScenario ProcessCalculation(ReadPipingCalculation readCalculation)
        {
            var pipingCalculation = new PipingCalculationScenario(new GeneralPipingInput())
            {
                Name = readCalculation.Name
            };

            try
            {
                ReadHydraulicBoundaryData(readCalculation, pipingCalculation);
                ReadSurfaceLine(readCalculation, pipingCalculation);
                ReadEntryExitPoint(readCalculation, pipingCalculation);
                ReadStochasticSoilModel(readCalculation, pipingCalculation);
                ReadStochasticSoilProfile(readCalculation, pipingCalculation);
                ReadStochasts(readCalculation, pipingCalculation);
            }
            catch (Exception e) when (e is CriticalFileValidationException || e is ArgumentOutOfRangeException)
            {
                string message = string.Format(Resources.PipingConfigurationImporter_ValidateCalculation_Error_message_0_calculation_1_skipped,
                                               e.Message,
                                               readCalculation.Name);
                log.Error(message, e);
                return null;
            }

            return pipingCalculation;
        }

        /// <summary>
        /// Reads the hydraulic boundary location or the assessment level that is manually set.
        /// </summary>
        /// <param name="readCalculation">The calculation read from the imported file.</param>
        /// <param name="pipingCalculation">The calculation to configure.</param>
        /// <exception cref="CriticalFileValidationException">Thrown when the <paramref name="readCalculation"/>
        /// has a <see cref="HydraulicBoundaryLocation"/> set which is not available in <see cref="hydraulicBoundaryLocations"/>.</exception>
        private void ReadHydraulicBoundaryData(ReadPipingCalculation readCalculation, PipingCalculationScenario pipingCalculation)
        {
            if (readCalculation.HydraulicBoundaryLocation != null)
            {
                HydraulicBoundaryLocation location = hydraulicBoundaryLocations
                    .FirstOrDefault(l => l.Name == readCalculation.HydraulicBoundaryLocation);

                if (location == null)
                {
                    throw new CriticalFileValidationException(string.Format(Resources.PipingConfigurationImporter_ReadHydraulicBoundaryLocation_Hydraulic_boundary_location_0_does_not_exist,
                                                                            readCalculation.HydraulicBoundaryLocation));
                }

                pipingCalculation.InputParameters.HydraulicBoundaryLocation = location;
            }
            else if (readCalculation.AssessmentLevel.HasValue)
            {
                pipingCalculation.InputParameters.UseAssessmentLevelManualInput = true;
                pipingCalculation.InputParameters.AssessmentLevel = (RoundedDouble) readCalculation.AssessmentLevel;
            }
        }

        /// <summary>
        /// Reads the surface line.
        /// </summary>
        /// <param name="readCalculation">The calculation read from the imported file.</param>
        /// <param name="pipingCalculation">The calculation to configure.</param>
        /// <exception cref="CriticalFileValidationException">Thrown when the <paramref name="readCalculation"/>
        /// has a <see cref="RingtoetsPipingSurfaceLine"/> set which is not available in the failure mechanism.</exception>
        private void ReadSurfaceLine(ReadPipingCalculation readCalculation, PipingCalculationScenario pipingCalculation)
        {
            if (readCalculation.SurfaceLine != null)
            {
                RingtoetsPipingSurfaceLine surfaceLine = failureMechanism.SurfaceLines
                    .FirstOrDefault(sl => sl.Name == readCalculation.SurfaceLine);

                if (surfaceLine == null)
                {
                    throw new CriticalFileValidationException(string.Format(Resources.PipingConfigurationImporter_ReadSurfaceLine_SurfaceLine_0_does_not_exist,
                                                                            readCalculation.SurfaceLine));
                }

                pipingCalculation.InputParameters.SurfaceLine = surfaceLine;
            }
        }

        private static void ReadEntryExitPoint(ReadPipingCalculation readCalculation, PipingCalculationScenario pipingCalculation)
        {
            if (readCalculation.EntryPointL.HasValue)
            {
                var entryPoint = (double) readCalculation.EntryPointL;

                PerformActionHandlingAnyArgumentOutOfRangeException(
                    () => pipingCalculation.InputParameters.EntryPointL = (RoundedDouble) entryPoint,
                    string.Format(Resources.PipingConfigurationImporter_ReadEntryExitPoint_Entry_point_invalid, entryPoint));
            }

            if (readCalculation.ExitPointL.HasValue)
            {
                var exitPoint = (double) readCalculation.ExitPointL;

                PerformActionHandlingAnyArgumentOutOfRangeException(
                    () => pipingCalculation.InputParameters.ExitPointL = (RoundedDouble) exitPoint,
                    string.Format(Resources.PipingConfigurationImporter_ReadEntryExitPoint_Exit_point_invalid, exitPoint));
            }
        }

        /// <summary>
        /// Reads the stochastic soil model.
        /// </summary>
        /// <param name="readCalculation">The calculation read from the imported file.</param>
        /// <param name="pipingCalculation">The calculation to configure.</param>
        /// <exception cref="CriticalFileValidationException">Thrown when
        /// <list type="bullet">
        /// <item>the <paramref name="readCalculation"/> has a <see cref="StochasticSoilModel"/> set
        /// which is not available in the failure mechanism.</item>
        /// <item>The <see cref="StochasticSoilModel"/> does not intersect with the <see cref="RingtoetsPipingSurfaceLine"/>
        /// when this is set.</item>
        /// </list>
        /// </exception>
        private void ReadStochasticSoilModel(ReadPipingCalculation readCalculation, PipingCalculationScenario pipingCalculation)
        {
            if (readCalculation.StochasticSoilModel != null)
            {
                StochasticSoilModel soilModel = failureMechanism.StochasticSoilModels
                    .FirstOrDefault(ssm => ssm.Name == readCalculation.StochasticSoilModel);

                if (soilModel == null)
                {
                    throw new CriticalFileValidationException(string.Format(Resources.PipingConfigurationImporter_ReadStochasticSoilModel_Stochastische_soil_model_0_does_not_exist,
                                                                            readCalculation.StochasticSoilModel));
                }

                if (pipingCalculation.InputParameters.SurfaceLine != null
                    && !soilModel.IntersectsWithSurfaceLineGeometry(pipingCalculation.InputParameters.SurfaceLine))
                {
                    throw new CriticalFileValidationException(string.Format(Resources.PipingConfigurationImporter_ReadStochasticSoilModel_Stochastische_soil_model_0_does_not_intersect_with_surfaceLine_1,
                                                                            readCalculation.StochasticSoilModel,
                                                                            readCalculation.SurfaceLine));
                }

                pipingCalculation.InputParameters.StochasticSoilModel = soilModel;
            }
        }

        /// <summary>
        /// Reads the stochastic soil profile.
        /// </summary>
        /// <param name="readCalculation">The calculation read from the imported file.</param>
        /// <param name="pipingCalculation">The calculation to configure.</param>
        /// <exception cref="CriticalFileValidationException">Thrown when the <paramref name="readCalculation"/>
        /// has a <see cref="StochasticSoilProfile"/> set which is not available in the <see cref="StochasticSoilModel"/>.</exception>
        private static void ReadStochasticSoilProfile(ReadPipingCalculation readCalculation,
                                                      PipingCalculationScenario pipingCalculation)
        {
            if (readCalculation.StochasticSoilProfile != null)
            {
                if (pipingCalculation.InputParameters.StochasticSoilModel == null)
                {
                    throw new CriticalFileValidationException(string.Format(Resources.PipingConfigurationImporter_ReadStochasticSoilProfile_No_soil_model_provided_for_soil_profile_with_name_0,
                                                                            readCalculation.StochasticSoilProfile));
                }

                StochasticSoilProfile soilProfile = pipingCalculation.InputParameters.StochasticSoilModel.StochasticSoilProfiles
                                                                     .FirstOrDefault(ssp => ssp.SoilProfile.Name == readCalculation.StochasticSoilProfile);

                if (soilProfile == null)
                {
                    throw new CriticalFileValidationException(string.Format(Resources.PipingConfigurationImporter_ReadStochasticSoilProfile_Stochastic_soil_profile_0_does_not_exist_within_soil_model_1,
                                                                            readCalculation.StochasticSoilProfile, readCalculation.StochasticSoilModel));
                }

                pipingCalculation.InputParameters.StochasticSoilProfile = soilProfile;
            }
        }

        private static void ReadStochasts(ReadPipingCalculation readCalculation, PipingCalculationScenario pipingCalculation)
        {
            if (readCalculation.PhreaticLevelExitMean.HasValue && readCalculation.PhreaticLevelExitStandardDeviation.HasValue)
            {
                var normalDistribution = new NormalDistribution();

                var mean = (double) readCalculation.PhreaticLevelExitMean;
                PerformActionHandlingAnyArgumentOutOfRangeException(
                    () => normalDistribution.Mean = (RoundedDouble) mean,
                    string.Format(Resources.PipingConfigurationImporter_ReadStochasts_Invalid_mean_0_for_stochast_with_name_1, mean,
                                  PipingConfigurationSchemaIdentifiers.PhreaticLevelExitStochastName));

                var standardDeviation = (double) readCalculation.PhreaticLevelExitStandardDeviation;
                PerformActionHandlingAnyArgumentOutOfRangeException(
                    () => normalDistribution.StandardDeviation = (RoundedDouble) standardDeviation,
                    string.Format(Resources.PipingConfigurationImporter_ReadStochasts_Invalid_standard_deviation_0_for_stochast_with_name_1, mean,
                                  PipingConfigurationSchemaIdentifiers.PhreaticLevelExitStochastName));

                pipingCalculation.InputParameters.PhreaticLevelExit = normalDistribution;
            }

            if (readCalculation.DampingFactorExitMean.HasValue && readCalculation.DampingFactorExitStandardDeviation.HasValue)
            {
                var logNormalDistribution = new LogNormalDistribution();

                var mean = (double) readCalculation.DampingFactorExitMean;
                PerformActionHandlingAnyArgumentOutOfRangeException(
                    () => logNormalDistribution.Mean = (RoundedDouble) mean,
                    string.Format(Resources.PipingConfigurationImporter_ReadStochasts_Invalid_mean_0_for_stochast_with_name_1, mean,
                                  PipingConfigurationSchemaIdentifiers.DampingFactorExitStochastName));

                var standardDeviation = (double) readCalculation.DampingFactorExitStandardDeviation;
                PerformActionHandlingAnyArgumentOutOfRangeException(
                    () => logNormalDistribution.StandardDeviation = (RoundedDouble) standardDeviation,
                    string.Format(Resources.PipingConfigurationImporter_ReadStochasts_Invalid_standard_deviation_0_for_stochast_with_name_1, mean,
                                  PipingConfigurationSchemaIdentifiers.DampingFactorExitStochastName));

                pipingCalculation.InputParameters.DampingFactorExit = logNormalDistribution;
            }
        }

        /// <summary>
        /// Performs the provided <paramref name="action"/> and handles any thrown <see cref="ArgumentOutOfRangeException"/>.
        /// </summary>
        /// <param name="action">The action to perform.</param>
        /// <param name="errorMessage">The error message to provide when rethrowing any thrown <see cref="ArgumentOutOfRangeException"/>.</param>
        /// <exception cref="CriticalFileValidationException">Thrown when <paramref name="action"/> throws an <see cref="ArgumentOutOfRangeException"/>.</exception>
        private static void PerformActionHandlingAnyArgumentOutOfRangeException(Action action, string errorMessage)
        {
            try
            {
                action();
            }
            catch (ArgumentOutOfRangeException e)
            {
                throw new CriticalFileValidationException($"{errorMessage} {e.Message}");
            }
        }

        private void AddItemsToModel(IEnumerable<ICalculationBase> validCalculationItems)
        {
            foreach (ICalculationBase validCalculationItem in validCalculationItems)
            {
                ImportTarget.Children.Add(validCalculationItem);
            }
        }
    }
}