class EPG(xbmcgui.WindowXML):
CHANNELS_PER_PAGE = 9
C_MAIN_DATE = 4000
C_MAIN_TITLE = 4020
C_MAIN_TIME = 4021
C_MAIN_DESCRIPTION = 4022
C_MAIN_IMAGE = 4023
C_MAIN_LOGO = 4024
C_MAIN_TIMEBAR = 4100
C_MAIN_LOADING = 4200
C_MAIN_LOADING_PROGRESS = 4201
C_MAIN_LOADING_TIME_LEFT = 4202
C_MAIN_LOADING_CANCEL = 4203
C_MAIN_MOUSE_CONTROLS = 4300
C_MAIN_MOUSE_HOME = 4301
C_MAIN_MOUSE_LEFT = 4302
C_MAIN_MOUSE_UP = 4303
C_MAIN_MOUSE_DOWN = 4304
C_MAIN_MOUSE_RIGHT = 4305
C_MAIN_MOUSE_EXIT = 4306
C_MAIN_BACKGROUND = 4600
C_MAIN_EPG = 5000
C_MAIN_EPG_VIEW_MARKER = 5001
C_MAIN_OSD = 6000
C_MAIN_OSD_TITLE = 6001
C_MAIN_OSD_TIME = 6002
C_MAIN_OSD_DESCRIPTION = 6003
C_MAIN_OSD_CHANNEL_LOGO = 6004
C_MAIN_OSD_CHANNEL_TITLE = 6005
def __new__(cls, currentNr, premiumUser):
# GreenAir: change path
return super(EPG, cls).__new__(cls, 'script-tvguide-main.xml', ADDON.getAddonInfo('path') + '/resources/epg')
def __init__(self, currentNr, premiumUser):
super(EPG, self).__init__()
# self.notification = None
self.redrawingEPG = False
self.isClosing = False
self.controlAndProgramList = list()
self.ignoreMissingControlIds = list()
self.channelIdx = currentNr
self.focusPoint = Point()
self.epgView = EPGView()
self.lastAction={'action':'', 'time':time.time(), 'count':0}
self.premiumUser = premiumUser
# find nearest half hour
self.viewStartDate = datetime.datetime.today()
self.viewStartDate -= datetime.timedelta(minutes=self.viewStartDate.minute % 30, seconds=self.viewStartDate.second)
def getControl(self, controlId):
try:
return super(EPG, self).getControl(controlId)
except:
if controlId in self.ignoreMissingControlIds:
return None
if not self.isClosing:
xbmcgui.Dialog().ok("Control not found", strings(SKIN_ERROR_LINE1), strings(SKIN_ERROR_LINE2), strings(SKIN_ERROR_LINE3))
self.close()
return None
def close(self):
if not self.isClosing:
self.isClosing = True
super(EPG, self).close()
def onInit(self):
self.db = ZattooDB()
control = self.getControl(self.C_MAIN_EPG_VIEW_MARKER)
if control:
left, top = control.getPosition()
self.focusPoint.x = left
self.focusPoint.y = top
self.epgView.left = left
self.epgView.top = top
self.epgView.right = left + control.getWidth()
self.epgView.bottom = top + control.getHeight()
self.epgView.width = control.getWidth()
self.epgView.cellHeight = control.getHeight() / CHANNELS_PER_PAGE
# draw epg on open
self.onRedrawEPG(self.channelIdx, self.viewStartDate)
# self.notification = Notification(self.database, ADDON.getAddonInfo('path')+'/resources/epg')
self.updateTimebar()
def onAction(self, action):
actionId=action.getId()
#print('EPG:'+str(actionId))
if actionId in [ACTION_PARENT_DIR, KEY_NAV_BACK, ACTION_PREVIOUS_MENU]:
self.close()
return
elif actionId == ACTION_MOUSE_MOVE:
# GreenAir: don't show mouse-move infowindow
# self._showControl(self.C_MAIN_MOUSE_CONTROLS)
return
controlInFocus = None
currentFocus = self.focusPoint
try:
controlInFocus = self.getFocus()
if controlInFocus in [elem.control for elem in self.controlAndProgramList]:
(left, top) = controlInFocus.getPosition()
currentFocus = Point()
currentFocus.x = left + (controlInFocus.getWidth() / 2)
currentFocus.y = top + (controlInFocus.getHeight() / 2)
except Exception:
control = self._findControlAt(self.focusPoint)
if control is None and len(self.controlAndProgramList) > 0:
control = self.controlAndProgramList[0].control
if control is not None:
self.setFocus(control)
return
if actionId == ACTION_LEFT:
self._left(currentFocus)
elif actionId == ACTION_RIGHT:
self._right(currentFocus)
elif actionId == ACTION_UP:
self._up(currentFocus)
elif actionId == ACTION_DOWN:
self._down(currentFocus)
elif actionId in [ACTION_NEXT_ITEM, ACTION_6]:
self._nextDay()
elif actionId in [ACTION_PREV_ITEM, ACTION_4]:
self._previousDay()
elif actionId in [ACTION_PAGE_UP, ACTION_8]:
self._moveUp(CHANNELS_PER_PAGE)
elif actionId in [ACTION_PAGE_DOWN, ACTION_2]:
self._moveDown(CHANNELS_PER_PAGE)
elif actionId == ACTION_MOUSE_WHEEL_UP:
self._moveUp(scrollEvent=True)
elif actionId == ACTION_MOUSE_WHEEL_DOWN:
self._moveDown(scrollEvent=True)
elif actionId == KEY_HOME:
self.viewStartDate = datetime.datetime.today()
self.viewStartDate -= datetime.timedelta(minutes=self.viewStartDate.minute % 30,
seconds=self.viewStartDate.second)
self.onRedrawEPG(self.channelIdx, self.viewStartDate)
def onClick(self, controlId):
if controlId in [self.C_MAIN_LOADING_CANCEL, self.C_MAIN_MOUSE_EXIT]:
self.close()
return
if self.isClosing: return
program = self._getProgramFromControl(self.getControl(controlId))
if program is None: return
start = int(time.mktime(program['start_date'].timetuple()))
end = int(time.mktime(program['end_date'].timetuple()))
now = time.time()
# if startime is in the future -> setup recording
if start > now :
if not self.premiumUser: xbmcgui.Dialog().ok('Error',' ',strings(ERROR_NO_PREMIUM))
elif xbmcgui.Dialog().yesno(program['title'], strings(RECORD_SHOW)):
url = "plugin://"+__addonId__+"/?mode=record_p&program_id=" + program['showID']
else: return
# else if endtime is in the past -> recall
elif end < now:
if not self.premiumUser: xbmcgui.Dialog().ok('Error',' ',strings(ERROR_NO_PREMIUM))
elif __addon__.getSetting('epgPlay')=='true':
url = "plugin://"+__addonId__+"/?mode=watch_c&id=" + program['channel'] + "&showID=" + program['showID'] + "&start=" + str(start) + "&end=" + str(end)
else:
ret = xbmcgui.Dialog().select(program['channel']+': '+program['title']+' '+program['start_date'].strftime('%H:%M')+' - '+program['end_date'].strftime('%H:%M'),[strings(PLAY_FROM_START), strings(RECORD_SHOW)])
if ret==0: #recall
url = "plugin://"+__addonId__+"/?mode=watch_c&id=" + program['channel'] + "&showID=" + program['showID'] + "&start=" + str(start) + "&end=" + str(end)
elif ret==1: #record
url = "plugin://"+__addonId__+"/?mode=record_p&program_id=" + program['showID']
else: return
# else currently playing
else:
if (__addon__.getSetting('epgPlay')=='true') or (not self.premiumUser):
url = "plugin://"+__addonId__+"/?mode=watch_c&id=" + program['channel'] + "&showID=" + program['showID']
else:
ret = xbmcgui.Dialog().select(program['channel']+': '+program['title']+' '+program['start_date'].strftime('%H:%M')+' - '+program['end_date'].strftime('%H:%M'), [strings(WATCH_CHANNEL), strings(PLAY_FROM_START), strings(RECORD_SHOW)])
if ret==0: #watch live
url = "plugin://"+__addonId__+"/?mode=watch_c&id=" + program['channel'] + "&showID=" + program['showID']
elif ret==1: #recall
url = "plugin://"+__addonId__+"/?mode=watch_c&id=" + program['channel'] + "&showID=" + program['showID'] + "&start=" + str(start) + "&end=" + str(end)
elif ret==2: #record
url = "plugin://"+__addonId__+"/?mode=record_p&program_id=" + program['showID']
else: return
xbmc.executebuiltin('XBMC.RunPlugin(%s)' % url)
def setFocusId(self, controlId):
control = self.getControl(controlId)
if control: self.setFocus(control)
def setFocus(self, control):
debug('setFocus %d' % control.getId())
if control in [elem.control for elem in self.controlAndProgramList]:
debug('Focus before %s' % self.focusPoint)
(left, top) = control.getPosition()
if left > self.focusPoint.x or left + control.getWidth() < self.focusPoint.x:
self.focusPoint.x = left
self.focusPoint.y = top + (control.getHeight() / 2)
debug('New focus at %s' % self.focusPoint)
super(EPG, self).setFocus(control)
def onFocus(self, controlId):
try:
controlInFocus = self.getControl(controlId)
except Exception:
return
program = self._getProgramFromControl(controlInFocus)
if program is None: return
if program['description'] == None:
self.setControlLabel(self.C_MAIN_TITLE, '[B]%s[/B]' % program['title'])
else:
self.setControlLabel(self.C_MAIN_TITLE, '[B]%s[/B] - [B]%s[/B]' % (program['title'], program['description']))
if program['start_date'] or program['end_date']:
self.setControlLabel(self.C_MAIN_TIME,'[B]%s - %s[/B]' % (self.formatTime(program['start_date']), self.formatTime(program['end_date'])))
else:
self.setControlLabel(self.C_MAIN_TIME, '')
description = self.db.getShowLongDescription(program['showID'])
if description == '': description = strings(NO_DESCRIPTION)
self.setControlText(self.C_MAIN_DESCRIPTION, description)
self.setControlImage(self.C_MAIN_LOGO, program['channel_logo'])
if program['image_small'] is not None:
self.setControlImage(self.C_MAIN_IMAGE, program['image_small'])
self.setControlImage(self.C_MAIN_BACKGROUND, program['image_small'])
# if ADDON.getSetting('program.background.enabled') == 'true' and program['image_large'] is not None:
# self.setControlImage(self.C_MAIN_BACKGROUND, program['image_large'])
def _pressed(self,action):
pressed=False
last=self.lastAction
timeDiff=time.time()-last['time']
if last['count']>0: checkDiff=1
else: checkDiff=0.2
if last['action']==action and (timeDiff<checkDiff):
last['count']+=timeDiff
if last['count']>1:
pressed=2
last['count']=0.01
else: pressed=1
else:
last['action']=action
last['count']=0
last['time']=time.time()
return pressed
def _left(self, currentFocus):
'''
pressed=self._pressed('left')
if pressed==1: return
elif pressed==2: control=None
else: control = self._findControlOnLeft(currentFocus)
'''
control = self._findControlOnLeft(currentFocus)
if control is not None:
self.setFocus(control)
elif control is None:
self.viewStartDate -= datetime.timedelta(hours=2)
self.focusPoint.x = self.epgView.right
self.onRedrawEPG(self.channelIdx, self.viewStartDate, focusFunction=self._findControlOnLeft)
def _right(self, currentFocus):
'''
pressed=self._pressed('right')
if pressed==1: return
elif pressed==2: control=None
else: control = self._findControlOnRight(currentFocus)
'''
control = self._findControlOnRight(currentFocus)
if control is not None:
self.setFocus(control)
elif control is None:
self.viewStartDate += datetime.timedelta(hours=2)
self.focusPoint.x = self.epgView.left
self.onRedrawEPG(self.channelIdx, self.viewStartDate, focusFunction=self._findControlOnRight)
def _up(self, currentFocus):
'''
pressed=self._pressed('up')
if pressed==1: return
elif pressed==2:
self._nextDay()
return
'''
currentFocus.x = self.focusPoint.x
control = self._findControlAbove(currentFocus)
if control is not None:
self.setFocus(control)
elif control is None:
self.focusPoint.y = self.epgView.bottom
self.onRedrawEPG(self.channelIdx - CHANNELS_PER_PAGE, self.viewStartDate,
focusFunction=self._findControlAbove)
def _down(self, currentFocus):
'''
pressed=self._pressed('down')
if pressed==1: return
elif pressed==2:
self._previousDay()
return
'''
currentFocus.x = self.focusPoint.x
control = self._findControlBelow(currentFocus)
if control is not None:
self.setFocus(control)
elif control is None:
self.focusPoint.y = self.epgView.top
self.onRedrawEPG(self.channelIdx + CHANNELS_PER_PAGE, self.viewStartDate,
focusFunction=self._findControlBelow)
def _nextDay(self):
self.viewStartDate += datetime.timedelta(days=1)
self.onRedrawEPG(self.channelIdx, self.viewStartDate)
def _previousDay(self):
self.viewStartDate -= datetime.timedelta(days=1)
self.onRedrawEPG(self.channelIdx, self.viewStartDate)
def _moveUp(self, count=1, scrollEvent=False):
if scrollEvent:
self.onRedrawEPG(self.channelIdx - count, self.viewStartDate)
else:
self.focusPoint.y = self.epgView.bottom
self.onRedrawEPG(self.channelIdx - count, self.viewStartDate, focusFunction=self._findControlAbove)
def _moveDown(self, count=1, scrollEvent=False):
if scrollEvent:
self.onRedrawEPG(self.channelIdx + count, self.viewStartDate)
else:
self.focusPoint.y = self.epgView.top
self.onRedrawEPG(self.channelIdx + count, self.viewStartDate, focusFunction=self._findControlBelow)
def loadChannels(self, favourites):
self.favourites = favourites
def onRedrawEPG(self, channelStart, startTime, focusFunction=None):
if self.redrawingEPG or self.isClosing:
debug('onRedrawEPG - already redrawing')
return # ignore redraw request while redrawing
debug('onRedrawEPG')
self.redrawingEPG = True
self._showControl(self.C_MAIN_EPG)
self.updateTimebar(scheduleTimer=False)
# remove existing controls
self._clearEpg()
channels = self.db.getChannelList(self.favourites)
if channelStart < 0: channelStart = len(channels) - 1
elif channelStart > len(channels) - 1: channelStart = 0
self.channelIdx = channelStart
'''
#channels that are visible
channels={'index':[]}
for nr in range(0, CHANNELS_PER_PAGE):
id=allChannels['index'][channelStart+nr]
channels[id]= allChannels[id]
channels['index'].append(id)
'''
'''
programs = self.db.getPrograms(channels, False, startTime, startTime + datetime.timedelta(hours=2))
if programs is None:
self.onEPGLoadError()
return
# channelsWithoutPrograms = list(channels)
'''
self.db.updateProgram(startTime)
# date and time row
self.setControlLabel(self.C_MAIN_DATE, self.formatDate(self.viewStartDate))
time=startTime
for col in range(1, 5):
self.setControlLabel(4000 + col, self.formatTime(time))
time += HALF_HOUR
# set channel logo or text
for idx in range(0, CHANNELS_PER_PAGE):
if (channelStart+idx) >= len(channels['index']):
self.setControlImage(4110 + idx, ' ')
self.setControlLabel(4010 + idx, ' ')
else:
channel = channels[channels['index'][channelStart+idx]]
self.setControlLabel(4010 + idx, channel['title'])
if channel['logo'] is not None: self.setControlImage(4110 + idx, channel['logo'])
else: self.setControlImage(4110 + idx, ' ')
programs = self.db.getPrograms({channel['id']:''}, False, startTime, startTime + datetime.timedelta(hours=2))
for program in programs:
# add channel index and logo for control-handling
program['channel_index'] = idx
program['channel_logo'] = channel['logo']
startDelta = program['start_date'] - self.viewStartDate
stopDelta = program['end_date'] - self.viewStartDate
cellStart = self._secondsToXposition(startDelta.seconds)
if startDelta.days < 0: cellStart = self.epgView.left
cellWidth = self._secondsToXposition(stopDelta.seconds) - cellStart
if cellStart + cellWidth > self.epgView.right: cellWidth = self.epgView.right - cellStart
if cellWidth > 1:
noFocusTexture = 'tvguide-program-grey.png'
focusTexture = 'tvguide-program-grey-focus.png'
if cellWidth < 25: title = '' # Text will overflow outside the button if it is too narrow
else: title = program['title']
control = xbmcgui.ControlButton(
cellStart,
self.epgView.top + self.epgView.cellHeight * idx,
cellWidth - 2,
self.epgView.cellHeight - 2,
title,
noFocusTexture=noFocusTexture,
focusTexture=focusTexture
)
self.controlAndProgramList.append(ControlAndProgram(control, program))
# for channel in channelsWithoutPrograms:
# idx = channels.index(channel)
#
# control = xbmcgui.ControlButton(
# self.epgView.left,
# self.epgView.top + self.epgView.cellHeight * idx,
# (self.epgView.right - self.epgView.left) - 2,
# self.epgView.cellHeight - 2,
# strings(NO_PROGRAM_AVAILABLE),
# noFocusTexture='tvguide-program-grey.png',
# focusTexture='tvguide-program-grey-focus.png'
# )
#
# program = src.Program(channel, strings(NO_PROGRAM_AVAILABLE), None, None, None)
# self.controlAndProgramList.append(ControlAndProgram(control, program))
# add program controls
if focusFunction is None:
focusFunction = self._findControlAt
focusControl = focusFunction(self.focusPoint)
controls = [elem.control for elem in self.controlAndProgramList]
self.addControls(controls)
if focusControl is not None:
debug('onRedrawEPG - setFocus %d' % focusControl.getId())
self.setFocus(focusControl)
self.ignoreMissingControlIds.extend([elem.control.getId() for elem in self.controlAndProgramList])
if focusControl is None and len(self.controlAndProgramList) > 0:
self.setFocus(self.controlAndProgramList[0].control)
self.redrawingEPG = False
def _clearEpg(self):
controls = [elem.control for elem in self.controlAndProgramList]
try:
self.removeControls(controls)
except RuntimeError:
for elem in self.controlAndProgramList:
try:
self.removeControl(elem.control)
except RuntimeError:
pass # happens if we try to remove a control that doesn't exist
del self.controlAndProgramList[:]
def onEPGLoadError(self):
self.redrawingEPG = False
xbmcgui.Dialog().ok(strings(LOAD_ERROR_TITLE), strings(LOAD_ERROR_LINE1), strings(LOAD_ERROR_LINE2))
self.close()
def _secondsToXposition(self, seconds):
return self.epgView.left + (seconds * self.epgView.width / 7200)
def _findControlOnRight(self, point):
distanceToNearest = 10000
nearestControl = None
for elem in self.controlAndProgramList:
control = elem.control
(left, top) = control.getPosition()
x = left + (control.getWidth() / 2)
y = top + (control.getHeight() / 2)
if point.x < x and point.y == y:
distance = abs(point.x - x)
if distance < distanceToNearest:
distanceToNearest = distance
nearestControl = control
return nearestControl
def _findControlOnLeft(self, point):
distanceToNearest = 10000
nearestControl = None
for elem in self.controlAndProgramList:
control = elem.control
(left, top) = control.getPosition()
x = left + (control.getWidth() / 2)
y = top + (control.getHeight() / 2)
if point.x > x and point.y == y:
distance = abs(point.x - x)
if distance < distanceToNearest:
distanceToNearest = distance
nearestControl = control
return nearestControl
def _findControlBelow(self, point):
nearestControl = None
for elem in self.controlAndProgramList:
control = elem.control
(leftEdge, top) = control.getPosition()
y = top + (control.getHeight() / 2)
if point.y < y:
rightEdge = leftEdge + control.getWidth()
if leftEdge <= point.x < rightEdge and (nearestControl is None or nearestControl.getPosition()[1] > top):
nearestControl = control
return nearestControl
def _findControlAbove(self, point):
nearestControl = None
for elem in self.controlAndProgramList:
control = elem.control
(leftEdge, top) = control.getPosition()
y = top + (control.getHeight() / 2)
if point.y > y:
rightEdge = leftEdge + control.getWidth()
if leftEdge <= point.x < rightEdge and (nearestControl is None or nearestControl.getPosition()[1] < top):
nearestControl = control
return nearestControl
def _findControlAt(self, point):
for elem in self.controlAndProgramList:
control = elem.control
(left, top) = control.getPosition()
bottom = top + control.getHeight()
right = left + control.getWidth()
if left <= point.x <= right and top <= point.y <= bottom:
return control
return None
def _getProgramFromControl(self, control):
for elem in self.controlAndProgramList:
if elem.control == control:
return elem.program
return None
def _hideControl(self, *controlIds):
"""
Visibility is inverted in skin
"""
for controlId in controlIds:
control = self.getControl(controlId)
if control:
control.setVisible(True)
def _showControl(self, *controlIds):
"""
Visibility is inverted in skin
"""
for controlId in controlIds:
control = self.getControl(controlId)
if control:
control.setVisible(False)
def formatTime(self, timestamp):
if timestamp:
format = xbmc.getRegion('time').replace(':%S', '').replace('%H%H', '%H')
return timestamp.strftime(format)
else:
return ''
def formatDate(self, timestamp):
if timestamp:
# format = xbmc.getRegion('dateshort')
format = "%a %d.%m."
return timestamp.strftime(format)
else:
return ''
def setControlImage(self, controlId, image):
control = self.getControl(controlId)
if control:
control.setImage(image.encode('utf-8'), False)
def setControlLabel(self, controlId, label):
control = self.getControl(controlId)
if control and label:
control.setLabel(label)
def setControlText(self, controlId, text):
control = self.getControl(controlId)
if control:
control.setText(text)
def updateTimebar(self, scheduleTimer=True):
try:
# move timebar to current time
timeDelta = datetime.datetime.today() - self.viewStartDate
control = self.getControl(self.C_MAIN_TIMEBAR)
if control:
(x, y) = control.getPosition()
try:
# Sometimes raises:
# exceptions.RuntimeError: Unknown exception thrown from the call "setVisible"
control.setVisible(timeDelta.days == 0)
except:
pass
control.setPosition(self._secondsToXposition(timeDelta.seconds), y)
if scheduleTimer and not xbmc.abortRequested and not self.isClosing:
threading.Timer(1, self.updateTimebar).start()
except Exception:
pass
