computation.hpp

Go to the documentation of this file.

 
#ifndef RTCTK_EXAMPLEDATATASK_MEASURETEL_COMPUTATION_HPP
#define RTCTK_EXAMPLEDATATASK_MEASURETEL_COMPUTATION_HPP
 
#include <rtctk/componentFramework/stopToken.hpp>
#include <rtctk/dataTask/computationBase.hpp>
#include <rtctk/exampleTopics/topics.hpp>
 
#include <chrono>
#include <mutex>
#include <optional>
#include <vector>
 
namespace rtctk::exampleDataTask {
 
// the type of the shm supertopic
using TopicType = rtctk::exampleTopic::ScaoLoopTopic;
 
// type of the user-owned sample buffer
using BufferType = std::vector<decltype(decltype(TopicType::wfs)::slopes)>;
 
class Computation : public dataTask::ComputationBase<TopicType> {
public:
    using ServiceContainer = rtctk::componentFramework::ServiceContainer;
 
    enum class Algorithm : uint8_t { Mean, Median };
 
    struct Result {
        std::vector<float> avg_values;
        uint32_t some_scalar;
    };
 
    struct ResultIf {
        virtual ~ResultIf() = default;
        virtual void OnResultAvailable(const Result& result) = 0;
    };
 
    Computation(ServiceContainer& services,
                ResultIf& publisher,
                const std::string& shm_name,
                size_t to_read,
                size_t to_skip,
                std::chrono::milliseconds sample_timeout,
                std::optional<numapp::NumaPolicies> thread_policies);
 
    ~Computation();
 
    Result& ComputeSync(Algorithm algorithm);
 
private:
    void OnCycleStart(size_t to_read) override;
 
    void CopyData(size_t sample_idx, const TopicType& sample) noexcept override;
 
    void Compute() override;
 
    void Publish() override;
 
    ResultIf& m_publisher;
    Algorithm m_algorithm;
    BufferType m_sample_buffer;
    Result m_result;
};
 
}  // namespace rtctk::exampleDataTask
 
#endif  // RTCTK_EXAMPLEDATATASK_MEASURETEL_COMPUTATION_HPP