loc_tool.cpp 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306
  1. #include <vector>
  2. #include <log.h>
  3. #include <ant.h>
  4. #include <loc_tool.h>
  5. #include <message.h>
  6. std::vector<point> loc_tool_tdoa_3_base::calc_location(std::vector<loc_message>&locm)
  7. {
  8. return std::vector<point>();
  9. }
  10. int loc_tool_tdoa_3_base::index()
  11. {
  12. return 5;
  13. }
  14. std::vector<point> loc_tool_tdoa_2_base::calc_location(std::vector<loc_message>&locm)
  15. {
  16. return std::vector<point>();
  17. }
  18. int loc_tool_tdoa_2_base::index()
  19. {
  20. return 3;
  21. }
  22. std::vector<point> loc_tool_tdoa_1_base::calc_location(std::vector<loc_message>&locm)
  23. {
  24. return std::vector<point>();
  25. }
  26. int loc_tool_tdoa_1_base::index()
  27. {
  28. return 1;
  29. }
  30. std::vector<point> loc_tool_tof_3_base::calc_location(std::vector<loc_message>&locm)
  31. {
  32. return std::vector<point>();
  33. }
  34. int loc_tool_tof_3_base::index()
  35. {
  36. return 4;
  37. }
  38. std::vector<point> loc_tool_tof_2_base::calc_location(std::vector<loc_message>&locm)
  39. {
  40. return std::vector<point>();
  41. }
  42. int loc_tool_tof_2_base::index()
  43. {
  44. return 2;
  45. }
  46. std::vector<point> loc_tool_tof_1_base::calc_location(std::vector<loc_message>&locm)
  47. {
  48. int32_t last_ct = -1;
  49. std::vector<point> vec;
  50. std::vector<loc_message> lm;
  51. for(auto rit = locm.crbegin();rit != locm.crend();rit++)
  52. {
  53. site s = rit->m_sit;
  54. if(s.is_path_empty() || rit->m_num_ticks == 0)
  55. continue;
  56. if(last_ct == -1)
  57. last_ct = rit->m_card_ct;
  58. else if(last_ct != rit->m_card_ct)
  59. continue;
  60. double dist_tof=rit->m_num_ticks*15.65*2.996*1e-4/s.m_scale;
  61. auto v = s.solving(rit->m_ant_id,dist_tof);
  62. lm.insert(lm.begin(),*rit);
  63. vec.insert(std::end(vec),std::begin(v),std::end(v));
  64. }
  65. locm.swap(lm);
  66. return std::move(vec);
  67. }
  68. int loc_tool_tof_1_base::index()
  69. {
  70. return 0;
  71. }
  72. loc_tool_main::loc_tool_main()
  73. {
  74. set_tool(new loc_tool_tof_1_base());
  75. set_tool(new loc_tool_tof_2_base());
  76. set_tool(new loc_tool_tof_3_base());
  77. set_tool(new loc_tool_tdoa_1_base());
  78. set_tool(new loc_tool_tdoa_2_base());
  79. set_tool(new loc_tool_tdoa_3_base());
  80. }
  81. loc_tool_main::~loc_tool_main()
  82. {
  83. for(auto&tool:g_tool)
  84. delete tool;
  85. }
  86. void loc_tool_main::set_tool(loc_tool*tool)
  87. {
  88. int index=tool->index();
  89. if(g_tool[index])
  90. {
  91. delete g_tool[index];
  92. g_tool[index]=0;
  93. }
  94. g_tool[index]=tool;
  95. }
  96. std::vector<point> loc_tool_main::calc_location(std::vector<loc_message>&locm)
  97. {
  98. if(locm.empty()) return {};
  99. int tool_index=locm[0].tool_index(),i=1,len=locm.size();
  100. for(;i<len;i++)
  101. {
  102. if(tool_index!=locm[i].tool_index())
  103. break;
  104. }
  105. if(i==len)
  106. {
  107. return std::move(g_tool[tool_index]->calc_location(locm));
  108. }
  109. //包含至少两种定位方式的基站,目前只考虑两种
  110. std::vector<loc_message> locm1,locm2;
  111. locm1.assign(locm.begin(),locm.begin()+i);
  112. for(;i<len;i++)
  113. {
  114. if(tool_index!=locm[i].tool_index())
  115. locm2.push_back(locm[i]);
  116. else
  117. locm1.push_back(locm[i]);
  118. }
  119. bool flag = false;
  120. if(locm2[0].tool_index() > tool_index)
  121. flag = true;
  122. std::vector<point> rc;
  123. if(flag && (locm2[0].m_sit.config().best_msg_cnt<=(int)locm2.size()))
  124. {
  125. int index=locm2[0].tool_index();
  126. rc = std::move(g_tool[index]->calc_location(locm2));
  127. locm.swap(locm2);
  128. }
  129. else if(locm1[0].m_sit.config().best_msg_cnt<=(int)locm1.size())
  130. {
  131. rc = std::move(g_tool[tool_index]->calc_location(locm1));
  132. locm.swap(locm1);
  133. }
  134. return std::move(rc);
  135. }
  136. loc_tool* loc_tool_main::g_tool[6]={0,0,0,0,0,0};
  137. /*
  138. std::vector<point> loc_tool_tdoa_3_base::calc_location(const std::vector<loc_message>&locm)
  139. {
  140. return std::vector<point>();
  141. }
  142. int loc_tool_tdoa_3_base::index()
  143. {
  144. return 5;
  145. }
  146. std::vector<point> loc_tool_tdoa_2_base::calc_location(const std::vector<loc_message>&locm)
  147. {
  148. return std::vector<point>();
  149. }
  150. int loc_tool_tdoa_2_base::index()
  151. {
  152. return 3;
  153. }
  154. std::vector<point> loc_tool_tdoa_1_base::calc_location(const std::vector<loc_message>&locm)
  155. {
  156. return std::vector<point>();
  157. }
  158. int loc_tool_tdoa_1_base::index()
  159. {
  160. return 1;
  161. }
  162. std::vector<point> loc_tool_tof_3_base::calc_location(const std::vector<loc_message>&locm)
  163. {
  164. return std::vector<point>();
  165. }
  166. int loc_tool_tof_3_base::index()
  167. {
  168. return 4;
  169. }
  170. std::vector<point> loc_tool_tof_2_base::calc_location(const std::vector<loc_message>&locm)
  171. {
  172. return std::vector<point>();
  173. }
  174. int loc_tool_tof_2_base::index()
  175. {
  176. return 2;
  177. }
  178. void loc_tool_tof_1_base::on_loc_message(ant*,const message_locinfo&m)
  179. {
  180. log_info("tof1-message:site=%d,ant=%d,card=%d,ct=%d,tof=%lld,rav=%02X,acc=%02X,rssi=%d",
  181. m.m_site_id,m.m_ant_id,m.m_card_id,m.m_card_ct,m.m_tof,m.m_rav,m.m_acc,m.m_rssi);
  182. }
  183. std::vector<point> loc_tool_tof_1_base::calc_location(const std::vector<loc_message>&locm)
  184. {
  185. return std::vector<point>();
  186. }
  187. int loc_tool_tof_1_base::index()
  188. {
  189. return 0;
  190. }
  191. loc_tool_main::loc_tool_main()
  192. {
  193. set_tool(new loc_tool_tof_1_base());
  194. set_tool(new loc_tool_tof_2_base());
  195. set_tool(new loc_tool_tof_3_base());
  196. set_tool(new loc_tool_tdoa_1_base());
  197. set_tool(new loc_tool_tdoa_2_base());
  198. set_tool(new loc_tool_tdoa_3_base());
  199. }
  200. loc_tool_main::~loc_tool_main()
  201. {
  202. for(auto&tool:g_tool)
  203. delete tool;
  204. }
  205. loc_tool* loc_tool_main::get_tool(int index)
  206. {
  207. if((uint32_t)index >= sizeof(g_tool)/sizeof(loc_tool*))
  208. return nullptr;
  209. return g_tool[index];
  210. }
  211. void loc_tool_main::set_tool(loc_tool*tool)
  212. {
  213. int index=tool->index();
  214. if(g_tool[index])
  215. {
  216. delete g_tool[index];
  217. g_tool[index]=0;
  218. }
  219. g_tool[index]=tool;
  220. }
  221. void loc_tool_main::on_loc_message(ant*a, const message_locinfo&m)
  222. {
  223. loc_tool*lt=get_tool(a->index());
  224. if(lt==nullptr)
  225. {
  226. log_warn("无法找到对应的loctool-message:site=%d,ant=%d,card=%d,ct=%d,tof=%lld,rav=%02X,acc=%02X,rssi=%d",
  227. m.m_site_id,m.m_ant_id,m.m_card_id,m.m_card_ct,m.m_tof,m.m_rav,m.m_acc,m.m_rssi);
  228. return;
  229. }
  230. lt->on_loc_message(a, m);
  231. }
  232. std::vector<point> loc_tool_main::calc_location(const std::vector<loc_message>&locm)
  233. {
  234. if(locm.empty()) return {};
  235. int tool_index=locm[0].tool_index(),i=1,len=locm.size();
  236. for(;i<len;i++)
  237. {
  238. if(tool_index!=locm[i].tool_index())
  239. break;
  240. }
  241. if(i==len)
  242. {
  243. return std::move(g_tool[tool_index]->calc_location(locm));
  244. }
  245. //包含至少两种定位方式的基站,目前只考虑两种
  246. std::vector<loc_message> locm1,locm2;
  247. locm1.assign(locm.begin(),locm.begin()+i);
  248. for(;i<len;i++)
  249. {
  250. if(tool_index!=locm[i].tool_index())
  251. locm2.push_back(locm[i]);
  252. else
  253. locm1.push_back(locm[i]);
  254. }
  255. std::vector<point> rc;
  256. if(locm1[0].m_ant->config().best_msg_cnt<=(int)locm1.size())
  257. {
  258. rc=std::move(g_tool[tool_index]->calc_location(locm1));
  259. }
  260. if(locm2[0].m_ant->config().best_msg_cnt<=(int)locm2.size())
  261. {
  262. int index=locm2[0].tool_index();
  263. auto v=std::move(g_tool[index]->calc_location(locm2));
  264. rc.insert(rc.begin(),v.begin(),v.end());
  265. }
  266. return std::move(rc);
  267. }
  268. */