aboutsummaryrefslogtreecommitdiff
path: root/lib/embed.h
blob: b57d830f69567a498bd8179225faff8d3dfd833e (plain)
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
88
#pragma once

#include <array>
#include <string_view>
#include "compressionctx.h"

namespace embed
{

template <std::size_t N>
class Resources
{
public:
    constexpr explicit Resources(
        const std::array<const char *, N> &entries,
        const std::array<std::span<const uint8_t>, N> &values)
        : m_entries(entries)
        , m_values(values)
    {
    }

    [[nodiscard]] constexpr int id(const std::string_view &path) const
    {
        for(std::size_t i = 0; i < m_entries.size(); ++i) {
            if(m_entries.at(i) == path)
                return i;
        }
        return -1;
    }
    [[nodiscard]] constexpr std::span<const uint8_t> value(const int id_) const
    {
        if(id_ == -1)
            return {};
        else
            return m_values.at(id_);
    }
    [[nodiscard]] constexpr std::span<const uint8_t> value(const std::string_view &path) const
    {
        return value(id(path));
    }

    [[deprecated("With no compression, this function returns a copy; use ::value instead")]]
    [[nodiscard]] std::vector<uint8_t> decompress(const int id_) const
    {
        const auto v = value(id_);
        return { v.begin(), v.end() };
    }
    [[deprecated("With no compression, this function returns a copy; use ::value instead")]]
    [[nodiscard]] std::vector<uint8_t> decompress(const std::string_view &path) const
    {
        const auto v = value(path);
        return { v.begin(), v.end() };
    }

protected:
    const std::array<const char *, N> m_entries;
    const std::array<std::span<const uint8_t>, N> m_values;
};

template <std::size_t N>
class CompressedResources final : public Resources<N>
{
public:
    explicit CompressedResources(
        const std::array<const char *, N> &entries,
        const std::array<std::span<const uint8_t>, N> &values,
        const Compression algo,
        const std::span<const uint8_t> &dictionary = {})
        : Resources<N>(entries, values)
        , m_compression(make_compression_ctx(algo, dictionary))
    {
    }
    ~CompressedResources() = default;

    [[nodiscard]] std::vector<uint8_t> decompress(const int id_) const
    {
        return m_compression->decompress(Resources<N>::value(id_));
    }
    [[nodiscard]] std::vector<uint8_t> decompress(const std::string_view &path) const
    {
        return m_compression->decompress(Resources<N>::value(path));
    }

protected:
    const std::unique_ptr<CompressionCtx> m_compression;
};

} // namespace embed