C++程序  |  96行  |  3.75 KB

// [The "BSD licence"]
// Copyright (c) 2006-2007 Kay Roepke 2010 Alan Condit
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
// 1. Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
// 2. Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
// 3. The name of the author may not be used to endorse or promote products
//    derived from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
// IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
// OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
// IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
// NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
// THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.


#import <Cocoa/Cocoa.h>
#import <CoreFoundation/CoreFoundation.h>
#import "ANTLRToken.h"
#import "AMutableArray.h"

#define BITS (sizeof(NSUInteger) * 8)
#define LOG_BITS ((sizeof(NSUInteger)==8)?6:5)

// A simple wrapper around CoreFoundation bit vectors to shield the rest of the implementation
// from the specifics of the BitVector initialization and query functions.
// This is fast, so there is no need to reinvent the wheel just yet.

@interface ANTLRBitSet : NSObject < NSMutableCopying > {
	__strong CFMutableBitVectorRef bitVector;
}

#pragma mark Class Methods

+ (ANTLRBitSet *) newANTLRBitSet;
+ (ANTLRBitSet *) newANTLRBitSetWithType:(ANTLRTokenType)type;
/** Construct a ANTLRBitSet given the size
 * @param nbits The size of the ANTLRBitSet in bits
 */
+ (ANTLRBitSet *) newANTLRBitSetWithNBits:(NSUInteger)nbits;
+ (ANTLRBitSet *) newANTLRBitSetWithArray:(AMutableArray *)types;
+ (ANTLRBitSet *) newANTLRBitSetWithBits:(const unsigned long long *)theBits Count:(NSUInteger)longCount;

+ (ANTLRBitSet *) of:(NSUInteger)el;
+ (ANTLRBitSet *) of:(NSUInteger)a And2:(NSUInteger)b;
+ (ANTLRBitSet *) of:(NSUInteger)a And2:(NSUInteger)b And3:(NSUInteger)c;
+ (ANTLRBitSet *) of:(NSUInteger)a And2:(NSUInteger)b And3:(NSUInteger)c And4:(NSUInteger)d;

#pragma mark Initializer

- (ANTLRBitSet *) init;
- (ANTLRBitSet *) initWithType:(ANTLRTokenType)type;
- (ANTLRBitSet *) initWithNBits:(NSUInteger)nbits;
- (ANTLRBitSet *) initWithBitVector:(CFMutableBitVectorRef)theBitVector;
- (ANTLRBitSet *) initWithBits:(const unsigned long long const*)theBits Count:(NSUInteger)theCount;
- (ANTLRBitSet *) initWithArrayOfBits:(NSArray *)theArray;

#pragma mark Operations
- (ANTLRBitSet *) or:(ANTLRBitSet *) aBitSet;
- (void) orInPlace:(ANTLRBitSet *) aBitSet;
- (void) add:(NSUInteger) bit;
- (void) remove:(NSUInteger) bit;
- (void) setAllBits:(BOOL) aState;

- (NSInteger) numBits;
- (NSUInteger) size;
- (void) setSize:(NSUInteger) noOfWords;

#pragma mark Informational
- (unsigned long long) bitMask:(NSUInteger) bitNumber;
- (BOOL) member:(NSUInteger)bitNumber;
- (BOOL) isNil;
- (NSString *) toString;
- (NSString *) description;

#pragma mark NSCopying support

- (id) mutableCopyWithZone:(NSZone *) theZone;


//private
- (CFMutableBitVectorRef) _bitVector;
@property (getter=_bitVector) CFMutableBitVectorRef bitVector;
@end