change interface a bit
[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 //              eDebug("line %d, framing code: %02x, M=%02x, Y=%02x", line_offset, framing_code, M, Y);
111                 
112                 if (m_Y == 0) /* page header */
113                 {
114                         m_C = 0;
115                         
116                         m_S1 = decode_hamming_84(data + 2); /* S1 */
117                         int S2C4 = decode_hamming_84(data + 3);
118                         
119                         m_S2 = S2C4 & 7;
120                         m_C |= (S2C4 & 8) ? (1<<4) : 0;
121                         
122                         m_S3 = decode_hamming_84(data + 4);
123                         
124                         int S4C5C6 = decode_hamming_84(data + 5);
125                         
126                         m_S4 = S4C5C6 & 3;
127                         m_C |= (S4C5C6 & 0xC) << 3;
128                         
129                         m_C |= decode_hamming_84(data + 6) << 7;
130                         m_C |= decode_hamming_84(data + 7) << 11;
131                         
132                                 /* page on the same magazine? end current page. */
133                         if ((m_M == m_page_M) && (m_page_open))
134                         {
135                                 handlePageEnd();
136                                 m_page_open = 0;
137                         }
138                         
139                         m_X = decode_hamming_84(data+1) * 0x10 + decode_hamming_84(data);
140                         
141                                 /* correct page on correct magazine? open page. */
142                         if ((m_M == m_page_M) && (m_X == m_page_X))
143                         {
144                                 handlePageStart();
145                                 m_page_open = 1;
146                                 handleLine(data + 8, 32);
147                         }
148                 } else
149                 {
150                         /* data for the selected page? */
151                         if ((m_M == m_page_M) && m_page_open)
152                                 handleLine(data, 40);
153                 }
154         }
155 }
156
157 int eDVBTeletextParser::start(int pid)
158 {
159         m_page_open = 0;
160
161         if (m_pes_reader)
162                 return m_pes_reader->start(pid);
163         else
164                 return -1;
165 }
166
167 void eDVBTeletextParser::handlePageStart()
168 {
169 //      if (m_C & (1<<4)) /* erase flag set */
170
171                 /* we are always erasing the page, 
172                    even when the erase flag is not set. */
173         m_subtitle_page.clear();
174 }
175
176 void eDVBTeletextParser::handleLine(unsigned char *data, int len)
177 {
178         if (!m_Y) /* first line is page header, we don't need that. */
179                 return;
180
181         int last_was_white = 1, color = -1;
182         
183         std::string text;
184         
185         for (int i=0; i<len; ++i)
186         {
187                 unsigned char b = decode_odd_parity(data + i);
188         
189                 if (b < 0x10) /* spacing attribute */
190                 {
191                         if (b < 8) /* colors */
192                         {
193                                 if (b != color) /* new color is split into a new string */
194                                 {
195                                         if (color != -1)
196                                                 addSubtitleString(color, text);
197                                         text = "";
198                                         color = b;
199                                 }
200                         }
201                                 /* ignore other attributes */
202                 } else
203                 {
204                                 /* no more than one whitespace, only printable chars */
205                         if (((!last_was_white) || (b != ' ')) && (b >= 0x20))
206                         {
207                                 text += b;
208                                 last_was_white = b == ' ';
209                         }
210                 }
211         }
212         addSubtitleString(color, text);
213 }
214
215 void eDVBTeletextParser::handlePageEnd()
216 {
217         addSubtitleString(-1, ""); /* end last line */ 
218         sendSubtitlePage();  /* send assembled subtitle page to display */
219 }
220
221 void eDVBTeletextParser::setPage(int page)
222 {
223         m_page_M = (page >> 8) & 7; /* magazine to look for */
224         m_page_X = page & 0xFF;     /* page number */
225 }
226
227 void eDVBTeletextParser::connectNewPage(const Slot1<void, const eDVBTeletextSubtitlePage&> &slot, ePtr<eConnection> &connection)
228 {
229         connection = new eConnection(this, m_new_subtitle_page.connect(slot));
230 }
231
232 void eDVBTeletextParser::addSubtitleString(int color, const std::string &string)
233 {
234         gRGB rgbcol((color & 1) ? 255 : 128, (color & 2) ? 255 : 128, (color & 4) ? 255 : 128);
235         if ((color != m_subtitle_color) && !m_subtitle_text.empty())
236         {
237                 m_subtitle_page.m_elements.push_back(eDVBTeletextSubtitlePageElement(rgbcol, m_subtitle_text));
238                 m_subtitle_text = "";
239         } else if (!m_subtitle_text.empty() && m_subtitle_text[m_subtitle_text.size()-1] != ' ')
240                 m_subtitle_text += " ";
241         
242         m_subtitle_color = color;
243         m_subtitle_text += string;
244 }
245
246 void eDVBTeletextParser::sendSubtitlePage()
247 {
248         eDebug("subtitle page:");
249         for (unsigned int i = 0; i < m_subtitle_page.m_elements.size(); ++i)
250                 eDebug("%s", m_subtitle_page.m_elements[i].m_text.c_str());
251         m_new_subtitle_page(m_subtitle_page);
252 }