Nuke treeset
This commit is contained in:
parent
45ef1c4271
commit
02e80b3827
5 changed files with 184 additions and 0 deletions
12
filters/Makefile
Normal file
12
filters/Makefile
Normal file
|
|
@ -0,0 +1,12 @@
|
|||
CC ?= gcc
|
||||
CFLAGS ?= -Wall -O2
|
||||
|
||||
TARGET = main.elf
|
||||
SRC = message.c main_demo.c
|
||||
|
||||
$(TARGET): $(SRC)
|
||||
@echo CC $@
|
||||
@$(CC) $(CFLAGS) -o $@ $^ $(LDFLAGS)
|
||||
|
||||
clean:
|
||||
rm -f $(TARGET)
|
||||
71
filters/filters.c
Normal file
71
filters/filters.c
Normal file
|
|
@ -0,0 +1,71 @@
|
|||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#define N 8
|
||||
|
||||
uint16_t buffer[N];
|
||||
uint8_t idx = 0;
|
||||
|
||||
/*
|
||||
* Pros: Simple and smooths well.
|
||||
* Cons: Lag, slow to respond to real changes.
|
||||
*/
|
||||
uint16_t moving_average(uint16_t new_sample) {
|
||||
buffer[idx++] = new_sample;
|
||||
if (idx >= N)
|
||||
idx = 0;
|
||||
|
||||
uint32_t sum = 0;
|
||||
for (int i = 0; i < N; i++) sum += buffer[i];
|
||||
return sum / N;
|
||||
}
|
||||
|
||||
float alpha = 0.1f; // Between 0 (more smoothing) and 1 (less)
|
||||
float filtered = 0;
|
||||
|
||||
/*
|
||||
* Pros: Very lightweight, tunable.
|
||||
* Cons: Might not remove all burst noise
|
||||
*/
|
||||
uint16_t low_pass_filter(uint16_t new_sample) {
|
||||
filtered = alpha * new_sample + (1 - alpha) * filtered;
|
||||
return (uint16_t)filtered;
|
||||
}
|
||||
|
||||
int compare_uint16(const void *a, const void *b) {
|
||||
return (*(uint16_t *)a - *(uint16_t *)b);
|
||||
}
|
||||
|
||||
uint16_t median_filter(uint16_t new_sample) {
|
||||
static uint16_t window[N];
|
||||
static uint8_t index = 0;
|
||||
window[index++] = new_sample;
|
||||
if (index >= N)
|
||||
index = 0;
|
||||
|
||||
// Copy and sort
|
||||
uint16_t sorted[N];
|
||||
memcpy(sorted, window, sizeof(sorted));
|
||||
qsort(sorted, N, sizeof(uint16_t), compare_uint16);
|
||||
return sorted[N / 2];
|
||||
}
|
||||
|
||||
#undef N
|
||||
#define N 5
|
||||
float coeffs[N] = {0.1, 0.15, 0.5, 0.15, 0.1}; // Symmetric LPF
|
||||
float buffer[N] = {0};
|
||||
int index = 0;
|
||||
|
||||
float fir_filter(float sample) {
|
||||
buffer[index] = sample;
|
||||
float result = 0;
|
||||
int i, j = index;
|
||||
for (i = 0; i < N; i++) {
|
||||
result += coeffs[i] * buffer[j];
|
||||
j = (j - 1 + N) % N;
|
||||
}
|
||||
index = (index + 1) % N;
|
||||
return result;
|
||||
}
|
||||
|
||||
70
lua_advanced/esc.c
Normal file
70
lua_advanced/esc.c
Normal file
|
|
@ -0,0 +1,70 @@
|
|||
#include <lauxlib.h>
|
||||
#include <lua.h>
|
||||
#include <lualib.h>
|
||||
#include <stdbool.h>
|
||||
|
||||
struct EscConfig {
|
||||
int maxspeed;
|
||||
float odometer;
|
||||
bool started;
|
||||
};
|
||||
|
||||
static struct EscConfig esc_config = {
|
||||
.maxspeed = 100,
|
||||
.odometer = 12.5f,
|
||||
.started = false,
|
||||
};
|
||||
|
||||
/* Lua: get_esc_config() -> table */
|
||||
static int l_get_esc_config(lua_State *L) {
|
||||
lua_newtable(L);
|
||||
|
||||
lua_pushstring(L, "maxspeed");
|
||||
lua_pushinteger(L, esc_config.maxspeed);
|
||||
lua_settable(L, -3);
|
||||
|
||||
lua_pushstring(L, "odometer");
|
||||
lua_pushnumber(L, esc_config.odometer);
|
||||
lua_settable(L, -3);
|
||||
|
||||
lua_pushstring(L, "started");
|
||||
lua_pushboolean(L, esc_config.started);
|
||||
lua_settable(L, -3);
|
||||
|
||||
return 1; // return the table
|
||||
}
|
||||
|
||||
/* Lua: set_maxspeed(value) */
|
||||
static int l_set_maxspeed(lua_State *L) {
|
||||
int maxspeed = luaL_checkinteger(L, 1);
|
||||
esc_config.maxspeed = maxspeed;
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Lua: set_started(value) */
|
||||
static int l_set_started(lua_State *L) {
|
||||
int started = lua_toboolean(L, 1);
|
||||
esc_config.started = started;
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Lua: add_odometer(delta) */
|
||||
static int l_add_odometer(lua_State *L) {
|
||||
float delta = luaL_checknumber(L, 1);
|
||||
esc_config.odometer += delta;
|
||||
return 0;
|
||||
}
|
||||
|
||||
static const struct luaL_Reg esc_funcs[] = {
|
||||
{"get_esc_config", l_get_esc_config},
|
||||
{"set_maxspeed", l_set_maxspeed},
|
||||
{"set_started", l_set_started},
|
||||
{"add_odometer", l_add_odometer},
|
||||
{NULL, NULL},
|
||||
};
|
||||
|
||||
/* Register the functions into Lua as global "esc" */
|
||||
void register_esc(lua_State *L) {
|
||||
luaL_newlib(L, esc_funcs); // create table with all funcs
|
||||
lua_setglobal(L, "esc"); // set global "esc"
|
||||
}
|
||||
4
lua_advanced/esc.h
Normal file
4
lua_advanced/esc.h
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
#pragma once
|
||||
#include <lua.h>
|
||||
|
||||
void register_esc(lua_State *L);
|
||||
27
lua_advanced/esc.lua
Normal file
27
lua_advanced/esc.lua
Normal file
|
|
@ -0,0 +1,27 @@
|
|||
---@meta
|
||||
|
||||
---@class EscConfig
|
||||
---@field maxspeed integer
|
||||
---@field odometer number
|
||||
---@field started boolean
|
||||
|
||||
---@class esc
|
||||
esc = {}
|
||||
|
||||
---Get current ESC config
|
||||
---@return EscConfig
|
||||
function esc.get_esc_config() end
|
||||
|
||||
---Set maximum speed
|
||||
---@param value integer
|
||||
function esc.set_maxspeed(value) end
|
||||
|
||||
---Set started flag
|
||||
---@param value boolean
|
||||
function esc.set_started(value) end
|
||||
|
||||
---Add to odometer
|
||||
---@param delta number
|
||||
function esc.add_odometer(delta) end
|
||||
|
||||
return esc
|
||||
Loading…
Add table
Add a link
Reference in a new issue