mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-06 05:52:12 +08:00
George Hotz
7119a98414
* start v4l encoder * v4l encoder starts * start and stop * fill in proper controls * it dequeued a buffer * getting bytes * it made a video * it does make files * getting close * ahh, so that's how dequeue works * qcam works (no remuxing) * remuxing works * we just need to make shutdown and rollover graceful * graceful destruction * switch to polling * should work now * fix pc build * refactors, stop properly * touchups, remove a copy * add v4l encoder to release * inlcude file * move writing to it's own thread * fix minor memory leak * block instead of dropping frames * add counter, fix tests maybe * better debugging and test print * print file path in assert * format string in test * no more oversized qlogs * match qcam * touchups, remove omx encoder * remove omx include files * checked ioctl, better debugging, open by name * unused import * move linux includes to third_party/linux/include * simple encoderd * full packet * encoderd should be complete * lagging print * updates * name dq thread * subset idx * video file writing works * debug * potential bugfix * rotation works * iframe * keep writing support * ci should pass * loggerd, not encoderd * remote encoder code * support remote encoder * cereal to master, add encoderd * header no longer required * put that back there * realtime * lower decoder latency * don't use queue for VisionIpcBufExtra, disable realtime again * assert all written * hmm simpler * only push to to_write if we are writing * assert timestamp is right * use at and remove assert * revert to queue Co-authored-by: Comma Device <device@comma.ai> old-commit-hash: 0baa4c3e2ad9ee6f8daba8267db44c2cd44caa62
135 lines
3.5 KiB
C++
135 lines
3.5 KiB
C++
#pragma once
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <atomic>
|
|
#include <cassert>
|
|
#include <cerrno>
|
|
#include <condition_variable>
|
|
#include <mutex>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <unordered_map>
|
|
|
|
#include "cereal/messaging/messaging.h"
|
|
#include "cereal/services.h"
|
|
#include "cereal/visionipc/visionipc.h"
|
|
#include "cereal/visionipc/visionipc_client.h"
|
|
#include "selfdrive/camerad/cameras/camera_common.h"
|
|
#include "selfdrive/common/params.h"
|
|
#include "selfdrive/common/swaglog.h"
|
|
#include "selfdrive/common/timing.h"
|
|
#include "selfdrive/common/util.h"
|
|
#include "selfdrive/hardware/hw.h"
|
|
|
|
#include "selfdrive/loggerd/encoder.h"
|
|
#include "selfdrive/loggerd/logger.h"
|
|
#ifdef QCOM2
|
|
#include "selfdrive/loggerd/v4l_encoder.h"
|
|
#define Encoder V4LEncoder
|
|
#else
|
|
#include "selfdrive/loggerd/raw_logger.h"
|
|
#define Encoder RawLogger
|
|
#endif
|
|
|
|
constexpr int MAIN_FPS = 20;
|
|
const int MAIN_BITRATE = Hardware::TICI() ? 10000000 : 5000000;
|
|
const int DCAM_BITRATE = Hardware::TICI() ? MAIN_BITRATE : 2500000;
|
|
|
|
#define NO_CAMERA_PATIENCE 500 // fall back to time-based rotation if all cameras are dead
|
|
|
|
const bool LOGGERD_TEST = getenv("LOGGERD_TEST");
|
|
const int SEGMENT_LENGTH = LOGGERD_TEST ? atoi(getenv("LOGGERD_SEGMENT_LENGTH")) : 60;
|
|
|
|
struct LogCameraInfo {
|
|
CameraType type;
|
|
const char *filename;
|
|
VisionStreamType stream_type;
|
|
int frame_width, frame_height;
|
|
int fps;
|
|
int bitrate;
|
|
bool is_h265;
|
|
bool has_qcamera;
|
|
bool trigger_rotate;
|
|
bool enable;
|
|
bool record;
|
|
};
|
|
|
|
const LogCameraInfo cameras_logged[] = {
|
|
{
|
|
.type = RoadCam,
|
|
.stream_type = VISION_STREAM_ROAD,
|
|
.filename = "fcamera.hevc",
|
|
.fps = MAIN_FPS,
|
|
.bitrate = MAIN_BITRATE,
|
|
.is_h265 = true,
|
|
.has_qcamera = true,
|
|
.trigger_rotate = true,
|
|
.enable = true,
|
|
.record = true,
|
|
.frame_width = 1928,
|
|
.frame_height = 1208,
|
|
},
|
|
{
|
|
.type = DriverCam,
|
|
.stream_type = VISION_STREAM_DRIVER,
|
|
.filename = "dcamera.hevc",
|
|
.fps = MAIN_FPS,
|
|
.bitrate = DCAM_BITRATE,
|
|
.is_h265 = true,
|
|
.has_qcamera = false,
|
|
.trigger_rotate = true,
|
|
.enable = true,
|
|
.record = Params().getBool("RecordFront"),
|
|
.frame_width = 1928,
|
|
.frame_height = 1208,
|
|
},
|
|
{
|
|
.type = WideRoadCam,
|
|
.stream_type = VISION_STREAM_WIDE_ROAD,
|
|
.filename = "ecamera.hevc",
|
|
.fps = MAIN_FPS,
|
|
.bitrate = MAIN_BITRATE,
|
|
.is_h265 = true,
|
|
.has_qcamera = false,
|
|
.trigger_rotate = true,
|
|
.enable = Hardware::TICI(),
|
|
.record = Hardware::TICI(),
|
|
.frame_width = 1928,
|
|
.frame_height = 1208,
|
|
},
|
|
};
|
|
const LogCameraInfo qcam_info = {
|
|
.filename = "qcamera.ts",
|
|
.fps = MAIN_FPS,
|
|
.bitrate = 256000,
|
|
.is_h265 = false,
|
|
.enable = true,
|
|
.record = true,
|
|
.frame_width = Hardware::TICI() ? 526 : 480,
|
|
.frame_height = Hardware::TICI() ? 330 : 360 // keep pixel count the same?
|
|
};
|
|
|
|
struct LoggerdState {
|
|
LoggerState logger = {};
|
|
char segment_path[4096];
|
|
std::mutex rotate_lock;
|
|
std::condition_variable rotate_cv;
|
|
std::atomic<int> rotate_segment;
|
|
std::atomic<double> last_camera_seen_tms;
|
|
std::atomic<int> ready_to_rotate; // count of encoders ready to rotate
|
|
int max_waiting = 0;
|
|
double last_rotate_tms = 0.; // last rotate time in ms
|
|
|
|
// Sync logic for startup
|
|
std::atomic<int> encoders_ready = 0;
|
|
std::atomic<uint32_t> start_frame_id = 0;
|
|
bool camera_ready[WideRoadCam + 1] = {};
|
|
bool camera_synced[WideRoadCam + 1] = {};
|
|
};
|
|
|
|
bool sync_encoders(LoggerdState *s, CameraType cam_type, uint32_t frame_id);
|
|
bool trigger_rotate_if_needed(LoggerdState *s, int cur_seg, uint32_t frame_id);
|
|
void rotate_if_needed(LoggerdState *s);
|
|
void loggerd_thread();
|