plugins/video/video.py

   1 import transcode, os, socket, re
   2 from Cheetah.Template import Template
   3 from plugin import Plugin
   4 from urllib import unquote_plus, quote, unquote
   5 from urlparse import urlparse
   6 from xml.sax.saxutils import escape
   7 from lrucache import LRUCache
   8 import Config
   9
  10 SCRIPTDIR = os.path.dirname(__file__)
  11
  12
  13 class video(Plugin):
  14
  15     content_type = 'x-container/tivo-videos'
  16     playable_cache = LRUCache(1000)
  17
  18     def SendFile(self, handler, container, name):
  19
  20         #No longer a 'cheep' hack :p
  21         if handler.headers.getheader('Range') and not handler.headers.getheader('Range') == 'bytes=0-':
  22             handler.send_response(206)
  23             handler.send_header('Connection', 'close')
  24             handler.send_header('Content-Type', 'video/x-tivo-mpeg')
  25             handler.send_header('Transfer-Encoding', 'chunked')
  26             handler.send_header('Server', 'TiVo Server/1.4.257.475')
  27             handler.end_headers()
  28             handler.wfile.write("\x30\x0D\x0A")
  29             return
  30
  31         tsn =  handler.headers.getheader('tsn', '')
  32
  33         o = urlparse("http://fake.host" + handler.path)
  34         path = unquote_plus(o[2])
  35         handler.send_response(200)
  36         handler.end_headers()
  37         transcode.output_video(container['path'] + path[len(name)+1:], handler.wfile, tsn)
  38
  39
  40
  41     def QueryContainer(self, handler, query):
  42
  43         subcname = query['Container'][0]
  44         cname = subcname.split('/')[0]
  45
  46         if not handler.server.containers.has_key(cname) or not self.get_local_path(handler, query):
  47             handler.send_response(404)
  48             handler.end_headers()
  49             return
  50
  51         path = self.get_local_path(handler, query)
  52         def isdir(file):
  53             return os.path.isdir(os.path.join(path, file))
  54
  55         def duration(file):
  56             full_path = os.path.join(path, file)
  57             return self.playable_cache[full_path]
  58
  59         def est_size(file):
  60             full_path = os.path.join(path, file)
  61             #Size is estimated by taking audio and video bit rate adding 2%
  62
  63             if transcode.tivo_compatable(full_path):  # Is TiVo compatible mpeg2
  64                 return int(os.stat(full_path).st_size)
  65             else:  # Must be re-encoded
  66                 audioBPS = strtod(Config.getAudioBR())
  67                 videoBPS = strtod(Config.getVideoBR())
  68                 bitrate =  audioBPS + videoBPS
  69                 return int((duration(file)/1000)*(bitrate * 1.02 / 8))
  70
  71         def VideoFileFilter(file):
  72             full_path = os.path.join(path, file)
  73
  74             if full_path in self.playable_cache:
  75                 return self.playable_cache[full_path]
  76             if os.path.isdir(full_path):
  77                 self.playable_cache[full_path] = True
  78                 return True
  79             millisecs = transcode.suported_format(full_path)
  80             if millisecs:
  81                 self.playable_cache[full_path] = millisecs
  82                 return millisecs
  83             else:
  84                 self.playable_cache[full_path] = False
  85                 return False
  86
  87         handler.send_response(200)
  88         handler.end_headers()
  89         t = Template(file=os.path.join(SCRIPTDIR,'templates', 'container.tmpl'))
  90         t.name = subcname
  91         t.files, t.total, t.start = self.get_files(handler, query, VideoFileFilter)
  92         t.duration = duration
  93         t.est_size = est_size
  94         t.isdir = isdir
  95         t.quote = quote
  96         t.escape = escape
  97         handler.wfile.write(t)
  98
  99
 100 # Parse a bitrate using the SI/IEEE suffix values as if by ffmpeg
 101 # For example, 2K==2000, 2Ki==2048, 2MB==16000000, 2MiB==16777216
 102 # Algorithm: http://svn.mplayerhq.hu/ffmpeg/trunk/libavcodec/eval.c
 103 def strtod(value):
 104     prefixes = {"y":-24,"z":-21,"a":-18,"f":-15,"p":-12,"n":-9,"u":-6,"m":-3,"c":-2,"d":-1,"h":2,"k":3,"K":3,"M":6,"G":9,"T":12,"P":15,"E":18,"Z":21,"Y":24}
 105     p = re.compile(r'^(\d+)(?:([yzafpnumcdhkKMGTPEZY])(i)?)?([Bb])?$')
 106     m = p.match(value)
 107     if m is None:
 108         raise SyntaxError('Invalid bit value syntax')
 109     (coef, prefix, power, byte) = m.groups()
 110     if prefix is None:
 111         value = float(coef)
 112     else:
 113         exponent = float(prefixes[prefix])
 114         if power == "i":
 115             # Use powers of 2
 116             value = float(coef) * pow(2.0, exponent/0.3)
 117         else:
 118             # Use powers of 10
 119             value = float(coef) * pow(10.0, exponent)
 120     if byte == "B": # B==Byte, b=bit
 121         value *= 8;
 122     return value