ZLKMedia/3rdpart/media-server/libhls/demo/hls-segmenter-mp4.cpp

#if defined(_HAVE_FFMPEG_)

extern "C" 
{
#include "libavformat/avformat.h"
}
#include "hls-fmp4.h"
#include "hls-m3u8.h"
#include "hls-param.h"
#include "mov-format.h"
#include "mpeg-ps.h"
#include <assert.h>

static char s_packet[2 * 1024 * 1024];

static void ffmpeg_init()
{
	avformat_network_init();
}

static AVFormatContext* ffmpeg_open(const char* url)
{
	int r;
	AVFormatContext* ic;
	AVDictionary* opt = NULL;
	ic = avformat_alloc_context();
	if (NULL == ic)
	{
		printf("%s(%s): avformat_alloc_context failed.\n", __FUNCTION__, url);
		return NULL;
	}

	//if (!av_dict_get(ff->opt, "scan_all_pmts", NULL, AV_DICT_MATCH_CASE)) {
	//	av_dict_set(&ff->opt, "scan_all_pmts", "1", AV_DICT_DONT_OVERWRITE);
	//	scan_all_pmts_set = 1;
	//}

	r = avformat_open_input(&ic, url, NULL, &opt);
	if (0 != r)
	{
		printf("%s: avformat_open_input(%s) => %d\n", __FUNCTION__, url, r);
		return NULL;
	}

	//if (scan_all_pmts_set)
	//	av_dict_set(&format_opts, "scan_all_pmts", NULL, AV_DICT_MATCH_CASE);

	//ff->ic->probesize = 100 * 1024;
	//ff->ic->max_analyze_duration = 5 * AV_TIME_BASE;

	/* If not enough info to get the stream parameters, we decode the
	first frames to get it. (used in mpeg case for example) */
	r = avformat_find_stream_info(ic, NULL/*&opt*/);
	if (r < 0) {
		printf("%s(%s): could not find codec parameters\n", __FUNCTION__, url);
		return NULL;
	}

	av_dict_free(&opt);
	return ic;
}

static int hls_init_segment(hls_fmp4_t* hls, hls_m3u8_t* m3u)
{
	int bytes = hls_fmp4_init_segment(hls, s_packet, sizeof(s_packet));

	FILE* fp = fopen("hls/0.mp4", "wb");
	fwrite(s_packet, 1, bytes, fp);
	fclose(fp);

	return hls_m3u8_set_x_map(m3u, "hls/0.mp4");
}

static int hls_segment(void* m3u8, const void* data, size_t bytes, int64_t /*pts*/, int64_t dts, int64_t duration)
{
	static int i = 0;
	static char name[128] = { 0 };
	snprintf(name, sizeof(name) - 1, "hls/%d.mp4", ++i);
	FILE* fp = fopen(name, "wb");
	fwrite(data, 1, bytes, fp);
	fclose(fp);

	return hls_m3u8_add((hls_m3u8_t*)m3u8, name, dts, duration, 0);
}

void hls_segmenter_fmp4_test(const char* file)
{
	ffmpeg_init();

	AVPacket pkt;
	memset(&pkt, 0, sizeof(pkt));
	//av_init_packet(&pkt);

	AVFormatContext* ic = ffmpeg_open(file);
	hls_m3u8_t* m3u = hls_m3u8_create(0, 7);
	hls_fmp4_t* hls = hls_fmp4_create(HLS_DURATION * 1000, hls_segment, m3u);

	int track_aac = -1;
	int track_264 = -1;
	int track_265 = -1;
	for (unsigned int i = 0; i < ic->nb_streams; i++)
	{
		AVStream* st = ic->streams[i];
		if (AV_CODEC_ID_AAC == st->codecpar->codec_id)
			track_aac = hls_fmp4_add_audio(hls, MOV_OBJECT_AAC, st->codecpar->channels, st->codecpar->bits_per_coded_sample, st->codecpar->sample_rate, st->codecpar->extradata, st->codecpar->extradata_size);
		else if (AV_CODEC_ID_H264 == st->codecpar->codec_id)
			track_264 = hls_fmp4_add_video(hls, MOV_OBJECT_H264, st->codecpar->width, st->codecpar->height, st->codecpar->extradata, st->codecpar->extradata_size);
		else if(AV_CODEC_ID_H265 == st->codecpar->codec_id)
			track_265 = hls_fmp4_add_video(hls, MOV_OBJECT_HEVC, st->codecpar->width, st->codecpar->height, st->codecpar->extradata, st->codecpar->extradata_size);
	}

	// write init segment
	hls_init_segment(hls, m3u);

	int r = av_read_frame(ic, &pkt);
	while (0 == r)
	{
		AVStream* st = ic->streams[pkt.stream_index];
		int64_t pts = (int64_t)(pkt.pts * av_q2d(st->time_base) * 1000);
		int64_t dts = (int64_t)(pkt.dts * av_q2d(st->time_base) * 1000);
		if (AV_CODEC_ID_AAC == st->codecpar->codec_id)
		{
			//printf("[A] pts: %08lld, dts: %08lld\n", pts, dts);
			hls_fmp4_input(hls, track_aac, pkt.data, pkt.size, pts, dts, 0);
		}
		else if (AV_CODEC_ID_H264 == st->codecpar->codec_id)
		{
			//printf("[V] pts: %08lld, dts: %08lld\n", pts, dts);
			hls_fmp4_input(hls, track_264, pkt.data, pkt.size, pts, dts, (pkt.flags & AV_PKT_FLAG_KEY) ? MOV_AV_FLAG_KEYFREAME : 0);
		}
		else if (AV_CODEC_ID_H265 == st->codecpar->codec_id)
		{
			//printf("[V] pts: %08lld, dts: %08lld\n", pts, dts);
			hls_fmp4_input(hls, track_265, pkt.data, pkt.size, pts, dts, (pkt.flags & AV_PKT_FLAG_KEY) ? MOV_AV_FLAG_KEYFREAME : 0);
		}
		else
		{
			assert(0);
		}

		//av_packet_unref(&pkt);
		r = av_read_frame(ic, &pkt);
	}

	avformat_close_input(&ic);
	avformat_free_context(ic); 
	hls_fmp4_destroy(hls);

	// write m3u8 file
	hls_m3u8_playlist(m3u, 1, s_packet, sizeof(s_packet));
	hls_m3u8_destroy(m3u);

	FILE* fp = fopen("playlist.m3u8", "wb");
	fwrite(s_packet, 1, strlen(s_packet), fp);
	fclose(fp);
}
#endif