using System;
using System.Linq;

using Deltares.Dam.Data;
using Deltares.Dam.Data.DataPlugins.Configuration;
using Deltares.Dam.Data.Importers;
using Deltares.Geotechnics;
using Deltares.Geotechnics.IO;
using Deltares.Standard;
using Deltares.Standard.IO.Xml;

namespace Deltares.Dam.TestHelper
{
    public class ProjectLoader
    {
        private static DamProjectData LoadDike(string projectDataFolder)
        {
            var damProject = new DamProject{ DamProjectData = new DamProjectData() };
            var csvSource = new DataSource { DataSourceType = DataSourceType.CsvFiles, DataLocation = ""};
            var soilSource = new DataSource { DataSourceType = DataSourceType.MSoilBase, DataLocation = "./soilmaterials.mdb" };
            var container = new DataSourceContainer{ DataSourceList = {
                csvSource, soilSource
            }};

            damProject.Import(projectDataFolder, container, new[] { "Dike from CSV" }, DamType.Regional, DamProjectType.Design, null);

            return damProject.DamProjectData;
        }

        public static DamProjectData LoadProjectData(string dataFileOrPath)
        {
            if (dataFileOrPath.EndsWith(".damx"))
            {
                using (var damProject = new DamProject())
                {
                    return (damProject.OpenXMLProject(dataFileOrPath) as DamProjectData);
                }
            }

            return LoadProjectData(dataFileOrPath, CreateDefaultFailureMechanismeCalculationSpecification);
        }

        private static DamProjectData LoadProjectData(string projectDataFolder, Func<DamProjectData, string, DamFailureMechanismeCalculationSpecification> failureMechanismCalculationSpecificationFactoryMethod)
        {
            var damProjectData = LoadDike(projectDataFolder);
            var dike = damProjectData.WaterBoard.Dikes.First();
            var soilDatabaseName = dike.SoilDatabaseName;

            // Specify calculation
            var damCalculationSpecification = failureMechanismCalculationSpecificationFactoryMethod(damProjectData, soilDatabaseName);
            damProjectData.DamProjectCalculationSpecification.DamCalculationSpecifications.Add(damCalculationSpecification);

            return damProjectData;
        }

        public static DamFailureMechanismeCalculationSpecification CreateDefaultFailureMechanismeCalculationSpecification(
            DamProjectData damProjectData, string soilDatabaseName)
        {
            MStabParameters mstabParameters = FactoryForStabilityTests.CreateMStabParameters();
            mstabParameters.GridPosition = MStabGridPosition.Left;
            mstabParameters.SearchMethod = MStabSearchMethod.Grid;
            mstabParameters.SoilDatabaseName = soilDatabaseName;

            var damCalculationSpecification = new DamFailureMechanismeCalculationSpecification()
            {
                FailureMechanismSystemType = FailureMechanismSystemType.StabilityOutside,
                FailureMechanismeParamatersMStab = new FailureMechanismeParamatersMStab()
                {
                    MStabParameters = mstabParameters
                }
            };
            DamProjectCalculationSpecification.SelectedAnalysisType = AnalysisType.AdaptNWO;
            damProjectData.DamProjectCalculationSpecification.SelectedProbabilisticType = ProbabilisticType.Deterministic;
            return damCalculationSpecification;
        }
    }

}