2 Copyright 2005-2008 Mark Calderbank
3 Copyright 2007 Marten Richter (AC3 support)
5 This file is part of VOMP.
7 VOMP is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
12 VOMP is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with VOMP; if not, write to the Free Software Foundation, Inc.,
19 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
25 #include "dvbsubtitles.h"
32 #define DEMUXER_SEQ_HEAD 0x000001B3
33 #define DEMUXER_PIC_HEAD 0x00000101
35 #define DEMUXER_H264_ACCESS_UNIT 0x00000109
36 #define DEMUXER_H264_SEQ_PARAMETER_SET 0x00000107
39 #define SEEK_THRESHOLD 150000 // About 1.5 seconds
42 const int Demuxer::FrameRates[9] = { 0, 23, 24, 25, 29, 30, 50, 59, 60 };
43 Demuxer* Demuxer::instance = NULL;
47 NALUUnit(const UCHAR* buf,UINT length_buf);
50 inline UINT getBits(UINT num_bits);
53 bool isEonalu() {return eonalu;};
65 NALUUnit::NALUUnit(const UCHAR *buf, UINT length_buf)
77 UINT pattern =(((UINT)buf[ 0] << 16) |
81 while (pattern != 0x000001)
83 if (++nalu_start >= length_buf) return;
84 pattern = ((pattern << 8) | buf[nalu_start])&0x00FFFFFF;
86 nalu_end=nalu_start+1;
87 pattern = ((pattern << 8) | buf[nalu_end])&0x00FFFFFF;
89 while (pattern != 0x000001 && pattern != 0x000000)
91 if (++nalu_end >= length_buf) { nalu_end+=3;break;};
92 pattern = ((pattern << 8) | buf[nalu_end])&0x00FFFFFF;
95 nalu_end=min(length_buf-1,nalu_end);
96 nalu_length=nalu_end-nalu_start;
97 nalu_buf=(UCHAR*)malloc(nalu_length);
98 memcpy(nalu_buf,buf+nalu_start,nalu_length);
102 NALUUnit::~NALUUnit()
104 if (nalu_buf) free(nalu_buf);
107 inline UINT NALUUnit::getBits(UINT num_bits)
109 if (num_bits==0) return 0; //???
110 UINT remain_bits=num_bits;
112 //May be slow, but should work!
113 while (remain_bits>0) {
117 last_bytes=(last_bytes<<8) & nalu_buf[pos];
118 if ((last_bytes & 0x00FFFFFF) == 0x000003) pos++; //emulation prevention byte
121 working_byte=nalu_buf[pos];
137 UINT fetch_bits=min(remain_bits,8-bit_pos);
138 work=work <<fetch_bits;
139 //work|=((working_byte>>bit_pos) & (0xFF>>(8-fetch_bits)));
140 work|=(working_byte &(0xFF>>(bit_pos)))>>(8-fetch_bits-bit_pos);
141 remain_bits-=fetch_bits;
142 bit_pos=(bit_pos+fetch_bits)%8;
147 UINT NALUUnit::getUe()
151 for( bit = 0; !bit && !eonalu; leadbits++ )
153 if (eonalu) return true;
154 return ((1 << leadbits)-1)+getBits(leadbits);
157 int NALUUnit::getSe()
160 if (input==0) return 0;
161 int output=((input+1)>>1);
162 if (input & 0x1) output*=-1;
168 static const int PESPacket_initial_size = 2000;
171 PESPacket::PESPacket()
173 data_size = PESPacket_initial_size;
174 data = (UCHAR*)malloc(data_size);
181 PESPacket::PESPacket(const PESPacket& packet)
186 PESPacket& PESPacket::operator=(const PESPacket& packet)
190 if (data) free(data);
196 PESPacket::~PESPacket()
198 if (data) free(data);
201 void PESPacket::copyFrom(const PESPacket& packet)
203 length = packet.length;
205 packetType = packet.packetType;
206 substream = packet.substream;
207 seq_header = packet.seq_header;
209 data = (UCHAR*)malloc(data_size);
210 memcpy(data, packet.data, data_size);
213 void PESPacket::init(UCHAR type, UCHAR sub)
217 data[4] = data[5] = 0;
221 seq_header = 1; // Unknown seq_header status
224 void PESPacket::truncate()
226 init(packetType,substream);
229 int PESPacket::write(const UCHAR *buf, int len)
231 if (size + len > 0x10000) return 0;
232 if (size + len > data_size)
234 UINT new_data_size = max(data_size + data_size / 2, data_size + len);
235 if (new_data_size > 0x10000) new_data_size = 0x10000;
236 data_size = new_data_size;
237 data = (UCHAR*)realloc(data, data_size);
239 memcpy(data + size, buf, len);
242 data[4] = (length >> 8);
243 data[5] = (length & 0xFF);
244 // We have added data - reset seq_header indicator if necessary
245 if (seq_header == 0) seq_header = 1; // Reset to 'unknown'
249 ULLONG PESPacket::getPTS() const
251 if ( ( (packetType >= Demuxer::PESTYPE_AUD0 &&
252 packetType <= Demuxer::PESTYPE_AUDMAX)
254 (packetType >= Demuxer::PESTYPE_VID0 &&
255 packetType <= Demuxer::PESTYPE_VIDMAX)
257 packetType == Demuxer::PESTYPE_PRIVATE_1
259 && size >= 14 && data[7] & 0x80)
261 return ( (ULLONG)(data[ 9] & 0x0E) << 29) |
262 ( (ULLONG)(data[10]) << 22 ) |
263 ( (ULLONG)(data[11] & 0xFE) << 14 ) |
264 ( (ULLONG)(data[12]) << 7 ) |
265 ( (ULLONG)(data[13] & 0xFE) >> 1 );
267 else return PTS_INVALID;
270 UCHAR PESPacket::operator[] (UINT index) const
278 UINT PESPacket::findPictureHeader(bool h264) const
280 if (size < 12) return 0;
281 UINT pattern = ( ((UINT)data[ 8] << 24) |
282 ((UINT)data[ 9] << 16) |
283 ((UINT)data[10] << 8) |
288 while (pattern != DEMUXER_H264_ACCESS_UNIT)
290 if (++pos >= size) return 0;
291 pattern = (pattern << 8) | data[pos];
295 while (pattern != DEMUXER_PIC_HEAD)
297 if (++pos >= size) return 0;
298 pattern = (pattern << 8) | data[pos];
304 UINT PESPacket::countPictureHeaders(bool h264) const
306 if (size < 12) return 0;
307 UINT pattern = ( ((UINT)data[ 8] << 24) |
308 ((UINT)data[ 9] << 16) |
309 ((UINT)data[10] << 8) |
318 pattern = (pattern << 8) | data[pos];
319 if (pattern==DEMUXER_H264_ACCESS_UNIT) count++;
326 pattern = (pattern << 8) | data[pos];
327 if (pattern==DEMUXER_PIC_HEAD) count++;
333 UINT PESPacket::findSeqHeader(bool h264) const
335 if (seq_header != 1) return seq_header;
336 if (size < 12) return 0;
337 UINT pattern = ( ((UINT)data[ 8] << 24) |
338 ((UINT)data[ 9] << 16) |
339 ((UINT)data[10] << 8) |
343 while ((pattern & 0xFFFFFF1F) != DEMUXER_H264_SEQ_PARAMETER_SET)
350 pattern = (pattern << 8) | data[pos];
352 seq_header = pos - 3;
356 while (pattern != DEMUXER_SEQ_HEAD)
363 pattern = (pattern << 8) | data[pos];
365 seq_header = pos - 3;
373 if (instance) return;
378 vid_seeking = aud_seeking = false;
379 video_pts = audio_pts = 0;
380 ispre_1_3_19 = false;
392 Demuxer* Demuxer::getInstance()
397 int Demuxer::init(Callback* tcallback, DrainTarget* audio, DrainTarget* video, DrainTarget* teletext,
398 ULONG demuxMemoryV, ULONG demuxMemoryA, ULONG demuxMemoryT,double infps, DVBSubtitles* tsubtitles)
402 if ( !videostream.init(video, demuxMemoryV) ||
403 !audiostream.init(audio, demuxMemoryA) ||
404 !teletextstream.init(teletext, demuxMemoryT))
406 Log::getInstance()->log("Demuxer", Log::CRIT,
407 "Failed to initialize demuxer");
413 isteletextdecoded = true;
415 isteletextdecoded = false;
420 subtitles = tsubtitles;
421 callback = tcallback;
425 void Demuxer::reset()
427 Log::getInstance()->log("Demuxer", Log::DEBUG, "Reset called");
429 video_current = audio_current = teletext_current = subtitle_current = -1;
430 horizontal_size = vertical_size = 0;
431 aspect_ratio = (enum AspectRatio) 0;
432 frame_rate = bit_rate = 0;
433 ispre_1_3_19 = false;
437 for (int i = 0; i <= (PESTYPE_AUDMAX - PESTYPE_AUD0); i++)
439 avail_mpaudchan[i] = false;
441 for (int i = 0; i <= (PESTYPE_SUBSTREAM_AC3MAX - PESTYPE_SUBSTREAM_AC30); i++)
443 avail_ac3audchan[i] = false;
445 for (int i = 0; i <= (PESTYPE_SUBSTREAM_DVBSUBTITLEMAX - PESTYPE_SUBSTREAM_DVBSUBTITLE0); i++)
447 avail_dvbsubtitlechan[i] = false;
451 int Demuxer::shutdown()
453 videostream.shutdown();
454 audiostream.shutdown();
455 teletextstream.shutdown();
460 void Demuxer::flush()
462 Log::getInstance()->log("Demuxer", Log::DEBUG, "Flush called");
466 teletextstream.flush();
469 void Demuxer::flushAudio()
476 vid_seeking = aud_seeking = true;
477 video_pts = audio_pts = teletext_pts = 0;
480 void Demuxer::setAudioStream(int id)
485 void Demuxer::setVideoStream(int id)
490 void Demuxer::setTeletextStream(int id)
492 teletext_current = id;
495 void Demuxer::setDVBSubtitleStream(int id)
497 subtitle_current = id;
500 void Demuxer::setAspectRatio(enum AspectRatio ar)
502 if (aspect_ratio != ar)
504 Log::getInstance()->log("Demux", Log::DEBUG,
505 "Aspect ratio difference signalled");
506 if (++arcnt > 3) // avoid changing aspect ratio if glitch in signal
510 if (callback) callback->call(this);
517 bool Demuxer::writeAudio()
519 return audiostream.drain();
522 bool Demuxer::writeVideo()
524 return videostream.drain();
527 bool Demuxer::writeTeletext()
529 return teletextstream.drain();
532 bool Demuxer::submitPacket(PESPacket& packet)
535 UCHAR packet_type = packet.getPacketType();
536 const UCHAR* packetdata = packet.getData();
537 if (packet_type >= PESTYPE_VID0 && packet_type <= PESTYPE_VIDMAX)
539 if (video_current == -1) video_current = packet_type;
540 if (video_current == packet_type && !vid_seeking)
542 sent = videostream.put(&packetdata[0], packet.getSize(), h264?MPTYPE_VIDEO_H264:MPTYPE_VIDEO_MPEG2,packetnum);
543 if (sent) packetnum++;
546 sent = packet.getSize();
548 else if (packet_type >= PESTYPE_AUD0 && packet_type <= PESTYPE_AUDMAX)
550 if (audio_current == -1) audio_current = packet_type;
551 avail_mpaudchan[packet_type - PESTYPE_AUD0] = true;
552 if (audio_current == packet_type && !aud_seeking)
554 sent = audiostream.put(&packetdata[0], packet.getSize(), MPTYPE_MPEG_AUDIO,packetnum);
555 if (sent) packetnum++;
558 sent = packet.getSize();
560 else if (packet_type == PESTYPE_PRIVATE_1 &&
561 packet.getSubstream() >= PESTYPE_SUBSTREAM_AC30 &&
562 packet.getSubstream() <= PESTYPE_SUBSTREAM_AC3MAX)
564 avail_ac3audchan[packet.getSubstream() - PESTYPE_SUBSTREAM_AC30] = true;
565 if (packet.getSubstream() == audio_current)
567 sent = audiostream.put(&packetdata[0], packet.getSize(), (ispre_1_3_19)? MPTYPE_AC3_PRE13 : MPTYPE_AC3,packetnum);
568 if (sent) packetnum++;
572 sent = packet.getSize();
575 else if (packet_type == PESTYPE_PRIVATE_1 &&
576 packet.getSubstream() >= PESTYPE_SUBSTREAM_DVBSUBTITLE0 &&
577 packet.getSubstream() <= PESTYPE_SUBSTREAM_DVBSUBTITLEMAX)
579 avail_dvbsubtitlechan[packet.getSubstream()-PESTYPE_SUBSTREAM_DVBSUBTITLE0]=true;
580 if (subtitle_current == -1) subtitle_current = packet.getSubstream();
581 if (subtitles && packet.getSubstream()==subtitle_current)
583 subtitles->put(packet);
585 sent = packet.getSize();
587 else if (isteletextdecoded && packet_type == PESTYPE_PRIVATE_1 &&
588 packet.getSubstream() >= PESTYPE_SUBSTREAM_TELETEXT0 &&
589 packet.getSubstream() <= PESTYPE_SUBSTREAM_TELETEXTMAX)
592 if (teletext_current == -1) teletext_current = packet.getSubstream();
593 if (teletext_current == packet.getSubstream() )
595 sent = teletextstream.put(&packetdata[0], packet.getSize(), MPTYPE_TELETEXT,packetnum);
599 sent = packet.getSize();
604 sent = packet.getSize();
607 if (sent < packet.getSize()) // Stream is full.
613 void Demuxer::parsePacketDetails(PESPacket& packet)
615 if (packet.getPacketType() >= PESTYPE_AUD0 &&
616 packet.getPacketType() <= PESTYPE_AUDMAX)
618 // Extract audio PTS if it exists
621 audio_pts = packet.getPTS();
622 // We continue to seek on the audio if the video PTS that we
623 // are trying to match is ahead of the audio PTS by at most
624 // SEEK_THRESHOLD. We consider the possibility of PTS wrap.
625 if (aud_seeking && !vid_seeking &&
626 !( (video_pts_seek > audio_pts &&
627 video_pts_seek - audio_pts < SEEK_THRESHOLD)
629 (video_pts_seek < audio_pts &&
630 video_pts_seek + (1LL<<33) - audio_pts < SEEK_THRESHOLD) ))
633 Log::getInstance()->log("Demuxer", Log::DEBUG,
634 "Leaving audio sync: Audio PTS = %llu", audio_pts);
638 else if (packet.getPacketType() == PESTYPE_PRIVATE_1) // Private stream
640 //Inspired by vdr's device.c
641 int payload_begin = packet[8]+9;
642 unsigned char substream_id = packet[payload_begin];
643 unsigned char substream_type = substream_id & 0xF0;
644 unsigned char substream_index = substream_id & 0x1F;
645 pre_1_3_19_Recording: //This is for old recordings stuff and live TV
648 substream_id = PESTYPE_PRIVATE_1;
649 substream_type = 0x80;
652 switch (substream_type)
656 packet.setSubstream(substream_id);
658 case 0xA0: //LPCM //not supported yet, is there any LPCM transmissio out there?
660 case 0x80: //ac3, currently only one ac3 track per recording supported
661 packet.setSubstream(substream_type+substream_index);
663 // Extract audio PTS if it exists
666 audio_pts = packet.getPTS();
667 // We continue to seek on the audio if the video PTS that we
668 // are trying to match is ahead of the audio PTS by at most
669 // SEEK_THRESHOLD. We consider the possibility of PTS wrap.
670 if (aud_seeking && !vid_seeking &&
671 !( (video_pts_seek > audio_pts &&
672 video_pts_seek - audio_pts < SEEK_THRESHOLD)
674 (video_pts_seek < audio_pts &&
675 video_pts_seek + (1LL<<33) - audio_pts < SEEK_THRESHOLD) ))
678 Log::getInstance()->log("Demuxer", Log::DEBUG, "Leaving audio sync: Audio PTS = %llu", audio_pts);
682 case 0x10: //Teletext Is this correct?
683 packet.setSubstream(substream_id);
684 // Extract teletxt PTS if it exists
687 teletext_pts = packet.getPTS();
693 ispre_1_3_19=true; //switching to compat mode and live tv mode
694 goto pre_1_3_19_Recording;
698 packet.setSubstream(0);
703 else if (packet.getPacketType() >= PESTYPE_VID0 &&
704 packet.getPacketType() <= PESTYPE_VIDMAX)
706 // Extract video PTS if it exists
707 if (packet.hasPTS()) video_pts = packet.getPTS();
709 // If there is a sequence header, extract information
710 UINT pos = packet.findSeqHeader(h264);
715 if (pos+6 >= packet.getSize()) return;
716 horizontal_size = ((int)packet[pos] << 4) | ((int)packet[pos+1] >> 4);
718 vertical_size = (((int)packet[pos+1] & 0xf) << 8) | (int)packet[pos+2];
720 setAspectRatio((enum AspectRatio)(packet[pos+3] >> 4));
721 frame_rate = packet[pos+3] & 0x0f;
722 if (frame_rate >= 1 && frame_rate <= 8)
723 frame_rate = FrameRates[frame_rate];
726 bit_rate = ((int)packet[pos+4] << 10) |
727 ((int)packet[pos+5] << 2) |
728 ((int)packet[pos+6] >> 6);
732 /* Chris and Mark I know this is ugly, should we move this to a method of PESPacket or to NALUUnit what would be better?
733 This looks so ugly since the header includes variable length parts and I have to parse through the whole header to get the wanted information*/
734 NALUUnit nalu(packet.getData()+pos,packet.getSize()-pos);
735 profile=nalu.getBits(8);
736 nalu.getBits(8); //constraints
737 nalu.getBits(8); //level_idc
738 nalu.getUe(); //seq_parameter_set_id
740 if (profile==100 || profile==110 || profile==122 || profile==144)
747 nalu.getUe(); //bit depth lume
748 nalu.getUe(); //bit depth chrome
752 for (int i=0;i<8;i++){
759 for (int j=0;j<16;j++) {
761 UINT delta=nalu.getSe();
762 nextscale=(lastscale+delta+256)%256;
764 lastscale=(nextscale==0)?lastscale:nextscale;
771 for (int j=0;j<64;j++) {
773 UINT delta=nalu.getSe();
774 nextscale=(lastscale+delta+256)%256;
776 lastscale=(nextscale==0)?lastscale:nextscale;
787 chromunitx=chromunity=1; break;
789 chromunitx=chromunity=2; break;
791 chromunitx=2;chromunity=1; break;
793 chromunitx=chromunity=1; break;
796 nalu.getUe(); //log2framenum
797 UINT temp=nalu.getUe();
798 if (temp==0) //pict order
804 UINT temp2=nalu.getUe();
805 for (int i=0;i<temp2;i++)
808 nalu.getUe(); //Num refframes
810 horizontal_size=(nalu.getUe()+1)*16;
812 vertical_size=(nalu.getUe()+1)*16;
813 int interlaced=nalu.getBits(1);
814 vertical_size*=(2-interlaced);
816 if (!interlaced) nalu.getBits(1);
820 horizontal_size-=nalu.getUe()*chromunitx;
821 horizontal_size-=nalu.getUe()*chromunitx;
822 vertical_size-=nalu.getUe()*(2-interlaced)*chromunity;
823 vertical_size-=nalu.getUe()*(2-interlaced)*chromunity;
829 UINT aspectratioidc=nalu.getBits(8);
830 bool hasaspect=false;
831 const float aspects[]={1., 1./1.,12./11.,10./11.,16./11.,40./33.,
832 24./11.,20./11.,32./11.,80./33.,18./11.,15./11.,64./33.,160./99.,4./3.,3./2.,2./1.};
834 float aspectratio=((float) horizontal_size)/((float) vertical_size);
835 if (aspectratioidc<=16)
838 aspectratio*=aspects[aspectratioidc];
841 else if (aspectratioidc==255)
843 int t_sar_width=nalu.getBits(16);
844 int t_sar_height=nalu.getBits(16);
845 if (t_sar_width!=0 && t_sar_height!=0)
848 aspectratio*=((float)t_sar_width)/((float)t_sar_height);
853 if (fabs(aspectratio-16./9.)<0.1) setAspectRatio(ASPECT_16_9);
854 else if (fabs(aspectratio-4./3.)<0.1) setAspectRatio(ASPECT_4_3);
865 video_pts_seek = video_pts;
866 Log::getInstance()->log("Demuxer", Log::DEBUG,
867 "Entering audio sync: Video PTS = %llu", video_pts);
868 Log::getInstance()->log("Demuxer", Log::DEBUG,
869 "Entering audio sync: Audio PTS = %llu", audio_pts);
876 UINT Demuxer::stripAudio(UCHAR* buf, UINT len)
878 UINT read_pos = 0, write_pos = 0;
879 UINT pattern, packet_length;
880 if (len < 4) return 0;
881 pattern = (buf[0] << 16) | (buf[1] << 8) | (buf[2]);
882 while (read_pos + 7 <= len)
884 pattern = ((pattern & 0xFFFFFF) << 8) | buf[read_pos+3];
885 if (pattern < (0x100|PESTYPE_VID0) || pattern > (0x100|PESTYPE_VIDMAX))
889 packet_length = ((buf[read_pos+4] << 8) | (buf[read_pos+5])) + 6;
890 if (read_pos + packet_length > len)
894 if (read_pos != write_pos)
895 memmove(buf+write_pos, buf+read_pos, packet_length);
896 read_pos += packet_length;
897 write_pos += packet_length;
898 pattern = (buf[read_pos] << 16) | (buf[read_pos+1] << 8)
906 bool Demuxer::scanForVideo(UCHAR* buf, UINT len, bool &ish264)
911 if (len < 4) return false;
912 pattern = (buf[0] << 16) | (buf[1] << 8) | (buf[2]);
915 pattern = ((pattern & 0xFFFFFF) << 8) | buf[pos++];
916 if (pattern >= (0x100|PESTYPE_VID0) && pattern <= (0x100|PESTYPE_VIDMAX))
922 bool* Demuxer::getmpAudioChannels()
924 return avail_mpaudchan;
927 bool* Demuxer::getac3AudioChannels()
929 return avail_ac3audchan;
932 bool* Demuxer::getSubtitleChannels()
934 return avail_dvbsubtitlechan;
937 int Demuxer::getselSubtitleChannel()
939 return subtitle_current;
942 int Demuxer::getselAudioChannel()
944 return audio_current;