You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.5 KiB
95 lines
2.5 KiB
4 weeks ago
|
#pragma once
|
||
|
|
|
||
|
|
#include <ctime>
|
||
|
|
#include <map>
|
||
|
|
#include <memory>
|
||
|
|
#include <mutex>
|
||
|
|
#include <string>
|
||
|
|
#include <thread>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "tools/replay/framereader.h"
|
||
|
|
#include "tools/replay/logreader.h"
|
||
|
|
#include "tools/replay/util.h"
|
||
|
|
|
||
|
|
enum class RouteLoadError {
|
||
|
|
None,
|
||
|
|
Unauthorized,
|
||
|
|
AccessDenied,
|
||
|
|
NetworkError,
|
||
|
|
FileNotFound,
|
||
|
|
UnknownError
|
||
|
|
};
|
||
|
|
|
||
|
|
struct RouteIdentifier {
|
||
|
|
std::string dongle_id;
|
||
|
|
std::string timestamp;
|
||
|
|
int begin_segment = 0;
|
||
|
|
int end_segment = -1;
|
||
|
|
std::string str;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct SegmentFile {
|
||
|
|
std::string rlog;
|
||
|
|
std::string qlog;
|
||
|
|
std::string road_cam;
|
||
|
|
std::string driver_cam;
|
||
|
|
std::string wide_road_cam;
|
||
|
|
std::string qcamera;
|
||
|
|
};
|
||
|
|
|
||
|
|
class Route {
|
||
|
|
public:
|
||
|
|
Route(const std::string &route, const std::string &data_dir = {}, bool auto_source = false);
|
||
|
|
bool load();
|
||
|
|
RouteLoadError lastError() const { return err_; }
|
||
|
|
inline const std::string &name() const { return route_.str; }
|
||
|
|
inline const std::time_t datetime() const { return date_time_; }
|
||
|
|
inline const std::string &dir() const { return data_dir_; }
|
||
|
|
inline const RouteIdentifier &identifier() const { return route_; }
|
||
|
|
inline const std::map<int, SegmentFile> &segments() const { return segments_; }
|
||
|
|
inline const SegmentFile &at(int n) { return segments_.at(n); }
|
||
|
|
static RouteIdentifier parseRoute(const std::string &str);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
bool loadSegments();
|
||
|
|
bool loadFromAutoSource();
|
||
|
|
bool loadFromLocal();
|
||
|
|
bool loadFromServer(int retries = 3);
|
||
|
|
bool loadFromJson(const std::string &json);
|
||
|
|
void addFileToSegment(int seg_num, const std::string &file);
|
||
|
|
RouteIdentifier route_ = {};
|
||
|
|
std::string data_dir_;
|
||
|
|
std::map<int, SegmentFile> segments_;
|
||
|
|
std::time_t date_time_ = 0;
|
||
|
|
RouteLoadError err_ = RouteLoadError::None;
|
||
|
|
bool auto_source_ = false;
|
||
|
|
std::string route_string_;
|
||
|
|
};
|
||
|
|
|
||
|
|
class Segment {
|
||
|
|
public:
|
||
|
|
enum class LoadState {Loading, Loaded, Failed};
|
||
|
|
|
||
|
|
Segment(int n, const SegmentFile &files, uint32_t flags, const std::vector<bool> &filters,
|
||
|
|
std::function<void(int, bool)> callback);
|
||
|
|
~Segment();
|
||
|
|
LoadState getState();
|
||
|
|
|
||
|
|
const int seg_num = 0;
|
||
|
|
std::unique_ptr<LogReader> log;
|
||
|
|
std::unique_ptr<FrameReader> frames[MAX_CAMERAS] = {};
|
||
|
|
|
||
|
|
protected:
|
||
|
|
void loadFile(int id, const std::string file);
|
||
|
|
|
||
|
|
std::atomic<bool> abort_ = false;
|
||
|
|
std::atomic<int> loading_ = 0;
|
||
|
|
std::mutex mutex_;
|
||
|
|
std::vector<std::thread> threads_;
|
||
|
|
std::function<void(int, bool)> on_load_finished_ = nullptr;
|
||
|
|
uint32_t flags;
|
||
|
|
std::vector<bool> filters_;
|
||
|
|
LoadState load_state_ = LoadState::Loading;
|
||
|
|
};
|