59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from moviepy.decorators import apply_to_mask
|
||
|
|
from moviepy.video.VideoClip import ImageClip
|
||
|
|
|
||
|
|
|
||
|
|
@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)
|