// Copyright (C) Stichting Deltares 2023. All rights reserved.
// 
// This file is part of the application DAM - UI.
// 
// DAM - UI 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 NetTopologySuite.Geometries;
using NetTopologySuite.IO;

namespace Deltares.Maps;

/// <summary>
///     Reads data from ShapeFiles
/// </summary>
/// <remarks>
/// </remarks>
public class ShapeFileReader : FileReader<IFile, IFeature>, IFeatureReader
{
    public ShapeFileReader(IFile file)
        : base(file) {}

    public ShapeFileReader(string fileName)
        : base(new ShapeFileLocation(fileName)) {}

    /// <summary>
    ///     Gets or sets the value indicating geometry attribute or feature data should be read
    ///     If set to false the attribute data will be read form the .dbf file (default)
    /// </summary>
    public bool IgnoreAttributeData { get; set; }

    #region IFeatureReader Members

    public override IEnumerable<IFeature> Read()
    {
        using (var reader = new ShapefileDataReader(File.FullPath, GeometryFactory.Default))
        {
            var count = 0;

            while (reader.Read())
            {
                var feature = new NtsFeature
                {
                    Geometry = reader.Geometry
                };

                if (!IgnoreAttributeData)
                {
                    for (var i = 0; i < reader.DbaseHeader.NumFields; i++)
                    {
                        string colName = reader.DbaseHeader.Fields[i].Name;
                        object rowValue = reader.GetValue(i);
                        if (!feature.Attributes.Exists(colName))
                        {
                            feature.Attributes.AddAttribute(colName, rowValue);
                        }
                    }
                }

                count++;
                yield return feature;
            }

            if (reader.RecordCount != count)
            {
                string message = $"The shape file '{File.FullPath}' contains multiple records but only one (composite) shape. " +
                                 "This is not yet supported. The current implementation of the shape file reader only works " +
                                 "with shape files where each shape has exactly one database record.";
                throw new NotSupportedException(message);
            }
        }
    }

    #endregion
}