using System;
using System.Collections.Generic;
namespace Python.Runtime
{
///
/// Represents a standard Python iterator object. See the documentation at
/// PY2: https://docs.python.org/2/c-api/iterator.html
/// PY3: https://docs.python.org/3/c-api/iterator.html
/// for details.
///
public class PyIter : PyObject, IEnumerator
{
private PyObject _current;
///
/// PyIter Constructor
///
///
/// Creates a new PyIter from an existing iterator reference. Note
/// that the instance assumes ownership of the object reference.
/// The object reference is not checked for type-correctness.
///
public PyIter(IntPtr ptr) : base(ptr)
{
}
///
/// PyIter Constructor
///
///
/// Creates a Python iterator from an iterable. Like doing "iter(iterable)" in python.
///
public PyIter(PyObject iterable)
{
obj = Runtime.PyObject_GetIter(iterable.obj);
if (obj == IntPtr.Zero)
{
throw new PythonException();
}
}
protected override void Dispose(bool disposing)
{
if (null != _current)
{
_current.Dispose();
_current = null;
}
base.Dispose(disposing);
}
public bool MoveNext()
{
// dispose of the previous object, if there was one
if (null != _current)
{
_current.Dispose();
_current = null;
}
IntPtr next = Runtime.PyIter_Next(obj);
if (next == IntPtr.Zero)
{
return false;
}
_current = new PyObject(next);
return true;
}
public void Reset()
{
//Not supported in python.
}
public object Current
{
get { return _current; }
}
}
}