get rid of some floating point values
[enigma2.git] / lib / python / Components / AVSwitch.py
1 from config import config, ConfigSlider, ConfigSelection, ConfigYesNo, ConfigEnableDisable, ConfigSubsection, ConfigBoolean
2 from enigma import eAVSwitch, getDesktop
3 from SystemInfo import SystemInfo
4
5 class AVSwitch:
6         INPUT = { "ENCODER": (0, 4), "SCART": (1, 3), "AUX": (2, 4) }
7
8         def setInput(self, input):
9                 eAVSwitch.getInstance().setInput(self.INPUT[input][0])
10                 if self.INPUT[input][1] == 4:
11                         aspect = self.getAspectRatioSetting()
12                         self.setAspectWSS(aspect)
13                         self.setAspectSlowBlank(aspect)
14                 else:
15                         eAVSwitch.getInstance().setSlowblank(self.INPUT[input][1])
16                 # FIXME why do we have to reset the colorformat? bug in avs-driver?
17                 map = {"cvbs": 0, "rgb": 1, "svideo": 2, "yuv": 3}
18                 eAVSwitch.getInstance().setColorFormat(map[config.av.colorformat.value])
19
20         def setColorFormat(self, value):
21                 eAVSwitch.getInstance().setColorFormat(value)
22
23         def setAspectRatio(self, value):
24                 eAVSwitch.getInstance().setAspectRatio(value)
25                 self.setAspectWSS(value)
26                 self.setAspectSlowBlank(value)
27
28         def setSystem(self, value):
29                 eAVSwitch.getInstance().setVideomode(value)
30
31         def getOutputAspect(self):
32                 if valstr in ("4_3_letterbox", "4_3_panscan"): # 4:3
33                         return (4,3)
34                 elif valstr == "16_9": # auto ... 4:3 or 16:9
35                         try:
36                                 aspect_str = open("/proc/stb/vmpeg/0/aspect", "r").read()
37                                 if aspect_str == "1": # 4:3
38                                         return (4,3)
39                         except IOError:
40                                 pass
41                 elif valstr in ("16_9_always", "16_9_letterbox"): # 16:9
42                         pass
43                 elif valstr in ("16_10_letterbox", "16_10_panscan"): # 16:10
44                         return (16,10)
45                 return (16,9)
46
47         def getFramebufferScale(self):
48                 aspect = self.getOutputAspect()
49                 fb_size = getDesktop(0).size()
50                 return (aspect[0] * fb_size.height(), aspect[1] * fb_size.width())
51
52         def getAspectRatioSetting(self):
53                 valstr = config.av.aspectratio.value
54                 if valstr == "4_3_letterbox":
55                         val = 0
56                 elif valstr == "4_3_panscan":
57                         val = 1
58                 elif valstr == "16_9":
59                         val = 2
60                 elif valstr == "16_9_always":
61                         val = 3
62                 elif valstr == "16_10_letterbox":
63                         val = 4
64                 elif valstr == "16_10_panscan":
65                         val = 5
66                 elif valstr == "16_9_letterbox":
67                         val = 6
68                 return val
69
70         def setAspectWSS(self, aspect=None):
71                 if aspect is None:
72                         aspect = self.getAspectRatioSetting()
73                 if aspect == 0 or aspect == 1: # letterbox or panscan
74                         if not config.av.wss.value:
75                                 value = 0 # wss off
76                         else:
77                                 value = 3 # 4:3_full_format
78                 elif aspect == 2: # 16:9
79                         if not config.av.wss.value:
80                                 value = 2 # auto(4:3_off)
81                         else:
82                                 value = 1 # auto
83                 elif aspect == 3 or aspect == 6: # always 16:9
84                         value = 4 # 16:9_full_format
85                 elif aspect == 4 or aspect == 5: # 16:10
86                         value = 10 # 14:9_full_format
87                 eAVSwitch.getInstance().setWSS(value)
88
89         def setAspectSlowBlank(self, aspect=None):
90                 if aspect is None:
91                         aspect = self.getAspectRatioSetting()
92                 if aspect == 0 or aspect == 1: # letterbox or panscan
93                         value = 2 # 12 V
94                 elif aspect == 2: # 16:9
95                         value = 4 # auto
96                 elif aspect == 3 or aspect == 4 or aspect == 5 or aspect == 6: # always 16:9
97                         value = 1 # 6V
98                 eAVSwitch.getInstance().setSlowblank(value)
99
100 def InitAVSwitch():
101         config.av = ConfigSubsection()
102         config.av.yuvenabled = ConfigBoolean(default=False)
103         colorformat_choices = {"cvbs": _("CVBS"), "rgb": _("RGB"), "svideo": _("S-Video")}
104         
105         # when YUV is not enabled, don't let the user select it
106         if config.av.yuvenabled.value:
107                 colorformat_choices["yuv"] = _("YPbPr")
108
109         config.av.colorformat = ConfigSelection(choices=colorformat_choices, default="rgb")
110         config.av.aspectratio = ConfigSelection(choices={
111                         "4_3_letterbox": _("4:3 Letterbox"),
112                         "4_3_panscan": _("4:3 PanScan"), 
113                         "16_9": _("16:9"), 
114                         "16_9_always": _("16:9 always"),
115                         "16_10_letterbox": _("16:10 Letterbox"),
116                         "16_10_panscan": _("16:10 PanScan"), 
117                         "16_9_letterbox": _("16:9 Letterbox")}, 
118                         default = "4_3_letterbox")
119
120         config.av.aspect = ConfigSelection(choices={
121                         "4_3": _("4:3"),
122                         "16_9": _("16:9"), 
123                         "16_10": _("16:10"),
124                         "auto": _("Automatic")},
125                         default = "auto")
126         config.av.policy_169 = ConfigSelection(choices={
127                                 # TRANSLATORS: (aspect ratio policy: black bars on top/bottom) in doubt, keep english term.
128                         "letterbox": _("Letterbox"), 
129                                 # TRANSLATORS: (aspect ratio policy: cropped content on left/right) in doubt, keep english term
130                         "panscan": _("Pan&Scan"),  
131                                 # TRANSLATORS: (aspect ratio policy: display as fullscreen, even if this breaks the aspect)
132                         "scale": _("Just Scale")},
133                         default = "letterbox")
134         config.av.policy_43 = ConfigSelection(choices={
135                                 # TRANSLATORS: (aspect ratio policy: black bars on left/right) in doubt, keep english term.
136                         "pillarbox": _("Pillarbox"), 
137                                 # TRANSLATORS: (aspect ratio policy: cropped content on left/right) in doubt, keep english term
138                         "panscan": _("Pan&Scan"),  
139                                 # TRANSLATORS: (aspect ratio policy: display as fullscreen, with stretching the left/right)
140                         "nonlinear": _("Nonlinear"),  
141                                 # TRANSLATORS: (aspect ratio policy: display as fullscreen, even if this breaks the aspect)
142                         "scale": _("Just Scale")},
143                         default = "pillarbox")
144         config.av.tvsystem = ConfigSelection(choices = {"pal": _("PAL"), "ntsc": _("NTSC"), "multinorm": _("multinorm")}, default="pal")
145         config.av.wss = ConfigEnableDisable(default = True)
146         config.av.defaultac3 = ConfigYesNo(default = False)
147         config.av.vcrswitch = ConfigEnableDisable(default = False)
148
149         iAVSwitch = AVSwitch()
150
151         def setColorFormat(configElement):
152                 map = {"cvbs": 0, "rgb": 1, "svideo": 2, "yuv": 3}
153                 iAVSwitch.setColorFormat(map[configElement.value])
154
155         def setAspectRatio(configElement):
156                 map = {"4_3_letterbox": 0, "4_3_panscan": 1, "16_9": 2, "16_9_always": 3, "16_10_letterbox": 4, "16_10_panscan": 5, "16_9_letterbox" : 6}
157                 iAVSwitch.setAspectRatio(map[configElement.value])
158
159         def setSystem(configElement):
160                 map = {"pal": 0, "ntsc": 1, "multinorm" : 2}
161                 iAVSwitch.setSystem(map[configElement.value])
162
163         def setWSS(configElement):
164                 iAVSwitch.setAspectWSS()
165
166         # this will call the "setup-val" initial
167         config.av.colorformat.addNotifier(setColorFormat)
168         config.av.aspectratio.addNotifier(setAspectRatio)
169         config.av.tvsystem.addNotifier(setSystem)
170         config.av.wss.addNotifier(setWSS)
171
172         iAVSwitch.setInput("ENCODER") # init on startup
173         SystemInfo["ScartSwitch"] = eAVSwitch.getInstance().haveScartSwitch()
174
175         try:
176                 can_downmix = open("/proc/stb/audio/ac3_choices", "r").read()[:-1].find("downmix") != -1
177         except:
178                 can_downmix = False
179
180         SystemInfo["CanDownmixAC3"] = can_downmix
181         if can_downmix:
182                 def setAC3Downmix(configElement):
183                         open("/proc/stb/audio/ac3", "w").write(configElement.value and "downmix" or "passthrough")
184                 config.av.downmix_ac3 = ConfigYesNo(default = True)
185                 config.av.downmix_ac3.addNotifier(setAC3Downmix)
186
187         try:
188                 can_osd_alpha = open("/proc/stb/video/alpha", "r") and True or False
189         except:
190                 can_osd_alpha = False
191
192         SystemInfo["CanChangeOsdAlpha"] = can_osd_alpha
193
194         def setAlpha(config):
195                 open("/proc/stb/video/alpha", "w").write(str(config.value))
196
197         if can_osd_alpha:
198                 config.av.osd_alpha = ConfigSlider(default=255, limits=(0,255))
199                 config.av.osd_alpha.addNotifier(setAlpha)