Esempio n. 1
0
class BaseClip:
    def __init__(self, clip):
        self.clip = CompositeVideoClip(clips=[clip])
        self.duration = self.clip.duration

    def resize(self, new_size):
        """
        Uses moviepy.video.fx.all.resize module
        :param new_size: Can be wither(width,height) in pixels or a float
                         A scaling factor, like 0.5
                         A function of time returning one of these.
        """
        self.clip = self.clip.resize(new_size)

    def crop(self,
             aspectRatio=None,
             x1=None,
             y1=None,
             x2=None,
             y2=None,
             width=None,
             height=None,
             x_center=None,
             y_center=None):
        """
        Uses moviepy.video.fx.crop module. From documentation:
        Returns a new clip in which just a rectangular subregion of the
        original clip is conserved. x1,y1 indicates the top left corner and
        x2,y2 is the lower right corner of the croped region.
        All coordinates are in pixels. Float numbers are accepted.
        :param x1: top left corner x-axis
        :param y1: top left corner y-axis
        :param x2: bottom right corner x-axis
        :param y2: bottom right corner y-axis
        :param width: width of rectangle
        :param height: height of rectangle
        :param x_center: x-axis center
        :param y_center: y-axis center
        """

        # If a preselected aspect ratio was selected.
        if aspectRatio:
            if not x_center:
                x_center = self.clip.w / 2
            if not y_center:
                y_center = self.clip.h / 2

            # Vertical/Phone ratio
            if aspectRatio == "vertical" or aspectRatio == "9:16" or aspectRatio == "phone":
                self.clip = self.clip.crop(width=self.clip.h * 9 / 16,
                                           height=self.clip.h,
                                           x_center=x_center,
                                           y_center=y_center)

            # Square ratio
            elif aspectRatio == "square" or aspectRatio == "1:1":
                self.clip = self.clip.crop(width=self.clip.h,
                                           height=self.clip.h,
                                           x_center=x_center,
                                           y_center=y_center)

            # 4:3/Letterbox ratio
            elif aspectRatio == "4:3" or aspectRatio == "1.33:1" or aspectRatio == "letterbox":
                self.clip = self.clip.crop(width=self.clip.h * 1.33,
                                           height=self.clip.h,
                                           x_center=x_center,
                                           y_center=y_center)

            # 16:9/Widescreen ratio
            elif aspectRatio == "16:9" or aspectRatio == "widescreen" or aspectRatio == "1.77:1":
                self.clip = self.clip.crop(width=self.clip.w,
                                           height=self.clip.w / 1.77,
                                           x_center=x_center,
                                           y_center=y_center)

            # 21:9/Cinemascope ratio
            elif aspectRatio == "cinemascope" or aspectRatio == "21:9" or aspectRatio == "2.33:1":
                self.clip = self.clip.crop(width=self.clip.w,
                                           height=self.clip.w / 2.33,
                                           x_center=x_center,
                                           y_center=y_center)

            # 2.35:1/Anamorphic ratio
            elif aspectRatio == "anamorphic" or aspectRatio == "2.35:1":
                self.clip = self.clip.crop(width=self.clip.w,
                                           height=self.clip.w / 2.35,
                                           x_center=x_center,
                                           y_center=y_center)

            # 2.39:1/DCI ratio
            elif aspectRatio == "DCI" or aspectRatio == "2.39:1":
                self.clip = self.clip.crop(width=self.clip.w,
                                           height=self.clip.w / 2.39,
                                           x_center=x_center,
                                           y_center=y_center)

            # 2.9:1/Digital IMAX ratio
            elif aspectRatio == "Digital IMAX" or aspectRatio == "2.9:1":
                self.clip = self.clip.crop(width=self.clip.w,
                                           height=self.clip.w / 2.9,
                                           x_center=x_center,
                                           y_center=y_center)

            # If an invalid aspect ratio was specified, raise an exception.
            else:
                raise AttributeError("Invalid Aspect Ratio specified: '" +
                                     str(aspectRatio) + "'")

        # If no preset ratio was selected, use other crop parameters.
        else:
            self.clip = self.clip.crop(x1=x1,
                                       y1=y1,
                                       x2=x2,
                                       y2=y2,
                                       width=width,
                                       height=height,
                                       x_center=x_center,
                                       y_center=y_center)

    def add_text(self, text, font_size, color, font, interline, posString,
                 duration):
        """
        Add a layer of text over the selected clip.

        :param text:
        :param font_size:
        :param color:
        :param font:
        :param interline:
        :param pos:
        :param duration:
        :return:
        """
        pos = (self.clip.w / 2, self.clip.h / 2)
        if posString == 'top':
            pos = (self.clip.w / 2, self.clip.h / (self.clip.h - 0.5))
        elif posString == 'left':
            pos = (self.clip.w / (self.clip.w - 0.5), self.clip.h / 2)
        elif posString == 'bottom':
            pos = (self.clip.w / 2, self.clip.h / 1.1)
        elif posString == 'right':
            pos = (self.clip.w / 1.1, self.clip.h / 2)
        elif posString == 'top-left':
            pos = (self.clip.w / (self.clip.w - 0.5),
                   self.clip.h / (self.clip.h - 0.5))
        elif posString == 'top-right':
            pos = (self.clip.w / 1.1, self.clip.h / (self.clip.h - 0.5))
        elif posString == 'bottom-left':
            pos = (self.clip.w / (self.clip.w - 0.5), self.clip.h / 1.1)
        elif posString == 'bottom-right':
            pos = (self.clip.w / 1.1, self.clip.h / 1.1)
        text = TextClip(
            text,
            fontsize=font_size,
            color=color,
            font=font,
            interline=interline).set_pos(pos).set_duration(duration)
        self.clip = CompositeVideoClip([self.clip, text])

    def addAudioFromFile(self, audio, start_time, end_time):
        """
        Uses moviepy.audio.io.AudioFileClip module. from Doc:
        An audio clip read from a sound file, or an array. The whole file is not loaded in memory.
        Instead, only a portion is read and stored in memory. this portion includes frames before and after the
        last frames read, so that it is fast to read the sound backward and forward.

        :param audio: audio file taken from directory (mp3, wav, etc)
        :return: adds audio to the clip being worked on (self.clip)


        This method works with the clip that was made and is stored on self.clip, which means it will alter the
        a clip that is already being made, not a new external clip. This is to avoid discrepancies when making
        new clips with or without overlay audio.
        """

        thisAudio = AudioFileClip(audio)
        changedAudio = thisAudio.subclip(start_time, end_time)
        self.clip = self.clip.set_audio(changedAudio)

    def addAudioFromClip(self, clipToExtract, start_time, end_time):
        """
        Instead of using an audio file like the method before this, it takes another video such as an mp4 file
        and rips the audio out of it, converts it into an AudioClip, and overlays it on the clip that is
        currently being worked on.

        ****This DOES NOT work with clips made through the VideoFileClip() method, since they have been processed
        as a different file type, and already have their own audio attribute. To access such, one just needs to call
        'clip'.audio, clip being your target clip for audio extraction.

        :param clipToExtract: video from directory (mp4, etc)
        :return: adds audio to the clip being worked on (self.clip)

        """

        thisAudio = AudioFileClip(clipToExtract)
        changedAudio = thisAudio.subclip(start_time, end_time)
        self.clip = self.clip.set_audio(changedAudio)

    def writeVideo(self, filename):
        """
        Write the video to a file.
        :param filename: name and format of output file.
        :return:
        """
        self.clip.write_videofile(filename)

    def create_gif(self, filename):
        # TODO: gif that loops fluidly
        self.clip.write_gif(filename)
Esempio n. 2
0
    def crop_scale(self, dimensions: Tuple[int, int]) -> 'Segment':
        """
        Returns
        -------
        A new Segment, cropped and/or scaled as necessary to reach specified dimensions
        """
        segment = self.copy()
        dimensions = Dimensions(*dimensions)

        def blur(image):
            #return cv2.GaussianBlur(image.astype(float),(99,99),0)
            return cv2.blur(image.astype(float), (30, 30), 0)

        #if segment.aspect_ratio != dimensions.aspect_ratio:
        # Crop segment to match aspect ratio
        #segment = segment.crop_to_aspect_ratio(dimensions.aspect_ratio)

        #if segment.dimensions != dimensions:
        # Resize segment to reach final dimensions
        #segment = segment.resize(dimensions)

        replace_width = dimensions.width
        replace_height = dimensions.height

        if segment.aspect_ratio != replace_width / replace_height:

            ##########################################Below 1 AR##################################################
            if segment.aspect_ratio <= 1:
                #print("below 1")
                if segment.size[0] != replace_width:
                    segment = segment.resize(width=replace_width)
                if segment.size[1] != replace_height:
                    segment = segment.resize(height=replace_height)

                segment = segment.set_position("center")
                background1 = segment.crop(x1=0, width=(segment.w / 2))
                background2 = segment.crop(x1=(segment.w / 2),
                                           width=(segment.w / 2))

                if segment.aspect_ratio != 1:
                    #print("Not 1:1")
                    background1 = background1.resize(
                        width=(replace_width - segment.w) / 2)
                    background2 = background2.resize(
                        width=((replace_width - segment.w) / 2) + 1)

                background1 = background1.set_position(
                    ("left", 'center')).fl_image(blur)
                background2 = background2.set_position(
                    ("right", 'center')).fl_image(blur)

                segment = CompositeVideoClip(
                    [background1, background2, segment],
                    size=(replace_width, replace_height))
                segment.effects = self.effects

        #########################################Above 1080 ratio###############################################
            elif segment.aspect_ratio > round(replace_width / replace_height,
                                              2):
                #print("above 1.7")
                if segment.size[1] != replace_height:
                    segment = segment.resize(height=replace_height)
                if segment.size[0] != replace_width:
                    segment = segment.resize(width=replace_width)

                test = (replace_height - segment.h) / 2
                segment = segment.set_position("center")
                background1 = segment.crop(
                    y1=0, height=((replace_height - segment.h) / 2))
                background2 = segment.crop(y1=segment.h - test, height=test)

                background1 = background1.set_position(
                    ('center', 'top')).fl_image(blur)
                background2 = background2.set_position(
                    ('center', 'bottom')).fl_image(blur)
                segment = CompositeVideoClip(
                    [background1, background2, segment],
                    size=(replace_width, replace_height))
                segment.effects = self.effects

    ######################################other#######################################################
            elif segment.aspect_ratio > 1 and segment.aspect_ratio < round(
                    replace_width / replace_height, 2):
                #print('midway between')
                if segment.size[0] != replace_width:
                    segment = segment.resize(width=replace_width)
                if segment.size[1] != replace_height:
                    segment = segment.resize(height=replace_height)

                segment = segment.set_position("center")
                background1 = segment.crop(x1=0, width=(segment.w / 2))
                background2 = segment.crop(x1=(segment.w / 2),
                                           width=(segment.w / 2))

                background1 = background1.set_position(
                    ("left", 'center')).fl_image(blur)
                background2 = background2.set_position(
                    ("right", 'center')).fl_image(blur)
                segment = CompositeVideoClip(
                    [background1, background2, segment],
                    size=(replace_width, replace_height))
                segment.effects = self.effects

        #############################################################################################

        if segment.w != replace_width and segment.h != replace_height:
            segment = segment.resize((replace_width, replace_height))
            #print("On Aspect, too big or small")

        return segment