library/cpp/codecs/codecs_registry.h


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

#pragma once 
 
#include "codecs.h" 
#include <util/string/cast.h>
 
namespace NCodecs { 
    struct TNoCodecException : TCodecException {
        TNoCodecException(TStringBuf name) {
            (*this) << "unknown codec: " << name; 
        } 
    }; 
 
    struct ICodecFactory : TAtomicRefCount<ICodecFactory> {
        virtual ~ICodecFactory() = default; 
        virtual TCodecPtr MakeCodec(TStringBuf name) const = 0; 
        virtual TVector<TString> ListNames() const = 0;
    }; 
 
    typedef TIntrusivePtr<ICodecFactory> TCodecFactoryPtr; 
 
    namespace NPrivate { 
        template <typename TCodec>
        struct TInstanceFactory : ICodecFactory {
            TCodecPtr MakeCodec(TStringBuf) const override { 
                return new TCodec; 
            } 
 
            TVector<TString> ListNames() const override {
                TVector<TString> vs;
                vs.push_back(ToString(TCodec::MyName()));
                return vs; 
            } 
        }; 
 
        class TCodecRegistry { 
            using TRegistry = THashMap<TString, TIntrusivePtr<ICodecFactory>>;
            TRegistry Registry; 
 
        public: 
            using TFactoryPtr = TIntrusivePtr<ICodecFactory>; 
 
            TCodecRegistry(); 
 
            void RegisterFactory(TFactoryPtr fac); 
 
            TCodecPtr GetCodec(TStringBuf name) const; 
 
            TVector<TString> GetCodecsList() const;
        }; 
 
    } 
 
    void RegisterCodecFactory(TCodecFactoryPtr fact); 
 
    template <typename TCodec>
    void RegisterCodec() { 
        RegisterCodecFactory(new NPrivate::TInstanceFactory<TCodec>()); 
    } 
 
}