8bitlenser/lenser/nes/convert/__init__.py
2026-07-03 19:35:35 -07:00

19 lines
748 B
Python

"""Nintendo Entertainment System conversion dispatch."""
from __future__ import annotations
from ... import imageprep
from . import bg, mono
_MODULES = {"bg": bg, "mono": mono}
MODES = list(_MODULES.keys())
def convert_image(path_or_img, mode="bg", palette_name="nes",
dither_mode="floyd", intensive=False, prep_opt=None,
base_color=None):
prep_opt = prep_opt or imageprep.PrepOptions()
module = _MODULES.get(mode, bg)
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)