lib/python/Plugins/SystemPlugins/Satfinder/plugin.py

   1 from enigma import eTimer, eDVBSatelliteEquipmentControl, eDVBResourceManager, eDVBDiseqcCommand, eDVBResourceManagerPtr, iDVBChannelPtr, iDVBFrontendPtr, iDVBFrontend, eDVBFrontendParametersSatellite, eDVBFrontendParameters
   2 from Screens.Screen import Screen
   3 from Screens.ScanSetup import ScanSetup
   4 from Screens.MessageBox import MessageBox
   5 from Plugins.Plugin import PluginDescriptor
   6
   7 from Components.Label import Label
   8 from Components.ConfigList import ConfigList
   9 from Components.TunerInfo import TunerInfo
  10 from Components.ActionMap import ActionMap
  11 from Components.NimManager import nimmanager
  12 from Components.MenuList import MenuList
  13 from Components.config import config, ConfigSubsection, configElement_nonSave, configNothing, getConfigListEntry, configSelection, currentConfigSelectionElement, configSatlist
  14
  15 class Tuner:
  16         def __init__(self, frontend):
  17                 self.frontend = frontend
  18
  19         def tune(self, transponder):
  20                 print "tuning to transponder with data", transponder
  21                 parm = eDVBFrontendParametersSatellite()
  22                 parm.frequency = transponder[0] * 1000
  23                 parm.symbol_rate = transponder[1] * 1000
  24                 parm.polarisation = transponder[2]
  25                 parm.fec = transponder[3]
  26                 parm.inversion = transponder[4]
  27                 parm.orbital_position = transponder[5]
  28                 parm.system = 0  # FIXMEE !! HARDCODED DVB-S (add support for DVB-S2)
  29                 parm.modulation = 1 # FIXMEE !! HARDCODED QPSK
  30                 feparm = eDVBFrontendParameters()
  31                 feparm.setDVBS(parm, True)
  32                 self.lastparm = feparm
  33                 self.frontend.tune(feparm)
  34
  35         def retune(self):
  36                 self.frontend.tune(self.lastparm)
  37
  38         def getTransponderData(self):
  39                 return self.frontend.readTransponderData(True)
  40
  41 class Diseqc:
  42         def __init__(self, feid = 0):
  43                 self.ready = False
  44                 self.feid = feid
  45                 res_mgr = eDVBResourceManagerPtr()
  46                 if eDVBResourceManager.getInstance(res_mgr) == 0:
  47                         self.raw_channel = iDVBChannelPtr()
  48                         if res_mgr.allocateRawChannel(self.raw_channel, self.feid) == 0:
  49                                 self.frontend = iDVBFrontendPtr()
  50                                 if self.raw_channel.getFrontend(self.frontend) == 0:
  51                                         self.ready = True
  52                                 else:
  53                                         print "getFrontend failed"
  54                         else:
  55                                 print "getRawChannel failed"
  56                 else:
  57                         print "getResourceManager instance failed"
  58
  59         def getFrontend(self):
  60                 return self.frontend
  61
  62         def command(self, what, param = 0):
  63                 if self.ready:
  64                         cmd = eDVBDiseqcCommand()
  65                         if what == "moveWest":
  66                                 string = 'e03169' + ("%02x" % param)
  67                         elif what == "moveEast":
  68                                 string = 'e03168' + ("%02x" % param)
  69                         elif what == "moveTo":
  70                                 string = 'e0316b' + ("%02x" % param)
  71                         elif what == "store":
  72                                 string = 'e0316a' + ("%02x" % param)
  73                         elif what == "limitOn":
  74                                 string = 'e0316a00'
  75                         elif what == "limitOff":
  76                                 string = 'e03163'
  77                         elif what == "limitEast":
  78                                 string = 'e03166'
  79                         elif what == "limitWest":
  80                                 string = 'e03167'
  81                         else:
  82                                 string = 'e03160' #positioner stop
  83                         print "diseqc command:",
  84                         print string
  85
  86                         cmd.setCommandString(string)
  87                         self.frontend.sendDiseqc(cmd)
  88
  89 class Satfinder(ScanSetup):
  90         skin = """
  91                 <screen position="90,100" size="520,400" title="Tune">
  92                         <widget name="config" position="20,10" size="460,240" scrollbarMode="showOnDemand" />
  93                         <widget name="introduction" position="20,360" zPosition="-10" size="350,30" font="Regular;23" />
  94                         <widget name="snr" text="SNR:" position="0,245" size="60,22" font="Regular;21" />
  95                         <widget name="agc" text="AGC:" position="0,270" size="60,22" font="Regular;21" />
  96                         <widget name="ber" text="BER:" position="0,295" size="60,22" font="Regular;21" />
  97                         <widget name="lock" text="Lock:" position="0,320" size="60,22" font="Regular;21" />
  98                         <widget name="snr_percentage" position="220,245" size="60,22" font="Regular;21" />
  99                         <widget name="agc_percentage" position="220,270" size="60,22" font="Regular;21" />
 100                         <widget name="ber_value" position="220,295" size="60,22" font="Regular;21" />
 101                         <widget name="lock_state" position="60,320" size="150,22" font="Regular;21" />
 102                         <widget name="snr_bar" position="60,245" size="150,22" />
 103                         <widget name="agc_bar" position="60,270" size="150,22" />
 104                         <widget name="ber_bar" position="60,295" size="150,22" />
 105                 </screen>"""
 106
 107         def __init__(self, session, feid):
 108                 self.initcomplete = False
 109                 self.feid = feid
 110
 111                 ScanSetup.__init__(self, session)
 112                 self.session.nav.stopService()
 113
 114                 self.diseqc = Diseqc(self.feid)
 115                 self.tuner = Tuner(self.diseqc.getFrontend())
 116
 117                 self["snr"] = Label()
 118                 self["agc"] = Label()
 119                 self["ber"] = Label()
 120                 self["lock"] = Label()
 121                 self["snr_percentage"] = TunerInfo(TunerInfo.SNR_PERCENTAGE, frontendfkt = self.diseqc.getFrontend)
 122                 self["agc_percentage"] = TunerInfo(TunerInfo.AGC_PERCENTAGE, frontendfkt = self.diseqc.getFrontend)
 123                 self["ber_value"] = TunerInfo(TunerInfo.BER_VALUE, frontendfkt = self.diseqc.getFrontend)
 124                 self["snr_bar"] = TunerInfo(TunerInfo.SNR_BAR, frontendfkt = self.diseqc.getFrontend)
 125                 self["agc_bar"] = TunerInfo(TunerInfo.AGC_BAR, frontendfkt = self.diseqc.getFrontend)
 126                 self["ber_bar"] = TunerInfo(TunerInfo.BER_BAR, frontendfkt = self.diseqc.getFrontend)
 127                 self["lock_state"] = TunerInfo(TunerInfo.LOCK_STATE, frontendfkt = self.diseqc.getFrontend)
 128
 129                 self["introduction"].setText("")
 130
 131                 self.statusTimer = eTimer()
 132                 self.statusTimer.timeout.get().append(self.updateStatus)
 133                 self.statusTimer.start(50, False)
 134
 135                 self.initcomplete = True
 136
 137         def updateStatus(self):
 138                 self["snr_percentage"].update()
 139                 self["agc_percentage"].update()
 140                 self["ber_value"].update()
 141                 self["snr_bar"].update()
 142                 self["agc_bar"].update()
 143                 self["ber_bar"].update()
 144                 self["lock_state"].update()
 145
 146         def createSetup(self):
 147                 self.typeOfTuningEntry = None
 148                 self.satEntry = None
 149
 150                 self.list = []
 151                 self.typeOfTuningEntry = getConfigListEntry(_('Tune'), config.tuning.type)
 152                 self.list.append(self.typeOfTuningEntry)
 153                 self.satEntry = getConfigListEntry(_('Satellite'), config.tuning.sat)
 154                 self.list.append(self.satEntry)
 155                 if currentConfigSelectionElement(config.tuning.type) == "manual_transponder":
 156                         self.list.append(getConfigListEntry(_('Frequency'), config.scan.sat.frequency))
 157                         self.list.append(getConfigListEntry(_('Inversion'), config.scan.sat.inversion))
 158                         self.list.append(getConfigListEntry(_('Symbol Rate'), config.scan.sat.symbolrate))
 159                         self.list.append(getConfigListEntry(_("Polarity"), config.scan.sat.polarization))
 160                         self.list.append(getConfigListEntry(_("FEC"), config.scan.sat.fec))
 161                 elif currentConfigSelectionElement(config.tuning.type) == "predefined_transponder":
 162                         self.list.append(getConfigListEntry(_("Transponder"), config.tuning.transponder))
 163                 self["config"].list = self.list
 164                 self["config"].l.setList(self.list)
 165
 166         def newConfig(self):
 167                 if self["config"].getCurrent() == self.typeOfTuningEntry:
 168                         self.createSetup()
 169                 elif self["config"].getCurrent() == self.satEntry:
 170                         self.updateSats()
 171                         self.createSetup()
 172
 173         def retune(self, configElement):
 174                 returnvalue = (0, 0, 0, 0, 0, 0, 0)
 175                 satpos = config.tuning.sat.vals[config.tuning.sat.value][1]
 176                 if currentConfigSelectionElement(config.tuning.type) == "manual_transponder":
 177                         returnvalue = (config.scan.sat.frequency.value[0], config.scan.sat.symbolrate.value[0], config.scan.sat.polarization.value, config.scan.sat.fec.value, config.scan.sat.inversion.value, satpos)
 178                 elif currentConfigSelectionElement(config.tuning.type) == "predefined_transponder":
 179                         transponder = nimmanager.getTransponders(config.tuning.sat.vals[config.tuning.sat.value][1])[config.tuning.transponder.value]
 180                         returnvalue = (int(transponder[1] / 1000), int(transponder[2] / 1000), transponder[3], transponder[4], 2, config.tuning.sat.vals[config.tuning.sat.value][1], satpos)
 181                 self.tune(returnvalue)
 182
 183         def createConfig(self, foo):
 184                 config.tuning = ConfigSubsection()
 185
 186                 config.tuning.type = configElement_nonSave("config.tuning.type", configSelection, 0, (("manual_transponder", _("Manual transponder")), ("predefined_transponder", _("Predefined satellite"))))
 187                 config.tuning.sat = configElement_nonSave("config.tuning.sat", configSatlist, 192, nimmanager.getRotorSatListForNim(self.feid))
 188                 ScanSetup.createConfig(self, None)
 189
 190                 self.updateSats()
 191
 192                 config.tuning.type.addNotifier(self.retune)
 193                 config.tuning.sat.addNotifier(self.retune)
 194                 config.scan.sat.frequency.addNotifier(self.retune)
 195                 config.scan.sat.inversion.addNotifier(self.retune)
 196                 config.scan.sat.symbolrate.addNotifier(self.retune)
 197                 config.scan.sat.polarization.addNotifier(self.retune)
 198                 config.scan.sat.fec.addNotifier(self.retune)
 199                 config.tuning.transponder.addNotifier(self.retune)
 200
 201         def updateSats(self):
 202                 satnum = config.tuning.sat.value
 203                 satlist = config.tuning.sat.vals
 204                 if len(satlist):
 205                         transponderlist = nimmanager.getTransponders(satlist[satnum][1])
 206                         list = []
 207                         for x in transponderlist:
 208                                 if x[3] == 0:
 209                                         pol = "H"
 210                                 elif x[3] == 1:
 211                                         pol = "V"
 212                                 elif x[3] == 2:
 213                                         pol = "CL"
 214                                 elif x[3] == 3:
 215                                         pol = "CR"
 216                                 if x[4] == 0:
 217                                         fec = "FEC_AUTO"
 218                                 elif x[4] == 1:
 219                                         fec = "FEC_1_2"
 220                                 elif x[4] == 2:
 221                                         fec = "FEC_2_3"
 222                                 elif x[4] == 3:
 223                                         fec = "FEC_3_4"
 224                                 elif x[4] == 4:
 225                                         fec = "FEC_5_6"
 226                                 elif x[4] == 5:
 227                                         fec = "FEC_7_8"
 228                                 elif x[4] == 5:
 229                                         fec = "FEC_8_9"
 230                                 elif x[4] == 6:
 231                                         fec = "FEC_None"
 232                                 list.append(str(x[1]) + "," + str(x[2]) + "," + pol + "," + fec)
 233                         config.tuning.transponder = configElement_nonSave("config.tuning.transponder", configSelection, 0, list)
 234
 235         def keyGo(self):
 236                 returnvalue = (0, 0, 0, 0, 0, 0, 0)
 237                 satpos = config.tuning.sat.vals[config.tuning.sat.value][1]
 238                 if currentConfigSelectionElement(config.tuning.type) == "manual_transponder":
 239                         returnvalue = (config.scan.sat.frequency.value[0], config.scan.sat.symbolrate.value[0], config.scan.sat.polarization.value, config.scan.sat.fec.value, config.scan.sat.inversion.value, satpos)
 240                 elif currentConfigSelectionElement(config.tuning.type) == "predefined_transponder":
 241                         transponder = nimmanager.getTransponders(config.tuning.sat.vals[config.tuning.sat.value][1])[config.tuning.transponder.value]
 242                         returnvalue = (int(transponder[1] / 1000), int(transponder[2] / 1000), transponder[3], transponder[4], 2, config.tuning.sat.vals[config.tuning.sat.value][1], satpos)
 243                 self.close(returnvalue)
 244
 245         def keyCancel(self):
 246                 self.close(None)
 247
 248         def tune(self, transponder):
 249                 if self.initcomplete:
 250                         if transponder is not None:
 251                                 self.tuner.tune(transponder)
 252
 253 class NimSelection(Screen):
 254         skin = """
 255                 <screen position="140,165" size="400,100" title="select Slot">
 256                         <widget name="nimlist" position="20,10" size="360,75" />
 257                 </screen>"""
 258
 259         def __init__(self, session):
 260                 Screen.__init__(self, session)
 261
 262                 nimlist = nimmanager.getNimListOfType(nimmanager.nimType["DVB-S"])
 263                 nimMenuList = []
 264                 for x in nimlist:
 265                         nimMenuList.append((_("NIM ") + (["A", "B", "C", "D"][x]) + ": " + nimmanager.getNimName(x) + " (" + nimmanager.getNimTypeName(x) + ")", x))
 266
 267                 self["nimlist"] = MenuList(nimMenuList)
 268
 269                 self["actions"] = ActionMap(["OkCancelActions"],
 270                 {
 271                         "ok": self.okbuttonClick ,
 272                         "cancel": self.close
 273                 }, -1)
 274
 275         def okbuttonClick(self):
 276                 selection = self["nimlist"].getCurrent()
 277                 self.session.open(Satfinder, selection[1])
 278
 279 def SatfinderMain(session, **kwargs):
 280         nimList = nimmanager.getNimListOfType(nimmanager.nimType["DVB-S"])
 281         if len(nimList) == 0:
 282                 session.open(MessageBox, _("No positioner capable frontend found."), MessageBox.TYPE_ERROR)
 283         else:
 284                 if session.nav.RecordTimer.isRecording():
 285                         session.open(MessageBox, _("A recording is currently running. Please stop the recording before trying to configure the positioner."), MessageBox.TYPE_ERROR)
 286                 else:
 287                         if len(nimList) == 1:
 288                                 session.open(Satfinder, nimList[0])
 289                         elif len(nimList) > 1:
 290                                 session.open(NimSelection)
 291                         else:
 292                                 session.open(MessageBox, _("No tuner is configured for use with a diseqc positioner!"), MessageBox.TYPE_ERROR)
 293
 294 def Plugins(**kwargs):
 295         return PluginDescriptor(name="Satfinder", description="Helps setting up your dish", where = PluginDescriptor.WHERE_PLUGINMENU, fnc=SatfinderMain)