qzl
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from enum import IntEnum
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
|
|
class SystemVisibilityBit(IntEnum):
|
|
UI_HISTORY = 0
|
|
CONTEXT_ASSEMBLY = 1
|
|
|
|
|
|
class VisibilityMask(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
value: int = Field(..., ge=0, le=(1 << 63) - 1)
|
|
|
|
@classmethod
|
|
def from_bits(cls, *, bits: list[int]) -> "VisibilityMask":
|
|
mask = 0
|
|
for bit in bits:
|
|
validate_visibility_bit(bit=bit)
|
|
mask |= 1 << bit
|
|
return cls(value=mask)
|
|
|
|
def contains(self, *, bit: int) -> bool:
|
|
validate_visibility_bit(bit=bit)
|
|
return bool(self.value & (1 << bit))
|
|
|
|
|
|
class VisibilityBitRef(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
bit: int = Field(..., ge=0, le=63)
|
|
|
|
@field_validator("bit")
|
|
@classmethod
|
|
def _validate_bit(cls, value: int) -> int:
|
|
validate_visibility_bit(bit=value)
|
|
return value
|
|
|
|
|
|
def validate_visibility_bit(*, bit: int) -> None:
|
|
if bit < 0 or bit > 63:
|
|
raise ValueError("visibility bit must be in range [0, 63]")
|
|
|
|
|
|
def bit_mask(*, bit: int) -> int:
|
|
validate_visibility_bit(bit=bit)
|
|
return 1 << bit
|