抽出 entitiesOf

Author: xieyuheng

TODO.md

@@ -1,5 +1,3 @@
-抽出 `entitiesOf`
-
 修复 `commonAttributes` -- 应该是交集不是并集
 
 `concept-graph/` -- 从 `generateConcepts` 生成有向图

src/context/attributesOf.ts

@@ -1,6 +1,9 @@
 import type { Attribute, Context, Entity } from "./Context.js"
 
-export function attributesOf(context: Context, entity: Entity): Set<Attribute> {
+export function attributesOf(
+  context: Context,
+  entity: Entity,
+): ReadonlySet<Attribute> {
   const attributes = context.entityAttributeIndex.get(entity)
   if (attributes === undefined) {
     return new Set()

src/context/entitiesOf.ts

@@ -0,0 +1,16 @@
+import type { Attribute, Context, Entity } from "./Context.js"
+import { attributesOf } from "./attributesOf.js"
+
+export function entitiesOf(
+  context: Context,
+  attribute: Attribute,
+): ReadonlySet<Entity> {
+  const entities = new Set<Entity>()
+  for (const entity of context.entities) {
+    if (attributesOf(context, entity).has(attribute)) {
+      entities.add(entity)
+    }
+  }
+
+  return entities
+}

src/context/index.ts

@@ -1,8 +1,8 @@
 export * from "./attributesOf.js"
-// export * from "./entitiesOf.js"
 export * from "./commonAttributes.js"
 export * from "./commonEntities.js"
 export * from "./Context.js"
 export * from "./createContextFromCrossTable.js"
 export * from "./CrossTable.js"
+export * from "./entitiesOf.js"
 export * from "./entityHasAttribute.js"