stage_registry.h

Go to the documentation of this file.

#pragma once
 
#include "stage/stage.h"
#include "fzm_format.h"
 
#include <cstddef>
#include <cstdint>
 
namespace fz {
 
using StageHeaderFactory = Stage* (*)(const uint8_t* config, size_t config_size);
 
void registerStageHeaderFactory(StageType type, StageHeaderFactory fn);
 
bool hasStageHeaderFactory(StageType type);
 
Stage* createStage(StageType type, const uint8_t* config, size_t config_size);
 
namespace detail {
struct StageFactoryRegistrar {
    StageFactoryRegistrar(StageType type, StageHeaderFactory fn) {
        registerStageHeaderFactory(type, fn);
    }
};
}  // namespace detail
 
}  // namespace fz
 
/* Token-paste helpers so __LINE__ expands before concatenation. */
#define FZ_STAGE_CONCAT_(a, b) a##b
#define FZ_STAGE_CONCAT(a, b) FZ_STAGE_CONCAT_(a, b)
 
#define FZ_REGISTER_STAGE_FACTORY(TYPE, FN)                                     \
    namespace {                                                                 \
    const ::fz::detail::StageFactoryRegistrar                                   \
        FZ_STAGE_CONCAT(FZ_STAGE_FACTORY_REGISTRAR_, __LINE__){(TYPE), (FN)};                    \
    }
 
#define FZ_REGISTER_SIMPLE_STAGE(TYPE, STAGE_CLASS)                             \
    namespace {                                                                 \
    ::fz::Stage* FZ_STAGE_CONCAT(FZ_STAGE_SIMPLE_FACTORY_, __LINE__)(const uint8_t* config,      \
                                                    size_t config_size) {       \
        auto* s = new STAGE_CLASS();                                            \
        s->deserializeHeader(config, config_size);                             \
        return s;                                                               \
    }                                                                           \
    const ::fz::detail::StageFactoryRegistrar                                   \
        FZ_STAGE_CONCAT(FZ_STAGE_SIMPLE_REGISTRAR_, __LINE__){(TYPE),                            \
            &FZ_STAGE_CONCAT(FZ_STAGE_SIMPLE_FACTORY_, __LINE__)};                              \
    }