OSDN Git Service

change encoding.
[rec10/rec10-git.git] / rec10 / trunk / src / tv2mp4.py
index 56608cb..b9e705c 100644 (file)
-#!/usr/bin/python
-# coding: UTF-8
-# Rec10 TS Recording Tools
-# Copyright (C) 2009 Yukikaze
-import sys
-import commands
-import tv2avi
-import os
-import os.path
-import time
-import configreader
-def ts2mp4(pin, pout, opt):
-    dir=os.path.split(pout)[0]
-    title=os.path.split(pout)[1]
-    title=os.path.splitext(title)[0]
-    tpavi=os.path.join(dir, title+".avi")
-    tv2avi.ts2avi(pin, tpavi, opt)
-    time.sleep(10)
-    if not os.path.exists(tpavi):
-        if os.path.exists(pin):
-            tpavi=pin
-    avi2mp4(tpavi,pout)
-def avi2mp4(pin,pout):
-    exe = configreader.getpath(u"MP4Box")
-    dir=os.path.split(pin)[0]
-    title=os.path.split(pin)[1]
-    title=os.path.splitext(title)[0]
-    audiopath=os.path.join(dir,title+u"_audio.mp3")
-    videopath=os.path.join(dir,title+u"_video.h264")
-    #doexe = exe.encode('utf-8') + u' \"' + pin.encode('utf-8') + u'\" \"' + pout.encode('utf-8') + u'\"'
-    exe1=exe+u" -aviraw audio u'"+pin+u"\'"
-    exe2=exe+u" -aviraw video \'"+pin+u"\'"
-    exefin=exe+u" -new -fps 29.97 -add \'"+videopath+u"\'#video -add \'"+audiopath+u"\'#audio \'"+pout+u"\'"
-    logo=pin.replace(u".avi",u".log")
-    os.environ['LANG']="ja_JP.UTF-8"
-    txt= exe1+u"\n"+exe2+u"\n"+exefin+u"\n"
-    print txt
-    txt = txt+commands.getoutput(exe1.encode('utf-8'))
-    txt = txt+commands.getoutput(exe2.encode('utf-8'))
-    txt = txt+commands.getoutput(exefin.encode('utf-8'))
-    os.remove(audiopath)
-    os.remove(videopath)
-    txt = "\n####MP4Box-log####\n"+txt
-    f=open(logo,'a')
-    f.write(txt)
-    f.close()
-    sys.stdout.flush()
-    time.sleep(10)
-    if os.path.exists(pout):
-        if os.path.exists(audiopath):
-            os.remove(audiopath)
-        if os.path.exists(videopath):
-            os.remove(videopath)
-def mkv2mp4(pin,pout):
-    exeb = configreader.getpath("mkvextract")
-    exe = configreader.getpath("MP4Box")
-    dir=os.path.split(pin)[0]
-    title=os.path.split(pin)[1]
-    title=os.path.splitext(title)[0]
-    audiopath=os.path.join(dir,title+"_audio.mp3")
-    if not os.path.exists(audiopath):
-        if os.path.exists(os.path.join(dir,title+"_audio.raw")):
-            audiopath=os.path.join(dir,title+"_audio.raw")
-        if os.path.exists(os.path.join(dir,title+"_audio.aac")):
-            audiopath=os.path.join(dir,title+"_audio.aac")
-    videopath=os.path.join(dir,title+"_video.h264")
-    #doexe = exe.encode('utf-8') + u' \"' + pin.encode('utf-8') + u'\" \"' + pout.encode('utf-8') + u'\"'
-    exe0=exeb+" tracks \'"+pin+"\' 1:\'"+videopath+"\' 2:\'"+audiopath+"\'"
-    exe1=exe+" -aviraw audio \'"+pin+"\'"
-    exe2=exe+" -aviraw video \'"+pin+"\'"
-    exefin=exe+" -new -fps 29.97 -add \'"+videopath+"\'#video -add \'"+audiopath+"\'#audio \'"+pout+"\'"
-    logo=pin.replace(".avi",".log")
-    os.environ['LANG']="ja_JP.UTF-8"
-    txt= exe0+"\n"+exe1+"\n"+exe2+"\n"+exefin+"\n"
-    txt = txt+commands.getoutput(exe0.encode('utf-8'))
-    txt = txt+commands.getoutput(exe1.encode('utf-8'))
-    txt = txt+commands.getoutput(exe2.encode('utf-8'))
-    txt = txt+commands.getoutput(exefin.encode('utf-8'))
-    os.remove(audiopath)
-    os.remove(videopath)
-    txt = "\n####MKV2MP4-log####\n"+txt
-    f=open(logo,'a')
-    f.write(txt)
-    f.close()
-    sys.stdout.flush()
-    time.sleep(10)
-    if os.path.exists(pout):
-        if os.path.exists(audiopath):
-            os.remove(audiopath)
-        if os.path.exists(videopath):
-            os.remove(videopath)
+#!/usr/bin/python\r
+# coding: UTF-8\r
+# Rec10 TS Recording Tools\r
+# Copyright (C) 2009-2010 Yukikaze\r
+import commands\r
+import shutil\r
+import tv2avi\r
+import os\r
+import re\r
+import os.path\r
+import string\r
+import base64\r
+import time\r
+import recdblist\r
+import configreader\r
+\r
+def ts2mp4(pin, pout, opt):\r
+    dir=os.path.split(pout)[0]\r
+    title=os.path.split(pout)[1]\r
+    title=os.path.splitext(title)[0]\r
+    tpavi=os.path.join(dir, title+".avi")\r
+    tv2avi.ts2avi(pin, tpavi, opt)\r
+    time.sleep(10)\r
+    if not os.path.exists(tpavi):\r
+        if os.path.exists(pin):\r
+            tpavi=pin\r
+    avi2mp4(tpavi,pout,opt)\r
+def raw2mp4(pin,pout,opt):\r
+    dir=os.path.split(pout)[0]\r
+    title=os.path.split(pout)[1]\r
+    title=os.path.splitext(title)[0]\r
+    duration="-fps 29.970030 "\r
+    isdualaac=0\r
+    ispentaaudio=0\r
+    if re.search("a",opt):\r
+        duration="-fps 23.976023 "\r
+    elif re.search("v",opt):\r
+        duration="-fps 23.976023 "\r
+    if re.search("d",opt):\r
+        isdualaac=1\r
+        duration="-fps 29.970030 "##ffmpegが24fpsに対応していないための措置\r
+    elif re.search("5",opt):\r
+        ispentaaudio=1\r
+        duration="-fps 29.970030 "##ffmpegが24fpsに対応していないための措置\r
+    exe = configreader.getpath("mp4box")\r
+    txt=""\r
+    os.environ['LANG']="ja_JP.UTF-8"\r
+    if isdualaac==1:\r
+        wineexe=configreader.getpath("wine")\r
+        pincap=pin.replace(".264",".srt")\r
+        cap2ass=configreader.getpath("Caption2Ass")\r
+        pints=pin.replace(".264",".ts")\r
+        paac1=pin.replace(".264","_1.aac")\r
+        paac2=pin.replace(".264","_2.aac")\r
+        recdblist.printutf8(paac1)\r
+        if not os.path.exists(paac1):\r
+            paac1=pin.replace(".264","_1.mp3")\r
+        if not os.path.exists(paac2):\r
+            paac2=pin.replace(".264","_2.mp3")\r
+        e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pints+"\" \"Z:\\"+pincap+"\""\r
+        e1=exe +u" "+duration+" -add \""+pin+" -new \""+pout+"\""\r
+        e1a1=exe +u" -add \""+paac1+" \""+pout+"\""\r
+        e1a2=exe +u" -add \""+paac2+" \""+pout+"\""\r
+        e1s=exe +u" -add \""+pincap+" \""+pout+"\""\r
+        execmp4box(pin, pout, e1)\r
+        addmp4(paac1, pout, e1a1)\r
+        addmp4(paac2, pout, e1a2)\r
+        if os.path.isfile(cap2ass):\r
+            txt=txt+execcomd(e0)\r
+        if os.path.exists(pincap) and (os.path.getsize(pincap)>1000):\r
+            addmp4(pincap, pout, e1s)\r
+    elif ispentaaudio==1:\r
+        wineexe=configreader.getpath("wine")\r
+        pincap=pin.replace(".264",".srt")\r
+        cap2ass=configreader.getpath("Caption2Ass")\r
+        pints=pin.replace(".264",".ts")\r
+        paac1=pin.replace(".264","_1.aac")\r
+        paac2=pin.replace(".264","_2.aac")\r
+        recdblist.printutf8(paac1)\r
+        if not os.path.exists(paac1):\r
+            paac1=pin.replace(".264","_1.mp3")\r
+        if not os.path.exists(paac2):\r
+            paac2=pin.replace(".264","_2.mp3")\r
+        e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pints+"\" \"Z:\\"+pincap+"\""\r
+        e1=exe +u" "+duration+" -add \""+pin+" -new \""+pout+"\""\r
+        e1a1=exe +u" -add \""+paac1+" \""+pout+"\""\r
+        e1a2=exe +u" -add \""+paac2+" \""+pout+"\""\r
+        e1s=exe +u" -add \""+pincap+" \""+pout+"\""\r
+        execmp4box(pin, pout, e1)\r
+        addmp4(paac1, pout, e1a1)\r
+        addmp4(paac2, pout, e1a2)\r
+        if os.path.isfile(cap2ass):\r
+            txt=txt+execcomd(e0)\r
+        if os.path.exists(pincap) and (os.path.getsize(pincap)>1000):\r
+            addmp4(pincap, pout, e1s)\r
+    else:\r
+        wineexe=configreader.getpath("wine")\r
+        pincap=pin.replace(".264",".srt")\r
+        cap2ass=configreader.getpath("Caption2Ass")\r
+        pints=pin.replace(".264",".ts")\r
+        pinaac=pin.replace(".264",".aac")\r
+        e0=wineexe+u" "+cap2ass+u" -format srt \"Z:\\"+pints+"\" \"Z:\\"+pincap+"\""\r
+        e1="ffmpeg -i \""+pints+"\" -vn -f aac -acodec copy \""+pinaac+"\""\r
+        e2=exe +u" "+duration+" -add \""+pin+" -new \""+pout+"\""\r
+        e2a=exe +u" -add \""+pinaac+" \""+pout+"\""\r
+        e2s=exe +u" -add \""+pincap+" \""+pout+"\""\r
+        execmp4box(pin, pout, e2)\r
+        addmp4(pinaac, pout, e2a)\r
+        addmp4(pincap, pout, e2s)\r
+        if os.path.isfile(cap2ass):\r
+            txt=txt+execcomd(e0)\r
+        if os.path.exists(pincap) and (os.path.getsize(pincap)>1000):\r
+            addmp4(pincap, pout, e2s)\r
+    recdblist.addlog(pout, txt, "mp4box-log")\r
+    if status.getSettings_auto_del_tmp()==1:\r
+        if os.path.exists(pout):\r
+            if os.path.getsize(pout)>100*1000*1000:\r
+                deltitle(dir,title)\r
+def avi2mp4(pin,pout,opt):\r
+    isvfr=0\r
+    if re.search(u"v", opt):\r
+        isvfr=1\r
+    exe = configreader.getpath(u"MP4Box")\r
+    dir=os.path.split(pin)[0]\r
+    title=os.path.split(pin)[1]\r
+    title=os.path.splitext(title)[0]\r
+    etitle=base64.b16encode(title.encode('utf-8'))\r
+    audiopath=os.path.join(dir,etitle+u"_audio.raw")\r
+    videopath=os.path.join(dir,etitle+u"_video.264")\r
+    exes=[]\r
+    os.environ['LANG']="ja_JP.UTF-8"\r
+    txt=""\r
+    if isvfr==1:\r
+        wineexe=configreader.getpath("wine")\r
+        cfr2tc=configreader.getpath("cfr2tc")\r
+        dtsedit=configreader.getpath("DtsEdit")\r
+        caption2ass=configreader.getpath("Caption2Ass")\r
+        pin120=os.path.join(dir,etitle+u".120.avi")\r
+        pints=os.path.join(dir,title+u".ts")\r
+        pmp4=os.path.join(dir,etitle+u".tmp.mp4")\r
+        pintimecode=os.path.join(dir,etitle+u".timecode.txt")\r
+        pinaac=os.path.join(dir,etitle+u".aac")\r
+        pincat=os.path.join(dir,etitle+u".srt")\r
+        evfr0="ffmpeg -i '"+pints+"' -vn -f aac -acodec copy '"+pinaac+"'"\r
+        evfr1=wineexe+u" "+cfr2tc+u" '"+"Z:\\"+pin+"' '"+"Z:\\"+pin120+u"' 'Z:\\"+pintimecode+u"' 2"\r
+        evfr2=exe+u" -aviraw video \'"+pin120+u"\'"\r
+        videotpath=os.path.join(dir,etitle+u".120_video.h264")\r
+        evfr3=exe+u" -fps 29.970030 -add \'"+videotpath+"\' -add \'"+pinaac+"\' -new \'"+pmp4+"\'"\r
+        evfr4=wineexe+u" "+dtsedit+u" -tc \'Z:\\"+pintimecode+u"\' \'Z:\\"+pmp4+"\' -o \'Z:\\"+pout+"\'"\r
+        try:\r
+            recdblist.printutf8(evfr0)\r
+            txt=txt+evfr0+"\n"\r
+            txt=txt+unicode(commands.getoutput(evfr0.encode('utf-8')),'utf-8')\r
+        except:\r
+            ""\r
+        try:\r
+            recdblist.printutf8(evfr1)\r
+            txt=txt+evfr1+"\n"\r
+            txt=txt+unicode(commands.getoutput(evfr1.encode('utf-8')),'utf-8')\r
+        except:\r
+            ""\r
+        try:\r
+            if not os.path.exists(pin120):\r
+                evfr2=exe+u" -aviraw video \'"+pin+u"\'"\r
+                videotpath=os.path.join(dir,etitle+u"_video.h264")\r
+                evfr3=exe+u" -fps 29.970030 -add \'"+videotpath+"\' -add \'"+pinaac+"\' -new \'"+pmp4+"\'"\r
+            recdblist.printutf8(evfr2)\r
+            txt=txt+evfr2+"\n"\r
+            txt=txt+unicode(commands.getoutput(evfr2.encode('utf-8')),'utf-8')\r
+        except:\r
+            ""\r
+        try:\r
+            recdblist.printutf8(evfr3)\r
+            txt=txt+evfr3+"\n"\r
+            txt=txt+unicode(commands.getoutput(evfr3.encode('utf-8')),'utf-8')\r
+        except:\r
+            ""\r
+        try:\r
+            recdblist.printutf8(evfr4)\r
+            txt=txt+evfr4+"\n"\r
+            txt=txt+unicode(commands.getoutput(evfr4.encode('utf-8')),'utf-8')\r
+        except:\r
+            ""\r
+        if (not os.path.exists(pin.replace(".avi",".120.avi"))and (not os.path.exists(pout))):\r
+            txt=txt+"CFR2TC error. maybe avi file is over 1GB.(ODML error would have happened.)"\r
+            ptmp=pin.replace(".avi",".noodml.avi")\r
+            e="mencoder -noodml -forceidx -oac copy -ovc copy -o \'"+ptmp+"\'"+" \'"+pin+"\'"\r
+            ex="nice -n 17 "+e\r
+            e2=wineexe+u" "+cfr2tc+u" '"+"Z:\\"+ptmp+"' '"+"Z:\\"+pin120+u"' '"+"Z:\\"+pintimecode+u"' 2"\r
+            ex2="nice -n 17 "+e2\r
+            e3=exe+u" -aviraw video \'"+pin120+u"\'"\r
+            ex3="nice -n 17 "+e3\r
+            e4=exe+u" -fps 29.97 -add \'"+videotpath+"\' -add \'"+pinaac+"\' -new \'"+pmp4+"\'"\r
+            ex4="nice -n 17 "+e4\r
+            e5=wineexe+u" "+dtsedit+u" -tc \'Z:\\"+pintimecode+u"\' \'Z:\\"+pmp4+"\' -o \'Z:\\"+pout+"\'"\r
+            ex5="nice -n 17 "+e5\r
+            recdblist.printutf8(e)\r
+            txt=txt+e+"\n"\r
+            try:\r
+                cmdt=unicode(commands.getoutput(ex.encode('utf-8')),'utf-8')\r
+            except:\r
+                ""\r
+            try:\r
+                cmdt=unicode(commands.getoutput(ex2.encode('utf-8')),'utf-8')\r
+            except:\r
+                ""\r
+            try:\r
+                if os.path.exists(pin120):\r
+                    cmdt=unicode(commands.getoutput(ex3.encode('utf-8')),'utf-8')\r
+            except:\r
+                ""\r
+            try:\r
+                cmdt=unicode(commands.getoutput(ex4.encode('utf-8')),'utf-8')\r
+            except:\r
+                ""\r
+            try:\r
+                cmdt=unicode(commands.getoutput(ex5.encode('utf-8')),'utf-8')\r
+            except:\r
+                ""\r
+            try:\r
+                os.remove(ptmp)\r
+            except:\r
+                ""\r
+            txt=txt+cmdt+"\n"\r
+    else:\r
+        try:\r
+            shutil.move(pin,os.path.join(dir,etitle+".avi"))\r
+        except:\r
+            ""\r
+        if os.path.exists(os.path.join(dir,etitle+".avi")):\r
+            e0=exe+u" -aviraw audio \'"+os.path.join(dir,etitle+".avi")+u"\'"\r
+            e1=exe+u" -aviraw video \'"+os.path.join(dir,etitle+".avi")+u"\'"\r
+            os.system(e0)\r
+            os.system(e1)\r
+        if not os.path.exists(audiopath):\r
+            if os.path.exists(os.path.join(dir,etitle+u"_audio.mp3")):\r
+                audiopath=os.path.join(dir,etitle+u"_audio.mp3")\r
+            elif os.path.exists(os.path.join(dir,etitle+u"_audio.aac")):\r
+                audiopath=os.path.join(dir,etitle+u"_audio.aac")\r
+        else:\r
+            shutil.move(audiopath,os.path.join(dir,etitle+u"_audio.aac"))\r
+            audiopath=os.path.join(dir,etitle+u"_audio.aac")\r
+        ptmp=os.path.exists(os.path.join(dir,etitle+u".mp4"))\r
+        e2=exe+u" -fps 29.970030 -add \'"+videopath+u"\' -add \'"+audiopath+u"\' -new \'"+ptmp+u"\'"\r
+        os.system(e2)\r
+        shutil.move(ptmp,pout)\r
+    recdblist.addlog(pin,txt,"MP4Box-log")\r
+    time.sleep(10)\r
+    delpath=[]\r
+    delpath.append(pin.replace(u".avi",u".120.avi"))\r
+    delpath.append(pin.replace(u".avi",u".sa.avi"))\r
+    delpath.append(pin.replace(u".avi",u".m2v"))\r
+    delpath.append(pin.replace(u".avi",u"_1.aac"))\r
+    delpath.append(pin.replace(u".avi",u"_2.aac"))\r
+    delpath.append(pin.replace(u".avi",u"_1.mp3"))\r
+    delpath.append(pin.replace(u".avi",u"_2.mp3"))\r
+    delpath.append(pin.replace(u".avi",u".ts"))\r
+    delpath.append(pin.replace(u".avi",u".aac"))\r
+    delpath.append(pin.replace(u".avi",u".m2v"))\r
+    delpath.append(pin.replace(u".avi",u".120_video.264"))\r
+    delpath.append(pin.replace(u".avi",u".ts.b25"))\r
+    if os.path.exists(pout):\r
+        if os.path.exists(audiopath):\r
+            os.remove(audiopath)\r
+        if os.path.exists(videopath):\r
+            os.remove(videopath)\r
+        """\r
+        for dp in delpath:\r
+            try:\r
+                os.remove(dp)\r
+            except:\r
+                ""\r
+        """\r
+def mkv2mp4(pin,pout):\r
+    exeb = configreader.getpath(u"mkvextract")\r
+    exe = configreader.getpath(u"MP4Box")\r
+    dtsedit=configreader.getpath("DtsEdit")\r
+    wineexe=configreader.getpath("wine")\r
+    dir=os.path.split(pin)[0]\r
+    title=os.path.split(pin)[1]\r
+    title=os.path.splitext(title)[0]\r
+    etitle=base64.b16encode(title.encode('utf-8'))\r
+    audiopath=os.path.join(dir,etitle+u"_audio.aac")\r
+    videopath=os.path.join(dir,etitle+u"_video.264")\r
+    timecodepath=os.path.join(dir,etitle+u"_1_timecode.txt")\r
+    tmpmp4=os.path.join(dir,etitle+u".tmp.mp4")\r
+    exe0=exeb+u" tracks \'"+pin+u"\' 1:\'"+videopath+u"\' 2:\'"+audiopath+u"\'"\r
+    exe1=exeb+u" timecodes_v2 \'"+pin+"\' 1:\'"+timecodepath+"\'"\r
+    exe2=exe+u" -fps 29.970030 -add \'"+videopath+u"\' -add \'"+audiopath+u"\' -new \'"+tmpmp4+u"\'"\r
+    exe3=wineexe+u" "+dtsedit+u" -tc \'Z:\\"+timecodepath+u"\' \'Z:\\"+tmpmp4+u"\' -o \'Z:\\"+pout+u"\'"\r
+    os.environ['LANG']="ja_JP.UTF-8"\r
+    txt=""\r
+    try:\r
+        txt=txt+execcomd(exe0)+"\n"\r
+        txt=txt+execcomd(exe1)+"\n"\r
+        txt=txt+execcomd(exe2)+"\n"\r
+        txt=txt+execcomd(exe3)+"\n"\r
+    except:\r
+        ""\r
+    recdblist.addlog(pin, txt, "MKV2MP4-log")\r
+    txt = "\n####MKV2MP4-log####\n"+txt\r
+    time.sleep(10)\r
+    if os.path.exists(pout):\r
+        if os.path.exists(audiopath):\r
+            os.remove(audiopath)\r
+        if os.path.exists(videopath):\r
+            os.remove(videopath)\r
+        if os.path.exists(timecodepath):\r
+            os.remove(timecodepath)\r
+        if os.path.exists(tmpmp4):\r
+            os.remove(tmpmp4)\r
+def execmp4box(pin,pout,cmd):\r
+    ptin=os.path.join(os.path.dirname(pin),base64.b32encode(os.path.splitext(os.path.split(pin)[1])[0])+".264")\r
+    shutil.move(pin,ptin)\r
+    ptout=os.path.join(os.path.dirname(pout),base64.b32encode(os.path.splitext(os.path.split(pout)[1])[0])+".mp4")\r
+    cmdn=string.replace(cmd,pin,ptin)\r
+    cmdn=string.replace(cmdn,pout,ptout)\r
+    txt=""\r
+    try:\r
+        txt=execcomd(cmdn)\r
+    except Exception, inst:\r
+        txt= "error occures in execmp4box\n"\r
+        txt=txt+ str(type(inst))+"\n"\r
+        txt=txt+str(inst)\r
+    recdblist.addlog(pin, txt, "MP4Box-log")\r
+    time.sleep(3)\r
+    shutil.move(ptout,pout)\r
+def addmp4(padd,pout,cmd):#without video\r
+    ptadd=os.path.join(os.path.dirname(padd),base64.b32encode(os.path.splitext(os.path.split(padd)[1])[0])+"."+os.path.splitext(os.path.split(padd)[1])[1])\r
+    shutil.move(padd,ptadd)\r
+    ptout=os.path.join(os.path.dirname(pout),base64.b32encode(os.path.splitext(os.path.split(pout)[1])[0])+".mp4")\r
+    cmdn=string.replace(cmd,padd,ptadd)\r
+    cmdn=string.replace(cmdn,pout,ptout)\r
+    txt=""\r
+    try:\r
+        txt=execcomd(cmdn)\r
+    except Exception, inst:\r
+        txt= "error occures in addmp4\n"\r
+        txt=txt+ str(type(inst))+"\n"\r
+        txt=txt+str(inst)\r
+    recdblist.addlog(pin, txt, "MP4Box-log-add")\r
+    time.sleep(3)\r
+    shutil.move(ptout,pout)\r
+def execcomd(cmd):\r
+    txt=""\r
+    try:\r
+        txt=u"Cmd : "+cmd+"\n"\r
+        txt2=commands.getoutput(cmd.encode('utf-8'))\r
+        txt=txt+unicode(txt2,"utf-8")+"\n"\r
+    except:\r
+        ""\r
+    return txt\r