pandora_mlt/timeline.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
from __future__ import division, with_statement
import os
import sys
import json

import ox

MIN_DURATION = 2

profiles = {
    'pal': '<profile width="720" display_aspect_den="3" colorspace="601" frame_rate_den="1" description="DV/DVD PAL" height="576" display_aspect_num="4" frame_rate_num="25" progressive="0" sample_aspect_num="16" sample_aspect_den="15"/>'
}

class Video:
    position = 0
    playlist = []
    producers = {}

    template = '''<?xml version='1.0' encoding='utf-8'?>
<mlt title="%(title)s" version="0.8.0" root="%(root)s" LC_NUMERIC="en_US.UTF-8">
 %(profile)s
 <producer in="0" out="72869" id="black">
  <property name="mlt_type">producer</property>
  <property name="length">72870</property>
  <property name="eof">pause</property>
  <property name="resource">black</property>
  <property name="aspect_ratio">0</property>
  <property name="mlt_service">colour</property>
 </producer>
 <playlist id="black_track">
  <entry in="0" out="12047" producer="black"/>
 </playlist>
 <playlist id="playlist1"/>
 <playlist id="playlist2"/>
 <playlist id="playlist3"/>

 %(producers)s

 <playlist id="playlist4">
  %(playlist)s
 </playlist>
 <playlist id="playlist5"/>
 <tractor title="%(title)s" global_feed="1" in="0" out="12047" id="maintractor">
  <property name="meta.volume">1</property>
  <track producer="black_track"/>
  <track hide="video" producer="playlist1"/>
  <track hide="video" producer="playlist2"/>
  <track producer="playlist3"/>
  <track producer="playlist4"/>
  <track producer="playlist5"/>
  <transition id="transition0">
   <property name="a_track">1</property>
   <property name="b_track">2</property>
   <property name="mlt_type">transition</property>
   <property name="mlt_service">mix</property>
   <property name="always_active">1</property>
   <property name="combine">1</property>
   <property name="internal_added">237</property>
  </transition>
  <transition id="transition1">
   <property name="a_track">1</property>
   <property name="b_track">3</property>
   <property name="mlt_type">transition</property>
   <property name="mlt_service">mix</property>
   <property name="always_active">1</property>
   <property name="combine">1</property>
   <property name="internal_added">237</property>
  </transition>
  <transition id="transition2">
   <property name="a_track">1</property>
   <property name="b_track">4</property>
   <property name="mlt_type">transition</property>
   <property name="mlt_service">mix</property>
   <property name="always_active">1</property>
   <property name="combine">1</property>
   <property name="internal_added">237</property>
  </transition>
  <transition id="transition3">
   <property name="a_track">1</property>
   <property name="b_track">5</property>
   <property name="mlt_type">transition</property>
   <property name="mlt_service">mix</property>
   <property name="always_active">1</property>
   <property name="combine">1</property>
   <property name="internal_added">237</property>
  </transition>
 </tractor>
 <kdenlivedoc profile="dv_pal" kdenliveversion="0.9.2" version="0.88" projectfolder="%(root)s">
  <customeffects/>
  <documentproperties proxyimageminsize="2000" zonein="0" enableproxy="0" zoneout="100" generateproxy="0" zoom="9" verticalzoom="1" proxyextension="ts" position="1" documentid="1353193897205" generateimageproxy="0" proxyminsize="1000" proxyparams="-f mpegts -acodec libmp3lame -ac 2 -ab 128k -ar 48000 -vcodec mpeg2video -g 5 -deinterlace -s 480x270 -vb 400k"/>
  <documentmetadata/>
  <documentnotes>&lt;!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.0//EN" "http://www.w3.org/TR/REC-html40/strict.dtd">
&lt;html>&lt;head>&lt;meta name="qrichtext" content="1" />&lt;style type="text/css">
p, li { white-space: pre-wrap; }
&lt;/style>&lt;/head>&lt;body style=" font-family:'Sans Serif'; font-size:8pt; font-weight:400; font-style:normal;">
&lt;p style="-qt-paragraph-type:empty; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;">&lt;br />&lt;/p>&lt;/body>&lt;/html></documentnotes>
  <profileinfo width="720" display_aspect_den="3" frame_rate_den="1" description="DV/DVD PAL" height="576" frame_rate_num="25" display_aspect_num="4" progressive="0" sample_aspect_num="16" sample_aspect_den="15"/>
  <tracksinfo>
   <trackinfo blind="1" mute="0" locked="0" trackname="Audio 2" type="audio"/>
   <trackinfo blind="1" mute="0" locked="0" trackname="Audio 1" type="audio"/>
   <trackinfo blind="0" mute="0" locked="0" trackname="Video 3"/>
   <trackinfo blind="0" mute="0" locked="0" trackname="Video 2"/>
   <trackinfo blind="0" mute="0" locked="0" trackname="Video 1"/>
  </tracksinfo>
  %(kdenlive_producers)s
  <markers/>
  <groups/>
 </kdenlivedoc>
</mlt>
'''

    def __init__(self, data, title, url, root):
        files = []
        items = []
        for c in data:
            files.append(c['path'])
            items.append(c['id'])

        files = list(set(files))
        items = list(set(items))
        self.files = files
        self.clips = data

        self.fps = 25
        profile = profiles[self.fps == 25 and 'pal' or 'ntsc']

        self.info = {
            'profile': profile,
            'root': root,
            'title': title,
            'url': url,
            'producers': '',
            'kdenlive_producers': '',
            'duration': 0,
            'stats': '%d clips from %d films' % (len(self.clips), len(items))
        }
        self.add_title()
        for clip in self.clips:
            path = clip['path']
            if not path.startswith('/'):
                path = os.path.join('/data/Cinema', clip['path'])
            out = clip['out']
            if out-clip['in'] < MIN_DURATION:
                out = clip['in'] + MIN_DURATION
            duration = int((out-clip['in']) * self.fps)
            self.add_clip(path, clip['in'], out)
            self.info['duration'] += duration

    def render(self, output):
        xml = output
        with open(xml, 'w') as f:
            self.info['playlist'] = '\n'.join(self.playlist)
            data = self.template % self.info
            f.write(data.strip())
        #os.system('melt -progress "%s" -consumer "avformat:%s"'%(xml, output))

    def add_title(self):
        self.info['duration'] = 5*self.fps
        self.info['producers'] += '''
         <producer in="0" out="%(duration)s" id="1">
          <property name="mlt_type">producer</property>
          <property name="length">%(duration)s</property>
          <property name="eof">pause</property>
          <property name="resource"/>
          <property name="mlt_service">kdenlivetitle</property>
          <property name="xmldata">&lt;kdenlivetitle width="768" height="576" out="125" LC_NUMERIC="en_US.UTF-8">
         &lt;item z-index="2" type="QGraphicsTextItem">
          &lt;position x="86" y="123">
           &lt;transform>1,0,0,0,1,0,0,0,1&lt;/transform>
          &lt;/position>
          &lt;content font-color="255,255,255,255" font-outline-color="0,0,0,255" font-pixel-size="151" font-italic="0" alignment="4" font-underline="0" font-weight="50" font="Arial Black" font-outline="0.5">%(title)s&lt;/content>
         &lt;/item>
         &lt;item z-index="1" type="QGraphicsTextItem">
          &lt;position x="79" y="321">
           &lt;transform>1,0,0,0,1,0,0,0,1&lt;/transform>
          &lt;/position>
          &lt;content font-color="255,255,255,255" font-outline-color="0,0,0,255" font-pixel-size="24" font-italic="0" alignment="4" font-underline="0" font-weight="50" font="Arial Black" font-outline="0.5">%(url)s
%(stats)s&lt;/content>
         &lt;/item>
         &lt;startviewport rect="0,0,768,576"/>
         &lt;endviewport rect="0,0,768,576"/>
         &lt;background color="0,0,0,0"/>
        &lt;/kdenlivetitle>
        </property>
          <property name="force_reload">0</property>
         </producer>
        ''' % self.info
        self.playlist.append('<entry in="%s" out="%s" producer="1"/>' % (0, self.info['duration']))
        self.info['kdenlive_producers'] += '''
            <kdenlive_producer audio_max="0" id="1" default_video="0" xmldata="&lt;kdenlivetitle width=&quot;768&quot; height=&quot;576&quot; out=&quot;125&quot; LC_NUMERIC=&quot;en_US.UTF-8&quot;>&#xa; &lt;item z-index=&quot;2&quot; type=&quot;QGraphicsTextItem&quot;>&#xa;  &lt;position x=&quot;86&quot; y=&quot;123&quot;>&#xa;   &lt;transform>1,0,0,0,1,0,0,0,1&lt;/transform>&#xa;  &lt;/position>&#xa;  &lt;content font-color=&quot;255,255,255,255&quot; font-outline-color=&quot;0,0,0,255&quot; font-pixel-size=&quot;151&quot; font-italic=&quot;0&quot; alignment=&quot;4&quot; font-underline=&quot;0&quot; font-weight=&quot;50&quot; font=&quot;Arial Black&quot; font-outline=&quot;0.5&quot;>%(title)s&lt;/content>&#xa; &lt;/item>&#xa; &lt;item z-index=&quot;1&quot; type=&quot;QGraphicsTextItem&quot;>&#xa;  &lt;position x=&quot;79&quot; y=&quot;321&quot;>&#xa;   &lt;transform>1,0,0,0,1,0,0,0,1&lt;/transform>&#xa;  &lt;/position>&#xa;  &lt;content font-color=&quot;255,255,255,255&quot; font-outline-color=&quot;0,0,0,255&quot; font-pixel-size=&quot;24&quot; font-italic=&quot;0&quot; alignment=&quot;4&quot; font-underline=&quot;0&quot; font-weight=&quot;50&quot; font=&quot;Arial Black&quot; font-outline=&quot;0.5&quot;>%(url)s&#xa;%(stats)s&lt;/content>&#xa; &lt;/item>&#xa; &lt;startviewport rect=&quot;0,0,768,576&quot;/>&#xa; &lt;endviewport rect=&quot;0,0,768,576&quot;/>&#xa; &lt;background color=&quot;0,0,0,0&quot;/>&#xa;&lt;/kdenlivetitle>&#xa;" name="Title" in="0" thumbnail="100" transparency="1" default_audio="0" duration="125" aspect_ratio="0.000000" channels="0" frequency="0" out="125" video_max="0" progressive="0" type="6" frame_size="720x576"/>
        ''' % self.info
    def add_producer(self, uri):
        c = len(self.producers.keys()) + 2
        self.producers[uri] = c
        info = ox.avinfo(uri)
        i = {
            'id': c,
            'duration': int(info['duration'] * self.fps),
            'filename': uri.encode('utf-8'),
            'basename': os.path.basename(uri).encode('utf-8')
        }
        i['out'] = i['duration'] - 1
        self.info['producers'] += '''
         <producer in="0" out="%(out)s" id="%(id)s_1">
          <property name="mlt_type">producer</property>
          <property name="length">%(duration)s</property>
          <property name="eof">pause</property>
          <property name="resource">%(filename)s</property>
          <property name="seekable">1</property>
          <property name="mlt_service">avformat</property>
         </producer>
         ''' % i
        self.info['kdenlive_producers'] += '''
          <kdenlive_producer
              id="%(id)s"
              name="%(basename)s"
              resource="%(filename)s"
          />
         ''' % i

    def get_producer(self, uri):
        if not uri in self.producers:
            self.add_producer(uri)
        return self.producers[uri]

    def add_clip(self, uri, start, end):
        start = int(start * self.fps)
        end = int(end * self.fps) - 1
        o = '<entry in="%s" out="%s" producer="%d_1"/>' % (start, end, self.get_producer(uri))
        self.playlist.append(o)

if __name__ == '__main__':

    if len(sys.argv) != 5:
        print 'usage: %s <json input> <video prefix> <output> <title>' % sys.argv[0]
        sys.exit(1)
    json_input = sys.argv[1]
    video_prefix = sys.argv[2]
    root = sys.argv[3]
    title = sys.argv[4]
    output = json_input + '.kdenlive'
    url = 'http://pan.do/ra'
    
    data = json.load(open(json_input))
    c = Video(data,
        title,
        url,
        root)
    c.render(output)