4 #define ENABLE_PRIVATE_EPG 1
5 //#define ENABLE_MHW_EPG 1
11 #include <ext/hash_map>
12 #include <ext/hash_set>
16 #include <lib/dvb/eit.h>
17 #include <lib/dvb/lowlevel/eit.h>
19 #include <lib/dvb/lowlevel/mhw.h>
21 #include <lib/dvb/idvb.h>
22 #include <lib/dvb/demux.h>
23 #include <lib/dvb/dvbtime.h>
24 #include <lib/base/ebase.h>
25 #include <lib/base/thread.h>
26 #include <lib/base/message.h>
27 #include <lib/service/event.h>
28 #include <lib/python/python.h>
30 #define CLEAN_INTERVAL 60000 // 1 min
31 #define UPDATE_INTERVAL 3600000 // 60 min
32 #define ZAP_DELAY 2000 // 2 sek
34 #define HILO(x) (x##_hi << 8 | x##_lo)
37 class eServiceReferenceDVB;
38 class eDVBServicePMTHandler;
43 uniqueEPGKey( const eServiceReference &ref )
44 :sid( ref.type != eServiceReference::idInvalid ? ((eServiceReferenceDVB&)ref).getServiceID().get() : -1 )
45 ,onid( ref.type != eServiceReference::idInvalid ? ((eServiceReferenceDVB&)ref).getOriginalNetworkID().get() : -1 )
46 ,tsid( ref.type != eServiceReference::idInvalid ? ((eServiceReferenceDVB&)ref).getTransportStreamID().get() : -1 )
50 :sid(-1), onid(-1), tsid(-1)
53 uniqueEPGKey( int sid, int onid, int tsid )
54 :sid(sid), onid(onid), tsid(tsid)
57 bool operator <(const uniqueEPGKey &a) const
59 return memcmp( &sid, &a.sid, sizeof(int)*3)<0;
63 return !(sid == -1 && onid == -1 && tsid == -1);
65 bool operator==(const uniqueEPGKey &a) const
67 return !memcmp( &sid, &a.sid, sizeof(int)*3);
71 bool operator()(const uniqueEPGKey &a, const uniqueEPGKey &b) const
73 return !memcmp( &a.sid, &b.sid, sizeof(int)*3);
78 //eventMap is sorted by event_id
79 #define eventMap std::map<__u16, eventData*>
80 //timeMap is sorted by beginTime
81 #define timeMap std::map<time_t, eventData*>
83 #define channelMapIterator std::map<iDVBChannel*, channel_data*>::iterator
84 #define updateMap std::map<eDVBChannelID, time_t>
86 struct hash_uniqueEPGKey
88 inline size_t operator()( const uniqueEPGKey &x) const
90 return (x.onid << 16) | x.tsid;
94 #define tidMap std::set<__u32>
95 #if defined(__GNUC__) && ((__GNUC__ == 3 && __GNUC_MINOR__ >= 1) || __GNUC__ == 4 ) // check if gcc version >= 3.1
96 #define eventCache __gnu_cxx::hash_map<uniqueEPGKey, std::pair<eventMap, timeMap>, hash_uniqueEPGKey, uniqueEPGKey::equal>
97 #ifdef ENABLE_PRIVATE_EPG
98 #define contentTimeMap __gnu_cxx::hash_map<time_t, std::pair<time_t, __u16> >
99 #define contentMap __gnu_cxx::hash_map<int, contentTimeMap >
100 #define contentMaps __gnu_cxx::hash_map<uniqueEPGKey, contentMap, hash_uniqueEPGKey, uniqueEPGKey::equal >
102 #else // for older gcc use following
103 #define eventCache std::hash_map<uniqueEPGKey, std::pair<eventMap, timeMap>, hash_uniqueEPGKey, uniqueEPGKey::equal >
104 #ifdef ENABLE_PRIVATE_EPG
105 #define contentTimeMap std::hash_map<time_t, std::pair<time_t, __u16> >
106 #define contentMap std::hash_map<int, contentTimeMap >
107 #define contentMaps std::hash_map<uniqueEPGKey, contentMap, hash_uniqueEPGKey, uniqueEPGKey::equal>
111 #define descriptorPair std::pair<int,__u8*>
112 #define descriptorMap std::map<__u32, descriptorPair >
116 friend class eEPGCache;
121 static descriptorMap descriptors;
122 static __u8 data[4108];
123 static int CacheSize;
124 static void load(FILE *);
125 static void save(FILE *);
127 eventData(const eit_event_struct* e=NULL, int size=0, int type=0);
129 const eit_event_struct* get() const;
130 operator const eit_event_struct*() const
136 return (EITdata[0] << 8) | EITdata[1];
138 time_t getStartTime()
140 return parseDVBtime(EITdata[2], EITdata[3], EITdata[4], EITdata[5], EITdata[6]);
144 return fromBCD(EITdata[7])*3600+fromBCD(EITdata[8])*60+fromBCD(EITdata[9]);
149 class eEPGCache: public eMainloop, private eThread, public Object
152 DECLARE_REF(eEPGCache)
153 struct channel_data: public Object
155 pthread_mutex_t channel_active;
156 channel_data(eEPGCache*);
158 eTimer abortTimer, zapTimer;
159 int prevChannelState;
160 __u8 state, isRunning, haveData;
161 ePtr<eDVBChannel> channel;
162 ePtr<eConnection> m_stateChangedConn, m_NowNextConn, m_ScheduleConn, m_ScheduleOtherConn;
163 ePtr<iDVBSectionReader> m_NowNextReader, m_ScheduleReader, m_ScheduleOtherReader;
164 tidMap seenSections[3], calcedSections[3];
165 #ifdef ENABLE_PRIVATE_EPG
166 eTimer startPrivateTimer;
169 uniqueEPGKey m_PrivateService;
170 ePtr<eConnection> m_PrivateConn;
171 ePtr<iDVBSectionReader> m_PrivateReader;
172 std::set<__u8> seenPrivateSections;
173 void readPrivateData(const __u8 *data);
174 void startPrivateReader();
176 #ifdef ENABLE_MHW_EPG
177 std::vector<mhw_channel_name_t> m_channels;
178 std::map<__u8, mhw_theme_name_t> m_themes;
179 std::map<__u32, mhw_title_t> m_titles;
180 std::map<__u32, __u32> m_program_ids;
181 ePtr<eConnection> m_MHWConn;
182 ePtr<iDVBSectionReader> m_MHWReader;
183 eDVBSectionFilterMask m_MHWFilterMask;
184 eTimer m_MHWTimeoutTimer;
186 void MHWTimeout() { m_MHWTimeoutet=true; }
187 void readMHWData(const __u8 *data);
188 void startMHWReader(__u16 pid, __u8 tid);
189 void startTimeout(int msek);
190 bool checkTimeout() { return m_MHWTimeoutet; }
192 __u8 *delimitName( __u8 *in, __u8 *out, int len_in );
193 void timeMHW2DVB( u_char hours, u_char minutes, u_char *return_time);
194 void timeMHW2DVB( int minutes, u_char *return_time);
195 void timeMHW2DVB( u_char day, u_char hours, u_char minutes, u_char *return_time);
196 void storeTitle(std::map<__u32, mhw_title_t>::iterator itTitle, std::string sumText, const __u8 *data);
198 void readData(const __u8 *data);
203 void abortNonAvail();
205 void FixOverlapping(std::pair<eventMap,timeMap> &servicemap, time_t TM, int duration, const timeMap::iterator &tm_it, const uniqueEPGKey &service);
207 enum {PRIVATE=0, NOWNEXT=1, SCHEDULE=2, SCHEDULE_OTHER=4
208 #ifdef ENABLE_MHW_EPG
228 iDVBChannel *channel;
229 uniqueEPGKey service;
240 Message(int type, bool b)
241 :type(type), avail(b) {}
242 Message(int type, iDVBChannel *channel, int err=0)
243 :type(type), channel(channel), err(err) {}
244 Message(int type, const eServiceReference& service, int err=0)
245 :type(type), service(service), err(err) {}
246 Message(int type, time_t time)
247 :type(type), time(time) {}
249 eFixedMessagePump<Message> messages;
251 friend class channel_data;
252 static eEPGCache *instance;
255 std::map<iDVBChannel*, channel_data*> m_knownChannels;
256 ePtr<eConnection> m_chanAddedConn;
259 updateMap channelLastUpdated;
260 static pthread_mutex_t cache_lock, channel_map_lock;
262 #ifdef ENABLE_PRIVATE_EPG
263 contentMaps content_time_tables;
266 void thread(); // thread function
268 // called from epgcache thread
271 #ifdef ENABLE_PRIVATE_EPG
272 void privateSectionRead(const uniqueEPGKey &, const __u8 *);
274 void sectionRead(const __u8 *data, int source, channel_data *channel);
275 void gotMessage(const Message &message);
276 void flushEPG(const uniqueEPGKey & s=uniqueEPGKey());
279 // called from main thread
281 void DVBChannelAdded(eDVBChannel*);
282 void DVBChannelStateChanged(iDVBChannel*);
283 void DVBChannelRunning(iDVBChannel *);
285 timeMap::iterator m_timemap_cursor, m_timemap_end;
286 int currentQueryTsidOnid; // needed for getNextTimeEntry.. only valid until next startTimeQuery call
292 static eEPGCache *getInstance() { return instance; }
298 // called from main thread
300 inline void Unlock();
301 #ifdef ENABLE_PRIVATE_EPG
302 void PMTready(eDVBServicePMTHandler *pmthandler);
304 void PMTready(eDVBServicePMTHandler *pmthandler) {}
307 // at moment just for one service..
308 RESULT startTimeQuery(const eServiceReference &service, time_t begin=-1, int minutes=-1);
311 // eventData's are plain entrys out of the cache.. it's not safe to use them after cache unlock
312 // but its faster in use... its not allowed to delete this pointers via delete or free..
313 SWIG_VOID(RESULT) lookupEventId(const eServiceReference &service, int event_id, const eventData *&SWIG_OUTPUT);
314 SWIG_VOID(RESULT) lookupEventTime(const eServiceReference &service, time_t, const eventData *&SWIG_OUTPUT, int direction=0);
315 SWIG_VOID(RESULT) getNextTimeEntry(const eventData *&SWIG_OUTPUT);
317 // eit_event_struct's are plain dvb eit_events .. it's not safe to use them after cache unlock
318 // its not allowed to delete this pointers via delete or free..
319 RESULT lookupEventId(const eServiceReference &service, int event_id, const eit_event_struct *&);
320 RESULT lookupEventTime(const eServiceReference &service, time_t , const eit_event_struct *&, int direction=0);
321 RESULT getNextTimeEntry(const eit_event_struct *&);
323 // Event's are parsed epg events.. it's safe to use them after cache unlock
324 // after use this Events must be deleted (memleaks)
325 RESULT lookupEventId(const eServiceReference &service, int event_id, Event* &);
326 RESULT lookupEventTime(const eServiceReference &service, time_t, Event* &, int direction=0);
327 RESULT getNextTimeEntry(Event *&);
330 SIMILAR_BROADCASTINGS_SEARCH,
338 PyObject *lookupEvent(PyObject *list, PyObject *convertFunc=NULL);
339 PyObject *search(PyObject *);
341 // eServiceEvent are parsed epg events.. it's safe to use them after cache unlock
342 // for use from python ( members: m_start_time, m_duration, m_short_description, m_extended_description )
343 SWIG_VOID(RESULT) lookupEventId(const eServiceReference &service, int event_id, ePtr<eServiceEvent> &SWIG_OUTPUT);
344 SWIG_VOID(RESULT) lookupEventTime(const eServiceReference &service, time_t, ePtr<eServiceEvent> &SWIG_OUTPUT, int direction=0);
345 SWIG_VOID(RESULT) getNextTimeEntry(ePtr<eServiceEvent> &SWIG_OUTPUT);
349 inline void eEPGCache::Lock()
351 pthread_mutex_lock(&cache_lock);
354 inline void eEPGCache::Unlock()
356 pthread_mutex_unlock(&cache_lock);