locate_manager.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746
  1. //
  2. // Created by zx on 2019/12/30.
  3. //
  4. #include "locate_manager.h"
  5. #include "../tool/proto_tool.h"
  6. #include "locate_parameter.pb.h"
  7. #include <pcl/filters//voxel_grid.h>
  8. #include <pcl/filters/passthrough.h>
  9. #include <glog/logging.h>
  10. #include <string.h>
  11. Locate_manager::Locate_manager()
  12. {
  13. mp_point_sift = NULL;
  14. mp_cnn3d = NULL;
  15. m_locate_manager_status = LOCATE_MANAGER_UNKNOW;
  16. m_locate_manager_working_flag = false;
  17. mp_locate_manager_thread = NULL;
  18. mp_locate_manager_task = NULL;
  19. mp_locate_information = NULL;
  20. m_save_flag = false;
  21. // m_save_path = "";
  22. }
  23. Locate_manager::~Locate_manager()
  24. {
  25. Locate_manager_uninit();
  26. }
  27. //初始化 定位 管理模块。如下三选一LOCATE_PARAMETER_PATH
  28. Error_manager Locate_manager::Locate_manager_init()
  29. {
  30. return Locate_manager_init_from_protobuf(LOCATE_PARAMETER_PATH);
  31. }
  32. //初始化 定位 管理模块。从文件读取
  33. Error_manager Locate_manager::Locate_manager_init_from_protobuf(std::string prototxt_path)
  34. {
  35. Measure::LocateParameter t_locate_parameters;
  36. if(! proto_tool::read_proto_param(prototxt_path,t_locate_parameters) )
  37. {
  38. return Error_manager(LOCATER_MANAGER_READ_PROTOBUF_ERROR,MINOR_ERROR,"Locate_manager read_proto_param failed");
  39. }
  40. return Locate_manager_init_from_protobuf(t_locate_parameters);
  41. }
  42. //初始化 定位 管理模块。从protobuf读取
  43. Error_manager Locate_manager::Locate_manager_init_from_protobuf(Measure::LocateParameter& locate_parameters)
  44. {
  45. m_locate_manager_working_flag = false;
  46. m_locate_manager_status = LOCATE_MANAGER_READY;
  47. //启动雷达管理模块的内部线程。默认wait。
  48. m_locate_manager_condition.reset(false, false, false);
  49. mp_locate_manager_thread = new std::thread(&Locate_manager::thread_work, this);
  50. return Error_code::SUCCESS;
  51. LOG(INFO) << " ---Locate_manager_init_from_protobuf run--- "<< this;
  52. Error_manager t_error;
  53. m_locate_manager_working_flag = false;
  54. int point_size = locate_parameters.seg_parameter().point_size();
  55. int cls_num = locate_parameters.seg_parameter().cls_num();
  56. double freq = locate_parameters.seg_parameter().freq();
  57. std::string graph = locate_parameters.seg_parameter().graph();
  58. std::string cpkt = locate_parameters.seg_parameter().cpkt();
  59. Measure::Area3d area = locate_parameters.seg_parameter().area();
  60. Cloud_box t_cloud_box;
  61. t_cloud_box.x_min = area.x_min();
  62. t_cloud_box.x_max = area.x_max();
  63. t_cloud_box.y_min = area.y_min();
  64. t_cloud_box.y_max = area.y_max();
  65. t_cloud_box.z_min = area.z_min();
  66. t_cloud_box.z_max = area.z_max();
  67. mp_point_sift = new Point_sift_segmentation(point_size,cls_num,freq,t_cloud_box);
  68. std::string graph_file = graph;
  69. std::string cpkt_file = cpkt;
  70. std::cout << "graph_file" << graph_file << std::endl;
  71. std::cout << "cpkt_file" << cpkt_file << std::endl;
  72. t_error=mp_point_sift->init(graph_file,cpkt_file);
  73. if(t_error!=SUCCESS)
  74. {
  75. LOG(ERROR)<<t_error.to_string();
  76. return t_error;
  77. }
  78. LOG(INFO)<<" pointSift init SUCCESS, graph = "<< graph;
  79. int t_cnn3d_length = locate_parameters.net_3dcnn_parameter().length();
  80. int t_cnn3d_width = locate_parameters.net_3dcnn_parameter().width();
  81. int t_cnn3d_height = locate_parameters.net_3dcnn_parameter().height();
  82. int t_cnn3d_nclass = locate_parameters.net_3dcnn_parameter().nclass();
  83. int t_cnn3d_freq = locate_parameters.net_3dcnn_parameter().freq();
  84. mp_cnn3d = new Cnn3d_segmentation(t_cnn3d_length, t_cnn3d_width, t_cnn3d_height, t_cnn3d_freq, t_cnn3d_nclass);
  85. std::string weights = locate_parameters.net_3dcnn_parameter().weights_file();
  86. t_error=mp_cnn3d->init(weights);
  87. if(t_error!=SUCCESS)
  88. {
  89. LOG(ERROR)<<t_error.to_string();
  90. return t_error;
  91. }
  92. LOG(INFO)<<" 3dcnn Init SUCCESS ";
  93. m_locate_manager_status = LOCATE_MANAGER_READY;
  94. //启动雷达管理模块的内部线程。默认wait。
  95. m_locate_manager_condition.reset(false, false, false);
  96. mp_locate_manager_thread = new std::thread(&Locate_manager::thread_work, this);
  97. return SUCCESS;
  98. }
  99. //反初始化 定位 管理模块。
  100. Error_manager Locate_manager::Locate_manager_uninit()
  101. {
  102. LOG(INFO) << " ---Locate_manager_uninit run--- "<< this;
  103. //关闭线程
  104. if (mp_locate_manager_thread)
  105. {
  106. m_locate_manager_condition.kill_all();
  107. }
  108. //回收线程的资源
  109. if (mp_locate_manager_thread)
  110. {
  111. mp_locate_manager_thread->join();
  112. delete mp_locate_manager_thread;
  113. mp_locate_manager_thread = NULL;
  114. }
  115. //回收内存
  116. if ( mp_point_sift !=NULL )
  117. {
  118. delete mp_point_sift;
  119. mp_point_sift = NULL;
  120. }
  121. if ( mp_cnn3d !=NULL )
  122. {
  123. delete mp_cnn3d;
  124. mp_cnn3d = NULL;
  125. }
  126. m_locate_manager_status = LOCATE_MANAGER_UNKNOW;
  127. m_locate_manager_working_flag = false;
  128. return Error_code::SUCCESS;
  129. }
  130. //对外的接口函数,负责接受并处理任务单,
  131. //input:p_locate_task 定位任务单,基类的指针,指向子类的实例,(多态)
  132. Error_manager Locate_manager::execute_task(Task_Base* p_locate_task)
  133. {
  134. LOG(INFO) << " ---Locate_manager::execute_task run--- "<< this;
  135. Error_manager t_error;
  136. Error_manager t_result;
  137. //检查指针
  138. if (p_locate_task == NULL) {
  139. return Error_manager(Error_code::POINTER_IS_NULL, Error_level::MINOR_ERROR,
  140. "Laser_manager::execute_task failed, POINTER_IS_NULL");
  141. }
  142. //检查任务类型,
  143. if (p_locate_task->get_task_type() != LOCATE_MANGER_TASK)
  144. {
  145. return Error_manager(Error_code::LOCATER_MANAGER_TASK_TYPE_ERROR, Error_level::MINOR_ERROR,
  146. "Locate_manager::execute_task get_task_type() != LOCATE_MANGER_TASK ");
  147. }
  148. //检查接收方的状态
  149. std::cout << "m_locate_manager_status"<<m_locate_manager_status << std::endl;
  150. t_error = check_status();
  151. if ( t_error != SUCCESS )
  152. {
  153. t_result.compare_and_cover_error(t_error);
  154. }
  155. else
  156. {
  157. //接受任务,并将任务的状态改为TASK_SIGNED已签收
  158. mp_locate_manager_task = (Locate_manager_task *) p_locate_task;
  159. mp_locate_manager_task->set_task_statu(TASK_SIGNED);
  160. //检查消息内容是否正确,
  161. //检查三维点云指针
  162. if (mp_locate_manager_task->get_task_point_cloud_map() == NULL)
  163. {
  164. t_error.error_manager_reset(Error_code::POINTER_IS_NULL,
  165. Error_level::MINOR_ERROR,
  166. "execute_task mp_task_point_cloud is null");
  167. t_result.compare_and_cover_error(t_error);
  168. }
  169. else if ( mp_locate_manager_task->get_task_cloud_lock() == NULL )
  170. {
  171. t_error.error_manager_reset(Error_code::POINTER_IS_NULL,
  172. Error_level::MINOR_ERROR,
  173. "execute_task mp_task_cloud_lock is null");
  174. t_result.compare_and_cover_error(t_error);
  175. }
  176. else
  177. {
  178. // std::cout << "--------------------------------------" << std::endl;
  179. //
  180. // std::cout << mp_locate_manager_task->get_task_point_cloud_map()->size() << std::endl;
  181. // std::cout << mp_locate_manager_task->get_cloud_aggregation_flag() << std::endl;
  182. // std::cout << "--------------------------------------" << std::endl;
  183. //校验map的大小
  184. if ( (mp_locate_manager_task->get_task_point_cloud_map()->size() ==1 && mp_locate_manager_task->get_cloud_aggregation_flag() == true)
  185. || ( (mp_locate_manager_task->get_task_point_cloud_map()->size() == CNN3D_WHEEL_NUMBER
  186. || mp_locate_manager_task->get_task_point_cloud_map()->size() == CNN3D_WHEEL_NUMBER-1 )
  187. && mp_locate_manager_task->get_cloud_aggregation_flag() == false) )
  188. {
  189. //解析任务单,将任务单的数据保存在本地.
  190. mp_locate_information = mp_locate_manager_task->get_task_locate_information_ex();
  191. m_save_flag = mp_locate_manager_task->get_task_save_flag();
  192. m_save_path = mp_locate_manager_task->get_task_save_path();
  193. //启动定位管理模块,的核心工作线程
  194. m_locate_manager_status = LOCATE_MANAGER_SIFT;
  195. m_locate_manager_working_flag = true;
  196. m_locate_manager_condition.notify_all(true);
  197. //通知 thread_work 子线程启动。
  198. //将任务的状态改为 TASK_WORKING 处理中
  199. mp_locate_manager_task->set_task_statu(TASK_WORKING);
  200. }
  201. else
  202. {
  203. t_error.error_manager_reset(Error_code::LOCATER_MANAGER_CLOUD_MAP_ERROR,
  204. Error_level::MINOR_ERROR,
  205. "execute_task input map error");
  206. t_result.compare_and_cover_error(t_error);
  207. }
  208. }
  209. //return 之前要填充任务单里面的错误码.
  210. if (t_result != Error_code::SUCCESS)
  211. {
  212. //忽略轻微故障
  213. if ( t_result.get_error_level() >= Error_level::MINOR_ERROR)
  214. {
  215. //将任务的状态改为 TASK_ERROR 结束错误
  216. mp_locate_manager_task->set_task_statu(TASK_ERROR);
  217. }
  218. //返回错误码
  219. mp_locate_manager_task->set_task_error_manager(t_result);
  220. }
  221. }
  222. return t_result;
  223. }
  224. //检查状态,是否正常运行
  225. Error_manager Locate_manager::check_status()
  226. {
  227. if ( m_locate_manager_status == LOCATE_MANAGER_READY )
  228. {
  229. return Error_code::SUCCESS;
  230. }
  231. else if ( m_locate_manager_status == LOCATE_MANAGER_SIFT
  232. || m_locate_manager_status == LOCATE_MANAGER_CAR
  233. || m_locate_manager_status == LOCATE_MANAGER_WHEEL)
  234. {
  235. return Error_manager(Error_code::LOCATER_MANAGER_STATUS_BUSY, Error_level::NEGLIGIBLE_ERROR,
  236. " Locate_manager::check_status error ");
  237. }
  238. else
  239. {
  240. return Error_manager(Error_code::LOCATER_MANAGER_STATUS_ERROR, Error_level::MINOR_ERROR,
  241. " Locate_manager::check_status error ");
  242. }
  243. return Error_code::SUCCESS;
  244. }
  245. //结束任务单,里面会根据任务的故障等级修正雷达管理模块的状态和任务单的状态
  246. Error_manager Locate_manager::end_task()
  247. {
  248. LOG(INFO) << " ---Locate_manager::end_task run---"<< this;
  249. //关闭子线程
  250. m_locate_manager_working_flag=false;
  251. m_locate_manager_condition.notify_all(false);
  252. //释放缓存
  253. mp_locate_information = NULL;
  254. m_cloud_wheel_map.clear();
  255. m_cloud_car_map.clear();
  256. if ( m_locate_manager_status == LOCATE_MANAGER_SIFT
  257. || m_locate_manager_status == LOCATE_MANAGER_CAR
  258. || m_locate_manager_status == LOCATE_MANAGER_WHEEL)
  259. {
  260. m_locate_manager_status = LOCATE_MANAGER_READY;
  261. }
  262. //else 状态不变
  263. //在结束任务单时,将雷达任务状态改为 TASK_OVER 已结束
  264. if(mp_locate_manager_task !=NULL)
  265. {
  266. //判断任务单的错误等级,
  267. if ( mp_locate_manager_task->get_task_error_manager().get_error_level() < Error_level::MINOR_ERROR)
  268. {
  269. //强制改为TASK_OVER,不管它当前在做什么。
  270. mp_locate_manager_task->set_task_statu(TASK_OVER);
  271. }
  272. else
  273. {
  274. //强制改为 TASK_ERROR,不管它当前在做什么。
  275. mp_locate_manager_task->set_task_statu(TASK_ERROR);
  276. }
  277. }
  278. return Error_code::SUCCESS;
  279. }
  280. //取消任务单,由发送方提前取消任务单
  281. Error_manager Locate_manager::cancel_task()
  282. {
  283. //关闭子线程
  284. m_locate_manager_working_flag=false;
  285. m_locate_manager_condition.notify_all(false);
  286. //确保内部线程已经停下
  287. while (m_locate_manager_condition.is_working())
  288. {
  289. }
  290. //释放缓存
  291. mp_locate_information = NULL;
  292. m_cloud_wheel_map.clear();
  293. m_cloud_car_map.clear();
  294. //强制改为 TASK_DEAD,不管它当前在做什么。
  295. mp_locate_manager_task->set_task_statu(TASK_DEAD);
  296. if ( m_locate_manager_status == LOCATE_MANAGER_SIFT
  297. || m_locate_manager_status == LOCATE_MANAGER_CAR
  298. || m_locate_manager_status == LOCATE_MANAGER_WHEEL)
  299. {
  300. m_locate_manager_status = LOCATE_MANAGER_READY;
  301. }
  302. //else 状态不变
  303. return Error_code::SUCCESS;
  304. }
  305. //判断是否为待机,如果已经准备好,则可以执行任务。
  306. bool Locate_manager::is_ready()
  307. {
  308. return m_locate_manager_status == LOCATE_MANAGER_READY;;
  309. }
  310. Locate_manager::Locate_manager_status Locate_manager::get_locate_manager_status()
  311. {
  312. return m_locate_manager_status;
  313. }
  314. //mp_locate_manager_thread 线程执行函数,
  315. //thread_work 内部线程负责locate定位分析整车的信息,并且回收汇总雷达的数据
  316. void Locate_manager::thread_work()
  317. {
  318. LOG(INFO) << " -------------------------mp_locate_manager_thread start "<< this;
  319. Error_manager t_error;
  320. Error_manager t_result;
  321. //定位管理的独立线程,
  322. while (m_locate_manager_condition.is_alive())
  323. {
  324. m_locate_manager_condition.wait();
  325. if ( m_locate_manager_condition.is_alive() )
  326. {
  327. std::this_thread::yield();
  328. //重新循环必须清除错误码.
  329. t_error.error_manager_clear_all();
  330. t_result.error_manager_clear_all();
  331. if ( mp_locate_manager_task == NULL )
  332. {
  333. m_locate_manager_status = LOCATE_MANAGER_FAULT;
  334. m_locate_manager_working_flag = false;
  335. m_locate_manager_condition.notify_all(false);
  336. }
  337. else if (mp_locate_manager_task->get_task_point_cloud_map() == NULL)
  338. {
  339. t_error.error_manager_reset(Error_code::POINTER_IS_NULL,
  340. Error_level::MINOR_ERROR,
  341. "thread_work mp_task_point_cloud is null");
  342. t_result.compare_and_cover_error(t_error);
  343. //因为故障,而提前结束任务.
  344. mp_locate_manager_task->compare_and_cover_task_error_manager(t_result);
  345. end_task();
  346. }
  347. else if ( mp_locate_manager_task->get_task_cloud_lock() == NULL )
  348. {
  349. t_error.error_manager_reset(Error_code::POINTER_IS_NULL,
  350. Error_level::MINOR_ERROR,
  351. "thread_work mp_task_cloud_lock is null");
  352. t_result.compare_and_cover_error(t_error);
  353. //因为故障,而提前结束任务.
  354. mp_locate_manager_task->compare_and_cover_task_error_manager(t_result);
  355. end_task();
  356. }
  357. //第一步
  358. //point_sift 点筛选分割 , 将输入点云分割为车身和轮胎,
  359. //注:地面会在box切割的时候,使用z轴 z_min 直接切除
  360. else if(m_locate_manager_status == LOCATE_MANAGER_SIFT)
  361. {
  362. m_cloud_wheel_map.clear();
  363. m_cloud_car_map.clear();
  364. //定位筛选,将输入点云map拆分为车轮和车身的map
  365. t_error = locate_manager_sift();
  366. t_result.compare_and_cover_error(t_error);
  367. //故障汇总, 只处理2级以上的故障
  368. if(t_result.get_error_level() >= MINOR_ERROR)
  369. {
  370. m_locate_manager_status = LOCATE_MANAGER_FAULT;
  371. mp_locate_manager_task->set_task_statu(TASK_ERROR);
  372. //因为故障,而提前结束任务.
  373. mp_locate_manager_task->compare_and_cover_task_error_manager(t_result);
  374. end_task();
  375. }
  376. else
  377. {
  378. //成功则 进入到下一个阶段,
  379. m_locate_manager_status = LOCATE_MANAGER_CAR;
  380. }
  381. }
  382. //第二步
  383. //_measure_height 计算汽车的长宽高
  384. else if(m_locate_manager_status == LOCATE_MANAGER_CAR)
  385. {
  386. //计算汽车长宽高,
  387. t_error = locate_manager_locate_car();
  388. t_result.compare_and_cover_error(t_error);
  389. //故障汇总, 只处理2级以上的故障
  390. if(t_result.get_error_level() >= MINOR_ERROR)
  391. {
  392. m_locate_manager_status = LOCATE_MANAGER_FAULT;
  393. mp_locate_manager_task->set_task_statu(TASK_ERROR);
  394. //因为故障,而提前结束任务.
  395. mp_locate_manager_task->compare_and_cover_task_error_manager(t_result);
  396. end_task();
  397. }
  398. else
  399. {
  400. //成功则 进入到下一个阶段,
  401. m_locate_manager_status = LOCATE_MANAGER_WHEEL;
  402. }
  403. }
  404. //第三步
  405. //locate_wheel 解析车轮信息
  406. else if(m_locate_manager_status == LOCATE_MANAGER_WHEEL)
  407. {
  408. //解析车轮信息,
  409. t_error = locate_manager_locate_wheel();
  410. t_result.compare_and_cover_error(t_error);
  411. //故障汇总, 只处理2级以上的故障
  412. if(t_result.get_error_level() >= MINOR_ERROR)
  413. {
  414. m_locate_manager_status = LOCATE_MANAGER_FAULT;
  415. mp_locate_manager_task->set_task_statu(TASK_ERROR);
  416. //因为故障,而提前结束任务.
  417. mp_locate_manager_task->compare_and_cover_task_error_manager(t_result);
  418. end_task();
  419. }
  420. else
  421. {
  422. m_locate_manager_working_flag = false;
  423. m_locate_manager_condition.set_pass_ever(false);
  424. //正常结束任务
  425. mp_locate_manager_task->compare_and_cover_task_error_manager(t_result);
  426. end_task();
  427. }
  428. }
  429. }
  430. }
  431. LOG(INFO) << " ---------------------------mp_locate_manager_thread end :"<<this;
  432. return;
  433. }
  434. //定位筛选,将输入点云map拆分为车轮和车身的map.(数据可以直接从locate_manager内部获取)
  435. Error_manager Locate_manager::locate_manager_sift()
  436. {
  437. if ( mp_locate_manager_task == NULL )
  438. {
  439. return Error_manager(Error_code::POINTER_IS_NULL, Error_level::MINOR_ERROR,
  440. " Locate_manager::locate_manager_sift() error ");
  441. }
  442. else if (mp_locate_manager_task->get_task_point_cloud_map() == NULL)
  443. {
  444. return Error_manager(Error_code::POINTER_IS_NULL,
  445. Error_level::MINOR_ERROR,
  446. "locate_manager_sift mp_task_point_cloud is null");
  447. }
  448. else if ( mp_locate_manager_task->get_task_cloud_lock() == NULL )
  449. {
  450. return Error_manager(Error_code::POINTER_IS_NULL,
  451. Error_level::MINOR_ERROR,
  452. "locate_manager_sift mp_task_cloud_lock is null");
  453. }
  454. Error_manager t_error;
  455. Error_manager t_result;
  456. //获取任务单的输入点云
  457. std::unique_lock<std::mutex> lck (*(mp_locate_manager_task->get_task_cloud_lock()));
  458. std::map<int, pcl::PointCloud<pcl::PointXYZ>::Ptr> *tp_task_point_cloud_map = mp_locate_manager_task->get_task_point_cloud_map();
  459. //清除map
  460. m_cloud_wheel_map.clear();
  461. m_cloud_car_map.clear();
  462. //遍历任务单的输入点云map
  463. for (const auto &map_iter : *tp_task_point_cloud_map)
  464. {
  465. if (map_iter.second.get() == NULL)
  466. {
  467. return Error_manager(Error_code::POINTER_IS_NULL, Error_level::MINOR_ERROR,
  468. " Locate_manager::locate_manager_sift() pcl::PointCloud<pcl::PointXYZ>::Ptr is NULL ");
  469. }
  470. //提前为拆分之后的点云分配内存.
  471. pcl::PointCloud<pcl::PointXYZRGB>::Ptr tp_cloud_wheel(new pcl::PointCloud<pcl::PointXYZRGB>);
  472. pcl::PointCloud<pcl::PointXYZRGB>::Ptr tp_cloud_car(new pcl::PointCloud<pcl::PointXYZRGB>);
  473. //利用PointSift从场景中分割车辆点云
  474. t_error = locate_sift(map_iter.second, tp_cloud_wheel, tp_cloud_car, m_save_flag, m_save_path);
  475. if (t_error != Error_code::SUCCESS)
  476. {
  477. char buf[256] = {0};
  478. sprintf(buf, " map.id = %d, locate_sift error", map_iter.first );
  479. t_error.add_error_description(buf, strlen(buf) );
  480. t_result.compare_and_cover_error(t_error);
  481. //注:这里不直接return,而是要把map全部执行完成
  482. }
  483. else
  484. {
  485. //将结果存入本地的中间缓存,此时map点云是有内存的,后续要记得回收
  486. m_cloud_wheel_map[map_iter.first] = tp_cloud_wheel;
  487. m_cloud_car_map[map_iter.first] = tp_cloud_car;
  488. }
  489. }
  490. return t_result;
  491. }
  492. //定位筛选,对具体的点云进行操作,分离出车轮和车身.
  493. //input::cloud输入点云
  494. //output::cloud_wheel输出车轮点云
  495. //output::cloud_car输出车身点云
  496. //work_dir:中间文件保存路径
  497. Error_manager Locate_manager::locate_sift(pcl::PointCloud<pcl::PointXYZ>::Ptr p_cloud_in,
  498. pcl::PointCloud<pcl::PointXYZRGB>::Ptr& p_cloud_wheel,pcl::PointCloud<pcl::PointXYZRGB>::Ptr& p_cloud_car,
  499. bool save_flag, std::string work_dir)
  500. {
  501. if(p_cloud_in.get()==0)
  502. {
  503. return Error_manager(LOCATER_SIFT_INPUT_CLOUD_UNINIT,MINOR_ERROR,"sift input cloud uninit");
  504. }
  505. if(p_cloud_in->size()==0)
  506. {
  507. return Error_manager(LOCATER_SIFT_INPUT_CLOUD_EMPTY,MINOR_ERROR,"locate_sift input cloud empty");
  508. }
  509. if(mp_point_sift==NULL)
  510. {
  511. return Error_manager(LOCATER_POINTSIFT_UNINIT,MINOR_ERROR,"Point Sift unInit");
  512. }
  513. Error_manager t_error;
  514. //分割车辆点云
  515. std::vector<pcl::PointCloud<pcl::PointXYZRGB>::Ptr> segmentation_clouds;
  516. t_error=mp_point_sift->segmentation(p_cloud_in, segmentation_clouds, save_flag, work_dir);
  517. if(t_error!=SUCCESS)
  518. {
  519. return t_error;
  520. }
  521. //第0类即是轮胎点云,第1类为车身点云
  522. pcl::copyPointCloud(*segmentation_clouds[0], *p_cloud_wheel);
  523. pcl::copyPointCloud(*segmentation_clouds[1], *p_cloud_car);
  524. return SUCCESS;
  525. }
  526. //计算汽车高度,
  527. Error_manager Locate_manager::locate_manager_locate_car()
  528. {
  529. if ( mp_locate_manager_task == NULL )
  530. {
  531. return Error_manager(Error_code::POINTER_IS_NULL, Error_level::MINOR_ERROR,
  532. " Locate_manager::locate_manager_locate_car() error ");
  533. }
  534. else if (mp_locate_manager_task->get_task_point_cloud_map() == NULL)
  535. {
  536. return Error_manager(Error_code::POINTER_IS_NULL,
  537. Error_level::MINOR_ERROR,
  538. "locate_manager_locate_car mp_task_point_cloud is null");
  539. }
  540. else if ( mp_locate_manager_task->get_task_cloud_lock() == NULL )
  541. {
  542. return Error_manager(Error_code::POINTER_IS_NULL,
  543. Error_level::MINOR_ERROR,
  544. "locate_manager_locate_car mp_task_cloud_lock is null");
  545. }
  546. Error_manager t_error;
  547. pcl::PointXYZRGB t_point3d_min, t_point3d_max;
  548. std::unique_lock<std::mutex> lck (*(mp_locate_manager_task->get_task_cloud_lock()));
  549. for (auto iter = m_cloud_car_map.begin(); iter != m_cloud_car_map.end(); ++iter)
  550. {
  551. pcl::PointXYZRGB min, max;
  552. t_error = locate_car(iter->second, min, max);
  553. if ( t_error != Error_code::SUCCESS )
  554. {
  555. return t_error;
  556. }
  557. if ( iter == m_cloud_car_map.begin() )
  558. {
  559. t_point3d_min.x = min.x;
  560. t_point3d_min.y = min.y;
  561. t_point3d_min.z = min.z;
  562. t_point3d_max.x = max.x;
  563. t_point3d_max.y = max.y;
  564. t_point3d_max.z = max.z;
  565. }
  566. else
  567. {
  568. //比较选出整个map点云的边界
  569. if ( t_point3d_min.x <= min.x )
  570. {
  571. t_point3d_min.x = min.x;
  572. }
  573. if ( t_point3d_min.y <= min.y )
  574. {
  575. t_point3d_min.y = min.y;
  576. }
  577. if ( t_point3d_min.z <= min.z )
  578. {
  579. t_point3d_min.z = min.z;
  580. }
  581. if ( t_point3d_max.x >= max.x )
  582. {
  583. t_point3d_max.x = max.x;
  584. }
  585. if ( t_point3d_max.y >= max.y )
  586. {
  587. t_point3d_max.y = max.y;
  588. }
  589. if ( t_point3d_max.z >= max.z )
  590. {
  591. t_point3d_max.z = max.z;
  592. }
  593. }
  594. }
  595. mp_locate_information->locate_length = t_point3d_max.y - t_point3d_min.y;
  596. mp_locate_information->locate_width = t_point3d_max.x - t_point3d_min.x;;
  597. mp_locate_information->locate_height = t_point3d_max.z;
  598. //注意了:在雷达扫描时, 就已经将地面标定位为 z = 0
  599. return Error_code::SUCCESS;
  600. }
  601. //根据汽车点云计算汽车的边界
  602. Error_manager Locate_manager::locate_car(pcl::PointCloud<pcl::PointXYZRGB>::Ptr cloud_car, pcl::PointXYZRGB& min, pcl::PointXYZRGB& max)
  603. {
  604. if(cloud_car.get()==0)
  605. {
  606. return Error_manager(LOCATER_MEASURE_HEIGHT_CLOUD_UNINIT,MINOR_ERROR,"measure height input cloud uninit");
  607. }
  608. if(cloud_car->size()==0)
  609. {
  610. return Error_manager(LOCATER_MEASURE_HEIGHT_CLOUD_EMPTY,MINOR_ERROR,"measure height input cloud is empty");
  611. }
  612. //获取点云的边界
  613. pcl::getMinMax3D(*cloud_car,min,max);
  614. return SUCCESS;
  615. }
  616. //计算汽车的车轮信息
  617. Error_manager Locate_manager::locate_manager_locate_wheel()
  618. {
  619. if ( mp_locate_manager_task == NULL )
  620. {
  621. return Error_manager(Error_code::POINTER_IS_NULL, Error_level::MINOR_ERROR,
  622. " Locate_manager::locate_manager_locate_car() error ");
  623. }
  624. else if (mp_locate_manager_task->get_task_point_cloud_map() == NULL)
  625. {
  626. return Error_manager(Error_code::POINTER_IS_NULL,
  627. Error_level::MINOR_ERROR,
  628. "locate_manager_locate_car mp_task_point_cloud is null");
  629. }
  630. else if ( mp_locate_manager_task->get_task_cloud_lock() == NULL )
  631. {
  632. return Error_manager(Error_code::POINTER_IS_NULL,
  633. Error_level::MINOR_ERROR,
  634. "locate_manager_locate_car mp_task_cloud_lock is null");
  635. }
  636. if(mp_cnn3d==NULL)
  637. {
  638. return Error_manager(LOCATER_3DCNN_UNINIT,MINOR_ERROR,"locate_wheel 3dcnn is not init");
  639. }
  640. Error_manager t_error;
  641. std::unique_lock<std::mutex> lck (*(mp_locate_manager_task->get_task_cloud_lock()));
  642. //解析车轮
  643. t_error=mp_cnn3d->analytic_wheel(m_cloud_wheel_map,mp_locate_manager_task->get_cloud_aggregation_flag(),
  644. mp_locate_manager_task->get_task_locate_information_ex(),
  645. m_save_flag,m_save_path);
  646. if ( t_error != Error_code::SUCCESS )
  647. {
  648. return t_error;
  649. }
  650. return Error_code::SUCCESS;
  651. }
  652. //保存点云成txt到文件
  653. void Locate_manager::save_cloud_txt(pcl::PointCloud<pcl::PointXYZ>::Ptr cloud,std::string save_file)
  654. {
  655. FILE* pfile=fopen(save_file.c_str(),"w");
  656. for(int i=0;i<cloud->size();++i)
  657. {
  658. fprintf(pfile,"%.3f %.3f %.3f\n",cloud->points[i].x,cloud->points[i].y,cloud->points[i].z);
  659. }
  660. fclose(pfile);
  661. }