81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
try:
|
|
from . import category_template_metadata as template_policy
|
|
from . import hardcore_position_config as hardcore_position_policy
|
|
from .hardcore_action_metadata import source_hardcore_action_family
|
|
except ImportError: # Allows local smoke tests from the repository root.
|
|
import category_template_metadata as template_policy
|
|
import hardcore_position_config as hardcore_position_policy
|
|
from hardcore_action_metadata import source_hardcore_action_family
|
|
|
|
|
|
EMPTY_ACTION_POSITION_ROUTE = {
|
|
"position_family": "",
|
|
"position_keys": [],
|
|
"position_key": "",
|
|
"action_family": "",
|
|
}
|
|
|
|
|
|
def empty_action_position_route() -> dict[str, Any]:
|
|
return {
|
|
"position_family": "",
|
|
"position_keys": [],
|
|
"position_key": "",
|
|
"action_family": "",
|
|
}
|
|
|
|
|
|
def resolve_action_position_route(
|
|
*,
|
|
is_pose_category: bool,
|
|
subcategory: dict[str, Any],
|
|
hardcore_position_config: dict[str, Any] | None,
|
|
item_template_metadata: dict[str, Any] | None,
|
|
item_text: Any,
|
|
source_role_graph: Any,
|
|
source_composition: Any,
|
|
pose: Any,
|
|
item_axis_values: dict[str, Any] | None = None,
|
|
) -> dict[str, Any]:
|
|
if not is_pose_category:
|
|
return empty_action_position_route()
|
|
|
|
metadata = item_template_metadata or {}
|
|
position_family = template_policy.template_position_family(
|
|
metadata
|
|
) or hardcore_position_policy.hardcore_source_position_family(
|
|
subcategory,
|
|
hardcore_position_config,
|
|
)
|
|
inferred_position_keys = hardcore_position_policy.hardcore_position_keys(
|
|
item_text,
|
|
source_role_graph,
|
|
source_composition,
|
|
pose,
|
|
axis_values=item_axis_values,
|
|
)
|
|
position_keys = template_policy.merge_position_keys(
|
|
template_policy.template_position_keys(metadata),
|
|
inferred_position_keys,
|
|
)
|
|
action_family = template_policy.template_action_family(metadata)
|
|
if not action_family:
|
|
action_family = source_hardcore_action_family(
|
|
position_family,
|
|
source_role_graph,
|
|
item_text,
|
|
source_composition,
|
|
item_axis_values,
|
|
)
|
|
|
|
return {
|
|
"position_family": position_family,
|
|
"position_keys": position_keys,
|
|
"position_key": position_keys[0] if position_keys else "",
|
|
"action_family": action_family,
|
|
}
|