fix some bugs, collect subtitle pages
[enigma2.git] / lib / dvb / teletext.cpp
1 #include <lib/base/eerror.h>
2 #include <lib/dvb/teletext.h>
3 #include <lib/dvb/idemux.h>
4 #include <lib/gdi/gpixmap.h>
5
6 /*
7  This is a very simple en300 706 telext decoder.
8  
9  It can only decode a single page at a time, thus it's only used
10  for subtitles.
11  
12  */
13
14 DEFINE_REF(eDVBTeletextParser);
15
16         /* we asumme error free transmission! */
17 static inline unsigned char decode_odd_parity(unsigned char *b)
18 {
19         int i;
20         unsigned char res = 0;
21         for (i=0; i<7; ++i)
22                 if (*b & (0x80 >> i))
23                         res |= 1<<i;
24         return res;
25 }
26
27 static inline unsigned char decode_hamming_84(unsigned char *b)
28 {
29         unsigned char res = 0;
30         res |= (*b << 3) & 8;
31         res |= (*b     ) & 4;
32         res |= (*b >> 3) & 2;
33         res |= (*b >> 6) & 1;
34         return res;
35 }
36
37 static inline unsigned long decode_hamming_2418(unsigned char *b)
38 {
39         unsigned long h24 = b[0] | (b[1] << 8) | (b[2] << 16);
40         
41         return
42                 ((h24 & 0x000004) >> 2) |
43                 ((h24 & 0x000070) >> 3) |
44                 ((h24 & 0x007f00) >> 4) |
45                 ((h24 & 0x7f0000) >> 5);
46 }
47
48 eDVBTeletextParser::eDVBTeletextParser(iDVBDemux *demux)
49 {
50         setStreamID(0xBD); /* as per en 300 472 */
51         
52         setPage(0x150);
53         
54         if (demux->createPESReader(eApp, m_pes_reader))
55                 eDebug("failed to create PES reader!");
56         else
57                 m_pes_reader->connectRead(slot(*this, &eDVBTeletextParser::processData), m_read_connection);
58 }
59
60 eDVBTeletextParser::~eDVBTeletextParser()
61 {
62 }
63
64 void eDVBTeletextParser::processPESPacket(__u8 *pkt, int len)
65 {
66         unsigned char *p = pkt;
67         
68         p += 4; len -= 4; /* start code, already be verified by pes parser */
69         p += 2; len -= 2; /* length, better use the argument */ 
70         
71         p += 3; len -= 3; /* pes header */
72         
73         p += 0x24; len -= 0x24; /* skip header */
74         
75 //      eDebug("data identifier: %02x", *p);
76         
77         p++; len--;
78         
79         while (len > 2)
80         {
81                 unsigned char data_unit_id = *p++;
82                 unsigned char data_unit_length = *p++;
83                 len -= 2;
84                 
85                 if (len < data_unit_length)
86                 {
87                         eDebug("data_unit_length > len");
88                         break;
89                 }
90                 
91                 if (data_unit_length != 44)
92                 {
93                         eDebug("illegal data unit length %d", data_unit_length);
94                         break;
95                 }
96                 
97                 unsigned char line_offset = *p++; len--;
98                 unsigned char framing_code = *p++; len--;
99
100                 int magazine_and_packet_address = decode_hamming_84(p++); len--;
101                 magazine_and_packet_address |= decode_hamming_84(p++)<<4; len--;
102                 
103                 unsigned char *data = p; p += 40; len -= 40;
104                 
105                 if (framing_code != 0xe4) /* no teletxt data */
106                         continue;
107                 
108                 m_M = magazine_and_packet_address & 7;
109                 m_Y = magazine_and_packet_address >> 3;
110
111 //                      eDebug("line %d, framing code: %02x, M=%02x, Y=%02x", line_offset, framing_code, m_M, m_Y);
112                 
113                 if (m_Y == 0) /* page header */
114                 {
115                         m_C = 0;
116                         
117                         m_S1 = decode_hamming_84(data + 2); /* S1 */
118                         int S2C4 = decode_hamming_84(data + 3);
119                         
120                         m_S2 = S2C4 & 7;
121                         m_C |= (S2C4 & 8) ? (1<<4) : 0;
122                         
123                         m_S3 = decode_hamming_84(data + 4);
124                         
125                         int S4C5C6 = decode_hamming_84(data + 5);
126                         
127                         m_S4 = S4C5C6 & 3;
128                         m_C |= (S4C5C6 & 0xC) << 3;
129                         
130                         m_C |= decode_hamming_84(data + 6) << 7;
131                         m_C |= decode_hamming_84(data + 7) << 11;
132                         
133                         int serial_mode = m_C & (1<<11);
134                         
135                                 /* page on the same magazine? end current page. */
136                         if ((serial_mode || (m_M == m_page_M)) && (m_page_open))
137                         {
138                                 handlePageEnd();
139                                 m_page_open = 0;
140                         }
141                         
142                         m_X = decode_hamming_84(data+1) * 0x10 + decode_hamming_84(data);
143                         
144                         if (m_C & (1<<6))
145                                 m_found_subtitle_pages.insert((m_M << 8) | m_X);
146                         
147                                 /* correct page on correct magazine? open page. */
148                         if ((m_M == m_page_M) && (m_X == m_page_X))
149                         {
150                                 handlePageStart();
151                                 m_page_open = 1;
152                                 handleLine(data + 8, 32);
153                         }
154                 } else
155                 {
156                         /* data for the selected page? */
157                         if ((m_M == m_page_M) && m_page_open)
158                                 handleLine(data, 40);
159                 }
160         }
161 }
162
163 int eDVBTeletextParser::start(int pid)
164 {
165         m_page_open = 0;
166
167         if (m_pes_reader)
168                 return m_pes_reader->start(pid);
169         else
170                 return -1;
171 }
172
173 void eDVBTeletextParser::handlePageStart()
174 {
175 //      if (m_C & (1<<4)) /* erase flag set */
176
177                 /* we are always erasing the page, 
178                    even when the erase flag is not set. */
179         m_subtitle_page.clear();
180 }
181
182 void eDVBTeletextParser::handleLine(unsigned char *data, int len)
183 {
184         if (!m_Y) /* first line is page header, we don't need that. */
185                 return;
186
187         int last_was_white = 1, color = 7; /* start with whitespace. start with color=white. (that's unrelated.) */
188         
189         std::string text;
190         
191 //      eDebug("handle subtitle line: %d len", len);
192         for (int i=0; i<len; ++i)
193         {
194                 unsigned char b = decode_odd_parity(data + i);
195         
196                 if (b < 0x10) /* spacing attribute */
197                 {
198                         if (b < 8) /* colors */
199                         {
200                                 if (b != color) /* new color is split into a new string */
201                                 {
202                                         addSubtitleString(color, text);
203                                         text = "";
204                                         color = b;
205                                 }
206                         }
207                                 /* ignore other attributes */
208                 } else
209                 {
210                         eDebugNoNewLine("%c", b);
211                                 /* no more than one whitespace, only printable chars */
212                         if (((!last_was_white) || (b != ' ')) && (b >= 0x20))
213                         {
214                                 text += b;
215                                 last_was_white = b == ' ';
216                         }
217                 }
218         }
219         eDebug("");
220         addSubtitleString(color, text);
221 }
222
223 void eDVBTeletextParser::handlePageEnd()
224 {
225         eDebug("handle page end");
226         addSubtitleString(-1, ""); /* end last line */ 
227         sendSubtitlePage();  /* send assembled subtitle page to display */
228 }
229
230 void eDVBTeletextParser::setPage(int page)
231 {
232         m_page_M = (page >> 8) & 7; /* magazine to look for */
233         m_page_X = page & 0xFF;     /* page number */
234 }
235
236 void eDVBTeletextParser::connectNewPage(const Slot1<void, const eDVBTeletextSubtitlePage&> &slot, ePtr<eConnection> &connection)
237 {
238         connection = new eConnection(this, m_new_subtitle_page.connect(slot));
239 }
240
241 void eDVBTeletextParser::addSubtitleString(int color, const std::string &string)
242 {
243 //      eDebug("add subtitle string: %s, col %d", string.c_str(), color);
244         gRGB rgbcol((color & 1) ? 255 : 128, (color & 2) ? 255 : 128, (color & 4) ? 255 : 128);
245         if ((color != m_subtitle_color) && !m_subtitle_text.empty())
246         {
247                 m_subtitle_page.m_elements.push_back(eDVBTeletextSubtitlePageElement(rgbcol, m_subtitle_text));
248                 m_subtitle_text = "";
249         } else if (!m_subtitle_text.empty() && m_subtitle_text[m_subtitle_text.size()-1] != ' ')
250                 m_subtitle_text += " ";
251         
252         m_subtitle_color = color;
253         m_subtitle_text += string;
254 }
255
256 void eDVBTeletextParser::sendSubtitlePage()
257 {
258         eDebug("subtitle page:");
259         for (unsigned int i = 0; i < m_subtitle_page.m_elements.size(); ++i)
260                 eDebug("%s", m_subtitle_page.m_elements[i].m_text.c_str());
261         m_new_subtitle_page(m_subtitle_page);
262 }