test/loops/CollectionWordLoops.java

/*
 * Written by Doug Lea with assistance from members of JCP JSR-166
 * Expert Group and released to the public domain, as explained at
 * http://creativecommons.org/publicdomain/zero/1.0/
 */
import java.io.*;
import java.util.*;

public class CollectionWordLoops {

    static final String[] WORDS_FILES = {
        "kw.txt",
        "class.txt",
        //        "dir.txt",
        //        "ids.txt",
        //        "/usr/dict/words",
    };

    static final int MAX_WORDS = 500000;
    static final int pinsert   = 80;
    static final int premove   = 2;
    static final int NOPS      = 100000;
    static final int numTests  = 2;

    public static void main(String[] args) {
        Class<?> collectionClass = null;
        try {
            collectionClass = Class.forName(args[0]);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException("Class " + args[0] + " not found.");
        }

        System.out.println("Testing " + collectionClass.getName());

        for (int s = 0; s < WORDS_FILES.length; ++s)
            tests(collectionClass, numTests, s);

        for (int s = WORDS_FILES.length-1; s >= 0; --s)
            tests(collectionClass, numTests, s);
    }

    static void tests(Class<?> collectionClass, int numTests, int sizeIndex) {
        try {
            String[] key = readWords(sizeIndex);
            int size = key.length;

            System.out.print("n = " +LoopHelpers.rightJustify(size) +" : ");
            long least = Long.MAX_VALUE;

            for (int i = 0; i < numTests; ++i) {
                Collection<String> m = newCollection(collectionClass);
                long t = doTest(collectionClass.getName(), m, key);
                if (t < least) least = t;
                m.clear();
                m = null;
            }

            long nano = Math.round(1000000.0 * (least) / NOPS);
            System.out.println(LoopHelpers.rightJustify(nano) + " ns per op");
        } catch (IOException ignore) {
            return; // skip test if can't read file
        }
    }

    static Collection<String> newCollection(Class<?> cl) {
        try {
            Collection m = (Collection<String>) cl.newInstance();
            return m;
        } catch (Exception e) {
            throw new RuntimeException("Can't instantiate " + cl + ": " + e);
        }
    }

    static void pause() {
        try { Thread.sleep(100); }
        catch (InterruptedException ie) { return; }
    }

    static String[] readWords(int sizeIndex) throws IOException {
        String[] l = new String[MAX_WORDS];
        String[] array = null;
        try {
            FileReader fr = new FileReader(WORDS_FILES[sizeIndex]);
            BufferedReader reader = new BufferedReader(fr);
            int k = 0;
            for (;;) {
                String s = reader.readLine();
                if (s == null) break;
                l[k++] = s;
            }
            array = new String[k];
            for (int i = 0; i < k; ++i) {
                array[i] = l[i];
                l[i] = null;
            }
            l = null;
            reader.close();
        }
        catch (IOException ex) {
            System.out.println("Can't read words file:" + ex);
            throw ex;
        }
        return array;
    }

    static long doTest(String name,
                       final Collection<String> m,
                       final String[] key) {

        //    System.out.print(name + "\t");
        Runner runner = new Runner(m, key);
        long startTime = System.currentTimeMillis();
        runner.run();
        long afterRun = System.currentTimeMillis();
        long runTime = afterRun - startTime;
        int np = runner.total;
        if (runner.total == runner.hashCode())
            System.out.println("Useless Number" + runner.total);
        int sz = runner.maxsz;
        if (sz == runner.hashCode())
            System.out.println("Useless Number" + sz);
        //        System.out.print(" m = " + sz);
        return runTime;
    }

    static class Runner implements Runnable {
        final Collection<String> collection;
        final String[] key;
        LoopHelpers.SimpleRandom rng = new LoopHelpers.SimpleRandom();
        final int pctrem;
        final int pctins;
        int nputs = 0;
        int npgets = 0;
        int nagets = 0;
        int nremoves = 0;
        volatile int total;
        int maxsz;

        Runner(Collection<String> m, String[] k) {
            collection = m; key = k;
            pctrem = (int)(((long)premove * (long)(Integer.MAX_VALUE/2)) / 50);
            pctins = (int)(((long)pinsert * (long)(Integer.MAX_VALUE/2)) / 50);
        }

        int oneStep(int j) {
            int n = key.length;
            int r = rng.next() & 0x7FFFFFFF;
            int jinc = (r & 7);
            j += jinc - 3;
            if (j >= n) j -= n;
            if (j < 0) j += n;

            int l = n / 4 + j;
            if (l >= n) l -= n;

            String k = key[j];

            if (!collection.contains(k)) {
                ++nagets;
                if (r < pctins) {
                    collection.add(k);
                    ++nputs;
                    int csz = nputs - nremoves;
                    if (csz > maxsz) maxsz = csz;
                }
            }
            else {
                ++npgets;
                if (r < pctrem) {
                    collection.remove(k);
                    ++nremoves;
                    j += ((r >>> 8) & 7) + n / 2;
                    if (j >= n) j -= n;
                }
            }
            return j;
        }

        public void run() {
            int j = key.length / 2;
            for (int i = 0; i < NOPS; ++i) {
                j = oneStep(j);
            }
            total = nputs + npgets + nagets + nremoves;
        }
    }

}
Revision:	1.11
Committed:	Mon Aug 10 03:13:33 2015 UTC (8 years, 9 months ago) by jsr166
Branch:	MAIN
Changes since 1.10:	+0 -1 lines
Log Message:	delete unwanted blank lines
#	User	Rev	Content
1	dl	1.1	/*
2			* Written by Doug Lea with assistance from members of JCP JSR-166
3			* Expert Group and released to the public domain, as explained at
4	jsr166	1.6	* http://creativecommons.org/publicdomain/zero/1.0/
5	dl	1.1	*/
6	jsr166	1.9	import java.io.*;
7	dl	1.1	import java.util.*;
8
9			public class CollectionWordLoops {
10
11			static final String[] WORDS_FILES = {
12	jsr166	1.3	"kw.txt",
13	dl	1.1	"class.txt",
14	dl	1.2	// "dir.txt",
15	jsr166	1.3	// "ids.txt",
16			// "/usr/dict/words",
17	dl	1.1	};
18
19			static final int MAX_WORDS = 500000;
20			static final int pinsert = 80;
21			static final int premove = 2;
22			static final int NOPS = 100000;
23			static final int numTests = 2;
24
25			public static void main(String[] args) {
26	jsr166	1.8	Class<?> collectionClass = null;
27	dl	1.1	try {
28			collectionClass = Class.forName(args[0]);
29	jsr166	1.4	} catch (ClassNotFoundException e) {
30	dl	1.1	throw new RuntimeException("Class " + args[0] + " not found.");
31			}
32
33			System.out.println("Testing " + collectionClass.getName());
34
35	jsr166	1.3	for (int s = 0; s < WORDS_FILES.length; ++s)
36	dl	1.1	tests(collectionClass, numTests, s);
37
38	jsr166	1.3	for (int s = WORDS_FILES.length-1; s >= 0; --s)
39	dl	1.1	tests(collectionClass, numTests, s);
40			}
41
42	jsr166	1.8	static void tests(Class<?> collectionClass, int numTests, int sizeIndex) {
43	jsr166	1.3	try {
44	dl	1.1	String[] key = readWords(sizeIndex);
45			int size = key.length;
46	jsr166	1.3
47	dl	1.1	System.out.print("n = " +LoopHelpers.rightJustify(size) +" : ");
48			long least = Long.MAX_VALUE;
49	jsr166	1.3
50	dl	1.1	for (int i = 0; i < numTests; ++i) {
51			Collection<String> m = newCollection(collectionClass);
52			long t = doTest(collectionClass.getName(), m, key);
53			if (t < least) least = t;
54			m.clear();
55			m = null;
56			}
57	jsr166	1.3
58	dl	1.1	long nano = Math.round(1000000.0 * (least) / NOPS);
59			System.out.println(LoopHelpers.rightJustify(nano) + " ns per op");
60			} catch (IOException ignore) {
61			return; // skip test if can't read file
62			}
63			}
64
65	jsr166	1.8	static Collection<String> newCollection(Class<?> cl) {
66	dl	1.1	try {
67	jsr166	1.5	Collection m = (Collection<String>) cl.newInstance();
68	dl	1.1	return m;
69	jsr166	1.4	} catch (Exception e) {
70	dl	1.1	throw new RuntimeException("Can't instantiate " + cl + ": " + e);
71			}
72			}
73
74			static void pause() {
75	jsr166	1.4	try { Thread.sleep(100); }
76			catch (InterruptedException ie) { return; }
77	dl	1.1	}
78
79			static String[] readWords(int sizeIndex) throws IOException {
80			String[] l = new String[MAX_WORDS];
81			String[] array = null;
82			try {
83			FileReader fr = new FileReader(WORDS_FILES[sizeIndex]);
84			BufferedReader reader = new BufferedReader(fr);
85			int k = 0;
86			for (;;) {
87			String s = reader.readLine();
88			if (s == null) break;
89			l[k++] = s;
90			}
91			array = new String[k];
92			for (int i = 0; i < k; ++i) {
93			array[i] = l[i];
94			l[i] = null;
95			}
96			l = null;
97			reader.close();
98			}
99			catch (IOException ex) {
100			System.out.println("Can't read words file:" + ex);
101			throw ex;
102			}
103			return array;
104			}
105
106			static long doTest(String name,
107	jsr166	1.3	final Collection<String> m,
108	dl	1.1	final String[] key) {
109
110			// System.out.print(name + "\t");
111			Runner runner = new Runner(m, key);
112			long startTime = System.currentTimeMillis();
113			runner.run();
114			long afterRun = System.currentTimeMillis();
115	jsr166	1.7	long runTime = afterRun - startTime;
116	dl	1.1	int np = runner.total;
117	jsr166	1.3	if (runner.total == runner.hashCode())
118	dl	1.1	System.out.println("Useless Number" + runner.total);
119			int sz = runner.maxsz;
120	jsr166	1.3	if (sz == runner.hashCode())
121	dl	1.1	System.out.println("Useless Number" + sz);
122			// System.out.print(" m = " + sz);
123			return runTime;
124			}
125
126			static class Runner implements Runnable {
127			final Collection<String> collection;
128			final String[] key;
129			LoopHelpers.SimpleRandom rng = new LoopHelpers.SimpleRandom();
130			final int pctrem;
131			final int pctins;
132			int nputs = 0;
133			int npgets = 0;
134			int nagets = 0;
135			int nremoves = 0;
136			volatile int total;
137			int maxsz;
138
139			Runner(Collection<String> m, String[] k) {
140	jsr166	1.3	collection = m; key = k;
141	dl	1.1	pctrem = (int)(((long)premove * (long)(Integer.MAX_VALUE/2)) / 50);
142			pctins = (int)(((long)pinsert * (long)(Integer.MAX_VALUE/2)) / 50);
143			}
144
145			int oneStep(int j) {
146			int n = key.length;
147			int r = rng.next() & 0x7FFFFFFF;
148			int jinc = (r & 7);
149			j += jinc - 3;
150	jsr166	1.3	if (j >= n) j -= n;
151	dl	1.1	if (j < 0) j += n;
152
153			int l = n / 4 + j;
154	jsr166	1.3	if (l >= n) l -= n;
155
156	dl	1.1	String k = key[j];
157	jsr166	1.3
158	dl	1.1	if (!collection.contains(k)) {
159			++nagets;
160			if (r < pctins) {
161			collection.add(k);
162			++nputs;
163			int csz = nputs - nremoves;
164			if (csz > maxsz) maxsz = csz;
165			}
166			}
167			else {
168			++npgets;
169			if (r < pctrem) {
170			collection.remove(k);
171			++nremoves;
172	jsr166	1.7	j += ((r >>> 8) & 7) + n / 2;
173	dl	1.1	if (j >= n) j -= n;
174			}
175			}
176			return j;
177			}
178
179			public void run() {
180			int j = key.length / 2;
181			for (int i = 0; i < NOPS; ++i) {
182			j = oneStep(j);
183			}
184			total = nputs + npgets + nagets + nremoves;
185			}
186			}
187
188			}