mirror of
https://github.com/MarioSpore/Grinch-AP.git
synced 2025-10-21 20:21:32 -06:00
40 lines
847 B
Python
40 lines
847 B
Python
|
from functools import singledispatch
|
||
|
|
from typing import Set
|
||
|
|
|
||
|
|
from . import StardewRule, Reach, Count, AggregatingStardewRule, Has
|
||
|
|
|
||
|
|
|
||
|
|
def look_for_indirect_connection(rule: StardewRule) -> Set[str]:
|
||
|
|
required_regions = set()
|
||
|
|
_find(rule, required_regions)
|
||
|
|
return required_regions
|
||
|
|
|
||
|
|
|
||
|
|
@singledispatch
|
||
|
|
def _find(rule: StardewRule, regions: Set[str]):
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
@_find.register
|
||
|
|
def _(rule: AggregatingStardewRule, regions: Set[str]):
|
||
|
|
for r in rule.original_rules:
|
||
|
|
_find(r, regions)
|
||
|
|
|
||
|
|
|
||
|
|
@_find.register
|
||
|
|
def _(rule: Count, regions: Set[str]):
|
||
|
|
for r in rule.rules:
|
||
|
|
_find(r, regions)
|
||
|
|
|
||
|
|
|
||
|
|
@_find.register
|
||
|
|
def _(rule: Has, regions: Set[str]):
|
||
|
|
r = rule.other_rules[rule.item]
|
||
|
|
_find(r, regions)
|
||
|
|
|
||
|
|
|
||
|
|
@_find.register
|
||
|
|
def _(rule: Reach, regions: Set[str]):
|
||
|
|
if rule.resolution_hint == "Region":
|
||
|
|
regions.add(rule.spot)
|