pythonnet/src/runtime/iterator.cs at refactor-qc-pythonnet5 · QuantConnect/pythonnet · GitHub

History

55 lines (50 loc) · 1.45 KB

Raw

using System;

using System.Collections;

namespace Python.Runtime

{

/// <summary>

/// Implements a generic Python iterator for IEnumerable objects and

/// managed array objects. This supports 'for i in object:' in Python.

/// </summary>

internal class Iterator : ExtensionType

{

private IEnumerator iter;

private Type elemType;

public Iterator(IEnumerator e, Type elemType)

{

iter = e;

this.elemType = elemType;

}

/// <summary>

/// Implements support for the Python iteration protocol.

/// </summary>

public static IntPtr tp_iternext(IntPtr ob)

{

var self = GetManagedObject(ob) as Iterator;

try

{

if (!self.iter.MoveNext())

{

Exceptions.SetError(Exceptions.StopIteration, Runtime.PyNone);

return IntPtr.Zero;

}

catch (Exception e)

{

if (e.InnerException != null)

{

e = e.InnerException;

}

Exceptions.SetError(e);

return IntPtr.Zero;

}

object item = self.iter.Current;

return Converter.ToPython(item, self.elemType);

}

public static IntPtr tp_iter(IntPtr ob)

{

Runtime.XIncref(ob);

return ob;

}

pythonnet/src/runtime/iterator.cs at refactor-qc-pythonnet5 · QuantConnect/pythonnet · GitHub

Uh oh!

Expand file tree

Breadcrumbs

iterator.cs

Latest commit

History

Breadcrumbs

iterator.cs

File metadata and controls