doomcc/dcc/text.py

import dcc.config
import dcc.doom_base
import sys
import wand.drawing
import wand.image


def draw_text(self, img, text, font_size=64):
    with wand.drawing.Drawing() as draw:
        draw.font = self.thumbnail_font
        draw.font_size = font_size
        draw.fill_color = wand.color.Color(self.thumbnail_text_fill)
        draw.stroke_color = wand.color.Color(self.thumbnail_text_stroke)
        draw.stroke_width = font_size * 5 / 32
        draw.text_interline_spacing = -font_size / 4
        draw.text(5, int(draw.font_size) + 5, text)
        draw(img)
        draw.stroke_color = wand.color.Color("none")
        draw.stroke_width = 0
        draw.text(5, int(draw.font_size)+5, text)
        draw(img)


dcc.config.Base.draw_text = draw_text


class Text(dcc.doom_base.WadMap):
    def get_parser(self, prog_name):
        parser = super().get_parser(prog_name)
        parser.add_argument("--nomap", action="store_true")
        parser.add_argument("--demotype", default="UV-Max Demo")
        return parser

    def take_action(self, parsed_args):
        text = sys.stdin.read().rstrip()
        if not parsed_args.nomap:
            text = "MAP{}: {}".format(parsed_args.map, text)
        text = "{}\n{}".format(text, parsed_args.demotype)
        with wand.image.Image(
            height=self.thumbnail_height,
            width=self.thumbnail_width
        ) as img:
            self.draw_text(
                img, text,
            )
            img.trim()
            img.reset_coords()
            img.save(filename=self.text_thumb_path())