tangent128/base2020
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Better typings for Lookup

Browse source
This commit is contained in:
Tangent Wantwight 2020-01-16 00:03:24 -05:00
parent 8434e7046c
commit a080dabc67
2 changed files with 12 additions and 43 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
src/Ecs/Data.ts
View file

@ -77,45 +77,14 @@ export function Remove<DATA extends Data>(data: DATA, [id, generation]: Id, stat
} }
// Ergonomic Lookup typings // Ergonomic Lookup typings
export function Lookup< type ItemType<S> = S extends Record<number, infer T> ? T : never;
DATA extends Data, type StoreType<DATA extends Data, K> = K extends keyof DATA ? ItemType<DATA[K]> : never;
A extends keyof DATA, type StoreTypes<DATA extends Data, K extends (keyof DATA)[]> = {
> ( [I in keyof K]: StoreType<DATA, K[I]>;
data: DATA, };
id: Id, type MaybeStoreTypes<DATA extends Data, K extends (keyof DATA)[]> = {
a: A, [I in keyof K]: StoreType<DATA, K[I]> | null;
): [ };
DATA[A][number] | null
];
export function Lookup<
DATA extends Data,
A extends keyof DATA,
B extends keyof DATA,
> (
data: DATA,
id: Id,
a: A,
b: B,
): [
DATA[A][number] | null,
DATA[B][number] | null
];
export function Lookup<
DATA extends Data,
A extends keyof DATA,
B extends keyof DATA,
C extends keyof DATA,
> (
data: DATA,
id: Id,
a: A,
b: B,
c: C,
): [
DATA[A][number] | null,
DATA[B][number] | null,
DATA[C][number]
];
/** /**
* Look up components that may or may not exist for an entity * Look up components that may or may not exist for an entity
@ -124,7 +93,7 @@ export function Lookup<
* @param components names of components to look for * @param components names of components to look for
* @returns the cooresponding components, with unfound ones replaced by nulls * @returns the cooresponding components, with unfound ones replaced by nulls
*/ */
export function Lookup<DATA extends Data, K extends keyof DATA>(data: DATA, [id, generation]: Id, ...components: K[]): ({} | null)[] { export function Lookup<DATA extends Data, K extends (keyof DATA)[]>(data: DATA, [id, generation]: Id, ...components: K): MaybeStoreTypes<DATA, K> {
const entity = data.entity[id]; const entity = data.entity[id];
// inactive entities are fine to lookup, but dead ones are not // inactive entities are fine to lookup, but dead ones are not
if(entity && entity.generation == generation && entity.alive != Liveness.DEAD) { if(entity && entity.generation == generation && entity.alive != Liveness.DEAD) {
@ -135,9 +104,9 @@ export function Lookup<DATA extends Data, K extends keyof DATA>(data: DATA, [id,
} else { } else {
return null; return null;
} }
}); }) as MaybeStoreTypes<DATA, K>;
} else { } else {
return components.map(() => null); return components.map(() => null) as MaybeStoreTypes<DATA, K>;
} }
} }

2
src/Ecs/test.ts
View file

@ -144,7 +144,7 @@ export class LoopTest {
DumbMotion(this.data, interval); DumbMotion(this.data, interval);
const [triangleDebug] = Lookup(this.data, triangleId, "renderBounds"); const [triangleDebug] = Lookup(this.data, triangleId, "renderBounds");
(triangleDebug as RenderBounds).color = "#d40"; if(triangleDebug) triangleDebug.color = "#d40";
FindCollisions(this.data, 500, (className, sourceId, targetId) => { FindCollisions(this.data, 500, (className, sourceId, targetId) => {
switch(className) { switch(className) {