press_down_detect.c 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639
  1. #include "press_down_detect.h"
  2. #define down_thresh 2000.0f
  3. int back_jump_stage = 0;
  4. int cancel_down = 0;
  5. int virus_flag = 0;
  6. float real_front_min_left = 50000.f;
  7. float real_front_min_right = 50000.f;
  8. void dual_foot_detect_up_trend(float* left_mag_up_min, int *left_up_count, float* left_mag_window,
  9. float* right_mag_up_min, int *right_up_count ,float* right_mag_window, int window_size)
  10. {
  11. float left_max_val = left_mag_window[0];
  12. float left_min_val = left_mag_window[0];
  13. int left_max_index = 0;
  14. int left_min_index = 0;
  15. float right_max_val = right_mag_window[0];
  16. float right_min_val = right_mag_window[0];
  17. int right_max_index = 0;
  18. int right_min_index = 0;
  19. for (int i = 1; i < window_size; i++)
  20. {
  21. if (left_max_val < left_mag_window[i])
  22. {
  23. left_max_val = left_mag_window[i];
  24. left_max_index = i;
  25. }
  26. if (left_min_val > left_mag_window[i])
  27. {
  28. left_min_val = left_mag_window[i];
  29. left_min_index = i;
  30. }
  31. if (right_max_val < right_mag_window[i])
  32. {
  33. right_max_val = right_mag_window[i];
  34. right_max_index = i;
  35. }
  36. if (right_min_val > right_mag_window[i])
  37. {
  38. right_min_val = right_mag_window[i];
  39. right_min_index = i;
  40. }
  41. }
  42. if (((left_max_val - left_min_val > 100.0f && left_max_index > left_min_index) && left_max_val - left_mag_window[window_size - 1] < 1000.f) &&
  43. ((right_max_val - right_min_val > 100.0f && right_max_index > right_min_index) && right_max_val - right_mag_window[window_size - 1] < 1000.f))
  44. {
  45. if (left_min_val < *left_mag_up_min)
  46. {
  47. *left_mag_up_min = left_min_val;
  48. }
  49. if (right_min_val < *right_mag_up_min)
  50. {
  51. *right_mag_up_min = right_min_val;
  52. }
  53. if (*left_up_count > 0)
  54. {
  55. *left_up_count = *left_up_count + 1;
  56. }
  57. else
  58. {
  59. int index = left_min_index > right_min_index ? left_min_index : right_min_index;
  60. *left_mag_up_min = left_mag_window[index];
  61. for (int i = 0; i <= index; i++)
  62. {
  63. left_mag_window[i] = left_mag_window[index];
  64. }
  65. *left_up_count = window_size - index;
  66. }
  67. if (*right_up_count > 0)
  68. {
  69. *right_up_count = *right_up_count + 1;
  70. }
  71. else
  72. {
  73. int index = left_min_index > right_min_index ? left_min_index : right_min_index;
  74. *right_mag_up_min = right_mag_window[index];
  75. for (int i = 0; i <= index; i++)
  76. {
  77. right_mag_window[i] = right_mag_window[index];
  78. }
  79. *right_up_count = window_size - index;
  80. }
  81. }
  82. else
  83. {
  84. *left_mag_up_min = 40000.f;
  85. *right_mag_up_min = 40000.f;
  86. *left_up_count = 0;
  87. *right_up_count = 0;
  88. }
  89. }
  90. int avoid_down_during_change_road(float* left_mag_window, float* right_mag_window, int window_size)
  91. {
  92. float left_max_val = left_mag_window[0];
  93. float left_min_val = left_mag_window[0];
  94. int left_max_index = 0;
  95. int left_min_index = 0;
  96. float right_max_val = right_mag_window[0];
  97. float right_min_val = right_mag_window[0];
  98. int right_max_index = 0;
  99. int right_min_index = 0;
  100. for (int i = 1; i < window_size; i++)
  101. {
  102. if (left_max_val < left_mag_window[i])
  103. {
  104. left_max_val = left_mag_window[i];
  105. left_max_index = i;
  106. }
  107. if (left_min_val > left_mag_window[i])
  108. {
  109. left_min_val = left_mag_window[i];
  110. left_min_index = i;
  111. }
  112. if (right_max_val < right_mag_window[i])
  113. {
  114. right_max_val = right_mag_window[i];
  115. right_max_index = i;
  116. }
  117. if (right_min_val > right_mag_window[i])
  118. {
  119. right_min_val = right_mag_window[i];
  120. right_min_index = i;
  121. }
  122. }
  123. if (left_max_index > left_min_index && right_max_index < right_min_index
  124. && left_max_val - left_min_val > 2000.f && right_max_val - right_min_val > 2000.f)
  125. {
  126. return 1;
  127. }
  128. if (left_max_index < left_min_index && right_max_index > right_min_index
  129. && left_max_val - left_min_val > 2000.f && right_max_val - right_min_val > 2000.f)
  130. {
  131. return 1;
  132. }
  133. return 0;
  134. }
  135. int avoid_down_during_change_road_by_acc(float* left_acc_window, float* right_acc_window, int window_size)
  136. {
  137. float left_max_val = left_acc_window[0];
  138. float left_min_val = left_acc_window[0];
  139. float right_max_val = right_acc_window[0];
  140. float right_min_val = right_acc_window[0];
  141. for (int i = 1; i < window_size; i++)
  142. {
  143. if (left_max_val < left_acc_window[i])
  144. {
  145. left_max_val = left_acc_window[i];
  146. }
  147. if (left_min_val > left_acc_window[i])
  148. {
  149. left_min_val = left_acc_window[i];
  150. }
  151. if (right_max_val < right_acc_window[i])
  152. {
  153. right_max_val = right_acc_window[i];
  154. }
  155. if (right_min_val > right_acc_window[i])
  156. {
  157. right_min_val = right_acc_window[i];
  158. }
  159. }
  160. if (left_max_val > left_min_val + 0.3f || right_max_val > right_min_val + 0.3f)
  161. {
  162. return 1;
  163. }
  164. return 0;
  165. }
  166. #define BIG_WINDOW_SIZE 20
  167. int press_down_detect_new(int index, float front_mag_left, float back_mag_left,
  168. float front_mag_right, float back_mag_right, int left_zupt, int right_zupt,
  169. float left_acc_x, float left_acc_y, float left_acc_z, float right_acc_x, float right_acc_y, float right_acc_z,
  170. int* front_down, int* back_down)
  171. {
  172. static float front_mag_left_big_buff[BIG_WINDOW_SIZE];
  173. static float front_mag_right_big_buff[BIG_WINDOW_SIZE];
  174. static float back_mag_left_big_buff[BIG_WINDOW_SIZE];
  175. static float back_mag_right_big_buff[BIG_WINDOW_SIZE];
  176. static float left_acc_z_big_buff[BIG_WINDOW_SIZE];
  177. static float right_acc_z_big_buff[BIG_WINDOW_SIZE];
  178. static float left_front_up_min = 40000.0f;
  179. static float left_back_up_min = 40000.0f;
  180. static float right_front_up_min = 40000.0f;
  181. static float right_back_up_min = 40000.0f;
  182. static int left_front_up_count = 0;
  183. static int right_front_up_count = 0;
  184. static int left_back_up_count = 0;
  185. static int right_back_up_count = 0;
  186. memcpy(front_mag_left_big_buff, front_mag_left_big_buff + 1, (BIG_WINDOW_SIZE - 1) * sizeof(float));
  187. memcpy(front_mag_right_big_buff, front_mag_right_big_buff + 1, (BIG_WINDOW_SIZE - 1) * sizeof(float));
  188. memcpy(back_mag_left_big_buff, back_mag_left_big_buff + 1, (BIG_WINDOW_SIZE - 1) * sizeof(float));
  189. memcpy(back_mag_right_big_buff, back_mag_right_big_buff + 1, (BIG_WINDOW_SIZE - 1) * sizeof(float));
  190. memcpy(left_acc_z_big_buff, left_acc_z_big_buff + 1, (BIG_WINDOW_SIZE - 1) * sizeof(float));
  191. memcpy(right_acc_z_big_buff, right_acc_z_big_buff + 1, (BIG_WINDOW_SIZE - 1) * sizeof(float));
  192. front_mag_left_big_buff[(BIG_WINDOW_SIZE - 1)] = front_mag_left;
  193. front_mag_right_big_buff[(BIG_WINDOW_SIZE - 1)] = front_mag_right;
  194. back_mag_left_big_buff[(BIG_WINDOW_SIZE - 1)] = back_mag_left;
  195. back_mag_right_big_buff[(BIG_WINDOW_SIZE - 1)] = back_mag_right;
  196. left_acc_z_big_buff[(BIG_WINDOW_SIZE - 1)] = left_acc_z;
  197. right_acc_z_big_buff[(BIG_WINDOW_SIZE - 1)] = right_acc_z;
  198. dual_foot_detect_up_trend(&left_front_up_min, &left_front_up_count, front_mag_left_big_buff,
  199. &right_front_up_min, &right_front_up_count,front_mag_right_big_buff, BIG_WINDOW_SIZE);
  200. dual_foot_detect_up_trend(&left_back_up_min,&left_back_up_count, back_mag_left_big_buff,
  201. &right_back_up_min, &right_back_up_count, back_mag_right_big_buff, BIG_WINDOW_SIZE);
  202. // if (front_mag_left - left_front_up_min > 2000.f && front_mag_right - right_front_up_min > 2000.f)
  203. if (front_mag_left - left_front_up_min > 1000.f && front_mag_right - right_front_up_min > 1000.f)
  204. {
  205. *front_down = 1;
  206. //printf( "left_front_up_count: %d , right_front_up_count : %d \n" , left_front_up_count , right_front_up_count);
  207. }
  208. if (back_mag_left - left_back_up_min > 2000.0f && back_mag_right - right_back_up_min > 2000.0f)
  209. {
  210. *back_down = 1;
  211. //printf("left_front_up_count: %d , right_front_up_count : %d \n", left_front_up_count , right_front_up_count);
  212. }
  213. if (avoid_down_during_change_road_by_acc(left_acc_z_big_buff, right_acc_z_big_buff, BIG_WINDOW_SIZE))
  214. {
  215. if (*front_down == 1)
  216. {
  217. //printf("delete front_down when change road \n");
  218. }
  219. if (*back_down == 1)
  220. {
  221. //printf("delete back_down when change road \n");
  222. }
  223. *front_down = 0;
  224. }
  225. return 0;
  226. }
  227. short pos_jump_detect(int* h_pos, int* s_pos, int left_zupt, int right_zupt)
  228. {
  229. static int last_h_z;
  230. static int last_s_z;
  231. static int left_up;
  232. static int right_up;
  233. static int left_up_count;
  234. static int right_up_count;
  235. static int left_zupt_count;
  236. static int right_zupt_count;
  237. if (left_zupt)
  238. {
  239. left_zupt_count = 10;
  240. }
  241. else
  242. {
  243. left_zupt_count--;
  244. }
  245. if (right_zupt)
  246. {
  247. right_zupt_count = 10;
  248. }
  249. else
  250. {
  251. right_zupt_count--;
  252. }
  253. if (h_pos[2] - last_h_z > 0 && h_pos[2] > 0)
  254. {
  255. left_up = 1;
  256. }
  257. else if ((h_pos[2] - last_h_z < 0) || h_pos[2] <= 0)
  258. {
  259. left_up = -1;
  260. }
  261. if (left_up == 1)
  262. {
  263. left_up_count++;
  264. }
  265. else
  266. {
  267. left_up_count = 0;
  268. }
  269. if (s_pos[2] - last_s_z > 0 && s_pos[2] > 0)
  270. {
  271. right_up = 1;
  272. }
  273. else if ((s_pos[2] - last_s_z < 0) || s_pos[2] <= 0)
  274. {
  275. right_up = -1;
  276. }
  277. if (right_up == 1)
  278. {
  279. right_up_count++;
  280. }
  281. else
  282. {
  283. right_up_count = 0;
  284. }
  285. last_h_z = h_pos[2];
  286. last_s_z = s_pos[2];
  287. if (left_up == 1 && right_up == 1 && right_up_count < 15 && left_up_count < 15
  288. && right_zupt_count > 0 && left_zupt_count > 0 &&
  289. ((right_up_count > 2 && left_up_count > 2) ||
  290. (right_up_count > 0 && left_up_count > 0 && h_pos[2] > 1 && s_pos[2] > 1)))
  291. {
  292. return 1;
  293. }
  294. return 0;
  295. }
  296. //由于仅靠IMU来探测触底了,高度估计得不太对,所以用加速度来算起跳动作
  297. void max_min_window(float* data, int length, float* max_val, int* max_index, float* min_val, int* min_index)
  298. {
  299. *max_val = data[0];
  300. *max_index = 0;
  301. *min_val = data[0];
  302. *min_index = 0;
  303. for (int i = 0; i < length; i++)
  304. {
  305. if (*max_val < data[i])
  306. {
  307. *max_val = data[i];
  308. *max_index = i;
  309. }
  310. if (*min_val > data[i])
  311. {
  312. *min_val = data[i];
  313. *min_index = i;
  314. }
  315. }
  316. }
  317. int avoid_error_jump_by_press(float* mag_window, int window_size)
  318. {
  319. float min_val = mag_window[0];
  320. float max_val = mag_window[0];
  321. int max_index = 0;
  322. for (int i = 0; i < window_size; i++)
  323. {
  324. if (min_val > mag_window[0])
  325. {
  326. min_val = mag_window[0];
  327. }
  328. if (max_val < mag_window[0])
  329. {
  330. max_val = mag_window[0];
  331. max_index = i;
  332. }
  333. }
  334. if (max_val - min_val > 2000.f && max_index == window_size - 1)
  335. {
  336. return 1;
  337. }
  338. return 0;
  339. }
  340. short press_jump_detect(float left_acc_z, float right_acc_z, int left_zupt, int right_zupt, int left_front_press, int right_front_press)
  341. {
  342. static float right_data_z_buff[10];
  343. static float left_data_z_buff[10];
  344. static float left_front_mag_buff[10];
  345. static float right_front_mag_buff[10];
  346. static int last_right_press;
  347. static int last_left_press;
  348. static int left_max_press;
  349. static int right_max_press;
  350. static int press_wait;
  351. static int acc_wait;
  352. static int wait;
  353. /*
  354. * 存储数据
  355. */
  356. memcpy(right_data_z_buff, right_data_z_buff + 1, 9 * sizeof(float));
  357. memcpy(left_data_z_buff, left_data_z_buff + 1, 9 * sizeof(float));
  358. right_data_z_buff[9] = right_acc_z;
  359. left_data_z_buff[9] = left_acc_z;
  360. memcpy(left_front_mag_buff, left_front_mag_buff + 1, 9 * sizeof(float));
  361. memcpy(right_front_mag_buff, right_front_mag_buff + 1, 9 * sizeof(float));
  362. left_front_mag_buff[9] = left_front_press;
  363. right_front_mag_buff[9] = right_front_press;
  364. float max_val_right_mag, min_val_right_mag, max_val_left_mag, min_val_left_mag;
  365. int max_index_right_mag, min_index_right_mag, max_index_left_mag, min_index_left_mag;
  366. max_min_window(right_front_mag_buff, 10, &max_val_right_mag, &max_index_right_mag, &min_val_right_mag, &min_index_right_mag);
  367. max_min_window(left_front_mag_buff, 10, &max_val_left_mag, &max_index_left_mag, &min_val_left_mag, &min_index_left_mag);
  368. if (max_val_right_mag > right_front_press + 3000.f && max_val_left_mag > left_front_press + 3000.f)
  369. {
  370. int start_index = max_index_left_mag > max_index_right_mag ? max_index_left_mag : max_index_right_mag;
  371. int down_flag = 1;
  372. // 遍历双脚持续下降的所用
  373. for (int i = start_index + 1; i < 10; i++)
  374. {
  375. if (right_front_mag_buff[i - 1] - right_front_mag_buff[i] < 100.f)
  376. {
  377. down_flag = 0;
  378. break;
  379. }
  380. if (left_front_mag_buff[i - 1] - left_front_mag_buff[i] < 100.f)
  381. {
  382. down_flag = 0;
  383. break;
  384. }
  385. }
  386. if (down_flag == 1)
  387. {
  388. press_wait = 10;
  389. }
  390. }
  391. /*
  392. * 加速变化估计
  393. */
  394. float max_val_right, min_val_right, max_val_left, min_val_left;
  395. int max_index_right, min_index_right, max_index_left, min_index_left;
  396. max_min_window(right_data_z_buff, 10, &max_val_right, &max_index_right, &min_val_right, &min_index_right);
  397. max_min_window(left_data_z_buff, 10, &max_val_left, &max_index_left, &min_val_left, &min_index_left);
  398. /*
  399. * 1.0f的阈值轻轻跳就可以达到了,现在改为1.3f的阈值
  400. */
  401. int jump = 0;
  402. if (max_index_right < min_index_right && max_index_left < min_index_left
  403. && max_val_right - min_val_right > 0.8f && max_val_left - min_val_left > 0.8f
  404. && (left_zupt == 0 && right_zupt == 0))
  405. {
  406. acc_wait = 10;
  407. }
  408. if ( acc_wait > 0 && press_wait > 0)
  409. {
  410. jump = 1;
  411. }
  412. if (acc_wait > 0)
  413. {
  414. acc_wait--;
  415. }
  416. if (press_wait > 0)
  417. {
  418. press_wait--;
  419. }
  420. //if (avoid_error_jump_by_press(left_front_mag_buff, 10))
  421. //{
  422. // jump = 0;
  423. //}
  424. //if (avoid_error_jump_by_press(right_front_mag_buff, 10))
  425. //{
  426. // jump = 0;
  427. //}
  428. return jump;
  429. }
  430. int att_jump_detect(float left_pitch, float right_pitch, int left_zupt, int right_zupt)
  431. {
  432. static int num;
  433. static float left_pitch_window[8];
  434. static float right_pitch_window[8];
  435. //用俯仰角判断好了, 现在的IMU方向 下标1为俯仰角
  436. memcpy(left_pitch_window, left_pitch_window + 1, 7 * sizeof(float));
  437. memcpy(right_pitch_window, right_pitch_window + 1, 7 * sizeof(float));
  438. left_pitch_window[7] = left_pitch;
  439. right_pitch_window[7] = right_pitch;
  440. num++;
  441. if (num < 8)
  442. {
  443. return 0;
  444. }
  445. else
  446. {
  447. num = 8;
  448. }
  449. //要求起跳要同步
  450. static int zupt_count = 0;
  451. if (left_zupt && right_zupt)
  452. {
  453. zupt_count = 21;
  454. }
  455. if (zupt_count > 0)
  456. {
  457. zupt_count--;
  458. }
  459. for (int i = 7; i > 1; i--)
  460. {
  461. if ((left_pitch_window[i] > left_pitch_window[i - 1] || left_pitch_window[i] > left_pitch_window[i - 2])
  462. && (right_pitch_window[i] > right_pitch_window[i - 1] || right_pitch_window[i] > right_pitch_window[i - 2]))
  463. {
  464. if ((left_pitch > left_pitch_window[i - 1] + 0.1f || left_pitch > left_pitch_window[i - 2] + 0.1f)
  465. && (right_pitch > right_pitch_window[i - 1] + 0.1f || right_pitch > right_pitch_window[i - 2] + 0.1f)
  466. && zupt_count > 0)
  467. {
  468. return 1;
  469. }
  470. }
  471. else
  472. {
  473. return 0;
  474. }
  475. }
  476. return 0;
  477. }