ZLKMedia/3rdpart/media-server/librtsp/test/media/mp4-file-source.h

#pragma once

#include "mpeg4-avc.h"
#include "mpeg4-aac.h"
#include "mpeg4-hevc.h"
#include "media-source.h"
#include "avpacket-queue.h"
#include "mov-reader.h"
#include "rtp-sender.h"
#include <string>
#include <stdio.h>
#include <stdint.h>

class MP4FileSource : public IMediaSource
{
public:
	MP4FileSource(const char *file);
	virtual ~MP4FileSource();

public:
	virtual int Play();
	virtual int Pause();
	virtual int Seek(int64_t pos);
	virtual int SetSpeed(double speed);
	virtual int GetDuration(int64_t& duration) const;
	virtual int GetSDPMedia(std::string& sdp) const;
	virtual int GetRTPInfo(const char* uri, char *rtpinfo, size_t bytes) const;
	virtual int SetTransport(const char* track, std::shared_ptr<IRTPTransport> transport);

	int SendRTCP(uint64_t clock);

private:
	struct media_t;
	struct media_t* FetchNextPacket();

	static void OnRTCPEvent(void* param, const struct rtcp_msg_t* msg);
	void OnRTCPEvent(const struct rtcp_msg_t* msg);
	
	int SendBye();

	static int OnRTPPacket(void* param, const void *packet, int bytes, uint32_t timestamp, int flags);

	static void MP4OnVideo(void* param, uint32_t track, uint8_t object, int width, int height, const void* extra, size_t bytes);
	static void MP4OnAudio(void* param, uint32_t track, uint8_t object, int channel_count, int bit_per_sample, int sample_rate, const void* extra, size_t bytes);
	static void MP4OnRead(void* param, uint32_t track, const void* buffer, size_t bytes, int64_t pts, int64_t dts, int flags);

private:
	int64_t m_dts;
	uint64_t m_clock;
	mov_reader_t* m_reader;
	FILE* m_fp;

	std::string m_sdp;

	int m_status;
	int64_t m_pos;
	double m_speed;

	struct mpeg4_aac_t m_aac;
	struct mpeg4_avc_t m_avc;
	struct mpeg4_hevc_t m_hevc;
	uint8_t m_packet[2 * 1024 * 1024];

	struct media_t
	{
		struct rtp_sender_t rtp;

		int64_t dts_first; // first frame timestamp
		int64_t dts_last; // last frame timestamp
		uint64_t rtcp_clock;

		struct avpacket_queue_t* pkts;
		std::shared_ptr<IRTPTransport> transport;
		int track; // mp4 track
	};

	struct media_t m_media[2]; // 0-video, 1-audio
	int m_count;
};