def prepare_drivethrucards(img, drawable, output_folder):
""" Prepare an image for DriveThruCards printing.
"""
gimp.progress_init('Prepare an image for DriveThruCards printing...')
pdb.gimp_undo_push_group_start(img)
try:
file_name, back_side = _get_filename_backside(img, 'jpg')
except Exception: # pylint: disable=W0703
pdb.gimp_undo_push_group_end(img)
return
rotation = _get_rotation(drawable)
clip_size = _get_dtc_clip_size(drawable)
if rotation:
_rotate(drawable, back_side)
if clip_size:
_clip(img, drawable, clip_size, rotation and back_side)
pdb.file_jpeg_save(img, drawable,
os.path.join(output_folder, file_name), file_name,
1, 0, 1, 0, '', 2, 1, 0, 0)
pdb.gimp_undo_push_group_end(img)
def prepare_tts(img, _, output_folder): # pylint: disable=R0914
""" Prepare a TTS sheet image.
"""
gimp.progress_init('Prepare a TTS sheet image...')
pdb.gimp_undo_push_group_start(img)
try:
file_name, _ = _get_filename_backside(img, 'jpg')
except Exception: # pylint: disable=W0703
pdb.gimp_undo_push_group_end(img)
return
parts = file_name.split('_')
num = int(parts[-3])
rows = int(parts[-4])
columns = int(parts[-5])
new_width = columns * 750
new_height = rows * 1050
pdb.gimp_image_scale(img, new_width, new_height)
json_path = re.sub(r'\.jpg$', '.json', pdb.gimp_image_get_filename(img))
try:
with open(json_path, 'r') as fobj:
cards = json.load(fobj)
except Exception: # pylint: disable=W0703
pdb.gimp_undo_push_group_end(img)
return
cards = [c['path'] for c in cards]
if len(cards) != num:
pdb.gimp_undo_push_group_end(img)
return
card_rows = [
cards[i * columns:(i + 1) * columns]
for i in range((len(cards) + columns - 1) // columns)
]
if len(card_rows) != rows:
pdb.gimp_undo_push_group_end(img)
return
for i, card_row in enumerate(card_rows):
for j, card_path in enumerate(card_row):
if not os.path.exists(card_path):
pdb.gimp_undo_push_group_end(img)
return
card_layer = pdb.gimp_file_load_layer(img, card_path)
pdb.gimp_image_insert_layer(img, card_layer, None, -1)
rotation = _get_rotation(card_layer)
if rotation:
_rotate(card_layer, True)
pdb.gimp_layer_set_offsets(card_layer, j * 750, i * 1050)
pdb.gimp_image_merge_down(img, card_layer, 1)
pdb.file_jpeg_save(img, img.layers[0],
os.path.join(output_folder, file_name), file_name, 1, 0,
1, 0, '', 2, 1, 0, 0)
pdb.gimp_undo_push_group_end(img)
def captcha_generate(imageDir, goal):
"""Make sure there are as many catchas in image_dir as goal."""
needed = goal - countImages(imageDir)
if needed < 1:
return
answers = [selectAnswer(CAPTCHA_LETTERS) for i in xrange(0, needed)]
for answer in answers:
imageFile = '{0}{1}'.format(answer, CAPTCHA_FILE_EXT)
imagePath = os.path.join(imageDir, imageFile)
imageTmp = '%s.tmp' % imagePath
img, drawable = make_captcha(CAPTCHA_WIDTH, CAPTCHA_HEIGHT,
FONT_HEIGHT, LETTER_SPACING, LEFT_MARGIN,
ANGLE_RANGE, FONTS, answer)
try:
if CAPTCHA_FILE_EXT == ".jpg":
gpdb.file_jpeg_save(img, drawable,
imageTmp, imageTmp,
JPG_QUALITY,
JPG_SMOOTHING,
JPG_OPTIMIZE,
JPG_PROGRESSIVE,
JPG_COMMENT,
JPG_SUBSMP,
JPG_BASELINE,
JPG_MARKERS,
JPG_DCTALGO)
elif CAPTCHA_FILE_EXT == ".png":
gpdb.file_png_save2(img, drawable,
imageTmp, imageTmp,
PNG_INTERLACE,
PNG_COMPRESSION,
PNG_BKGD,
PNG_GAMA,
PNG_OFFSET,
PNG_PHYS,
PNG_TIME,
PNG_COMMENT,
PNG_SVGTRANS)
else:
return SystemExit("Image extension %r is not supported!"
% CAPTCHA_FILE_EXT)
except Exception as err:
print(err)
if os.path.isfile(imageTmp):
os.unlink(imageTmp)
else:
os.rename(imageTmp, imagePath)
def flip(file):
fileabs = os.path.abspath(file)
print(fileabs)
file_dir, filename = os.path.split(fileabs)
filename = filename.lower().replace('.png', '_mini.jpg')
file_new = os.path.join(file_dir, filename)
image = pdb.gimp_file_load(file, file)
drawable = pdb.gimp_image_get_active_layer(image)
#pdb.gimp_image_flip(image, ORIENTATION_HORIZONTAL)
#pdb.gimp_file_save(image, drawable, file_new, file_new)
pdb.file_jpeg_save(image, drawable, file_new, file_new, 0.8, 0, 1, 1, '', 0, 1, 0, 0)
pdb.gimp_image_delete(image)
def plugin_main(file):
"""
Main plugin method
"""
image = pdb.gimp_file_load(file, file)
drawable = pdb.gimp_image_get_active_layer(image)
fileabs = os.path.abspath(file)
#gprint('=== Input ==== ' + fileabs)
file_dir, filename = os.path.split(fileabs)
filename = filename[:-4] + '.jpg'
file_new = os.path.join(OUTPUT_PATH, filename)
pdb.file_jpeg_save(image, drawable, file_new, file_new, QUALITY, 0, 1, 1, '', 0, 1, 0, 0)
pdb.gimp_image_delete(image)
gprint('=== Output === ' + file_new)
def prepare_mbprint_jpg(img, drawable, output_folder):
""" Prepare a JPG image for MBPrint printing.
"""
gimp.progress_init('Prepare a JPG image for MBPrint printing...')
pdb.gimp_undo_push_group_start(img)
try:
file_name, back_side = _get_filename_backside(img, 'jpg')
except Exception: # pylint: disable=W0703
pdb.gimp_undo_push_group_end(img)
return
rotation = _get_rotation(drawable)
if rotation:
_rotate(drawable, back_side)
pdb.file_jpeg_save(img, drawable, os.path.join(output_folder, file_name),
file_name, 1, 0, 1, 0, '', 2, 1, 0, 0)
pdb.gimp_undo_push_group_end(img)
def captcha_generate(imageDir, goal):
"""Make sure there are as many catchas in image_dir as goal."""
needed = goal - countImages(imageDir)
if needed < 1:
return
answers = [selectAnswer(CAPTCHA_LETTERS) for i in xrange(0, needed)]
for answer in answers:
imageFile = '{0}{1}'.format(answer, CAPTCHA_FILE_EXT)
imagePath = os.path.join(imageDir, imageFile)
imageTmp = '%s.tmp' % imagePath
img, drawable = make_captcha(CAPTCHA_WIDTH, CAPTCHA_HEIGHT,
FONT_HEIGHT, LETTER_SPACING, LEFT_MARGIN,
ANGLE_RANGE, FONTS, answer)
try:
if CAPTCHA_FILE_EXT == ".jpg":
gpdb.file_jpeg_save(img, drawable, imageTmp, imageTmp,
JPG_QUALITY, JPG_SMOOTHING, JPG_OPTIMIZE,
JPG_PROGRESSIVE, JPG_COMMENT, JPG_SUBSMP,
JPG_BASELINE, JPG_MARKERS, JPG_DCTALGO)
elif CAPTCHA_FILE_EXT == ".png":
gpdb.file_png_save2(img, drawable, imageTmp, imageTmp,
PNG_INTERLACE, PNG_COMPRESSION, PNG_BKGD,
PNG_GAMA, PNG_OFFSET, PNG_PHYS, PNG_TIME,
PNG_COMMENT, PNG_SVGTRANS)
else:
return SystemExit("Image extension %r is not supported!" %
CAPTCHA_FILE_EXT)
except Exception as err:
print(err)
if os.path.isfile(imageTmp):
os.unlink(imageTmp)
else:
os.rename(imageTmp, imagePath)
def cutter_single_image(
image, drawable,
limit, sl_thresh, sz_thresh,
bg_manual, bg_color, bg_corner, bg_x, bg_y,
padding, deskew, sq_crop, autoclose,
save_same, save_dir, save_ftype, save_dpi, jpg_qual, save_suffix
):
img_width = pdb.gimp_image_width(image)
img_height = pdb.gimp_image_height(image)
img_fullpath = pdb.gimp_image_get_filename(image)
img_filename = os.path.basename(img_fullpath)
img_name = '.'.join(img_filename.split('.')[:-1])
img_ext = save_ftype or img_filename.split('.')[-1].lower()
new_filename_tpl = ''.join([
img_name, '-', save_suffix, '-',
'%0', str(len(str(int(limit + 1)))), 'd',
'.', img_ext
])
new_fullpath_tpl = os.path.join(
os.path.dirname(img_fullpath) if save_same else save_dir,
new_filename_tpl
)
# gimp.message(new_fullpath_tpl)
# function code goes here...
gimp.context_push()
pdb.gimp_image_undo_disable(image)
# If the background wasn't manually defined, pick the colour from one of the four corners
# (using radius 5 average)
if not bg_manual:
if bg_corner in (1, 3):
bg_x = img_width - bg_x
if bg_corner in (2, 3):
bg_y = img_height - bg_y
bg_color = pdb.gimp_image_pick_color(image, drawable, bg_x, bg_y, True, True, 5)
pdb.gimp_context_set_defaults()
pdb.gimp_context_set_antialias(True)
pdb.gimp_context_set_sample_transparent(True)
pdb.gimp_context_set_sample_threshold_int(sl_thresh)
pdb.gimp_context_set_feather(True)
fr = min(img_width, img_height) / 100.0 # NOTE why???
pdb.gimp_context_set_feather_radius(fr, fr)
pdb.gimp_context_set_background(bg_color)
pdb.gimp_image_select_color(image, gimpfu.CHANNEL_OP_REPLACE, drawable, bg_color)
# convert inverted copy of the background selection to a path
pdb.gimp_selection_sharpen(image)
pdb.gimp_selection_invert(image)
# _, before = pdb.gimp_image_get_vectors(image)
pdb.plug_in_sel2path(image, drawable)
# _, after = pdb.gimp_image_get_vectors(image)
# newpath_id = list(set(after) - set(before))[0]
# newpath = gimp.Vectors.from_id(newpath_id)
# looks like newly created vector is always active, so this should be sufficent
newpath = pdb.gimp_image_get_active_vectors(image)
pdb.gimp_context_set_feather(False)
_, strokes = pdb.gimp_vectors_get_strokes(newpath)
extracted = 0
for stroke_id in strokes:
stroke_points = pdb.gimp_vectors_stroke_get_points(newpath, stroke_id)
# skip not closed paths
if not stroke_points[3]:
continue
temp_vector = pdb.gimp_vectors_new(image, '-temp-')
pdb.gimp_image_insert_vectors(image, temp_vector, None, -1)
pdb.gimp_vectors_stroke_new_from_points(temp_vector, *stroke_points)
pdb.gimp_image_select_item(image, gimpfu.CHANNEL_OP_REPLACE, temp_vector)
pdb.gimp_image_remove_vectors(image, temp_vector)
# check for minimum size
bounds = pdb.gimp_selection_bounds(image)
sizex = bounds[3] - bounds[1]
sizey = bounds[4] - bounds[2]
if (min(sizex, sizey) < sz_thresh or
sizex >= img_width or
sizey >= img_height):
continue
buffname = "dsibuff"
if deskew and pdb.gimp_procedural_db_proc_exists('gimp_deskew_plugin'):
pdb.gimp_progress_set_text('Running deskew plugin...')
pdb.gimp_image_select_rectangle(
image, gimpfu.CHANNEL_OP_REPLACE,
bounds[1], bounds[2], sizex, sizey
)
buffname = pdb.gimp_edit_named_copy(drawable, buffname)
temp_image = pdb.gimp_edit_named_paste_as_new(buffname)
temp_layer = pdb.gimp_image_get_active_layer(temp_image)
pdb.gimp_image_undo_disable(temp_image)
pdb.gimp_layer_flatten(temp_layer)
# RUN_NONINTERACTIVE causes 'calling error' exception
pdb.gimp_deskew_plugin(temp_image, temp_layer, 0, 0, 0, 0, True,
run_mode=gimpfu.RUN_INTERACTIVE)
pdb.gimp_image_resize_to_layers(temp_image)
pdb.gimp_layer_flatten(temp_layer)
pdb.gimp_image_select_contiguous_color(
temp_image, gimpfu.CHANNEL_OP_REPLACE, temp_layer, 0, 0
)
pdb.gimp_selection_invert(temp_image)
bounds = pdb.gimp_selection_bounds(temp_image)
sizex = bounds[3] - bounds[1]
sizey = bounds[4] - bounds[2]
pdb.gimp_selection_none(temp_image)
pdb.gimp_image_crop(temp_image, sizex, sizey, bounds[1], bounds[2])
if (sq_crop and sizex != sizey
and pdb.gimp_procedural_db_proc_exists('script_fu_addborder')):
if sizex > sizey:
dx = 0
dy = (sizex - sizey) * 0.5
else:
dx = (sizey - sizex) * 0.5
dy = 0
pdb.script_fu_addborder(temp_image, temp_layer, dx, dy, bg_color, 0)
pdb.gimp_image_raise_item_to_top(temp_image, temp_layer)
pdb.gimp_image_merge_visible_layers(temp_image, gimpfu.EXPAND_AS_NECESSARY)
temp_layer = pdb.gimp_image_get_active_layer(temp_image)
else:
temp_image = image
pdb.gimp_image_undo_disable(temp_image)
temp_layer = pdb.gimp_image_get_active_layer(temp_image)
if sq_crop:
c_x = 0.5 * (bounds[1] + bounds[3])
c_y = 0.5 * (bounds[2] + bounds[4])
hl = padding + max(sizex, sizey) * 0.5
sel_x = c_x - hl
sel_y = c_y - hl
sel_w = sel_h = 2 * hl
else:
sel_x = bounds[1]
sel_y = bounds[2]
sel_w = sizex
sel_h = sizey
pdb.gimp_image_select_rectangle(
temp_image, gimpfu.CHANNEL_OP_REPLACE,
sel_x, sel_y, sel_w, sel_h
)
buffname = pdb.gimp_edit_named_copy(drawable, buffname)
temp_image = pdb.gimp_edit_named_paste_as_new(buffname)
temp_layer = pdb.gimp_image_get_active_layer(temp_image)
if padding and pdb.gimp_procedural_db_proc_exists('script_fu_addborder'):
pdb.script_fu_addborder(temp_image, temp_layer, padding, padding, bg_color, 0)
pdb.gimp_image_merge_visible_layers(temp_image, gimpfu.EXPAND_AS_NECESSARY)
temp_layer = pdb.gimp_image_get_active_layer(temp_image)
pdb.gimp_image_undo_enable(temp_image)
temp_display = pdb.gimp_display_new(temp_image)
extracted += 1
filename = new_fullpath_tpl % (extracted, )
pdb.gimp_image_set_resolution(temp_image, save_dpi, save_dpi)
if img_ext == 'jpg':
pdb.file_jpeg_save(
temp_image, temp_layer, filename, filename,
jpg_qual, 0.1, 1, 1, '', 2, 0, 0, 1
)
else:
pdb.gimp_file_save(temp_image, temp_layer, filename, filename)
if autoclose:
pdb.gimp_display_delete(temp_display)
if extracted >= limit:
break
pdb.gimp_progress_set_text('Extracted %d images' % (extracted, ))
pdb.gimp_image_remove_vectors(image, newpath)
pdb.gimp_selection_none(image)
pdb.gimp_image_undo_enable(image)
pdb.gimp_progress_end()
pdb.gimp_displays_flush()
gimp.context_pop()
return extracted