1 // Copyright 2004 Adam Megacz, see the COPYING file for licensing [GPL]
4 import org.ibex.util.*;
7 /** A JavaScript JSArray */
8 public class JSArray extends JS.BT {
9 private static final Object NULL = new Object();
12 public JSArray(int size) { setSize(size); }
14 /*private static int intVal(Object o) {
15 if (o instanceof Number) {
16 int intVal = ((Number)o).intValue();
17 if (intVal == ((Number)o).doubleValue()) return intVal;
18 return Integer.MIN_VALUE;
20 if (!(o instanceof String)) return Integer.MIN_VALUE;
22 for(int i=0; i<s.length(); i++) if (s.charAt(i) < '0' || s.charAt(i) > '9') return Integer.MIN_VALUE;
23 return Integer.parseInt(s);
26 public JS callMethod(JS method, JS a0, JS a1, JS a2, JS[] rest, int nargs) throws JSExn {
27 if(isString(method)) {
28 //#switch(JS.toString(method))
31 if(oldSize == 0) return null;
32 return removeElementAt(oldSize-1);
34 case "reverse": return reverse();
35 case "toString": return join(",");
37 if(length() == 0) return null;
38 return removeElementAt(0);
40 return join(nargs == 0 ? "," : JS.toString(a0));
42 return sort(nargs < 1 ? null : a0);
44 int start = toInt(nargs < 1 ? null : a0);
45 int end = nargs < 2 ? length() : toInt(a1);
46 return slice(start, end);
49 for(int i=0; i<nargs; i++) insertElementAt(i==0?a0:i==1?a1:i==2?a2:rest[i-3],oldSize+i);
50 return N(oldSize + nargs);
53 for(int i=0; i<nargs; i++) insertElementAt(i==0?a0:i==1?a1:i==2?a2:rest[i-3],i);
56 JSArray array = new JSArray();
57 for(int i=0; i<nargs; i++) array.addElement(i==0?a0:i==1?a1:i==2?a2:rest[i-3]);
61 return super.callMethod(method, a0, a1, a2, rest, nargs);
64 public JS get(JS key) throws JSExn {
67 if (i < 0 || i >= size()) return null;
71 //#switch(JS.toString(key))
72 case "pop": return METHOD;
73 case "reverse": return METHOD;
74 case "toString": return METHOD;
75 case "shift": return METHOD;
76 case "join": return METHOD;
77 case "sort": return METHOD;
78 case "slice": return METHOD;
79 case "push": return METHOD;
80 case "unshift": return METHOD;
81 case "splice": return METHOD;
82 case "length": return N(size());
85 return super.get(key);
88 public void put(JS key, JS val) throws JSExn {
95 if(i > oldSize) setSize(i);
96 insertElementAt(val,i);
101 if (JS.toString(key).equals("length")) {
102 setSize(JS.toInt(val));
109 public Enumeration keys() throws JSExn {
110 return new Enumeration(super.keys()) {
112 public boolean _hasMoreElements() { return n < size(); }
113 public JS _nextElement() {
114 return n >= size() ? null : JS.N(n++);
119 public final void setSize(int newSize) {
120 // FEATURE: This could be done a lot more efficiently in BalancedTree
121 int oldSize = size();
122 for(int i=oldSize;i<newSize;i++) insertElementAt(null,i);
123 for(int i=oldSize-1;i>=newSize;i--) removeElementAt(i);
126 public final int length() { return size(); }
127 public final JS elementAt(int i) {
128 if(i < 0 || i >= size()) throw new ArrayIndexOutOfBoundsException(i);
129 Object o = getNode(i);
130 return o == NULL ? (JS)null : (JS)o;
132 public final void addElement(JS o) {
133 insertNode(size(),o==null ? NULL : o);
135 public final void setElementAt(JS o, int i) {
136 if(i < 0 || i >= size()) throw new ArrayIndexOutOfBoundsException(i);
137 replaceNode(i,o==null ? NULL : o);
139 public final void insertElementAt(JS o, int i) {
140 if(i < 0 || i > size()) throw new ArrayIndexOutOfBoundsException(i);
141 insertNode(i,o==null ? NULL : o);
143 public final JS removeElementAt(int i) {
144 if(i < 0 || i >= size()) throw new ArrayIndexOutOfBoundsException(i);
145 Object o = deleteNode(i);
146 return o == NULL ? (JS)null : (JS)o;
149 public final int size() { return treeSize(); }
151 private JS join(String sep) throws JSExn {
153 if(length == 0) return JS.S("");
154 StringBuffer sb = new StringBuffer(64);
158 if(o != null) sb.append(JS.toString(o));
159 if(++i == length) break;
162 return JS.S(sb.toString());
165 // FEATURE: Implement this more efficiently
166 private JS reverse() {
168 if(size < 2) return this;
171 for(int i=size-1,j=0;i>=0;i--,j++) insertElementAt((JS)vec.elementAt(i),j);
175 private JS slice(int start, int end) {
176 int length = length();
177 if(start < 0) start = length+start;
178 if(end < 0) end = length+end;
179 if(start < 0) start = 0;
181 if(start > length) start = length;
182 if(end > length) end = length;
183 JSArray a = new JSArray(end-start);
184 for(int i=0;i<end-start;i++)
185 a.setElementAt(elementAt(start+i),i);
189 private static final Vec.CompareFunc defaultSort = new Vec.CompareFunc() {
190 public int compare(Object a, Object b) {
192 return JS.toString((JS)a).compareTo(JS.toString((JS)b));
193 } catch(JSExn e) { throw new JSExn.Wrapper(e); }
196 private JS sort(JS tmp) throws JSExn {
199 if(tmp instanceof JS) {
200 final JS jsFunc = (JS) tmp;
201 vec.sort(new Vec.CompareFunc() {
202 public int compare(Object a, Object b) {
204 return JS.toInt(jsFunc.call((JS)a, (JS)b, null, null, 2));
205 } catch(JSExn e) { throw new JSExn.Wrapper(e); }
209 vec.sort(defaultSort);
211 } catch(JSExn.Wrapper e) {
218 private JS splice(JSArray args) throws JSExn {
219 int oldLength = length();
220 int start = JS.toInt(args.length() < 1 ? null : args.elementAt(0));
221 int deleteCount = JS.toInt(args.length() < 2 ? null : args.elementAt(1));
222 int newCount = args.length() - 2;
223 if(newCount < 0) newCount = 0;
224 if(start < 0) start = oldLength+start;
225 if(start < 0) start = 0;
226 if(start > oldLength) start = oldLength;
227 if(deleteCount < 0) deleteCount = 0;
228 if(deleteCount > oldLength-start) deleteCount = oldLength-start;
229 int newLength = oldLength - deleteCount + newCount;
230 int lengthChange = newLength - oldLength;
231 JSArray ret = new JSArray(deleteCount);
232 for(int i=0;i<deleteCount;i++)
233 ret.setElementAt(elementAt(start+i),i);
234 if(lengthChange > 0) {
236 for(int i=newLength-1;i>=start+newCount;i--)
237 setElementAt(elementAt(i-lengthChange),i);
238 } else if(lengthChange < 0) {
239 for(int i=start+newCount;i<newLength;i++)
240 setElementAt(elementAt(i-lengthChange),i);
243 for(int i=0;i<newCount;i++)
244 setElementAt(args.elementAt(i+2),start+i);
248 protected Vec toVec() {
252 for(int i=0;i<count;i++) {
253 Object o = getNode(i);
254 vec.setElementAt(o == NULL ? null : o,i);
259 protected void setFromVec(Vec vec) {
260 int count = vec.size();
262 for(int i=0;i<count;i++) {
263 Object o = vec.elementAt(i);
264 insertNode(i,o==null ? NULL : o);
268 String coerceToString() throws JSExn { return JS.toString(join(",")); }