/*

* java-tron is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

*

* java-tron is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <http://www.gnu.org/licenses/>.

*/

package org.tron.utils;

import org.tron.db.ByteArrayWrapper;

import java.util.Collection;

import java.util.HashSet;

import java.util.Iterator;

import java.util.Set;

public class ByteArraySet implements Set<byte[]> {

Set<ByteArrayWrapper> delegate;

public ByteArraySet() {

this(new HashSet<ByteArrayWrapper>());

}

ByteArraySet(Set<ByteArrayWrapper> delegate) {

this.delegate = delegate;

}

@Override

public int size() {

return delegate.size();

}

@Override

public boolean isEmpty() {

return delegate.isEmpty();

}

@Override

public boolean contains(Object o) {

return delegate.contains(new ByteArrayWrapper((byte[]) o));

}

@Override

public Iterator<byte[]> iterator() {

return new Iterator<byte[]>() {

Iterator<ByteArrayWrapper> it = delegate.iterator();

@Override

public boolean hasNext() {

return it.hasNext();

}

@Override

public byte[] next() {

return it.next().getData();

}

@Override

public void remove() {

it.remove();

}

};

}

@Override

public Object[] toArray() {

byte[][] ret = new byte[size()][];

ByteArrayWrapper[] arr = delegate.toArray(new ByteArrayWrapper[size()]);

for (int i = 0; i < arr.length; i++) {

ret[i] = arr[i].getData();

}

return ret;

}

@Override

public <T> T[] toArray(T[] a) {

return (T[]) toArray();

}

@Override

public boolean add(byte[] bytes) {

return delegate.add(new ByteArrayWrapper(bytes));

}

@Override

public boolean remove(Object o) {

return delegate.remove(new ByteArrayWrapper((byte[]) o));

}

@Override

public boolean containsAll(Collection<?> c) {

throw new RuntimeException("Not implemented");

}

@Override

public boolean addAll(Collection<? extends byte[]> c) {

boolean ret = false;

for (byte[] bytes : c) {

ret |= add(bytes);

}

return ret;

}

@Override

public boolean retainAll(Collection<?> c) {

throw new RuntimeException("Not implemented");

}

@Override

public boolean removeAll(Collection<?> c) {

throw new RuntimeException("Not implemented");

}

@Override

public void clear() {

delegate.clear();

}

@Override

public boolean equals(Object o) {

throw new RuntimeException("Not implemented");

}

@Override

public int hashCode() {

throw new RuntimeException("Not implemented");

}

}