service/event.cpp: convert country code to lower case before compare it
[vuplus_dvbapp] / lib / service / event.cpp
1 #include <lib/service/event.h>
2 #include <lib/base/estring.h>
3 #include <lib/base/encoding.h>
4 #include <lib/dvb/dvbtime.h>
5 #include <lib/dvb/idvb.h>
6 #include <dvbsi++/event_information_section.h>
7 #include <dvbsi++/short_event_descriptor.h>
8 #include <dvbsi++/extended_event_descriptor.h>
9 #include <dvbsi++/linkage_descriptor.h>
10 #include <dvbsi++/component_descriptor.h>
11 #include <dvbsi++/descriptor_tag.h>
12
13 #include <sys/types.h>
14 #include <fcntl.h>
15
16 // static members / methods
17 std::string eServiceEvent::m_language = "de_DE";
18
19 void eServiceEvent::setEPGLanguage( const std::string language )
20 {
21         m_language = language;
22 }
23 ///////////////////////////
24
25 DEFINE_REF(eServiceEvent);
26 DEFINE_REF(eComponentData);
27
28 const char MAX_LANG = 37;
29 /* OSD language (see /share/locales/locales) to iso639 conversion table */
30 std::string ISOtbl[MAX_LANG][2] =
31 {
32         {"ar_AE","ara"},
33         {"C","eng"},
34         {"cs_CZ","ces"},     /* or 'cze' */
35         {"cs_CZ","cze"},
36         {"da_DK","dan"},
37         {"de_DE","deu"},     /* also 'ger' is valid iso639 code!! */
38         {"de_DE","ger"},
39         {"el_GR","gre"},     /* also 'ell' is valid */
40         {"el_GR","ell"},
41         {"es_ES","esl"},     /* also 'spa' is ok */
42         {"es_ES","spa"},
43         {"et_EE","est"},
44         {"fi_FI","fin"},
45         {"fr_FR","fra"},
46         {"hr_HR","hrv"},     /* or 'scr' */
47         {"hr_HR","scr"},
48         {"hu_HU","hun"},
49         {"is_IS","isl"},     /* or 'ice' */
50         {"is_IS","ice"},
51         {"it_IT","ita"},
52         {"lt_LT","lit"},
53         {"nl_NL","nld"},     /* or 'dut' */
54         {"nl_NL","dut"},
55         {"no_NO","nor"},
56         {"pl_PL","pol"},
57         {"pt_PT","por"},
58         {"ro_RO","ron"},     /* or 'rum' */
59         {"ro_RO","rum"},
60         {"ru_RU","rus"},
61         {"sk_SK","slk"},     /* or 'slo' */
62         {"sk_SK","slo"},
63         {"sl_SI","slv"},
64         {"sr_YU","srp"},     /* or 'scc' */
65         {"sr_YU","scc"},
66         {"sv_SE","swe"},
67         {"tr_TR","tur"},
68         {"ur_IN","urd"}
69 };
70
71 /* search for the presence of language from given EIT event descriptors*/
72 bool eServiceEvent::loadLanguage(Event *evt, std::string lang, int tsidonid)
73 {
74         bool retval=0;
75         for (DescriptorConstIterator desc = evt->getDescriptors()->begin(); desc != evt->getDescriptors()->end(); ++desc)
76         {
77                 switch ((*desc)->getTag())
78                 {
79                         case LINKAGE_DESCRIPTOR:
80                                 m_linkage_services.clear();
81                                 break;
82                         case SHORT_EVENT_DESCRIPTOR:
83                         {
84                                 const ShortEventDescriptor *sed = (ShortEventDescriptor*)*desc;
85                                 const std::string &cc = sed->getIso639LanguageCode();
86                                 int table=encodingHandler.getCountryCodeDefaultMapping(cc);
87                                 std::string s2;
88                                 std::transform(cc.begin(), cc.end(), s2.begin(), tolower);
89                                 if (lang.empty())
90                                         lang = s2;  // use first found language
91                                 if (lang == s2)
92                                 {
93                                         m_event_name = convertDVBUTF8(replace_all(replace_all(sed->getEventName(), "\n", " "), "\t", " "), table, tsidonid);
94                                         m_short_description = convertDVBUTF8(sed->getText(), table, tsidonid);
95                                         retval=1;
96                                 }
97                                 break;
98                         }
99                         case EXTENDED_EVENT_DESCRIPTOR:
100                         {
101                                 const ExtendedEventDescriptor *eed = (ExtendedEventDescriptor*)*desc;
102                                 const std::string &cc = eed->getIso639LanguageCode();
103                                 int table=encodingHandler.getCountryCodeDefaultMapping(cc);
104                                 std::string s2;
105                                 std::transform(cc.begin(), cc.end(), s2.begin(), tolower);
106                                 if (lang.empty())
107                                         lang = s2;  // use first found language
108                                 if (lang == s2)
109                                 {
110                                         m_extended_description += convertDVBUTF8(eed->getText(), table, tsidonid);
111                                         retval=1;
112                                 }
113 #if 0
114                                 const ExtendedEventList *itemlist = eed->getItems();
115                                 for (ExtendedEventConstIterator it = itemlist->begin(); it != itemlist->end(); ++it)
116                                 {
117                                         m_extended_description += '\n';
118                                         m_extended_description += convertDVBUTF8((*it)->getItemDescription());
119                                         m_extended_description += ' ';
120                                         m_extended_description += convertDVBUTF8((*it)->getItem());
121                                 }
122 #endif
123                                 break;
124                         }
125                         default:
126                                 break;
127                 }
128         }
129         if ( retval == 1 )
130         {
131                 for (DescriptorConstIterator desc = evt->getDescriptors()->begin(); desc != evt->getDescriptors()->end(); ++desc)
132                 {
133                         switch ((*desc)->getTag())
134                         {
135                                 case COMPONENT_DESCRIPTOR:
136                                 {
137                                         const ComponentDescriptor *cp = (ComponentDescriptor*)*desc;
138                                         eComponentData data;
139                                         data.m_streamContent = cp->getStreamContent();
140                                         data.m_componentType = cp->getComponentType();
141                                         data.m_componentTag = cp->getComponentTag();
142                                         data.m_iso639LanguageCode = cp->getIso639LanguageCode();
143                                         int table=encodingHandler.getCountryCodeDefaultMapping(data.m_iso639LanguageCode);
144                                         data.m_text = convertDVBUTF8(cp->getText(),table,tsidonid);
145                                         m_component_data.push_back(data);
146                                         break;
147                                 }
148                                 case LINKAGE_DESCRIPTOR:
149                                 {
150                                         const LinkageDescriptor  *ld = (LinkageDescriptor*)*desc;
151                                         if ( ld->getLinkageType() == 0xB0 )
152                                         {
153                                                 eServiceReference ref;
154                                                 ref.type = eServiceReference::idDVB;
155                                                 eServiceReferenceDVB &dvb_ref = (eServiceReferenceDVB&) ref;
156                                                 dvb_ref.setServiceType(1);
157                                                 dvb_ref.setTransportStreamID(ld->getTransportStreamId());
158                                                 dvb_ref.setOriginalNetworkID(ld->getOriginalNetworkId());
159                                                 dvb_ref.setServiceID(ld->getServiceId());
160                                                 const PrivateDataByteVector *privateData = ld->getPrivateDataBytes();
161                                                 dvb_ref.name = convertDVBUTF8((const unsigned char*)&((*privateData)[0]), privateData->size(), 1, tsidonid);
162                                                 m_linkage_services.push_back(ref);
163                                         }
164                                         break;
165                                 }
166                         }
167                 }
168         }
169         if ( m_extended_description.find(m_short_description) == 0 )
170                 m_short_description="";
171         return retval;
172 }
173
174 RESULT eServiceEvent::parseFrom(Event *evt, int tsidonid)
175 {
176         uint16_t stime_mjd = evt->getStartTimeMjd();
177         uint32_t stime_bcd = evt->getStartTimeBcd();
178         uint32_t duration = evt->getDuration();
179         m_begin = parseDVBtime(
180                 stime_mjd >> 8,
181                 stime_mjd&0xFF,
182                 stime_bcd >> 16,
183                 (stime_bcd >> 8)&0xFF,
184                 stime_bcd & 0xFF
185         );
186         m_event_id = evt->getEventId();
187         m_duration = fromBCD(duration>>16)*3600+fromBCD(duration>>8)*60+fromBCD(duration);
188         for (int i=0; i < MAX_LANG; i++)
189                 if (m_language==ISOtbl[i][0])
190                         if (loadLanguage(evt, ISOtbl[i][1], tsidonid))
191                                 return 0;
192         if (loadLanguage(evt, "eng", tsidonid))
193                 return 0;
194         if (loadLanguage(evt, std::string(), tsidonid))
195                 return 0;
196         return 0;
197 }
198
199 RESULT eServiceEvent::parseFrom(const std::string filename, int tsidonid)
200 {
201         if (!filename.empty())
202         {
203                 int fd = ::open( filename.c_str(), O_RDONLY );
204                 if ( fd > -1 )
205                 {
206                         __u8 buf[4096];
207                         int rd = ::read(fd, buf, 4096);
208                         ::close(fd);
209                         if ( rd > 12 /*EIT_LOOP_SIZE*/ )
210                         {
211                                 Event ev(buf);
212                                 parseFrom(&ev, tsidonid);
213                                 return 0;
214                         }
215                 }
216         }
217         return -1;
218 }
219
220 std::string eServiceEvent::getBeginTimeString() const
221 {
222         tm t;
223         localtime_r(&m_begin, &t);
224         char tmp[13];
225         snprintf(tmp, 13, "%02d.%02d, %02d:%02d",
226                 t.tm_mday, t.tm_mon+1,
227                 t.tm_hour, t.tm_min);
228         return std::string(tmp, 12);
229 }
230
231 RESULT eServiceEvent::getComponentData(ePtr<eComponentData> &dest, int tagnum) const
232 {
233         std::list<eComponentData>::const_iterator it =
234                 m_component_data.begin();
235         for(;it != m_component_data.end(); ++it)
236         {
237                 if ( it->m_componentTag == tagnum )
238                 {
239                         dest=new eComponentData(*it);
240                         return 0;
241                 }
242         }
243         dest=0;
244         return -1;
245 }
246
247 PyObject *eServiceEvent::getComponentData() const
248 {
249         ePyObject ret = PyList_New(m_component_data.size());
250         int cnt=0;
251         for (std::list<eComponentData>::const_iterator it(m_component_data.begin()); it != m_component_data.end(); ++it)
252         {
253                 ePyObject tuple = PyTuple_New(5);
254                 PyTuple_SET_ITEM(tuple, 0, PyInt_FromLong(it->m_componentTag));
255                 PyTuple_SET_ITEM(tuple, 1, PyInt_FromLong(it->m_componentType));
256                 PyTuple_SET_ITEM(tuple, 2, PyInt_FromLong(it->m_streamContent));
257                 PyTuple_SET_ITEM(tuple, 3, PyString_FromString(it->m_iso639LanguageCode.c_str()));
258                 PyTuple_SET_ITEM(tuple, 4, PyString_FromString(it->m_text.c_str()));
259                 PyList_SET_ITEM(ret, cnt++, tuple);
260         }
261         return ret;
262 }
263
264 RESULT eServiceEvent::getLinkageService(eServiceReference &service, eServiceReference &parent, int num) const
265 {
266         std::list<eServiceReference>::const_iterator it =
267                 m_linkage_services.begin();
268         while( it != m_linkage_services.end() && num-- )
269                 ++it;
270         if ( it != m_linkage_services.end() )
271         {
272                 service = *it;
273                 eServiceReferenceDVB &subservice = (eServiceReferenceDVB&) service;
274                 eServiceReferenceDVB &current = (eServiceReferenceDVB&) parent;
275                 subservice.setDVBNamespace(current.getDVBNamespace());
276                 if ( current.getParentTransportStreamID().get() )
277                 {
278                         subservice.setParentTransportStreamID( current.getParentTransportStreamID() );
279                         subservice.setParentServiceID( current.getParentServiceID() );
280                 }
281                 else
282                 {
283                         subservice.setParentTransportStreamID( current.getTransportStreamID() );
284                         subservice.setParentServiceID( current.getServiceID() );
285                 }
286                 if ( subservice.getParentTransportStreamID() == subservice.getTransportStreamID() &&
287                         subservice.getParentServiceID() == subservice.getServiceID() )
288                 {
289                         subservice.setParentTransportStreamID( eTransportStreamID(0) );
290                         subservice.setParentServiceID( eServiceID(0) );
291                 }
292                 return 0;
293         }
294         service.type = eServiceReference::idInvalid;
295         return -1;
296 }
297
298 void setServiceEventLanguage(const std::string language)
299 {
300         eServiceEvent::setEPGLanguage(language);
301 }
302
303 DEFINE_REF(eDebugClass);