// Copyright (C) Stichting Deltares 2024. 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.Collections.Generic;
using System.Linq;
using System.Xml.Linq;
using Deltares.DamEngine.Calculators.General;
using Deltares.DamEngine.Data.Geotechnics;

namespace Deltares.DamEngine.Calculators.KernelWrappers.Assemblers;

/// <summary>
/// 
/// </summary>    
public class Geometry2DSectionAssembler : DtoDocumentAssembler<Geometry2DSectionParameters, Geometry2DSectionParameters>
{
    /// <summary>
    /// Holds the xml element name
    /// </summary>
    public const string XmlElementNameGeometry2DSectionInput = "Geometry2DSectionInput";

    /// <summary>
    /// Holds the Geometry2DSectionParameters xml element and attribute names
    /// </summary>
    public const string XmlElementGeometry2DSectionInput = "Geometry2DSectionInput";

    public const string XmlAttributeSoilGeometry2DFilename = "SoilGeometry2DFilename";
    public const string XmlAttributeXCoordinateSection = "XCoordinateSection";

    /// <summary>
    /// 
    /// </summary>
    public Geometry2DSectionAssembler() :
        base(XMLAttributes.XmlElementName, XMLAttributes.XmlElementNamespace, XMLAttributes.XsdEmbeddedResourcePath) {}

    public override Geometry2DSectionParameters CreateOutputObject(XDocument dtoObject)
    {
        var geometry2DSectionParameters = new Geometry2DSectionParameters();
        // Get the dto
        XElement dto = (from x in dtoObject.Descendants()
                        where x.Name.LocalName == XmlElementGeometry2DSectionInput
                        select x).FirstOrDefault();
        if (dto != null)
        {
            geometry2DSectionParameters.SoilGeometry2DName = dto.AttributeAs<string>(XmlAttributeSoilGeometry2DFilename);
            geometry2DSectionParameters.XCoordinateSection = dto.AttributeAs<double>(XmlAttributeXCoordinateSection);
        }

        // Get the dto
        XElement profileDto = (from x in dtoObject.Descendants()
                               where x.Name.LocalName == XMLAttributes.XmlElementProfile
                               select x).FirstOrDefault();
        if (profileDto != null)
        {
            var soilProfile = new SoilProfile1D
            {
                Name = profileDto.AttributeAs<string>(XMLAttributes.XmlAttributeName),
                BottomLevel = profileDto.AttributeAs<double>(XMLAttributes.XmlAttributeBottomLevel)
            };
            // Layers
            XElement layerCollectionElement = (from element in profileDto.Descendants()
                                               where element.Name.LocalName == XMLAttributes.XmlElementLayers
                                               select element).Single();
            IEnumerable<XElement> layerElementCollection = from element in layerCollectionElement.Descendants()
                                                           where element.Name.LocalName == XMLAttributes.XmlElementLayer
                                                           select element;
            foreach (XElement layerElement in layerElementCollection)
            {
                var soilLayer = new SoilLayer1D
                {
                    Name = layerElement.AttributeAs<string>("ID"),
                    SoilName = layerElement.AttributeAs<string>(XMLAttributes.XmlAttributeSoilID),
                    TopLevel = layerElement.AttributeAs<double>(XMLAttributes.XmlAttributeTopLevel)
                };
                soilProfile.Layers.Add(soilLayer);
            }

            geometry2DSectionParameters.SoilProfile = soilProfile;
        }

        return geometry2DSectionParameters;
    }

    /// <summary>
    /// Assembles the GeometryCreationOptions element 
    /// </summary>
    /// <param name="geometry2DSectionParameters"></param>
    /// <param name="tnsb"></param>
    /// <returns></returns>
    private static XElement GetGeometry2DSectionParametersElement(Geometry2DSectionParameters geometry2DSectionParameters, XNamespace tnsb)
    {
        return new XElement(tnsb + XmlElementGeometry2DSectionInput,
                            new XAttribute(XmlAttributeSoilGeometry2DFilename, geometry2DSectionParameters.SoilGeometry2DName),
                            new XAttribute(XmlAttributeXCoordinateSection, geometry2DSectionParameters.XCoordinateSection)
        );
    }

    /// <summary>
    /// 
    /// </summary>
    /// <param name="profile"></param>
    /// <param name="tns"></param>
    /// <param name="tnsa"></param>
    /// <returns></returns>
    private static XElement GetProfileElement(SoilProfile1D profile, XNamespace tns, XNamespace tnsa)
    {
        var profileElement = new XElement(tns + XMLAttributes.XmlElementProfile);
        profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeName, profile.Name));
        // Next four attributes are fake but never used in DAM anyway so ok.
        profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeXCoordinate, 0));
        profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeYCoordinate, 0));
        profileElement.Add(new XAttribute(XMLAttributes.XmlAttributePhreaticLevel, 0));
        profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeHasPhreaticLevel, false));

        profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeBottomLevel, profile.BottomLevel));

        if (profile.Layers.Count > 0)
        {
            var layersElement = new XElement(tnsa + XMLAttributes.XmlElementLayers);
            GetProfileLayersElement(profile, tnsa, layersElement);
            profileElement.Add(layersElement);
        }

        // Bottom sand layer
        if (profile.BottomAquiferLayer != null)
        {
            profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeBottomSandLayerID, profile.BottomAquiferLayer.Name));
        }

        // In-between sand layer
        if (profile.InBetweenAquiferLayer != null)
        {
            profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeInBetweenSandLayerID, profile.InBetweenAquiferLayer.Name));
        }

        // Infiltration layer
        if (profile.InfiltrationLayer != null)
        {
            profileElement.Add(new XAttribute(XMLAttributes.XmlAttributeInfiltrationLayerID, profile.InfiltrationLayer.Name));
        }

        return profileElement;
    }

    // Assembles the layer elements for the profile
    private static void GetProfileLayersElement(SoilProfile1D profile, XNamespace tnsa, XElement layersElement)
    {
        foreach (SoilLayer1D layer in profile.Layers)
        {
            if (layer.Name == null)
            {
                layer.Name = profile.GetNewUniqueLayerName();
            }

            var layerElement = new XElement(tnsa + XMLAttributes.XmlElementLayer);
            layersElement.Add(layerElement);
            layerElement.Add(new XAttribute(XMLAttributes.XmlAttributeID, $"{layer.Name}"));
            layerElement.Add(new XAttribute(XMLAttributes.XmlAttributeSoilID, layer.Soil.Name));
            layerElement.Add(new XAttribute(XMLAttributes.XmlAttributeTopLevel, layer.TopLevel));
        }
    }
}