Mercurial > pub > ImplabNet
annotate Implab/Parallels/AsyncQueue.cs @ 205:8200ab154c8a v2
Added ResetState to RunnableComponent to reset in case of failure
Added StateChanged event to IRunnable
Renamed Promise.SUCCESS -> Promise.Success
Added Promise.FromException
Renamed Bundle -> PromiseAll in PromiseExtensions
| author | cin |
|---|---|
| date | Tue, 25 Oct 2016 17:40:33 +0300 |
| parents | 238e15580926 |
| children | 8d5de4eb9c2c |
| rev | line source |
|---|---|
| 119 | 1 using System.Threading; |
| 2 using System.Collections.Generic; | |
| 3 using System; | |
| 4 using System.Collections; | |
| 124 | 5 using System.Diagnostics; |
| 119 | 6 |
| 7 namespace Implab.Parallels { | |
| 8 public class AsyncQueue<T> : IEnumerable<T> { | |
| 9 class Chunk { | |
| 10 public Chunk next; | |
| 11 | |
| 12 int m_low; | |
| 13 int m_hi; | |
| 14 int m_alloc; | |
| 15 readonly int m_size; | |
| 16 readonly T[] m_data; | |
| 17 | |
| 18 public Chunk(int size) { | |
| 19 m_size = size; | |
| 20 m_data = new T[size]; | |
| 21 } | |
| 22 | |
| 23 public Chunk(int size, T value) { | |
| 24 m_size = size; | |
| 25 m_hi = 1; | |
| 26 m_alloc = 1; | |
| 27 m_data = new T[size]; | |
| 28 m_data[0] = value; | |
| 29 } | |
| 30 | |
| 121 | 31 public Chunk(int size, T[] data, int offset, int length, int alloc) { |
| 32 m_size = size; | |
| 33 m_hi = length; | |
| 34 m_alloc = alloc; | |
| 35 m_data = new T[size]; | |
| 36 Array.Copy(data, offset, m_data, 0, length); | |
| 37 } | |
| 38 | |
| 119 | 39 public int Low { |
| 40 get { return m_low; } | |
| 41 } | |
| 42 | |
| 43 public int Hi { | |
| 44 get { return m_hi; } | |
| 45 } | |
| 46 | |
| 127 | 47 public int Size { |
| 48 get { return m_size; } | |
| 49 } | |
| 50 | |
| 121 | 51 public bool TryEnqueue(T value, out bool extend) { |
| 120 | 52 var alloc = Interlocked.Increment(ref m_alloc) - 1; |
| 119 | 53 |
| 120 | 54 if (alloc >= m_size) { |
| 55 extend = alloc == m_size; | |
| 119 | 56 return false; |
| 57 } | |
| 120 | 58 |
| 59 extend = false; | |
| 119 | 60 m_data[alloc] = value; |
| 61 | |
| 62 while (alloc != Interlocked.CompareExchange(ref m_hi, alloc + 1, alloc)) { | |
| 63 // spin wait for commit | |
| 64 } | |
| 65 return true; | |
| 66 } | |
| 67 | |
| 124 | 68 /// <summary> |
| 69 /// Prevents from allocating new space in the chunk and waits for all write operations to complete | |
| 70 /// </summary> | |
| 71 public void Commit() { | |
| 72 var actual = Math.Min(Interlocked.Exchange(ref m_alloc, m_size + 1), m_size); | |
| 73 | |
| 74 while (m_hi != actual) | |
| 75 Thread.MemoryBarrier(); | |
| 76 } | |
| 77 | |
| 121 | 78 public bool TryDequeue(out T value, out bool recycle) { |
| 119 | 79 int low; |
| 80 do { | |
| 81 low = m_low; | |
| 82 if (low >= m_hi) { | |
| 83 value = default(T); | |
| 84 recycle = (low == m_size); | |
| 85 return false; | |
| 86 } | |
| 87 } while(low != Interlocked.CompareExchange(ref m_low, low + 1, low)); | |
| 88 | |
| 89 recycle = (low == m_size - 1); | |
| 90 value = m_data[low]; | |
| 91 | |
| 92 return true; | |
| 93 } | |
| 94 | |
| 120 | 95 public bool TryEnqueueBatch(T[] batch, int offset, int length, out int enqueued, out bool extend) { |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
96 //int alloc; |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
97 //int allocSize; |
| 120 | 98 |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
99 var alloc = Interlocked.Add(ref m_alloc, length) - length; |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
100 if (alloc > m_size) { |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
101 // the chunk is full and someone already |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
102 // creating the new one |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
103 enqueued = 0; // nothing was added |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
104 extend = false; // the caller shouldn't try to extend the queue |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
105 return false; // nothing was added |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
106 } |
| 120 | 107 |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
108 enqueued = Math.Min(m_size - alloc, length); |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
109 extend = length > enqueued; |
| 120 | 110 |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
111 if (enqueued == 0) |
| 120 | 112 return false; |
| 113 | |
| 114 | |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
115 Array.Copy(batch, offset, m_data, alloc, enqueued); |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
116 |
|
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
117 while (alloc != Interlocked.CompareExchange(ref m_hi, alloc + enqueued, alloc)) { |
| 120 | 118 // spin wait for commit |
| 119 } | |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
120 |
| 120 | 121 return true; |
| 122 } | |
| 123 | |
| 121 | 124 public bool TryDequeueBatch(T[] buffer, int offset, int length,out int dequeued, out bool recycle) { |
| 125 int low, hi, batchSize; | |
| 126 | |
| 127 do { | |
| 128 low = m_low; | |
| 129 hi = m_hi; | |
| 130 if (low >= hi) { | |
| 131 dequeued = 0; | |
| 132 recycle = (low == m_size); // recycling could be restarted and we need to signal again | |
| 133 return false; | |
| 134 } | |
| 135 batchSize = Math.Min(hi - low, length); | |
| 136 } while(low != Interlocked.CompareExchange(ref m_low, low + batchSize, low)); | |
| 137 | |
| 138 recycle = (low == m_size - batchSize); | |
| 139 dequeued = batchSize; | |
| 140 | |
| 141 Array.Copy(m_data, low, buffer, offset, batchSize); | |
| 142 | |
| 143 return true; | |
| 144 } | |
| 145 | |
| 119 | 146 public T GetAt(int pos) { |
| 147 return m_data[pos]; | |
| 148 } | |
| 149 } | |
| 150 | |
| 151 public const int DEFAULT_CHUNK_SIZE = 32; | |
| 121 | 152 public const int MAX_CHUNK_SIZE = 262144; |
| 119 | 153 |
| 154 Chunk m_first; | |
| 155 Chunk m_last; | |
| 156 | |
| 121 | 157 /// <summary> |
| 158 /// Adds the specified value to the queue. | |
| 159 /// </summary> | |
| 160 /// <param name="value">Tha value which will be added to the queue.</param> | |
| 137 | 161 public virtual void Enqueue(T value) { |
| 119 | 162 var last = m_last; |
| 163 // spin wait to the new chunk | |
| 164 bool extend = true; | |
| 121 | 165 while (last == null || !last.TryEnqueue(value, out extend)) { |
| 119 | 166 // try to extend queue |
| 167 if (extend || last == null) { | |
| 125 | 168 var chunk = new Chunk(DEFAULT_CHUNK_SIZE, value); |
| 119 | 169 if (EnqueueChunk(last, chunk)) |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
170 break; // success! exit! |
| 119 | 171 last = m_last; |
| 172 } else { | |
| 121 | 173 while (last == m_last) { |
| 119 | 174 Thread.MemoryBarrier(); |
| 175 } | |
| 121 | 176 last = m_last; |
| 119 | 177 } |
| 178 } | |
| 179 } | |
| 180 | |
| 121 | 181 /// <summary> |
| 182 /// Adds the specified data to the queue. | |
| 183 /// </summary> | |
| 184 /// <param name="data">The buffer which contains the data to be enqueued.</param> | |
| 185 /// <param name="offset">The offset of the data in the buffer.</param> | |
| 186 /// <param name="length">The size of the data to read from the buffer.</param> | |
| 137 | 187 public virtual void EnqueueRange(T[] data, int offset, int length) { |
| 121 | 188 if (data == null) |
| 189 throw new ArgumentNullException("data"); | |
|
130
671f60cd0250
fixed Resove method bug when calling it on already cancelled promise
cin
parents:
129
diff
changeset
|
190 if (length == 0) |
|
671f60cd0250
fixed Resove method bug when calling it on already cancelled promise
cin
parents:
129
diff
changeset
|
191 return; |
| 121 | 192 if (offset < 0) |
| 193 throw new ArgumentOutOfRangeException("offset"); | |
| 194 if (length < 1 || offset + length > data.Length) | |
| 195 throw new ArgumentOutOfRangeException("length"); | |
| 196 | |
| 197 var last = m_last; | |
| 198 | |
| 199 bool extend; | |
| 200 int enqueued; | |
| 201 | |
| 202 while (length > 0) { | |
| 203 extend = true; | |
| 204 if (last != null && last.TryEnqueueBatch(data, offset, length, out enqueued, out extend)) { | |
| 205 length -= enqueued; | |
| 206 offset += enqueued; | |
| 207 } | |
| 208 | |
| 209 if (extend) { | |
| 210 // there was no enough space in the chunk | |
| 211 // or there was no chunks in the queue | |
| 212 | |
| 213 while (length > 0) { | |
| 214 | |
| 215 var size = Math.Min(length, MAX_CHUNK_SIZE); | |
| 216 | |
| 217 var chunk = new Chunk( | |
| 125 | 218 Math.Max(size, DEFAULT_CHUNK_SIZE), |
| 121 | 219 data, |
| 220 offset, | |
| 221 size, | |
| 222 length // length >= size | |
| 223 ); | |
| 224 | |
| 225 if (!EnqueueChunk(last, chunk)) { | |
| 226 // looks like the queue has been updated then proceed from the beginning | |
| 227 last = m_last; | |
| 228 break; | |
| 229 } | |
| 230 | |
| 231 // we have successfully added the new chunk | |
| 232 last = chunk; | |
| 233 length -= size; | |
| 234 offset += size; | |
| 235 } | |
| 236 } else { | |
| 237 // we don't need to extend the queue, if we successfully enqueued data | |
| 238 if (length == 0) | |
| 239 break; | |
| 240 | |
| 241 // if we need to wait while someone is extending the queue | |
| 242 // spinwait | |
| 243 while (last == m_last) { | |
| 244 Thread.MemoryBarrier(); | |
| 245 } | |
| 246 | |
| 247 last = m_last; | |
| 248 } | |
| 249 } | |
| 250 } | |
| 251 | |
| 252 /// <summary> | |
| 253 /// Tries to retrieve the first element from the queue. | |
| 254 /// </summary> | |
| 255 /// <returns><c>true</c>, if element is dequeued, <c>false</c> otherwise.</returns> | |
| 256 /// <param name="value">The value of the dequeued element.</param> | |
| 119 | 257 public bool TryDequeue(out T value) { |
| 258 var chunk = m_first; | |
| 259 bool recycle; | |
| 260 while (chunk != null) { | |
| 261 | |
| 262 var result = chunk.TryDequeue(out value, out recycle); | |
| 263 | |
| 264 if (recycle) // this chunk is waste | |
| 265 RecycleFirstChunk(chunk); | |
| 266 else | |
| 267 return result; // this chunk is usable and returned actual result | |
| 268 | |
| 269 if (result) // this chunk is waste but the true result is always actual | |
| 270 return true; | |
| 271 | |
| 272 // try again | |
| 273 chunk = m_first; | |
| 274 } | |
| 275 | |
| 276 // the queue is empty | |
| 277 value = default(T); | |
| 278 return false; | |
| 279 } | |
| 280 | |
| 121 | 281 /// <summary> |
| 282 /// Tries to dequeue the specified amount of data from the queue. | |
| 283 /// </summary> | |
| 284 /// <returns><c>true</c>, if data was deuqueued, <c>false</c> otherwise.</returns> | |
| 285 /// <param name="buffer">The buffer to which the data will be written.</param> | |
| 286 /// <param name="offset">The offset in the buffer at which the data will be written.</param> | |
| 287 /// <param name="length">The maximum amount of data to be retrieved.</param> | |
| 288 /// <param name="dequeued">The actual amout of the retrieved data.</param> | |
| 289 public bool TryDequeueRange(T[] buffer, int offset, int length, out int dequeued) { | |
| 290 if (buffer == null) | |
| 291 throw new ArgumentNullException("buffer"); | |
| 292 if (offset < 0) | |
| 293 throw new ArgumentOutOfRangeException("offset"); | |
| 294 if (length < 1 || offset + length > buffer.Length) | |
| 295 throw new ArgumentOutOfRangeException("length"); | |
| 296 | |
| 297 var chunk = m_first; | |
| 298 bool recycle; | |
| 299 dequeued = 0; | |
| 300 while (chunk != null) { | |
| 301 | |
| 302 int actual; | |
| 303 if (chunk.TryDequeueBatch(buffer, offset, length, out actual, out recycle)) { | |
| 304 offset += actual; | |
| 305 length -= actual; | |
| 306 dequeued += actual; | |
| 307 } | |
| 308 | |
| 309 if (recycle) // this chunk is waste | |
| 310 RecycleFirstChunk(chunk); | |
| 311 else if (actual == 0) | |
| 312 break; // the chunk is usable, but doesn't contain any data (it's the last chunk in the queue) | |
| 313 | |
| 314 if (length == 0) | |
| 315 return true; | |
| 316 | |
| 317 // we still may dequeue something | |
| 318 // try again | |
| 319 chunk = m_first; | |
| 320 } | |
| 321 | |
| 322 return dequeued != 0; | |
| 323 } | |
| 324 | |
| 325 /// <summary> | |
| 326 /// Tries to dequeue all remaining data in the first chunk. | |
| 327 /// </summary> | |
| 328 /// <returns><c>true</c>, if data was dequeued, <c>false</c> otherwise.</returns> | |
|
122
0c8685c8b56b
minor fixes and improvements of AsyncQueue, additional tests
cin
parents:
121
diff
changeset
|
329 /// <param name="buffer">The buffer to which the data will be written.</param> |
| 121 | 330 /// <param name="offset">The offset in the buffer at which the data will be written.</param> |
| 331 /// <param name="length">Tha maximum amount of the data to be dequeued.</param> | |
| 332 /// <param name="dequeued">The actual amount of the dequeued data.</param> | |
| 333 public bool TryDequeueChunk(T[] buffer, int offset, int length, out int dequeued) { | |
| 334 if (buffer == null) | |
| 335 throw new ArgumentNullException("buffer"); | |
| 336 if (offset < 0) | |
| 337 throw new ArgumentOutOfRangeException("offset"); | |
| 338 if (length < 1 || offset + length > buffer.Length) | |
| 339 throw new ArgumentOutOfRangeException("length"); | |
| 340 | |
| 341 var chunk = m_first; | |
| 342 bool recycle; | |
| 343 dequeued = 0; | |
| 344 | |
| 345 while (chunk != null) { | |
| 346 | |
| 347 int actual; | |
| 348 if (chunk.TryDequeueBatch(buffer, offset, length, out actual, out recycle)) { | |
| 349 dequeued = actual; | |
| 350 } | |
| 351 | |
| 352 if (recycle) // this chunk is waste | |
| 353 RecycleFirstChunk(chunk); | |
| 354 | |
| 355 // if we have dequeued any data, then return | |
| 356 if (dequeued != 0) | |
| 357 return true; | |
| 358 | |
| 359 // we still may dequeue something | |
| 360 // try again | |
| 361 chunk = m_first; | |
| 362 } | |
| 363 | |
| 364 return false; | |
| 365 } | |
| 366 | |
| 119 | 367 bool EnqueueChunk(Chunk last, Chunk chunk) { |
| 368 if (Interlocked.CompareExchange(ref m_last, chunk, last) != last) | |
| 369 return false; | |
| 370 | |
| 371 if (last != null) | |
| 372 last.next = chunk; | |
| 124 | 373 else { |
| 119 | 374 m_first = chunk; |
| 124 | 375 } |
| 119 | 376 return true; |
| 377 } | |
| 378 | |
| 379 void RecycleFirstChunk(Chunk first) { | |
| 380 var next = first.next; | |
| 381 | |
| 124 | 382 if (first != Interlocked.CompareExchange(ref m_first, next, first)) |
| 383 return; | |
| 384 | |
| 119 | 385 if (next == null) { |
| 124 | 386 |
| 119 | 387 if (first != Interlocked.CompareExchange(ref m_last, null, first)) { |
| 124 | 388 /*while (first.next == null) |
| 389 Thread.MemoryBarrier();*/ | |
| 390 | |
| 119 | 391 // race |
| 124 | 392 // someone already updated the tail, restore the pointer to the queue head |
| 393 m_first = first; | |
| 119 | 394 } |
| 395 // the tail is updated | |
| 396 } | |
| 397 | |
| 398 // we need to update the head | |
| 124 | 399 //Interlocked.CompareExchange(ref m_first, next, first); |
| 119 | 400 // if the head is already updated then give up |
| 124 | 401 //return; |
| 119 | 402 |
| 403 } | |
| 404 | |
| 123 | 405 public void Clear() { |
| 406 // start the new queue | |
| 125 | 407 var chunk = new Chunk(DEFAULT_CHUNK_SIZE); |
| 124 | 408 |
| 409 do { | |
| 410 Thread.MemoryBarrier(); | |
| 411 var first = m_first; | |
| 412 var last = m_last; | |
| 413 | |
| 414 if (last == null) // nothing to clear | |
| 415 return; | |
| 123 | 416 |
| 124 | 417 if (first == null || (first.next == null && first != last)) // inconcistency |
| 418 continue; | |
| 419 | |
| 420 // here we will create inconsistency which will force others to spin | |
| 421 // and prevent from fetching. chunk.next = null | |
| 422 if (first != Interlocked.CompareExchange(ref m_first, chunk, first)) | |
| 423 continue;// inconsistent | |
| 424 | |
| 425 m_last = chunk; | |
| 426 | |
| 427 return; | |
| 428 | |
| 429 } while(true); | |
| 123 | 430 } |
| 431 | |
| 432 public T[] Drain() { | |
| 433 // start the new queue | |
| 125 | 434 var chunk = new Chunk(DEFAULT_CHUNK_SIZE); |
| 123 | 435 |
| 436 do { | |
| 124 | 437 Thread.MemoryBarrier(); |
| 438 var first = m_first; | |
| 439 var last = m_last; | |
| 440 | |
| 441 if (last == null) | |
| 442 return new T[0]; | |
| 443 | |
| 444 if (first == null || (first.next == null && first != last)) | |
| 445 continue; | |
| 123 | 446 |
| 124 | 447 // here we will create inconsistency which will force others to spin |
| 448 // and prevent from fetching. chunk.next = null | |
| 449 if (first != Interlocked.CompareExchange(ref m_first, chunk, first)) | |
| 450 continue;// inconsistent | |
| 123 | 451 |
| 124 | 452 last = Interlocked.Exchange(ref m_last, chunk); |
| 453 | |
| 454 return ReadChunks(first, last); | |
| 455 | |
| 456 } while(true); | |
| 123 | 457 } |
| 458 | |
|
128
6241bff0cd64
Added Signal class a lightweight alternative to ManualResetEvent
cin
parents:
127
diff
changeset
|
459 static T[] ReadChunks(Chunk chunk, object last) { |
| 123 | 460 var result = new List<T>(); |
| 125 | 461 var buffer = new T[DEFAULT_CHUNK_SIZE]; |
| 123 | 462 int actual; |
| 463 bool recycle; | |
| 464 while (chunk != null) { | |
| 124 | 465 // ensure all write operations on the chunk are complete |
| 466 chunk.Commit(); | |
| 467 | |
| 123 | 468 // we need to read the chunk using this way |
| 469 // since some client still may completing the dequeue | |
| 470 // operation, such clients most likely won't get results | |
| 471 while (chunk.TryDequeueBatch(buffer, 0, buffer.Length, out actual, out recycle)) | |
| 472 result.AddRange(new ArraySegmentCollection(buffer, 0, actual)); | |
| 473 | |
| 124 | 474 if (chunk == last) { |
| 475 chunk = null; | |
| 476 } else { | |
| 477 while (chunk.next == null) | |
| 478 Thread.MemoryBarrier(); | |
| 479 chunk = chunk.next; | |
| 480 } | |
| 123 | 481 } |
| 482 | |
| 483 return result.ToArray(); | |
| 484 } | |
| 485 | |
| 486 struct ArraySegmentCollection : ICollection<T> { | |
| 487 readonly T[] m_data; | |
| 488 readonly int m_offset; | |
| 489 readonly int m_length; | |
| 490 | |
| 491 public ArraySegmentCollection(T[] data, int offset, int length) { | |
| 492 m_data = data; | |
| 493 m_offset = offset; | |
| 494 m_length = length; | |
| 495 } | |
| 496 | |
| 497 #region ICollection implementation | |
| 498 | |
| 499 public void Add(T item) { | |
| 129 | 500 throw new NotSupportedException(); |
| 123 | 501 } |
| 502 | |
| 503 public void Clear() { | |
| 129 | 504 throw new NotSupportedException(); |
| 123 | 505 } |
| 506 | |
| 507 public bool Contains(T item) { | |
| 508 return false; | |
| 509 } | |
| 510 | |
| 511 public void CopyTo(T[] array, int arrayIndex) { | |
| 512 Array.Copy(m_data,m_offset,array,arrayIndex, m_length); | |
| 513 } | |
| 514 | |
| 515 public bool Remove(T item) { | |
| 129 | 516 throw new NotSupportedException(); |
| 123 | 517 } |
| 518 | |
| 519 public int Count { | |
| 520 get { | |
| 521 return m_length; | |
| 522 } | |
| 523 } | |
| 524 | |
| 525 public bool IsReadOnly { | |
| 526 get { | |
| 527 return true; | |
| 528 } | |
| 529 } | |
| 530 | |
| 531 #endregion | |
| 532 | |
| 533 #region IEnumerable implementation | |
| 534 | |
| 535 public IEnumerator<T> GetEnumerator() { | |
| 536 for (int i = m_offset; i < m_length + m_offset; i++) | |
| 537 yield return m_data[i]; | |
| 538 } | |
| 539 | |
| 540 #endregion | |
| 541 | |
| 542 #region IEnumerable implementation | |
| 543 | |
| 544 IEnumerator IEnumerable.GetEnumerator() { | |
| 545 return GetEnumerator(); | |
| 546 } | |
| 547 | |
| 548 #endregion | |
| 549 } | |
| 550 | |
| 119 | 551 #region IEnumerable implementation |
| 552 | |
| 553 class Enumerator : IEnumerator<T> { | |
| 554 Chunk m_current; | |
| 555 int m_pos = -1; | |
| 556 | |
| 557 public Enumerator(Chunk fisrt) { | |
| 558 m_current = fisrt; | |
| 559 } | |
| 560 | |
| 561 #region IEnumerator implementation | |
| 562 | |
| 563 public bool MoveNext() { | |
| 564 if (m_current == null) | |
| 565 return false; | |
| 566 | |
| 567 if (m_pos == -1) | |
| 568 m_pos = m_current.Low; | |
| 569 else | |
| 570 m_pos++; | |
| 127 | 571 |
| 119 | 572 if (m_pos == m_current.Hi) { |
| 127 | 573 |
| 574 m_current = m_pos == m_current.Size ? m_current.next : null; | |
| 575 | |
| 119 | 576 m_pos = 0; |
| 127 | 577 |
| 578 if (m_current == null) | |
| 579 return false; | |
| 119 | 580 } |
| 581 | |
| 582 return true; | |
| 583 } | |
| 584 | |
| 585 public void Reset() { | |
| 586 throw new NotSupportedException(); | |
| 587 } | |
| 588 | |
| 589 object IEnumerator.Current { | |
| 590 get { | |
| 591 return Current; | |
| 592 } | |
| 593 } | |
| 594 | |
| 595 #endregion | |
| 596 | |
| 597 #region IDisposable implementation | |
| 598 | |
| 599 public void Dispose() { | |
| 600 } | |
| 601 | |
| 602 #endregion | |
| 603 | |
| 604 #region IEnumerator implementation | |
| 605 | |
| 606 public T Current { | |
| 607 get { | |
| 608 if (m_pos == -1 || m_current == null) | |
| 609 throw new InvalidOperationException(); | |
| 610 return m_current.GetAt(m_pos); | |
| 611 } | |
| 612 } | |
| 613 | |
| 614 #endregion | |
| 615 } | |
| 616 | |
| 617 public IEnumerator<T> GetEnumerator() { | |
| 618 return new Enumerator(m_first); | |
| 619 } | |
| 620 | |
| 621 #endregion | |
| 622 | |
| 623 #region IEnumerable implementation | |
| 624 | |
| 625 IEnumerator IEnumerable.GetEnumerator() { | |
| 626 return GetEnumerator(); | |
| 627 } | |
| 628 | |
| 629 #endregion | |
| 630 } | |
| 631 } |