git.plutz.net Git - viper/blob - plugins/v_x264.py

   1 #encoding: utf-8
   2 #Copyright 2011 Paul Hänsch
   3
   4 #This file is part of Viper.
   5
   6 #Viper is free software: you can redistribute it and/or modify
   7 #it under the terms of the GNU General Public License as published by
   8 #the Free Software Foundation, either version 3 of the License, or
   9 #(at your option) any later version.
  10
  11 #Viper is distributed in the hope that it will be useful,
  12 #but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 #GNU General Public License for more details.
  15
  16 #You should have received a copy of the GNU General Public License
  17 #along with Viper.  If not, see <http://www.gnu.org/licenses/>
  18
  19 from plugin import *
  20
  21 class V_x264(V_Plugin):
  22     def build_dialog(self):
  23         self.subgets['l_profile'] = Label(self, text = 'Profile:')
  24         self.subgets['l_profile'].grid(row = 0, column = 0)
  25         self.subgets['o_profile'] = Pmw.OptionMenu(self, initialitem = 2,
  26                                                 items = ['baseline', 'main', 'high'])
  27         self.subgets['o_profile'].grid(row = 1, column = 0)
  28
  29         self.subgets['l_preset'] = Label(self, text = 'Preset:')
  30         self.subgets['l_preset'].grid(row = 0, column = 1)
  31         self.subgets['o_preset'] = Pmw.OptionMenu(self, initialitem = 5,
  32                                                 items = ['ultrafast', 'superfast', 'veryfast', 'faster',
  33                                                          'fast', 'medium', 'slow', 'slower', 'veryslow',
  34                                                          'placebo'])
  35         self.subgets['o_preset'].grid(row = 1, column = 1)
  36
  37         self.subgets['l_keyint'] = Label(self, text = 'Keyframe\nInterval:')
  38         self.subgets['l_keyint'].grid(row = 0, column = 2)
  39         self.subgets['s_keyint'] = Spinbox(self, from_ = 1, to = 2000, increment = 25, width = 5)
  40         self.subgets['s_keyint'].delete(0, END)
  41         self.subgets['s_keyint'].insert(END, 250)
  42         self.subgets['s_keyint'].grid(row = 1, column = 2)
  43
  44         self.subgets['psy'] = False
  45         self.subgets['c_psy'] = Checkbutton(self, text = 'Psychovisual\nOptimization:', command = self.c_psy)
  46         self.subgets['c_psy'].deselect()
  47         self.subgets['c_psy'].grid(row = 0, column = 3)
  48         self.subgets['o_psy'] = Pmw.OptionMenu(self, initialitem = 0, items = ['          '])
  49         self.subgets['o_psy'].grid(row = 1, column = 3)
  50
  51         self.subgets['bitrate'] = False
  52         self.subgets['c_bitrate'] = Checkbutton(self, text = 'Force avg.\nBitrate (Kb/s):',
  53                                                 command = self.c_bitrate)
  54         self.subgets['c_bitrate'].deselect()
  55         self.subgets['c_bitrate'].grid(row = 0, column = 4)
  56         self.subgets['s_bitrate'] = Spinbox(self, from_ = 0, to = 20000, increment = 100, width = 6)
  57         self.subgets['s_bitrate'].delete(0, END)
  58         self.subgets['s_bitrate'].insert(END, 4000)
  59         self.subgets['s_bitrate'].config(state = 'disabled')
  60         self.subgets['s_bitrate'].grid(row = 1, column = 4)
  61
  62         self.subgets['fastdecode'] = False
  63         self.subgets['c_fastdecode'] = Checkbutton(self, text = 'Fastdecode', command = self.c_fastdecode)
  64         self.subgets['c_fastdecode'].deselect()
  65         self.subgets['c_fastdecode'].grid(row = 2, column = 1, columnspan = 2)
  66
  67         self.subgets['zerolatency'] = False
  68         self.subgets['c_zerolatency'] = Checkbutton(self, text = 'Zerolatency', command = self.c_zerolatency)
  69         self.subgets['c_zerolatency'].deselect()
  70         self.subgets['c_zerolatency'].grid(row = 2, column = 3, columnspan = 2)
  71
  72     def c_psy(self):
  73         if self.subgets['psy']:
  74             self.subgets['o_psy'].setitems(['          '])
  75         else:
  76             self.subgets['o_psy'].setitems(['film', 'animation', 'grain', 'stillimage', 'psnr', 'ssim'])
  77         self.subgets['psy'] = not self.subgets['psy']
  78
  79     def c_bitrate(self):
  80         if self.subgets['bitrate']:
  81             self.subgets['s_bitrate'].config(state = 'disabled')
  82         else:
  83             self.subgets['s_bitrate'].config(state = 'normal')
  84         self.subgets['bitrate'] = not self.subgets['bitrate']
  85
  86     def c_fastdecode(self):
  87         self.subgets['fastdecode'] = not self.subgets['fastdecode']
  88
  89     def c_zerolatency(self):
  90         self.subgets['zerolatency'] = not self.subgets['zerolatency']
  91
  92     def name(self):
  93         return 'h264 (via x264)'
  94
  95     def subcommand(self):
  96         opts = ('profile=' + self.subgets['o_profile'].getvalue() +
  97                 ':preset=' + self.subgets['o_preset'].getvalue() +
  98                 ':keyint=' + self.subgets['s_keyint'].get())
  99         if self.subgets['psy']:
 100             opts += ':tune=' + self.subgets['o_psy'].getvalue()
 101             if self.subgets['fastdecode']:
 102                 opts += ',fastdecode'
 103             if self.subgets['zerolatency']:
 104                 opts += ',zerolatency'
 105         elif self.subgets['fastdecode'] and self.subgets['zerolatency']:
 106             opts += ':tune=fastdecode,zerolatency'
 107         elif self.subgets['fastdecode']:
 108             opts += ':tune=fastdecode'
 109         elif self.subgets['zerolatency']:
 110             opts += ':tune=zerolatency'
 111
 112         return ['-ovc', 'x264', '-x264encopts', opts]
 113
 114     def is_available(self):
 115         result = False
 116         slave = os.popen2(['mencoder', '-ovc', 'help'])
 117         line = '#'
 118         while line != '':
 119             line = slave[1].readline()
 120             if line.lstrip().find('x264') == 0:
 121                 result = True
 122         return result