code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Plugins/SystemPlugins/PositionerSetup/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 ConfigSatlist, ConfigNothing, ConfigSelection, ConfigSubsection, KEY_LEFT, KEY_RIGHT, getConfigListEntry
  14
  15 from time import sleep
  16
  17 class PositionerSetup(Screen):
  18         skin = """
  19                 <screen position="100,100" size="560,400" title="Positioner setup..." >
  20                         <widget name="list" position="100,0" size="350,155" />
  21
  22                         <widget name="red" position="0,155" size="140,80" backgroundColor="red" halign="center" valign="center" font="Regular;21" />
  23                         <widget name="green" position="140,155" size="140,80" backgroundColor="green" halign="center" valign="center" font="Regular;21" />
  24                         <widget name="yellow" position="280,155" size="140,80" backgroundColor="yellow" halign="center" valign="center" font="Regular;21" />
  25                         <widget name="blue" position="420,155" size="140,80" backgroundColor="blue" halign="center" valign="center" font="Regular;21" />
  26
  27                         <widget name="snr" text="SNR:" position="0,245" size="60,22" font="Regular;21" />
  28                         <widget name="agc" text="AGC:" position="0,270" size="60,22" font="Regular;21" />
  29                         <widget name="ber" text="BER:" position="0,295" size="60,22" font="Regular;21" />
  30                         <widget name="lock" text="Lock:" position="0,320" size="60,22" font="Regular;21" />
  31                         <widget name="snr_percentage" position="220,245" size="60,22" font="Regular;21" />
  32                         <widget name="agc_percentage" position="220,270" size="60,22" font="Regular;21" />
  33                         <widget name="ber_value" position="220,295" size="60,22" font="Regular;21" />
  34                         <widget name="lock_state" position="60,320" size="150,22" font="Regular;21" />
  35                         <widget name="snr_bar" position="60,245" size="150,22" />
  36                         <widget name="agc_bar" position="60,270" size="150,22" />
  37                         <widget name="ber_bar" position="60,295" size="150,22" />
  38
  39                         <widget name="frequency" text="Frequency:" position="300,245" size="120,22" font="Regular;21" />
  40                         <widget name="symbolrate" text="Symbolrate:" position="300,270" size="120,22" font="Regular;21" />
  41                         <widget name="fec" text="FEC:" position="300,295" size="120,22" font="Regular;21" />
  42                         <widget name="frequency_value" position="420,245" size="120,22" font="Regular;21" />
  43                         <widget name="symbolrate_value" position="420,270" size="120,22" font="Regular;21" />
  44                         <widget name="fec_value" position="420,295" size="120,22" font="Regular;21" />
  45                 </screen>"""
  46         def __init__(self, session, feid):
  47                 self.skin = PositionerSetup.skin
  48                 Screen.__init__(self, session)
  49
  50                 self.session.nav.stopService()
  51
  52                 self.feid = feid
  53
  54                 self.diseqc = Diseqc(self.feid)
  55                 self.tuner = Tuner(self.diseqc.getFrontend())
  56                 self.tuner.tune((0,0,0,0,0,0))
  57
  58                 #self.session.nav.stopService()
  59
  60                 self.createConfig()
  61
  62                 self.isMoving = False
  63                 self.stopOnLock = False
  64
  65                 self.red = Label("")
  66                 self["red"] = self.red
  67                 self.green = Label("")
  68                 self["green"] = self.green
  69                 self.yellow = Label("")
  70                 self["yellow"] = self.yellow
  71                 self.blue = Label("")
  72                 self["blue"] = self.blue
  73
  74                 self.list = []
  75                 self["list"] = ConfigList(self.list)
  76                 self.createSetup()
  77
  78                 self["snr"] = Label()
  79                 self["agc"] = Label()
  80                 self["ber"] = Label()
  81                 self["lock"] = Label()
  82                 self["snr_percentage"] = TunerInfo(TunerInfo.SNR_PERCENTAGE, frontendfkt = self.diseqc.getFrontend)
  83                 self["agc_percentage"] = TunerInfo(TunerInfo.AGC_PERCENTAGE, frontendfkt = self.diseqc.getFrontend)
  84                 self["ber_value"] = TunerInfo(TunerInfo.BER_VALUE, frontendfkt = self.diseqc.getFrontend)
  85                 self["snr_bar"] = TunerInfo(TunerInfo.SNR_BAR, frontendfkt = self.diseqc.getFrontend)
  86                 self["agc_bar"] = TunerInfo(TunerInfo.AGC_BAR, frontendfkt = self.diseqc.getFrontend)
  87                 self["ber_bar"] = TunerInfo(TunerInfo.BER_BAR, frontendfkt = self.diseqc.getFrontend)
  88                 self["lock_state"] = TunerInfo(TunerInfo.LOCK_STATE, frontendfkt = self.diseqc.getFrontend)
  89
  90                 self["frequency"] = Label()
  91                 self["symbolrate"] = Label()
  92                 self["fec"] = Label()
  93
  94                 self["frequency_value"] = Label("")
  95                 self["symbolrate_value"] = Label("")
  96                 self["fec_value"] = Label("")
  97
  98                 self["actions"] = ActionMap(["DirectionActions", "OkCancelActions", "ColorActions"],
  99                 {
 100                         "ok": self.go,
 101                         "cancel": self.close,
 102                         "up": self.up,
 103                         "down": self.down,
 104                         "left": self.left,
 105                         "right": self.right,
 106                         "red": self.redKey,
 107                         "green": self.greenKey,
 108                         "yellow": self.yellowKey,
 109                         "blue": self.blueKey,
 110                 }, -1)
 111
 112                 self.updateColors("tune")
 113
 114                 self.statusTimer = eTimer()
 115                 self.statusTimer.timeout.get().append(self.updateStatus)
 116                 self.statusTimer.start(50, False)
 117
 118         def createConfig(self):
 119                 self.positioner_tune = ConfigNothing()
 120                 self.positioner_move = ConfigNothing()
 121                 self.positioner_finemove = ConfigNothing()
 122                 self.positioner_limits = ConfigNothing()
 123                 self.positioner_goto0 = ConfigNothing()
 124                 storepos = []
 125                 for x in range(1,255):
 126                         storepos.append(str(x))
 127                 self.positioner_storage = ConfigSelection(choices = storepos)
 128
 129         def createSetup(self):
 130                 self.list.append((_("Tune"), self.positioner_tune, "tune"))
 131                 self.list.append((_("Positioner movement"), self.positioner_move, "move"))
 132                 self.list.append((_("Positioner fine movement"), self.positioner_finemove, "finemove"))
 133                 self.list.append((_("Set limits"), self.positioner_limits, "limits"))
 134                 self.list.append((_("Positioner storage"), self.positioner_storage, "storage"))
 135                 self.list.append((_("Goto 0"), self.positioner_goto0, "goto0"))
 136                 self["list"].l.setList(self.list)
 137
 138         def go(self):
 139                 pass
 140
 141         def getCurrentConfigPath(self):
 142                 return self["list"].getCurrent()[2]
 143
 144         def up(self):
 145                 if not self.isMoving:
 146                         self["list"].instance.moveSelection(self["list"].instance.moveUp)
 147                         self.updateColors(self.getCurrentConfigPath())
 148
 149         def down(self):
 150                 if not self.isMoving:
 151                         self["list"].instance.moveSelection(self["list"].instance.moveDown)
 152                         self.updateColors(self.getCurrentConfigPath())
 153
 154         def left(self):
 155                 self["list"].handleKey(KEY_LEFT)
 156
 157         def right(self):
 158                 self["list"].handleKey(KEY_RIGHT)
 159
 160         def updateColors(self, entry):
 161                 if entry == "tune":
 162                         self.red.setText(_("Tune"))
 163                         self.green.setText("")
 164                         self.yellow.setText("")
 165                         self.blue.setText("")
 166                 elif entry == "move":
 167                         if self.isMoving:
 168                                 self.red.setText(_("Stop"))
 169                                 self.green.setText(_("Stop"))
 170                                 self.yellow.setText(_("Stop"))
 171                                 self.blue.setText(_("Stop"))
 172                         else:
 173                                 self.red.setText(_("Move west"))
 174                                 self.green.setText(_("Search west"))
 175                                 self.yellow.setText(_("Search east"))
 176                                 self.blue.setText(_("Move east"))
 177                 elif entry == "finemove":
 178                         self.red.setText("")
 179                         self.green.setText(_("Step west"))
 180                         self.yellow.setText(_("Step east"))
 181                         self.blue.setText("")
 182                 elif entry == "limits":
 183                         self.red.setText(_("Limits off"))
 184                         self.green.setText(_("Limit west"))
 185                         self.yellow.setText(_("Limit east"))
 186                         self.blue.setText(_("Limits on"))
 187                 elif entry == "storage":
 188                         self.red.setText("")
 189                         self.green.setText(_("Store position"))
 190                         self.yellow.setText(_("Goto position"))
 191                         self.blue.setText("")
 192                 elif entry == "goto0":
 193                         self.red.setText(_("Goto 0"))
 194                         self.green.setText("")
 195                         self.yellow.setText("")
 196                         self.blue.setText("")
 197                 else:
 198                         self.red.setText("")
 199                         self.green.setText("")
 200                         self.yellow.setText("")
 201                         self.blue.setText("")
 202
 203         def redKey(self):
 204                 entry = self.getCurrentConfigPath()
 205                 if entry == "move":
 206                         if self.isMoving:
 207                                 self.diseqccommand("stop")
 208                                 self.isMoving = False
 209                                 self.stopOnLock = False
 210                         else:
 211                                 self.diseqccommand("moveWest", 0)
 212                                 self.isMoving = True
 213                         self.updateColors("move")
 214                 elif entry == "limits":
 215                         self.diseqccommand("limitOff")
 216                 elif entry == "tune":
 217                         self.session.openWithCallback(self.tune, TunerScreen, self.feid)
 218                 elif entry == "goto0":
 219                         print "move to position 0"
 220                         self.diseqccommand("moveTo", 0)
 221
 222         def greenKey(self):
 223                 entry = self.getCurrentConfigPath()
 224                 if entry == "move":
 225                         if self.isMoving:
 226                                 self.diseqccommand("stop")
 227                                 self.isMoving = False
 228                                 self.stopOnLock = False
 229                         else:
 230                                 self.isMoving = True
 231                                 self.stopOnLock = True
 232                                 self.diseqccommand("moveWest", 0)
 233                         self.updateColors("move")
 234                 elif entry == "finemove":
 235                         print "stepping west"
 236                         self.diseqccommand("moveWest", 0xFF) # one step
 237                 elif entry == "storage":
 238                         print "store at position", int(self.positioner_storage.value)
 239                         self.diseqccommand("store", int(self.positioner_storage.value))
 240                 elif entry == "limits":
 241                         self.diseqccommand("limitWest")
 242
 243         def yellowKey(self):
 244                 entry = self.getCurrentConfigPath()
 245                 if entry == "move":
 246                         if self.isMoving:
 247                                 self.diseqccommand("stop")
 248                                 self.isMoving = False
 249                                 self.stopOnLock = False
 250                         else:
 251                                 self.isMoving = True
 252                                 self.stopOnLock = True
 253                                 self.diseqccommand("moveEast", 0)
 254                         self.updateColors("move")
 255                 elif entry == "finemove":
 256                         print "stepping east"
 257                         self.diseqccommand("moveEast", 0xFF) # one step
 258                 elif entry == "storage":
 259                         print "move to position", int(self.positioner_storage.value)
 260                         self.diseqccommand("moveTo", int(self.positioner_storage.value))
 261                 elif entry == "limits":
 262                         self.diseqccommand("limitEast")
 263 #
 264         def blueKey(self):
 265                 entry = self.getCurrentConfigPath()
 266                 if entry == "move":
 267                         if self.isMoving:
 268                                 self.diseqccommand("stop")
 269                                 self.isMoving = False
 270                                 self.stopOnLock = False
 271                         else:
 272                                 self.diseqccommand("moveEast", 0)
 273                                 self.isMoving = True
 274                         self.updateColors("move")
 275                         print "moving east"
 276                 elif entry == "limits":
 277                         self.diseqccommand("limitOn")
 278
 279         def diseqccommand(self, cmd, param = 0):
 280                 self.diseqc.command(cmd, param)
 281                 self.tuner.retune()
 282
 283         def updateStatus(self):
 284                 self["snr_percentage"].update()
 285                 self["agc_percentage"].update()
 286                 self["ber_value"].update()
 287                 self["snr_bar"].update()
 288                 self["agc_bar"].update()
 289                 self["ber_bar"].update()
 290                 self["lock_state"].update()
 291                 transponderdata = self.tuner.getTransponderData()
 292                 self["frequency_value"].setText(str(transponderdata["frequency"]))
 293                 self["symbolrate_value"].setText(str(transponderdata["symbol_rate"]))
 294                 self["fec_value"].setText(str(transponderdata["fec_inner"]))
 295                 if transponderdata["tuner_locked"] == 1 and self.isMoving and self.stopOnLock:
 296                         self.diseqccommand("stop")
 297                         self.isMoving = False
 298                         self.stopOnLock = False
 299                         self.updateColors(self.getCurrentConfigPath())
 300
 301         def tune(self, transponder):
 302                 if transponder is not None:
 303                         self.tuner.tune(transponder)
 304
 305 class Diseqc:
 306         def __init__(self, feid = 0):
 307                 self.ready = False
 308                 self.feid = feid
 309                 res_mgr = eDVBResourceManagerPtr()
 310                 if eDVBResourceManager.getInstance(res_mgr) == 0:
 311                         self.raw_channel = iDVBChannelPtr()
 312                         if res_mgr.allocateRawChannel(self.raw_channel, self.feid) == 0:
 313                                 self.frontend = iDVBFrontendPtr()
 314                                 if self.raw_channel.getFrontend(self.frontend) == 0:
 315                                         self.ready = True
 316                                 else:
 317                                         print "getFrontend failed"
 318                         else:
 319                                 print "getRawChannel failed"
 320                 else:
 321                         print "getResourceManager instance failed"
 322
 323         def getFrontend(self):
 324                 return self.frontend
 325
 326         def command(self, what, param = 0):
 327                 if self.ready:
 328                         cmd = eDVBDiseqcCommand()
 329                         if what == "moveWest":
 330                                 string = 'e03169' + ("%02x" % param)
 331                         elif what == "moveEast":
 332                                 string = 'e03168' + ("%02x" % param)
 333                         elif what == "moveTo":
 334                                 string = 'e0316b' + ("%02x" % param)
 335                         elif what == "store":
 336                                 string = 'e0316a' + ("%02x" % param)
 337                         elif what == "limitOn":
 338                                 string = 'e0316a00'
 339                         elif what == "limitOff":
 340                                 string = 'e03163'
 341                         elif what == "limitEast":
 342                                 string = 'e03166'
 343                         elif what == "limitWest":
 344                                 string = 'e03167'
 345                         else:
 346                                 string = 'e03160' #positioner stop
 347
 348                         print "diseqc command:",
 349                         print string
 350                         cmd.setCommandString(string)
 351                         self.frontend.sendDiseqc(cmd)
 352                         if string == 'e03160': #positioner stop
 353                                 sleep(0.05)
 354                                 self.frontend.sendDiseqc(cmd) # send 2nd time
 355
 356 class Tuner:
 357         def __init__(self, frontend):
 358                 self.frontend = frontend
 359
 360         def tune(self, transponder):
 361                 print "tuning to transponder with data", transponder
 362                 parm = eDVBFrontendParametersSatellite()
 363                 parm.frequency = transponder[0] * 1000
 364                 parm.symbol_rate = transponder[1] * 1000
 365                 parm.polarisation = transponder[2]
 366                 parm.fec = transponder[3]
 367                 parm.inversion = transponder[4]
 368                 parm.orbital_position = transponder[5]
 369                 parm.system = 0  # FIXMEE !! HARDCODED DVB-S (add support for DVB-S2)
 370                 parm.modulation = 1 # FIXMEE !! HARDCODED QPSK
 371                 feparm = eDVBFrontendParameters()
 372                 feparm.setDVBS(parm, True)
 373                 self.lastparm = feparm
 374                 self.frontend.tune(feparm)
 375
 376         def retune(self):
 377                 self.frontend.tune(self.lastparm)
 378
 379         def getTransponderData(self):
 380                 return self.frontend.readTransponderData(True)
 381
 382 tuning = None
 383
 384 class TunerScreen(ScanSetup):
 385         skin = """
 386                 <screen position="90,100" size="520,400" title="Tune">
 387                         <widget name="config" position="20,10" size="460,350" scrollbarMode="showOnDemand" />
 388                         <widget name="introduction" position="20,360" size="350,30" font="Regular;23" />
 389                 </screen>"""
 390
 391         def __init__(self, session, feid):
 392                 self.feid = feid
 393                 ScanSetup.__init__(self, session)
 394
 395                 self["introduction"].setText("")
 396
 397         def createSetup(self):
 398                 self.typeOfTuningEntry = None
 399                 self.satEntry = None
 400
 401                 self.list = []
 402                 self.typeOfTuningEntry = getConfigListEntry(_('Tune'), tuning.type)
 403                 self.list.append(self.typeOfTuningEntry)
 404                 self.satEntry = getConfigListEntry(_('Satellite'), tuning.sat)
 405                 self.list.append(self.satEntry)
 406                 if tuning.type.value == "manual_transponder":
 407                         self.list.append(getConfigListEntry(_('Frequency'), self.scan_sat.frequency))
 408                         self.list.append(getConfigListEntry(_('Inversion'), self.scan_sat.inversion))
 409                         self.list.append(getConfigListEntry(_('Symbol Rate'), self.scan_sat.symbolrate))
 410                         self.list.append(getConfigListEntry(_("Polarity"), self.scan_sat.polarization))
 411                         self.list.append(getConfigListEntry(_("FEC"), self.scan_sat.fec))
 412                 elif tuning.type.value == "predefined_transponder":
 413                         self.list.append(getConfigListEntry(_("Transponder"), tuning.transponder))
 414                 self["config"].list = self.list
 415                 self["config"].l.setList(self.list)
 416
 417         def newConfig(self):
 418                 if self["config"].getCurrent() == self.typeOfTuningEntry:
 419                         self.createSetup()
 420                 elif self["config"].getCurrent() == self.satEntry:
 421                         self.createSetup()
 422
 423         def createConfig(self, foo):
 424                 if not tuning:
 425                         global tuning
 426                         tuning = ConfigSubsection()
 427                         tuning.type = ConfigSelection(
 428                                 default = "manual_transponder",
 429                                 choices = { "manual_transponder" : _("Manual transponder"),
 430                                                         "predefined_transponder" : _("Predefined transponder") } )
 431                         tuning.sat = ConfigSatlist(list=nimmanager.getRotorSatListForNim(self.feid))
 432                         tuning.sat.addNotifier(self.tuningSatChanged)
 433                         self.updateTransponders()
 434                         TunerScreenConfigCreated = True
 435                 ScanSetup.createConfig(self, None)
 436
 437         def tuningSatChanged(self, *parm):
 438                 self.updateTransponders()
 439
 440         def updateTransponders(self):
 441                 if len(tuning.sat.choices):
 442                         transponderlist = nimmanager.getTransponders(int(tuning.sat.value))
 443                         tps = []
 444                         cnt=0
 445                         for x in transponderlist:
 446                                 if x[3] == 0:
 447                                         pol = "H"
 448                                 elif x[3] == 1:
 449                                         pol = "V"
 450                                 elif x[3] == 2:
 451                                         pol = "CL"
 452                                 elif x[3] == 3:
 453                                         pol = "CR"
 454                                 if x[4] == 0:
 455                                         fec = "FEC_AUTO"
 456                                 elif x[4] == 1:
 457                                         fec = "FEC_1_2"
 458                                 elif x[4] == 2:
 459                                         fec = "FEC_2_3"
 460                                 elif x[4] == 3:
 461                                         fec = "FEC_3_4"
 462                                 elif x[4] == 4:
 463                                         fec = "FEC_5_6"
 464                                 elif x[4] == 5:
 465                                         fec = "FEC_7_8"
 466                                 elif x[4] == 5:
 467                                         fec = "FEC_8_9"
 468                                 elif x[4] == 6:
 469                                         fec = "FEC_None"
 470                                 tps.append(str(x[1]) + "," + str(x[2]) + "," + pol + "," + fec)
 471                         tuning.transponder = ConfigSelection(choices=tps)
 472
 473         def keyGo(self):
 474                 returnvalue = (0, 0, 0, 0, 0, 0)
 475                 satpos = int(tuning.sat.value)
 476                 if tuning.type.value == "manual_transponder":
 477                         returnvalue = (
 478                                 self.scan_sat.frequency.value,
 479                                 self.scan_sat.symbolrate.value,
 480                                 self.scan_sat.polarization.index,
 481                                 self.scan_sat.fec.index,
 482                                 self.scan_sat.inversion.index,
 483                                 satpos)
 484                 elif tuning.type.value == "predefined_transponder":
 485                         transponder = nimmanager.getTransponders(satpos)[tuning.transponder.index]
 486                         returnvalue = (int(transponder[1] / 1000), int(transponder[2] / 1000), transponder[3], transponder[4], 2, satpos)
 487                 self.close(returnvalue)
 488
 489         def keyCancel(self):
 490                 self.close(None)
 491
 492 class NimSelection(Screen):
 493         skin = """
 494                 <screen position="140,165" size="400,100" title="select Slot">
 495                         <widget name="nimlist" position="20,10" size="360,75" />
 496                 </screen>"""
 497
 498         def __init__(self, session):
 499                 Screen.__init__(self, session)
 500
 501                 nimlist = nimmanager.getNimListOfType(nimmanager.nimType["DVB-S"])
 502                 nimMenuList = []
 503                 for x in nimlist:
 504                         nimMenuList.append((_("NIM ") + (["A", "B", "C", "D"][x]) + ": " + nimmanager.getNimName(x) + " (" + nimmanager.getNimTypeName(x) + ")", x))
 505
 506                 self["nimlist"] = MenuList(nimMenuList)
 507
 508                 self["actions"] = ActionMap(["OkCancelActions"],
 509                 {
 510                         "ok": self.okbuttonClick ,
 511                         "cancel": self.close
 512                 }, -1)
 513
 514         def okbuttonClick(self):
 515                 selection = self["nimlist"].getCurrent()
 516                 self.session.open(PositionerSetup, selection[1])
 517
 518 def PositionerMain(session, **kwargs):
 519         nimList = nimmanager.getNimListOfType(nimmanager.nimType["DVB-S"])
 520         if len(nimList) == 0:
 521                 session.open(MessageBox, _("No positioner capable frontend found."), MessageBox.TYPE_ERROR)
 522         else:
 523                 if session.nav.RecordTimer.isRecording():
 524                         session.open(MessageBox, _("A recording is currently running. Please stop the recording before trying to configure the positioner."), MessageBox.TYPE_ERROR)
 525                 else:
 526                         usableNims = []
 527                         for x in nimList:
 528                                 configured_rotor_sats = nimmanager.getRotorSatListForNim(x)
 529                                 if len(configured_rotor_sats) != 0:
 530                                         usableNims.append(x)
 531                         if len(usableNims) == 1:
 532                                 session.open(PositionerSetup, usableNims[0])
 533                         elif len(usableNims) > 1:
 534                                 session.open(NimSelection)
 535                         else:
 536                                 session.open(MessageBox, _("No tuner is configured for use with a diseqc positioner!"), MessageBox.TYPE_ERROR)
 537
 538 def Plugins(**kwargs):
 539         return PluginDescriptor(name="Positioner setup", description="Setup your positioner", where = PluginDescriptor.WHERE_PLUGINMENU, fnc=PositionerMain)