bug fix

zhangmeng

2019-09-24 9792631e9d038ac34e287323961b78cc3d470873

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#ifndef _cffmpeg_wrapper_hpp_
#define _cffmpeg_wrapper_hpp_
 
#include <stdint.h>
 
#include <string>
#include <list>
#include <thread>
#include <atomic>
#include <mutex>
#include <memory>
#include "common/callback.hpp"
 
 
 
namespace ffwrapper{
    class FormatIn;
    class CodedData;
    class VideoProp;
}
 
namespace cffmpeg_wrap{
 
    class stream;
    class decoder;
    class rec;
 
    class Wrapper{
    public:
        Wrapper();
        ~Wrapper ();
    private: 
        std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input);
 
        void init_worker(ffwrapper::FormatIn *in);
        void run_worker(ffwrapper::FormatIn *in, std::shared_ptr<ffwrapper::CodedData> data, int64_t &id);
        void deinit_worker();
    public: 
        int RunStream(const char* input);
    private: 
        void run_stream_thread();
    public: //recorder
        void BuildRecorder(const char* id,const char *dir, const int mind, const int maxd, const bool audio);
        int FireRecorder(const char* sid,const int64_t &id);
        void GetInfoRecorder(std::string &recID, int &index, std::string &path);
        void ScalePicture(const int w, const int h, const int flags){
            scale_w_ = w;
            scale_h_ = h;
            scale_f_ = flags;
        }
        void GB28181(){gb_ = 1;}
        void CPUDec(){cpu_ = 1;}
        void AudioSwitch(const bool a);
    public: //decoder
        void BuildDecoder();
        void GetPicDecoder(unsigned char **data, int *w, int *h, int64_t *id);
    public: // push stream
        void GetPacket(unsigned char **pktData, int *size, int *key);
    private:
        // stream 参数
        std::string input_url_;
        int scale_w_, scale_h_, scale_f_;
        bool audio_;
        int gb_, cpu_;
        bool run_dec_;
        // decoder 参数
        std::unique_ptr<std::thread> thread_;
        std::atomic_bool    stop_stream_;
        // 业务类
        // 推流类
        stream* stream_;
        // 解码类
        decoder* decoder_;
        // 录像类,一直存在
        rec* rec_;
        // 录像请求缓存,等待runstream后添加
        std::function<void()> fn_rec_lazy_;
    };
 
    uint8_t* Decode(const char *file, const int gb, int *w, int *h);
    void *CreateEncoder(const int w, const int h, const int fps, const int br, const int scale_flag, const int gi);
    void DestroyEncoder(void *h);
    int Encode(void *hdl, uint8_t *in, const int w, const int h, uint8_t **out, int *size, int *key);
 
}
 
#endif