1. # this is actually part of the Array class
3. my role Shaped1Array does ShapedArray {
4. multi method AT-POS(::?CLASS:D: int \one) is raw {
5. nqp::ifnull(
6. nqp::atpos(
7. nqp::getattr(self,List,'$!reified'),
8. one),
9. AT-POS-CONTAINER(self,one)
10. )
11. }
12. multi method AT-POS(::?CLASS:D: Int:D \one) is raw {
13. nqp::ifnull(
14. nqp::atpos(
15. nqp::getattr(self,List,'$!reified'),
16. one),
17. AT-POS-CONTAINER(self,one)
18. )
19. }
20. sub AT-POS-CONTAINER(\array, int \one) is raw {
21. nqp::p6bindattrinvres(
22. (my $scalar := nqp::p6scalarfromdesc(
23. nqp::getattr(array,Array,'$!descriptor'))),
24. Scalar,
25. '$!whence',
26. -> { nqp::bindpos(
27. nqp::getattr(array,List,'$!reified'),
28. one, $scalar) }
29. )
30. }
32. multi method ASSIGN-POS(::?CLASS:D: int \one, \value) {
33. nqp::ifnull(
34. nqp::atpos(
35. nqp::getattr(self,List,'$!reified'),
36. one),
37. nqp::bindpos(
38. nqp::getattr(self,List,'$!reified'),
39. one,
40. nqp::p6scalarfromdesc(nqp::getattr(self,Array,'$!descriptor')))
41. ) = value
42. }
43. multi method ASSIGN-POS(::?CLASS:D: Int:D \one, \value) {
44. nqp::ifnull(
45. nqp::atpos(
46. nqp::getattr(self,List,'$!reified'),
47. one),
48. nqp::bindpos(
49. nqp::getattr(self,List,'$!reified'),
50. one,
51. nqp::p6scalarfromdesc(nqp::getattr(self,Array,'$!descriptor')))
52. ) = value
53. }
55. multi method EXISTS-POS(::?CLASS:D: int \one) {
56. nqp::p6bool(
57. nqp::islt_i(one,nqp::elems(nqp::getattr(self,List,'$!reified')))
58. && nqp::not_i(nqp::isnull(
59. nqp::atpos(nqp::getattr(self,List,'$!reified'),one)
60. ))
61. )
62. }
63. multi method EXISTS-POS(::?CLASS:D: Int:D \one) {
64. nqp::p6bool(
65. nqp::islt_i(one,nqp::elems(nqp::getattr(self,List,'$!reified')))
66. && nqp::not_i(nqp::isnull(
67. nqp::atpos(nqp::getattr(self,List,'$!reified'),one)
68. ))
69. )
70. }
72. multi method DELETE-POS(::?CLASS:D: int \one) is raw {
73. nqp::if(
74. nqp::isnull(my $value := nqp::atpos(
75. nqp::getattr(self,List,'$!reified'),
76. one)),
77. Nil,
78. nqp::stmts(
79. nqp::bindpos(
80. nqp::getattr(self,List,'$!reified'),
81. one, nqp::null),
82. $value
83. )
84. )
85. }
86. multi method DELETE-POS(::?CLASS:D: Int:D \one) is raw {
87. nqp::if(
88. nqp::isnull(my $value := nqp::atpos(
89. nqp::getattr(self,List,'$!reified'),
90. one)),
91. Nil,
92. nqp::stmts(
93. nqp::bindpos(
94. nqp::getattr(self,List,'$!reified'),
95. one, nqp::null),
96. $value
97. )
98. )
99. }
101. multi method BIND-POS(::?CLASS:D: int \one, \value) {
102. nqp::bindpos(
103. nqp::getattr(self,List,'$!reified'),
104. one, value
105. )
106. }
107. multi method BIND-POS(::?CLASS:D: Int:D \one, \value) {
108. nqp::bindpos(
109. nqp::getattr(self,List,'$!reified'),
110. one, value
111. )
112. }
114. multi method STORE(::?CLASS:D: ::?CLASS:D \from) {
115. nqp::stmts(
116. (my $to := nqp::getattr(self,List,'$!reified')),
117. (my $from := nqp::getattr(from,List,'$!reified')),
118. nqp::if(
119. nqp::iseq_i(
120. (my int $elems = nqp::elems($to)),nqp::elems($from)),
121. nqp::stmts(
122. (my $desc := nqp::getattr(self,Array,'$!descriptor')),
123. (my int $i = -1),
124. nqp::while(
125. nqp::islt_i(($i = nqp::add_i($i,1)),$elems),
126. # always create a new container in case the from list
127. # contains containers already existing in the to list
128. # e.g. after having done a .reverse or .rotate
129. nqp::bindpos($to,$i,nqp::p6scalarfromdesc($desc)) =
130. nqp::atpos($from,$i)
131. ),
132. self
133. ),
134. X::Assignment::ArrayShapeMismatch.new(
135. source-shape => from.shape,
136. target-shape => self.shape
137. ).throw
138. )
139. )
140. }
141. multi method STORE(::?CLASS:D: Iterable:D \in) {
142. nqp::stmts(
143. (my \list := nqp::getattr(self,List,'$!reified')),
144. (my \desc := nqp::getattr(self,Array,'$!descriptor')),
145. (my \iter := in.iterator),
146. (my int $elems = nqp::elems(list)),
147. (my int $i = -1),
148. nqp::until(
149. nqp::eqaddr((my $pulled := iter.pull-one),IterationEnd)
150. || nqp::iseq_i(($i = nqp::add_i($i,1)),$elems),
151. nqp::ifnull(
152. nqp::atpos(list,$i),
153. nqp::bindpos(list,$i,nqp::p6scalarfromdesc(desc))
154. ) = $pulled
155. ),
156. nqp::unless(
157. nqp::islt_i($i,$elems) || iter.is-lazy,
158. nqp::atpos(list,$i) # too many values on non-lazy iter, error
159. ),
160. self
161. )
162. }
163. multi method STORE(::?CLASS:D: Mu \item) {
164. nqp::stmts(
165. (nqp::ifnull(
166. nqp::atpos(nqp::getattr(self,List,'$!reified'),0),
167. nqp::bindpos(nqp::getattr(self,List,'$!reified'),0,
168. nqp::p6scalarfromdesc(
169. nqp::getattr(self,Array,'$!descriptor')))
170. ) = item),
171. self
172. )
173. }
175. multi method keys(::?CLASS:D:) {
176. Seq.new(Rakudo::Iterator.IntRange(0,self.shape.AT-POS(0) - 1))
177. }
178. multi method kv(::?CLASS:D:) {
179. Seq.new(Rakudo::Iterator.KeyValue(self.iterator))
180. }
181. multi method pairs(::?CLASS:D:) {
182. Seq.new(Rakudo::Iterator.Pair(self.iterator))
183. }
184. multi method antipairs(::?CLASS:D:) {
185. Seq.new(Rakudo::Iterator.AntiPair(self.iterator))
186. }
188. method iterator(::?CLASS:D:) {
189. class :: does Iterator {
190. has Mu $!reified;
191. has Mu $!desc;
192. has int $!pos;
193. method !SET-SELF(Mu \list) {
194. nqp::stmts(
195. ($!reified := nqp::getattr(list,List,'$!reified')),
196. ($!desc := nqp::getattr(list,Array,'$!descriptor')),
197. ($!pos = -1),
198. self
199. )
200. }
201. method new(Mu \list) { nqp::create(self)!SET-SELF(list) }
202. method pull-one() is raw {
203. nqp::if(
204. nqp::islt_i(
205. ($!pos = nqp::add_i($!pos,1)),
206. nqp::elems($!reified)
207. ),
208. nqp::ifnull(
209. nqp::atpos($!reified,$!pos),
210. nqp::p6bindattrinvres(
211. (my $scalar := nqp::p6scalarfromdesc($!desc)),
212. Scalar,
213. '$!whence',
214. -> { nqp::bindpos($!reified,$!pos,$scalar) }
215. )
216. ),
217. IterationEnd
218. )
219. }
220. method push-all($target --> IterationEnd) {
221. nqp::stmts(
222. (my int $elems = nqp::elems($!reified)),
223. (my int $i = $!pos),
224. nqp::while(
225. nqp::islt_i(($i = nqp::add_i($i,1)),$elems),
226. $target.push(
227. nqp::ifnull(
228. nqp::atpos($!reified,$i),
229. nqp::p6bindattrinvres(
230. (my $scalar := nqp::p6scalarfromdesc($!desc)),
231. Scalar,
232. '$!whence',
233. -> { nqp::bindpos($!reified,$i,$scalar) }
234. )
235. )
236. )
237. ),
238. ($!pos = $i) # mark as done
239. )
240. }
241. method count-only() { nqp::p6box_i(nqp::elems($!reified)) }
242. method bool-only() { nqp::p6bool(nqp::elems($!reified)) }
243. method sink-all(--> IterationEnd) {
244. $!pos = nqp::elems($!reified)
245. }
246. }.new(self)
247. }
249. method reverse(::?CLASS:D:) is nodal {
250. Seq.new(nqp::if(
251. (my int $elems = nqp::elems(
252. my $from := nqp::getattr(self,List,'$!reified')
253. )),
254. nqp::stmts(
255. (my int $last = nqp::sub_i($elems,1)),
256. (my int $i = -1),
257. (my $to := nqp::setelems(nqp::create(IterationBuffer),$elems)),
258. nqp::while(
259. nqp::islt_i(($i = nqp::add_i($i,1)),$elems),
260. nqp::bindpos($to,nqp::sub_i($last,$i),
261. nqp::decont(nqp::atpos($from,$i)))
262. ),
263. Rakudo::Iterator.ReifiedList($to)
264. ),
265. Rakudo::Iterator.Empty
266. ))
267. }
268. method rotate(::?CLASS:D: Int(Cool) $rotate = 1) is nodal {
269. Rakudo::Internals.RotateListToList(
270. self, $rotate, self.new(:shape(self.shape)))
271. }
272. method sum() is nodal { self.List::sum }
273. }