mirror of
https://gitee.com/xia-chu/ZLMediaKit.git
synced 2026-05-23 01:57:50 +08:00
141 lines
3.5 KiB
C++
141 lines
3.5 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 HLSMAKERSUB_H
|
|
#define HLSMAKERSUB_H
|
|
|
|
#include <deque>
|
|
#include <tuple>
|
|
#include "Common/config.h"
|
|
#include "Util/TimeTicker.h"
|
|
#include "Util/File.h"
|
|
#include "Util/util.h"
|
|
#include "Util/logger.h"
|
|
|
|
namespace mediakit {
|
|
|
|
class HlsMakerSub {
|
|
public:
|
|
/**
|
|
* @param seg_duration 切片文件长度
|
|
* @param seg_number 切片个数
|
|
* @param seg_keep 是否保留切片文件
|
|
*/
|
|
HlsMakerSub(float seg_duration = 5, uint32_t seg_number = 3, bool seg_keep = false);
|
|
virtual ~HlsMakerSub();
|
|
|
|
/**
|
|
* 写入ts数据
|
|
* @param data 数据
|
|
* @param len 数据长度
|
|
* @param timestamp 毫秒时间戳
|
|
* @param is_idr_fast_packet 是否为关键帧第一个包
|
|
*/
|
|
void inputData(void *data, size_t len, uint64_t timestamp, bool is_idr_fast_packet);
|
|
|
|
/**
|
|
* 是否为直播
|
|
*/
|
|
bool isLive();
|
|
|
|
/**
|
|
* 是否保留切片文件
|
|
*/
|
|
bool isKeep();
|
|
|
|
/**
|
|
* 清空记录
|
|
*/
|
|
void clear();
|
|
//设置是否录像标志
|
|
void startRecord(bool isRecord);
|
|
|
|
protected:
|
|
/**
|
|
* 创建ts切片文件回调
|
|
* @param index
|
|
* @return
|
|
*/
|
|
virtual std::string onOpenSegment(uint64_t index) = 0;
|
|
|
|
/**
|
|
* 删除ts切片文件回调
|
|
* @param index
|
|
*/
|
|
virtual void onDelSegment(uint64_t index) = 0;
|
|
|
|
/**
|
|
* 写ts切片文件回调
|
|
* @param data
|
|
* @param len
|
|
*/
|
|
virtual void onWriteSegment(const char *data, size_t len) = 0;
|
|
|
|
/**
|
|
* 写m3u8文件回调
|
|
*/
|
|
virtual void onWriteHls(const std::string &data) = 0;
|
|
|
|
/**
|
|
* 上一个 ts 切片写入完成, 可在这里进行通知处理
|
|
* @param duration_ms 上一个 ts 切片的时长, 单位为毫秒
|
|
*/
|
|
virtual void onFlushLastSegment(uint64_t duration_ms) {};
|
|
virtual std::string getPathPrefix() = 0;
|
|
|
|
/**
|
|
* 关闭上个ts切片并且写入m3u8索引
|
|
* @param eof HLS直播是否已结束
|
|
*/
|
|
void flushLastSegment(bool eof);
|
|
|
|
private:
|
|
/**
|
|
* 生成m3u8文件
|
|
* @param eof true代表点播
|
|
*/
|
|
void makeIndexFile(bool eof = false);
|
|
|
|
/**
|
|
* 删除旧的ts切片
|
|
*/
|
|
void delOldSegment();
|
|
|
|
/**
|
|
* 添加新的ts切片
|
|
* @param timestamp
|
|
*/
|
|
void addNewSegment(uint64_t timestamp);
|
|
|
|
//新增函数，实现录像功能
|
|
std::string getTsFile(const std::string &file_content);
|
|
std::string getM3u8TSBody(const std::string &file_content);
|
|
void createM3u8FileForRecord();
|
|
|
|
private:
|
|
float _seg_duration = 0;
|
|
uint32_t _seg_number = 0;
|
|
bool _seg_keep = false;
|
|
uint64_t _last_timestamp = 0;
|
|
uint64_t _last_seg_timestamp = 0;
|
|
uint64_t _file_index = 0;
|
|
std::string _last_file_name;
|
|
std::deque<std::tuple<int,std::string> > _seg_dur_list;
|
|
bool _is_record = false;
|
|
bool _is_close_stream = false;
|
|
std::string _m3u8_file_path;
|
|
|
|
public:
|
|
std::map<uint64_t /*index*/, std::string /*file_path*/> _segment_file_paths;
|
|
};
|
|
|
|
}//namespace mediakit
|
|
#endif //HLSMAKERSUB_H
|