OSDN Git Service

fix some miss.
[rec10/rec10-git.git] / rec10 / trunk / src / ts2x264.py
index d72e6b5..08a64a1 100644 (file)
 # Rec10 TS Recording Tools
 # Copyright (C) 2009-2010 Yukikaze
 import commands
+import configreader
 import os
+import os.path
 import re
-import time
-import tv2avi
+import random
+
 import recdblist
 def ts2x264(pin, pout, opts):#sizeは"HD"か"SD"
     """
     pinで指定されたファイルをpoutにx264でエンコードして書き出す
     """
     dualpass = 0
-    isvfr=0
     is24fps=0
-    islight=0
-    size="Tall"
-    quality=5
-    if re.search("2", opts):
-        dualpass = 1
+    size="HD"
+    crf=18
+    quality=4
+    quality=int(configreader.getenv("x264_preset"))
+    crf=int(configreader.getenv("crf"))
     if re.search("H", opts):
         size = "HD"
-        quality=7
     if re.search("S", opts):
         size = "SD"
-        quality=3
     if re.search("F", opts):
-        size = "FHD"
-        quality=5
-    if re.search("L", opts):
-        size = "Short"
-        quality=3
-    if re.search("G", opts):
-        size = "LowHD"
-        quality=5
+        size = "FullHD"
+    if re.search("W",opts):
+        size = "WVGA"
+    if re.search("MW8", opts):
+        size = "QVGA_BASE"
+        crf=crf+4
+    if re.search("MW9", opts):
+        size = "WVGA_BASE"
+        crf=crf+2
     if re.search("v", opts):
         is24fps=1
+        crf=int(configreader.getenv("animation_crf"))
     if re.search("a", opts):
         is24fps=1
-        quality=quality+2
-    if re.search("l",opts):
+        crf=int(configreader.getenv("animation_crf"))
+    if re.search("q",opts):
         quality=quality-2
-    if re.search("h",opts):
+    if re.search("w",opts):
+        quality=quality-1
+    if re.search("e",opts):
+        quality=quality+1
+    if re.search("r",opts):
         quality=quality+2
+    if re.search("u",opts):
+        crf=crf+2
+    if re.search("i",opts):
+        crf=crf+1
+    if re.search("o",opts):
+        crf=crf-1
+    if re.search("p",opts):
+        crf=crf-2
     if re.search("d",opts):
-        encode_ffmpeg(pin, pout, size, dualpass ,is24fps , quality)
+        encode_ffmpeg(pin,pout,size,is24fps,quality,crf)
+    elif re.search("5",opts):
+        encode_ffmpeg(pin,pout,size,is24fps,quality,crf)
     else:
-        encode(pin, pout,size,dualpass,is24fps,quality)
-def encode(pin,pout,size,is2pass,is24fps,quality):
+        try:
+            encode(pin, pout,size,is24fps,quality,crf)
+        except Exception, inst:
+            recdblist.printutf8("error occures in ts2x264.py encode")
+            recdblist.printutf8(str(type(inst)))
+            recdblist.printutf8(str(inst))
+def encode(pin,pout,size,is24fps,quality,crf):
+    mencoder=configreader.getpath("mencoder")
+    x264=configreader.getpath("x264")
     encvf=""
     txt=""
     encvf="-vf yadif=0,pp=l5"
     harddup=",hqdn3d=2:1:2,unsharp=l3x3:0.75:c3x3:0.75,harddup"
     ofps="-ofps 30000/1001"
     fps="-fps 30000/1001"
-    bm=1
-    scale=""
-    bnum=2000
-    if size == "HD":
-        encvf = encvf + ",scale=1280:720::0:3"+harddup
-        bitrate = "bitrate=" + str(tv2avi.Bitrate_HD*bm)
-        bnum=tv2avi.Bitrate_HD*bm
-        crf="crf=20"
-    elif size == "SD":
-        encvf = encvf + ",scale=720:480::0:3"+harddup
-        bitrate = "bitrate=" + str(tv2avi.Bitrate_SD*bm)
-        bnum=tv2avi.Bitrate_SD*bm
-        crf="crf=20"
-    elif size == "FHD":
-        encvf = encvf + ",scale=1920:1080::0:3"+harddup
-        bitrate = "bitrate=" + str(tv2avi.Bitrate_FHD*bm)
-        bnum=tv2avi.Bitrate_FHD*bm
-        crf="crf=22"
-    elif size == "Short":
-        encvf = encvf +harddup
-        bitrate = "bitrate=" + str(tv2avi.Bitrate_Short*bm)
-        bnum=tv2avi.Bitrate_Short*bm
-        crf="crf=20"
-    elif size == "LowHD":
-        encvf = encvf + ",scale=1280:720::0:3"+harddup
-        bitrate = "bitrate=" + str(tv2avi.Bitrate_LowHD*bm)
-        bnum=tv2avi.Bitrate_LowHD*bm
-        crf="crf=22"
-    else:
-        encvf = encvf + ",scale=1280:720::0:3"+harddup
-        bitrate = "bitrate=" + str(tv2avi.Bitrate_HD*bm)
-        bnum=tv2avi.Bitrate_HD*bm
-        crf="crf=22"
-    x264enc=""
-    x264enc1=""
-    x264enc2=""
-
-    if quality<3:
-        x264enc=":me=dia:partitions=none:frameref=1:subme=0:scenecut=0:trellis=0"
-        x264enc1=x264enc
-        x264enc2=x264enc1
-    elif quality<5:
-        x264enc=":me=dia:partitions=i8x8,i4x4:frameref=1:subme=1:trellis=0"
-        x264enc1=x264enc
-        x264enc2=x264enc1
-    elif quality<7:
-        x264enc=""
-        x264enc1=x264enc
-        x264enc2=x264enc1
-    elif quality<9:
-        x264enc=":direct_pred=auto:frameref=5:subme=8:bframes=1:trellis=1"
-        x264enc1=x264enc
-        x264enc2=x264enc1
-    elif quality<11:
-        x264enc=":frameref=6:subme=8:subq=7:trellis=2:partitions=all:8x8dct:direct_pred=auto:me=umh:bframes=3"
-        x264enc1=x264enc
-        x264enc2=x264enc1
-    crf=""
+    x264fps="--fps 30000/1001"
+    x264streamsize=""
+    x264preset=""
+    x264tune=""
+    x264_bitrate="5000"
+    x264_addline=configreader.getenv("x264_addline")
     if is24fps==1:
         ofps="-ofps 24000/1001"
         fps="-fps 30000/1001"
+        x264fps="--fps 24000/1001"
+        x264tune="--tune animation"
         encvf="-vf pullup,softskip"
         harddup=",pp=l5,unsharp=l3x3:0.75:c3x3:0.75,hqdn3d=2:1:2,harddup"
-        bm=1
-        if quality<3:
-            vbv=":vbv_maxrate="+str(bnum*2)
-            x264enc=":me=dia:aq-strength=0.6:deblock=1,1:psy-rd=0.4:partitions=none:frameref=1:subme=0:scenecut=0:trellis=0"
-            x264enc1=x264enc+vbv
-            x264enc2=x264enc1
-        elif quality<5:
-            vbv=":vbv_maxrate="+str(bnum*2)
-            x264enc=":me=dia:aq-strength=0.6:deblock=1,1:psy-rd=0.4:partitions=i8x8,i4x4:frameref=1:subme=1:trellis=0"
-            x264enc1=x264enc+vbv
-            x264enc2=x264enc1
-        elif quality<7:
-            vbv=":vbv_maxrate="+str(bnum*2)
-            x264enc=":aq-strength=0.6:deblock=1,1:psy-rd=0.4"
-            x264enc1=x264enc+vbv
-            x264enc2=x264enc1
-        elif quality<9:
-            vbv=":vbv_maxrate="+str(bnum*2)
-            x264enc=":direct_pred=auto:aq-strength=0.6:deblock=1,1:psy-rd=0.4:frameref=5:subme=8:bframes=1:trellis=1"
-            x264enc1=x264enc+vbv
-            x264enc2=x264enc1
-        elif quality<11:
-            vbv=":vbv_maxrate="+str(bnum*2)
-            x264enc=":frameref=6:subme=8:subq=7:trellis=2:aq-strength=0.6:deblock=1,1:psy-rd=0.4:partitions=all:8x8dct:direct_pred=auto:me=umh:bframes=3"
-            x264enc1=x264enc+vbv
-            x264enc2=x264enc1
-    if is2pass == 1:
-        pas1exe = "mencoder \'" + pin + "\' -vfm ffmpeg -quiet -nosound -sws 9 -ovc x264 " + encvf +" "+fps+" "+ofps+" -x264encopts " + bitrate +x264enc1+ ":threads=auto:level_idc=42:pass=1:turbo=2 -passlogfile \'" + pin + ".log\' " + " -o /dev/null"
-        pas2exe = "mencoder \'" + pin + "\' -vfm ffmpeg -quiet -sws 9 -ovc x264 " + encvf +" "+fps+ " "+ofps+" -x264encopts " + bitrate +x264enc2+ ":level_idc=42:threads=auto:pass=2 -passlogfile \'" + pin + ".log\' -oac mp3lame -lameopts cbr:br=256 -o \'" + pout + "\'"
-        pas1exe = "nice -n 19 " + pas1exe
-        pas2exe = "nice -n 19 " + pas2exe
-        recdblist.printutf8(pas1exe)
-        txt=txt+unicode(commands.getoutput(pas1exe.encode('utf-8')),'utf-8')
-        recdblist.printutf8(pas2exe)
-        time.sleep(5)
-        txt=txt+unicode(commands.getoutput(pas2exe.encode('utf-8')),'utf-8')
+    if size == "HD":
+        encvf = encvf + ",scale=-2:720::0:3,expand=1280:720"+harddup
+        x264streamsize=u"1280x720"
+        x264_bitrate="2500"
+    elif size == "WVGA":
+        encvf = encvf + ",scale=-2:480::0:3,expand=854:480"+harddup
+        x264streamsize=u"854x480"
+        x264_bitrate="1500"
+    elif size == "FullHD":
+        encvf = encvf + ",scale=-2:1080::0:3,expand=1920:1080"+harddup
+        x264streamsize=u"1920x1080"
+        x264_bitrate="5000"
+    elif size == "SD":
+        encvf = encvf + ",scale=720:-2::0:3,expand=720:480"+harddup
+        x264streamsize=u"720x480"
+        x264_bitrate="1250"
+    elif size == "QVGA_BASE":
+        encvf = encvf + ",scale=320:-2::0:3,expand=320:240"+harddup
+        x264streamsize=u"320x240"
+        x264_bitrate="300"
+    elif size == "WVGA_BASE":
+        encvf = encvf + ",scale=-2:480::0:3,expand=854:480"+harddup
+        x264streamsize=u"854x480"
+        x264_bitrate="1500"
     else:
-        pas1exe = "mencoder \'" + pin + "\' -vfm ffmpeg -quiet -sws 9 -ovc x264 " + encvf+" "+fps + " "+ofps+" -x264encopts " + bitrate +x264enc+":level_idc=42:threads=auto -oac mp3lame -lameopts cbr:br=256 -o \'" + pout + "\'"
-        pas1exe = "nice -n 19 " + pas1exe
-        recdblist.printutf8(pas1exe)
-        txt=txt+unicode(commands.getoutput(pas1exe.encode('utf-8')),'utf-8')
-
+        encvf = encvf + ",scale=-2:720::0:3,expand=1280:720"+harddup
+        x264streamsize=u"1280x720"
+        x264_bitrate="2500"
+    if quality==1:
+        x264preset=u"--preset ultrafast"
+    elif quality==2:
+        x264preset=u"--preset veryfast"
+    elif quality==3:
+        x264preset=u"--preset fast"
+    elif quality==4:
+        x264preset=u"--preset medium"
+    elif quality==5:
+        x264preset=u"--preset slow"
+    elif quality==6:
+        x264preset=u"--preset slower"
+    if size == "WVGA_BASE" or size == "QVGA_BASE":
+        x264profile=" --level 32 --profile baseline "
+    else:
+        x264profile=" --level 41 --profile high "
+    x264crf=u"--crf "+str(crf)
+    os.environ['LANG']="ja_JP.UTF-8"
+    random.seed(pin)
+    random.jumpahead(10)
+    streampath=os.path.join(os.path.dirname(pin),str(random.randint(10000, 999999)))
+    os.system(u"mkfifo "+streampath)
+    encexe=mencoder+u" \""+pin+u"\" -vfm ffmpeg -quiet -sws 9 "+encvf+u",format=i420 "+fps+" "+ofps+" -oac mp3lame -ovc raw -of rawvideo -o \""+streampath+"\" &"
+    encexe=encexe+u" nice -n 19 "+x264+" "+x264crf+u" "+x264_addline+u"  --threads auto"+x264profile+x264preset+" "+x264tune+" "+x264fps+" -o \""+pout+"\" "+streampath+" "+x264streamsize
+    encexe=u"nice -n 19 " +encexe
+    recdblist.printutf8(encexe)
+    txt=txt+"Cmd : "+encexe+"\n"
+    txtt=""
+    try:
+        txtt=unicode(commands.getoutput(encexe.encode('utf-8')),'utf-8')
+    except:
+        ""
+    os.system("rm "+streampath)
+    txt=txt+txtt
     recdblist.addlog( pin, txt, "mencoder-log")
 
-def encode_ffmpeg(pin,pout,size,is2pass,is24fps,quality):
+def encode_ffmpeg(pin,pout,size,is24fps,quality,crf):
     """
     
     """
-    ffmpeg="ffmpeg"
-    vf1="-me dia -subme 4 "
-    vf1="-vpre hq -level 42 -refs 3 -subme 8 -bf 3 "
-    if quality<3:
-        vf1="-me dia -refs 1 -subme 0 -trellis 0 -partitions none "
-        vf2="-me dia -refs 1 -subme 0 -trellis 0 -partitions none "
-    elif quality<5:
-        vf1="-me dia -refs 1 -subme 1 -trellis 0 "
-        vf2="-vpre hq -level 42 -me dia -refs 1 -subme 1 -trellis 0 "
-    elif quality<7:
-        vf1=""
-        vf2="-vpre hq -level 42 "
-    elif quality<9:
-        vf1="-directpred auto -refs 5 -subme 8 -bf 1 -trellis 1 "
-        vf2="-vpre hq -level 42 -directpred auto -refs 5 -subme 8 -bf 1 -trellis 1 "
-    elif quality<11:
-        vf1="-me hex -bf 3 "
-        vf2="-vpre hq -level 42 -directpred auto -refs 6 -subme 8 -trellis 2 -me umh -bf 3 "
-    if is24fps==1:
-        if quality<3:
-            vf1="-me dia -refs 1 -subme 0 -trellis 0 -partitions none -deblockalpha 1 -deblockbeta 1 "
-            vf2="-me dia -refs 1 -subme 0 -trellis 0 -partitions none -deblockalpha 1 -deblockbeta 1 "
-        elif quality<5:
-            vf1="-me dia -refs 1 -subme 1 -trellis 0 -deblockalpha 1 -deblockbeta 1 "
-            vf2="-vpre hq -level 42 -me dia -refs 1 -subme 1 -trellis 0 -deblockalpha 1 -deblockbeta 1 "
-        elif quality<7:
-            vf1="-deblockalpha 1 -deblockbeta 1 "
-            vf2="-vpre hq -level 42 -deblockalpha 1 -deblockbeta 1 "
-        elif quality<9:
-            vf1="-directpred auto -refs 5 -subme 8 -bf 1 -trellis 1 -deblockalpha 1 -deblockbeta 1 "
-            vf2="-vpre hq -level 42 -directpred auto -refs 5 -subme 8 -bf 1 -trellis 1 -deblockalpha 1 -deblockbeta 1 "
-        elif quality<11:
-            vf1="-directpred auto -me hex -bf 3 -deblockalpha 1 -deblockbeta 1 "
-            vf2="-vpre hq -level 42 -directpred auto -refs 6 -subme 8 -subq 7 -trellis 2 -me umh -bf 3 -deblockalpha 1 -deblockbeta 1 "
+    ffmpeg=configreader.getpath("ffmpeg")
+    x264=configreader.getpath("x264")
+    fps=u"-r 29.970030 "
+    x264fps="--fps 30000/1001"
+    x264streamsize=""
+    x264preset=""
+    x264tune=""
+    x264_bitrate="2500"
+    x264_addline=configreader.getenv("x264_addline")
     if size == "HD":
         s = "-s 1280x720 "
-        bitrate = str(tv2avi.Bitrate_HD)
-        crf="crf=20"
+        x264streamsize=u"1280x720"
+        x264_bitrate="2500"
+    elif size == "WVGA":
+        s = "-s 854x480 "
+        x264streamsize=u"854x480"
+        x264_bitrate="1500"
+    elif size == "FullHD":
+        s = "-s 1920x1080 "
+        x264streamsize=u"1920x1080"
+        x264_bitrate="5000"
     elif size == "SD":
-        s ="-s 720x480 "
-        bitrate = str(tv2avi.Bitrate_SD)
-        crf="crf=20"
-    elif size == "FHD":
-        s ="-s 1920x1080 "
-        bitrate = str(tv2avi.Bitrate_FHD)
-        crf="crf=22"
-    elif size == "Short":
-        s= ""
-        bitrate = str(tv2avi.Bitrate_Short)
-        crf="crf=20"
-    elif size == "LowHD":
-        s = "-s 1280x720 "
-        bitrate = str(tv2avi.Bitrate_LowHD)
-        crf="crf=22"
+        s = "-s 720x480 "
+        x264streamsize=u"720x480"
+        x264_bitrate="1250"
+    elif size == "QVGA_BASE":
+        s = "-s 320x240 "
+        x264streamsize=u"320x240"
+        x264_bitrate="300"
+    elif size == "WVGA_BASE":
+        s = "-s 854x480 "
+        x264streamsize=u"854x480"
+        x264_bitrate="1500"
     else:
         s = "-s 1280x720 "
-        bitrate = str(tv2avi.Bitrate_LowHD)
-        crf="crf=22"
-    txt=""
-    if is2pass==1:
-        os.environ['LANG']="ja_JP.UTF-8"
-        exe1=ffmpeg+" -y -i \""+pin+"\" -pass 1 -vsync 400 -vcodec libx264 -b "+bitrate+"k "+vf1+s+"-deinterlace -an -passlogfile \'" + pin + ".log\' " + " /dev/null"
-        exe2=ffmpeg+" -y -i \""+pin+"\" -pass 2 -vsync 400 -vcodec libx264 -b "+bitrate+"k "+vf2+s+"-deinterlace -an -passlogfile \'" + pin + ".log\' \"" +pout+"\"" #aviにすべきかどうか
-        txt=txt+exe1+"\n"+exe2+"\m"
-        recdblist.printutf8(exe1)
-        exe1 = "nice -n 19 " + exe1
-        exe2 = "nice -n 19 " + exe2
-        os.system(exe1.encode('utf-8'))
-        os.system(exe2.encode('utf-8'))
+        x264streamsize=u"1280x720"
+        x264_bitrate="2500"
+    if quality==1:
+        x264preset=u"--preset ultrafast"
+    elif quality==2:
+        x264preset=u"--preset veryfast"
+    elif quality==3:
+        x264preset=u"--preset fast"
+    elif quality==4:
+        x264preset=u"--preset medium"
+    elif quality==5:
+        x264preset=u"--preset slow"
+    elif quality==6:
+        x264preset=u"--preset slower"
+    if size == "WVGA_BASE" or size == "QVGA_BASE":
+        x264profile=" --level 32 --profile baseline "
     else:
-        os.environ['LANG']="ja_JP.UTF-8"
-        exe=ffmpeg+" -y -i \""+pin+"\" -vsync 400 -vcodec libx264 -b "+bitrate+"k "+vf2+s+"-deinterlace -an \"" +pout+"\""
-        txt=txt+exe+"\n"
-        exe = "nice -n 19 " + exe
-        os.system(exe.encode('utf-8'))
+        x264profile=" --level 41 --profile high "
+    x264crf=u"--crf "+str(crf)
+    txt=""
+    os.environ['LANG']="ja_JP.UTF-8"
+    exe=ffmpeg+u" -y -i \""+pin+"\" -vsync 400 -vcodec rawvideo -pix_fmt yuv420p "+s+fps+"-deinterlace -an -f rawvideo - |"
+    exe=exe+u" nice -n 19 "+x264+" "+x264crf+u" "+x264_addline+u" --threads auto "+x264profile+x264preset+" "+x264tune+" "+x264fps+" -o \""+pout+"\" - "+x264streamsize
+    txt=txt+"Cmd : "+exe+"\n"
+    exe = "nice -n 19 " + exe
+    txt=txt+"Cmd : "+exe+"\n"
+    txtt=""
+    recdblist.printutf8(exe)
+    try:
+        txtt=unicode(commands.getoutput(exe.encode('utf-8')),'utf-8')
+    except:
+        ""
+    txt=txt+txtt
     recdblist.addlog( pin, txt, "dual audio ffmpeg-log")
     
\ No newline at end of file