1 import ConfigParser
, os
3 from ConfigParser
import NoOptionError
5 BLACKLIST_169
= ('540', '649')
7 config
= ConfigParser
.ConfigParser()
8 p
= os
.path
.dirname(__file__
)
9 config
.read(os
.path
.join(p
, 'pyTivo.conf'))
12 if config
.has_option('Server', 'GUID'):
13 guid
= config
.get('Server', 'GUID')
18 def getBeaconAddresses():
19 if config
.has_option('Server', 'beacon'):
20 beacon_ips
= config
.get('Server', 'beacon')
22 beacon_ips
= '255.255.255.255'
26 return config
.get('Server', 'Port')
28 def get169Setting(tsn
):
32 if config
.has_section('_tivo_' + tsn
):
33 if config
.has_option('_tivo_' + tsn
, 'aspect169'):
34 if config
.get('_tivo_' + tsn
, 'aspect169').lower() == 'true':
39 if tsn
[:3] in BLACKLIST_169
:
45 shares
= [ (section
, dict(config
.items(section
))) for section
in config
.sections() if not(section
.startswith('_tivo_') or section
== 'Server') ]
47 for name
, data
in shares
:
48 if not data
.get('auto_subshares', 'False').lower() == 'true':
51 base_path
= data
['path']
52 for item
in os
.listdir(base_path
):
53 item_path
= os
.path
.join(base_path
, item
)
54 if not os
.path
.isdir(item_path
):
57 new_name
= name
+ '/' + item
59 new_data
['path'] = item_path
61 shares
.append( (new_name
, new_data
) )
68 debug
= config
.get('Server', 'debug')
69 if debug
.lower() == 'true':
78 optres
= config
.get('Server', 'optres')
79 if optres
.lower() == 'true':
86 def get(section
, key
):
87 return config
.get(section
, key
)
89 def getFFMPEGTemplate(tsn
):
90 if tsn
and config
.has_section('_tivo_' + tsn
):
92 return config
.get('_tivo_' + tsn
, 'ffmpeg_prams', raw
= True)
97 return config
.get('Server', 'ffmpeg_prams', raw
= True)
98 except NoOptionError
: #default
99 return '-vcodec mpeg2video -r 29.97 -b %(video_br)s -maxrate %(max_video_br)s -bufsize %(buff_size)s %(aspect_ratio)s -comment pyTivo.py %(audio_codec)s -ab %(audio_br)s -f vob -'
101 def getHDtivos(): # tsn's of High Definition Tivo's
102 return ['648', '652']
104 def getValidWidths():
105 return [1920, 1440, 1280, 720, 704, 544, 480, 352]
107 def getValidHeights():
108 return [1080, 720, 480] # Technically 240 is also supported
110 # Return the number in list that is nearest to x
111 # if two values are equidistant, return the larger
112 def nearest(x
, list):
113 return reduce(lambda a
, b
: closest(x
,a
,b
), list)
116 if abs(x
-a
) < abs(x
-b
) or (abs(x
-a
) == abs(x
-b
)and a
>b
):
121 def nearestTivoHeight(height
):
122 return nearest(height
, getValidHeights())
124 def nearestTivoWidth(width
):
125 return nearest(width
, getValidWidths())
127 def getTivoHeight(tsn
):
128 if tsn
and config
.has_section('_tivo_' + tsn
):
130 height
= int(config
.get('_tivo_' + tsn
, 'height'))
131 return nearestTivoHeight(height
)
132 except NoOptionError
:
136 height
= int(config
.get('Server', 'height'))
137 return nearestTivoHeight(height
)
138 except NoOptionError
: #default
141 def getTivoWidth(tsn
):
142 if tsn
and config
.has_section('_tivo_' + tsn
):
144 width
= int(config
.get('_tivo_' + tsn
, 'width'))
145 return nearestTivoWidth(width
)
146 except NoOptionError
:
150 width
= int(config
.get('Server', 'width'))
151 return nearestTivoWidth(width
)
152 except NoOptionError
: #default
155 def getAudioBR(tsn
= None):
156 #convert to non-zero multiple of 64 to ensure ffmpeg compatibility
157 #compare audio_br to max_audio_br and return lowest
158 if tsn
and config
.has_section('_tivo_' + tsn
):
160 audiobr
= int(max(int(strtod(config
.get('_tivo_' + tsn
, 'audio_br'))/1000), 64)/64)*64
161 return str(min(audiobr
, getMaxAudioBR(tsn
))) + 'k'
162 except NoOptionError
:
166 audiobr
= int(max(int(strtod(config
.get('Server', 'audio_br'))/1000), 64)/64)*64
167 return str(min(audiobr
, getMaxAudioBR(tsn
))) + 'k'
168 except NoOptionError
: #default to 192
171 def getAudioCodec(tsn
= None):
172 #check for HD tivo and return compatible audio parameters
173 if tsn
and tsn
[:3] in getHDtivos():
174 return '-acodec ac3 -ar 48000'
176 return '-acodec mp2 -ac 2 -ar 44100'
178 def getVideoBR(tsn
= None):
179 if tsn
and config
.has_section('_tivo_' + tsn
):
181 return config
.get('_tivo_' + tsn
, 'video_br')
182 except NoOptionError
:
186 return config
.get('Server', 'video_br')
187 except NoOptionError
: #default to 4096K
192 return str(int(strtod(config
.get('Server', 'max_video_br'))/1000)) + 'k'
193 except NoOptionError
: #default to 17Mi
198 return config
.get('Server', 'bufsize')
199 except NoOptionError
: #default 1024k
202 def getMaxAudioBR(tsn
= None):
203 #convert to non-zero multiple of 64 for ffmpeg compatibility
204 if tsn
and config
.has_section('_tivo_' + tsn
):
206 return int(int(strtod(config
.get('_tivo_' + tsn
, 'max_audio_br'))/1000)/64)*64
207 except NoOptionError
:
211 return int(int(strtod(config
.get('Server', 'max_audio_br'))/1000)/64)*64
212 except NoOptionError
:
213 if tsn
and tsn
[:3] in getHDtivos():
214 return int(448) #default to 448, max supported by HD TiVo's
216 return int(384) #default to 384, max supported by mp2 audio (S2 TiVo)
219 # Parse a bitrate using the SI/IEEE suffix values as if by ffmpeg
220 # For example, 2K==2000, 2Ki==2048, 2MB==16000000, 2MiB==16777216
221 # Algorithm: http://svn.mplayerhq.hu/ffmpeg/trunk/libavcodec/eval.c
223 prefixes
= {"y":-24,"z":-21,"a":-18,"f":-15,"p":-12,"n":-9,"u":-6,"m":-3,"c":-2,"d":-1,"h":2,"k":3,"K":3,"M":6,"G":9,"T":12,"P":15,"E":18,"Z":21,"Y":24}
224 p
= re
.compile(r
'^(\d+)(?:([yzafpnumcdhkKMGTPEZY])(i)?)?([Bb])?$')
227 raise SyntaxError('Invalid bit value syntax')
228 (coef
, prefix
, power
, byte
) = m
.groups()
232 exponent
= float(prefixes
[prefix
])
235 value
= float(coef
) * pow(2.0, exponent
/0.3)
238 value
= float(coef
) * pow(10.0, exponent
)
239 if byte
== "B": # B==Byte, b=bit