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 commands
   6 import tv2avi
   7 import os
   8 import re
   9 import os.path
  10 import time
  11 import recdblist
  12 import configreader
  13
  14 def ts2mp4(pin, pout, opt):
  15     dir=os.path.split(pout)[0]
  16     title=os.path.split(pout)[1]
  17     title=os.path.splitext(title)[0]
  18     tpavi=os.path.join(dir, title+".avi")
  19     tv2avi.ts2avi(pin, tpavi, opt)
  20     time.sleep(10)
  21     if not os.path.exists(tpavi):
  22         if os.path.exists(pin):
  23             tpavi=pin
  24     avi2mp4(tpavi,pout,opt)
  25 def avi2mp4(pin,pout,opt):
  26     isvfr=0
  27     if re.search(u"v", opt):
  28         isvfr=1
  29     exe = configreader.getpath(u"MP4Box")
  30     dir=os.path.split(pin)[0]
  31     title=os.path.split(pin)[1]
  32     title=os.path.splitext(title)[0]
  33     audiopath=os.path.join(dir,title+u"_audio.raw")
  34     videopath=os.path.join(dir,title+u"_video.h264")
  35     exes=[]
  36     os.environ['LANG']="ja_JP.UTF-8"
  37     txt=""
  38     if isvfr==1:
  39         wineexe=configreader.getpath("wine")
  40         cfr2tc=configreader.getpath("cfr2tc")
  41         dtsedit=configreader.getpath("DtsEdit")
  42         pin120=pin.replace(".avi",".120.avi")
  43         pints=pin.replace(".avi",".ts")
  44         pmp4=pin.replace(".avi",".tmp.mp4")
  45         pintimecode=pin.replace(".avi",".timecode.txt")
  46         pinaac=pin.replace(".avi",".aac")
  47         evfr=[]
  48         evfr.append("ffmpeg -i '"+pints+"' -vn -f aac -acodec copy '"+pinaac+"'")
  49         evfr.append(wineexe+u" "+cfr2tc+u" '"+"Z:\\"+pin+"' '"+"Z:\\"+pin120+u"' 'Z:\\"+pintimecode+u"' 2")
  50         evfr.append(exe+u" -aviraw video \'"+pin120+u"\'")
  51         videotpath=os.path.join(dir,title+u".120_video.h264")
  52         evfr.append(exe+u" -fps 29.970030 -add \'"+videotpath+"\' -add \'"+pinaac+"\' -new \'"+pmp4+"\'")
  53         #evfr.append(exe+u" -add \'"+pinaac+"\' \'"+pmp4+"\'")
  54         evfr.append(wineexe+u" "+dtsedit+u" -tc \'Z:\\"+pintimecode+u"\' \'Z:\\"+pmp4+"\' -o \'Z:\\"+pout+"\'")
  55         exes=evfr
  56         for e in exes:
  57             recdblist.printutf8(e)
  58             txt=txt+e+"\n"
  59             try:
  60                 txt=txt+unicode(commands.getoutput(e.encode('utf-8')),'utf-8')
  61             except:
  62                 ""
  63         if (not os.path.exists(pin.replace(".avi",".120.avi"))and (not os.path.exists(pout))):
  64             txt=txt+"CFR2TC error. maybe avi file is over 1GB.(ODML error would have happend.)"
  65             ptmp=pin.replace(".avi",".noodml.avi")
  66             e="mencoder -noodml -forceidx -oac copy -ovc copy -o \'"+ptmp+"\'"+" \'"+pin+"\'"
  67             ex="nice -n 17 "+e
  68             e2=wineexe+u" "+cfr2tc+u" '"+"Z:\\"+ptmp+"' '"+"Z:\\"+pin120+u"' '"+"Z:\\"+pintimecode+u"' 2"
  69             ex2="nice -n 17 "+e2
  70             e3=exe+u" -aviraw video \'"+pin120+u"\'"
  71             ex3="nice -n 17 "+e3
  72             e4=exe+u" -fps 29.97 -add \'"+videotpath+"\' -add \'"+pinaac+"\' -new \'"+pmp4+"\'"
  73             ex4="nice -n 17 "+e4
  74             e5=wineexe+u" "+dtsedit+u" -tc \'Z:\\"+pintimecode+u"\' \'Z:\\"+pmp4+"\' -o \'Z:\\"+pout+"\'"
  75             ex5="nice -n 17 "+e5
  76             recdblist.printutf8(e)
  77             txt=txt+e+"\n"
  78             try:
  79                 cmdt=unicode(commands.getoutput(ex.encode('utf-8')),'utf-8')
  80                 cmdt=cmdt+unicode(commands.getoutput(ex2.encode('utf-8')),'utf-8')
  81                 cmdt=cmdt+unicode(commands.getoutput(ex3.encode('utf-8')),'utf-8')
  82                 cmdt=cmdt+unicode(commands.getoutput(ex4.encode('utf-8')),'utf-8')
  83                 cmdt=cmdt+unicode(commands.getoutput(ex5.encode('utf-8')),'utf-8')
  84                 os.remove(ptmp)
  85             except:
  86                 ""
  87             txt=txt+cmdt+"\n"
  88     else:
  89         exes.append(exe+u" -aviraw audio \'"+pin+u"\'")
  90         exes.append(exe+u" -aviraw video \'"+pin+u"\'")
  91         exes.append(exe+u" -new -fps 29.970030 -add \'"+videopath+u"\'#video -add \'"+audiopath+u"\'#audio \'"+pout+u"\'")
  92         for e in exes:
  93             recdblist.printutf8(e)
  94             txt=txt+e+"\n"
  95             try:
  96                 txt=txt+unicode(commands.getoutput(e.encode('utf-8')),'utf-8')
  97             except:
  98                 ""
  99     logo=pin.replace(u".avi",u".log")
 100     txt = "\n####MP4Box-log####\n"+txt
 101     f=open(logo,'a')
 102     f.write(txt.encode('utf-8'))
 103     f.close()
 104     time.sleep(10)
 105     delpath=[]
 106     delpath.append(pin.replace(u".avi",u".120.avi"))
 107     delpath.append(pin.replace(u".avi",u".sa.avi"))
 108     delpath.append(pin.replace(u".avi",u".ts"))
 109     delpath.append(pin.replace(u".avi",u".aac"))
 110     delpath.append(pin.replace(u".avi",u".m2v"))
 111     delpath.append(pin.replace(u".avi",u".120_video.h264"))
 112     delpath.append(pin.replace(u".avi",u".ts.b25"))
 113     if os.path.exists(pout):
 114         if os.path.exists(audiopath):
 115             os.remove(audiopath)
 116         if os.path.exists(videopath):
 117             os.remove(videopath)
 118         """
 119         for dp in delpath:
 120             try:
 121                 os.remove(dp)
 122             except:
 123                 ""
 124         """
 125 def mkv2mp4(pin,pout):
 126     exeb = configreader.getpath(u"mkvextract")
 127     exe = configreader.getpath(u"MP4Box")
 128     dtsedit=configreader.getpath("DtsEdit")
 129     wineexe=configreader.getpath("wine")
 130     dir=os.path.split(pin)[0]
 131     title=os.path.split(pin)[1]
 132     title=os.path.splitext(title)[0]
 133     audiopath=os.path.join(dir,title+u"_audio.raw")
 134     videopath=os.path.join(dir,title+u"_video.h264")
 135     timecodepath=os.path.join(dir,title+u"_1_timecode.txt")
 136     tmpmp4=os.path.join(dir,title+u".tmp.mp4")
 137     exe0=exeb+u" tracks \'"+pin+u"\' 1:\'"+videopath+u"\' 2:\'"+audiopath+u"\'"
 138     exe1=exeb+u" timecodes_v2 \'"+pin+"\' 1:\'"+timecodepath+"\'"
 139     exe2=exe+u" -new -fps 29.970030 -add \'"+videopath+u"\'#video -add \'"+audiopath+u"\'#audio \'"+tmpmp4+u"\'"
 140     exe3=wineexe+u" "+dtsedit+u" -tc \'Z:\\"+timecodepath+u"\' \'Z:\\"+tmpmp4+u"\' -o \'Z:\\"+pout+u"\'"
 141     logo=pin.replace(".avi",".log")
 142     os.environ['LANG']="ja_JP.UTF-8"
 143     txt= exe0+"\n"+exe1+"\n"+exe2+"\n"+exe3+"\n"
 144     try:
 145         txt = txt+unicode(commands.getoutput(exe0.encode('utf-8')),'utf-8')
 146         txt = txt+unicode(commands.getoutput(exe1.encode('utf-8')),'utf-8')
 147         txt = txt+unicode(commands.getoutput(exe2.encode('utf-8')),'utf-8')
 148         txt = txt+unicode(commands.getoutput(exe3.encode('utf-8')),'utf-8')
 149     except:
 150         ""
 151     os.remove(audiopath)
 152     os.remove(videopath)
 153     txt = "\n####MKV2MP4-log####\n"+txt
 154     f=open(logo,'a')
 155     f.write(txt.encode('utf-8'))
 156     f.close()
 157     time.sleep(10)
 158     if os.path.exists(pout):
 159         if os.path.exists(audiopath):
 160             os.remove(audiopath)
 161         if os.path.exists(videopath):
 162             os.remove(videopath)