import numpy as np
from moviepy.decorators import apply_to_mask
from moviepy.video.VideoClip import ImageClip
[docs]@apply_to_mask
def margin(clip, mar=None, left=0, right=0, top=0,
bottom=0, color=(0, 0, 0), opacity = 1.0):
"""
Draws an external margin all around the frame.
:param mar: if not ``None``, then the new clip has a margin of
size ``mar`` in pixels on the left, right, top, and bottom.
:param left, right, top, bottom: width of the margin in pixel
in these directions.
:param color: color of the margin.
:param mask_margin: value of the mask on the margin. Setting
this value to 0 yields transparent margins.
"""
if (opacity != 1.0) and (clip.mask is None) and not (clip.ismask):
clip = clip.add_mask()
if mar is not None:
left = right = top = bottom = mar
def make_bg(w,h):
new_w, new_h = w + left + right, h + top + bottom
if clip.ismask:
shape = (new_h, new_w)
bg = ( np.tile(opacity, (new_h, new_w))
.astype(float)
.reshape(shape))
else:
shape = (new_h, new_w, 3)
bg = np.tile(color, (new_h, new_w)).reshape(shape)
return bg
if isinstance(clip, ImageClip):
im = make_bg(clip.w,clip.h)
im[top:top + clip.h, left:left + clip.w] = clip.img
return clip.fl_image(lambda pic:im)
else:
def fl(gf, t):
pic = gf(t)
h,w = pic.shape[:2]
im = make_bg(w,h)
im[top:top + h, left:left + w] = pic
return im
return clip.fl(fl)