src/org/ibex/util/Vec.java

   1 // Copyright 2000-2005 the Contributors, as shown in the revision logs.
   2 // Licensed under the Apache Public Source License 2.0 ("the License").
   3 // You may not use this file except in compliance with the License.
   4
   5 package org.ibex.util;
   6 import java.io.*;
   7
   8 /**
   9  *  An unsynchronized Vector implementation; same semantics as
  10  *  java.util.Vector. Useful for JDK1.1 platforms that don't have
  11  *  java.util.ArrayList.
  12  *
  13  *  May contain nulls.
  14  *
  15  *  @see java.util.Vector
  16  */
  17 public final class Vec implements Serializable, Cloneable {
  18
  19     private Object[] store;
  20     private int size = 0;
  21
  22     public Vec() { this(10); }
  23     public Vec(int i) { store = new Object[i]; }
  24     public Vec(int i, Object[] store) { size = i; this.store = store; }
  25     public Vec(Vec old) {
  26         store = new Object[old.store.length];
  27         System.arraycopy(old.store, 0, store, 0, old.store.length);
  28         this.size = old.size;
  29     }
  30
  31     public Object clone() { return new Vec(this); }
  32     private void grow() { grow(store.length * 2); }
  33     private void grow(int newsize) {
  34         Object[] newstore = new Object[newsize];
  35         System.arraycopy(store, 0, newstore, 0, size);
  36         store = newstore;
  37     }
  38
  39     public void removeAllElements() {
  40         for(int i=0; i<size; i++) store[i] = null;
  41         size = 0;
  42     }
  43
  44     public void toArray(Object[] o) {
  45         for(int i=0; i<size; i++)
  46             o[i] = store[i];
  47     }
  48
  49     public int indexOf(Object o) {
  50         for(int i=0; i<size; i++)
  51             if (o == null ? store[i] == null : store[i].equals(o)) return i;
  52
  53         return -1;
  54     }
  55
  56     public void addElement(Object o) {
  57         if (size >= store.length - 1) grow();
  58         store[size++] = o;
  59     }
  60
  61     public Object peek() {
  62         return lastElement();
  63     }
  64
  65     public Object elementAt(int i) {
  66         return store[i];
  67     }
  68
  69     public Object lastElement() {
  70         if (size == 0) return null;
  71         return store[size - 1];
  72     }
  73
  74     public void push(Object o) { addElement(o); }
  75     public Object pop() {
  76         Object ret = lastElement();
  77         if (size > 0) store[--size] = null;
  78         return ret;
  79     }
  80
  81     public int size() { return size; }
  82
  83     public void setSize(int newSize) {
  84         if (newSize < 0) throw new RuntimeException("tried to set size to negative value");
  85         if (newSize > store.length) grow(newSize * 2);
  86         if (newSize < size)
  87             for(int i=newSize; i<size; i++)
  88                 store[i] = null;
  89         size = newSize;
  90     }
  91
  92     public Object[] copyInto(Object[] out) {
  93         for(int i=0; i<size; i++)
  94             out[i] = store[i];
  95         return out;
  96     }
  97
  98     public void fromArray(Object[] in) {
  99         setSize(in.length);
 100         for(int i=0; i<size; i++)
 101             store[i] = in[i];
 102     }
 103
 104     public void removeElementAt(int i) {
 105         if (i >= size || i < 0) throw new RuntimeException("tried to remove an element outside the vector's limits");
 106         for(int j=i; j<size - 1; j++)
 107             store[j] = store[j + 1];
 108         setSize(size - 1);
 109     }
 110
 111     public void setElementAt(Object o, int i) {
 112         if (i >= size) setSize(i);
 113         store[i] = o;
 114     }
 115
 116     public void removeElement(Object o) {
 117         int idx = indexOf(o);
 118         if (idx != -1) removeElementAt(idx);
 119     }
 120
 121     public void insertElementAt(Object o, int at) {
 122         if (size == store.length) grow();
 123         for(int i=size; i>at; i--)
 124             store[i] = store[i-1];
 125         store[at] = o;
 126         size++;
 127     }
 128
 129     public interface CompareFunc {
 130         public int compare(Object a, Object b);
 131     }
 132
 133     public void sort(CompareFunc c) {
 134         sort(this, null, c, 0, size-1);
 135     }
 136
 137     public static void sort(Vec a, Vec b, CompareFunc c) {
 138         if (b != null && a.size != b.size) throw new IllegalArgumentException("Vec a and b must be of equal size");
 139         sort(a, b, c, 0, a.size-1);
 140     }
 141
 142     private static final void sort(Vec a, Vec b, CompareFunc c, int start, int end) {
 143         Object tmpa, tmpb = null;
 144         if(start >= end) return;
 145         if(end-start <= 6) {
 146             for(int i=start+1;i<=end;i++) {
 147                 tmpa = a.store[i];
 148                 if (b != null) tmpb = b.store[i];
 149                 int j;
 150                 for(j=i-1;j>=start;j--) {
 151                     if(c.compare(a.store[j],tmpa) <= 0) break;
 152                     a.store[j+1] = a.store[j];
 153                     if (b != null) b.store[j+1] = b.store[j];
 154                 }
 155                 a.store[j+1] = tmpa;
 156                 if (b != null) b.store[j+1] = tmpb;
 157             }
 158             return;
 159         }
 160
 161         Object pivot = a.store[end];
 162         int lo = start - 1;
 163         int hi = end;
 164
 165         do {
 166             while(c.compare(a.store[++lo],pivot) < 0) { }
 167             while((hi > lo) && c.compare(a.store[--hi],pivot) > 0) { }
 168             swap(a, lo,hi);
 169             if (b != null) swap(b, lo,hi);
 170         } while(lo < hi);
 171
 172         swap(a, lo,end);
 173         if (b != null) swap(b, lo,end);
 174
 175         sort(a, b, c, start, lo-1);
 176         sort(a, b, c, lo+1, end);
 177     }
 178
 179     private static final void swap(Vec vec, int a, int b) {
 180         if(a != b) {
 181             Object tmp = vec.store[a];
 182             vec.store[a] = vec.store[b];
 183             vec.store[b] = tmp;
 184         }
 185     }
 186
 187     public static final void sortInts(int[] a, int start, int end) {
 188         int tmpa;
 189         if(start >= end) return;
 190         if(end-start <= 6) {
 191             for(int i=start+1;i<=end;i++) {
 192                 tmpa = a[i];
 193                 int j;
 194                 for(j=i-1;j>=start;j--) {
 195                     if(a[j] <= tmpa) break;
 196                     a[j+1] = a[j];
 197                 }
 198                 a[j+1] = tmpa;
 199             }
 200             return;
 201         }
 202         int pivot = a[end];
 203         int lo = start - 1;
 204         int hi = end;
 205         do {
 206             while(a[++lo] < pivot) { }
 207             while((hi > lo) && a[--hi] > pivot) { }
 208             swapInts(a, lo, hi);
 209         } while(lo < hi);
 210         swapInts(a, lo, end);
 211         sortInts(a, start, lo-1);
 212         sortInts(a, lo+1, end);
 213     }
 214     private static final void swapInts(int[] vec, int a, int b) {
 215         if(a != b) {
 216             int tmp = vec[a];
 217             vec[a] = vec[b];
 218             vec[b] = tmp;
 219         }
 220     }
 221
 222
 223
 224     // just a cut-and-paste copy of Vec with int storage
 225     public static class Int {
 226         private int[] store;
 227         private int size = 0;
 228
 229         public Int() { this(10); }
 230         public Int(int i) { store = new int[i]; }
 231         public Int(int i, int[] store) { size = i; this.store = store; }
 232
 233         private void grow() { grow(store.length * 2); }
 234         private void grow(int newsize) {
 235             int[] newstore = new int[newsize];
 236             System.arraycopy(store, 0, newstore, 0, size);
 237             store = newstore;
 238         }
 239
 240         public void removeAllElements() {
 241             for(int i=0; i<size; i++) store[i] = 0;
 242             size = 0;
 243         }
 244
 245         public void toArray(int[] o) {
 246             for(int i=0; i<size; i++)
 247                 o[i] = store[i];
 248         }
 249
 250         public int[] dump() { int[] o = new int[size]; toArray(o); return o; }
 251
 252         public int indexOf(int o) {
 253             for(int i=0; i<size; i++)
 254                 if (o == store[i]) return i;
 255
 256             return -1;
 257         }
 258
 259         public void addElement(int o) {
 260             if (size >= store.length - 1) grow();
 261             store[size++] = o;
 262         }
 263
 264         public int peek() {
 265             return lastElement();
 266         }
 267
 268         public int elementAt(int i) {
 269             return store[i];
 270         }
 271
 272         public int lastElement() {
 273             if (size == 0) return 0;
 274             return store[size - 1];
 275         }
 276
 277         public void push(int o) { addElement(o); }
 278         public int pop() {
 279             int ret = lastElement();
 280             if (size > 0) store[size--] = 0;
 281             return ret;
 282         }
 283
 284         public int size() { return size; }
 285
 286         public void setSize(int newSize) {
 287             if (newSize < 0) throw new RuntimeException("tried to set size to negative value");
 288             if (newSize > store.length) grow(newSize * 2);
 289             if (newSize < size)
 290                 for(int i=newSize; i<size; i++)
 291                     store[i] = 0;
 292             size = newSize;
 293         }
 294
 295         public int[] copyInto(int[] out) {
 296             for(int i=0; i<size; i++)
 297                 out[i] = store[i];
 298             return out;
 299         }
 300
 301         public void fromArray(int[] in) {
 302             setSize(in.length);
 303             for(int i=0; i<size; i++)
 304                 store[i] = in[i];
 305         }
 306
 307         public void removeElementAt(int i) {
 308             if (i >= size || i < 0) throw new RuntimeException("tried to remove an element outside the vector's limits");
 309             for(int j=i; j<size - 1; j++)
 310                 store[j] = store[j + 1];
 311             setSize(size - 1);
 312         }
 313
 314         public void setElementAt(int o, int i) {
 315             if (i >= size) setSize(i);
 316             store[i] = o;
 317         }
 318
 319         public void removeElement(int o) {
 320             int idx = indexOf(o);
 321             if (idx != -1) removeElementAt(idx);
 322         }
 323
 324         public void insertElementAt(int o, int at) {
 325             if (size == store.length) grow();
 326             for(int i=size; i>at; i--)
 327                 store[i] = store[i-1];
 328             store[at] = o;
 329             size++;
 330         }
 331
 332         public void sort() { sort(this, null, 0, size-1); }
 333
 334         public static void sort(Vec.Int a, Vec.Int b) {
 335             if (b != null && a.size != b.size) throw new IllegalArgumentException("Vec.Int a and b must be of equal size");
 336             sort(a, b, 0, a.size-1);
 337         }
 338
 339         private static final void sort(Vec.Int a, Vec.Int b, int start, int end) {
 340             int tmpa, tmpb = 0;
 341             if(start >= end) return;
 342             if(end-start <= 6) {
 343                 for(int i=start+1;i<=end;i++) {
 344                     tmpa = a.store[i];
 345                     if (b != null) tmpb = b.store[i];
 346                     int j;
 347                     for(j=i-1;j>=start;j--) {
 348                         if((a.store[j]-tmpa) <= 0) break;
 349                         a.store[j+1] = a.store[j];
 350                         if (b != null) b.store[j+1] = b.store[j];
 351                     }
 352                     a.store[j+1] = tmpa;
 353                     if (b != null) b.store[j+1] = tmpb;
 354                 }
 355                 return;
 356             }
 357
 358             int pivot = a.store[end];
 359             int lo = start - 1;
 360             int hi = end;
 361
 362             do {
 363                 while((a.store[++lo]-pivot) < 0) { }
 364                 while((hi > lo) && (a.store[--hi]-pivot) > 0) { }
 365                 swap(a, lo,hi);
 366                 if (b != null) swap(b, lo,hi);
 367             } while(lo < hi);
 368
 369             swap(a, lo,end);
 370             if (b != null) swap(b, lo,end);
 371
 372             sort(a, b, start, lo-1);
 373             sort(a, b, lo+1, end);
 374         }
 375
 376         private static final void swap(Vec.Int vec, int a, int b) {
 377             if(a != b) {
 378                 int tmp = vec.store[a];
 379                 vec.store[a] = vec.store[b];
 380                 vec.store[b] = tmp;
 381             }
 382         }
 383
 384         public static final void sortInts(int[] a, int start, int end) {
 385             int tmpa;
 386             if(start >= end) return;
 387             if(end-start <= 6) {
 388                 for(int i=start+1;i<=end;i++) {
 389                     tmpa = a[i];
 390                     int j;
 391                     for(j=i-1;j>=start;j--) {
 392                         if(a[j] <= tmpa) break;
 393                         a[j+1] = a[j];
 394                     }
 395                     a[j+1] = tmpa;
 396                 }
 397                 return;
 398             }
 399
 400             int pivot = a[end];
 401             int lo = start - 1;
 402             int hi = end;
 403
 404             do {
 405                 while(a[++lo] < pivot) { }
 406                 while((hi > lo) && a[--hi] > pivot) { }
 407                 swapInts(a, lo, hi);
 408             } while(lo < hi);
 409             swapInts(a, lo, end);
 410             sortInts(a, start, lo-1);
 411             sortInts(a, lo+1, end);
 412         }
 413
 414         private static final void swapInts(int[] vec, int a, int b) {
 415             if(a != b) {
 416                 int tmp = vec[a];
 417                 vec[a] = vec[b];
 418                 vec[b] = tmp;
 419             }
 420         }
 421     }
 422
 423
 424
 425     public static final void sortFloats(float[] a, int start, int end) {
 426         float tmpa;
 427         if(start >= end) return;
 428         if(end-start <= 6) {
 429             for(int i=start+1;i<=end;i++) {
 430                 tmpa = a[i];
 431                 int j;
 432                 for(j=i-1;j>=start;j--) {
 433                     if(a[j] <= tmpa) break;
 434                     a[j+1] = a[j];
 435                 }
 436                 a[j+1] = tmpa;
 437             }
 438             return;
 439         }
 440         float pivot = a[end];
 441         int lo = start - 1;
 442         int hi = end;
 443         do {
 444             while(a[++lo] < pivot) { }
 445             while((hi > lo) && a[--hi] > pivot) { }
 446             swapFloats(a, lo, hi);
 447         } while(lo < hi);
 448         swapFloats(a, lo, end);
 449         sortFloats(a, start, lo-1);
 450         sortFloats(a, lo+1, end);
 451     }
 452     private static final void swapFloats(float[] vec, int a, int b) {
 453         if(a != b) {
 454             float tmp = vec[a];
 455             vec[a] = vec[b];
 456             vec[b] = tmp;
 457         }
 458     }
 459 }