don't die when entering ScanSetup
[vuplus_dvbapp] / lib / python / Screens / ScanSetup.py
1 from Screen import Screen
2 from ServiceScan import *
3 from Components.config import *
4 from Components.ActionMap import NumberActionMap
5 from Components.ConfigList import ConfigList
6 from Components.config import config
7 from Components.config import getConfigListEntry
8 from Components.NimManager import nimmanager
9 from Components.Label import Label
10 from enigma import eDVBFrontendParametersSatellite, eComponentScan
11
12 def getInitialTransponderList(tlist, pos):
13         print pos
14         list = nimmanager.getTransponders(pos)
15
16         for x in list:
17                 if x[0] == 0:           #SAT
18                         parm = eDVBFrontendParametersSatellite()
19                         parm.frequency = x[1]
20                         parm.symbol_rate = x[2]
21                         parm.polarisation = x[3] # eDVBFrontendParametersSatellite.Polarisation.Vertical
22                         #parm.fec = x[4]                        # eDVBFrontendParametersSatellite.FEC.f3_4;
23                         parm.fec = 6                                    # AUTO
24                         #parm.inversion = 1     #eDVBFrontendParametersSatellite.Inversion.Off;
25                         parm.inversion = 2              #AUTO
26                         parm.orbital_position = pos
27                         tlist.append(parm)
28
29 class ScanSetup(Screen):
30         def __init__(self, session):
31                 Screen.__init__(self, session)
32
33                 self.updateSatList()
34                 self.createConfig()
35
36
37                 self["actions"] = NumberActionMap(["SetupActions"],
38                 {
39                         "ok": self.keyGo,
40                         "cancel": self.keyCancel,
41                         "left": self.keyLeft,
42                         "right": self.keyRight,
43                         "1": self.keyNumberGlobal,
44                         "2": self.keyNumberGlobal,
45                         "3": self.keyNumberGlobal,
46                         "4": self.keyNumberGlobal,
47                         "5": self.keyNumberGlobal,
48                         "6": self.keyNumberGlobal,
49                         "7": self.keyNumberGlobal,
50                         "8": self.keyNumberGlobal,
51                         "9": self.keyNumberGlobal,
52                         "0": self.keyNumberGlobal
53                 }, -1)
54
55                 self.list = []
56                 self["config"] = ConfigList(self.list)
57                 self.createSetup()
58
59                 self["introduction"] = Label("Press OK to start the scan")
60
61         def updateSatList(self):
62                 self.satList = []
63                 for slot in nimmanager.nimslots:
64                         if (nimmanager.getNimType(slot.slotid) == nimmanager.nimType["DVB-S"]):
65                                 self.satList.append(nimmanager.getSatListForNim(slot.slotid))
66
67         def createSetup(self):
68                 self.list = []
69
70                 self.list.append(getConfigListEntry("Type of scan", config.scan.type))
71                 self.list.append(getConfigListEntry("Tuner", config.scan.nims))
72
73                 # single transponder scan
74                 if (config.scan.type.value == 0):
75                         if (nimmanager.getNimType(config.scan.nims.value) == nimmanager.nimType["DVB-S"]):
76                                 self.list.append(getConfigListEntry(_('Satellite'), config.scan.satselection[config.scan.nims.value]))
77                                 self.list.append(getConfigListEntry(_('Frequency'), config.scan.sat.frequency))
78                                 self.list.append(getConfigListEntry(_('Inversion'), config.scan.sat.inversion))
79                                 self.list.append(getConfigListEntry(_('Symbolrate'), config.scan.sat.symbolrate))
80                                 self.list.append(getConfigListEntry("Polarity", config.scan.sat.polarization))
81                                 self.list.append(getConfigListEntry("FEC", config.scan.sat.fec))
82                         if (nimmanager.getNimType(config.scan.nims.value) == nimmanager.nimType["DVB-C"]):
83                                 self.list.append(getConfigListEntry("Frequency", config.scan.cab.frequency))
84                                 self.list.append(getConfigListEntry("Inversion", config.scan.cab.inversion))
85                                 self.list.append(getConfigListEntry("Symbolrate", config.scan.cab.symbolrate))
86                                 self.list.append(getConfigListEntry("Modulation", config.scan.cab.modulation))
87                                 self.list.append(getConfigListEntry("FEC", config.scan.cab.fec))
88                         if (nimmanager.getNimType(config.scan.nims.value) == nimmanager.nimType["DVB-T"]):
89                                 self.list.append(getConfigListEntry("Frequency", config.scan.ter.frequency))
90                                 self.list.append(getConfigListEntry("Inversion", config.scan.ter.inversion))
91                                 self.list.append(getConfigListEntry("Bandwidth", config.scan.ter.bandwidth))
92                                 self.list.append(getConfigListEntry("Code rate high", config.scan.ter.fechigh))
93                                 self.list.append(getConfigListEntry("Code rate low", config.scan.ter.feclow))
94                                 self.list.append(getConfigListEntry("Modulation", config.scan.ter.modulation))
95                                 self.list.append(getConfigListEntry("Transmission mode", config.scan.ter.transmission))
96                                 self.list.append(getConfigListEntry("Guard interval mode", config.scan.ter.guard))
97                                 self.list.append(getConfigListEntry("Hierarchy mode", config.scan.ter.hierarchy))
98
99                 # single satellite scan
100                 print config.scan.satselection
101                 if (config.scan.type.value == 1):
102                         self.updateSatList()
103                         print config.scan.satselection[config.scan.nims.value]
104                         self.list.append(getConfigListEntry("Satellite", config.scan.satselection[config.scan.nims.value]))
105
106
107                 # multi sat scan
108                 if (config.scan.type.value == 2):
109                         # if (norotor)
110                         tlist = []
111                         SatList = nimmanager.getSatListForNim(config.scan.nims.value)
112
113                         for x in SatList:
114                                 if self.Satexists(tlist, x[1]) == 0:
115                                         tlist.append(x[1])
116                                         sat = configElement_nonSave(x[1], configSelection, 0, ("Enable", "Disable"))
117                                         self.list.append(getConfigListEntry(nimmanager.getSatDescription(x[1]), sat))
118
119                         # if (rotor):
120                    # for sat in nimmanager.satList:
121                         #       self.list.append(getConfigListEntry(sat[0], config.scan.scansat[sat[1]]))
122
123                 self["config"].list = self.list
124                 self["config"].l.setList(self.list)
125
126         def Satexists(self, tlist, pos):
127                 for x in tlist:
128                         if x == pos:
129                                 return 1
130                 return 0
131
132         def newConfig(self):
133                 print self["config"].getCurrent()
134                 if self["config"].getCurrent()[0] == "Type of scan":
135                         self.createSetup()
136                 if self["config"].getCurrent()[0] == "Tuner":
137                         self.createSetup()
138
139         def createConfig(self):
140                         config.scan = ConfigSubsection()
141                         config.scan.sat = ConfigSubsection()
142                         config.scan.cab = ConfigSubsection()
143                         config.scan.ter = ConfigSubsection()
144
145                         config.scan.type = configElement_nonSave("config.scan.type", configSelection, 0, ("Single transponder", "Single satellite", "Multisat"))
146                         nimList = [ ]
147                         for nim in nimmanager.nimList():
148                                 nimList.append(nim[0])
149                         nimList.append("all")
150                         config.scan.nims = configElement_nonSave("config.scan.nims", configSelection, 0, nimList)
151
152                         # sat
153                         config.scan.sat.frequency = configElement_nonSave("config.scan.sat.frequency", configSequence, [11836], configsequencearg.get("INTEGER", (10000, 14000)))
154                         config.scan.sat.inversion = configElement_nonSave("config.scan.sat.inversion", configSelection, 2, ("on", "off", "auto"))
155                         config.scan.sat.symbolrate = configElement_nonSave("config.scan.sat.symbolrate", configSequence, [27500], configsequencearg.get("INTEGER", (1, 30000)))
156                         config.scan.sat.polarization = configElement_nonSave("config.scan.sat.polarization", configSelection, 0, ("horizontal", "vertical",  "circular left", "circular right"))
157                         config.scan.sat.fec = configElement_nonSave("config.scan.sat.fec", configSelection, 7, ("None", "1/2", "2/3", "3/4", "5/6", "7/8", "auto"))
158
159
160                         # cable
161                         config.scan.cab.frequency = configElement_nonSave("config.scan.cab.frequency", configSequence, [466], configsequencearg.get("INTEGER", (10000, 14000)))
162                         config.scan.cab.inversion = configElement_nonSave("config.scan.cab.inversion", configSelection, 0, ("auto", "off", "on"))
163                         config.scan.cab.modulation = configElement_nonSave("config.scan.cab.modulation", configSelection, 0, ("Auto", "16-QAM", "32-QAM", "64-QAM", "128-QAM", "256-QAM"))
164                         config.scan.cab.fec = configElement_nonSave("config.scan.cab.fec", configSelection, 0, ("Auto", "1/2", "2/3", "3/4", "4/5", "5/6", "7/8", "8/9"))
165                         config.scan.cab.symbolrate = configElement_nonSave("config.scan.cab.symbolrate", configSequence, [6900], configsequencearg.get("INTEGER", (1, 30000)))
166
167                         # terrestial
168                         config.scan.ter.frequency = configElement_nonSave("config.scan.ter.frequency", configSequence, [466], configsequencearg.get("INTEGER", (10000, 14000)))
169                         config.scan.ter.inversion = configElement_nonSave("config.scan.ter.inversion", configSelection, 0, ("auto", "off", "on"))
170                         config.scan.ter.bandwidth = configElement_nonSave("config.scan.ter.bandwidth", configSelection, 0, ("Auto", "6 MHz", "7MHz", "8MHz"))
171                         config.scan.ter.fechigh = configElement_nonSave("config.scan.ter.fechigh", configSelection, 0, ("Auto", "1/2", "2/3", "3/4", "4/5", "5/6", "7/8", "8/9"))
172                         config.scan.ter.feclow = configElement_nonSave("config.scan.ter.feclow", configSelection, 0, ("Auto", "1/2", "2/3", "3/4", "4/5", "5/6", "7/8", "8/9"))
173                         config.scan.ter.modulation = configElement_nonSave("config.scan.ter.modulation", configSelection, 0, ("Auto", "16-QAM", "32-QAM", "64-QAM", "128-QAM", "256-QAM"))
174                         config.scan.ter.transmission = configElement_nonSave("config.scan.ter.transmission", configSelection, 0, ("Auto", "2K", "8K"))
175                         config.scan.ter.guard = configElement_nonSave("config.scan.ter.guard", configSelection, 0, ("Auto", "1/4", "1/8", "1/16", "1/32"))
176                         config.scan.ter.hierarchy = configElement_nonSave("config.scan.ter.hierarchy", configSelection, 0, ("Auto", "1", "2", "4"))
177
178                         config.scan.scansat = {}
179                         for sat in nimmanager.satList:
180                                 #print sat[1]
181                                 config.scan.scansat[sat[1]] = configElement_nonSave("config.scan.scansat[" + str(sat[1]) + "]", configSelection, 0, ("yes", "no"))
182
183                         config.scan.satselection = []
184                         slotid = 0
185                         for slot in nimmanager.nimslots:
186                                 if (nimmanager.getNimType(slot.slotid) == nimmanager.nimType["DVB-S"]):
187                                         config.scan.satselection.append(configElement_nonSave("config.scan.satselection[" + str(slot.slotid) + "]", configSatlist, 0, self.satList[slot.slotid]))
188
189         def keyLeft(self):
190                 self["config"].handleKey(config.key["prevElement"])
191                 self.newConfig()
192
193         def keyRight(self):
194                 self["config"].handleKey(config.key["nextElement"])
195                 self.newConfig()
196
197         def keyNumberGlobal(self, number):
198                 print "You pressed number " + str(number)
199                 if (self["config"].getCurrent()[1].parent.enabled == True):
200                         self["config"].handleKey(config.key[str(number)])
201
202         def addSatTransponder(self, tlist, frequency, symbol_rate, polarisation, fec, inversion, orbital_position):
203                 print "Add Sat: frequ: " + str(frequency) + " symbol: " + str(symbol_rate) + " pol: " + str(polarisation) + " fec: " + str(fec) + " inversion: " + str(inversion)
204                 print "orbpos: " + str(orbital_position)
205                 parm = eDVBFrontendParametersSatellite()
206                 parm.frequency = frequency * 1000
207                 parm.symbol_rate = symbol_rate * 1000
208                 parm.polarisation = polarisation # eDVBFrontendParametersSatellite.Polarisation.Verti   
209                 parm.fec = fec                  # eDVBFrontendParametersSatellite.FEC.f3_4;
210                 #parm.fec = 6                                   # AUTO
211                 parm.inversion = inversion      #eDVBFrontendParametersSatellite.Inversion.Off;
212                 #parm.inversion = 2             #AUTO
213                 parm.orbital_position = int(orbital_position)
214                 tlist.append(parm)
215
216         # FIXME use correct parameters
217         def addCabTransponder(self, tlist, frequency, symbol_rate, polarisation, fec, inversion, orbital_position):
218                 print "Add Sat: frequ: " + str(frequency) + " symbol: " + str(symbol_rate) + " pol: " + str(polarisation) + " fec: " + str(fec) + " inversion: " + str(inversion)
219                 print "orbpos: " + str(orbital_position)
220                 parm = eDVBFrontendParametersCable()
221                 parm.frequency = frequency * 1000
222                 parm.symbol_rate = symbol_rate * 1000
223                 parm.polarisation = polarisation # eDVBFrontendParametersSatellite.Polarisation.Verti   
224                 parm.fec = fec                  # eDVBFrontendParametersSatellite.FEC.f3_4;
225                 #parm.fec = 6                                   # AUTO
226                 parm.inversion = inversion      #eDVBFrontendParametersSatellite.Inversion.Off;
227                 #parm.inversion = 2             #AUTO
228                 parm.orbital_position = int(orbital_position)
229                 tlist.append(parm)
230
231         # FIXME use correct parameters
232         def addTerTransponder(self, tlist, frequency, symbol_rate, polarisation, fec, inversion, orbital_position):
233                 print "Add Sat: frequ: " + str(frequency) + " symbol: " + str(symbol_rate) + " pol: " + str(polarisation) + " fec: " + str(fec) + " inversion: " + str(inversion)
234                 print "orbpos: " + str(orbital_position)
235                 parm = eDVBFrontendParametersTerrestrial()
236                 parm.frequency = frequency * 1000
237                 parm.symbol_rate = symbol_rate * 1000
238                 parm.polarisation = polarisation # eDVBFrontendParametersSatellite.Polarisation.Verti   
239                 parm.fec = fec                  # eDVBFrontendParametersSatellite.FEC.f3_4;
240                 #parm.fec = 6                                   # AUTO
241                 parm.inversion = inversion      #eDVBFrontendParametersSatellite.Inversion.Off;
242                 #parm.inversion = 2             #AUTO
243                 parm.orbital_position = int(orbital_position)
244                 tlist.append(parm)
245
246         def keyGo(self):
247                 tlist = []
248                 flags = 0
249                 if (config.scan.type.value == 0): # single transponder scan
250                         if (nimmanager.getNimType(config.scan.nims.value) == nimmanager.nimType["DVB-S"]):
251                                 self.addSatTransponder(tlist, config.scan.sat.frequency.value[0],
252                                                                                           config.scan.sat.symbolrate.value[0],
253                                                                                           config.scan.sat.polarization.value,
254                                                                                           config.scan.sat.fec.value,
255                                                                                           config.scan.sat.inversion.value,
256                                                                                           self.satList[config.scan.nims.value][config.scan.satselection[config.scan.nims.value].value][1])
257                         if (nimmanager.getNimType(config.scan.nims.value) == nimmanager.nimType["DVB-C"]):
258                                 self.addCabTransponder(tlist, config.scan.cab.frequency.value[0],
259                                                                                           config.scan.cab.symbolrate.value[0],
260                                                                                           config.scan.cab.polarization.value,
261                                                                                           config.scan.cab.fec.value,
262                                                                                           config.scan.cab.inversion.value,
263                                                                                           self.satList[config.scan.nims.value][config.scan.satselection[config.scan.nims.value].value][1])
264                         if (nimmanager.getNimType(config.scan.nims.value) == nimmanager.nimType["DVB-T"]):
265                                 self.addTerTransponder(tlist, config.scan.sat.frequency.value[0],
266                                                                                           config.scan.sat.symbolrate.value[0],
267                                                                                           config.scan.sat.polarization.value,
268                                                                                           config.scan.sat.fec.value,
269                                                                                           config.scan.sat.inversion.value,
270                                                                                           self.satList[config.scan.nims.value][config.scan.satselection[config.scan.nims.value].value][1])
271
272                 if (config.scan.type.value == 1): # single sat scan
273                         getInitialTransponderList(tlist, int(self.satList[config.scan.nims.value][config.scan.satselection[config.scan.nims.value].value][1]))
274                         flags |= eComponentScan.scanNetworkSearch
275
276                 if (config.scan.type.value == 2): # multi sat scan
277                         SatList = nimmanager.getSatListForNim(config.scan.nims.value)
278
279                         for x in self.list:
280                                 if x[1].parent.value == 0:
281                                         print "   " + str(x[1].parent.configPath)
282                                         getInitialTransponderList(tlist, x[1].parent.configPath)
283                         flags |= eComponentScan.scanNetworkSearch
284
285                 for x in self["config"].list:
286                         x[1].save()
287
288                 feid = config.scan.nims.value
289                 # flags |= eComponentScan.scanSearchBAT
290                 self.session.openWithCallback(self.keyCancel, ServiceScan, tlist, feid, flags)
291
292                 #self.close()
293
294         def keyCancel(self):
295                 for x in self["config"].list:
296                         x[1].cancel()
297                 self.close()
298
299 class ScanSimple(Screen):
300
301         def keyOK(self):
302                 print "start scan for sats:"
303                 tlist = [ ]
304                 for x in self.list:
305                         if x[1].parent.value == 0:
306                                 print "   " + str(x[1].parent.configPath)
307                                 getInitialTransponderList(tlist, x[1].parent.configPath)
308
309                 feid = 0 # FIXME
310                 self.session.openWithCallback(self.keyCancel, ServiceScan, tlist, feid, eComponentScan.scanNetworkSearch)
311
312         def keyCancel(self):
313                 self.close()
314
315         def keyLeft(self):
316                 self["config"].handleKey(config.key["prevElement"])
317
318         def keyRight(self):
319                 self["config"].handleKey(config.key["nextElement"])
320
321         def Satexists(self, tlist, pos):
322                 for x in tlist:
323                         if x == pos:
324                                 return 1
325                 return 0
326
327         def __init__(self, session):
328                 Screen.__init__(self, session)
329
330                 self["actions"] = ActionMap(["SetupActions"],
331                 {
332                         "ok": self.keyOK,
333                         "cancel": self.keyCancel,
334                         "left": self.keyLeft,
335                         "right": self.keyRight,
336                 }, -1)
337
338                 self.list = []
339                 tlist = []
340
341                 SatList = nimmanager.getConfiguredSats()
342
343                 for x in SatList:
344                         if self.Satexists(tlist, x) == 0:
345                                 tlist.append(x)
346                                 sat = configElement_nonSave(x, configSelection, 0, ("Enable", "Disable"))
347                                 self.list.append(getConfigListEntry(nimmanager.getSatDescription(x), sat))
348
349                 self["config"] = ConfigList(self.list)
350                 self["header"] = Label("Automatic Scan")
351                 self["footer"] = Label("Press OK to scan")