main.cpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375
  1. //
  2. // Created by zx on 2020/6/18.
  3. //
  4. #include <iostream>
  5. #include "./error_code/error_code.h"
  6. #include "LogFiles.h"
  7. #include <glog/logging.h>
  8. #include "./laser/laser_manager.h"
  9. #include "./laser/Laser.h"
  10. #include "./laser/LivoxLaser.h"
  11. #include "./locate/locate_manager.h"
  12. #include "./communication/communication_socket_base.h"
  13. #include "./tool/thread_pool.h"
  14. #include "./system/system_communication.h"
  15. #include "./system/system_executor.h"
  16. #include "./wanji_lidar/wanji_lidar_device.h"
  17. #include "./tool/proto_tool.h"
  18. #include "./wanji_lidar/wanji_manager.h"
  19. #define LIVOX_NUMBER 2
  20. GOOGLE_GLOG_DLL_DECL void shut_down_logging(const char* data, int size)
  21. {
  22. time_t tt;
  23. time( &tt );
  24. tt = tt + 8*3600; // transform the time zone
  25. tm* t= gmtime( &tt );
  26. char buf[255]={0};
  27. sprintf(buf,"./%d%02d%02d-%02d%02d%02d-dump.txt",
  28. t->tm_year + 1900,
  29. t->tm_mon + 1,
  30. t->tm_mday,
  31. t->tm_hour,
  32. t->tm_min,
  33. t->tm_sec);
  34. FILE* tp_file=fopen(buf,"w");
  35. fprintf(tp_file,data,strlen(data));
  36. fclose(tp_file);
  37. }
  38. void prev_test_pred_task()
  39. {
  40. char ch;
  41. while (1)
  42. {
  43. std::cin >> ch;
  44. if (ch == 'b')
  45. {
  46. break;
  47. }
  48. //第一步, 创建点云缓存, 和 指定雷达,目前就一个
  49. std::map<int, pcl::PointCloud<pcl::PointXYZ>::Ptr> point_cloud_map;
  50. std::mutex cloud_lock;
  51. pcl::PointCloud<pcl::PointXYZ>::Ptr p_cloud(new pcl::PointCloud<pcl::PointXYZ>);
  52. std::ifstream is;
  53. is.open("/home/huli/huli/Terminal_project/hl_measure/Terminal_process_CT/test_data/car_cloud/20200907-084334_src - Cloud.txt", std::ios::out);
  54. if(!is.is_open())
  55. continue;
  56. char t_buf[256];
  57. while (!is.eof())
  58. {
  59. is.getline(t_buf, 256);
  60. pcl::PointXYZ t_point;
  61. sscanf(t_buf, "%f %f %f", &t_point.x, &t_point.y, &t_point.z);
  62. t_point.x = t_point.x / 1000.0;
  63. t_point.y = t_point.y / 1000.0;
  64. t_point.z = t_point.z / 1000.0;
  65. p_cloud->push_back(t_point);
  66. }
  67. std::cout << " huli test :::: "
  68. << " p_cloud->size() = " << p_cloud->size() << std::endl;
  69. point_cloud_map[0] = p_cloud;
  70. Locate_manager_task locate_manager_task;
  71. Common_data::Car_measure_information t_car_information_by_livox;
  72. //第五步, 创建定位模块的任务单, 并发送到 Locate_manager
  73. locate_manager_task.task_init(TASK_CREATED, "", NULL, std::chrono::milliseconds(15000),
  74. true, "/home/huli/huli/Terminal_project/hl_measure/Terminal_process_CT/data",
  75. &cloud_lock, &point_cloud_map, true);
  76. Error_manager t_error = Task_command_manager::get_instance_references().execute_task(&locate_manager_task);
  77. //第六步, 等待任务单完成
  78. if (t_error != Error_code::SUCCESS)
  79. {
  80. LOG(INFO) << " Locate_manager execute_task error ";
  81. }
  82. else
  83. {
  84. //判断任务是否结束, 阻塞 等待, 直到任务完成或者超时
  85. while (locate_manager_task.is_task_end() == false)
  86. {
  87. if (locate_manager_task.is_over_time())
  88. {
  89. //超时处理。取消任务。
  90. Locate_manager::get_instance_pointer()->cancel_task();
  91. locate_manager_task.set_task_statu(TASK_DEAD);
  92. t_error.error_manager_reset(Error_code::LOCATER_MANAGER_TASK_OVER_TIME, Error_level::MINOR_ERROR,
  93. " locate_manager_task is_over_time ");
  94. locate_manager_task.set_task_error_manager(t_error);
  95. }
  96. else
  97. {
  98. //继续等待
  99. std::this_thread::sleep_for(std::chrono::microseconds(1));
  100. std::this_thread::yield();
  101. }
  102. }
  103. //提取任务单 的错误码
  104. t_error = locate_manager_task.get_task_error_manager();
  105. if (t_error == Error_code::SUCCESS)
  106. {
  107. t_car_information_by_livox.center_x = locate_manager_task.get_task_locate_information_ex()->locate_x;
  108. t_car_information_by_livox.center_y = locate_manager_task.get_task_locate_information_ex()->locate_y;
  109. t_car_information_by_livox.car_angle = locate_manager_task.get_task_locate_information_ex()->locate_angle;
  110. t_car_information_by_livox.car_length = locate_manager_task.get_task_locate_information_ex()->locate_length;
  111. t_car_information_by_livox.car_width = locate_manager_task.get_task_locate_information_ex()->locate_width;
  112. t_car_information_by_livox.car_height = locate_manager_task.get_task_locate_information_ex()->locate_height;
  113. t_car_information_by_livox.wheel_base = locate_manager_task.get_task_locate_information_ex()->locate_wheel_base;
  114. t_car_information_by_livox.wheel_width = locate_manager_task.get_task_locate_information_ex()->locate_wheel_width;
  115. t_car_information_by_livox.front_theta = 0;
  116. t_car_information_by_livox.correctness = true;
  117. std::cout << " huli test :::: "
  118. << " t_car_information_by_livox.center_x = " << t_car_information_by_livox.center_x << std::endl;
  119. std::cout << " huli test :::: "
  120. << " t_car_information_by_livox.center_y = " << t_car_information_by_livox.center_y << std::endl;
  121. std::cout << " huli test :::: "
  122. << " t_car_information_by_livox.car_angle = " << t_car_information_by_livox.car_angle << std::endl;
  123. std::cout << " huli test :::: "
  124. << " t_car_information_by_livox.car_length = " << t_car_information_by_livox.car_length << std::endl;
  125. std::cout << " huli test :::: "
  126. << " t_car_information_by_livox.car_width = " << t_car_information_by_livox.car_width << std::endl;
  127. std::cout << " huli test :::: "
  128. << " t_car_information_by_livox.car_height = " << t_car_information_by_livox.car_height << std::endl;
  129. std::cout << " huli test :::: "
  130. << " t_car_information_by_livox.wheel_base = " << t_car_information_by_livox.wheel_base << std::endl;
  131. std::cout << " huli test :::: "
  132. << " t_car_information_by_livox.wheel_width = " << t_car_information_by_livox.wheel_width << std::endl;
  133. std::cout << " huli test :::: "
  134. << " t_car_information_by_livox.front_theta = " << t_car_information_by_livox.front_theta << std::endl;
  135. std::cout << " huli test :::: "
  136. << " t_car_information_by_livox.correctness = " << t_car_information_by_livox.correctness << std::endl;
  137. }
  138. else
  139. {
  140. LOG(INFO) << " locate_manager_task error :::::::" << locate_manager_task.get_task_error_manager().to_string();
  141. }
  142. }
  143. }
  144. }
  145. void prev_test_sample_pred_process()
  146. {
  147. std::cout << "sample pred process" << std::endl;
  148. Error_manager t_error;
  149. Point_sift_segmentation *mp_point_sift;
  150. int point_size = 8192;
  151. int cls_num = 3;
  152. double freq = 20.0;
  153. std::string graph = "../model_param/seg_model_404500.ckpt.meta";
  154. std::string cpkt = "../model_param/seg_model_404500.ckpt";
  155. pcl::PointXYZ minp(-10000, -10000, -10000);
  156. pcl::PointXYZ maxp(10000, 10000, 10000);
  157. while (1)
  158. {
  159. std::cout << "huli 401 connect_laser Error_code::SUCCESS" << std::endl;
  160. std::cout << " press to start" << std::endl;
  161. char ch;
  162. // = getchar();
  163. std::this_thread::sleep_for(std::chrono::seconds(2));
  164. std::cin >> ch;
  165. if (ch == 'b')
  166. {
  167. std::cout << " end scan ------------" << std::endl;
  168. break;
  169. }
  170. std::cout << " start scan ------------" << std::endl;
  171. int n = 1;
  172. while (n)
  173. {
  174. n--;
  175. std::map<int, pcl::PointCloud<pcl::PointXYZ>::Ptr> point_cloud_map;
  176. std::mutex cloud_lock;
  177. std::vector<int> select_laser_id_vector;
  178. select_laser_id_vector.push_back(0);
  179. select_laser_id_vector.push_back(1);
  180. select_laser_id_vector.push_back(2);
  181. select_laser_id_vector.push_back(3);
  182. time_t nowTime;
  183. nowTime = time(NULL);
  184. struct tm *sysTime = localtime(&nowTime);
  185. char t_save_path[256] = {0};
  186. sprintf(t_save_path, "../data/%04d%02d%02d_%02d%02d%02d",
  187. sysTime->tm_year + 1900, sysTime->tm_mon + 1, sysTime->tm_mday, sysTime->tm_hour, sysTime->tm_min, sysTime->tm_sec);
  188. Laser_manager_task *laser_manager_task = new Laser_manager_task;
  189. laser_manager_task->task_init(TASK_CREATED, "", NULL, std::chrono::milliseconds(15000),
  190. true, t_save_path, &cloud_lock, &point_cloud_map, false,
  191. 1000, false, select_laser_id_vector);
  192. t_error = Task_command_manager::get_instance_references().execute_task(laser_manager_task);
  193. // tp_laser_manager->execute_task(laser_manager_task);
  194. if (t_error != Error_code::SUCCESS)
  195. {
  196. std::cout << "huli Laser_manager:::::" << t_error.to_string() << std::endl;
  197. }
  198. else
  199. {
  200. while (laser_manager_task->is_task_end() == false)
  201. {
  202. if (laser_manager_task->is_over_time())
  203. {
  204. //超时处理。取消任务。
  205. Laser_manager::get_instance_pointer()->cancel_task();
  206. laser_manager_task->set_task_statu(TASK_DEAD);
  207. t_error.error_manager_reset(Error_code::LASER_MANAGER_TASK_OVER_TIME, Error_level::MINOR_ERROR,
  208. " laser_manager_task is_over_time ");
  209. laser_manager_task->set_task_error_manager(t_error);
  210. }
  211. else
  212. {
  213. std::this_thread::yield();
  214. }
  215. }
  216. std::cout << "huli task_error::::" << laser_manager_task->get_task_error_manager().to_string() << std::endl;
  217. }
  218. delete (laser_manager_task);
  219. std::cout << "huli laser result---------------------" << t_error.to_string() << std::endl;
  220. /*
  221. pcl::PointCloud<pcl::PointXYZ>::Ptr scan_cloud ;
  222. scan_cloud = point_cloud_map[0];
  223. //locate
  224. pcl::getMinMax3D(*scan_cloud,minp,maxp);
  225. std::cout << "huli 112" << std::endl;
  226. mp_point_sift = new Point_sift_segmentation(point_size,cls_num,freq,minp,maxp);
  227. std::cout << t_error.to_string() << std::endl;
  228. t_error = mp_point_sift->init(graph,cpkt);
  229. std::cout << t_error.to_string() << std::endl;
  230. std::vector<pcl::PointCloud<pcl::PointXYZRGB>::Ptr> segmentation_clouds;
  231. t_error = mp_point_sift->segmentation(scan_cloud, segmentation_clouds, true, "../data/locate_data");
  232. std::cout << t_error.to_string() << std::endl;
  233. */
  234. Locate_manager_task *locate_manager_task = new Locate_manager_task;
  235. locate_manager_task->task_init(TASK_CREATED, "", NULL, std::chrono::milliseconds(15000),
  236. true, t_save_path, &cloud_lock, &point_cloud_map, false);
  237. t_error = Task_command_manager::get_instance_references().execute_task(locate_manager_task);
  238. if (t_error != Error_code::SUCCESS)
  239. {
  240. std::cout << "huli Locate_manager:::::" << t_error.to_string() << std::endl;
  241. }
  242. else
  243. {
  244. while (locate_manager_task->is_task_end() == false)
  245. {
  246. if (locate_manager_task->is_over_time())
  247. {
  248. //超时处理。取消任务。
  249. Locate_manager::get_instance_pointer()->cancel_task();
  250. locate_manager_task->set_task_statu(TASK_DEAD);
  251. t_error.error_manager_reset(Error_code::LOCATER_MANAGER_TASK_OVER_TIME, Error_level::MINOR_ERROR,
  252. " locate_manager_task is_over_time ");
  253. locate_manager_task->set_task_error_manager(t_error);
  254. }
  255. else
  256. {
  257. std::this_thread::yield();
  258. }
  259. }
  260. std::cout << "huli task_error:::::" << locate_manager_task->get_task_error_manager().to_string() << std::endl;
  261. std::cout << "locate_x = " << locate_manager_task->get_task_locate_information_ex()->locate_x << std::endl;
  262. std::cout << "locate_y = " << locate_manager_task->get_task_locate_information_ex()->locate_y << std::endl;
  263. std::cout << "locate_length = " << locate_manager_task->get_task_locate_information_ex()->locate_length << std::endl;
  264. std::cout << "locate_width = " << locate_manager_task->get_task_locate_information_ex()->locate_width << std::endl;
  265. std::cout << "locate_height = " << locate_manager_task->get_task_locate_information_ex()->locate_height << std::endl;
  266. std::cout << "locate_wheel_base = " << locate_manager_task->get_task_locate_information_ex()->locate_wheel_base << std::endl;
  267. std::cout << "locate_wheel_width = " << locate_manager_task->get_task_locate_information_ex()->locate_wheel_width << std::endl;
  268. std::cout << "locate_angle = " << locate_manager_task->get_task_locate_information_ex()->locate_angle << std::endl;
  269. std::cout << "locate_correct = " << locate_manager_task->get_task_locate_information_ex()->locate_correct << std::endl;
  270. }
  271. delete (locate_manager_task);
  272. std::cout << "huli locate result:::::" << t_error.to_string() << std::endl;
  273. }
  274. std::cout << "err code :" << t_error.to_string() << std::endl;
  275. std::this_thread::sleep_for(std::chrono::seconds(2));
  276. }
  277. }
  278. void test_whole_process()
  279. {
  280. System_executor::get_instance_references().execute_for_measure("key_for_test_only", 5, std::chrono::system_clock::now());
  281. }
  282. int main(int argc,char* argv[])
  283. {
  284. Error_manager t_error;
  285. Error_manager t_result ;
  286. const char* logPath = "./";
  287. google::InitGoogleLogging("LidarMeasurement");
  288. google::SetStderrLogging(google::INFO);
  289. google::SetLogDestination(0, logPath);
  290. google::SetLogFilenameExtension("zxlog");
  291. google::InstallFailureSignalHandler();
  292. google::InstallFailureWriter(&shut_down_logging);
  293. FLAGS_colorlogtostderr = true; // Set log color
  294. FLAGS_logbufsecs = 0; // Set log output speed(s)
  295. FLAGS_max_log_size = 1024; // Set max log file size(GB)
  296. FLAGS_stop_logging_if_full_disk = true;
  297. // Wanji_manager::get_instance_references().wanji_manager_init();
  298. // std::cout << " huli test :::: " << " wanji_manager_init = " << 1 << std::endl;
  299. // 定义服务的终端id
  300. int t_terminal_id = 0;
  301. if ( argc == 2 )
  302. {
  303. std::cout << " huli test :::: " << " argv[1] = " << argv[1] << std::endl;
  304. t_terminal_id = atoi(argv[1]);
  305. }
  306. std::cout << " huli test :::: " << " t_terminal_id = " << t_terminal_id << std::endl;
  307. // 初始化
  308. Laser_manager::get_instance_references().laser_manager_init();
  309. std::cout << "Laser_manager = " << Laser_manager::get_instance_references().get_laser_manager_status() << std::endl;
  310. Locate_manager::get_instance_references().Locate_manager_init();
  311. std::cout << "Locate_manager = " << Locate_manager::get_instance_references().get_locate_manager_status() << std::endl;
  312. System_executor::get_instance_references().system_executor_init(4, t_terminal_id);
  313. std::cout << "System_executor = " << System_executor::get_instance_references().get_system_executor_status() << std::endl;
  314. System_communication::get_instance_references().communication_init();
  315. // prev_test_pred_task();
  316. test_whole_process();
  317. // 反初始化
  318. System_communication::get_instance_references().communication_uninit();
  319. System_executor::get_instance_references().system_executor_uninit();
  320. Locate_manager::get_instance_references().Locate_manager_uninit();
  321. Laser_manager::get_instance_references().laser_manager_uninit();
  322. return 0;
  323. }