from uniperceiver.utils.registry import Registry

DECODE_STRATEGY_REGISTRY = Registry("DECODE_STRATEGY")
DECODE_STRATEGY_REGISTRY.__doc__ = """
Registry for decode strategy
"""

def build_beam_searcher(cfg):
    beam_search = None if cfg.DECODE_STRATEGY.NAME.lower() == "none" else DECODE_STRATEGY_REGISTRY.get(cfg.DECODE_STRATEGY.NAME)(cfg)
    return beam_search

def build_greedy_decoder(cfg):
    greedy_decoder = DECODE_STRATEGY_REGISTRY.get("GreedyDecoder")(cfg)
    return greedy_decoder