lib/dvb/epgcache.h

   1 #ifndef __epgcache_h_
   2 #define __epgcache_h_
   3
   4 #define ENABLE_PRIVATE_EPG 1
   5
   6 #ifndef SWIG
   7
   8 #include <vector>
   9 #include <list>
  10 #include <ext/hash_map>
  11 #include <ext/hash_set>
  12
  13 #include <errno.h>
  14
  15 #include <lib/dvb/eit.h>
  16 #include <lib/dvb/lowlevel/eit.h>
  17 #include <lib/dvb/idvb.h>
  18 #include <lib/dvb/demux.h>
  19 #include <lib/dvb/dvbtime.h>
  20 #include <lib/base/ebase.h>
  21 #include <lib/base/thread.h>
  22 #include <lib/base/message.h>
  23 #include <lib/service/event.h>
  24 #include <lib/python/python.h>
  25
  26 #define CLEAN_INTERVAL 60000    //  1 min
  27 #define UPDATE_INTERVAL 3600000  // 60 min
  28 #define ZAP_DELAY 2000          // 2 sek
  29
  30 #define HILO(x) (x##_hi << 8 | x##_lo)
  31
  32 class eventData;
  33 class eServiceReferenceDVB;
  34 class eDVBServicePMTHandler;
  35
  36 struct uniqueEPGKey
  37 {
  38         int sid, onid, tsid;
  39         uniqueEPGKey( const eServiceReference &ref )
  40                 :sid( ref.type != eServiceReference::idInvalid ? ((eServiceReferenceDVB&)ref).getServiceID().get() : -1 )
  41                 ,onid( ref.type != eServiceReference::idInvalid ? ((eServiceReferenceDVB&)ref).getOriginalNetworkID().get() : -1 )
  42                 ,tsid( ref.type != eServiceReference::idInvalid ? ((eServiceReferenceDVB&)ref).getTransportStreamID().get() : -1 )
  43         {
  44         }
  45         uniqueEPGKey()
  46                 :sid(-1), onid(-1), tsid(-1)
  47         {
  48         }
  49         uniqueEPGKey( int sid, int onid, int tsid )
  50                 :sid(sid), onid(onid), tsid(tsid)
  51         {
  52         }
  53         bool operator <(const uniqueEPGKey &a) const
  54         {
  55                 return memcmp( &sid, &a.sid, sizeof(int)*3)<0;
  56         }
  57         operator bool() const
  58         {
  59                 return !(sid == -1 && onid == -1 && tsid == -1);
  60         }
  61         bool operator==(const uniqueEPGKey &a) const
  62         {
  63                 return !memcmp( &sid, &a.sid, sizeof(int)*3);
  64         }
  65         struct equal
  66         {
  67                 bool operator()(const uniqueEPGKey &a, const uniqueEPGKey &b) const
  68                 {
  69                         return !memcmp( &a.sid, &b.sid, sizeof(int)*3);
  70                 }
  71         };
  72 };
  73
  74 //eventMap is sorted by event_id
  75 #define eventMap std::map<__u16, eventData*>
  76 //timeMap is sorted by beginTime
  77 #define timeMap std::map<time_t, eventData*>
  78
  79 #define channelMapIterator std::map<iDVBChannel*, channel_data*>::iterator
  80 #define updateMap std::map<eDVBChannelID, time_t>
  81
  82 struct hash_uniqueEPGKey
  83 {
  84         inline size_t operator()( const uniqueEPGKey &x) const
  85         {
  86                 return (x.onid << 16) | x.tsid;
  87         }
  88 };
  89
  90 #define tidMap std::set<__u32>
  91 #if defined(__GNUC__) && ((__GNUC__ == 3 && __GNUC_MINOR__ >= 1) || __GNUC__ == 4 )  // check if gcc version >= 3.1
  92         #define eventCache __gnu_cxx::hash_map<uniqueEPGKey, std::pair<eventMap, timeMap>, hash_uniqueEPGKey, uniqueEPGKey::equal>
  93         #ifdef ENABLE_PRIVATE_EPG
  94                 #define contentTimeMap __gnu_cxx::hash_map<time_t, std::pair<time_t, __u16> >
  95                 #define contentMap __gnu_cxx::hash_map<int, contentTimeMap >
  96                 #define contentMaps __gnu_cxx::hash_map<uniqueEPGKey, contentMap, hash_uniqueEPGKey, uniqueEPGKey::equal >
  97         #endif
  98 #else // for older gcc use following
  99         #define eventCache std::hash_map<uniqueEPGKey, std::pair<eventMap, timeMap>, hash_uniqueEPGKey, uniqueEPGKey::equal >
 100         #ifdef ENABLE_PRIVATE_EPG
 101                 #define contentTimeMap std::hash_map<time_t, std::pair<time_t, __u16> >
 102                 #define contentMap std::hash_map<int, contentTimeMap >
 103                 #define contentMaps std::hash_map<uniqueEPGKey, contentMap, hash_uniqueEPGKey, uniqueEPGKey::equal>
 104         #endif
 105 #endif
 106
 107 #define descriptorPair std::pair<int,__u8*>
 108 #define descriptorMap std::map<__u32, descriptorPair >
 109
 110 class eventData
 111 {
 112         friend class eEPGCache;
 113 private:
 114         __u8* EITdata;
 115         __u8 ByteSize;
 116         __u8 type;
 117         static descriptorMap descriptors;
 118         static __u8 data[4108];
 119         static int CacheSize;
 120         static void load(FILE *);
 121         static void save(FILE *);
 122 public:
 123         eventData(const eit_event_struct* e=NULL, int size=0, int type=0);
 124         ~eventData();
 125         const eit_event_struct* get() const;
 126         operator const eit_event_struct*() const
 127         {
 128                 return get();
 129         }
 130         int getEventID()
 131         {
 132                 return (EITdata[0] << 8) | EITdata[1];
 133         }
 134         time_t getStartTime()
 135         {
 136                 return parseDVBtime(EITdata[2], EITdata[3], EITdata[4], EITdata[5], EITdata[6]);
 137         }
 138         int getDuration()
 139         {
 140                 return fromBCD(EITdata[7])*3600+fromBCD(EITdata[8])*60+fromBCD(EITdata[9]);
 141         }
 142 };
 143 #endif
 144
 145 class eEPGCache: public eMainloop, private eThread, public Object
 146 {
 147 #ifndef SWIG
 148         DECLARE_REF(eEPGCache)
 149         struct channel_data: public Object
 150         {
 151                 channel_data(eEPGCache*);
 152                 eEPGCache *cache;
 153                 eTimer abortTimer, zapTimer;
 154                 int prevChannelState;
 155                 __u8 state, isRunning, haveData;
 156                 ePtr<eDVBChannel> channel;
 157                 ePtr<eConnection> m_stateChangedConn, m_NowNextConn, m_ScheduleConn, m_ScheduleOtherConn;
 158                 ePtr<iDVBSectionReader> m_NowNextReader, m_ScheduleReader, m_ScheduleOtherReader;
 159                 tidMap seenSections[3], calcedSections[3];
 160 #ifdef ENABLE_PRIVATE_EPG
 161                 eTimer startPrivateTimer;
 162                 int m_PrevVersion;
 163                 int m_PrivatePid;
 164                 uniqueEPGKey m_PrivateService;
 165                 ePtr<eConnection> m_PrivateConn;
 166                 ePtr<iDVBSectionReader> m_PrivateReader;
 167                 std::set<__u8> seenPrivateSections;
 168                 void readPrivateData(const __u8 *data);
 169                 void startPrivateReader();
 170                 bool canDelete() { return !isRunning && !m_PrivateConn; }
 171 #else
 172                 bool canDelete() { return !isRunning; }
 173 #endif
 174                 void readData(const __u8 *data);
 175                 void startChannel();
 176                 void startEPG();
 177                 bool finishEPG();
 178                 void abortEPG();
 179                 void abortNonAvail();
 180         };
 181         void FixOverlapping(std::pair<eventMap,timeMap> &servicemap, time_t TM, int duration, const timeMap::iterator &tm_it, const uniqueEPGKey &service);
 182 public:
 183         enum {PRIVATE=0, NOWNEXT=1, SCHEDULE=2, SCHEDULE_OTHER=4};
 184         struct Message
 185         {
 186                 enum
 187                 {
 188                         flush,
 189                         startChannel,
 190                         leaveChannel,
 191                         pause,
 192                         restart,
 193                         updated,
 194                         isavail,
 195                         quit,
 196                         got_private_pid,
 197                         timeChanged
 198                 };
 199                 int type;
 200                 iDVBChannel *channel;
 201                 uniqueEPGKey service;
 202                 union {
 203                         int err;
 204                         time_t time;
 205                         bool avail;
 206                         int pid;
 207                 };
 208                 Message()
 209                         :type(0), time(0) {}
 210                 Message(int type)
 211                         :type(type) {}
 212                 Message(int type, bool b)
 213                         :type(type), avail(b) {}
 214                 Message(int type, iDVBChannel *channel, int err=0)
 215                         :type(type), channel(channel), err(err) {}
 216                 Message(int type, const eServiceReference& service, int err=0)
 217                         :type(type), service(service), err(err) {}
 218                 Message(int type, time_t time)
 219                         :type(type), time(time) {}
 220         };
 221         eFixedMessagePump<Message> messages;
 222 private:
 223         friend class channel_data;
 224         static eEPGCache *instance;
 225
 226         eTimer cleanTimer;
 227         std::map<iDVBChannel*, channel_data*> m_knownChannels;
 228         ePtr<eConnection> m_chanAddedConn;
 229
 230         eventCache eventDB;
 231         updateMap channelLastUpdated;
 232         static pthread_mutex_t cache_lock, channel_map_lock;
 233
 234 #ifdef ENABLE_PRIVATE_EPG
 235         contentMaps content_time_tables;
 236 #endif
 237
 238         void thread();  // thread function
 239
 240 // called from epgcache thread
 241         void save();
 242         void load();
 243 #ifdef ENABLE_PRIVATE_EPG
 244         void privateSectionRead(const uniqueEPGKey &, const __u8 *);
 245 #endif
 246         void sectionRead(const __u8 *data, int source, channel_data *channel);
 247         void gotMessage(const Message &message);
 248         void flushEPG(const uniqueEPGKey & s=uniqueEPGKey());
 249         void cleanLoop();
 250
 251 // called from main thread
 252         void timeUpdated();
 253         void DVBChannelAdded(eDVBChannel*);
 254         void DVBChannelStateChanged(iDVBChannel*);
 255         void DVBChannelRunning(iDVBChannel *);
 256
 257         timeMap::iterator m_timemap_cursor, m_timemap_end;
 258         int currentQueryTsidOnid; // needed for getNextTimeEntry.. only valid until next startTimeQuery call
 259 #else
 260         eEPGCache();
 261         ~eEPGCache();
 262 #endif // SWIG
 263 public:
 264         static eEPGCache *getInstance() { return instance; }
 265 #ifndef SWIG
 266         eEPGCache();
 267         ~eEPGCache();
 268 #endif
 269
 270         // called from main thread
 271         inline void Lock();
 272         inline void Unlock();
 273 #ifdef ENABLE_PRIVATE_EPG
 274         void PMTready(eDVBServicePMTHandler *pmthandler);
 275 #else
 276         void PMTready(eDVBServicePMTHandler *pmthandler) {}
 277 #endif
 278
 279         // at moment just for one service..
 280         RESULT startTimeQuery(const eServiceReference &service, time_t begin=-1, int minutes=-1);
 281
 282 #ifndef SWIG
 283         // eventData's are plain entrys out of the cache.. it's not safe to use them after cache unlock
 284         // but its faster in use... its not allowed to delete this pointers via delete or free..
 285         SWIG_VOID(RESULT) lookupEventId(const eServiceReference &service, int event_id, const eventData *&SWIG_OUTPUT);
 286         SWIG_VOID(RESULT) lookupEventTime(const eServiceReference &service, time_t, const eventData *&SWIG_OUTPUT, int direction=0);
 287         SWIG_VOID(RESULT) getNextTimeEntry(const eventData *&SWIG_OUTPUT);
 288
 289         // eit_event_struct's are plain dvb eit_events .. it's not safe to use them after cache unlock
 290         // its not allowed to delete this pointers via delete or free..
 291         RESULT lookupEventId(const eServiceReference &service, int event_id, const eit_event_struct *&);
 292         RESULT lookupEventTime(const eServiceReference &service, time_t , const eit_event_struct *&, int direction=0);
 293         RESULT getNextTimeEntry(const eit_event_struct *&);
 294
 295         // Event's are parsed epg events.. it's safe to use them after cache unlock
 296         // after use this Events must be deleted (memleaks)
 297         RESULT lookupEventId(const eServiceReference &service, int event_id, Event* &);
 298         RESULT lookupEventTime(const eServiceReference &service, time_t, Event* &, int direction=0);
 299         RESULT getNextTimeEntry(Event *&);
 300 #endif
 301         enum {
 302                 SIMILAR_BROADCASTINGS_SEARCH,
 303                 EXAKT_TITLE_SEARCH,
 304                 PARTIAL_TITLE_SEARCH
 305         };
 306         enum {
 307                 CASE_CHECK,
 308                 NO_CASE_CHECK
 309         };
 310         PyObject *lookupEvent(PyObject *list, PyObject *convertFunc=NULL);
 311         PyObject *search(PyObject *);
 312
 313         // eServiceEvent are parsed epg events.. it's safe to use them after cache unlock
 314         // for use from python ( members: m_start_time, m_duration, m_short_description, m_extended_description )
 315         SWIG_VOID(RESULT) lookupEventId(const eServiceReference &service, int event_id, ePtr<eServiceEvent> &SWIG_OUTPUT);
 316         SWIG_VOID(RESULT) lookupEventTime(const eServiceReference &service, time_t, ePtr<eServiceEvent> &SWIG_OUTPUT, int direction=0);
 317         SWIG_VOID(RESULT) getNextTimeEntry(ePtr<eServiceEvent> &SWIG_OUTPUT);
 318 };
 319
 320 #ifndef SWIG
 321 inline void eEPGCache::Lock()
 322 {
 323         pthread_mutex_lock(&cache_lock);
 324 }
 325
 326 inline void eEPGCache::Unlock()
 327 {
 328         pthread_mutex_unlock(&cache_lock);
 329 }
 330 #endif
 331
 332 #endif