mirrors/neutrino
1
0
Fork 0
mirror of https://github.com/tuxbox-neutrino/neutrino.git synced 2025-08-28 16:01:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

driver/record.cpp, driver/streamts.cpp: add more formats, for testing

Browse Source
This commit is contained in:
[CST] Focus
2016-03-12 14:53:11 +03:00
parent 5c49d9293a
commit f3a2657c9e
4 changed files with 66 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/driver/record.cpp
View File

@@ -1882,9 +1882,11 @@ void CStreamRec::Close()
} }
avformat_free_context(ofcx); avformat_free_context(ofcx);
} }
if (bsfc)
av_bitstream_filter_close(bsfc);
ifcx = NULL; ifcx = NULL;
ofcx = NULL; ofcx = NULL;
bsfc = NULL;
} }
void CStreamRec::GetPids(CZapitChannel * channel) void CStreamRec::GetPids(CZapitChannel * channel)
@@ -2085,7 +2087,11 @@ bool CStreamRec::Open(CZapitChannel * channel)
printf("%s: Cannot find stream info [%s]!\n", __FUNCTION__, channel->getUrl().c_str()); printf("%s: Cannot find stream info [%s]!\n", __FUNCTION__, channel->getUrl().c_str());
return false; return false;
} }
if (!strstr(ifcx->iformat->name, "applehttp") && !strstr(ifcx->iformat->name, "mpegts")) { if (!strstr(ifcx->iformat->name, "applehttp") &&
!strstr(ifcx->iformat->name, "mpegts") &&
!strstr(ifcx->iformat->name, "matroska") &&
!strstr(ifcx->iformat->name, "avi") &&
!strstr(ifcx->iformat->name, "mp4")) {
printf("%s: not supported format [%s]!\n", __FUNCTION__, ifcx->iformat->name); printf("%s: not supported format [%s]!\n", __FUNCTION__, ifcx->iformat->name);
return false; return false;
} }
@@ -2132,6 +2138,9 @@ bool CStreamRec::Open(CZapitChannel * channel)
av_log_set_level(AV_LOG_VERBOSE); av_log_set_level(AV_LOG_VERBOSE);
av_dump_format(ofcx, 0, ofcx->filename, 1); av_dump_format(ofcx, 0, ofcx->filename, 1);
av_log_set_level(AV_LOG_WARNING); av_log_set_level(AV_LOG_WARNING);
bsfc = av_bitstream_filter_init("h264_mp4toannexb");
if (!bsfc)
printf("%s: av_bitstream_filter_init h264_mp4toannexb failed!\n", __FUNCTION__);
return true; return true;
} }
@@ -2156,6 +2165,25 @@ void CStreamRec::run()
break; break;
if (pkt.stream_index < 0) if (pkt.stream_index < 0)
continue; continue;
AVCodecContext *codec = ifcx->streams[pkt.stream_index]->codec;
if (bsfc && codec->codec_id == CODEC_ID_H264) {
AVPacket newpkt = pkt;
int len;
if ((len = av_bitstream_filter_filter(bsfc, codec, NULL, &newpkt.data, &newpkt.size, pkt.data, pkt.size, pkt.flags & AV_PKT_FLAG_KEY)) < 0) {
av_free_packet(&pkt);
continue;
}
av_free_packet(&pkt);
newpkt.buf = av_buffer_create(newpkt.data, newpkt.size, av_buffer_default_free, NULL, 0);
pkt = newpkt;
}
pkt.pts = av_rescale_q(pkt.pts, ifcx->streams[pkt.stream_index]->time_base, ofcx->streams[pkt.stream_index]->time_base);
pkt.dts = av_rescale_q(pkt.dts, ifcx->streams[pkt.stream_index]->time_base, ofcx->streams[pkt.stream_index]->time_base);
av_write_frame(ofcx, &pkt); av_write_frame(ofcx, &pkt);
av_free_packet(&pkt); av_free_packet(&pkt);

1
src/driver/record.h
View File

@@ -242,6 +242,7 @@ class CStreamRec : public CRecordInstance, OpenThreads::Thread
private: private:
AVFormatContext *ifcx; AVFormatContext *ifcx;
AVFormatContext *ofcx; AVFormatContext *ofcx;
AVBitStreamFilterContext *bsfc;
bool stopped; bool stopped;
bool interrupt; bool interrupt;
time_t time_started; time_t time_started;

34
src/driver/streamts.cpp
View File

@@ -749,8 +749,12 @@ void CStreamStream::Close()
if (avio_ctx) if (avio_ctx)
av_free(avio_ctx); av_free(avio_ctx);
if (bsfc)
av_bitstream_filter_close(bsfc);
ifcx = NULL; ifcx = NULL;
ofcx = NULL; ofcx = NULL;
bsfc = NULL;
avio_ctx = NULL; avio_ctx = NULL;
} }
@@ -788,7 +792,11 @@ bool CStreamStream::Open()
printf("%s: Cannot find stream info [%s]!\n", __FUNCTION__, channel->getUrl().c_str()); printf("%s: Cannot find stream info [%s]!\n", __FUNCTION__, channel->getUrl().c_str());
return false; return false;
} }
if (!strstr(ifcx->iformat->name, "applehttp") && !strstr(ifcx->iformat->name, "mpegts")) { if (!strstr(ifcx->iformat->name, "applehttp") &&
!strstr(ifcx->iformat->name, "mpegts") &&
!strstr(ifcx->iformat->name, "matroska") &&
!strstr(ifcx->iformat->name, "avi") &&
!strstr(ifcx->iformat->name, "mp4")) {
printf("%s: not supported format [%s]!\n", __FUNCTION__, ifcx->iformat->name); printf("%s: not supported format [%s]!\n", __FUNCTION__, ifcx->iformat->name);
return false; return false;
} }
@@ -830,6 +838,9 @@ bool CStreamStream::Open()
av_log_set_level(AV_LOG_VERBOSE); av_log_set_level(AV_LOG_VERBOSE);
av_dump_format(ofcx, 0, ofcx->filename, 1); av_dump_format(ofcx, 0, ofcx->filename, 1);
av_log_set_level(AV_LOG_WARNING); av_log_set_level(AV_LOG_WARNING);
bsfc = av_bitstream_filter_init("h264_mp4toannexb");
if (!bsfc)
printf("%s: av_bitstream_filter_init h264_mp4toannexb failed!\n", __FUNCTION__);
return true; return true;
} }
@@ -872,6 +883,27 @@ void CStreamStream::run()
av_init_packet(&pkt); av_init_packet(&pkt);
if (av_read_frame(ifcx, &pkt) < 0) if (av_read_frame(ifcx, &pkt) < 0)
break; break;
if (pkt.stream_index < 0)
continue;
AVCodecContext *codec = ifcx->streams[pkt.stream_index]->codec;
if (bsfc && codec->codec_id == CODEC_ID_H264 ) {
AVPacket newpkt = pkt;
int len;
if ((len = av_bitstream_filter_filter(bsfc, codec, NULL, &newpkt.data, &newpkt.size, pkt.data, pkt.size, pkt.flags & AV_PKT_FLAG_KEY)) < 0) {
av_free_packet(&pkt);
continue;
}
av_free_packet(&pkt);
newpkt.buf = av_buffer_create(newpkt.data, newpkt.size, av_buffer_default_free, NULL, 0);
pkt = newpkt;
}
pkt.pts = av_rescale_q(pkt.pts, ifcx->streams[pkt.stream_index]->time_base, ofcx->streams[pkt.stream_index]->time_base);
pkt.dts = av_rescale_q(pkt.dts, ifcx->streams[pkt.stream_index]->time_base, ofcx->streams[pkt.stream_index]->time_base);
av_write_frame(ofcx, &pkt); av_write_frame(ofcx, &pkt);
av_free_packet(&pkt); av_free_packet(&pkt);
} }

1
src/driver/streamts.h
View File

@@ -73,6 +73,7 @@ class CStreamStream : public CStreamInstance
private: private:
AVFormatContext *ifcx; AVFormatContext *ifcx;
AVFormatContext *ofcx; AVFormatContext *ofcx;
AVBitStreamFilterContext *bsfc;
AVIOContext *avio_ctx; AVIOContext *avio_ctx;
bool stopped; bool stopped;