rec10/trunk/src/ts2x264.py

   1 #!/usr/bin/python
   2 # coding: UTF-8
   3 # Rec10 TS Recording Tools
   4 # Copyright (C) 2009-2010 Yukikaze
   5 import commands
   6 import configreader
   7 import os
   8 import os.path
   9 import re
  10 import random
  11
  12 import recdblist
  13 def ts2x264(pin, pout, opts):#sizeは"HD"か"SD"
  14     """
  15     pinで指定されたファイルをpoutにx264でエンコードして書き出す
  16     """
  17     dualpass = 0
  18     is24fps=0
  19     size="HD"
  20     crf=18
  21     quality=4
  22     quality=int(configreader.getenv("x264_preset"))
  23     crf=int(configreader.getenv("crf"))
  24     if re.search("H", opts):
  25         size = "HD"
  26     if re.search("S", opts):
  27         size = "SD"
  28     if re.search("F", opts):
  29         size = "FullHD"
  30     if re.search("W",opts):
  31         size = "WVGA"
  32     if re.search("v", opts):
  33         is24fps=1
  34         crf=int(configreader.getenv("animation_crf"))
  35     if re.search("a", opts):
  36         is24fps=1
  37         crf=int(configreader.getenv("animation_crf"))
  38     if re.search("q",opts):
  39         quality=quality-2
  40     if re.search("w",opts):
  41         quality=quality-1
  42     if re.search("e",opts):
  43         quality=quality+1
  44     if re.search("r",opts):
  45         quality=quality+2
  46     if re.search("u",opts):
  47         crf=crf+2
  48     if re.search("i",opts):
  49         crf=crf+1
  50     if re.search("o",opts):
  51         crf=crf-1
  52     if re.search("p",opts):
  53         crf=crf-2
  54     if re.search("d",opts):
  55         encode_ffmpeg(pin,pout,size,is24fps,quality,crf)
  56     elif re.search("5",opts):
  57         encode_ffmpeg(pin,pout,size,is24fps,quality,crf)
  58     else:
  59         try:
  60             encode(pin, pout,size,is24fps,quality,crf)
  61         except Exception, inst:
  62             print "error occures in ts2x264.py encode"
  63             print type(inst)
  64             print inst
  65 def encode(pin,pout,size,is24fps,quality,crf):
  66     encvf=""
  67     txt=""
  68     encvf="-vf yadif=0,pp=l5"
  69     harddup=",hqdn3d=2:1:2,unsharp=l3x3:0.75:c3x3:0.75,harddup"
  70     ofps="-ofps 30000/1001"
  71     fps="-fps 30000/1001"
  72     x264fps="--fps 30000/1001"
  73     x264streamsize=""
  74     x264preset=""
  75     x264tune=""
  76     x264_maxrate="10000"
  77     if is24fps==1:
  78         ofps="-ofps 24000/1001"
  79         fps="-fps 30000/1001"
  80         x264fps="--fps 24000/1001"
  81         x264tune="--tune animation"
  82         encvf="-vf pullup,softskip"
  83         harddup=",pp=l5,unsharp=l3x3:0.75:c3x3:0.75,hqdn3d=2:1:2,harddup"
  84     if size == "HD":
  85         encvf = encvf + ",scale=1280:720::0:3"+harddup
  86         x264streamsize=u"1280x720"
  87         x264_maxrate="7500"
  88     elif size == "WVGA":
  89         encvf = encvf + ",scale=854:480::0:3"+harddup
  90         x264streamsize=u"854x480"
  91         x264_maxrate="2500"
  92     elif size == "FullHD":
  93         encvf = encvf + ",scale=1920:1080::0:3"+harddup
  94         x264streamsize=u"1920x1080"
  95         x264_maxrate="10000"
  96     elif size == "SD":
  97         encvf = encvf + ",scale=720:480::0:3"+harddup
  98         x264streamsize=u"720x480"
  99         x264_maxrate="2500"
 100     else:
 101         encvf = encvf + ",scale=1280:720::0:3"+harddup
 102         x264streamsize=u"1280x720"
 103         x264_maxrate="7500"
 104     if quality==1:
 105         x264preset=u"--preset ultrafast"
 106     elif quality==2:
 107         x264preset=u"--preset veryfast"
 108     elif quality==3:
 109         x264preset=u"--preset fast"
 110     elif quality==4:
 111         x264preset=u"--preset medium"
 112     elif quality==5:
 113         x264preset=u"--preset slow"
 114     elif quality==6:
 115         x264preset=u"--preset slower"
 116     x264crf=u"--crf "+str(crf)
 117     os.environ['LANG']="ja_JP.UTF-8"
 118     random.seed(pin)
 119     random.jumpahead(10)
 120     streampath=os.path.join(os.path.dirname(pin),str(random.randint(10000, 999999)))
 121     os.system(u"mkfifo "+streampath)
 122     encexe=u"mencoder \""+pin+u"\" -vfm ffmpeg -quiet -sws 9 "+encvf+u",format=i420 "+fps+" "+ofps+" -oac mp3lame -ovc raw -of rawvideo -o \""+streampath+"\" &"
 123     encexe=encexe+u" nice -n 19 x264 "+x264crf+" --vbv-bufsize "+str(int(x264_maxrate)*2)+u" --vbv-maxrate "+x264_maxrate+u" --level 41 --profile high "+x264preset+" "+x264tune+" "+x264fps+" -o \""+pout+"\" "+streampath+" "+x264streamsize
 124     encexe=u"nice -n 19 " +encexe
 125     recdblist.printutf8(encexe)
 126     txt=txt+"Cmd : "+encexe+"\n"
 127     txtt=""
 128     try:
 129         txtt=unicode(commands.getoutput(encexe.encode('utf-8')),'utf-8')
 130     except:
 131         ""
 132     os.system("rm "+streampath)
 133     txt=txt+txtt
 134     recdblist.addlog( pin, txt, "mencoder-log")
 135
 136 def encode_ffmpeg(pin,pout,size,is24fps,quality,crf):
 137     """
 138
 139     """
 140     ffmpeg="ffmpeg"
 141     fps=u"-r 29.970030 "
 142     x264fps="--fps 30000/1001"
 143     x264streamsize=""
 144     x264preset=""
 145     x264tune=""
 146     x264_maxrate="10000"
 147     if size == "HD":
 148         s = "-s 1280x720 "
 149         x264streamsize=u"1280x720"
 150         x264_maxrate="7500"
 151     elif size == "WVGA":
 152         s = "-s 854x480 "
 153         x264streamsize=u"854x480"
 154         x264_maxrate="2500"
 155     elif size == "FullHD":
 156         s = "-s 1920x1080 "
 157         x264streamsize=u"1920x1080"
 158         x264_maxrate="10000"
 159     elif size == "SD":
 160         s = "-s 720x480 "
 161         x264streamsize=u"720x480"
 162         x264_maxrate="2500"
 163     else:
 164         s = "-s 1280x720 "
 165         x264streamsize=u"1280x720"
 166         x264_maxrate="7500"
 167     if quality==1:
 168         x264preset=u"--preset ultrafast"
 169     elif quality==2:
 170         x264preset=u"--preset veryfast"
 171     elif quality==3:
 172         x264preset=u"--preset fast"
 173     elif quality==4:
 174         x264preset=u"--preset medium"
 175     elif quality==5:
 176         x264preset=u"--preset slow"
 177     elif quality==6:
 178         x264preset=u"--preset slower"
 179     x264crf=u"--crf "+str(crf)
 180     txt=""
 181     os.environ['LANG']="ja_JP.UTF-8"
 182     exe=ffmpeg+u" -y -i \""+pin+"\" -vsync 400 -vcodec rawvideo -pix_fmt yuv420p "+s+fps+"-deinterlace -an -f rawvideo - |"
 183     exe=exe+u" nice -n 19 x264 "+x264crf+" --vbv-bufsize "+str(int(x264_maxrate)*2)+u" --vbv-maxrate "+x264_maxrate+u" --level 41 --profile high "+x264preset+" "+x264tune+" "+x264fps+" -o \""+pout+"\" - "+x264streamsize
 184     txt=txt+"Cmd : "+exe+"\n"
 185     exe = "nice -n 19 " + exe
 186     txt=txt+"Cmd : "+exe+"\n"
 187     txtt=""
 188     recdblist.printutf8(exe)
 189     try:
 190         txtt=unicode(commands.getoutput(exe.encode('utf-8')),'utf-8')
 191     except:
 192         ""
 193     txt=txt+txtt
 194     recdblist.addlog( pin, txt, "dual audio ffmpeg-log")
 195