mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-12 12:52:13 +08:00
Mitchell Goff
e6e19f8d3e
* Added all maneuvers to navInstruction message * Added nav instruction logic to modeld * New model: fcee01c1-96bb-414f-b00d-e4a994a00922/700 * Fixed bug in navd * Added sharp/slight modifiers * Updated refs old-commit-hash: c04e5d12fce1801cd555a4846768b68db7090b17
58 lines
1.9 KiB
C++
58 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include "cereal/messaging/messaging.h"
|
|
#include "cereal/visionipc/visionipc_client.h"
|
|
#include "common/util.h"
|
|
#include "common/modeldata.h"
|
|
#include "selfdrive/modeld/models/commonmodel.h"
|
|
#include "selfdrive/modeld/runners/run.h"
|
|
|
|
constexpr int NAV_INPUT_SIZE = 256*256;
|
|
constexpr int NAV_FEATURE_LEN = 256;
|
|
constexpr int NAV_INSTRUCTION_LEN = 150;
|
|
constexpr int NAV_DESIRE_LEN = 32;
|
|
|
|
struct NavModelOutputXY {
|
|
float x;
|
|
float y;
|
|
};
|
|
static_assert(sizeof(NavModelOutputXY) == sizeof(float)*2);
|
|
|
|
struct NavModelOutputPlan {
|
|
std::array<NavModelOutputXY, TRAJECTORY_SIZE> mean;
|
|
std::array<NavModelOutputXY, TRAJECTORY_SIZE> std;
|
|
};
|
|
static_assert(sizeof(NavModelOutputPlan) == sizeof(NavModelOutputXY)*TRAJECTORY_SIZE*2);
|
|
|
|
struct NavModelOutputDesirePrediction {
|
|
std::array<float, NAV_DESIRE_LEN> values;
|
|
};
|
|
static_assert(sizeof(NavModelOutputDesirePrediction) == sizeof(float)*NAV_DESIRE_LEN);
|
|
|
|
struct NavModelOutputFeatures {
|
|
std::array<float, NAV_FEATURE_LEN> values;
|
|
};
|
|
static_assert(sizeof(NavModelOutputFeatures) == sizeof(float)*NAV_FEATURE_LEN);
|
|
|
|
struct NavModelResult {
|
|
const NavModelOutputPlan plan;
|
|
const NavModelOutputDesirePrediction desire_pred;
|
|
const NavModelOutputFeatures features;
|
|
float dsp_execution_time;
|
|
};
|
|
static_assert(sizeof(NavModelResult) == sizeof(NavModelOutputPlan) + sizeof(NavModelOutputDesirePrediction) + sizeof(NavModelOutputFeatures) + sizeof(float));
|
|
|
|
constexpr int NAV_OUTPUT_SIZE = sizeof(NavModelResult) / sizeof(float);
|
|
constexpr int NAV_NET_OUTPUT_SIZE = NAV_OUTPUT_SIZE - 1;
|
|
|
|
struct NavModelState {
|
|
RunModel *m;
|
|
uint8_t net_input_buf[NAV_INPUT_SIZE];
|
|
float output[NAV_OUTPUT_SIZE];
|
|
};
|
|
|
|
void navmodel_init(NavModelState* s);
|
|
NavModelResult* navmodel_eval_frame(NavModelState* s, VisionBuf* buf);
|
|
void navmodel_publish(PubMaster &pm, VisionIpcBufExtra &extra, const NavModelResult &model_res, float execution_time, bool route_valid);
|
|
void navmodel_free(NavModelState* s);
|