doomcc/dcc/extract.py

import dcc.doom_base
import omg
import numpy as np
import wand.color
import wand.image


class Extract(dcc.doom_base.Wad):
    def get_parser(self, prog_name):
        parser = super().get_parser(prog_name)
        parser.add_argument('lump')
        return parser

    def take_action(self, parsed_args):
        for w in self.load_order()
            try:
                # TODO: handle anything other than graphics.
                wad = omg.WadIO(w)
                gl = omg.Graphic(wad.read(parsed_args.lump))
                # With no arguments, convert() changes a paletted image to an
                # RGB one.
                with wand.image.Image.from_array(
                    np.array(gl.to_Image().convert())
                ) as img:
                    img.transparent_color(wand.color.Color("#ff00ff"), 0.0)
                    img.save(
                        filename=self.fabricate.joinpath(parsed_args.wad)
                        .joinpath(parsed_args.lump + ".png")
                    )
                return
            except Exception as e:
                print(
                    f"Wad {w} likely has no lump {parsed_args.lump} "
                    + f"(exception {e})."
                )

        print(
            f"Lump {parsed_args.lump} not found in any wad in "
            + f"{parsed_args.wad}"
        )