using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Linq;
using ThinkGeo.MapSuite.Core;

namespace CSHowDoISamples.Getting_Started
{
    public static class FeatureLayersExtension
    {
        public static Collection<Feature> GetAllFeatures(this Collection<Layer> layers, ReturningColumnsType returningColumnsType)
        {
            return layers.OfType<FeatureLayer>().GetAllFeatures(returningColumnsType);
        }

        public static Collection<Feature> GetAllFeatures(this IEnumerable<FeatureLayer> featureLayers, ReturningColumnsType returningColumnsType)
        {
            Collection<Feature> features = new Collection<Feature>();
            foreach (FeatureLayer featureLayer in featureLayers)
            {
                featureLayer.Open();
                Collection<Feature> queriedFeatures = featureLayer.QueryTools.GetAllFeatures(returningColumnsType);
                foreach (Feature feature in queriedFeatures)
                {
                    features.Add(feature);
                }
                featureLayer.Close();
            }

            return features;
        }
    }
}