// Copyright (C) Stichting Deltares 2019. 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 NUnit.Framework;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Deltares.DamEngine.Data.General;
using Deltares.DamEngine.Data.Geotechnics;

namespace Deltares.DamEngine.Data.Tests.General
{
    [TestFixture]
    public class SegmentTests
    {
        [Test]
        public void CanGetMostProbableProbableProfile1DForStability()
        {
            var segment = CreateSegment(FailureMechanismSystemType.StabilityInside, SoilProfileType.ProfileType1D);
            Assert.AreEqual("profile 1D 2", segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.StabilityInside).SoilProfile1D.Name);
            Assert.AreEqual("profile 1D 2", segment.GetMostProbableProfile1D(FailureMechanismSystemType.StabilityInside).Name);
            Assert.AreEqual(null, segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.StabilityInside));
            Assert.AreEqual(null, segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.Piping));
            Assert.AreEqual(null, segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.Piping));
            Assert.AreEqual(null, segment.GetMostProbableProfile1D(FailureMechanismSystemType.Piping));
        }

        [Test]
        public void CanGetMostProbableProbableProfile1DForPiping()
        {
            var segment = CreateSegment(FailureMechanismSystemType.Piping, SoilProfileType.ProfileType1D);
            Assert.AreEqual("profile 1D 2", segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.Piping).SoilProfile1D.Name);
            Assert.AreEqual("profile 1D 2", segment.GetMostProbableProfile1D(FailureMechanismSystemType.Piping).Name);
            Assert.AreEqual(null, segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.Piping));
            Assert.AreEqual(null, segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.StabilityInside));
            Assert.AreEqual(null, segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.StabilityInside));
            Assert.AreEqual(null, segment.GetMostProbableProfile1D(FailureMechanismSystemType.StabilityInside));
        }

        [Test]
        public void CanGetMostProbableProbableProfile2DStiFileNameForStability()
        {
            var segment = CreateSegment(FailureMechanismSystemType.StabilityInside, SoilProfileType.ProfileTypeStiFile);
            Assert.AreEqual("sti filename 2", segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.StabilityInside).FullStiFileName);
            Assert.AreEqual("sti filename 2", segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.StabilityInside));
            Assert.AreEqual(null, segment.GetMostProbableProfile1D(FailureMechanismSystemType.StabilityInside));
            Assert.AreEqual(null, segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.Piping));
            Assert.AreEqual(null, segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.Piping));
            Assert.AreEqual(null, segment.GetMostProbableProfile1D(FailureMechanismSystemType.Piping));
        }

        [Test]
        public void CanGetMostProbableProbableProfile2DStiFileNameForPiping()
        {
            var segment = CreateSegment(FailureMechanismSystemType.Piping, SoilProfileType.ProfileTypeStiFile);
            Assert.AreEqual("sti filename 2", segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.Piping).FullStiFileName);
            Assert.AreEqual("sti filename 2", segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.Piping));
            Assert.AreEqual(null, segment.GetMostProbableProfile1D(FailureMechanismSystemType.Piping));
            Assert.AreEqual(null, segment.GetMostProbableSoilGeometryProbability(FailureMechanismSystemType.StabilityInside));
            Assert.AreEqual(null, segment.GetMostProbableProfile2DStiFileName(FailureMechanismSystemType.StabilityInside));
            Assert.AreEqual(null, segment.GetMostProbableProfile1D(FailureMechanismSystemType.StabilityInside));
        }
        private static Segment CreateSegment(FailureMechanismSystemType failureMechanismSystemType, SoilProfileType soilProfileType)
        {
            var segment = new Segment();
            segment.SoilProfileProbabilities.Add(new SoilGeometryProbability()
            {
                Probability = 0.1,
                SegmentFailureMechanismType = failureMechanismSystemType,
                SoilProfileType = soilProfileType,
                FullStiFileName = "sti filename 1",
                SoilProfile1D = new SoilProfile1D()
                {
                    Name = "profile 1D 1"
                }
            });
            segment.SoilProfileProbabilities.Add(new SoilGeometryProbability()
            {
                Probability = 0.4,
                SegmentFailureMechanismType = failureMechanismSystemType,
                SoilProfileType = soilProfileType,
                FullStiFileName = "sti filename 2",
                SoilProfile1D = new SoilProfile1D()
                {
                    Name = "profile 1D 2"
                }
            });
            segment.SoilProfileProbabilities.Add(new SoilGeometryProbability()
            {
                Probability = 0.3,
                SegmentFailureMechanismType = failureMechanismSystemType,
                SoilProfileType = soilProfileType,
                FullStiFileName = "sti filename 3",
                SoilProfile1D = new SoilProfile1D()
                {
                    Name = "profile 1D 3"
                }
            });
            segment.SoilProfileProbabilities.Add(new SoilGeometryProbability()
            {
                Probability = 0.2,
                SegmentFailureMechanismType = failureMechanismSystemType,
                SoilProfileType = soilProfileType,
                FullStiFileName = "sti filename 4",
                SoilProfile1D = new SoilProfile1D()
                {
                    Name = "profile 1D 4"
                }
            });
            return segment;
        }
    }
}