flash/flash.hpp

120 lines
2.6 KiB
C++

#pragma once
#include <type_traits>
#include <cassert>
#include <cstddef>
#include <avr/pgmspace.h>
#include "../util/util.hpp"
#define F(str) (reinterpret_cast<const ::detail::FlashString *>(PSTR(str)))
#define GF(name, str) \
const char __##name[] PROGMEM = str; \
const auto *name = reinterpret_cast<const ::detail::FlashString *>(__##name)
namespace detail {
// Only used for C++ type safety, because otherwise a PSTR would be indistinguishable from a normal c-string
struct FlashString;
} // namespace detail
namespace flash {
template <typename T>
static inline T load(const T &object)
{
T buffer;
auto byteBuffer = reinterpret_cast<std::byte *>(&buffer);
for (auto i = std::size_t{0}; i < sizeof(T); ++i) {
byteBuffer[i] = static_cast<std::byte>(pgm_read_byte(&reinterpret_cast<const std::byte *>(&object)[i]));
}
return buffer;
}
template <typename T>
struct [[gnu::progmem]] Wrapper
{
using value_type = T;
inline T operator*() const
{
return load(value);
}
inline std::byte operator[](const std::size_t idx) const
{
const auto bytePtr = reinterpret_cast<const std::byte *>(&value);
return load(bytePtr[idx]);
}
const T value;
};
template <typename T>
struct RamWrapper {
using value_type = T;
inline constexpr T &operator*()
{
return value;
}
inline constexpr const T &operator*() const
{
return value;
}
inline std::byte &operator[](const std::size_t idx)
{
const auto bytePtr = reinterpret_cast<std::byte *>(&value);
return bytePtr[idx];
}
inline const std::byte &operator[](const std::size_t idx) const
{
const auto bytePtr = reinterpret_cast<const std::byte *>(&value);
return bytePtr[idx];
}
T value;
};
template <typename WrapperT>
struct is_flash_wrapper : std::false_type {
};
template <typename T>
struct is_flash_wrapper<Wrapper<T>> : std::true_type {
};
template <typename WrapperT>
static inline constexpr auto is_flash_wrapper_v = is_flash_wrapper<WrapperT>::value;
template <typename WrapperT>
struct is_ram_wrapper : std::false_type {
};
template <typename T>
struct is_ram_wrapper<RamWrapper<T>> : std::true_type {
};
template <typename WrapperT>
static inline constexpr auto is_ram_wrapper_v = is_ram_wrapper<WrapperT>::value;
template <typename WrapperT, typename T>
static inline decltype(auto) loadLike(const T &object)
{
if constexpr (is_ram_wrapper_v<std::remove_cvref_t<WrapperT>>) {
return object;
} else if constexpr (is_flash_wrapper_v<std::remove_cvref_t<WrapperT>>) {
return load(object);
} else {
static_assert(util::always_false_v<T>, "Invalid wrapper type");
}
}
} // namespace flash