initial commit

include/adc_sweep/sweep_core.hpp

@@ -0,0 +1,71 @@
+#pragma once
+
+#include <chrono>
+#include <cstdint>
+#include <deque>
+#include <functional>
+#include <optional>
+#include <vector>
+
+namespace adc_sweep {
+
+struct SweepMeta {
+    uint32_t sweep_idx = 0;
+    int32_t ch_primary = 0;
+    uint32_t ch_mask = 0;
+    float n_valid = 0.0f;
+    float min = 0.0f;
+    float max = 0.0f;
+    float mean = 0.0f;
+    float std = 0.0f;
+    float dt_ms = 0.0f;
+    uint64_t ts_mono_ns = 0;
+};
+
+struct SweepResult {
+    std::vector<float> sweep;
+    SweepMeta meta;
+};
+
+int32_t u32_to_i32(uint32_t v);
+
+class SweepFinalizer {
+public:
+    SweepFinalizer(uint32_t sweep_width, float invert_threshold, bool fancy_fill);
+
+    std::optional<SweepResult> finalize(
+        const std::vector<int>& xs,
+        const std::vector<int32_t>& ys,
+        uint32_t ch_mask,
+        int32_t ch_primary);
+
+private:
+    uint32_t sweep_width_;
+    float invert_threshold_;
+    bool fancy_fill_;
+    uint32_t max_width_seen_ = 0;
+    uint32_t sweep_idx_ = 0;
+    std::optional<std::chrono::steady_clock::time_point> last_sweep_ts_;
+    std::deque<std::pair<std::chrono::steady_clock::time_point, int>> n_valid_hist_;
+};
+
+class BinarySweepParser {
+public:
+    using SweepCallback = std::function<void(const std::vector<int>&, const std::vector<int32_t>&, uint32_t, int32_t)>;
+
+    void feed(const uint8_t* data, size_t size, const SweepCallback& on_sweep);
+    void flush(const SweepCallback& on_sweep);
+
+private:
+    void emit_current(const SweepCallback& on_sweep);
+
+    std::vector<int> xs_;
+    std::vector<int32_t> ys_;
+    bool has_cur_channel_ = false;
+    int32_t cur_channel_ = 0;
+    uint32_t ch_mask_ = 0;
+    std::deque<uint16_t> words_;
+    std::optional<uint8_t> odd_byte_;
+};
+
+}  // namespace adc_sweep