dynamicDataPointRecordingUnit.hpp

Go to the documentation of this file.

 
#ifndef RTCTK_COMPONENTFRAMEWORK_DYNAMICDATAPOINTRECORDINGUNIT_HPP
#define RTCTK_COMPONENTFRAMEWORK_DYNAMICDATAPOINTRECORDINGUNIT_HPP
 
#include <atomic>
#include <boost/signals2/connection.hpp>
#include <functional>
#include <mutex>
#include <typeindex>
#include <vector>
 
#include <rtctk/componentFramework/recordingUnit.hpp>
#include <rtctk/componentFramework/serviceContainer.hpp>
namespace rtctk::componentFramework {
 
void* FitsDataRecorderFactory(std::type_index);
 
class DynamicDataPointRecordingUnit : public RecordingUnit {
public:
    DynamicDataPointRecordingUnit(
        const std::string& comp_id,
        const std::string& unit_id,
        ServiceContainer& services,
        std::function<void*(std::type_index)> recorder_factory = &FitsDataRecorderFactory);
 
    void Prepare(const std::filesystem::path& file_path) override;
 
    void Start() override;
 
    std::vector<std::filesystem::path> Stop() override;
 
    void Update() override;
 
    void CheckErrors() override;
 
    using DataRecorderFactory = std::function<void*(std::type_index)>;
 
    virtual ~DynamicDataPointRecordingUnit();
 
private:
    std::vector<std::unique_ptr<RecordingUnit>> m_recording_units;
    std::vector<boost::signals2::scoped_connection> m_recording_unit_connections;
 
    ServiceContainer& m_services;
 
    inline static constexpr std::string_view RTR_PATH_DATA_POINT_LIST =
        "/{}/dynamic/rec_units/{}/datapoint_list";
 
    std::function<void*(std::type_index)> m_recorder_factory;
 
    void StateTransitionHandler(const std::string& unit_id, const State& state);
 
    std::map<std::string, State> m_state_map{};
    std::mutex m_state_map_mutex{};
 
    // state counters for the sub recording units units
    std::atomic_size_t m_unit_count{0};
    std::atomic_size_t m_units_idle{0};
    std::atomic_size_t m_units_finished{0};
};
 
}  // namespace rtctk::componentFramework
 
#endif  // RTCTK_COMPONENTFRAMEWORK_DYNAMICDATAPOINTRECORDINGUNIT_HPP