136 lines
3.2 KiB
C++
136 lines
3.2 KiB
C++
/*
|
|
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
|
|
*
|
|
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
|
*
|
|
* Use of this source code is governed by MIT-like 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 ZLMEDIAKIT_MP4_H
|
|
#define ZLMEDIAKIT_MP4_H
|
|
|
|
#if defined(ENABLE_MP4)
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include "mp4-writer.h"
|
|
#include "mov-writer.h"
|
|
#include "mov-reader.h"
|
|
#include "mpeg4-hevc.h"
|
|
#include "mpeg4-avc.h"
|
|
#include "mpeg4-aac.h"
|
|
#include "mov-buffer.h"
|
|
#include "mov-format.h"
|
|
|
|
namespace mediakit {
|
|
|
|
//mp4文件IO的抽象接口类
|
|
class MP4FileIO : public std::enable_shared_from_this<MP4FileIO> {
|
|
public:
|
|
using Ptr = std::shared_ptr<MP4FileIO>;
|
|
using Writer = std::shared_ptr<mp4_writer_t>;
|
|
using Reader = std::shared_ptr<mov_reader_t>;
|
|
|
|
virtual ~MP4FileIO() = default;
|
|
|
|
/**
|
|
* 创建mp4复用器
|
|
* @param flags 支持0、MOV_FLAG_FASTSTART、MOV_FLAG_SEGMENT
|
|
* @param is_fmp4 是否为fmp4还是普通mp4
|
|
* @return mp4复用器
|
|
*/
|
|
virtual Writer createWriter(int flags, bool is_fmp4 = false);
|
|
|
|
/**
|
|
* 创建mp4解复用器
|
|
* @return mp4解复用器
|
|
*/
|
|
virtual Reader createReader();
|
|
|
|
/**
|
|
* 获取文件读写位置
|
|
*/
|
|
virtual uint64_t onTell() = 0;
|
|
|
|
/**
|
|
* seek至文件某处
|
|
* @param offset 文件偏移量
|
|
* @return 是否成功(0成功)
|
|
*/
|
|
virtual int onSeek(uint64_t offset) = 0;
|
|
|
|
/**
|
|
* 从文件读取一定数据
|
|
* @param data 数据存放指针
|
|
* @param bytes 指针长度
|
|
* @return 是否成功(0成功)
|
|
*/
|
|
virtual int onRead(void *data, size_t bytes) = 0;
|
|
|
|
/**
|
|
* 写入文件一定数据
|
|
* @param data 数据指针
|
|
* @param bytes 数据长度
|
|
* @return 是否成功(0成功)
|
|
*/
|
|
virtual int onWrite(const void *data, size_t bytes) = 0;
|
|
};
|
|
|
|
//磁盘MP4文件类
|
|
class MP4FileDisk : public MP4FileIO {
|
|
public:
|
|
using Ptr = std::shared_ptr<MP4FileDisk>;
|
|
|
|
/**
|
|
* 打开磁盘文件
|
|
* @param file 文件路径
|
|
* @param mode fopen的方式
|
|
*/
|
|
void openFile(const char *file, const char *mode);
|
|
|
|
/**
|
|
* 关闭磁盘文件
|
|
*/
|
|
void closeFile();
|
|
|
|
protected:
|
|
uint64_t onTell() override;
|
|
int onSeek(uint64_t offset) override;
|
|
int onRead(void *data, size_t bytes) override;
|
|
int onWrite(const void *data, size_t bytes) override;
|
|
|
|
private:
|
|
std::shared_ptr<FILE> _file;
|
|
};
|
|
|
|
class MP4FileMemory : public MP4FileIO{
|
|
public:
|
|
using Ptr = std::shared_ptr<MP4FileMemory>;
|
|
|
|
/**
|
|
* 获取文件大小
|
|
*/
|
|
size_t fileSize() const;
|
|
|
|
/**
|
|
* 获取并清空文件缓存
|
|
*/
|
|
std::string getAndClearMemory();
|
|
|
|
protected:
|
|
uint64_t onTell() override;
|
|
int onSeek(uint64_t offset) override;
|
|
int onRead(void *data, size_t bytes) override;
|
|
int onWrite(const void *data, size_t bytes) override;
|
|
|
|
private:
|
|
uint64_t _offset = 0;
|
|
std::string _memory;
|
|
};
|
|
|
|
}//namespace mediakit
|
|
#endif //defined(ENABLE_MP4)
|
|
#endif //ZLMEDIAKIT_MP4_H
|