-
Notifications
You must be signed in to change notification settings - Fork 226
Expand file tree
/
Copy pathrepeat.java
More file actions
156 lines (134 loc) · 4.05 KB
/
repeat.java
File metadata and controls
156 lines (134 loc) · 4.05 KB
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) 2012 Jython Developers */
package org.python.modules.itertools;
import org.python.core.ArgParser;
import org.python.core.Py;
import org.python.core.PyIterator;
import org.python.core.PyObject;
import org.python.core.PyString;
import org.python.core.PyTuple;
import org.python.core.PyType;
import org.python.core.Visitproc;
import org.python.expose.ExposedMethod;
import org.python.expose.ExposedNew;
import org.python.expose.ExposedType;
@ExposedType(name = "itertools.repeat", base = PyObject.class, doc = repeat.repeat_doc)
public class repeat extends PyIterator {
public static final PyType TYPE = PyType.fromClass(repeat.class);
private PyIterator iter;
private PyObject object;
private int counter;
public static final String repeat_doc =
"'repeat(element [,times]) -> create an iterator which returns the element\n" +
"for the specified number of times. If not specified, returns the element\n" +
"endlessly.";
public repeat() {
super();
}
public repeat(PyType subType) {
super(subType);
}
public repeat(PyObject object) {
super();
repeat___init__(object);
}
public repeat(PyObject object, int times) {
super();
repeat___init__(object, times);
}
@ExposedNew
@ExposedMethod
final void repeat___init__(final PyObject[] args, String[] kwds) {
ArgParser ap = new ArgParser("repeat", args, kwds, new String[] {"object", "times"}, 1);
PyObject object = ap.getPyObject(0);
if (args.length == 1) {
repeat___init__(object);
}
else {
int times = ap.getInt(1);
repeat___init__(object, times);
}
}
/**
* Creates an iterator that returns the same object the number of times given by
* <code>times</code>.
*/
private void repeat___init__(final PyObject object, final int times) {
this.object = object;
if (times < 0) {
counter = 0;
}
else {
counter = times;
}
iter = new PyIterator() {
public PyObject __iternext__() {
if (counter > 0) {
counter--;
return object;
}
return null;
}
};
}
/**
* Creates an iterator that returns the same object over and over again.
*/
private void repeat___init__(final PyObject object) {
this.object = object;
counter = -1;
iter = new PyIterator() {
public PyObject __iternext__() {
return object;
}
};
}
@ExposedMethod
final PyObject __copy__() {
return new repeat(object, counter);
}
@ExposedMethod
public int __len__() {
if (counter < 0) {
throw Py.TypeError("object of type 'itertools.repeat' has no len()");
}
return counter;
}
@ExposedMethod
public PyString __repr__() {
if (counter >= 0) {
return (PyString)(Py.newString("repeat(%r, %d)").
__mod__(new PyTuple(object, Py.newInteger(counter))));
}
else {
return (PyString)(Py.newString("repeat(%r)").
__mod__(new PyTuple(object)));
}
}
public PyObject __iternext__() {
return iter.__iternext__();
}
@ExposedMethod
@Override
public PyObject next() {
return doNext(__iternext__());
}
/* Traverseproc implementation */
@Override
public int traverse(Visitproc visit, Object arg) {
int retVal = super.traverse(visit, arg);
if (retVal != 0) {
return retVal;
}
if (object != null) {
retVal = visit.visit(object, arg);
if (retVal != 0) {
return retVal;
}
}
return iter != null ? visit.visit(iter, arg) : 0;
}
@Override
public boolean refersDirectlyTo(PyObject ob) {
return ob != null && (iter == ob || object == ob || super.refersDirectlyTo(ob));
}
}