HlsMakerImp.h 1.93 KB
Newer Older
xiongziliang committed
1
/*
xiongziliang committed
2
 * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
3 4 5
 *
 * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
 *
xiongziliang committed
6 7 8
 * 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.
9 10 11 12 13 14 15 16 17
 */

#ifndef HLSMAKERIMP_H
#define HLSMAKERIMP_H

#include <memory>
#include <string>
#include <stdlib.h>
#include "HlsMaker.h"
18
#include "HlsMediaSource.h"
19

20 21 22 23 24 25 26 27 28 29 30
using namespace std;

namespace mediakit {

class HlsMakerImp : public HlsMaker{
public:
    HlsMakerImp(const string &m3u8_file,
                const string &params,
                uint32_t bufSize  = 64 * 1024,
                float seg_duration = 5,
                uint32_t seg_number = 3);
31 32

    ~HlsMakerImp() override;
33 34 35 36 37 38 39

    /**
     * 设置媒体信息
     * @param vhost 虚拟主机
     * @param app 应用名
     * @param stream_id 流id
     */
40 41 42 43 44 45
    void setMediaSource(const string &vhost, const string &app, const string &stream_id);

    /**
     * 获取MediaSource
     * @return
     */
46 47 48 49 50 51 52
    HlsMediaSource::Ptr getMediaSource() const;

     /**
      * 清空缓存
      */
     void clearCache();

53
protected:
xiongziliang committed
54 55 56
    string onOpenSegment(int index) override ;
    void onDelSegment(int index) override;
    void onWriteSegment(const char *data, int len) override;
57
    void onWriteHls(const char *data, int len) override;
58
    void onFlushLastSegment(uint32_t duration_ms) override;
59

60 61
private:
    std::shared_ptr<FILE> makeFile(const string &file,bool setbuf = false);
62

63
private:
64 65 66 67
    int _buf_size;
    string _params;
    string _path_hls;
    string _path_prefix;
68
    RecordInfo _info;
69 70
    std::shared_ptr<FILE> _file;
    std::shared_ptr<char> _file_buf;
71 72
    HlsMediaSource::Ptr _media_src;
    map<int /*index*/,string/*file_path*/> _segment_file_paths;
73 74 75 76
};

}//namespace mediakit
#endif //HLSMAKERIMP_H