//
// Created by alex on 2021/7/5.
//

#ifndef HTTP_TSPLAYERIMP_H
#define HTTP_TSPLAYERIMP_H

#include <unordered_set>
#include "Util/util.h"
#include "Poller/Timer.h"
#include "Http/HttpDownloader.h"
#include "Player/MediaPlayer.h"
#include "Rtp/Decoder.h"
#include "Rtp/TSDecoder.h"
#include "TsPlayer.h"

using namespace toolkit;
namespace mediakit {

    class TsDemuxer : public MediaSinkInterface, public TrackSource, public std::enable_shared_from_this<TsDemuxer> {
    public:
        TsDemuxer() = default;

        ~TsDemuxer() override { _timer = nullptr; }

        void start(const EventPoller::Ptr &poller, TrackListener *listener);

        bool inputFrame(const Frame::Ptr &frame) override;

        bool addTrack(const Track::Ptr &track) override {
            return _delegate.addTrack(track);
        }

        void addTrackCompleted() override {
            _delegate.addTrackCompleted();
        }

        void resetTracks() override {
            ((MediaSink &) _delegate).resetTracks();
        }

        vector<Track::Ptr> getTracks(bool ready = true) const override {
            return _delegate.getTracks(ready);
        }

    private:
        void onTick();

        int64_t getBufferMS();

        int64_t getPlayPosition();

        void setPlayPosition(int64_t pos);

    private:
        int64_t _ticker_offset = 0;
        Ticker _ticker;
        Stamp _stamp[2];
        Timer::Ptr _timer;
        MediaSinkDelegate _delegate;
        multimap<int64_t, Frame::Ptr> _frame_cache;
    };


    class TsPlayerImp : public PlayerImp<TsPlayer, PlayerBase>, private TrackListener {
    public:
        typedef std::shared_ptr<TsPlayerImp> Ptr;

        TsPlayerImp(const EventPoller::Ptr &poller = nullptr);

        ~TsPlayerImp() override = default;

    private:
        //// HlsPlayer override////
        void onPacket(const char *data, size_t len) override;

    private:
        //// PlayerBase override////
        void onPlayResult(const SockException &ex) override;

        vector<Track::Ptr> getTracks(bool ready = true) const override;

        void onShutdown(const SockException &ex) override;

    private:
        //// TrackListener override////
        bool addTrack(const Track::Ptr &track) override { return true; };

        void addTrackCompleted() override;

    private:
        DecoderImp::Ptr _decoder;
        MediaSinkInterface::Ptr _demuxer;
    };
}//namespace mediakit
#endif //HTTP_TSPLAYERIMP_H