allow better and smarter configuration for > 2 nims
[enigma2.git] / lib / python / Screens / Satconfig.py
1 from enigma import eDVBDB
2 from Screen import Screen
3 from Components.SystemInfo import SystemInfo
4 from Components.ActionMap import ActionMap
5 from Components.ConfigList import ConfigListScreen
6 from Components.MenuList import MenuList
7 from Components.NimManager import nimmanager
8 from Components.config import getConfigListEntry, config, ConfigNothing, ConfigSelection
9 from Screens.MessageBox import MessageBox
10
11 from time import mktime, localtime
12 from datetime import datetime
13
14 class NimSetup(Screen, ConfigListScreen):
15         def createSimpleSetup(self, list, mode):
16                 if mode == "single":
17                         list.append(getConfigListEntry(_("Satellite"), self.nimConfig.diseqcA))
18                 else:
19                         list.append(getConfigListEntry(_("Port A"), self.nimConfig.diseqcA))
20
21                 if mode in ["toneburst_a_b", "diseqc_a_b", "diseqc_a_b_c_d"]:
22                         list.append(getConfigListEntry(_("Port B"), self.nimConfig.diseqcB))
23                         if mode == "diseqc_a_b_c_d":
24                                 list.append(getConfigListEntry(_("Port C"), self.nimConfig.diseqcC))
25                                 list.append(getConfigListEntry(_("Port D"), self.nimConfig.diseqcD))
26
27         def createPositionerSetup(self, list):
28                 nim = self.nimConfig
29                 list.append(getConfigListEntry(_("Longitude"), nim.longitude))
30                 list.append(getConfigListEntry(" ", nim.longitudeOrientation))
31                 list.append(getConfigListEntry(_("Latitude"), nim.latitude))
32                 list.append(getConfigListEntry(" ", nim.latitudeOrientation))
33                 if SystemInfo["CanMeasureFrontendInputPower"]:
34                         self.advancedPowerMeasurement = getConfigListEntry(_("Use Power Measurement"), nim.powerMeasurement)
35                         list.append(self.advancedPowerMeasurement)
36                         if nim.powerMeasurement.value:
37                                 list.append(getConfigListEntry(_("Power threshold in mA"), nim.powerThreshold))
38                                 self.turningSpeed = getConfigListEntry(_("Rotor turning speed"), nim.turningSpeed)
39                                 list.append(self.turningSpeed)
40                                 if nim.turningSpeed.value == "fast epoch":
41                                         self.turnFastEpochBegin = getConfigListEntry(_("Begin time"), nim.fastTurningBegin)
42                                         self.turnFastEpochEnd = getConfigListEntry(_("End time"), nim.fastTurningEnd)
43                                         list.append(self.turnFastEpochBegin)
44                                         list.append(self.turnFastEpochEnd)
45                 else:
46                         if nim.powerMeasurement.value:
47                                 nim.powerMeasurement.value = False
48                                 nim.powerMeasurement.save()
49                                 
50         def createSetup(self):
51                 print "Creating setup"
52                 self.list = [ ]
53
54                 self.configMode = None
55                 self.diseqcModeEntry = None
56                 self.advancedSatsEntry = None
57                 self.advancedLnbsEntry = None
58                 self.advancedDiseqcMode = None
59                 self.advancedUsalsEntry = None
60                 self.advancedLof = None
61                 self.advancedPowerMeasurement = None
62                 self.turningSpeed = None
63                 self.turnFastEpochBegin = None
64                 self.turnFastEpochEnd = None
65                 self.uncommittedDiseqcCommand = None
66                 self.cableScanType = None
67                 self.have_advanced = False
68
69                 if self.nim.isCompatible("DVB-S"):
70                         self.configMode = getConfigListEntry(_("Configuration Mode"), self.nimConfig.configMode)
71                         self.list.append(self.configMode)
72
73                         if self.nimConfig.configMode.value == "simple":                 #simple setup
74                                 self.diseqcModeEntry = getConfigListEntry(_("DiSEqC Mode"), self.nimConfig.diseqcMode)
75                                 self.list.append(self.diseqcModeEntry)
76                                 if self.nimConfig.diseqcMode.value in ["single", "toneburst_a_b", "diseqc_a_b", "diseqc_a_b_c_d"]:
77                                         self.createSimpleSetup(self.list, self.nimConfig.diseqcMode.value)
78                                 if self.nimConfig.diseqcMode.value == "positioner":
79                                         self.createPositionerSetup(self.list)
80                         elif self.nimConfig.configMode.value in ["satposdepends", "equal"]:
81                                 choices = []
82                                 nimlist = nimmanager.getNimListOfType(self.nim.type, exception = self.nim.slot)
83                                 for id in nimlist:
84                                         #choices.append((str(id), str(chr(65 + id))))
85                                         choices.append((str(id), nimmanager.getNimDescription(id)))
86                                 self.nimConfig.connectedTo = ConfigSelection(choices = choices)
87                                 self.list.append(getConfigListEntry(_("Tuner"), self.nimConfig.connectedTo))
88                         elif self.nimConfig.configMode.value == "loopthrough":
89                                 choices = []
90                                 print "connectable to:", nimmanager.connectableTo(self.slotid)
91                                 connectable = nimmanager.connectableTo(self.slotid) 
92                                 for id in connectable:
93                                         choices.append((str(id), nimmanager.getNimDescription(id)))
94                                 self.nimConfig.connectedTo = ConfigSelection(choices = choices)
95                                 self.list.append(getConfigListEntry(_("Connected to"), self.nimConfig.connectedTo))
96                         elif self.nimConfig.configMode.value == "nothing":
97                                 pass
98                         elif self.nimConfig.configMode.value == "advanced": # advanced
99                                 # SATs
100                                 self.advancedSatsEntry = getConfigListEntry(_("Satellite"), self.nimConfig.advanced.sats)
101                                 self.list.append(self.advancedSatsEntry)
102                                 cur_orb_pos = self.nimConfig.advanced.sats.orbital_position
103                                 satlist = self.nimConfig.advanced.sat.keys()
104                                 if cur_orb_pos is not None:
105                                         if cur_orb_pos not in satlist:
106                                                 cur_orb_pos = satlist[0]
107                                         currSat = self.nimConfig.advanced.sat[cur_orb_pos]
108                                         self.fillListWithAdvancedSatEntrys(currSat)
109                                 self.have_advanced = True
110                 elif self.nim.isCompatible("DVB-C"):
111                         self.configMode = getConfigListEntry(_("Configuration Mode"), self.nimConfig.configMode)
112                         self.list.append(self.configMode)
113                         if self.nimConfig.configMode.value == "enabled":
114                                 self.cableScanType=getConfigListEntry(_("Used service scan type"), self.nimConfig.cable.scan_type)
115                                 self.list.append(self.cableScanType)
116                                 if self.nimConfig.cable.scan_type.value == "provider":
117                                         self.list.append(getConfigListEntry(_("Provider to scan"), self.nimConfig.cable.scan_provider))
118                                 else:
119                                         if self.nimConfig.cable.scan_type.value == "bands":
120                                                 self.list.append(getConfigListEntry(_("Scan band EU VHF I"), self.nimConfig.cable.scan_band_EU_VHF_I))
121                                                 self.list.append(getConfigListEntry(_("Scan band EU MID"), self.nimConfig.cable.scan_band_EU_MID))
122                                                 self.list.append(getConfigListEntry(_("Scan band EU VHF III"), self.nimConfig.cable.scan_band_EU_VHF_III))
123                                                 self.list.append(getConfigListEntry(_("Scan band EU UHF IV"), self.nimConfig.cable.scan_band_EU_UHF_IV))
124                                                 self.list.append(getConfigListEntry(_("Scan band EU UHF V"), self.nimConfig.cable.scan_band_EU_UHF_V))
125                                                 self.list.append(getConfigListEntry(_("Scan band EU SUPER"), self.nimConfig.cable.scan_band_EU_SUPER))
126                                                 self.list.append(getConfigListEntry(_("Scan band EU HYPER"), self.nimConfig.cable.scan_band_EU_HYPER))
127                                                 self.list.append(getConfigListEntry(_("Scan band US LOW"), self.nimConfig.cable.scan_band_US_LOW))
128                                                 self.list.append(getConfigListEntry(_("Scan band US MID"), self.nimConfig.cable.scan_band_US_MID))
129                                                 self.list.append(getConfigListEntry(_("Scan band US HIGH"), self.nimConfig.cable.scan_band_US_HIGH))
130                                                 self.list.append(getConfigListEntry(_("Scan band US SUPER"), self.nimConfig.cable.scan_band_US_SUPER))
131                                                 self.list.append(getConfigListEntry(_("Scan band US HYPER"), self.nimConfig.cable.scan_band_US_HYPER))
132                                         elif self.nimConfig.cable.scan_type.value == "steps":
133                                                 self.list.append(getConfigListEntry(_("Frequency scan step size(khz)"), self.nimConfig.cable.scan_frequency_steps))
134                                         self.list.append(getConfigListEntry(_("Scan QAM16"), self.nimConfig.cable.scan_mod_qam16))
135                                         self.list.append(getConfigListEntry(_("Scan QAM32"), self.nimConfig.cable.scan_mod_qam32))
136                                         self.list.append(getConfigListEntry(_("Scan QAM64"), self.nimConfig.cable.scan_mod_qam64))
137                                         self.list.append(getConfigListEntry(_("Scan QAM128"), self.nimConfig.cable.scan_mod_qam128))
138                                         self.list.append(getConfigListEntry(_("Scan QAM256"), self.nimConfig.cable.scan_mod_qam256))
139                                         self.list.append(getConfigListEntry(_("Scan SR6900"), self.nimConfig.cable.scan_sr_6900))
140                                         self.list.append(getConfigListEntry(_("Scan SR6875"), self.nimConfig.cable.scan_sr_6875))
141                                         self.list.append(getConfigListEntry(_("Scan additional SR"), self.nimConfig.cable.scan_sr_ext1))
142                                         self.list.append(getConfigListEntry(_("Scan additional SR"), self.nimConfig.cable.scan_sr_ext2))
143                         self.have_advanced = False
144                 elif self.nim.isCompatible("DVB-T"):
145                         self.configMode = getConfigListEntry(_("Configuration Mode"), self.nimConfig.configMode)
146                         self.list.append(self.configMode)
147                         self.have_advanced = False
148                         if self.nimConfig.configMode.value == "enabled":
149                                 self.list.append(getConfigListEntry(_("Terrestrial provider"), self.nimConfig.terrestrial))
150                                 self.list.append(getConfigListEntry(_("Enable 5V for active antenna"), self.nimConfig.terrestrial_5V))
151                 else:
152                         self.have_advanced = False
153                 self["config"].list = self.list
154                 self["config"].l.setList(self.list)
155
156         def newConfig(self):
157                 checkList = (self.configMode, self.diseqcModeEntry, self.advancedSatsEntry, \
158                         self.advancedLnbsEntry, self.advancedDiseqcMode, self.advancedUsalsEntry, \
159                         self.advancedLof, self.advancedPowerMeasurement, self.turningSpeed, \
160                         self.uncommittedDiseqcCommand, self.cableScanType)
161                 for x in checkList:
162                         if self["config"].getCurrent() == x:
163                                 self.createSetup()
164
165         def run(self):
166                 if self.have_advanced and self.nim.config_mode == "advanced":
167                         self.fillAdvancedList()
168                 for x in self.list:
169                         if x in [self.turnFastEpochBegin, self.turnFastEpochEnd]:
170                                 # workaround for storing only hour*3600+min*60 value in configfile
171                                 # not really needed.. just for cosmetics..
172                                 tm = localtime(x[1].value)
173                                 dt = datetime(1970, 1, 1, tm.tm_hour, tm.tm_min)
174                                 x[1].value = int(mktime(dt.timetuple()))
175                         x[1].save()
176                 nimmanager.sec.update()
177
178         def fillListWithAdvancedSatEntrys(self, Sat):
179                 currLnb = self.nimConfig.advanced.lnb[int(Sat.lnb.value)]
180                 
181                 if isinstance(currLnb, ConfigNothing):
182                         currLnb = None
183
184                 self.list.append(getConfigListEntry(_("Voltage mode"), Sat.voltage))
185                 self.list.append(getConfigListEntry(_("Tone mode"), Sat.tonemode))
186                 if currLnb and currLnb.diseqcMode.value == "1_2":
187                         self.advancedUsalsEntry = getConfigListEntry(_("Use usals for this sat"), Sat.usals)
188                         self.list.append(self.advancedUsalsEntry)
189                         if not Sat.usals.value:
190                                 self.list.append(getConfigListEntry(_("Stored position"), Sat.rotorposition))
191
192                 # LNBs
193                 self.advancedLnbsEntry = getConfigListEntry(_("LNB"), Sat.lnb)
194                 self.list.append(self.advancedLnbsEntry)
195                 if currLnb:
196                         self.advancedDiseqcMode = getConfigListEntry(_("DiSEqC mode"), currLnb.diseqcMode)
197                         self.list.append(self.advancedDiseqcMode)
198                         if currLnb.diseqcMode.value != "none":
199                                 self.list.append(getConfigListEntry(_("Toneburst"), currLnb.toneburst))
200                                 self.list.append(getConfigListEntry(_("Committed DiSEqC command"), currLnb.commitedDiseqcCommand))
201                                 self.list.append(getConfigListEntry(_("Fast DiSEqC"), currLnb.fastDiseqc))
202                                 self.list.append(getConfigListEntry(_("Sequence repeat"), currLnb.sequenceRepeat))
203                                 if currLnb.diseqcMode.value == "1_0":
204                                         self.list.append(getConfigListEntry(_("Command order"), currLnb.commandOrder1_0))
205                                 else:
206                                         if currLnb.uncommittedDiseqcCommand.index:
207                                                 if currLnb.commandOrder.value == "ct":
208                                                         currLnb.commandOrder.value = "cut"
209                                                 elif currLnb.commandOrder.value == "tc":
210                                                         currLnb.commandOrder.value = "tcu"
211                                         else:
212                                                 if currLnb.commandOrder.index & 1:
213                                                         currLnb.commandOrder.value = "tc"
214                                                 else:
215                                                         currLnb.commandOrder.value = "ct"
216                                         self.list.append(getConfigListEntry(_("Command order"), currLnb.commandOrder))
217                                         self.uncommittedDiseqcCommand = getConfigListEntry(_("Uncommitted DiSEqC command"), currLnb.uncommittedDiseqcCommand)
218                                         self.list.append(self.uncommittedDiseqcCommand)
219                                         self.list.append(getConfigListEntry(_("DiSEqC repeats"), currLnb.diseqcRepeats))
220                                 if currLnb.diseqcMode.value == "1_2":
221                                         self.list.append(getConfigListEntry(_("Longitude"), currLnb.longitude))
222                                         self.list.append(getConfigListEntry(" ", currLnb.longitudeOrientation))
223                                         self.list.append(getConfigListEntry(_("Latitude"), currLnb.latitude))
224                                         self.list.append(getConfigListEntry(" ", currLnb.latitudeOrientation))
225                                         if SystemInfo["CanMeasureFrontendInputPower"]:
226                                                 self.advancedPowerMeasurement = getConfigListEntry(_("Use Power Measurement"), currLnb.powerMeasurement)
227                                                 self.list.append(self.advancedPowerMeasurement)
228                                                 if currLnb.powerMeasurement.value:
229                                                         self.list.append(getConfigListEntry(_("Power threshold in mA"), currLnb.powerThreshold))
230                                                         self.turningSpeed = getConfigListEntry(_("Rotor turning speed"), currLnb.turningSpeed)
231                                                         self.list.append(self.turningSpeed)
232                                                         if currLnb.turningSpeed.value == "fast epoch":
233                                                                 self.turnFastEpochBegin = getConfigListEntry(_("Begin time"), currLnb.fastTurningBegin)
234                                                                 self.turnFastEpochEnd = getConfigListEntry(_("End time"), currLnb.fastTurningEnd)
235                                                                 self.list.append(self.turnFastEpochBegin)
236                                                                 self.list.append(self.turnFastEpochEnd)
237                                         else:
238                                                 if currLnb.powerMeasurement.value:
239                                                         currLnb.powerMeasurement.value = False
240                                                         currLnb.powerMeasurement.save()
241                         self.advancedLof = getConfigListEntry(_("LOF"), currLnb.lof)
242                         self.list.append(self.advancedLof)
243                         if currLnb.lof.value == "user_defined":
244                                 self.list.append(getConfigListEntry(_("LOF/L"), currLnb.lofl))
245                                 self.list.append(getConfigListEntry(_("LOF/H"), currLnb.lofh))
246                                 self.list.append(getConfigListEntry(_("Threshold"), currLnb.threshold))
247 #                       self.list.append(getConfigListEntry(_("12V Output"), currLnb.output_12v))
248                         self.list.append(getConfigListEntry(_("Increased voltage"), currLnb.increased_voltage))
249
250         def fillAdvancedList(self):
251                 self.list = [ ]
252                 self.configMode = getConfigListEntry(_("Configuration Mode"), self.nimConfig.configMode)
253                 self.list.append(self.configMode)
254                 self.advancedSatsEntry = getConfigListEntry(_("Satellite"), self.nimConfig.advanced.sats)
255                 self.list.append(self.advancedSatsEntry)
256                 for x in nimmanager.satList:
257                         Sat = self.nimConfig.advanced.sat[x[0]]
258                         self.fillListWithAdvancedSatEntrys(Sat)
259                 self["config"].list = self.list
260
261         def keySave(self):
262                 old_configured_sats = nimmanager.getConfiguredSats()
263                 self.run()
264                 new_configured_sats = nimmanager.getConfiguredSats()
265                 self.unconfed_sats = old_configured_sats - new_configured_sats
266                 self.satpos_to_remove = None
267                 self.deleteConfirmed(False)
268
269         def deleteConfirmed(self, confirmed):
270                 if confirmed:
271                         eDVBDB.getInstance().removeServices(-1, -1, -1, self.satpos_to_remove)
272
273                 if self.satpos_to_remove is not None:
274                         self.unconfed_sats.remove(self.satpos_to_remove)
275
276                 self.satpos_to_remove = None
277                 for orbpos in self.unconfed_sats:
278                         self.satpos_to_remove = orbpos
279                         orbpos = self.satpos_to_remove
280                         try:
281                                 # why we need this cast?
282                                 sat_name = str(nimmanager.getSatDescription(orbpos))
283                         except:
284                                 if orbpos > 1800: # west
285                                         orbpos = 3600 - orbpos
286                                         h = _("W")
287                                 else:
288                                         h = _("E")
289                                 sat_name = ("%d.%d" + h) % (orbpos / 10, orbpos % 10)
290                         self.session.openWithCallback(self.deleteConfirmed, MessageBox, _("Delete no more configured satellite\n%s?") %(sat_name))
291                         break
292                 if not self.satpos_to_remove:
293                         self.close()
294
295         def __init__(self, session, slotid):
296                 Screen.__init__(self, session)
297                 self.list = [ ]
298
299                 ConfigListScreen.__init__(self, self.list)
300
301                 self["actions"] = ActionMap(["SetupActions"],
302                 {
303                         "ok": self.keySave,
304                         "cancel": self.keyCancel,
305                 }, -2)
306
307                 self.slotid = slotid
308                 self.nim = nimmanager.nim_slots[slotid]
309                 self.nimConfig = self.nim.config
310                 self.createSetup()
311
312         def keyLeft(self):
313                 ConfigListScreen.keyLeft(self)
314                 self.newConfig()
315
316         def keyRight(self):
317                 ConfigListScreen.keyRight(self)
318                 self.newConfig()
319
320 class NimSelection(Screen):
321         def __init__(self, session):
322                 Screen.__init__(self, session)
323                 
324                 menu = [ ]
325                 for x in nimmanager.nim_slots:
326                         menu.append((x.friendly_full_description, x))
327
328                 self["nimlist"] = MenuList(menu)
329
330                 self["actions"] = ActionMap(["OkCancelActions"],
331                 {
332                         "ok": self.okbuttonClick ,
333                         "cancel": self.close
334                 }, -2)
335
336         def okbuttonClick(self):
337                 nim = self["nimlist"].getCurrent()
338                 nim = nim and nim[1]
339                 if nim is not None and not nim.empty:
340                         self.session.open(NimSetup, nim.slot)