| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891 |
- #include "muxer_file_muxer.h"
- #include "../base/logger.h"
- #include "../base/media_common.h"
- #include <filesystem>
- #include <sstream>
- #include <iomanip>
- #include <algorithm>
- extern "C" {
- #include <libavformat/avformat.h>
- #include <libavutil/opt.h>
- #include <libavutil/time.h>
- #include <libavutil/mathematics.h>
- }
- namespace av {
- namespace muxer {
- FileMuxer::FileMuxer() {
- AV_LOGGER_DEBUG("创建文件复用器");
- lastFlushTime_ = std::chrono::steady_clock::now();
- }
- FileMuxer::~FileMuxer() {
- close();
- AV_LOGGER_DEBUG("文件复用器已销毁");
- }
- ErrorCode FileMuxer::initialize(const MuxerParams& params) {
- if (params.type != MuxerType::FILE_MUXER) {
- AV_LOGGER_ERROR("参数类型不是文件复用器");
- return ErrorCode::INVALID_PARAMS;
- }
-
- // 先验证参数,再进行类型转换
- if (!validateParams(params)) {
- return ErrorCode::INVALID_PARAMS;
- }
-
- // 安全地进行类型转换
- try {
- fileMuxerParams_ = static_cast<const FileMuxerParams&>(params);
- } catch (const std::exception& e) {
- AV_LOGGER_ERRORF("参数类型转换失败: {}", e.what());
- return ErrorCode::INVALID_PARAMS;
- }
-
- // 调用父类的initialize方法来处理streams
- ErrorCode result = AbstractMuxer::initialize(params);
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- // 设置输出文件
- if (!fileMuxerParams_.outputFile.empty()) {
- currentOutputFile_ = fileMuxerParams_.outputFile;
- } else {
- currentOutputFile_ = fileMuxerParams_.outputPath;
- }
-
- ErrorCode validateResult = validateFilePath(currentOutputFile_);
- if (validateResult != ErrorCode::SUCCESS) {
- return validateResult;
- }
-
- // 设置分段
- if (fileMuxerParams_.enableSegmentation) {
- segmentationEnabled_ = true;
- if (fileMuxerParams_.segmentPattern.empty()) {
- // 生成默认分段模式
- std::filesystem::path path(currentOutputFile_);
- std::string stem = path.stem().string();
- std::string ext = path.extension().string();
- fileMuxerParams_.segmentPattern = stem + "_%03d" + ext;
- }
- }
-
- AV_LOGGER_INFOF("文件复用器初始化成功: {}", currentOutputFile_);
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::start() {
- if (getState() != MuxerState::INITIALIZED) {
- AV_LOGGER_ERROR("复用器状态无效,无法启动");
- return ErrorCode::INVALID_STATE;
- }
-
- ErrorCode result = setupOutput();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- result = writeHeader();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- // 启动写入线程
- shouldStopWriting_ = false;
- try {
- writeThread_ = std::thread(&FileMuxer::writeThreadFunc, this);
- setState(MuxerState::STARTED);
-
- fileStartTime_ = std::chrono::steady_clock::now();
- AV_LOGGER_INFO("文件复用器已启动");
-
- return ErrorCode::SUCCESS;
- } catch (const std::exception& e) {
- AV_LOGGER_ERRORF("启动写入线程失败: {}", e.what());
- return ErrorCode::THREAD_ERROR;
- }
- }
- ErrorCode FileMuxer::stop() {
- if (getState() != MuxerState::STARTED && getState() != MuxerState::PAUSED) {
- return ErrorCode::SUCCESS;
- }
-
- // 停止写入线程
- shouldStopWriting_ = true;
- queueCondition_.notify_all();
-
- if (writeThread_.joinable()) {
- writeThread_.join();
- }
-
- // 刷新剩余数据
- flush();
-
- // 写入文件尾
- ErrorCode result = writeTrailer();
- if (result != ErrorCode::SUCCESS) {
- AV_LOGGER_ERRORF("写入文件尾失败: {}", static_cast<int>(result));
- }
-
- setState(MuxerState::STOPPED);
- AV_LOGGER_INFO("文件复用器已停止");
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::close() {
- stop();
-
- ErrorCode result = closeOutputFile();
-
- // 清空队列
- {
- std::lock_guard<std::mutex> lock(queueMutex_);
- while (!packetQueue_.empty()) {
- packetQueue_.pop();
- }
- }
-
- setState(MuxerState::IDLE);
- AV_LOGGER_INFO("文件复用器已关闭");
-
- return result;
- }
- ErrorCode FileMuxer::writePacket(AVPacket* packet) {
- if (!packet) {
- AV_LOGGER_ERROR("包指针为空");
- return ErrorCode::INVALID_PARAMS;
- }
-
- if (getState() != MuxerState::STARTED) {
- AV_LOGGER_ERROR("复用器未启动,无法写入包");
- return ErrorCode::INVALID_STATE;
- }
-
- // 检查队列大小
- {
- std::lock_guard<std::mutex> lock(queueMutex_);
- if (packetQueue_.size() >= MAX_QUEUE_SIZE) {
- AV_LOGGER_WARNING("包队列已满,丢弃包");
- {
- std::lock_guard<std::mutex> statsLock(statsMutex_);
- stats_.droppedPackets++;
- }
- return ErrorCode::QUEUE_FULL;
- }
- }
-
- // 复制包
- AVPacket* packetCopy = av_packet_alloc();
- if (!packetCopy) {
- AV_LOGGER_ERROR("分配包内存失败");
- return ErrorCode::MEMORY_ALLOC_FAILED;
- }
-
- int ret = av_packet_ref(packetCopy, packet);
- if (ret < 0) {
- av_packet_free(&packetCopy);
- AV_LOGGER_ERRORF("复制包失败: {}", ffmpeg_utils::errorToString(ret));
- return static_cast<ErrorCode>(ret);
- }
-
- // 添加到队列
- auto queueItem = std::make_unique<PacketQueueItem>(packetCopy, packet->stream_index);
-
- {
- std::lock_guard<std::mutex> lock(queueMutex_);
- packetQueue_.push(std::move(queueItem));
- queuedPackets_++;
- }
-
- queueCondition_.notify_one();
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::writeFrame(AVFrame* frame, int streamIndex) {
- if (!frame) {
- AV_LOGGER_ERROR("帧指针为空");
- return ErrorCode::INVALID_PARAMS;
- }
-
- // 这里需要编码器将帧编码为包,然后调用writePacket
- // 由于这是复用器,通常接收已编码的包
- AV_LOGGER_WARNING("文件复用器不直接支持写入原始帧,请先编码为包");
-
- return ErrorCode::NOT_SUPPORTED;
- }
- ErrorCode FileMuxer::flush() {
- if (!formatCtx_) {
- return ErrorCode::SUCCESS;
- }
-
- std::lock_guard<std::mutex> lock(fileMutex_);
-
- return flushInternal();
- }
- ErrorCode FileMuxer::flushInternal() {
- if (!formatCtx_) {
- return ErrorCode::SUCCESS;
- }
-
- // 注意:此方法假设调用者已经持有fileMutex_锁
-
- int ret = av_write_frame(formatCtx_, nullptr); // 刷新
- if (ret < 0) {
- AV_LOGGER_ERRORF("刷新复用器失败: {}", ffmpeg_utils::errorToString(ret));
- return static_cast<ErrorCode>(ret);
- }
-
- if (formatCtx_->pb) {
- avio_flush(formatCtx_->pb);
- }
-
- lastFlushTime_ = std::chrono::steady_clock::now();
- AV_LOGGER_DEBUG("复用器已刷新");
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::addStream(const StreamInfo& streamInfo) {
- if (getState() != MuxerState::INITIALIZED) {
- AV_LOGGER_ERROR("只能在初始化状态下添加流");
- return ErrorCode::INVALID_STATE;
- }
-
- std::lock_guard<std::mutex> lock(streamsMutex_);
-
- // 检查流索引是否已存在
- auto it = std::find_if(streams_.begin(), streams_.end(),
- [&streamInfo](const StreamInfo& info) {
- return info.index == streamInfo.index;
- });
-
- if (it != streams_.end()) {
- AV_LOGGER_ERRORF("流索引 {} 已存在", streamInfo.index);
- return ErrorCode::STREAM_EXISTS;
- }
-
- streams_.push_back(streamInfo);
-
- AV_LOGGER_INFOF("已添加流: 索引={}, 类型={}, 编解码器={}",
- streamInfo.index, static_cast<int>(streamInfo.type),
- streamInfo.codecName);
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::setOutputFile(const std::string& filename) {
- if (getState() != MuxerState::IDLE && getState() != MuxerState::INITIALIZED) {
- AV_LOGGER_ERROR("无法在运行时更改输出文件");
- return ErrorCode::INVALID_STATE;
- }
-
- ErrorCode result = validateFilePath(filename);
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- currentOutputFile_ = filename;
- fileMuxerParams_.outputFile = filename;
-
- AV_LOGGER_INFOF("输出文件已设置为: {}", filename);
-
- return ErrorCode::SUCCESS;
- }
- std::string FileMuxer::getOutputFile() const {
- return currentOutputFile_;
- }
- int64_t FileMuxer::getCurrentFileSize() const {
- return currentFileSize_;
- }
- double FileMuxer::getCurrentDuration() const {
- auto now = std::chrono::steady_clock::now();
- return std::chrono::duration<double>(now - fileStartTime_).count();
- }
- ErrorCode FileMuxer::enableSegmentation(bool enable, int duration) {
- if (getState() == MuxerState::STARTED) {
- AV_LOGGER_ERROR("无法在运行时更改分段设置");
- return ErrorCode::INVALID_STATE;
- }
-
- segmentationEnabled_ = enable;
- fileMuxerParams_.enableSegmentation = enable;
- fileMuxerParams_.segmentDuration = duration;
-
- AV_LOGGER_INFOF("Segmentation {}: duration={}s", enable ? "enabled" : "disabled", duration);
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::forceNewSegment() {
- if (!segmentationEnabled_) {
- AV_LOGGER_ERROR("分段功能未启用");
- return ErrorCode::INVALID_STATE;
- }
-
- if (getState() != MuxerState::STARTED) {
- AV_LOGGER_ERROR("复用器未启动");
- return ErrorCode::INVALID_STATE;
- }
-
- return createNewSegment();
- }
- std::vector<std::string> FileMuxer::getSegmentFiles() const {
- return segmentFiles_;
- }
- ErrorCode FileMuxer::setFastStart(bool enable) {
- fileMuxerParams_.enableFastStart = enable;
-
- if (enable) {
- fileMuxerParams_.movFlags |= 0x01; // AVFMT_FLAG_FASTSTART
- } else {
- fileMuxerParams_.movFlags &= ~0x01;
- }
-
- AV_LOGGER_INFOF("Fast start {}", enable ? "enabled" : "disabled");
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::setMovFlags(int flags) {
- fileMuxerParams_.movFlags = flags;
- AV_LOGGER_INFOF("MOV标志已设置为: 0x{:X}", flags);
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::setupOutput() {
- // 分配格式上下文
- const char* formatName = fileMuxerParams_.format.empty() ? nullptr : fileMuxerParams_.format.c_str();
- const char* filename = currentOutputFile_.c_str();
-
- int ret = avformat_alloc_output_context2(&formatCtx_, nullptr, formatName, filename);
- if (ret < 0) {
- AV_LOGGER_ERRORF("分配输出格式上下文失败: {}", ffmpeg_utils::errorToString(ret));
- return static_cast<ErrorCode>(ret);
- }
-
- if (!formatCtx_->oformat) {
- AV_LOGGER_ERROR("无法确定输出格式");
- return ErrorCode::FORMAT_NOT_SUPPORTED;
- }
-
- AV_LOGGER_INFOF("输出格式: {} ({})", formatCtx_->oformat->name, formatCtx_->oformat->long_name);
-
- // 设置流
- ErrorCode result = setupStreams();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- // 设置元数据
- for (const auto& meta : fileMuxerParams_.metadata) {
- av_dict_set(&formatCtx_->metadata, meta.first.c_str(), meta.second.c_str(), 0);
- }
-
- // 打开输出文件
- return openOutputFile();
- }
- ErrorCode FileMuxer::writeHeader() {
- if (!formatCtx_) {
- AV_LOGGER_ERROR("格式上下文未初始化");
- return ErrorCode::INVALID_STATE;
- }
-
- // 设置格式选项
- AVDictionary* options = nullptr;
-
- for (const auto& option : fileMuxerParams_.options) {
- av_dict_set(&options, option.first.c_str(), option.second.c_str(), 0);
- }
-
- // 设置MOV标志
- if (fileMuxerParams_.movFlags != 0) {
- av_dict_set_int(&options, "movflags", fileMuxerParams_.movFlags, 0);
- }
-
- // 写入文件头
- int ret = avformat_write_header(formatCtx_, &options);
- av_dict_free(&options);
-
- if (ret < 0) {
- AV_LOGGER_ERRORF("写入文件头失败: {}", ffmpeg_utils::errorToString(ret));
- return static_cast<ErrorCode>(ret);
- }
-
- AV_LOGGER_INFO("文件头已写入");
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::writeTrailer() {
- if (!formatCtx_) {
- return ErrorCode::SUCCESS;
- }
-
- std::lock_guard<std::mutex> lock(fileMutex_);
-
- int ret = av_write_trailer(formatCtx_);
- if (ret < 0) {
- AV_LOGGER_ERRORF("写入文件尾失败: {}", ffmpeg_utils::errorToString(ret));
- return static_cast<ErrorCode>(ret);
- }
-
- AV_LOGGER_INFO("文件尾已写入");
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::openOutputFile() {
- if (!formatCtx_) {
- return ErrorCode::INVALID_STATE;
- }
-
- // 检查是否需要覆盖文件
- if (!fileMuxerParams_.overwrite && std::filesystem::exists(currentOutputFile_)) {
- AV_LOGGER_ERRORF("文件已存在且不允许覆盖: {}", currentOutputFile_);
- return ErrorCode::FILE_EXISTS;
- }
-
- // 打开输出文件
- if (!(formatCtx_->oformat->flags & AVFMT_NOFILE)) {
- int ret = avio_open(&formatCtx_->pb, currentOutputFile_.c_str(), AVIO_FLAG_WRITE);
- if (ret < 0) {
- AV_LOGGER_ERRORF("打开输出文件失败: {} ({})",
- ffmpeg_utils::errorToString(ret), currentOutputFile_);
- return static_cast<ErrorCode>(ret);
- }
- }
-
- currentFileSize_ = 0;
- AV_LOGGER_INFOF("输出文件已打开: {}", currentOutputFile_);
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::closeOutputFile() {
- if (!formatCtx_) {
- return ErrorCode::SUCCESS;
- }
-
- std::lock_guard<std::mutex> lock(fileMutex_);
-
- // 关闭文件
- if (formatCtx_->pb && !(formatCtx_->oformat->flags & AVFMT_NOFILE)) {
- avio_closep(&formatCtx_->pb);
- }
-
- // 释放格式上下文
- avformat_free_context(formatCtx_);
- formatCtx_ = nullptr;
-
- // 清理流映射
- streamMap_.clear();
-
- AV_LOGGER_INFO("输出文件已关闭");
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::createNewSegment() {
- if (!segmentationEnabled_) {
- return ErrorCode::SUCCESS;
- }
-
- // 写入当前段的尾部
- ErrorCode result = writeTrailer();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- // 关闭当前文件
- result = closeOutputFile();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- // 添加到段文件列表
- segmentFiles_.push_back(currentOutputFile_);
-
- // 生成新的段文件名
- currentSegmentIndex_++;
- currentOutputFile_ = generateSegmentFilename(currentSegmentIndex_);
-
- // 重新设置输出
- result = setupOutput();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- // 写入新段的头部
- result = writeHeader();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
-
- fileStartTime_ = std::chrono::steady_clock::now();
-
- AV_LOGGER_INFOF("创建新段: {}", currentOutputFile_);
-
- return ErrorCode::SUCCESS;
- }
- ErrorCode FileMuxer::setupStreams() {
- std::lock_guard<std::mutex> lock(streamsMutex_);
-
- for (const auto& streamInfo : streams_) {
- AVStream* stream = createAVStream(streamInfo);
- if (!stream) {
- AV_LOGGER_ERRORF("创建流失败: 索引={}", streamInfo.index);
- return ErrorCode::STREAM_CREATE_FAILED;
- }
-
- streamMap_[streamInfo.index] = stream;
- }
-
- AV_LOGGER_INFOF("已设置 {} 个流", streams_.size());
-
- return ErrorCode::SUCCESS;
- }
- AVStream* FileMuxer::createAVStream(const StreamInfo& streamInfo) {
- if (!formatCtx_) {
- return nullptr;
- }
-
- AVStream* stream = avformat_new_stream(formatCtx_, nullptr);
- if (!stream) {
- AV_LOGGER_ERROR("创建新流失败");
- return nullptr;
- }
-
- stream->id = streamInfo.index;
- stream->time_base = streamInfo.timeBase;
-
- // 设置编解码器参数
- AVCodecParameters* codecpar = stream->codecpar;
- codecpar->codec_id = streamInfo.codecId;
- codecpar->codec_type = (streamInfo.type == StreamType::VIDEO) ? AVMEDIA_TYPE_VIDEO : AVMEDIA_TYPE_AUDIO;
-
- if (streamInfo.type == StreamType::VIDEO) {
- codecpar->width = streamInfo.width;
- codecpar->height = streamInfo.height;
- codecpar->format = streamInfo.pixelFormat;
- codecpar->bit_rate = streamInfo.bitrate;
-
- // 设置帧率
- stream->avg_frame_rate = streamInfo.frameRate;
- stream->r_frame_rate = streamInfo.frameRate;
- } else if (streamInfo.type == StreamType::AUDIO) {
- codecpar->sample_rate = streamInfo.sampleRate;
- av_channel_layout_copy(&codecpar->ch_layout, &streamInfo.channelLayout);
- codecpar->format = streamInfo.sampleFormat;
- codecpar->bit_rate = streamInfo.bitrate;
- }
-
- AV_LOGGER_INFOF("创建流成功: 索引={}, 类型={}", streamInfo.index, static_cast<int>(streamInfo.type));
-
- return stream;
- }
- ErrorCode FileMuxer::processPacket(AVPacket* packet) {
- if (!packet || !formatCtx_) {
- return ErrorCode::INVALID_PARAMS;
- }
-
- // 检查流索引
- auto it = streamMap_.find(packet->stream_index);
- if (it == streamMap_.end()) {
- AV_LOGGER_ERRORF("未找到流索引: {}", packet->stream_index);
- return ErrorCode::STREAM_NOT_FOUND;
- }
-
- // 检查是否需要创建新段
- if (shouldCreateNewSegment()) {
- ErrorCode result = createNewSegment();
- if (result != ErrorCode::SUCCESS) {
- return result;
- }
- }
-
- return writePacketInternal(packet);
- }
- ErrorCode FileMuxer::writePacketInternal(AVPacket* packet) {
- std::lock_guard<std::mutex> lock(fileMutex_);
-
- auto startTime = std::chrono::steady_clock::now();
-
- // 写入包
- int ret = av_interleaved_write_frame(formatCtx_, packet);
- if (ret < 0) {
- AV_LOGGER_ERRORF("写入包失败: {}", ffmpeg_utils::errorToString(ret));
- return static_cast<ErrorCode>(ret);
- }
-
- // 更新统计信息
- updateStats(packet);
- currentFileSize_ += packet->size;
- writtenPackets_++;
-
- // 更新平均写入时间
- auto endTime = std::chrono::steady_clock::now();
- double writeTime = std::chrono::duration<double, std::milli>(endTime - startTime).count();
- averageWriteTime_ = (averageWriteTime_ * 0.9) + (writeTime * 0.1);
-
- // 定期刷新 - 使用内部刷新方法避免死锁
- if (!fileMuxerParams_.syncMode) {
- auto now = std::chrono::steady_clock::now();
- auto elapsed = std::chrono::duration<double>(now - lastFlushTime_).count();
- if (elapsed >= fileMuxerParams_.flushInterval) {
- flushInternal();
- }
- }
-
- return ErrorCode::SUCCESS;
- }
- void FileMuxer::writeThreadFunc() {
- AV_LOGGER_INFO("写入线程已启动");
-
- while (!shouldStopWriting_) {
- std::unique_ptr<PacketQueueItem> item;
-
- // 从队列获取包
- {
- std::unique_lock<std::mutex> lock(queueMutex_);
- queueCondition_.wait(lock, [this] {
- return !packetQueue_.empty() || shouldStopWriting_;
- });
-
- if (shouldStopWriting_ && packetQueue_.empty()) {
- break;
- }
-
- if (!packetQueue_.empty()) {
- item = std::move(packetQueue_.front());
- packetQueue_.pop();
- }
- }
-
- if (item && item->packet) {
- ErrorCode result = processPacket(item->packet);
- if (result != ErrorCode::SUCCESS) {
- onError(result, "写入包失败");
- }
- }
- }
-
- // 处理剩余的包
- while (true) {
- std::unique_ptr<PacketQueueItem> item;
-
- {
- std::lock_guard<std::mutex> lock(queueMutex_);
- if (packetQueue_.empty()) {
- break;
- }
- item = std::move(packetQueue_.front());
- packetQueue_.pop();
- }
-
- if (item && item->packet) {
- processPacket(item->packet);
- }
- }
-
- AV_LOGGER_INFO("写入线程已退出");
- }
- std::string FileMuxer::generateSegmentFilename(int segmentIndex) {
- if (fileMuxerParams_.segmentPattern.empty()) {
- std::filesystem::path path(currentOutputFile_);
- std::string stem = path.stem().string();
- std::string ext = path.extension().string();
-
- std::ostringstream oss;
- oss << stem << "_" << std::setfill('0') << std::setw(3) << segmentIndex << ext;
- return oss.str();
- } else {
- char buffer[1024];
- snprintf(buffer, sizeof(buffer), fileMuxerParams_.segmentPattern.c_str(), segmentIndex);
- return std::string(buffer);
- }
- }
- bool FileMuxer::shouldCreateNewSegment() const {
- if (!segmentationEnabled_) {
- return false;
- }
-
- // 检查时长
- if (fileMuxerParams_.segmentDuration > 0) {
- double currentDuration = getCurrentDuration();
- if (currentDuration >= fileMuxerParams_.segmentDuration) {
- return true;
- }
- }
-
- // 检查文件大小
- if (fileMuxerParams_.maxFileSize > 0) {
- if (currentFileSize_ >= fileMuxerParams_.maxFileSize) {
- return true;
- }
- }
-
- return false;
- }
- ErrorCode FileMuxer::validateFilePath(const std::string& path) {
- if (path.empty()) {
- AV_LOGGER_ERROR("文件路径不能为空");
- return ErrorCode::INVALID_PARAMS;
- }
-
- try {
- std::filesystem::path filePath(path);
-
- // 检查父目录是否存在
- std::filesystem::path parentDir = filePath.parent_path();
- if (!parentDir.empty() && !std::filesystem::exists(parentDir)) {
- // 尝试创建目录
- std::filesystem::create_directories(parentDir);
- }
-
- // 检查文件扩展名
- std::string extension = filePath.extension().string();
- if (extension.empty()) {
- AV_LOGGER_WARNING("文件没有扩展名,可能导致格式检测问题");
- }
-
- } catch (const std::exception& e) {
- AV_LOGGER_ERRORF("文件路径验证失败: {}", e.what());
- return ErrorCode::INVALID_PATH;
- }
-
- return ErrorCode::SUCCESS;
- }
- bool FileMuxer::validateParams(const MuxerParams& params) {
- if (!AbstractMuxer::validateParams(params)) {
- return false;
- }
-
- // 首先验证基础参数
- if (params.outputPath.empty()) {
- AV_LOGGER_ERROR("输出文件路径不能为空");
- return false;
- }
-
- // 如果参数类型正确,进行更详细的验证
- if (params.type == MuxerType::FILE_MUXER) {
- try {
- const auto& fileParams = static_cast<const FileMuxerParams&>(params);
-
- if (fileParams.outputFile.empty() && fileParams.outputPath.empty()) {
- AV_LOGGER_ERROR("输出文件路径不能为空");
- return false;
- }
-
- if (fileParams.segmentDuration < 0) {
- AV_LOGGER_ERROR("分段时长不能为负数");
- return false;
- }
-
- if (fileParams.maxFileSize < 0) {
- AV_LOGGER_ERROR("最大文件大小不能为负数");
- return false;
- }
-
- if (fileParams.flushInterval <= 0) {
- AV_LOGGER_ERROR("刷新间隔必须大于0");
- return false;
- }
- } catch (const std::exception& e) {
- AV_LOGGER_ERRORF("参数验证时类型转换失败: {}", e.what());
- return false;
- }
- }
-
- return true;
- }
- // FileMuxerFactory 实现
- std::unique_ptr<AbstractMuxer> FileMuxer::FileMuxerFactory::createMuxer(MuxerType type) {
- if (type == MuxerType::FILE_MUXER) {
- return std::make_unique<FileMuxer>();
- }
-
- return nullptr;
- }
- bool FileMuxer::FileMuxerFactory::isTypeSupported(MuxerType type) const {
- return type == MuxerType::FILE_MUXER;
- }
- std::vector<MuxerType> FileMuxer::FileMuxerFactory::getSupportedTypes() const {
- return {MuxerType::FILE_MUXER};
- }
- std::unique_ptr<FileMuxer> FileMuxer::FileMuxerFactory::createFileMuxer(const std::string& filename) {
- auto muxer = std::make_unique<FileMuxer>();
-
- FileMuxerParams params;
- params.outputFile = filename;
- params.format = AbstractMuxer::getFormatFromExtension(filename);
-
- ErrorCode result = muxer->initialize(params);
- if (result != ErrorCode::SUCCESS) {
- AV_LOGGER_ERRORF("创建文件复用器失败: {}", static_cast<int>(result));
- return nullptr;
- }
-
- return muxer;
- }
- std::unique_ptr<FileMuxer> FileMuxer::FileMuxerFactory::createSegmentedMuxer(const std::string& pattern, int duration) {
- auto muxer = std::make_unique<FileMuxer>();
-
- FileMuxerParams params;
- params.outputFile = pattern;
- params.enableSegmentation = true;
- params.segmentDuration = duration;
- params.segmentPattern = pattern;
-
- // 从模式中推断格式
- std::string firstFile = pattern;
- size_t pos = firstFile.find("%");
- if (pos != std::string::npos) {
- firstFile.replace(pos, firstFile.find('d', pos) - pos + 1, "001");
- }
- params.format = AbstractMuxer::getFormatFromExtension(firstFile);
-
- ErrorCode result = muxer->initialize(params);
- if (result != ErrorCode::SUCCESS) {
- AV_LOGGER_ERRORF("创建分段复用器失败: {}", static_cast<int>(result));
- return nullptr;
- }
-
- return muxer;
- }
- } // namespace muxer
- } // namespace av
|