forked from ClearFoundry/ClearScript
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathEnumerableHelpers.cs
More file actions
120 lines (102 loc) · 3.26 KB
/
EnumerableHelpers.cs
File metadata and controls
120 lines (102 loc) · 3.26 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license.
using System;
using System.Collections;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Runtime.InteropServices.ComTypes;
using Microsoft.ClearScript.Util.COM;
namespace Microsoft.ClearScript.Util
{
// ReSharper disable once PartialTypeWithSinglePart
internal static partial class EnumerableHelpers
{
public static readonly HostType HostType = HostType.Wrap(typeof(EnumerableHelpers));
public static IList<T> ToIList<T>(this IEnumerable<T> source)
{
return (source as IList<T>) ?? source.ToList();
}
public static void ForEach<T>(this IEnumerable<T> source, Action<T> action)
{
foreach (var element in source)
{
action(element);
}
}
public static void ForEach<T>(this IEnumerable<T> source, Action<T, int> action)
{
var index = 0;
foreach (var element in source)
{
action(element, index++);
}
}
public static IEnumerable<T> Flatten<T>(this IEnumerable<T> source, Func<T, IEnumerable<T>> selector)
{
foreach (var element in source)
{
yield return element;
foreach (var descendant in selector(element).Flatten(selector))
{
yield return descendant;
}
}
}
public static IEnumerable<T> ToEnumerable<T>(this T element)
{
yield return element;
}
public static IEnumerable<string> ExcludeIndices(this IEnumerable<string> names)
{
foreach (var name in names)
{
if (!int.TryParse(name, NumberStyles.Integer, CultureInfo.InvariantCulture, out _))
{
yield return name;
}
}
}
public static IEnumerable<int> GetIndices(this IEnumerable<string> names)
{
foreach (var name in names)
{
if (int.TryParse(name, NumberStyles.Integer, CultureInfo.InvariantCulture, out var index))
{
yield return index;
}
}
}
public static IEnumerator<T> GetEnumerator<T>(IEnumerable<T> source)
{
return source.GetEnumerator();
}
public static IEnumerator GetEnumerator(IEnumerable source)
{
return source.GetEnumerator();
}
}
internal sealed class EnumeratorWrapper : IEnumerator
{
private readonly IEnumVARIANT enumVariant;
public EnumeratorWrapper(IEnumVARIANT enumVariant)
{
this.enumVariant = enumVariant;
}
public bool MoveNext()
{
var items = new object[1];
if (enumVariant.Next(1, items, IntPtr.Zero) == HResult.S_OK)
{
Current = items[0];
return true;
}
return false;
}
public void Reset()
{
enumVariant.Reset();
}
public object Current { get; private set; }
}
}