generated from thinkode/modelRepository
32 lines
947 B
Python
32 lines
947 B
Python
|
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
from moviepy.Clip import Clip
|
||
|
|
from moviepy.Effect import Effect
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class MultiplySpeed(Effect):
|
||
|
|
"""Returns a clip playing the current clip but at a speed multiplied by ``factor``.
|
||
|
|
|
||
|
|
Instead of factor one can indicate the desired ``final_duration`` of the clip, and
|
||
|
|
the factor will be automatically computed. The same effect is applied to the clip's
|
||
|
|
audio and mask if any.
|
||
|
|
"""
|
||
|
|
|
||
|
|
factor: float = None
|
||
|
|
final_duration: float = None
|
||
|
|
|
||
|
|
def apply(self, clip: Clip) -> Clip:
|
||
|
|
"""Apply the effect to the clip."""
|
||
|
|
if self.final_duration:
|
||
|
|
self.factor = 1.0 * clip.duration / self.final_duration
|
||
|
|
|
||
|
|
new_clip = clip.time_transform(
|
||
|
|
lambda t: self.factor * t, apply_to=["mask", "audio"]
|
||
|
|
)
|
||
|
|
|
||
|
|
if clip.duration is not None:
|
||
|
|
new_clip = new_clip.with_duration(1.0 * clip.duration / self.factor)
|
||
|
|
|
||
|
|
return new_clip
|