Initial PoC for Datalog editing API

datalog/src/main/native/cpp/DataLogEditor.cpp

@@ -0,0 +1,65 @@
+#include "wpi/datalog/DataLogEditor.hpp"
+#include <algorithm>
+#include <cstddef>
+#include <memory>
+#include <string>
+#include <utility>
+#include <vector>
+#include "wpi/datalog/DataLogReader.hpp"
+#include "wpi/datalog/DataLogReaderThread.hpp"
+
+namespace wpi::log {
+
+  DataLogEditor::DataLogEditor(std::unique_ptr<DataLogReaderThread> datalog) {
+    state = std::make_shared<DataLogEditInfo>();
+    state->readerThread = std::move(datalog);
+  }
+
+  DataLogEditor DataLogEditor::WithTimestamps(std::vector<std::pair<int64_t, int64_t>> timestamps) const {
+    state->filterTimestamps = true;
+    state->allowedTimestamps = timestamps;
+    return DataLogEditor{state};
+  }
+
+  DataLogEditor DataLogEditor::WithEntryNames(std::vector<std::string> entryNames) const {
+    state->filterEntryNames = true;
+    state->allowedEntryNames = entryNames;
+    return DataLogEditor{state};
+  }
+
+  DataLogEditor DataLogEditor::WithRenamedEntries(std::map<std::string, std::string> renames) const {
+    state->renameEntries = true;
+    state->entryRenames = std::move(renames);
+    return DataLogEditor{state};
+  }
+
+  std::vector<DataLogRecord> DataLogEditor::ExecuteEdits() {
+    // we need to do the renames before we drop entries
+    const DataLogReaderEntry* entryData;
+    for (const DataLogRecord& record : state->records) {
+      entryData = state->readerThread->GetEntry(record.GetEntry());
+      // exclude based on timestamp
+      auto timestamp = record.GetTimestamp();
+      // if we allow this timestamp or it is a start/finish record, add it
+      if (std::ranges::any_of(state->allowedTimestamps, [timestamp](const std::pair<int64_t, int64_t>& pair) {
+        return timestamp >= pair.first && timestamp <= pair.second;
+      }) || record.IsControl()) {
+        state->records.push_back(record);
+      }
+
+
+      if (record.IsStart() && state->entryRenames.contains(std::string{entryData->name})) {
+        // its the start record of the one we want and its guaranteed to be in the filtered list so lets replace it with a copy that includes the correct name
+        StartRecordData srd;
+        record.GetStartData(&srd);
+        srd.name = state->entryRenames[std::string{srd.name}];
+        auto it = std::find_if(state->records.begin(), state->records.end(), [srd](DataLogRecord& targetRecord) {
+          return targetRecord.GetEntry() == srd.entry;
+        });
+        ptrdiff_t idx = it - state->records.begin();
+        state->records.at(idx) = DataLogRecord{entryData->entry, timestamp, reinterpret_cast<std::span<uint8_t>>(srd)};
+      }
+    }
+    return state->records;
+  }
+}

datalog/src/main/native/include/wpi/datalog/DataLogEditor.hpp

@@ -0,0 +1,46 @@
+#pragma once
+
+#include <cstdint>
+#include <map>
+#include <memory>
+#include <utility>
+#include <vector>
+#include "wpi/datalog/DataLogReader.hpp"
+#include "wpi/datalog/DataLogReaderThread.hpp"
+
+namespace wpi::log {
+  class DataLogEditor {
+    public:
+    explicit DataLogEditor(std::unique_ptr<DataLogReaderThread> datalog);
+
+    DataLogEditor WithTimestamps(std::vector<std::pair<int64_t, int64_t>> timestamps) const;
+
+    DataLogEditor WithEntryNames(std::vector<std::string> entryNames) const;
+
+    DataLogEditor WithRenamedEntries(std::map<std::string, std::string> renames) const;
+
+    std::vector<DataLogRecord> ExecuteEdits();
+
+    private:
+    struct DataLogEditInfo {
+      // record list (input to the filter function)
+      std::vector<DataLogRecord> records;
+      // used to get the initial record list and to check what entry a record is from
+      std::unique_ptr<DataLogReaderThread> readerThread;
+
+      // edit stage enable flags
+      bool filterTimestamps = false;
+      bool filterEntryNames = false;
+      bool renameEntries = false;
+
+      // edit info
+      std::vector<std::pair<int64_t, int64_t>> allowedTimestamps;
+      std::vector<std::string> allowedEntryNames;
+      std::map<std::string, std::string> entryRenames;
+    };
+
+    explicit DataLogEditor(std::shared_ptr<DataLogEditInfo> config) : state(std::move(config)) {}
+
+    std::shared_ptr<DataLogEditInfo> state;
+  };
+}