113 lines
2.1 KiB
Python
113 lines
2.1 KiB
Python
"""
|
|
Base modifier dataclasses for ComfyUI Variationator
|
|
"""
|
|
|
|
from typing import List
|
|
from dataclasses import dataclass
|
|
from enum import Enum
|
|
|
|
|
|
class ModifierCategory(Enum):
|
|
"""Categories for modifiers that use wet/dry classification"""
|
|
DRY = "dry"
|
|
WET = "wet"
|
|
|
|
|
|
@dataclass
|
|
class LightingModifier:
|
|
"""Represents a lighting/time-of-day modifier"""
|
|
id: int
|
|
description: str
|
|
category: ModifierCategory
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class SeasonalModifier:
|
|
"""Represents a seasonal variation modifier"""
|
|
id: int
|
|
description: str
|
|
category: ModifierCategory
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class LocationModifier:
|
|
"""Represents a location/environment modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class WeatherModifier:
|
|
"""Represents a weather condition modifier"""
|
|
id: int
|
|
description: str
|
|
category: ModifierCategory
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class MoodModifier:
|
|
"""Represents a mood/atmosphere modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class CameraModifier:
|
|
"""Represents a camera/photography technique modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class ColorGradingModifier:
|
|
"""Represents a color grading modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class ActionModifier:
|
|
"""Represents an action/activity state modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class EraModifier:
|
|
"""Represents a time period/era modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class DetailModifier:
|
|
"""Represents a detail/focus control modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class CompositionModifier:
|
|
"""Represents a composition technique modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|
|
|
|
|
|
@dataclass
|
|
class DistanceModifier:
|
|
"""Represents a distance/proximity modifier"""
|
|
id: int
|
|
description: str
|
|
tags: List[str]
|