mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-23 11:17:09 +08:00
114 lines
3.3 KiB
C++
114 lines
3.3 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT license that can be found in the
|
||
* LICENSE file in the root of the source tree. All contributing project authors
|
||
* may be found in the AUTHORS file in the root of the source tree.
|
||
*/
|
||
|
||
#ifndef SRC_RTP_RTPPARSERTESTER_H_
|
||
#define SRC_RTP_RTPPARSERTESTER_H_
|
||
|
||
#include <memory>
|
||
#include <algorithm>
|
||
#include <functional>
|
||
#include "Common/config.h"
|
||
#include "RtspPlayer.h"
|
||
#include "RtspDemuxer.h"
|
||
#include "Poller/Timer.h"
|
||
#include "Util/TimeTicker.h"
|
||
|
||
using namespace std;
|
||
using namespace toolkit;
|
||
|
||
namespace mediakit {
|
||
|
||
class RtspPlayerImp : public PlayerImp<RtspPlayer, PlayerBase> ,private TrackListener {
|
||
public:
|
||
using Ptr = std::shared_ptr<RtspPlayerImp>;
|
||
using Super = PlayerImp<RtspPlayer, PlayerBase>;
|
||
|
||
RtspPlayerImp(const EventPoller::Ptr &poller) : Super(poller) {}
|
||
|
||
~RtspPlayerImp() override {
|
||
DebugL << endl;
|
||
}
|
||
|
||
float getProgress() const override {
|
||
if (getDuration() > 0) {
|
||
return getProgressMilliSecond() / (getDuration() * 1000);
|
||
}
|
||
return PlayerBase::getProgress();
|
||
}
|
||
|
||
uint32_t getProgressPos() const override {
|
||
if (getDuration() > 0) {
|
||
return getProgressMilliSecond();
|
||
}
|
||
return PlayerBase::getProgressPos();
|
||
}
|
||
|
||
void seekTo(float fProgress) override {
|
||
fProgress = MAX(float(0), MIN(fProgress, float(1.0)));
|
||
seekToMilliSecond((uint32_t) (fProgress * getDuration() * 1000));
|
||
}
|
||
|
||
void seekTo(uint32_t seekPos) override {
|
||
uint32_t pos = MAX(float(0), MIN(seekPos, getDuration())) * 1000;
|
||
seekToMilliSecond(pos);
|
||
}
|
||
|
||
float getDuration() const override {
|
||
return _demuxer ? _demuxer->getDuration() : 0;
|
||
}
|
||
|
||
vector<Track::Ptr> getTracks(bool ready = true) const override {
|
||
return _demuxer ? _demuxer->getTracks(ready) : Super::getTracks(ready);
|
||
}
|
||
|
||
private:
|
||
//派生类回调函数
|
||
bool onCheckSDP(const string &sdp) override {
|
||
_rtsp_media_src = dynamic_pointer_cast<RtspMediaSource>(_media_src);
|
||
if (_rtsp_media_src) {
|
||
_rtsp_media_src->setSdp(sdp);
|
||
}
|
||
_demuxer = std::make_shared<RtspDemuxer>();
|
||
_demuxer->setTrackListener(this, true);
|
||
_demuxer->loadSdp(sdp);
|
||
return true;
|
||
}
|
||
|
||
void onRecvRTP(RtpPacket::Ptr rtp, const SdpTrack::Ptr &track) override {
|
||
_demuxer->inputRtp(rtp);
|
||
if (_rtsp_media_src) {
|
||
// rtsp直接代理是无法判断该rtp是否是I帧，所以GOP缓存基本是无效的
|
||
// 为了减少内存使用，那么我们设置为一直关键帧以便清空GOP缓存
|
||
_rtsp_media_src->onWrite(std::move(rtp), true);
|
||
}
|
||
}
|
||
|
||
void onPlayResult(const SockException &ex) override {
|
||
if (ex) {
|
||
Super::onPlayResult(ex);
|
||
return;
|
||
}
|
||
}
|
||
|
||
bool addTrack(const Track::Ptr &track) override { return true; }
|
||
|
||
void addTrackCompleted() override {
|
||
Super::onPlayResult(SockException(Err_success, "play success"));
|
||
}
|
||
|
||
private:
|
||
RtspDemuxer::Ptr _demuxer;
|
||
RtspMediaSource::Ptr _rtsp_media_src;
|
||
};
|
||
|
||
} /* namespace mediakit */
|
||
|
||
#endif /* SRC_RTP_RTPPARSERTESTER_H_ */
|