demo.h

Go to the documentation of this file.

 
#ifndef INFUSE_SDK_INCLUDE_INFUSE_ALGORITHM_RUNNER_ALGORITHMS_DEMO_H_
#define INFUSE_SDK_INCLUDE_INFUSE_ALGORITHM_RUNNER_ALGORITHMS_DEMO_H_
 
#include <stdint.h>
 
#include <zephyr/toolchain.h>
 
#include <infuse/algorithm_runner/runner.h>
#include <infuse/zbus/channels.h>
 
#ifdef __cplusplus
extern "C" {
#endif

 
enum {
        ALGORITHM_DEMO_EVENT_LOG = BIT(0),
        ALGORITHM_DEMO_STATE_LOG = BIT(0),
        ALGORITHM_DEMO_METRIC_LOG = BIT(0),
};

struct algorithm_demo_common_args {
        struct kv_algorithm_logging logging;
        /* Algorithm specific arguments */
        union {
                uint8_t event_gen_chance;
                uint16_t compute_metric_len;
        } __packed;
} __packed;
 
/* Algorithm runtime state */
union algorithm_demo_common_data {
        uint32_t processed;
        uint8_t current_state;
};

void algorithm_demo_event_fn(const struct zbus_channel *chan,
                             const struct infuse_algorithm *algorithm, const void *args,
                             void *data);

void algorithm_demo_state_fn(const struct zbus_channel *chan,
                             const struct infuse_algorithm *algorithm, const void *args,
                             void *data);

void algorithm_demo_metric_fn(const struct zbus_channel *chan,
                              const struct infuse_algorithm *algorithm, const void *args,
                              void *data);

#define ALGORITHM_DEMO_EVENT_DEFINE(name, loggers_, tdfs, event_chance_percent)                    \
        static void name##_wrapper(const struct zbus_channel *chan,                                \
                                   const struct infuse_algorithm *algorithm, const void *args)     \
        {                                                                                          \
                static union algorithm_demo_common_data data;                                      \
                algorithm_demo_event_fn(chan, algorithm, args, &data);                             \
        }                                                                                          \
        static struct algorithm_demo_common_args name##_default_args = {                           \
                .logging =                                                                         \
                        {                                                                          \
                                .loggers = loggers_,                                               \
                                .tdf_mask = tdfs,                                                  \
                        },                                                                         \
                .event_gen_chance = event_chance_percent,                                          \
        };                                                                                         \
        static const struct infuse_algorithm name = {                                              \
                .fn = name##_wrapper,                                                              \
                .algorithm_id = 0xFFFFFFF0,                                                        \
                .zbus_channel = INFUSE_ZBUS_CHAN_IMU,                                              \
                .arguments = &name##_default_args,                                                 \
                .arguments_size = sizeof(struct algorithm_demo_common_args),                       \
        }

#define ALGORITHM_DEMO_STATE_DEFINE(name, loggers_, tdfs)                                          \
        static void name##_wrapper(const struct zbus_channel *chan,                                \
                                   const struct infuse_algorithm *algorithm, const void *args)     \
        {                                                                                          \
                static union algorithm_demo_common_data data;                                      \
                algorithm_demo_state_fn(chan, algorithm, args, &data);                             \
        }                                                                                          \
        static struct algorithm_demo_common_args name##_default_args = {                           \
                .logging =                                                                         \
                        {                                                                          \
                                .loggers = loggers_,                                               \
                                .tdf_mask = tdfs,                                                  \
                        },                                                                         \
        };                                                                                         \
        static const struct infuse_algorithm name = {                                              \
                .fn = name##_wrapper,                                                              \
                .algorithm_id = 0xFFFFFFF1,                                                        \
                .zbus_channel = INFUSE_ZBUS_CHAN_IMU,                                              \
                .arguments = &name##_default_args,                                                 \
                .arguments_size = sizeof(struct algorithm_demo_common_args),                       \
        }

#define ALGORITHM_DEMO_METRIC_DEFINE(name, loggers_, tdfs, metric_compute_len)                     \
        static void name##_wrapper(const struct zbus_channel *chan,                                \
                                   const struct infuse_algorithm *algorithm, const void *args)     \
        {                                                                                          \
                static union algorithm_demo_common_data data;                                      \
                algorithm_demo_metric_fn(chan, algorithm, args, &data);                            \
        }                                                                                          \
        static struct algorithm_demo_common_args name##_default_args = {                           \
                .logging =                                                                         \
                        {                                                                          \
                                .loggers = loggers_,                                               \
                                .tdf_mask = tdfs,                                                  \
                        },                                                                         \
                .compute_metric_len = metric_compute_len,                                          \
        };                                                                                         \
        static const struct infuse_algorithm name = {                                              \
                .fn = name##_wrapper,                                                              \
                .algorithm_id = 0xFFFFFFF2,                                                        \
                .zbus_channel = INFUSE_ZBUS_CHAN_IMU,                                              \
                .arguments = &name##_default_args,                                                 \
                .arguments_size = sizeof(struct algorithm_demo_common_args),                       \
        }

 
#ifdef __cplusplus
}
#endif
 
#endif /* INFUSE_SDK_INCLUDE_INFUSE_ALGORITHM_RUNNER_ALGORITHMS_DEMO_H_ */