public final class PGenerator extends PythonBuiltinObject implements PIterator {
public static final PythonBuiltinClass __class__ =
PythonContext.getBuiltinTypeFor(PGenerator.class);
protected final String name;
protected final RootCallTarget callTarget;
protected final FrameDescriptor frameDescriptor;
protected final Object[] arguments;
public static PGenerator create(
String name,
RootCallTarget callTarget,
FrameDescriptor frameDescriptor,
MaterializedFrame declarationFrame,
Object[] arguments,
int numOfActiveFlags,
int numOfGeneratorBlockNode,
int numOfGeneratorForNode) {
/** Setting up the persistent frame in {@link #arguments}. */
GeneratorControlData generatorArgs =
new GeneratorControlData(numOfActiveFlags, numOfGeneratorBlockNode, numOfGeneratorForNode);
MaterializedFrame generatorFrame =
Truffle.getRuntime().createMaterializedFrame(PArguments.create(), frameDescriptor);
PArguments.setDeclarationFrame(arguments, declarationFrame);
PArguments.setGeneratorFrame(arguments, generatorFrame);
PArguments.setControlData(arguments, generatorArgs);
return new PGenerator(name, callTarget, frameDescriptor, arguments);
}
public PGenerator(
String name, RootCallTarget callTarget, FrameDescriptor frameDescriptor, Object[] arguments) {
this.name = name;
this.callTarget = callTarget;
this.frameDescriptor = frameDescriptor;
this.arguments = arguments;
}
@Override
public PythonBuiltinClass __class__() {
return __class__;
}
public FrameDescriptor getFrameDescriptor() {
return frameDescriptor;
}
public RootCallTarget getCallTarget() {
return callTarget;
}
public Object[] getArguments() {
return arguments;
}
@Override
public Object __next__() throws StopIterationException {
return callTarget.call(arguments);
}
public Object send(Object value) throws StopIterationException {
PArguments.setSpecialArgument(arguments, value);
return callTarget.call(arguments);
}
@Override
public String toString() {
return "<generator object '" + name + "' at " + hashCode() + ">";
}
}
