// Copyright (C) Stichting Deltares 2017. All rights reserved.
//
// This file is part of the Dam Engine.
//
// The Dam Engine is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero 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 Deltares.DamEngine.Data.General;
using Deltares.DamEngine.Data.RWScenarios;

namespace Deltares.DamEngine.Calculators.Dikes_Assessment_Regional
{
    public class RWScenarioSelector
    {
        public RWScenarioSelector()
        {
        }

        public PipingModelType PipingModelType { get; set; }
        public MStabParameters MStabParameters { get; set; }

        private bool WillChoicesResultInScenarios(params Enum[] previousChoices)
        {
            if (previousChoices.Length >= 2)
            {
                if (previousChoices[0].Equals(LoadSituation.Dry) && previousChoices[1].Equals(DikeDrySensitivity.None))
                {
                    return false;
                }
            }

            return true;
        }

        public RWScenarioProfileResult[] GetScenarios(Location location, SoilGeometryProbability soilGeometryProbability)
        {
            List<RWScenarioProfileResult> scenarios = new List<RWScenarioProfileResult>();

            RWEvaluator[] evaluators = new RWEvaluator[]
                                           {
                                               new DikeMaterialRWEvaluator(), 
                                               new HydraulicShortcutRWEvaluator(), 
                                               new UpliftRWEvaluator()
                                           };

            Enum[] choices = new Enum[evaluators.Length + 1];
            foreach (LoadSituation loadSituation in Enum.GetValues(typeof(LoadSituation)))
            {
                choices[0] = loadSituation;

                bool resultsInScenarios = true;

                for (int i = 1; i < evaluators.Length + 1; i++)
                {
                    choices[i] = evaluators[i - 1].Evaluate(location, new SoilGeometry(soilGeometryProbability.SoilProfile, null), choices.ToArray());

                    //Is het een doorlopende tak? Zo niet break af (afbreken als geen veenkade en droge belastingsituatie
                    resultsInScenarios = WillChoicesResultInScenarios(choices.ToArray());

                    if (!resultsInScenarios)
                    {
                        break;
                    }
                }

                if (resultsInScenarios) 
                {
                    if (choices[0].Equals(LoadSituation.Wet))
                    {
                        if (choices[1].Equals(DikeDrySensitivity.None))
                        {
                            if (choices[2].Equals(HydraulicShortcutType.NoHydraulicShortcut))
                            {
                                if (choices[3].Equals(UpliftType.NoUplift))
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario08, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                }
                                else // UpliftType.Uplift 
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario04, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.UpliftVan));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario08, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario10, choices, location, soilGeometryProbability, FailureMechanismSystemType.Piping, PipingModelType));
                                }
                            }
                            else // HydraulicShortcutType.HydraulicShortcut
                            {
                                if (choices[3].Equals(UpliftType.NoUplift))
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario07, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                }
                                else // UpliftType.Uplift 
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario03, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.UpliftVan));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario07, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario10, choices, location, soilGeometryProbability, FailureMechanismSystemType.Piping, PipingModelType));
                                }
                            }
                        }
                        else // DikeDrySensitivity.Dry
                        {
                            if (choices[2].Equals(HydraulicShortcutType.NoHydraulicShortcut))
                            {
                                if (choices[3].Equals(UpliftType.NoUplift))
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario08, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                }
                                else // UpliftType.Uplift 
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario04, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.UpliftVan));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario08, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario10, choices, location, soilGeometryProbability, FailureMechanismSystemType.Piping, PipingModelType));
                                }
                            }
                            else // HydraulicShortcutType.HydraulicShortcut
                            {
                                if (choices[3].Equals(UpliftType.NoUplift))
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario07, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                }
                                else // UpliftType.Uplift 
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario03, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.UpliftVan));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario07, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario10, choices, location, soilGeometryProbability, FailureMechanismSystemType.Piping, PipingModelType));
                                }
                            }
                        }
                    }
                    else // LoadSituation.Dry
                    {
                        if (choices[1].Equals(DikeDrySensitivity.Dry))
                        {
                            if (choices[2].Equals(HydraulicShortcutType.NoHydraulicShortcut))
                            {
                                if (choices[3].Equals(UpliftType.NoUplift))
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario06, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                }
                                else // UpliftType.Uplift 
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario02, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.UpliftVan));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario06, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario09, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.HorizontalBalance));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario11, choices, location, soilGeometryProbability, FailureMechanismSystemType.Piping, PipingModelType));
                                }
                            }
                            else // HydraulicShortcutType.HydraulicShortcut
                            {
                                if (choices[3].Equals(UpliftType.NoUplift))
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario05, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                }
                                else // UpliftType.Uplift 
                                {
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario01, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.UpliftVan));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario05, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.Bishop));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario09, choices, location, soilGeometryProbability, FailureMechanismSystemType.StabilityInside, MStabModelType.HorizontalBalance));
                                    scenarios.Add(this.GetScenario(ScenarioType.Scenario11, choices, location, soilGeometryProbability, FailureMechanismSystemType.Piping, PipingModelType));
                                }
                            }
                        }
                    }
                }
            }

            return scenarios.ToArray();
        }

        private RWScenarioProfileResult GetScenario(ScenarioType scenarioType, Enum[] choices, Location location,
            SoilGeometryProbability soilGeometryProbability, FailureMechanismSystemType failureMechanismType, MStabModelType modelOption)
        {
            RWScenarioProfileResult scenario = new RWScenarioProfileResult(location, soilGeometryProbability); 
            scenario.FailureMechanismType = failureMechanismType;
            scenario.MstabModelOption = modelOption;
            scenario.PipingModelOption = PipingModelType;
            scenario.LoadSituation = (LoadSituation)choices[0];
            scenario.DikeDrySensitivity = (DikeDrySensitivity)choices[1];
            scenario.HydraulicShortcutType = (HydraulicShortcutType)choices[2];
            scenario.UpliftType = (UpliftType)choices[3];

            scenario.ScenarioType = scenarioType;

            //if (scenario.IsDry)
            //{
            //    // Soils are not transferred to the computational core at the moment, therefore this code has no meaning at the moment
            //    // You could say the same about soils in general in the DAM C# code
            //    SoilProfile drySoilProfile = soilProfile.Clone();
            //    foreach (Layer layer in drySoilProfile.Layers)
            //    {
            //        layer.Soil.AbovePhreaticLevel = layer.Soil.DryUnitWeight;
            //    }
            //}

            return scenario;
        }

        private RWScenarioProfileResult GetScenario(ScenarioType scenarioType, Enum[] choices, Location location,
            SoilGeometryProbability soilGeometryProbability, FailureMechanismSystemType failureMechanismType, PipingModelType pipingModelType)
        {
            RWScenarioProfileResult scenario = new RWScenarioProfileResult(location, soilGeometryProbability);
            scenario.FailureMechanismType = failureMechanismType;
            scenario.MstabModelOption = MStabModelType.Bishop;
            scenario.PipingModelOption = pipingModelType;

            scenario.LoadSituation = (LoadSituation)choices[0];
            scenario.DikeDrySensitivity = (DikeDrySensitivity)choices[1];
            scenario.HydraulicShortcutType = (HydraulicShortcutType)choices[2];
            scenario.UpliftType = (UpliftType)choices[3];

            scenario.ScenarioType = scenarioType;

            //if (scenario.IsDry)
            //{
            //    // Soils are not transferred to the computational core at the moment, therefore this code has no meaning at the moment
            //    // You could say the same about soils in general in the DAM C# code
            //    SoilProfile drySoilProfile = soilProfile.Clone();
            //    foreach (Layer layer in drySoilProfile.Layers)
            //    {
            //        layer.Soil.AbovePhreaticLevel = layer.Soil.DryUnitWeight;
            //    }
            //}

            return scenario;
        }
    }
}