java-decompiler: post-import cleanup (classes moved)
This commit is contained in:
204
src/org/jetbrains/java/decompiler/util/VBStyleCollection.java
Normal file
204
src/org/jetbrains/java/decompiler/util/VBStyleCollection.java
Normal file
@@ -0,0 +1,204 @@
|
||||
/*
|
||||
* Fernflower - The Analytical Java Decompiler
|
||||
* http://www.reversed-java.com
|
||||
*
|
||||
* (C) 2008 - 2010, Stiver
|
||||
*
|
||||
* This software is NEITHER public domain NOR free software
|
||||
* as per GNU License. See license.txt for more details.
|
||||
*
|
||||
* This software is distributed WITHOUT ANY WARRANTY; without
|
||||
* even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
||||
* A PARTICULAR PURPOSE.
|
||||
*/
|
||||
|
||||
package org.jetbrains.java.decompiler.util;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collection;
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
|
||||
public class VBStyleCollection<E, K> extends ArrayList<E> {
|
||||
|
||||
private HashMap<K, Integer> map = new HashMap<K, Integer>();
|
||||
|
||||
private ArrayList<K> lstKeys = new ArrayList<K>();
|
||||
|
||||
public VBStyleCollection() {
|
||||
super();
|
||||
}
|
||||
|
||||
public VBStyleCollection(int initialCapacity) {
|
||||
super(initialCapacity);
|
||||
lstKeys = new ArrayList<K>(initialCapacity);
|
||||
map = new HashMap<K, Integer>(initialCapacity);
|
||||
}
|
||||
|
||||
public VBStyleCollection(Collection<E> c) {
|
||||
super(c);
|
||||
}
|
||||
|
||||
public boolean add(E element) {
|
||||
lstKeys.add(null);
|
||||
super.add(element);
|
||||
return true;
|
||||
}
|
||||
|
||||
public boolean remove(Object element) { // TODO: error on void remove(E element)
|
||||
throw new RuntimeException("not implemented!");
|
||||
}
|
||||
|
||||
public boolean addAll(Collection<? extends E> c) {
|
||||
for(int i=c.size()-1;i>=0;i--) {
|
||||
lstKeys.add(null);
|
||||
}
|
||||
return super.addAll(c);
|
||||
}
|
||||
|
||||
public void addAllWithKey(VBStyleCollection<E, K> c) {
|
||||
for(int i=0;i<c.size();i++) {
|
||||
addWithKey(c.get(i), c.getKey(i));
|
||||
}
|
||||
}
|
||||
|
||||
public void addAllWithKey(Collection<E> elements, Collection<K> keys) {
|
||||
int index = super.size();
|
||||
|
||||
for(K key : keys) {
|
||||
map.put(key, index++);
|
||||
}
|
||||
|
||||
super.addAll(elements);
|
||||
lstKeys.addAll(keys);
|
||||
}
|
||||
|
||||
public void addWithKey(E element, K key) {
|
||||
map.put(key, super.size());
|
||||
super.add(element);
|
||||
lstKeys.add(key);
|
||||
}
|
||||
|
||||
// TODO: speed up the method
|
||||
public E putWithKey(E element, K key) {
|
||||
Integer index = map.get(key);
|
||||
if (index == null) {
|
||||
addWithKey(element, key);
|
||||
} else {
|
||||
return super.set(index, element);
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
public void add(int index, E element) {
|
||||
addToListIndex(index, 1);
|
||||
lstKeys.add(index, null);
|
||||
super.add(index, element);
|
||||
}
|
||||
|
||||
public void addWithKeyAndIndex(int index, E element, K key) {
|
||||
addToListIndex(index, 1);
|
||||
map.put(key, new Integer(index));
|
||||
super.add(index, element);
|
||||
lstKeys.add(index, key);
|
||||
}
|
||||
|
||||
public void removeWithKey(K key) {
|
||||
int index = ((Integer) map.get(key)).intValue();
|
||||
addToListIndex(index + 1, -1);
|
||||
super.remove(index);
|
||||
lstKeys.remove(index);
|
||||
map.remove(key);
|
||||
}
|
||||
|
||||
public E remove(int index) {
|
||||
addToListIndex(index + 1, -1);
|
||||
Object obj = lstKeys.get(index);
|
||||
if (obj != null) {
|
||||
map.remove(obj);
|
||||
}
|
||||
lstKeys.remove(index);
|
||||
return super.remove(index);
|
||||
}
|
||||
|
||||
public E getWithKey(K key) {
|
||||
Integer index = map.get(key);
|
||||
if (index == null) {
|
||||
return null;
|
||||
}
|
||||
return super.get(index.intValue());
|
||||
}
|
||||
|
||||
public int getIndexByKey(K key) {
|
||||
return map.get(key).intValue();
|
||||
}
|
||||
|
||||
public E getLast() {
|
||||
return super.get(super.size()-1);
|
||||
}
|
||||
|
||||
public boolean containsKey(K key) {
|
||||
return map.containsKey(key);
|
||||
}
|
||||
|
||||
public void clear() {
|
||||
map.clear();
|
||||
lstKeys.clear();
|
||||
super.clear();
|
||||
}
|
||||
|
||||
public VBStyleCollection<E, K> clone() {
|
||||
VBStyleCollection<E, K> c = new VBStyleCollection<E, K>();
|
||||
c.addAll(new ArrayList<E>(this));
|
||||
c.setMap(new HashMap<K, Integer>(map));
|
||||
c.setLstKeys(new ArrayList<K>(lstKeys));
|
||||
return c;
|
||||
}
|
||||
|
||||
public void swap(int index1, int index2) {
|
||||
|
||||
Collections.swap(this, index1, index2);
|
||||
Collections.swap(lstKeys, index1, index2);
|
||||
|
||||
K key = lstKeys.get(index1);
|
||||
if(key!=null) {
|
||||
map.put(key, new Integer(index1));
|
||||
}
|
||||
|
||||
key = lstKeys.get(index2);
|
||||
if(key!=null) {
|
||||
map.put(key, new Integer(index2));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public HashMap<K, Integer> getMap() {
|
||||
return map;
|
||||
}
|
||||
|
||||
public void setMap(HashMap<K, Integer> map) {
|
||||
this.map = map;
|
||||
}
|
||||
|
||||
public K getKey(int index) {
|
||||
return lstKeys.get(index);
|
||||
}
|
||||
|
||||
public ArrayList<K> getLstKeys() {
|
||||
return lstKeys;
|
||||
}
|
||||
|
||||
public void setLstKeys(ArrayList<K> lstKeys) {
|
||||
this.lstKeys = lstKeys;
|
||||
}
|
||||
|
||||
private void addToListIndex(int index, int diff) {
|
||||
for (int i = lstKeys.size() - 1; i >= index; i--) {
|
||||
K obj = lstKeys.get(i);
|
||||
if (obj != null) {
|
||||
map.put(obj, new Integer(i + diff));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user