src/share/classes/com/sun/tools/javac/util/Bits.java

changeset 1
9a66ca7c79fa
child 554
9d9f26857129
equal deleted inserted replaced
-1:000000000000 1:9a66ca7c79fa
1 /*
2 * Copyright 1999-2005 Sun Microsystems, Inc. All Rights Reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation. Sun designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Sun in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
22 * CA 95054 USA or visit www.sun.com if you need additional information or
23 * have any questions.
24 */
25
26 package com.sun.tools.javac.util;
27
28 /** A class for extensible, mutable bit sets.
29 *
30 * <p><b>This is NOT part of any API supported by Sun Microsystems. If
31 * you write code that depends on this, you do so at your own risk.
32 * This code and its internal interfaces are subject to change or
33 * deletion without notice.</b>
34 */
35 public class Bits {
36
37
38 private final static int wordlen = 32;
39 private final static int wordshift = 5;
40 private final static int wordmask = wordlen - 1;
41
42 private int[] bits;
43
44 /** Construct an initially empty set.
45 */
46 public Bits() {
47 this(new int[1]);
48 }
49
50 /** Construct a set consisting initially of given bit vector.
51 */
52 public Bits(int[] bits) {
53 this.bits = bits;
54 }
55
56 /** Construct a set consisting initially of given range.
57 */
58 public Bits(int start, int limit) {
59 this();
60 inclRange(start, limit);
61 }
62
63 private void sizeTo(int len) {
64 if (bits.length < len) {
65 int[] newbits = new int[len];
66 System.arraycopy(bits, 0, newbits, 0, bits.length);
67 bits = newbits;
68 }
69 }
70
71 /** This set = {}.
72 */
73 public void clear() {
74 for (int i = 0; i < bits.length; i++) bits[i] = 0;
75 }
76
77 /** Return a copy of this set.
78 */
79 public Bits dup() {
80 int[] newbits = new int[bits.length];
81 System.arraycopy(bits, 0, newbits, 0, bits.length);
82 return new Bits(newbits);
83 }
84
85 /** Include x in this set.
86 */
87 public void incl(int x) {
88 assert x >= 0;
89 sizeTo((x >>> wordshift) + 1);
90 bits[x >>> wordshift] = bits[x >>> wordshift] |
91 (1 << (x & wordmask));
92 }
93
94
95 /** Include [start..limit) in this set.
96 */
97 public void inclRange(int start, int limit) {
98 sizeTo((limit >>> wordshift) + 1);
99 for (int x = start; x < limit; x++)
100 bits[x >>> wordshift] = bits[x >>> wordshift] |
101 (1 << (x & wordmask));
102 }
103
104 /** Exclude x from this set.
105 */
106 public void excl(int x) {
107 assert x >= 0;
108 sizeTo((x >>> wordshift) + 1);
109 bits[x >>> wordshift] = bits[x >>> wordshift] &
110 ~(1 << (x & wordmask));
111 }
112
113 /** Is x an element of this set?
114 */
115 public boolean isMember(int x) {
116 return
117 0 <= x && x < (bits.length << wordshift) &&
118 (bits[x >>> wordshift] & (1 << (x & wordmask))) != 0;
119 }
120
121 /** this set = this set & xs.
122 */
123 public Bits andSet(Bits xs) {
124 sizeTo(xs.bits.length);
125 for (int i = 0; i < xs.bits.length; i++)
126 bits[i] = bits[i] & xs.bits[i];
127 return this;
128 }
129
130 /** this set = this set | xs.
131 */
132 public Bits orSet(Bits xs) {
133 sizeTo(xs.bits.length);
134 for (int i = 0; i < xs.bits.length; i++)
135 bits[i] = bits[i] | xs.bits[i];
136 return this;
137 }
138
139 /** this set = this set \ xs.
140 */
141 public Bits diffSet(Bits xs) {
142 for (int i = 0; i < bits.length; i++) {
143 if (i < xs.bits.length) {
144 bits[i] = bits[i] & ~xs.bits[i];
145 }
146 }
147 return this;
148 }
149
150 /** this set = this set ^ xs.
151 */
152 public Bits xorSet(Bits xs) {
153 sizeTo(xs.bits.length);
154 for (int i = 0; i < xs.bits.length; i++)
155 bits[i] = bits[i] ^ xs.bits[i];
156 return this;
157 }
158
159 /** Count trailing zero bits in an int. Algorithm from "Hacker's
160 * Delight" by Henry S. Warren Jr. (figure 5-13)
161 */
162 private static int trailingZeroBits(int x) {
163 assert wordlen == 32;
164 if (x == 0) return 32;
165 int n = 1;
166 if ((x & 0xffff) == 0) { n += 16; x >>>= 16; }
167 if ((x & 0x00ff) == 0) { n += 8; x >>>= 8; }
168 if ((x & 0x000f) == 0) { n += 4; x >>>= 4; }
169 if ((x & 0x0003) == 0) { n += 2; x >>>= 2; }
170 return n - (x&1);
171 }
172
173 /** Return the index of the least bit position >= x that is set.
174 * If none are set, returns -1. This provides a nice way to iterate
175 * over the members of a bit set:
176 * <pre>
177 * for (int i = bits.nextBit(0); i>=0; i = bits.nextBit(i+1)) ...
178 * </pre>
179 */
180 public int nextBit(int x) {
181 int windex = x >>> wordshift;
182 if (windex >= bits.length) return -1;
183 int word = bits[windex] & ~((1 << (x & wordmask))-1);
184 while (true) {
185 if (word != 0)
186 return (windex << wordshift) + trailingZeroBits(word);
187 windex++;
188 if (windex >= bits.length) return -1;
189 word = bits[windex];
190 }
191 }
192
193 /** a string representation of this set.
194 */
195 public String toString() {
196 char[] digits = new char[bits.length * wordlen];
197 for (int i = 0; i < bits.length * wordlen; i++)
198 digits[i] = isMember(i) ? '1' : '0';
199 return new String(digits);
200 }
201
202 /** Test Bits.nextBit(int). */
203 public static void main(String[] args) {
204 java.util.Random r = new java.util.Random();
205 Bits bits = new Bits();
206 int dupCount = 0;
207 for (int i=0; i<125; i++) {
208 int k;
209 do {
210 k = r.nextInt(250);
211 } while (bits.isMember(k));
212 System.out.println("adding " + k);
213 bits.incl(k);
214 }
215 int count = 0;
216 for (int i = bits.nextBit(0); i >= 0; i = bits.nextBit(i+1)) {
217 System.out.println("found " + i);
218 count ++;
219 }
220 if (count != 125) throw new Error();
221 }
222 }

mercurial