package org.jcodec.common;

import java.util.Arrays;

public class IntIntMap {
  private static final int GROW_BY = 128;
  
  private static final int MIN_VALUE = -2147483648;
  
  private int[] storage;
  
  private int _size;
  
  public IntIntMap() {
    this.storage = createArray(128);
    Arrays.fill(this.storage, Integer.MIN_VALUE);
  }
  
  public void put(int key, int val) {
    if (val == Integer.MIN_VALUE)
      throw new IllegalArgumentException("This implementation can not store -2147483648"); 
    if (this.storage.length <= key) {
      int[] ns = createArray(key + 128);
      System.arraycopy(this.storage, 0, ns, 0, this.storage.length);
      Arrays.fill(ns, this.storage.length, ns.length, Integer.MIN_VALUE);
      this.storage = ns;
    } 
    if (this.storage[key] == Integer.MIN_VALUE)
      this._size++; 
    this.storage[key] = val;
  }
  
  public int get(int key) {
    return (key >= this.storage.length) ? Integer.MIN_VALUE : this.storage[key];
  }
  
  public boolean contains(int key) {
    return (key >= 0 && key < this.storage.length);
  }
  
  public int[] keys() {
    int[] result = new int[this._size];
    for (int i = 0, r = 0; i < this.storage.length; i++) {
      if (this.storage[i] != Integer.MIN_VALUE)
        result[r++] = i; 
    } 
    return result;
  }
  
  public void clear() {
    for (int i = 0; i < this.storage.length; i++)
      this.storage[i] = Integer.MIN_VALUE; 
    this._size = 0;
  }
  
  public int size() {
    return this._size;
  }
  
  public void remove(int key) {
    if (this.storage[key] != Integer.MIN_VALUE)
      this._size--; 
    this.storage[key] = Integer.MIN_VALUE;
  }
  
  public int[] values() {
    int[] result = createArray(this._size);
    for (int i = 0, r = 0; i < this.storage.length; i++) {
      if (this.storage[i] != Integer.MIN_VALUE)
        result[r++] = this.storage[i]; 
    } 
    return result;
  }
  
  private static int[] createArray(int size) {
    return new int[size];
  }
}