computation.hpp

Go to the documentation of this file.

 
#ifndef RTCTK_EXAMPLEDATATASK_MEASURETEL_COMPUTATION_HPP
#define RTCTK_EXAMPLEDATATASK_MEASURETEL_COMPUTATION_HPP
 
#include "rtctk/componentFramework/exceptions.hpp"
#include "rtctk/exampleTopics/topics.hpp"
 
#include <vector>
 
class Computation {
public:
    using ScaoLoopTopic = rtctk::exampleTopic::ScaoLoopTopic;
 
    enum class Algorithm { Mean, Median };
 
    Computation() {
        // get number of slopes from array
        constexpr size_t NUM_SLOPES =
            std::tuple_size<decltype(decltype(ScaoLoopTopic::wfs)::slopes)>::value;
        m_result.resize(NUM_SLOPES);
        Reset();
    }
 
    void Reset() {
        m_sample_idx = 0;
    }
 
    void SetDynamicConfig(unsigned to_read) {
        // pre-allocate memory
        m_samples.resize(to_read);
 
        m_samples_to_read = to_read;
        Reset();
    }
 
    void OnDataAvailable(ScaoLoopTopic const& sample) {
        m_samples[m_sample_idx++] = sample.wfs.slopes;
    }
 
    struct Result {
        std::vector<float> const& avg_values;
    };
 
    Result Compute(Algorithm algorithm) {
        using rtctk::componentFramework::RtctkException;
 
        if (m_sample_idx != m_samples_to_read) {
            CII_THROW(RtctkException, "not all data available");
        }
 
        if (algorithm == Algorithm::Mean) {
            ComputeMean();
        } else {
            CII_THROW(RtctkException, "algorithm not implemented");
        }
 
        Reset();
        return {m_result};
    }
 
    void ComputeMean() {
        for (auto& slope : m_result) {
            slope = 0;
        }
 
        for (auto const& sample : m_samples) {
            size_t idx = 0;
            for (auto const& slope : sample) {
                m_result[idx++] += slope;
            }
        }
 
        for (auto& slope : m_result) {
            slope = slope / m_samples.size();
        }
    }
 
private:
    unsigned m_sample_idx;
    unsigned m_samples_to_read;
    std::vector<decltype(decltype(ScaoLoopTopic::wfs)::slopes)> m_samples;
    std::vector<float> m_result;
};
 
#endif  // RTCTK_EXAMPLEDATATASK_MEASURETEL_COMPUTATION_HPP