mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-14 19:44:05 +08:00
Dean Lee
0f49ecbff7
* simplify seek&merge * update test cases update test cases * cleanup test cases * new function currentSeconds * add TODO * thread safe publishFrame * cleanup * fix 'at x S' not printed if seek back to old time * exit replay if failed to load route * move out setCurrentSegment from if statement * cleanup * use std::find * const variables old-commit-hash: 5527736df6ffb1f94be48d6a84245165fdf0cac0
66 lines
1.5 KiB
C++
66 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <QDir>
|
|
#include <QObject>
|
|
#include <QString>
|
|
#include <vector>
|
|
|
|
#include "selfdrive/common/util.h"
|
|
#include "selfdrive/ui/replay/framereader.h"
|
|
#include "selfdrive/ui/replay/logreader.h"
|
|
|
|
const QDir CACHE_DIR(util::getenv("COMMA_CACHE", "/tmp/comma_download_cache/").c_str());
|
|
const int connections_per_file = 3;
|
|
|
|
struct SegmentFile {
|
|
QString rlog;
|
|
QString qlog;
|
|
QString road_cam;
|
|
QString driver_cam;
|
|
QString wide_road_cam;
|
|
QString qcamera;
|
|
};
|
|
|
|
class Route {
|
|
public:
|
|
Route(const QString &route);
|
|
bool load();
|
|
inline const QString &name() const { return route_; };
|
|
inline int size() const { return segments_.size(); }
|
|
inline SegmentFile &at(int n) { return segments_[n]; }
|
|
|
|
protected:
|
|
bool loadFromJson(const QString &json);
|
|
QString route_;
|
|
std::vector<SegmentFile> segments_;
|
|
};
|
|
|
|
class Segment : public QObject {
|
|
Q_OBJECT
|
|
|
|
public:
|
|
Segment(int n, const SegmentFile &segment_files, bool load_dcam, bool load_ecam);
|
|
~Segment();
|
|
inline bool isLoaded() const { return loaded_; }
|
|
|
|
std::unique_ptr<LogReader> log;
|
|
std::unique_ptr<FrameReader> frames[MAX_CAMERAS] = {};
|
|
|
|
signals:
|
|
void loadFinished();
|
|
|
|
protected:
|
|
void load();
|
|
void downloadFile(const QString &url);
|
|
QString localPath(const QUrl &url);
|
|
|
|
std::atomic<bool> loaded_ = false;
|
|
std::atomic<bool> aborting_ = false;
|
|
std::atomic<int> downloading_ = 0;
|
|
int seg_num_ = 0;
|
|
SegmentFile files_;
|
|
QString road_cam_path_;
|
|
QString log_path_;
|
|
std::vector<QThread*> download_threads_;
|
|
};
|