45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import asdict, dataclass
|
|
from typing import Any
|
|
|
|
COLORS = ("red", "blue", "green", "yellow")
|
|
SHAPES = ("circle", "square", "triangle", "star")
|
|
RELATIONS = ("left_of", "right_of", "above", "below")
|
|
CLASSES = ("alpha", "beta", "gamma", "delta")
|
|
OBJECT_IDS = ("obj1", "obj2", "obj3")
|
|
|
|
PREDICATES = ("color", "shape", "left_of", "right_of", "above", "below", "implies_class")
|
|
ENTITIES = OBJECT_IDS + COLORS + SHAPES + CLASSES
|
|
|
|
SUBJECT_TO_ID = {value: index for index, value in enumerate(ENTITIES)}
|
|
PREDICATE_TO_ID = {value: index for index, value in enumerate(PREDICATES)}
|
|
OBJECT_TO_ID = SUBJECT_TO_ID
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Fact:
|
|
subject: str
|
|
predicate: str
|
|
object: str
|
|
|
|
def to_dict(self) -> dict[str, str]:
|
|
return asdict(self)
|
|
|
|
@classmethod
|
|
def from_dict(cls, value: dict[str, str]) -> "Fact":
|
|
return cls(**value)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class SceneObject:
|
|
object_id: str
|
|
color: str
|
|
shape: str
|
|
x: int
|
|
y: int
|
|
|
|
def to_dict(self) -> dict[str, Any]:
|
|
return asdict(self)
|
|
|