rec10/trunk/src/tv2mp4.py

   1 #!/usr/bin/python\r
   2 # coding: UTF-8\r
   3 # Rec10 TS Recording Tools\r
   4 # Copyright (C) 2009-2010 Yukikaze\r
   5 import commands\r
   6 import shutil\r
   7 import tv2avi\r
   8 import os\r
   9 import re\r
  10 import os.path\r
  11 import string\r
  12 import base64\r
  13 import time\r
  14 import recdblist\r
  15 import configreader\r
  16 import status\r
  17 import subprocess\r
  18 import tv2audio\r
  19 \r
  20 def ts2mp4(pin, pout, opt):\r
  21     dir=os.path.split(pout)[0]\r
  22     title=os.path.split(pout)[1]\r
  23     title=os.path.splitext(title)[0]\r
  24     tpraw=os.path.join(dir, title+".264")\r
  25     tpmp4=os.path.join(dir, title+".mp4")\r
  26     if os.path.isfile(pin) and os.path.getsize(pin)>10*1000:\r
  27         tv2avi.ts2raw(pin, tpraw, opt)\r
  28         time.sleep(10)\r
  29         if os.path.isfile(tpraw) and os.path.getsize(tpraw)>10*1000:\r
  30             raw2mp4(tpraw, tpmp4, opt)\r
  31         time.sleep(10)\r
  32         if os.path.exists(tpraw):\r
  33             os.remove(tpraw)\r
  34 def raw2mp4(pin,pout,opt):\r
  35     dir=os.path.split(pout)[0]\r
  36     title=os.path.split(pout)[1]\r
  37     title=os.path.splitext(title)[0]\r
  38     duration="-fps 29.970030 "\r
  39     if re.search("a",opt):\r
  40         duration="-fps 23.976023 "\r
  41     elif re.search("v",opt):\r
  42         duration="-fps 23.976023 "\r
  43     if re.search("d",opt):\r
  44         duration="-fps 29.970030 "##ffmpegが24fpsに対応していないための措置\r
  45     elif re.search("5",opt):\r
  46         duration="-fps 29.970030 "##ffmpegが24fpsに対応していないための措置\r
  47     exe = configreader.getpath("mp4box")\r
  48     txt=""\r
  49     os.environ['LANG']="ja_JP.UTF-8"\r
  50     pints=pin.replace(".264",".ts")\r
  51     e1=exe +u" "+duration+" -add \""+pin+"\" -new \""+pout+"\""\r
  52     execmp4box(pin, pout, e1)\r
  53     addAudio(pints, pout, opt)\r
  54     addCaption(pints, pout)\r
  55     #recdblist.addlog(pout, txt, "mp4boxログ-コマンド")\r
  56     #recdblist.addlog(pout, txt, "mp4boxログ-詳細")\r
  57     if status.getSettings_auto_del_tmp()==1:\r
  58         if os.path.exists(pout):\r
  59             if re.search(opt,"MW8") or re.search(opt,"MW9"):\r
  60                 if os.path.getsize(pout)>50*1000*1000:\r
  61                     deltitle(dir,title)\r
  62             elif re.search(opt,"8") or re.search(opt,"9"):\r
  63                 ""\r
  64             else:\r
  65                 if os.path.getsize(pout)>100*1000*1000:\r
  66                     deltitle(dir,title)\r
  67 def mkv2mp4(pin,pout):\r
  68     exeb = configreader.getpath(u"mkvextract")\r
  69     exe = configreader.getpath(u"mp4Box")\r
  70     #dtsedit=configreader.getpath("DtsEdit")\r
  71     wineexe=configreader.getpath("wine")\r
  72     dir=os.path.split(pin)[0]\r
  73     title=os.path.split(pin)[1]\r
  74     title=os.path.splitext(title)[0]\r
  75     etitle=base64.b16encode(title.encode('utf-8'))\r
  76     audiopath=os.path.join(dir,etitle+u"_audio.aac")\r
  77     videopath=os.path.join(dir,etitle+u"_video.264")\r
  78     timecodepath=os.path.join(dir,etitle+u"_1_timecode.txt")\r
  79     tmpmp4=os.path.join(dir,etitle+u".tmp.mp4")\r
  80     exe0=exeb+u" tracks \'"+pin+u"\' 1:\'"+videopath+u"\' 2:\'"+audiopath+u"\'"\r
  81     exe1=exeb+u" timecodes_v2 \'"+pin+"\' 1:\'"+timecodepath+"\'"\r
  82     exe2=exe+u" -fps 29.970030 -add \'"+videopath+u"\' -add \'"+audiopath+u"\' -new \'"+tmpmp4+u"\'"\r
  83     exe3=wineexe+u" "+dtsedit+u" -tc \'Z:\\"+timecodepath+u"\' \'Z:\\"+tmpmp4+u"\' -o \'Z:\\"+pout+u"\'"\r
  84     os.environ['LANG']="ja_JP.UTF-8"\r
  85     txt=""\r
  86     try:\r
  87         txt=txt+execcomd(exe0)+"\n"\r
  88         txt=txt+execcomd(exe1)+"\n"\r
  89         txt=txt+execcomd(exe2)+"\n"\r
  90         txt=txt+execcomd(exe3)+"\n"\r
  91     except:\r
  92         ""\r
  93     recdblist.addlog(pin, txt, u"MKV2MP4-log")\r
  94     txt = "\n####MKV2MP4-log####\n"+txt\r
  95     time.sleep(10)\r
  96     if status.getSettings_auto_del_tmp()==1:\r
  97         if os.path.exists(pout):\r
  98             if os.path.getsize(pout)>100*1000*1000:\r
  99                 deltitle(dir,title)\r
 100 \r
 101 def addCaption(pts,pmp4):##字幕の追加を試みる。\r
 102     wineexe=configreader.getpath("wine")\r
 103     pincap=pts.replace(".ts",".srt")\r
 104     try:\r
 105         cap2ass=configreader.getpath("caption2ass")\r
 106     except:\r
 107         cap2ass=""\r
 108     if os.path.isfile(cap2ass):\r
 109         e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pts+"\" \"Z:\\"+pincap+"\""\r
 110         recdblist.printutf8(e0)\r
 111         p0=subprocess.Popen(e0,shell=True,stdout=subprocess.PIPE)\r
 112         time.sleep(300)\r
 113         if p0.poll==None:#実行中\r
 114             #if os.path.getsize(pincap)<1000:#2mで1kb以下の場合自動で終了\r
 115              #   os.kill(p0.pid,signal.SIGKILL)\r
 116             #else:\r
 117             os.waitpid(p0.pid, 0)\r
 118             logt=unicode(p0.communicate()[0], "UTF-8")\r
 119             recdblist.addlog(pts,e0, u"Captionログ-コマンド")\r
 120             recdblist.addlog(pts,logt, u"Captionログ-詳細")\r
 121         if os.path.getsize(pincap)>1000:\r
 122             exe = configreader.getpath("mp4box")\r
 123             e1s=exe +u" -add \""+pincap+"\" \""+pmp4+"\""\r
 124             addmp4(pincap,pmp4,e1s)\r
 125 def addAudio(pts,pmp4,opt):#オプションに応じた音声の追加を行う\r
 126     exe = configreader.getpath("mp4box")\r
 127     if re.search("d",opt) or re.search("5",opt):#二カ国語放送/5.1ch放送の場合\r
 128         paac1=pts.replace(".ts","_1.aac")\r
 129         paac2=pts.replace(".ts","_2.aac")\r
 130         recdblist.printutf8(paac1)\r
 131         if not os.path.exists(paac1):\r
 132             paac1=pts.replace(".ts","_1.mp3")\r
 133         if not os.path.exists(paac2):\r
 134             paac2=pts.replace(".ts","_2.mp3")\r
 135         e1a1=exe +u" -add \""+paac1+"\" \""+pmp4+"\""\r
 136         e1a2=exe +u" -add \""+paac2+"\" \""+pmp4+"\""\r
 137         addmp4(paac1, pmp4, e1a1)\r
 138         addmp4(paac2, pmp4, e1a2)\r
 139     else:\r
 140         tv2audio.ts2single_audio(pts)\r
 141         pinaac=pts.replace(".ts",".aac")\r
 142         if not os.path.exists(pinaac):\r
 143             pinaac=pinaac.replace(".aac",".mp3")\r
 144         e1a=exe +u" -add \""+pinaac+"\" \""+pmp4+"\""\r
 145         addmp4(pinaac, pmp4, e1a)\r
 146 def execmp4box(pin,pout,cmd):\r
 147     title=os.path.splitext(os.path.split(pin)[1])[0]\r
 148     nt=base64.b16encode(title.encode('utf-8'))\r
 149     if len(nt)>200:\r
 150         nt=nt[:180]\r
 151     ptin=os.path.join(os.path.dirname(pin),nt+".264")\r
 152     recdblist.printutf8(ptin)\r
 153     shutil.move(pin,ptin)\r
 154     time.sleep(10)\r
 155     ptout=os.path.join(os.path.dirname(pout),nt+".mp4")\r
 156     cmdn=string.replace(cmd,pin,ptin)\r
 157     cmdn=string.replace(cmdn,pout,ptout)\r
 158     recdblist.printutf8(cmdn)\r
 159     txt=""\r
 160     try:\r
 161         txt=execcomd(cmdn)\r
 162     except Exception, inst:\r
 163         txt= "error occures in execmp4box\n"\r
 164         txt=txt+ str(type(inst))+"\n"\r
 165         txt=txt+str(inst)\r
 166     recdblist.addlog(pin, cmdn, u"MP4Boxログ-コマンド")\r
 167     recdblist.addlog(pin, txt, u"MP4Boxログ-詳細")\r
 168     time.sleep(5)\r
 169     shutil.move(ptin,pin)\r
 170     shutil.move(ptout,pout)\r
 171     time.sleep(5)\r
 172 def addmp4(padd,pout,cmd):#without video\r
 173     title=os.path.splitext(os.path.split(padd)[1])[0]\r
 174     ext=os.path.splitext(os.path.split(padd)[1])[1]\r
 175     nt=base64.b16encode(title.encode('utf-8'))\r
 176     if len(nt)>200:\r
 177         nt=nt[:180]\r
 178     ptadd=os.path.join(os.path.dirname(padd),nt+ext)\r
 179     ptoutb=os.path.join(os.path.dirname(pout),nt+"_b.mp4")\r
 180     ptout=os.path.join(os.path.dirname(pout),nt+".mp4")\r
 181     shutil.move(padd,ptadd)\r
 182     if os.path.isfile(pout):\r
 183         shutil.move(pout,ptoutb)\r
 184     time.sleep(5)\r
 185     cmdn=string.replace(cmd,padd,ptadd)\r
 186     cmdn=string.replace(cmdn,u"-out \""+pout,u"-add \""+ptoutb+"\" -new \""+ptout)\r
 187     cmdn=string.replace(cmdn,u"\""+pout,u"-add \""+ptoutb+"\" -new \""+ptout)\r
 188     cmdn=string.replace(cmdn,pout,ptout)\r
 189     recdblist.printutf8(cmdn)\r
 190     txt=""\r
 191     try:\r
 192         txt=execcomd(cmdn)\r
 193     except Exception, inst:\r
 194         txt= "error occures in addmp4\n"\r
 195         txt=txt+ str(type(inst))+"\n"\r
 196         txt=txt+str(inst)\r
 197     recdblist.addlog(pout, cmdn, u"MP4Box追加ログ-コマンド")\r
 198     recdblist.addlog(pout, txt, u"MP4Box追加ログ-詳細")\r
 199     time.sleep(5)\r
 200     os.remove(ptoutb)\r
 201     shutil.move(ptadd,padd)\r
 202     shutil.move(ptout,pout)\r
 203     time.sleep(5)\r
 204 def execcomd(cmd):\r
 205     txt=""\r
 206     try:\r
 207         txt=u"Cmd : "+cmd+"\n"\r
 208         txt2=commands.getoutput(cmd.encode('utf-8'))\r
 209         txt=txt+unicode(txt2,"utf-8")+"\n"\r
 210     except:\r
 211         ""\r
 212     return txt\r
 213 \r
 214 def deltitle(path,title):\r
 215     dp=[]\r
 216     dp.append(os.path.join(path,title+".avi"))\r
 217     dp.append(os.path.join(path,title+".264"))\r
 218     dp.append(os.path.join(path,title+".120.avi"))\r
 219     dp.append(os.path.join(path,title+".noodml.avi"))\r
 220     dp.append(os.path.join(path,title+".aac"))\r
 221     dp.append(os.path.join(path,title+".m2v"))\r
 222     dp.append(os.path.join(path,title+"_1.aac"))\r
 223     dp.append(os.path.join(path,title+"_2.aac"))\r
 224     dp.append(os.path.join(path,title+"_1.mp3"))\r
 225     dp.append(os.path.join(path,title+"_2.mp3"))\r
 226     dp.append(os.path.join(path,title+".srt"))\r
 227     if configreader.getenv("remove_ts")=="1":\r
 228         dp.append(os.path.join(path,title+".ts"))\r
 229     if os.path.exists(os.path.join(path,title+".ts"))and os.path.exists(os.path.join(path,title+".ts.b25")):\r
 230         if os.path.getsize(os.path.join(path,title+".ts"))*10>os.path.getsize(os.path.join(path,title+".ts.b25")):\r
 231             if os.path.getsize(os.path.join(path,title+".ts"))>1000*1000*100:\r
 232                 dp.append(os.path.join(path,title+".ts.b25"))\r
 233     if os.path.exists(os.path.join(path,title+".mp4")):\r
 234         if os.path.getsize(os.path.join(path,title+".mp4"))>1000*1000*10:\r
 235             for ip in dp:\r
 236                 try:\r
 237                     os.remove(ip)\r
 238                 except:\r
 239                     ""\r