File size: 6,797 Bytes
9dce458 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 |
import tempfile
import subprocess
import math
import cv2
import platform
import glob
import os
from ..utils import Context
# convert alignment/direction to gimp's values
alignment_to_justification = {
"left": "TEXT-JUSTIFY-LEFT",
"right": "TEXT-JUSTIFY-RIGHT",
"center": "TEXT-JUSTIFY-CENTER",
}
direction_to_base_direction = {
"h": "TEXT-DIRECTION-LTR",
"v": "TEXT-DIRECTION-TTB-LTR-UPRIGHT",
"hr": "TEXT-DIRECTION-RTL",
"vr": "TEXT-DIRECTION-TTB-RTL-UPRIGHT",
}
text_init_template = '( text{n} ( car ( gimp-text-layer-new image "{text}" "{default_font}" {text_size} 0 ) ) )'
font_template = '( gimp-text-layer-set-font text{n} "{font}" )'
angle_template = "( gimp-item-transform-rotate text{n} {angle} TRUE 0 0 )"
text_template = """
( gimp-image-add-layer image text{n} 0 )
( gimp-text-layer-set-color text{n} (list {color}) )
( gimp-item-set-name text{n} "{name}" )
( gimp-layer-set-offsets text{n} {position} )
( gimp-text-layer-resize text{n} {size} )
( gimp-text-layer-set-language text{n} "{language}" )
( gimp-text-layer-set-letter-spacing text{n} {letter_spacing} )
( gimp-text-layer-set-line-spacing text{n} {line_spacing} )
( gimp-text-layer-set-base-direction text{n} {base_direction} )
( gimp-text-layer-set-justification text{n} {justify} )
{font}
{angle}
"""
save_templates = {
"xcf": '( gimp-xcf-save RUN-NONINTERACTIVE image background_layer "{out_file}" "{out_file}" )',
"psd": '( file-psd-save RUN-NONINTERACTIVE image background_layer "{out_file}" "{out_file}" 0 0 )',
"pdf": '( file-pdf-save RUN-NONINTERACTIVE image background_layer "{out_file}" "{out_file}" TRUE TRUE TRUE )',
}
create_mask = '( inpainting ( car ( gimp-file-load-layer RUN-NONINTERACTIVE image "{mask_file}" ) ) )'
rename_mask = '( gimp-image-add-layer image inpainting 0 ) ( gimp-item-set-name inpainting "mask" )'
script_template = """
( let* (
( image ( car ( gimp-file-load RUN-NONINTERACTIVE "{input_file}" "{input_file}" ) ) )
( layer-list (gimp-image-get-layers image))
( background_layer (car layer-list))
{create_mask}
{text_init}
)
{rename_mask}
( gimp-item-set-name background_layer "original image" )
( gimp-item-set-lock-content background_layer TRUE )
( gimp-item-set-lock-position background_layer TRUE )
{text}
{save}
( gimp-quit 0 )
)"""
def gimp_render(out_file, ctx: Context):
input_file = os.path.join(tempfile.gettempdir(), ".gimp_input.png")
mask_file = os.path.join(tempfile.gettempdir(), ".gimp_mask.png")
extension = out_file.split(".")[-1]
ctx.upscaled.save(input_file)
# If there is no text on the page, gimp_mask will be None and there is no
# need to add it as a layer.
if ctx.gimp_mask is not None:
cv2.imwrite(mask_file, ctx.gimp_mask)
else:
ctx.text_regions = []
filtered_text_regions = [
text_region for text_region in ctx.text_regions if text_region.translation != ""
]
text_init = "\n".join(
[
text_init_template.format(
n=n,
text=text_region.translation.replace('"', '\\"'),
text_size=text_region.font_size,
default_font=ctx.gimp_font
+ (" Bold" if text_region.bold else "")
+ (" Italic" if text_region.italic else ""),
)
for n, text_region in enumerate(filtered_text_regions)
]
)
text = "".join(
[
text_template.format(
n=n,
color=" ".join([str(value) for value in text_region.fg_colors]),
name=" ".join(text_region.text).replace('"', '\\"'),
position=str(text_region.xywh[0]) + " " + str(text_region.xywh[1]),
size=str(text_region.xywh[2]) + " " + str(text_region.xywh[3]),
justify=alignment_to_justification[text_region.alignment],
font=font_template.format(n=n, font=text_region.font_family)
if text_region.font_family != ""
else "",
# rotated text is weird in gimp so we don't do it unless it's over 10 degrees
angle=angle_template.format(n=n, angle=math.radians(text_region.angle))
if abs(text_region.angle) > 10
else "",
language=text_region.target_lang,
line_spacing=text_region.line_spacing,
letter_spacing=text_region.letter_spacing,
base_direction=direction_to_base_direction[text_region.direction],
)
for n, text_region in enumerate(filtered_text_regions)
]
)
# scheme script to be ran by gimp
full_script = script_template.format(
input_file=input_file.replace("\\", "\\\\"),
text_init=text_init,
text=text,
extension=extension,
save=save_templates[extension].format(out_file=out_file.replace("\\", "\\\\")),
create_mask=(
create_mask.format(mask_file=mask_file.replace("\\", "\\\\"))
if ctx.gimp_mask is not None
else ""
),
rename_mask=(rename_mask if ctx.gimp_mask is not None else ""),
)
gimp_batch(full_script)
# Delete Files
os.unlink(input_file)
# Deleting file only if it exists
if os.path.exists(mask_file):
os.unlink(mask_file)
def gimp_console_executable():
executable = "gimp"
if platform.system() == "Windows":
gimp_dir = os.getenv("LOCALAPPDATA") + "\\Programs\\GIMP 2\\bin\\"
executables = glob.glob(gimp_dir + "gimp-console-2.*.exe")
if len(executables) > 0:
return executables[0]
# may be in program files
gimp_dir = os.getenv("ProgramFiles") + "\\GIMP 2\\bin\\"
executables = glob.glob(gimp_dir + "gimp-console-2.*.exe")
if len(executables) == 0:
print("error: gimp not found in directory:", gimp_dir)
return
executable = executables[0]
return executable
def gimp_batch(script):
"""
Run a gimp script in batch mode. Quit gimp after running the script and on errors. Raise an exception if there is a GIMP error.
"""
# logging.info("=== Running GIMP script:")
# result =
result = subprocess.run(
[gimp_console_executable(), "-i", "-b", script, "-b", "(gimp-quit 0)"],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
universal_newlines=True,
)
print("=== Output")
print(result.stdout)
print("=== Error")
print(result.stderr)
if "Error:" in result.stderr:
raise Exception("GIMP Execution error")
# return result
|