88 lines
2.8 KiB
TypeScript
88 lines
2.8 KiB
TypeScript
|
import { Component } from '../../../src/ECS/Component';
|
|||
|
|
import { ComponentStorageManager, EnableSoA } from '../../../src/ECS/Core/ComponentStorage';
|
|||
|
|
|
|||
|
|
// 默认原始存储组件
|
|||
|
|
class PositionComponent extends Component {
|
|||
|
|
public x: number = 0;
|
|||
|
|
public y: number = 0;
|
|||
|
|
public z: number = 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 启用SoA优化的组件（用于大规模批量操作）
|
|||
|
|
@EnableSoA
|
|||
|
|
class LargeScaleComponent extends Component {
|
|||
|
|
public x: number = 0;
|
|||
|
|
public y: number = 0;
|
|||
|
|
public z: number = 0;
|
|||
|
|
public vx: number = 0;
|
|||
|
|
public vy: number = 0;
|
|||
|
|
public vz: number = 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
describe('SoA优化选择测试', () => {
|
|||
|
|
let manager: ComponentStorageManager;
|
|||
|
|
|
|||
|
|
beforeEach(() => {
|
|||
|
|
manager = new ComponentStorageManager();
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
test('默认使用原始存储', () => {
|
|||
|
|
const storage = manager.getStorage(PositionComponent);
|
|||
|
|
|
|||
|
|
// 添加组件
|
|||
|
|
manager.addComponent(1, new PositionComponent());
|
|||
|
|
|
|||
|
|
// 验证能正常工作
|
|||
|
|
const component = manager.getComponent(1, PositionComponent);
|
|||
|
|
expect(component).toBeTruthy();
|
|||
|
|
expect(component?.x).toBe(0);
|
|||
|
|
|
|||
|
|
// 验证使用原始存储
|
|||
|
|
expect(storage.constructor.name).toBe('ComponentStorage');
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
test('@EnableSoA装饰器启用优化', () => {
|
|||
|
|
const storage = manager.getStorage(LargeScaleComponent);
|
|||
|
|
|
|||
|
|
// 添加组件
|
|||
|
|
const component = new LargeScaleComponent();
|
|||
|
|
component.x = 100;
|
|||
|
|
component.vx = 10;
|
|||
|
|
manager.addComponent(1, component);
|
|||
|
|
|
|||
|
|
// 验证能正常工作
|
|||
|
|
const retrieved = manager.getComponent(1, LargeScaleComponent);
|
|||
|
|
expect(retrieved).toBeTruthy();
|
|||
|
|
expect(retrieved?.x).toBe(100);
|
|||
|
|
expect(retrieved?.vx).toBe(10);
|
|||
|
|
|
|||
|
|
// 验证使用SoA存储
|
|||
|
|
expect(storage.constructor.name).toBe('SoAStorage');
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
test('SoA存储功能验证', () => {
|
|||
|
|
const entityCount = 1000;
|
|||
|
|
|
|||
|
|
// 创建实体（使用SoA优化）
|
|||
|
|
for (let i = 0; i < entityCount; i++) {
|
|||
|
|
const component = new LargeScaleComponent();
|
|||
|
|
component.x = i;
|
|||
|
|
component.y = i * 2;
|
|||
|
|
component.vx = 1;
|
|||
|
|
component.vy = 2;
|
|||
|
|
manager.addComponent(i, component);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证数据正确性
|
|||
|
|
const testComponent = manager.getComponent(100, LargeScaleComponent);
|
|||
|
|
expect(testComponent?.x).toBe(100);
|
|||
|
|
expect(testComponent?.y).toBe(200);
|
|||
|
|
expect(testComponent?.vx).toBe(1);
|
|||
|
|
expect(testComponent?.vy).toBe(2);
|
|||
|
|
|
|||
|
|
// 验证存储类型
|
|||
|
|
const storage = manager.getStorage(LargeScaleComponent);
|
|||
|
|
expect(storage.constructor.name).toBe('SoAStorage');
|
|||
|
|
console.log(`成功创建 ${entityCount} 个SoA实体，数据验证通过`);
|
|||
|
|
});
|
|||
|
|
});
|