-
Notifications
You must be signed in to change notification settings - Fork 4k
/
SyntaxTokenList.Enumerator.cs
156 lines (135 loc) · 4.94 KB
/
SyntaxTokenList.Enumerator.cs
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
// Copyright (c) Microsoft. All Rights Reserved. Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Runtime.InteropServices;
namespace Microsoft.CodeAnalysis
{
/// <summary>
/// Represents a read-only list of <see cref="SyntaxToken"/>s.
/// </summary>
public partial struct SyntaxTokenList
{
/// <summary>
/// A structure for enumerating a <see cref="SyntaxTokenList"/>
/// </summary>
[SuppressMessage("Performance", "CA1067", Justification = "Equality not actually implemented")]
[StructLayout(LayoutKind.Auto)]
public struct Enumerator
{
// This enumerator allows us to enumerate through two types of lists.
// either it looks like:
//
// Parent
// |
// List
// | \
// c1 c2
//
// or
//
// Parent
// |
// c1
//
// I.e. in the single child case, we optimize and store the child
// directly (without any list parent).
//
// Enumerating over the single child case is simple. We just
// return it and we're done.
//
// In the multi child case, things are a bit more difficult. We need
// to return the children in order, while also keeping their offset
// correct.
private readonly SyntaxNode _parent;
private readonly GreenNode _singleNodeOrList;
private readonly int _baseIndex;
private readonly int _count;
private int _index;
private GreenNode _current;
private int _position;
internal Enumerator(in SyntaxTokenList list)
{
_parent = list._parent;
_singleNodeOrList = list.Node;
_baseIndex = list._index;
_count = list.Count;
_index = -1;
_current = null;
_position = list.Position;
}
/// <summary>
/// Advances the enumerator to the next token in the collection.
/// </summary>
/// <returns>true if the enumerator was successfully advanced to the next element; false if the enumerator
/// has passed the end of the collection.</returns>
public bool MoveNext()
{
if (_count == 0 || _count <= _index + 1)
{
// invalidate iterator
_current = null;
return false;
}
_index++;
// Add the length of the previous node to the offset so that
// the next node's offset is reported correctly.
if (_current != null)
{
_position += _current.FullWidth;
}
_current = GetGreenNodeAt(_singleNodeOrList, _index);
System.Diagnostics.Debug.Assert(_current != null);
return true;
}
/// <summary>
/// Gets the current element in the collection.
/// </summary>
public SyntaxToken Current
{
get
{
if (_current == null)
{
throw new InvalidOperationException();
}
// In both the list and the single node case we want to
// return the original root parent as the parent of this
// token.
return new SyntaxToken(_parent, _current, _position, _baseIndex + _index);
}
}
public override bool Equals(object obj)
{
throw new NotSupportedException();
}
public override int GetHashCode()
{
throw new NotSupportedException();
}
}
private class EnumeratorImpl : IEnumerator<SyntaxToken>
{
private Enumerator _enumerator;
// SyntaxTriviaList is a relatively big struct so is passed by ref
internal EnumeratorImpl(in SyntaxTokenList list)
{
_enumerator = new Enumerator(in list);
}
public SyntaxToken Current => _enumerator.Current;
object IEnumerator.Current => _enumerator.Current;
public bool MoveNext()
{
return _enumerator.MoveNext();
}
public void Reset()
{
throw new NotSupportedException();
}
public void Dispose()
{
}
}
}
}