rec10/trunk/src/tv2mp4.py

   1 #!/usr/bin/python
   2 # coding: UTF-8
   3 # Rec10 TS Recording Tools
   4 # Copyright (C) 2009 Yukikaze
   5 import sys
   6 import commands
   7 import tv2avi
   8 import os
   9 import os.path
  10 import time
  11 import configreader
  12 def ts2mp4(pin, pout, opt):
  13     dir=os.path.split(pout)[0]
  14     title=os.path.split(pout)[1]
  15     title=os.path.splitext(title)[0]
  16     tpavi=os.path.join(dir, title+".avi")
  17     tv2avi.ts2avi(pin, tpavi, opt)
  18     time.sleep(10)
  19     if not os.path.exists(tpavi):
  20         if os.path.exists(pin):
  21             tpavi=pin
  22     avi2mp4(tpavi,pout)
  23 def avi2mp4(pin,pout):
  24     exe = configreader.getpath(u"MP4Box")
  25     dir=os.path.split(pin)[0]
  26     title=os.path.split(pin)[1]
  27     title=os.path.splitext(title)[0]
  28     audiopath=os.path.join(dir,title+u"_audio.mp3")
  29     videopath=os.path.join(dir,title+u"_video.h264")
  30     #doexe = exe.encode('utf-8') + u' \"' + pin.encode('utf-8') + u'\" \"' + pout.encode('utf-8') + u'\"'
  31     exe1=exe+u" -aviraw audio u'"+pin+u"\'"
  32     exe2=exe+u" -aviraw video \'"+pin+u"\'"
  33     exefin=exe+u" -new -fps 29.97 -add \'"+videopath+u"\'#video -add \'"+audiopath+u"\'#audio \'"+pout+u"\'"
  34     logo=pin.replace(u".avi",u".log")
  35     os.environ['LANG']="ja_JP.UTF-8"
  36     txt= exe1+u"\n"+exe2+u"\n"+exefin+u"\n"
  37     txt = txt+unicode(commands.getoutput(exe1.encode('utf-8')),'utf-8')
  38     txt = txt+unicode(commands.getoutput(exe2.encode('utf-8')),'utf-8')
  39     txt = txt+unicode(commands.getoutput(exefin.encode('utf-8')),'utf-8')
  40     os.remove(audiopath)
  41     os.remove(videopath)
  42     txt = "\n####MP4Box-log####\n"+txt
  43     f=open(logo,'a')
  44     f.write(txt)
  45     f.close()
  46     sys.stdout.flush()
  47     time.sleep(10)
  48     if os.path.exists(pout):
  49         if os.path.exists(audiopath):
  50             os.remove(audiopath)
  51         if os.path.exists(videopath):
  52             os.remove(videopath)
  53 def mkv2mp4(pin,pout):
  54     exeb = configreader.getpath(u"mkvextract")
  55     exe = configreader.getpath(u"MP4Box")
  56     dir=os.path.split(pin)[0]
  57     title=os.path.split(pin)[1]
  58     title=os.path.splitext(title)[0]
  59     audiopath=os.path.join(dir,title+u"_audio.mp3")
  60     if not os.path.exists(audiopath):
  61         if os.path.exists(os.path.join(dir,title+u"_audio.raw")):
  62             audiopath=os.path.join(dir,title+u"_audio.raw")
  63         if os.path.exists(os.path.join(dir,title+u"_audio.aac")):
  64             audiopath=os.path.join(dir,title+u"_audio.aac")
  65     videopath=os.path.join(dir,title+u"_video.h264")
  66     #doexe = exe.encode('utf-8') + u' \"' + pin.encode('utf-8') + u'\" \"' + pout.encode('utf-8') + u'\"'
  67     exe0=exeb+u" tracks \'"+pin+u"\' 1:\'"+videopath+u"\' 2:\'"+audiopath+u"\'"
  68     exe1=exe+u" -aviraw audio \'"+pin+u"\'"
  69     exe2=exe+u" -aviraw video \'"+pin+u"\'"
  70     exefin=exe+u" -new -fps 29.97 -add \'"+videopath+u"\'#video -add \'"+audiopath+u"\'#audio \'"+pout+u"\'"
  71     logo=pin.replace(".avi",".log")
  72     os.environ['LANG']="ja_JP.UTF-8"
  73     txt= exe0+"\n"+exe1+"\n"+exe2+"\n"+exefin+"\n"
  74     txt = txt+unicode(commands.getoutput(exe0.encode('utf-8')),'utf-8')
  75     txt = txt+unicode(commands.getoutput(exe1.encode('utf-8')),'utf-8')
  76     txt = txt+unicode(commands.getoutput(exe2.encode('utf-8')),'utf-8')
  77     txt = txt+unicode(commands.getoutput(exefin.encode('utf-8')),'utf-8')
  78     os.remove(audiopath)
  79     os.remove(videopath)
  80     txt = "\n####MKV2MP4-log####\n"+txt
  81     f=open(logo,'a')
  82     f.write(txt)
  83     f.close()
  84     sys.stdout.flush()
  85     time.sleep(10)
  86     if os.path.exists(pout):
  87         if os.path.exists(audiopath):
  88             os.remove(audiopath)
  89         if os.path.exists(videopath):
  90             os.remove(videopath)