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