esengine/source/src/ECS/Utils/BitSet.ts

module es {
    /**
     * 这个类可以从两方面来考虑。你可以把它看成一个位向量或者一组非负整数。这个名字有点误导人。
     *
     * 它是由一个位向量实现的，但同样可以把它看成是一个非负整数的集合;集合中的每个整数由对应索引处的集合位表示。该结构的大小由集合中的最大整数决定。
     */
    export class BitSet{
        private static LONG_MASK: number = 0x3f;
        private _bits: number[];

        constructor(nbits: number = 64){
            let length = nbits >> 6;
            if ((nbits & BitSet.LONG_MASK) != 0)
                length ++;

            this._bits = new Array(length);
        }

        public and(bs: BitSet){
            let max = Math.min(this._bits.length, bs._bits.length);
            let i;
            for (let i = 0; i < max; ++i)
                this._bits[i] &= bs._bits[i];

            while (i < this._bits.length)
                this._bits[i ++] = 0;
        }

        public andNot(bs: BitSet){
            let i = Math.min(this._bits.length, bs._bits.length);
            while(--i >= 0)
                this._bits[i] &= ~bs._bits[i];
        }

        public cardinality(): number{
            let card = 0;
            for (let i = this._bits.length - 1; i >= 0; i --){
                let a = this._bits[i];

                if (a == 0)
                    continue;

                if (a == -1){
                    card += 64;
                    continue;
                }

                a = ((a >> 1) & 0x5555555555555555) + (a & 0x5555555555555555);
                a = ((a >> 2) & 0x3333333333333333) + (a & 0x3333333333333333);
                let b = ((a >> 32) + a);
                b = ((b >> 4) & 0x0f0f0f0f) + (b & 0x0f0f0f0f);
                b = ((b >> 8) & 0x00ff00ff) + (b & 0x00ff00ff);
                card += ((b >> 16) & 0x0000ffff) + (b & 0x0000ffff);
            }

            return card;
        }

        public clear(pos?: number){
            if (pos != undefined){
                let offset = pos >> 6;
                this.ensure(offset);
                this._bits[offset] &= ~(1 << pos);
            }else{
                for (let i = 0; i < this._bits.length; i ++)
                    this._bits[i] = 0;
            }
        }

        private ensure(lastElt: number){
            if (lastElt >= this._bits.length){
                let nd = new Number[lastElt + 1];
                nd = this._bits.copyWithin(0, 0, this._bits.length);
                this._bits = nd;
            }
        }

        public get(pos: number): boolean{
            let offset = pos >> 6;
            if (offset >= this._bits.length)
                return false;

            return (this._bits[offset] & (1 << pos)) != 0;
        }

        public intersects(set: BitSet){
            let i = Math.min(this._bits.length, set._bits.length);
            while (--i >= 0){
                if ((this._bits[i] & set._bits[i]) != 0)
                    return true;
            }

            return false;
        }

        public isEmpty(): boolean{
            for (let i = this._bits.length - 1; i >= 0; i --){
                if (this._bits[i])
                    return false;
            }

            return true;
        }

        public nextSetBit(from: number){
            let offset = from >> 6;
            let mask = 1 << from;
            while (offset < this._bits.length){
                let h = this._bits[offset];
                do {
                    if ((h & mask) != 0)
                        return from;

                    mask <<= 1;
                    from ++;
                } while (mask != 0);

                mask = 1;
                offset ++;
            }

            return -1;
        }

        public set(pos: number, value: boolean = true){
            if (value){
                let offset = pos >> 6;
                this.ensure(offset);
                this._bits[offset] |= 1 << pos;
            }else{
                this.clear(pos);
            }
        }
    }
}