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