重构位掩码数据结构,修复部分方法未考虑扩展位的问题

- 所有操作均考虑扩展位、扩展长度不一致的使用场景,无感扩容掩码位
- 使用定长数组存储高低位,遍历友好,为高效哈希计算提供结构支持
- 补充相应单元测试,覆盖所有方法及分支
This commit is contained in:
MirageTank
2025-10-03 16:55:07 +08:00
parent 316527c459
commit da8b7cf601
6 changed files with 482 additions and 529 deletions

View File

@@ -54,10 +54,7 @@ export class ComponentRegistry {
const typeName = getComponentTypeName(componentType); const typeName = getComponentTypeName(componentType);
throw new Error(`Component type ${typeName} is not registered`); throw new Error(`Component type ${typeName} is not registered`);
} }
return BitMask64Utils.create(bitIndex);
const mask: BitMask64Data = { lo: 0, hi: 0 };
BitMask64Utils.setBitExtended(mask, bitIndex);
return mask;
} }
/** /**

View File

@@ -4,9 +4,8 @@ import { ComponentRegistry, ComponentType } from './ComponentStorage';
import { BitMask64Utils, BitMask64Data } from '../Utils/BigIntCompatibility'; import { BitMask64Utils, BitMask64Data } from '../Utils/BigIntCompatibility';
import { createLogger } from '../../Utils/Logger'; import { createLogger } from '../../Utils/Logger';
import { getComponentTypeName } from '../Decorators'; import { getComponentTypeName } from '../Decorators';
import { Archetype, ArchetypeSystem } from './ArchetypeSystem';
import { ComponentPoolManager } from './ComponentPool'; import { ComponentTypeManager } from "../Utils";
import { ArchetypeSystem, Archetype, ArchetypeQueryResult } from './ArchetypeSystem';
/** /**
* 查询条件类型 * 查询条件类型
@@ -787,8 +786,7 @@ export class QuerySystem {
private createComponentMask(componentTypes: ComponentType[]): BitMask64Data { private createComponentMask(componentTypes: ComponentType[]): BitMask64Data {
// 生成缓存键 // 生成缓存键
const cacheKey = componentTypes.map(t => { const cacheKey = componentTypes.map(t => {
const name = getComponentTypeName(t); return getComponentTypeName(t);
return name;
}).sort().join(','); }).sort().join(',');
// 检查缓存 // 检查缓存
@@ -797,27 +795,10 @@ export class QuerySystem {
return cached; return cached;
} }
let mask = BitMask64Utils.clone(BitMask64Utils.ZERO); let mask = ComponentTypeManager.instance.getEntityBits(componentTypes);
let hasValidComponents = false;
for (const type of componentTypes) {
try {
const bitMask = ComponentRegistry.getBitMask(type);
BitMask64Utils.orInPlace(mask, bitMask);
hasValidComponents = true;
} catch (error) {
this._logger.warn(`组件类型 ${getComponentTypeName(type)} 未注册,跳过`);
}
}
// 如果没有有效的组件类型,返回一个不可能匹配的掩码
if (!hasValidComponents) {
mask = { lo: 0xFFFFFFFF, hi: 0xFFFFFFFF };
}
// 缓存结果 // 缓存结果
this.componentMaskCache.set(cacheKey, mask); this.componentMaskCache.set(cacheKey, mask.getValue());
return mask; return mask.getValue();
} }
/** /**

View File

@@ -191,7 +191,7 @@ export class Entity {
const maxBitIndex = ComponentRegistry.getRegisteredCount(); const maxBitIndex = ComponentRegistry.getRegisteredCount();
for (let bitIndex = 0; bitIndex < maxBitIndex; bitIndex++) { for (let bitIndex = 0; bitIndex < maxBitIndex; bitIndex++) {
if (BitMask64Utils.getBitExtended(mask, bitIndex)) { if (BitMask64Utils.getBit(mask, bitIndex)) {
const componentType = ComponentRegistry.getTypeByBitIndex(bitIndex); const componentType = ComponentRegistry.getTypeByBitIndex(bitIndex);
if (componentType) { if (componentType) {
let component: Component | null = null; let component: Component | null = null;
@@ -504,7 +504,7 @@ export class Entity {
this._localComponents.delete(componentType); this._localComponents.delete(componentType);
// 更新位掩码 // 更新位掩码
BitMask64Utils.clearBitExtended(this._componentMask, bitIndex); BitMask64Utils.clearBit(this._componentMask, bitIndex);
// 使缓存失效 // 使缓存失效
this._componentCache = null; this._componentCache = null;

View File

@@ -1,31 +1,31 @@
/**
* 位枚举
*/
export enum SegmentPart {
LOW = 0,
HIGH = 1,
}
/** /**
* 基础 64 位段结构 * 基础 64 位段结构
* [低32位高32位]
*/ */
export interface BitMask64Segment { export type BitMask64Segment = [number,number]
/** 低32位bit 0-31 */
lo: number;
/** 高32位bit 32-63 */
hi: number;
}
/** /**
* 位掩码数据结构 * 位掩码数据结构
* 基础模式64位使用 lo + hi 存储 64 位segments 为空 * 基础模式64位使用 base[lo , hi] 存储 64 位segments 为空
* 扩展模式128+位lo + hi 作为第一段segments 存储额外的 64 位段 * 扩展模式128+位):base[lo , hi] 作为第一段segments 存储额外的 64 位段
* segments[0] 对应 bit 64-127segments[1] 对应 bit 128-191以此类推 * segments[0] 对应 bit 64-127segments[1] 对应 bit 128-191以此类推
*/ */
export interface BitMask64Data { export interface BitMask64Data {
/** 低32位bit 0-31 */ base: BitMask64Segment;
lo: number;
/** 高32位bit 32-63 */
hi: number;
/** 扩展段数组,每个元素是一个 64 位段,用于超过 64 位的场景 */ /** 扩展段数组,每个元素是一个 64 位段,用于超过 64 位的场景 */
segments?: BitMask64Segment[]; segments?: BitMask64Segment[];
} }
export class BitMask64Utils { export class BitMask64Utils {
/** 零掩码常量所有位都为0 */ /** 零掩码常量所有位都为0 */
public static readonly ZERO: BitMask64Data = { lo: 0, hi: 0 }; public static readonly ZERO: Readonly<BitMask64Data> = { base: [0, 0], segments: undefined };
/** /**
* 根据位索引创建64位掩码 * 根据位索引创建64位掩码
@@ -34,15 +34,12 @@ export class BitMask64Utils {
* @throws 当位索引超出范围时抛出错误 * @throws 当位索引超出范围时抛出错误
*/ */
public static create(bitIndex: number): BitMask64Data { public static create(bitIndex: number): BitMask64Data {
if (bitIndex < 0 || bitIndex >= 64) { if (bitIndex < 0) {
throw new Error(`Bit index ${bitIndex} out of range [0, 63]`); throw new Error(`Bit index ${bitIndex} out of range [0, ∞)`);
}
if (bitIndex < 32) {
return { lo: 1 << bitIndex, hi: 0 };
} else {
return { lo: 0, hi: 1 << (bitIndex - 32) };
} }
const mask: BitMask64Data = { base: [0, 0], segments: undefined };
BitMask64Utils.setBit(mask, bitIndex);
return mask;
} }
/** /**
@@ -51,7 +48,7 @@ export class BitMask64Utils {
* @returns 低32位为输入值、高32位为0的掩码 * @returns 低32位为输入值、高32位为0的掩码
*/ */
public static fromNumber(value: number): BitMask64Data { public static fromNumber(value: number): BitMask64Data {
return { lo: value >>> 0, hi: 0 }; return { base: [value >>> 0, 0], segments: undefined};
} }
/** /**
@@ -61,75 +58,50 @@ export class BitMask64Utils {
* @returns 如果掩码包含bits中的任意位则返回true * @returns 如果掩码包含bits中的任意位则返回true
*/ */
public static hasAny(mask: BitMask64Data, bits: BitMask64Data): boolean { public static hasAny(mask: BitMask64Data, bits: BitMask64Data): boolean {
// 检查第一个 64 位段 const bitsBase = bits.base;
if ((mask.lo & bits.lo) !== 0 || (mask.hi & bits.hi) !== 0) { const maskBase = mask.base;
return true; const bitsSegments = bits.segments;
} const maskSegments = mask.segments;
const baseHasAny = (maskBase[SegmentPart.LOW] & bitsBase[SegmentPart.LOW]) !== 0 || (maskBase[SegmentPart.HIGH] & bitsBase[SegmentPart.HIGH]) !== 0;
// 如果 bits 没有扩展段,检查完成 // 基础区段就包含指定的位,或任意一个参数不含扩展段,直接短路
if (!bits.segments || bits.segments.length === 0) { if(baseHasAny || !bitsSegments || !maskSegments) return baseHasAny;
return false; // 额外检查扩展区域是否包含指定的位 - 如果bitsSegments[index]不存在会被转为NaNNaN的位运算始终返回0
} return maskSegments.some((seg, index) => (seg[SegmentPart.LOW] & bitsSegments[index][SegmentPart.LOW]) !== 0 || (seg[SegmentPart.HIGH] & bitsSegments[index][SegmentPart.HIGH]) !== 0)
// 如果 bits 有扩展段但 mask 没有,返回 false
if (!mask.segments || mask.segments.length === 0) {
return false;
}
// 检查每个扩展段
const minSegments = Math.min(mask.segments.length, bits.segments.length);
for (let i = 0; i < minSegments; i++) {
const maskSeg = mask.segments[i];
const bitsSeg = bits.segments[i];
if ((maskSeg.lo & bitsSeg.lo) !== 0 || (maskSeg.hi & bitsSeg.hi) !== 0) {
return true;
}
}
return false;
} }
/** /**
* 检查掩码是否包含所有指定的位 * 检查掩码是否包含所有指定的位
* 支持扩展模式,自动处理超过 64 位的掩码
* @param mask 要检查的掩码 * @param mask 要检查的掩码
* @param bits 指定的位模式 * @param bits 指定的位模式
* @returns 如果掩码包含bits中的所有位则返回true * @returns 如果掩码包含bits中的所有位则返回true
*/ */
public static hasAll(mask: BitMask64Data, bits: BitMask64Data): boolean { public static hasAll(mask: BitMask64Data, bits: BitMask64Data): boolean {
// 检查第一个 64 位段 const maskBase = mask.base;
if ((mask.lo & bits.lo) !== bits.lo || (mask.hi & bits.hi) !== bits.hi) { const bitsBase = bits.base;
return false; const maskSegments = mask.segments;
} const bitsSegments = bits.segments;
const baseHasAll = (maskBase[SegmentPart.LOW] & bitsBase[SegmentPart.LOW]) === bitsBase[SegmentPart.LOW] && (maskBase[SegmentPart.HIGH] & bitsBase[SegmentPart.HIGH]) === bitsBase[SegmentPart.HIGH];
// 基础区域就不包含指定的位或bits没有扩展区段直接短路。
if(!baseHasAll || !bitsSegments) return baseHasAll;
// 如果 bits 没有扩展段,检查完成 // 扩展区段的hasAll匹配逻辑
if (!bits.segments || bits.segments.length === 0) { const maskSegmentsLength = maskSegments?.length ?? 0;
return true; // 对mask/bits中都存在的区段进行hasAll判断
} if(maskSegments){
for (let i = 0; i < Math.min(maskSegmentsLength,bitsSegments.length); i++) {
// 如果 bits 有扩展段但 mask 没有,返回 false if((maskSegments[i][SegmentPart.LOW] & bitsSegments[i][SegmentPart.LOW]) !== bitsSegments[i][SegmentPart.LOW] || (maskSegments[i][SegmentPart.HIGH] & bitsSegments[i][SegmentPart.HIGH]) !== bitsSegments[i][SegmentPart.HIGH]){
if (!mask.segments || mask.segments.length === 0) { // 存在不匹配的位,直接短路
// 检查 bits 的扩展段是否全为 0
return bits.segments.every(seg => BitMask64Utils.isZero(seg));
}
// 检查每个扩展段
const minSegments = Math.min(mask.segments.length, bits.segments.length);
for (let i = 0; i < minSegments; i++) {
const maskSeg = mask.segments[i];
const bitsSeg = bits.segments[i];
if ((maskSeg.lo & bitsSeg.lo) !== bitsSeg.lo || (maskSeg.hi & bitsSeg.hi) !== bitsSeg.hi) {
return false; return false;
} }
} }
}
// 如果 bits 有更多段,检查这些段是否为空 // 对mask中不存在但bits中存在的区段进行isZero判断
for (let i = minSegments; i < bits.segments.length; i++) { for (let i = maskSegmentsLength; i < bitsSegments.length; i++) {
if (!BitMask64Utils.isZero(bits.segments[i])) { if(bitsSegments[i][SegmentPart.LOW] !== 0 || bitsSegments[i][SegmentPart.HIGH] !== 0){
// 存在不为0的区段直接短路
return false; return false;
} }
} }
return true; return true;
} }
@@ -140,7 +112,15 @@ export class BitMask64Utils {
* @returns 如果掩码不包含bits中的任何位则返回true * @returns 如果掩码不包含bits中的任何位则返回true
*/ */
public static hasNone(mask: BitMask64Data, bits: BitMask64Data): boolean { public static hasNone(mask: BitMask64Data, bits: BitMask64Data): boolean {
return (mask.lo & bits.lo) === 0 && (mask.hi & bits.hi) === 0; const maskBase = mask.base;
const bitsBase = bits.base;
const maskSegments = mask.segments;
const bitsSegments = bits.segments;
const baseHasNone = (maskBase[SegmentPart.LOW] & bitsBase[SegmentPart.LOW]) === 0 && (maskBase[SegmentPart.HIGH] & bitsBase[SegmentPart.HIGH]) === 0;
//不含扩展区域或基础区域就包含指定的位或bits不含拓展区段直接短路。
if(!maskSegments || !baseHasNone || !bitsSegments) return baseHasNone;
// 额外检查扩展区域是否都包含指定的位 - 此时bitsSegments存在,如果bitsSegments[index]不存在会被转为NaNNaN的位运算始终返回0
return maskSegments.every((seg, index) => (seg[SegmentPart.LOW] & bitsSegments[index][SegmentPart.LOW]) === 0 && (seg[SegmentPart.HIGH] & bitsSegments[index][SegmentPart.HIGH]) === 0);
} }
/** /**
@@ -148,8 +128,14 @@ export class BitMask64Utils {
* @param mask 要检查的掩码 * @param mask 要检查的掩码
* @returns 如果掩码所有位都为0则返回true * @returns 如果掩码所有位都为0则返回true
*/ */
public static isZero(mask: BitMask64Data | BitMask64Segment): boolean { public static isZero(mask: BitMask64Data): boolean {
return mask.lo === 0 && mask.hi === 0; const baseIsZero = mask.base[SegmentPart.LOW] === 0 && mask.base[SegmentPart.HIGH] === 0;
if(!mask.segments || !baseIsZero){
// 不含扩展区域或基础区域值就不为0直接短路
return baseIsZero;
}
// 额外检查扩展区域是否都为0
return mask.segments.every(seg => seg[SegmentPart.LOW] === 0 && seg[SegmentPart.HIGH] === 0);
} }
/** /**
@@ -159,42 +145,82 @@ export class BitMask64Utils {
* @returns 如果两个掩码完全相等则返回true * @returns 如果两个掩码完全相等则返回true
*/ */
public static equals(a: BitMask64Data, b: BitMask64Data): boolean { public static equals(a: BitMask64Data, b: BitMask64Data): boolean {
return a.lo === b.lo && a.hi === b.hi; let baseEquals = a.base[SegmentPart.LOW] === b.base[SegmentPart.LOW] && a.base[SegmentPart.HIGH] === b.base[SegmentPart.HIGH];
// base不相等或ab都没有扩展区域位直接返回base比较结果
if(!baseEquals || (!a.segments && !b.segments)) return baseEquals;
// 不能假设ab的segments都存在或长度相同.
const aSegments = a.segments ?? [];
const bSegments = b.segments ?? [];
for (let i = 0; i < Math.max(aSegments.length, bSegments.length); i++) {
const aSeg: BitMask64Segment | undefined = aSegments[i]; // 可能为undefined
const bSeg: BitMask64Segment | undefined = bSegments[i]; // 可能为undefined
if(aSeg && !bSeg){
//bSeg不存在则必须要求aSeg全为0
if(aSeg[SegmentPart.LOW] !== 0 || aSeg[SegmentPart.HIGH] !== 0) return false;
}else if(!aSeg && bSeg){
//aSeg不存在则必须要求bSeg全为0
if(bSeg[SegmentPart.LOW] !== 0 || bSeg[SegmentPart.HIGH] !== 0) return false;
}else{
//理想状态aSeg/bSeg都存在
if(aSeg[SegmentPart.LOW] !== bSeg[SegmentPart.LOW] || aSeg[SegmentPart.HIGH] !== bSeg[SegmentPart.HIGH]) return false;
}
}
return true;
} }
/** /**
* 设置掩码中指定位为1 * 设置掩码中指定位为1,必要时自动扩展
* @param mask 要修改的掩码(原地修改) * @param mask 要修改的掩码(原地修改)
* @param bitIndex 位索引,范围 [0, 63] * @param bitIndex 位索引,不小于零
* @throws 当位索引超出范围时抛出错误 * @throws 当位索引超出范围时抛出错误
*/ */
public static setBit(mask: BitMask64Data, bitIndex: number): void { public static setBit(mask: BitMask64Data, bitIndex: number): void {
if (bitIndex < 0 || bitIndex >= 64) { if (bitIndex < 0) {
throw new Error(`Bit index ${bitIndex} out of range [0, 63]`); throw new Error(`Bit index ${bitIndex} out of range [0, 63]`);
} }
const targetSeg = BitMask64Utils.getSegmentByBitIndex(mask, bitIndex)!;
if (bitIndex < 32) { const mod = bitIndex & 63; // bitIndex % 64 优化方案
mask.lo |= (1 << bitIndex); if(mod < 32){
targetSeg[SegmentPart.LOW] |= (1 << mod);
} else { } else {
mask.hi |= (1 << (bitIndex - 32)); targetSeg[SegmentPart.HIGH] |= (1 << (mod - 32));
} }
} }
/** /**
* 清除掩码中指定位0 * 获取掩码中指定位,如果位超出当前掩码的区段长度,则直接返回0
* @param mask 掩码
* @param bitIndex 位索引,不小于零
*/
public static getBit(mask: BitMask64Data, bitIndex: number): boolean {
if (bitIndex < 0) {
return false;
}
const targetSeg = BitMask64Utils.getSegmentByBitIndex(mask, bitIndex, false);
if(!targetSeg) return false;
const mod = bitIndex & 63; // bitIndex % 64 优化方案
if(mod < 32){
return (targetSeg[SegmentPart.LOW] & (1 << mod)) !== 0;
} else {
return (targetSeg[SegmentPart.HIGH] & (1 << (mod - 32))) !== 0;
}
}
/**
* 清除掩码中指定位为0如果位超出当前掩码的区段长度则什么也不做
* @param mask 要修改的掩码(原地修改) * @param mask 要修改的掩码(原地修改)
* @param bitIndex 位索引,范围 [0, 63] * @param bitIndex 位索引,不小于0
* @throws 当位索引超出范围时抛出错误
*/ */
public static clearBit(mask: BitMask64Data, bitIndex: number): void { public static clearBit(mask: BitMask64Data, bitIndex: number): void {
if (bitIndex < 0 || bitIndex >= 64) { if (bitIndex < 0) {
throw new Error(`Bit index ${bitIndex} out of range [0, 63]`); throw new Error(`Bit index ${bitIndex} out of range [0, 63]`);
} }
const targetSeg = BitMask64Utils.getSegmentByBitIndex(mask, bitIndex, false);
if (bitIndex < 32) { if(!targetSeg) return;
mask.lo &= ~(1 << bitIndex); const mod = bitIndex & 63; // bitIndex % 64 优化方案
if(mod < 32){
targetSeg[SegmentPart.LOW] &= ~(1 << mod);
} else { } else {
mask.hi &= ~(1 << (bitIndex - 32)); targetSeg[SegmentPart.HIGH] &= ~(1 << (mod - 32));
} }
} }
@@ -204,24 +230,26 @@ export class BitMask64Utils {
* @param other 用于按位或的掩码 * @param other 用于按位或的掩码
*/ */
public static orInPlace(target: BitMask64Data, other: BitMask64Data): void { public static orInPlace(target: BitMask64Data, other: BitMask64Data): void {
target.lo |= other.lo; target.base[SegmentPart.LOW] |= other.base[SegmentPart.LOW];
target.hi |= other.hi; target.base[SegmentPart.HIGH] |= other.base[SegmentPart.HIGH];
// 处理扩展段 // 处理扩展段
if (other.segments && other.segments.length > 0) { const otherSegments = other.segments;
if (otherSegments && otherSegments.length > 0) {
if (!target.segments) { if (!target.segments) {
target.segments = []; target.segments = [];
} }
const targetSegments = target.segments;
// 确保 target 有足够的段 // 确保 target 有足够的段
while (target.segments.length < other.segments.length) { while (targetSegments.length < otherSegments.length) {
target.segments.push({ lo: 0, hi: 0 }); targetSegments.push([0, 0]);
} }
// 对每个段执行或操作 // 对每个段执行或操作
for (let i = 0; i < other.segments.length; i++) { for (let i = 0; i < otherSegments.length; i++) {
target.segments[i].lo |= other.segments[i].lo; targetSegments[i][SegmentPart.LOW] |= otherSegments[i][SegmentPart.LOW];
target.segments[i].hi |= other.segments[i].hi; targetSegments[i][SegmentPart.HIGH] |= otherSegments[i][SegmentPart.HIGH];
} }
} }
} }
@@ -232,8 +260,27 @@ export class BitMask64Utils {
* @param other 用于按位与的掩码 * @param other 用于按位与的掩码
*/ */
public static andInPlace(target: BitMask64Data, other: BitMask64Data): void { public static andInPlace(target: BitMask64Data, other: BitMask64Data): void {
target.lo &= other.lo; target.base[SegmentPart.LOW] &= other.base[SegmentPart.LOW];
target.hi &= other.hi; target.base[SegmentPart.HIGH] &= other.base[SegmentPart.HIGH];
// 处理扩展段
const otherSegments = other.segments;
if (otherSegments && otherSegments.length > 0) {
if (!target.segments) {
target.segments = [];
}
const targetSegments = target.segments;
// 确保 target 有足够的段
while (targetSegments.length < otherSegments.length) {
targetSegments.push([0, 0]);
}
// 对每个段执行与操作
for (let i = 0; i < otherSegments.length; i++) {
targetSegments[i][SegmentPart.LOW] &= otherSegments[i][SegmentPart.LOW];
targetSegments[i][SegmentPart.HIGH] &= otherSegments[i][SegmentPart.HIGH];
}
}
} }
/** /**
@@ -242,8 +289,25 @@ export class BitMask64Utils {
* @param other 用于按位异或的掩码 * @param other 用于按位异或的掩码
*/ */
public static xorInPlace(target: BitMask64Data, other: BitMask64Data): void { public static xorInPlace(target: BitMask64Data, other: BitMask64Data): void {
target.lo ^= other.lo; target.base[SegmentPart.LOW] ^= other.base[SegmentPart.LOW];
target.hi ^= other.hi; target.base[SegmentPart.HIGH] ^= other.base[SegmentPart.HIGH];
// 处理扩展段
const otherSegments = other.segments;
if (!otherSegments || otherSegments.length == 0) return;
if (!target.segments) target.segments = [];
const targetSegments = target.segments;
// 确保 target 有足够的段
while (targetSegments.length < otherSegments.length) {
targetSegments.push([0, 0]);
}
// 对每个段执行异或操作
for (let i = 0; i < otherSegments.length; i++) {
targetSegments[i][SegmentPart.LOW] ^= otherSegments[i][SegmentPart.LOW];
targetSegments[i][SegmentPart.HIGH] ^= otherSegments[i][SegmentPart.HIGH];
}
} }
/** /**
@@ -251,18 +315,43 @@ export class BitMask64Utils {
* @param mask 要清除的掩码(原地修改) * @param mask 要清除的掩码(原地修改)
*/ */
public static clear(mask: BitMask64Data): void { public static clear(mask: BitMask64Data): void {
mask.lo = 0; mask.base[SegmentPart.LOW] = 0;
mask.hi = 0; mask.base[SegmentPart.HIGH] = 0;
for (let i = 0; i < (mask.segments?.length ?? 0); i++) {
const seg = mask.segments![i];
seg[SegmentPart.LOW] = 0;
seg[SegmentPart.HIGH] = 0;
}
} }
/** /**
* 将源掩码的值复制到目标掩码 * 将源掩码的值复制到目标掩码如果source包含扩展段则target也会至少扩展到source扩展段的长度
* @param source 源掩码 * @param source 源掩码
* @param target 目标掩码(原地修改) * @param target 目标掩码(原地修改)
*/ */
public static copy(source: BitMask64Data, target: BitMask64Data): void { public static copy(source: BitMask64Data, target: BitMask64Data): void {
target.lo = source.lo; BitMask64Utils.clear(target);
target.hi = source.hi; target.base[SegmentPart.LOW] = source.base[SegmentPart.LOW];
target.base[SegmentPart.HIGH] = source.base[SegmentPart.HIGH];
// source没有扩展段直接退出
if(!source.segments || source.segments.length == 0) return;
// 没有拓展段,则直接复制数组
if(!target.segments){
target.segments = source.segments.map(seg => [...seg]);
return;
}
// source有扩展段target扩展段不足则补充长度
const copyLength = source.segments.length - target.segments.length;
for (let i = 0; i < copyLength; i++) {
target.segments.push([0,0]);
}
// 逐个重写
for (let i = 0; i < length; i++) {
const targetSeg = target.segments![i];
const sourSeg = source.segments![i];
targetSeg[SegmentPart.LOW] = sourSeg[SegmentPart.LOW];
targetSeg[SegmentPart.HIGH] = sourSeg[SegmentPart.HIGH];
}
} }
/** /**
@@ -271,36 +360,57 @@ export class BitMask64Utils {
* @returns 新的掩码对象,内容与源掩码相同 * @returns 新的掩码对象,内容与源掩码相同
*/ */
public static clone(mask: BitMask64Data): BitMask64Data { public static clone(mask: BitMask64Data): BitMask64Data {
return { lo: mask.lo, hi: mask.hi }; return {
base: mask.base.slice() as BitMask64Segment,
segments: mask.segments ? mask.segments.map(seg => [...seg]) : undefined
};
} }
/** /**
* 将掩码转换为字符串表示 * 将掩码转换为字符串表示,每个区段之间将使用空格分割。
* @param mask 要转换的掩码 * @param mask 要转换的掩码
* @param radix 进制支持2二进制或16十六进制默认为2 * @param radix 进制支持2二进制或16十六进制默认为2其他的值被视为2
* @param printHead 打印头
* @returns 掩码的字符串表示二进制不带前缀十六进制带0x前缀 * @returns 掩码的字符串表示二进制不带前缀十六进制带0x前缀
* @throws 当进制不支持时抛出错误
*/ */
public static toString(mask: BitMask64Data, radix: number = 2): string { public static toString(mask: BitMask64Data, radix: 2 | 16 = 2, printHead: boolean = false): string {
if(radix != 2 && radix != 16) radix = 2;
const totalLength = mask.segments?.length ?? 0;
let result: string = '';
if(printHead){
let paddingLength = 0;
if(radix === 2){ if(radix === 2){
if (mask.hi === 0) { paddingLength = 64 + 1 + 1;
return mask.lo.toString(2);
}else{ }else{
const hiBits = mask.hi.toString(2); paddingLength = 16 + 2 + 1;
const loBits = mask.lo.toString(2).padStart(32, '0');
return hiBits + loBits;
} }
} else if (radix === 16) { for (let i = 0; i <= totalLength; i++) {
if (mask.hi === 0) { const title = i === 0 ? '0 (Base):' : `${i} (${64 * i}):`;
return '0x' + mask.lo.toString(16).toUpperCase(); result += title.toString().padEnd(paddingLength);
}
result += '\n';
}
for (let i = -1; i < totalLength; i++) {
let segResult = '';
const bitMaskData = i == -1 ? mask.base : mask.segments![i];
let hi = bitMaskData[SegmentPart.HIGH];
let lo = bitMaskData[SegmentPart.LOW];
if(radix == 2){
const hiBits = hi.toString(2).padStart(32, '0');
const loBits = lo.toString(2).padStart(32, '0');
segResult = hiBits + '_' + loBits; //高低位之间使用_隔离
}else{ }else{
const hiBits = mask.hi.toString(16).toUpperCase(); const hiBits = hi.toString(16).toUpperCase().padStart(8, '0');
const loBits = mask.lo.toString(16).toUpperCase().padStart(8, '0'); const loBits = lo.toString(16).toUpperCase().padStart(8, '0');
return '0x' + hiBits + loBits; segResult = '0x' + hiBits + loBits;
} }
} else { if(i === -1)
throw new Error('Only radix 2 and 16 are supported'); result += segResult;
else
result += ' ' + segResult; // 不同段之间使用空格隔离
} }
return result;
} }
/** /**
@@ -310,126 +420,49 @@ export class BitMask64Utils {
*/ */
public static popCount(mask: BitMask64Data): number { public static popCount(mask: BitMask64Data): number {
let count = 0; let count = 0;
let lo = mask.lo; for (let i = -1; i < (mask.segments?.length ?? 0); i++) {
let hi = mask.hi; const bitMaskData = i == -1 ? mask.base : mask.segments![i];
let lo = bitMaskData[SegmentPart.LOW];
let hi = bitMaskData[SegmentPart.HIGH];
while (lo) { while (lo) {
lo &= lo - 1; lo &= lo - 1;
count++; count++;
} }
while (hi) { while (hi) {
hi &= hi - 1; hi &= hi - 1;
count++; count++;
} }
}
return count; return count;
} }
/** /**
* 设置扩展位(支持超过 64 位的索引) * 获取包含目标位的BitMask64Segment
* @param mask 要修改的掩码(原地修改) * @param mask 要操作的掩码
* @param bitIndex 位索引(可以超过 63 * @param bitIndex 目标
* @param createNewSegment 如果bitIndex超过了当前范围是否自动补充扩展区域默认为真
* @private
*/ */
public static setBitExtended(mask: BitMask64Data, bitIndex: number): void { private static getSegmentByBitIndex(mask: BitMask64Data,bitIndex: number, createNewSegment: boolean = true): BitMask64Segment | null{
if (bitIndex < 0) { if(bitIndex <= 63){
throw new Error('Bit index cannot be negative'); // 基础位
} return mask.base;
if (bitIndex < 64) {
BitMask64Utils.setBit(mask, bitIndex);
return;
}
// 计算段索引和段内位索引
const segmentIndex = Math.floor(bitIndex / 64) - 1;
const localBitIndex = bitIndex % 64;
// 确保 segments 数组存在
if (!mask.segments) {
mask.segments = [];
}
// 扩展 segments 数组
while (mask.segments.length <= segmentIndex) {
mask.segments.push({ lo: 0, hi: 0 });
}
// 设置对应段的位
const segment = mask.segments[segmentIndex];
if (localBitIndex < 32) {
segment.lo |= (1 << localBitIndex);
}else{ }else{
segment.hi |= (1 << (localBitIndex - 32)); // 扩展位
let segments = mask.segments;
if(!segments) {
if(!createNewSegment) return null;
segments = mask.segments = [];
}
const targetSegIndex = (bitIndex >> 6) - 1; // Math.floor(bitIndex / 64) - 1的位运算优化
if(segments.length <= targetSegIndex){
if(!createNewSegment) return null;
const diff = targetSegIndex - segments.length + 1;
for (let i = 0; i < diff; i++) {
segments.push([0, 0]);
} }
} }
return segments[targetSegIndex];
/**
* 获取扩展位(支持超过 64 位的索引)
* @param mask 要检查的掩码
* @param bitIndex 位索引(可以超过 63
* @returns 如果位被设置则返回 true
*/
public static getBitExtended(mask: BitMask64Data, bitIndex: number): boolean {
if (bitIndex < 0) {
return false;
}
if (bitIndex < 64) {
const testMask = BitMask64Utils.create(bitIndex);
return BitMask64Utils.hasAny(mask, testMask);
}
if (!mask.segments) {
return false;
}
const segmentIndex = Math.floor(bitIndex / 64) - 1;
if (segmentIndex >= mask.segments.length) {
return false;
}
const localBitIndex = bitIndex % 64;
const segment = mask.segments[segmentIndex];
if (localBitIndex < 32) {
return (segment.lo & (1 << localBitIndex)) !== 0;
} else {
return (segment.hi & (1 << (localBitIndex - 32))) !== 0;
}
}
/**
* 清除扩展位(支持超过 64 位的索引)
* @param mask 要修改的掩码(原地修改)
* @param bitIndex 位索引(可以超过 63
*/
public static clearBitExtended(mask: BitMask64Data, bitIndex: number): void {
if (bitIndex < 0) {
return;
}
if (bitIndex < 64) {
BitMask64Utils.clearBit(mask, bitIndex);
return;
}
if (!mask.segments) {
return;
}
const segmentIndex = Math.floor(bitIndex / 64) - 1;
if (segmentIndex >= mask.segments.length) {
return;
}
const localBitIndex = bitIndex % 64;
const segment = mask.segments[segmentIndex];
if (localBitIndex < 32) {
segment.lo &= ~(1 << localBitIndex);
} else {
segment.hi &= ~(1 << (localBitIndex - 32));
} }
} }
} }

View File

@@ -1,9 +1,8 @@
import { BitMask64Data, BitMask64Utils } from './BigIntCompatibility'; import { SegmentPart, BitMask64Data, BitMask64Utils } from './BigIntCompatibility';
/** /**
* 位集合类,用于高效的位操作 * 位集合类,用于高效的位操作
* 自动扩展支持:默认 64 位,超过时自动扩展到 128/256 位 * 支持任意位的位运算操作.
* 扩展模式性能略有下降,但仍然比数组遍历快得多
*/ */
export class Bits { export class Bits {
/** 存储位数据的掩码,支持扩展 */ /** 存储位数据的掩码,支持扩展 */
@@ -37,7 +36,7 @@ export class Bits {
throw new Error('Bit index cannot be negative'); throw new Error('Bit index cannot be negative');
} }
BitMask64Utils.setBitExtended(this._value, index); BitMask64Utils.setBit(this._value, index);
} }
/** /**
@@ -50,7 +49,7 @@ export class Bits {
throw new Error('Bit index cannot be negative'); throw new Error('Bit index cannot be negative');
} }
BitMask64Utils.clearBitExtended(this._value, index); BitMask64Utils.clearBit(this._value, index);
} }
/** /**
@@ -59,7 +58,7 @@ export class Bits {
* @returns 如果位被设置为1则返回true否则返回false * @returns 如果位被设置为1则返回true否则返回false
*/ */
public get(index: number): boolean { public get(index: number): boolean {
return BitMask64Utils.getBitExtended(this._value, index); return BitMask64Utils.getBit(this._value, index);
} }
/** /**
@@ -163,16 +162,16 @@ export class Bits {
if (maxBits <= 32) { if (maxBits <= 32) {
const mask = (1 << maxBits) - 1; const mask = (1 << maxBits) - 1;
result._value.lo = (~result._value.lo) & mask; result._value.base[SegmentPart.LOW] = (~result._value.base[SegmentPart.LOW]) & mask;
result._value.hi = 0; result._value.base[SegmentPart.HIGH] = 0;
} else { } else {
result._value.lo = ~result._value.lo; result._value.base[SegmentPart.LOW] = ~result._value.base[SegmentPart.LOW];
if (maxBits < 64) { if (maxBits < 64) {
const remainingBits = maxBits - 32; const remainingBits = maxBits - 32;
const mask = (1 << remainingBits) - 1; const mask = (1 << remainingBits) - 1;
result._value.hi = (~result._value.hi) & mask; result._value.base[SegmentPart.HIGH] = (~result._value.base[SegmentPart.HIGH]) & mask;
} else { } else {
result._value.hi = ~result._value.hi; result._value.base[SegmentPart.HIGH] = ~result._value.base[SegmentPart.HIGH];
} }
} }
@@ -237,9 +236,10 @@ export class Bits {
* @param maxBits 最大位数默认为64 * @param maxBits 最大位数默认为64
* @returns 二进制字符串表示每8位用空格分隔 * @returns 二进制字符串表示每8位用空格分隔
*/ */
public toBinaryString(maxBits: number = 64): string { public toBinaryString(maxBits: number = 0): string {
if (maxBits > 64) maxBits = 64; if(maxBits == 0){
maxBits = 64 + (this._value.segments ? this._value.segments.length * 64 : 0);
}
let result = ''; let result = '';
for (let i = maxBits - 1; i >= 0; i--) { for (let i = maxBits - 1; i >= 0; i--) {
result += this.get(i) ? '1' : '0'; result += this.get(i) ? '1' : '0';
@@ -265,16 +265,16 @@ export class Bits {
*/ */
public static fromBinaryString(binaryString: string): Bits { public static fromBinaryString(binaryString: string): Bits {
const cleanString = binaryString.replace(/\s/g, ''); const cleanString = binaryString.replace(/\s/g, '');
let data: BitMask64Data; let data: BitMask64Data = { base: undefined!, segments: undefined};
if (cleanString.length <= 32) { if (cleanString.length <= 32) {
const num = parseInt(cleanString, 2); const num = parseInt(cleanString, 2);
data = { lo: num >>> 0, hi: 0 }; data.base = [num >>> 0, 0];
} else { } else {
const loBits = cleanString.substring(cleanString.length - 32); const loBits = cleanString.substring(cleanString.length - 32);
const hiBits = cleanString.substring(0, cleanString.length - 32); const hiBits = cleanString.substring(0, cleanString.length - 32);
const lo = parseInt(loBits, 2); const lo = parseInt(loBits, 2);
const hi = parseInt(hiBits, 2); const hi = parseInt(hiBits, 2);
data = { lo: lo >>> 0, hi: hi >>> 0 }; data.base = [lo >>> 0, hi >>> 0];
} }
return new Bits(data); return new Bits(data);
} }
@@ -286,16 +286,16 @@ export class Bits {
*/ */
public static fromHexString(hexString: string): Bits { public static fromHexString(hexString: string): Bits {
const cleanString = hexString.replace(/^0x/i, ''); const cleanString = hexString.replace(/^0x/i, '');
let data: BitMask64Data; let data: BitMask64Data = { base: undefined!, segments: undefined};
if (cleanString.length <= 8) { if (cleanString.length <= 8) {
const num = parseInt(cleanString, 16); const num = parseInt(cleanString, 16);
data = { lo: num >>> 0, hi: 0 }; data.base = [num >>> 0, 0];
} else { } else {
const loBits = cleanString.substring(cleanString.length - 8); const loBits = cleanString.substring(cleanString.length - 8);
const hiBits = cleanString.substring(0, cleanString.length - 8); const hiBits = cleanString.substring(0, cleanString.length - 8);
const lo = parseInt(loBits, 16); const lo = parseInt(loBits, 16);
const hi = parseInt(hiBits, 16); const hi = parseInt(hiBits, 16);
data = { lo: lo >>> 0, hi: hi >>> 0 }; data.base = [lo >>> 0, hi >>> 0];
} }
return new Bits(data); return new Bits(data);
} }
@@ -318,16 +318,16 @@ export class Bits {
return -1; return -1;
} }
if (this._value.hi !== 0) { if (this._value.base[SegmentPart.HIGH] !== 0) {
for (let i = 31; i >= 0; i--) { for (let i = 31; i >= 0; i--) {
if ((this._value.hi & (1 << i)) !== 0) { if ((this._value.base[SegmentPart.HIGH] & (1 << i)) !== 0) {
return i + 32; return i + 32;
} }
} }
} }
for (let i = 31; i >= 0; i--) { for (let i = 31; i >= 0; i--) {
if ((this._value.lo & (1 << i)) !== 0) { if ((this._value.base[SegmentPart.LOW] & (1 << i)) !== 0) {
return i; return i;
} }
} }
@@ -345,13 +345,13 @@ export class Bits {
} }
for (let i = 0; i < 32; i++) { for (let i = 0; i < 32; i++) {
if ((this._value.lo & (1 << i)) !== 0) { if ((this._value.base[SegmentPart.LOW] & (1 << i)) !== 0) {
return i; return i;
} }
} }
for (let i = 0; i < 32; i++) { for (let i = 0; i < 32; i++) {
if ((this._value.hi & (1 << i)) !== 0) { if ((this._value.base[SegmentPart.HIGH] & (1 << i)) !== 0) {
return i + 32; return i + 32;
} }
} }

View File

@@ -1,247 +1,189 @@
import { import { BitMask64Data, BitMask64Utils } from "../../../src";
BitMask64Data, import { SegmentPart } from "../../../src/ECS/Utils/BigIntCompatibility";
BitMask64Utils
} from '../../../src/ECS/Utils/BigIntCompatibility';
describe('64位掩码兼容性测试', () => { describe("BitMask64Utils 位掩码工具测试", () => {
describe('基本功能', () => { test("create() 应该在指定索引位置设置位", () => {
it('应该能够创建和检查掩码', () => { const mask = BitMask64Utils.create(0);
const zero = BitMask64Utils.ZERO; expect(mask.base[0]).toBe(1);
const mask1 = BitMask64Utils.create(0); expect(mask.base[1]).toBe(0);
const mask2 = BitMask64Utils.create(5);
expect(BitMask64Utils.isZero(zero)).toBe(true); const mask2 = BitMask64Utils.create(33);
expect(BitMask64Utils.isZero(mask1)).toBe(false); expect(mask2.base[0]).toBe(0);
expect(BitMask64Utils.isZero(mask2)).toBe(false); expect(mask2.base[1]).toBe(0b10);
}); });
it('应该支持数字创建', () => { test("fromNumber() 应该把数值放入低32位", () => {
const mask = BitMask64Utils.fromNumber(42); const mask = BitMask64Utils.fromNumber(123456);
expect(mask.lo).toBe(42); expect(mask.base[0]).toBe(123456);
expect(mask.hi).toBe(0); expect(mask.base[1]).toBe(0);
});
}); });
describe('位运算', () => { test("setBit/getBit/clearBit 应该正确设置、读取和清除位", () => {
let mask1: BitMask64Data; const mask: BitMask64Data = { base: [0, 0] };
let mask2: BitMask64Data;
beforeEach(() => {
mask1 = BitMask64Utils.create(2); // 位2
mask2 = BitMask64Utils.create(3); // 位3
});
it('hasAny运算', () => {
const combined = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.orInPlace(combined, mask1);
BitMask64Utils.orInPlace(combined, mask2);
expect(BitMask64Utils.hasAny(combined, mask1)).toBe(true);
expect(BitMask64Utils.hasAny(combined, mask2)).toBe(true);
const mask4 = BitMask64Utils.create(4);
expect(BitMask64Utils.hasAny(combined, mask4)).toBe(false);
});
it('hasAll运算', () => {
const combined = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.orInPlace(combined, mask1);
BitMask64Utils.orInPlace(combined, mask2);
expect(BitMask64Utils.hasAll(combined, mask1)).toBe(true);
expect(BitMask64Utils.hasAll(combined, mask2)).toBe(true);
const both = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.orInPlace(both, mask1);
BitMask64Utils.orInPlace(both, mask2);
expect(BitMask64Utils.hasAll(combined, both)).toBe(true);
});
it('hasNone运算', () => {
const mask4 = BitMask64Utils.create(4);
const mask5 = BitMask64Utils.create(5);
expect(BitMask64Utils.hasNone(mask1, mask2)).toBe(true);
expect(BitMask64Utils.hasNone(mask1, mask4)).toBe(true);
expect(BitMask64Utils.hasNone(mask1, mask1)).toBe(false);
});
it('原地位运算', () => {
const target = BitMask64Utils.clone(mask1);
// OR操作
BitMask64Utils.orInPlace(target, mask2);
expect(BitMask64Utils.hasAll(target, mask1)).toBe(true);
expect(BitMask64Utils.hasAll(target, mask2)).toBe(true);
// AND操作
const andTarget = BitMask64Utils.clone(target);
BitMask64Utils.andInPlace(andTarget, mask1);
expect(BitMask64Utils.hasAll(andTarget, mask1)).toBe(true);
expect(BitMask64Utils.hasAny(andTarget, mask2)).toBe(false);
// XOR操作
const xorTarget = BitMask64Utils.clone(target);
BitMask64Utils.xorInPlace(xorTarget, mask1);
expect(BitMask64Utils.hasAny(xorTarget, mask1)).toBe(false);
expect(BitMask64Utils.hasAll(xorTarget, mask2)).toBe(true);
});
it('设置和清除位', () => {
const mask = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.setBit(mask, 5); BitMask64Utils.setBit(mask, 5);
expect(BitMask64Utils.hasAny(mask, BitMask64Utils.create(5))).toBe(true); expect(BitMask64Utils.getBit(mask, 5)).toBe(true);
BitMask64Utils.clearBit(mask, 5); BitMask64Utils.clearBit(mask, 5);
expect(BitMask64Utils.isZero(mask)).toBe(true); expect(BitMask64Utils.getBit(mask, 5)).toBe(false);
});
// 测试扩展段
BitMask64Utils.setBit(mask, 70);
expect(mask.segments).toBeDefined();
expect(BitMask64Utils.getBit(mask, 70)).toBe(true);
}); });
describe('字符串表示', () => { test("hasAny/hasAll/hasNone 判断应正确", () => {
it('二进制字符串', () => { const maskA = BitMask64Utils.create(1);
const mask = BitMask64Utils.create(5); // 位5设置为1 const maskB = BitMask64Utils.create(1);
const binaryStr = BitMask64Utils.toString(mask, 2); const maskC = BitMask64Utils.create(2);
expect(binaryStr).toBe('100000'); // 位5为1
expect(BitMask64Utils.hasAny(maskA, maskB)).toBe(true);
expect(BitMask64Utils.hasAll(maskA, maskB)).toBe(true);
expect(BitMask64Utils.hasNone(maskA, maskC)).toBe(true);
}); });
it('十六进制字符串', () => { test("isZero 应正确判断", () => {
const mask = BitMask64Utils.fromNumber(255); const mask = BitMask64Utils.create(3);
const hexStr = BitMask64Utils.toString(mask, 16);
expect(hexStr).toBe('0xFF');
});
it('大数字的十六进制表示', () => {
const mask: BitMask64Data = { lo: 0xFFFFFFFF, hi: 0x12345678 };
const hexStr = BitMask64Utils.toString(mask, 16);
expect(hexStr).toBe('0x12345678FFFFFFFF');
});
});
describe('位计数', () => {
it('popCount应该正确计算设置位的数量', () => {
const mask = BitMask64Utils.clone(BitMask64Utils.ZERO);
expect(BitMask64Utils.popCount(mask)).toBe(0);
BitMask64Utils.setBit(mask, 0);
BitMask64Utils.setBit(mask, 2);
BitMask64Utils.setBit(mask, 4);
expect(BitMask64Utils.popCount(mask)).toBe(3);
});
it('大数的popCount', () => {
const mask = BitMask64Utils.fromNumber(0xFF); // 8个1
expect(BitMask64Utils.popCount(mask)).toBe(8);
});
});
describe('ECS组件掩码操作', () => {
it('多组件掩码组合', () => {
const componentMasks: BitMask64Data[] = [];
for (let i = 0; i < 10; i++) {
componentMasks.push(BitMask64Utils.create(i));
}
let combinedMask = BitMask64Utils.clone(BitMask64Utils.ZERO);
for (const mask of componentMasks) {
BitMask64Utils.orInPlace(combinedMask, mask);
}
expect(BitMask64Utils.popCount(combinedMask)).toBe(10);
// 检查所有位都设置了
for (let i = 0; i < 10; i++) {
expect(BitMask64Utils.hasAny(combinedMask, BitMask64Utils.create(i))).toBe(true);
}
});
it('实体匹配模拟', () => {
// 模拟实体具有组件0, 2, 4
const entityMask = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.setBit(entityMask, 0);
BitMask64Utils.setBit(entityMask, 2);
BitMask64Utils.setBit(entityMask, 4);
// 查询需要组件0和2
const queryMask = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.setBit(queryMask, 0);
BitMask64Utils.setBit(queryMask, 2);
expect(BitMask64Utils.hasAll(entityMask, queryMask)).toBe(true);
// 查询需要组件1和3
const queryMask2 = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.setBit(queryMask2, 1);
BitMask64Utils.setBit(queryMask2, 3);
expect(BitMask64Utils.hasAll(entityMask, queryMask2)).toBe(false);
});
});
describe('边界测试', () => {
it('应该处理64位边界', () => {
expect(() => BitMask64Utils.create(63)).not.toThrow();
expect(() => BitMask64Utils.create(64)).toThrow();
expect(() => BitMask64Utils.create(-1)).toThrow();
});
it('设置和清除边界位', () => {
const mask = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.setBit(mask, 63);
expect(BitMask64Utils.hasAny(mask, BitMask64Utils.create(63))).toBe(true);
expect(mask.hi).not.toBe(0);
expect(mask.lo).toBe(0);
BitMask64Utils.clearBit(mask, 63);
expect(BitMask64Utils.isZero(mask)).toBe(true);
});
it('高32位和低32位操作', () => {
const lowMask = BitMask64Utils.create(15); // 低32位
const highMask = BitMask64Utils.create(47); // 高32位
expect(lowMask.hi).toBe(0);
expect(lowMask.lo).not.toBe(0);
expect(highMask.hi).not.toBe(0);
expect(highMask.lo).toBe(0);
const combined = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.orInPlace(combined, lowMask);
BitMask64Utils.orInPlace(combined, highMask);
expect(combined.hi).not.toBe(0);
expect(combined.lo).not.toBe(0);
expect(BitMask64Utils.popCount(combined)).toBe(2);
});
});
describe('复制和相等性', () => {
it('clone应该创建独立副本', () => {
const original = BitMask64Utils.create(5);
const cloned = BitMask64Utils.clone(original);
expect(BitMask64Utils.equals(original, cloned)).toBe(true);
BitMask64Utils.setBit(cloned, 6);
expect(BitMask64Utils.equals(original, cloned)).toBe(false);
});
it('copy应该正确复制', () => {
const source = BitMask64Utils.create(10);
const target = BitMask64Utils.clone(BitMask64Utils.ZERO);
BitMask64Utils.copy(source, target);
expect(BitMask64Utils.equals(source, target)).toBe(true);
});
it('clear应该清除所有位', () => {
const mask = BitMask64Utils.create(20);
expect(BitMask64Utils.isZero(mask)).toBe(false); expect(BitMask64Utils.isZero(mask)).toBe(false);
BitMask64Utils.clear(mask); BitMask64Utils.clear(mask);
expect(BitMask64Utils.isZero(mask)).toBe(true); expect(BitMask64Utils.isZero(mask)).toBe(true);
}); });
test("equals 应正确判断两个掩码是否相等", () => {
const mask1 = BitMask64Utils.create(10);
const mask2 = BitMask64Utils.create(10);
const mask3 = BitMask64Utils.create(11);
expect(BitMask64Utils.equals(mask1, mask2)).toBe(true);
expect(BitMask64Utils.equals(mask1, mask3)).toBe(false);
}); });
test("orInPlace/andInPlace/xorInPlace 运算应正确", () => {
const mask1 = BitMask64Utils.create(1);
const mask2 = BitMask64Utils.create(2);
BitMask64Utils.orInPlace(mask1, mask2);
expect(BitMask64Utils.getBit(mask1, 1)).toBe(true);
expect(BitMask64Utils.getBit(mask1, 2)).toBe(true);
BitMask64Utils.andInPlace(mask1, mask2);
expect(BitMask64Utils.getBit(mask1, 1)).toBe(false);
expect(BitMask64Utils.getBit(mask1, 2)).toBe(true);
BitMask64Utils.xorInPlace(mask1, mask2);
expect(BitMask64Utils.getBit(mask1, 2)).toBe(false);
}); });
test("copy/clone 应正确复制数据", () => {
const source = BitMask64Utils.create(15);
const target: BitMask64Data = { base: [0, 0] };
BitMask64Utils.copy(source, target);
expect(BitMask64Utils.equals(source, target)).toBe(true);
const clone = BitMask64Utils.clone(source);
expect(BitMask64Utils.equals(source, clone)).toBe(true);
expect(clone).not.toBe(source); // 深拷贝
});
test("toString 应正确输出二进制和十六进制", () => {
const mask = BitMask64Utils.create(0);
expect(BitMask64Utils.toString(mask, 2)).toContain("1");
expect(BitMask64Utils.toString(mask, 16)).toMatch(/0x/i);
});
test("popCount 应返回正确的位数", () => {
const mask = BitMask64Utils.create(0);
BitMask64Utils.setBit(mask, 1);
BitMask64Utils.setBit(mask, 63);
expect(BitMask64Utils.popCount(mask)).toBe(3);
});
test("越界与非法输入处理", () => {
expect(() => BitMask64Utils.create(-1)).toThrow();
expect(BitMask64Utils.getBit({ base: [0,0] }, -5)).toBe(false);
expect(() => BitMask64Utils.clearBit({ base: [0,0] }, -2)).toThrow();
});
test("大于64位的扩展段逻辑 - hasAny/hasAll/hasNone/equals", () => {
// 掩码 A: 只在 bit 150 位置为 1
const maskA = BitMask64Utils.create(150);
// 掩码 B: 只在 bit 200 位置为 1
const maskB = BitMask64Utils.create(200);
// A 与 B 在不同扩展段,不存在重叠位
expect(BitMask64Utils.hasAny(maskA, maskB)).toBe(false);
expect(BitMask64Utils.hasNone(maskA, maskB)).toBe(true);
// C: 在 150 与 200 都置位
const maskC = BitMask64Utils.clone(maskA);
BitMask64Utils.setBit(maskC, 200);
// A 是 C 的子集
expect(BitMask64Utils.hasAll(maskC, maskA)).toBe(true);
// B 是 C 的子集
expect(BitMask64Utils.hasAll(maskC, maskB)).toBe(true);
// A 和 C 不相等
expect(BitMask64Utils.equals(maskA, maskC)).toBe(false);
// C 与自身相等
expect(BitMask64Utils.equals(maskC, maskC)).toBe(true);
//copy
const copyMask = BitMask64Utils.create(0);
BitMask64Utils.copy(maskA,copyMask);
expect(BitMask64Utils.equals(copyMask,maskA)).toBe(true);
// hasAll短路测试对第一个if的测试覆盖
BitMask64Utils.setBit(copyMask,64);
expect(BitMask64Utils.hasAll(maskA, copyMask)).toBe(false);
BitMask64Utils.clearBit(copyMask, 64);
// 扩展到350位对最后一个短路if的测试覆盖
BitMask64Utils.setBit(copyMask,350);
expect(BitMask64Utils.hasAll(maskA, copyMask)).toBe(false);
});
test("大于64位的逻辑运算 - or/and/xor 跨段处理", () => {
const maskA = BitMask64Utils.create(128); // 第一扩展段
const maskB = BitMask64Utils.create(190); // 同一扩展段但不同位置
const maskC = BitMask64Utils.create(300); // 不同扩展段
// OR: 合并不同扩展段
const orMask = BitMask64Utils.clone(maskA);
BitMask64Utils.orInPlace(orMask, maskC);
expect(BitMask64Utils.getBit(orMask, 128)).toBe(true);
expect(BitMask64Utils.getBit(orMask, 300)).toBe(true);
// AND: 交集为空
const andMask = BitMask64Utils.clone(maskA);
BitMask64Utils.andInPlace(andMask, maskB);
expect(BitMask64Utils.isZero(andMask)).toBe(true);
// XOR: 不同扩展段应该都保留
const xorMask = BitMask64Utils.clone(maskA);
BitMask64Utils.xorInPlace(xorMask, maskC);
expect(BitMask64Utils.getBit(xorMask, 128)).toBe(true);
expect(BitMask64Utils.getBit(xorMask, 300)).toBe(true);
});
test("扩展段 toString 与 popCount 验证", () => {
const mask = BitMask64Utils.create(130); // 扩展段,此时扩展段自动延长到2
BitMask64Utils.setBit(mask, 260); // 再设置另一个更高位,此时扩展段自动延长到3
const strBin = BitMask64Utils.toString(mask, 2);
const strHex = BitMask64Utils.toString(mask, 16);
// 第三个区段应该以100结尾130位为1
expect(strBin.split(' ')[2].endsWith('100')).toBe(true);
// 第三个区段应该以4结尾低32位为0x4
expect(strHex.split(' ')[2].endsWith('4')).toBe(true);
// 应该有两个置位
expect(BitMask64Utils.popCount(mask)).toBe(2);
});
});