datamosh.py

import bpy, os
from bpy.props import IntProperty, StringProperty, BoolProperty
import subprocess

from . import functions


proxy_qualities = [  ( "1", "25%", "" ), ( "2", "50%", "" ),
                    ( "3", "75%", "" ), ( "4", "100%", "" ),
                     ( "5", "none", "" )]


# functions

def createdatamosh(context, strip):
    preferences = context.user_preferences
    prefs = preferences.addons[__package__].preferences

    fileinput = bpy.path.abspath(strip.filepath)
    fileoutput = fileinput.rpartition(".")[0]+"_datamosh.avi"

    if prefs.all_keyframes:
        command = "datamosh '{}' -a -o '{}'".format(fileinput, fileoutput)
    else:
        command = "datamosh '{}' -o '{}'".format(fileinput, fileoutput)
    print(command)
    os.system(command)
    return fileoutput

def createavi(context, strip):
    preferences = context.user_preferences
    prefs = preferences.addons[__package__].preferences

    fileinput = bpy.path.abspath(strip.filepath)
    fileoutput = fileinput.rpartition(".")[0]+"_.avi"

    command = "ffmpeg -i '{}' -vcodec copy '{}'".format(fileinput, fileoutput)

    print(command)
    os.system(command)
    return fileoutput

def createavimjpeg(context, strip):
    preferences = context.user_preferences
    prefs = preferences.addons[__package__].preferences

    fileinput = bpy.path.abspath(strip.filepath)
    fileoutput = fileinput.rpartition(".")[0]+"_mjpeg.avi"

    command = "ffmpeg -i '{}' -vcodec mjpeg -q:v 1 '{}'".format(fileinput, fileoutput)

    print(command)
    os.system(command)
    return fileoutput




# classes

class CreateAvi(bpy.types.Operator):
    """ """
    bl_idname = "sequencer.createavi"
    bl_label = "create avi file"

    @classmethod
    def poll(self, context):
        strip = functions.act_strip(context)
        scn = context.scene
        if scn and scn.sequence_editor and scn.sequence_editor.active_strip:
            return strip.type in ('MOVIE')
        else:
            return False

    size = IntProperty(
    name='proxysize',
    default=1)
    bl_options = {'REGISTER', 'UNDO'}

    def execute(self, context):

        preferences = context.user_preferences
        strips = functions.get_selected_strips(context)

        for strip in strips:
            #deselect all other strips 
            for i in strips: i.select = False
            #select current strip
            strip.select = True
            if strip.type == "MOVIE":
                if self.size == 1:
                    fileoutput = createavi(context, strip)
                elif self.size == 2:
                    fileoutput = createavimjpeg(context, strip)
                strip.filepath = fileoutput
        #select all strips again
        for strip in strips:
            try: 
                strip.select=True
            except ReferenceError:
                pass
        bpy.ops.sequencer.reload()
        return {'FINISHED'}


class CreateDatamosh(bpy.types.Operator):
    """ """
    bl_idname = "sequencer.createdatamosh"
    bl_label = "create datamosh"

    @classmethod
    def poll(self, context):
        strip = functions.act_strip(context)
        scn = context.scene
        if scn and scn.sequence_editor and scn.sequence_editor.active_strip:
            return strip.type in ('MOVIE')
        else:
            return False

    def execute(self, context):

        preferences = context.user_preferences
        prefs = preferences.addons[__package__].preferences
        strips = functions.get_selected_strips(context)

        for strip in strips:
            #deselect all other strips 
            for i in strips: i.select = False
            #select current strip
            strip.select = True
            if strip.type == "MOVIE":
                fileoutput = createdatamosh(context, strip)
                if prefs.load_glitch:
                    strip.filepath = fileoutput
        #select all strips again
        for strip in strips:
            try: 
                strip.select=True
            except ReferenceError:
                pass
        bpy.ops.sequencer.reload()
        return {'FINISHED'}


class CreateGlitchToolPanel(bpy.types.Panel):
    """  """
    bl_label = "Glitch Tools"
    bl_idname = "OBJECT_PT_GlitchTool"
    bl_space_type = 'SEQUENCE_EDITOR'
    bl_region_type = 'UI'


    @classmethod
    def poll(self, context):
        if context.space_data.view_type in {'SEQUENCER',
            'SEQUENCER_PREVIEW'}:
            strip = functions.act_strip(context)
            scn = context.scene
            preferences = context.user_preferences
            prefs = preferences.addons[__package__].preferences
            if scn and scn.sequence_editor and scn.sequence_editor.active_strip:
                if prefs.use_glitch_panel:
                    return strip.type in ('MOVIE')
        else:
            return False

    def draw_header(self, context):
        layout = self.layout
        layout.label(text="", icon="GAME")

    def draw(self, context):

        preferences = context.user_preferences
        prefs = preferences.addons[__package__].preferences

        layout = self.layout

        layout.operator("sequencer.createavi", text="create avi (same codec)")
        layout.operator("sequencer.createavi", text="create avi (mjpeg)").size=2

        strip = functions.act_strip(context)

        layout.prop(prefs,"all_keyframes")
        layout.prop(prefs,"load_glitch")

        layout.operator("sequencer.createdatamosh")