"""TRS-80 Color Computer conversion dispatch."""
from __future__ import annotations
from ... import imageprep
from . import pmode3, pmode4, mono

_MODULES = {"pmode4": pmode4, "pmode3": pmode3, "mono": mono}
MODES = list(_MODULES.keys())


def convert_image(path_or_img, mode="pmode4", palette_name="mc6847",
                  dither_mode="floyd", intensive=False, prep_opt=None, base_color=None):
    prep_opt = prep_opt or imageprep.PrepOptions()
    module = _MODULES[mode]
    img_rgb = imageprep.prepare(path_or_img, module.WIDTH, module.HEIGHT,
                                module.PIXEL_ASPECT, prep_opt, border_rgb=(0, 0, 0))
    return module.convert(img_rgb, palette_name, dither_mode, intensive, base_color=base_color)