Skip to content

Commit 0c5c1b6

Browse files
authored
remove rtsp ps test deps on avstream stream_codecid (#357)
* fix: ps file read and write for rtsp * fix: ps file seek for rtsp test * fix: ps file seek reset for rtsp test * remove rtsp ps test deps on avstream stream_codecid
1 parent fdebfd9 commit 0c5c1b6

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

librtsp/test/media/ps-file-reader.cpp

+6-6
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
#include <map>
99

1010
PSFileReader::PSFileReader(const char* file)
11-
:m_fp(NULL), m_pos(0), m_v_start_ts(-1), m_v_end_ts(-1), m_duration(0), m_demuxer(NULL)
11+
:m_fp(NULL), m_pos(0), m_v_start_ts(-1), m_v_end_ts(-1), m_v_codecid(-1), m_a_codecid(-1), m_duration(0), m_demuxer(NULL)
1212
{
1313
memset(&m_utils, 0, sizeof(m_utils));
1414
m_fp = fopen(file, "rb");
@@ -116,7 +116,7 @@ int PSFileReader::GetNextFrame(int64_t& pts, int64_t& dts, const uint8_t*& ptr,
116116
pts = pkt->pts;
117117
dts = pkt->dts;
118118
flags = pkt->flags;
119-
codecid = pkt->stream->stream_codecid;
119+
codecid = (pkt->stream->codecid >= AVCODEC_VIDEO_MPEG1 && pkt->stream->codecid <= AVCODEC_VIDEO_SVAC) ? m_v_codecid : m_a_codecid;
120120

121121
return 0;
122122
}
@@ -133,13 +133,13 @@ void PSFileReader::PSOnStream(void* param, int stream, int codecid, const void*
133133
AVPACKET_CODEC_ID avcodecid = s_payloads[r].codecid;
134134
if (avcodecid >= AVCODEC_VIDEO_MPEG1 && avcodecid <= AVCODEC_VIDEO_SVAC)
135135
{
136-
struct avstream_t* avstream = avpktutil_addvideo(&self->m_utils, stream, avcodecid, 0, 0, extra, bytes);
137-
avstream->stream_codecid = codecid;
136+
avpktutil_addvideo(&self->m_utils, stream, avcodecid, 0, 0, extra, bytes);
137+
self->m_v_codecid = codecid;
138138
}
139139
else if (avcodecid >= AVCODEC_AUDIO_PCM && avcodecid <= AVCODEC_AUDIO_SVAC)
140140
{
141-
struct avstream_t* avstream = avpktutil_addaudio(&self->m_utils, stream, avcodecid, 0, 0, 0, extra, bytes);
142-
avstream->stream_codecid = codecid;
141+
avpktutil_addaudio(&self->m_utils, stream, avcodecid, 0, 0, 0, extra, bytes);
142+
self->m_a_codecid = codecid;
143143
}
144144
}
145145

librtsp/test/media/ps-file-reader.h

+2
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@ class PSFileReader : std::enable_shared_from_this<PSFileReader>
3030
public:
3131
int64_t m_v_start_ts;
3232
int64_t m_v_end_ts;
33+
int m_v_codecid;
34+
int m_a_codecid;
3335

3436
private:
3537
FILE* m_fp;

0 commit comments

Comments
 (0)