// Program that compares several structures for a large number of calls on
// these three operations: add, remove, contains.  
// It has several tests that it runs:
//     test1: using a sorted ArrayList
//     test2: using Java's TreeSet (binary search tree)
//     test3: using Java's HashSet
//     test4: using our HashStringSet

import java.util.*;
import java.io.*;

public class CollectionsTest {
   public static void main(String[] args) throws FileNotFoundException {
        Scanner console = new Scanner(System.in);
        System.out.print("input file? ");
        Scanner input = new Scanner(new File(console.nextLine()));
        List<String> data = new ArrayList<String>();
        while (input.hasNext())
            data.add(input.next());
        int max = 100;
        
        System.out.println("Total words = " + data.size());
        System.out.println("max         = " + max);
        
        int which = -1;
        do {
            System.out.println("Which test? (1=ArrayList,  2=Java's TreeSet,");
            System.out.println("     3=Java's HashSet, 4=our HashStringSet,");
            System.out.print("     Anything else=Quit)? ");
            which = console.nextInt();
            
            if (which == 1)
                test1(data, max);
            else if (which == 2)
                test2(data, max);
            else if (which == 3)
                test3(data, max);
            else if (which == 4)
                test4(data, max);
            else
                which = -1;
        } while (which != -1);
    }

    // run a test for an unsorted ArrayList
    public static void test1(List<String> data, int max) {
        long start = System.currentTimeMillis();
        List<String> lst = new ArrayList<String>(max);
        for (int i = 0; i < data.size(); i++)
            if (i % 5 == 0)
                lst.remove(data.get(i));
            else if (i % 5 == 1)
                lst.contains(data.get(i));
            else if (!lst.contains(data.get(i)))
                lst.add(data.get(i));
        System.out.println("list size after = " + lst.size());
        double elapsed = (System.currentTimeMillis() - start) / 1000.0;
        System.out.println("ArrayList time = " + elapsed + " seconds");
        System.out.println();
    }

    // run a test for a TreeSet (binary search tree)
    public static void test2(List<String> data, int max) {
        long start = System.currentTimeMillis();
        // no TreeSet constructor that takes a size, so max not used
        TreeSet<String> s = new TreeSet<String>();
        for (int i = 0; i < data.size(); i++)
            if (i % 5 == 0)
                s.remove(data.get(i));
            else if (i % 5 == 1)
                s.contains(data.get(i));
            else
                s.add(data.get(i));
        System.out.println("set size after = " + s.size());
        double elapsed = (System.currentTimeMillis() - start) / 1000.0;
        System.out.println("TreeSet time = " + elapsed + " seconds");
        System.out.println();
    }

    // run a test for Java's implementation of HashSet
    public static void test3(List<String> data, int max) {
        long start = System.currentTimeMillis();
        java.util.HashSet<String> s = new java.util.HashSet<String>(max);
        for (int i = 0; i < data.size(); i++)
            if (i % 5 == 0)
                s.remove(data.get(i));
            else if (i % 5 == 1)
                s.contains(data.get(i));
            else
                s.add(data.get(i));
        System.out.println("set size after = " + s.size());
        double elapsed = (System.currentTimeMillis() - start) / 1000.0;
        System.out.println("Java's HashSet time = " + elapsed + " seconds");
        System.out.println();
    }
    
    // run a test for our completed HashStringSet
    public static void test4(List<String> data, int max) {
        long start = System.currentTimeMillis();
        HashStringSetComplete s = new HashStringSetComplete(max);
        for (int i = 0; i < data.size(); i++)
            if (i % 5 == 0)
                s.remove(data.get(i));
            else if (i % 5 == 1)
                s.contains(data.get(i));
            else
                s.add(data.get(i));
        System.out.println("set size after = " + s.size());
        double elapsed = (System.currentTimeMillis() - start) / 1000.0;
        System.out.println("our HashSet time = " + elapsed + " seconds");
        System.out.println();
    }
}