lds_lidar.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780
  1. //
  2. // The MIT License (MIT)
  3. //
  4. // Copyright (c) 2019 Livox. All rights reserved.
  5. //
  6. // Permission is hereby granted, free of charge, to any person obtaining a copy
  7. // of this software and associated documentation files (the "Software"), to deal
  8. // in the Software without restriction, including without limitation the rights
  9. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10. // copies of the Software, and to permit persons to whom the Software is
  11. // furnished to do so, subject to the following conditions:
  12. //
  13. // The above copyright notice and this permission notice shall be included in
  14. // all copies or substantial portions of the Software.
  15. //
  16. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22. // SOFTWARE.
  23. //
  24. #include "lds_lidar.h"
  25. #include <stdio.h>
  26. #include <string.h>
  27. #include <memory>
  28. #include <mutex>
  29. #include <thread>
  30. #include "rapidjson/document.h"
  31. #include "rapidjson/filereadstream.h"
  32. #include "rapidjson/stringbuffer.h"
  33. using namespace std;
  34. namespace livox_ros {
  35. /** Const varible ------------------------------------------------------------*/
  36. /** For callback use only */
  37. LdsLidar *g_lds_ldiar = nullptr;
  38. /** Global function for common use -------------------------------------------*/
  39. /** Lds lidar function -------------------------------------------------------*/
  40. LdsLidar::LdsLidar(uint32_t interval_ms) : Lds(interval_ms, kSourceRawLidar) {
  41. auto_connect_mode_ = true;
  42. is_initialized_ = false;
  43. whitelist_count_ = 0;
  44. memset(broadcast_code_whitelist_, 0, sizeof(broadcast_code_whitelist_));
  45. ResetLdsLidar();
  46. }
  47. LdsLidar::~LdsLidar() {}
  48. void LdsLidar::ResetLdsLidar(void) { ResetLds(kSourceRawLidar); }
  49. int LdsLidar::InitLdsLidar(std::vector<std::string> &broadcast_code_strs,
  50. const char *user_config_path) {
  51. if (is_initialized_) {
  52. printf("LiDAR data source is already inited!\n");
  53. return -1;
  54. }
  55. if (!Init()) {
  56. Uninit();
  57. printf("Livox-SDK init fail!\n");
  58. return -1;
  59. }
  60. LivoxSdkVersion _sdkversion;
  61. GetLivoxSdkVersion(&_sdkversion);
  62. printf("Livox SDK version %d.%d.%d\n", _sdkversion.major, _sdkversion.minor,
  63. _sdkversion.patch);
  64. SetBroadcastCallback(OnDeviceBroadcast);
  65. SetDeviceStateUpdateCallback(OnDeviceChange);
  66. /** Add commandline input broadcast code */
  67. for (auto input_str : broadcast_code_strs) {
  68. AddBroadcastCodeToWhitelist(input_str.c_str());
  69. }
  70. ParseConfigFile(user_config_path);
  71. if (whitelist_count_) {
  72. DisableAutoConnectMode();
  73. printf("Disable auto connect mode!\n");
  74. printf("List all broadcast code in whiltelist:\n");
  75. for (uint32_t i = 0; i < whitelist_count_; i++) {
  76. printf("%s\n", broadcast_code_whitelist_[i]);
  77. }
  78. } else {
  79. EnableAutoConnectMode();
  80. printf(
  81. "No broadcast code was added to whitelist, swith to automatic "
  82. "connection mode!\n");
  83. }
  84. if (enable_timesync_) {
  85. timesync_ = TimeSync::GetInstance();
  86. if (timesync_->InitTimeSync(timesync_config_)) {
  87. printf("Timesync init fail\n");
  88. return -1;
  89. }
  90. if (timesync_->SetReceiveSyncTimeCb(ReceiveSyncTimeCallback, this)) {
  91. printf("Set Timesync callback fail\n");
  92. return -1;
  93. }
  94. timesync_->StartTimesync();
  95. }
  96. /** Start livox sdk to receive lidar data */
  97. if (!Start()) {
  98. Uninit();
  99. printf("Livox-SDK init fail!\n");
  100. return -1;
  101. }
  102. /** Add here, only for callback use */
  103. if (g_lds_ldiar == nullptr) {
  104. g_lds_ldiar = this;
  105. }
  106. is_initialized_ = true;
  107. printf("Livox-SDK init success!\n");
  108. return 0;
  109. }
  110. int LdsLidar::DeInitLdsLidar(void) {
  111. if (!is_initialized_) {
  112. printf("LiDAR data source is not exit");
  113. return -1;
  114. }
  115. Uninit();
  116. printf("Livox SDK Deinit completely!\n");
  117. if (timesync_) {
  118. timesync_->DeInitTimeSync();
  119. }
  120. return 0;
  121. }
  122. void LdsLidar::PrepareExit(void) { DeInitLdsLidar(); }
  123. /** Static function in LdsLidar for callback or event process ----------------*/
  124. /** Receiving point cloud data from Livox LiDAR. */
  125. void LdsLidar::OnLidarDataCb(uint8_t handle, LivoxEthPacket *data,
  126. uint32_t data_num, void *client_data) {
  127. using namespace std;
  128. LdsLidar *lds_lidar = static_cast<LdsLidar *>(client_data);
  129. LivoxEthPacket *eth_packet = data;
  130. if (!data || !data_num || (handle >= kMaxLidarCount)) {
  131. return;
  132. }
  133. lds_lidar->StorageRawPacket(handle, eth_packet);
  134. }
  135. void LdsLidar::OnDeviceBroadcast(const BroadcastDeviceInfo *info) {
  136. if (info == nullptr) {
  137. return;
  138. }
  139. if (info->dev_type == kDeviceTypeHub) {
  140. printf("In lidar mode, couldn't connect a hub : %s\n",
  141. info->broadcast_code);
  142. return;
  143. }
  144. if (g_lds_ldiar->IsAutoConnectMode()) {
  145. printf("In automatic connection mode, will connect %s\n",
  146. info->broadcast_code);
  147. } else {
  148. if (!g_lds_ldiar->IsBroadcastCodeExistInWhitelist(info->broadcast_code)) {
  149. printf("Not in the whitelist, please add %s to if want to connect!\n",
  150. info->broadcast_code);
  151. return;
  152. }
  153. }
  154. bool result = false;
  155. uint8_t handle = 0;
  156. result = AddLidarToConnect(info->broadcast_code, &handle);
  157. if (result == kStatusSuccess && handle < kMaxLidarCount) {
  158. SetDataCallback(handle, OnLidarDataCb, (void *)g_lds_ldiar);
  159. LidarDevice *p_lidar = &(g_lds_ldiar->lidars_[handle]);
  160. p_lidar->handle = handle;
  161. p_lidar->connect_state = kConnectStateOff;
  162. UserRawConfig config;
  163. if (g_lds_ldiar->GetRawConfig(info->broadcast_code, config)) {
  164. printf("Could not find raw config, set config to default!\n");
  165. config.enable_fan = 1;
  166. config.return_mode = kFirstReturn;
  167. config.coordinate = kCoordinateCartesian;
  168. config.imu_rate = kImuFreq200Hz;
  169. config.extrinsic_parameter_source = kNoneExtrinsicParameter;
  170. config.enable_high_sensitivity = false;
  171. }
  172. p_lidar->config.enable_fan = config.enable_fan;
  173. p_lidar->config.return_mode = config.return_mode;
  174. p_lidar->config.coordinate = config.coordinate;
  175. p_lidar->config.imu_rate = config.imu_rate;
  176. p_lidar->config.extrinsic_parameter_source =
  177. config.extrinsic_parameter_source;
  178. p_lidar->config.enable_high_sensitivity = config.enable_high_sensitivity;
  179. } else {
  180. printf("Add lidar to connect is failed : %d %d \n", result, handle);
  181. }
  182. }
  183. /** Callback function of changing of device state. */
  184. void LdsLidar::OnDeviceChange(const DeviceInfo *info, DeviceEvent type) {
  185. if (info == nullptr) {
  186. return;
  187. }
  188. uint8_t handle = info->handle;
  189. if (handle >= kMaxLidarCount) {
  190. return;
  191. }
  192. LidarDevice *p_lidar = &(g_lds_ldiar->lidars_[handle]);
  193. if (type == kEventConnect) {
  194. QueryDeviceInformation(handle, DeviceInformationCb, g_lds_ldiar);
  195. if (p_lidar->connect_state == kConnectStateOff) {
  196. p_lidar->connect_state = kConnectStateOn;
  197. p_lidar->info = *info;
  198. }
  199. } else if (type == kEventDisconnect) {
  200. printf("Lidar[%s] disconnect!\n", info->broadcast_code);
  201. ResetLidar(p_lidar, kSourceRawLidar);
  202. } else if (type == kEventStateChange) {
  203. p_lidar->info = *info;
  204. }
  205. if (p_lidar->connect_state == kConnectStateOn) {
  206. printf("Lidar[%s] status_code[%d] working state[%d] feature[%d]\n",
  207. p_lidar->info.broadcast_code,
  208. p_lidar->info.status.status_code.error_code, p_lidar->info.state,
  209. p_lidar->info.feature);
  210. SetErrorMessageCallback(handle, LidarErrorStatusCb);
  211. /** Config lidar parameter */
  212. if (p_lidar->info.state == kLidarStateNormal) {
  213. /** Ensure the thread safety for set_bits and connect_state */
  214. lock_guard<mutex> lock(g_lds_ldiar->config_mutex_);
  215. if (p_lidar->config.coordinate != 0) {
  216. SetSphericalCoordinate(handle, SetCoordinateCb, g_lds_ldiar);
  217. } else {
  218. SetCartesianCoordinate(handle, SetCoordinateCb, g_lds_ldiar);
  219. }
  220. p_lidar->config.set_bits |= kConfigCoordinate;
  221. if (kDeviceTypeLidarMid40 != info->type) {
  222. LidarSetPointCloudReturnMode(
  223. handle, (PointCloudReturnMode)(p_lidar->config.return_mode),
  224. SetPointCloudReturnModeCb, g_lds_ldiar);
  225. p_lidar->config.set_bits |= kConfigReturnMode;
  226. }
  227. if ((kDeviceTypeLidarMid70 != info->type) &&
  228. (kDeviceTypeLidarMid40 != info->type)) {
  229. LidarSetImuPushFrequency(handle, (ImuFreq)(p_lidar->config.imu_rate),
  230. SetImuRatePushFrequencyCb, g_lds_ldiar);
  231. p_lidar->config.set_bits |= kConfigImuRate;
  232. }
  233. if (p_lidar->config.extrinsic_parameter_source ==
  234. kExtrinsicParameterFromLidar) {
  235. LidarGetExtrinsicParameter(handle, GetLidarExtrinsicParameterCb,
  236. g_lds_ldiar);
  237. p_lidar->config.set_bits |= kConfigGetExtrinsicParameter;
  238. }
  239. if (kDeviceTypeLidarTele == info->type) {
  240. if (p_lidar->config.enable_high_sensitivity) {
  241. LidarEnableHighSensitivity(handle, SetHighSensitivityCb, g_lds_ldiar);
  242. printf("Enable high sensitivity\n");
  243. } else {
  244. LidarDisableHighSensitivity(handle, SetHighSensitivityCb,
  245. g_lds_ldiar);
  246. printf("Disable high sensitivity\n");
  247. }
  248. p_lidar->config.set_bits |= kConfigSetHighSensitivity;
  249. }
  250. p_lidar->connect_state = kConnectStateConfig;
  251. }
  252. }
  253. }
  254. /** Query the firmware version of Livox LiDAR. */
  255. void LdsLidar::DeviceInformationCb(livox_status status, uint8_t handle,
  256. DeviceInformationResponse *ack,
  257. void *clent_data) {
  258. if (status != kStatusSuccess) {
  259. printf("Device Query Informations Failed : %d\n", status);
  260. }
  261. if (ack) {
  262. printf("firmware version: %d.%d.%d.%d\n", ack->firmware_version[0],
  263. ack->firmware_version[1], ack->firmware_version[2],
  264. ack->firmware_version[3]);
  265. }
  266. }
  267. /** Callback function of Lidar error message. */
  268. void LdsLidar::LidarErrorStatusCb(livox_status status, uint8_t handle,
  269. ErrorMessage *message) {
  270. static uint32_t error_message_count = 0;
  271. if (message != NULL) {
  272. ++error_message_count;
  273. if (0 == (error_message_count % 100)) {
  274. printf("handle: %u\n", handle);
  275. printf("temp_status : %u\n", message->lidar_error_code.temp_status);
  276. printf("volt_status : %u\n", message->lidar_error_code.volt_status);
  277. printf("motor_status : %u\n", message->lidar_error_code.motor_status);
  278. printf("dirty_warn : %u\n", message->lidar_error_code.dirty_warn);
  279. printf("firmware_err : %u\n", message->lidar_error_code.firmware_err);
  280. printf("pps_status : %u\n", message->lidar_error_code.device_status);
  281. printf("fan_status : %u\n", message->lidar_error_code.fan_status);
  282. printf("self_heating : %u\n", message->lidar_error_code.self_heating);
  283. printf("ptp_status : %u\n", message->lidar_error_code.ptp_status);
  284. printf("time_sync_status : %u\n",
  285. message->lidar_error_code.time_sync_status);
  286. printf("system_status : %u\n", message->lidar_error_code.system_status);
  287. }
  288. }
  289. }
  290. void LdsLidar::ControlFanCb(livox_status status, uint8_t handle,
  291. uint8_t response, void *clent_data) {}
  292. void LdsLidar::SetPointCloudReturnModeCb(livox_status status, uint8_t handle,
  293. uint8_t response, void *clent_data) {
  294. LdsLidar *lds_lidar = static_cast<LdsLidar *>(clent_data);
  295. if (handle >= kMaxLidarCount) {
  296. return;
  297. }
  298. LidarDevice *p_lidar = &(lds_lidar->lidars_[handle]);
  299. if (status == kStatusSuccess) {
  300. printf("Set return mode success!\n");
  301. lock_guard<mutex> lock(lds_lidar->config_mutex_);
  302. p_lidar->config.set_bits &= ~((uint32_t)(kConfigReturnMode));
  303. if (!p_lidar->config.set_bits) {
  304. LidarStartSampling(handle, StartSampleCb, lds_lidar);
  305. p_lidar->connect_state = kConnectStateSampling;
  306. }
  307. } else {
  308. LidarSetPointCloudReturnMode(
  309. handle, (PointCloudReturnMode)(p_lidar->config.return_mode),
  310. SetPointCloudReturnModeCb, lds_lidar);
  311. printf("Set return mode fail, try again!\n");
  312. }
  313. }
  314. void LdsLidar::SetCoordinateCb(livox_status status, uint8_t handle,
  315. uint8_t response, void *clent_data) {
  316. LdsLidar *lds_lidar = static_cast<LdsLidar *>(clent_data);
  317. if (handle >= kMaxLidarCount) {
  318. return;
  319. }
  320. LidarDevice *p_lidar = &(lds_lidar->lidars_[handle]);
  321. if (status == kStatusSuccess) {
  322. printf("Set coordinate success!\n");
  323. lock_guard<mutex> lock(lds_lidar->config_mutex_);
  324. p_lidar->config.set_bits &= ~((uint32_t)(kConfigCoordinate));
  325. if (!p_lidar->config.set_bits) {
  326. LidarStartSampling(handle, StartSampleCb, lds_lidar);
  327. p_lidar->connect_state = kConnectStateSampling;
  328. }
  329. } else {
  330. if (p_lidar->config.coordinate != 0) {
  331. SetSphericalCoordinate(handle, SetCoordinateCb, lds_lidar);
  332. } else {
  333. SetCartesianCoordinate(handle, SetCoordinateCb, lds_lidar);
  334. }
  335. printf("Set coordinate fail, try again!\n");
  336. }
  337. }
  338. void LdsLidar::SetImuRatePushFrequencyCb(livox_status status, uint8_t handle,
  339. uint8_t response, void *clent_data) {
  340. LdsLidar *lds_lidar = static_cast<LdsLidar *>(clent_data);
  341. if (handle >= kMaxLidarCount) {
  342. return;
  343. }
  344. LidarDevice *p_lidar = &(lds_lidar->lidars_[handle]);
  345. if (status == kStatusSuccess) {
  346. printf("Set imu rate success!\n");
  347. lock_guard<mutex> lock(lds_lidar->config_mutex_);
  348. p_lidar->config.set_bits &= ~((uint32_t)(kConfigImuRate));
  349. if (!p_lidar->config.set_bits) {
  350. LidarStartSampling(handle, StartSampleCb, lds_lidar);
  351. p_lidar->connect_state = kConnectStateSampling;
  352. }
  353. } else {
  354. LidarSetImuPushFrequency(handle, (ImuFreq)(p_lidar->config.imu_rate),
  355. SetImuRatePushFrequencyCb, g_lds_ldiar);
  356. printf("Set imu rate fail, try again!\n");
  357. }
  358. }
  359. /** Callback function of get LiDARs' extrinsic parameter. */
  360. void LdsLidar::GetLidarExtrinsicParameterCb(
  361. livox_status status, uint8_t handle,
  362. LidarGetExtrinsicParameterResponse *response, void *clent_data) {
  363. LdsLidar *lds_lidar = static_cast<LdsLidar *>(clent_data);
  364. if (handle >= kMaxLidarCount) {
  365. return;
  366. }
  367. if (status == kStatusSuccess) {
  368. if (response != nullptr) {
  369. printf("Lidar[%d] get ExtrinsicParameter status[%d] response[%d]\n",
  370. handle, status, response->ret_code);
  371. LidarDevice *p_lidar = &(lds_lidar->lidars_[handle]);
  372. ExtrinsicParameter *p_extrinsic = &p_lidar->extrinsic_parameter;
  373. p_extrinsic->euler[0] = static_cast<float>(response->roll * PI / 180.0);
  374. p_extrinsic->euler[1] = static_cast<float>(response->pitch * PI / 180.0);
  375. p_extrinsic->euler[2] = static_cast<float>(response->yaw * PI / 180.0);
  376. p_extrinsic->trans[0] = static_cast<float>(response->x / 1000.0);
  377. p_extrinsic->trans[1] = static_cast<float>(response->y / 1000.0);
  378. p_extrinsic->trans[2] = static_cast<float>(response->z / 1000.0);
  379. EulerAnglesToRotationMatrix(p_extrinsic->euler, p_extrinsic->rotation);
  380. if (p_lidar->config.extrinsic_parameter_source) {
  381. p_extrinsic->enable = true;
  382. }
  383. printf("Lidar[%d] get ExtrinsicParameter success!\n", handle);
  384. lock_guard<mutex> lock(lds_lidar->config_mutex_);
  385. p_lidar->config.set_bits &= ~((uint32_t)(kConfigGetExtrinsicParameter));
  386. if (!p_lidar->config.set_bits) {
  387. LidarStartSampling(handle, StartSampleCb, lds_lidar);
  388. p_lidar->connect_state = kConnectStateSampling;
  389. }
  390. } else {
  391. printf("Lidar[%d] get ExtrinsicParameter fail!\n", handle);
  392. }
  393. } else if (status == kStatusTimeout) {
  394. printf("Lidar[%d] get ExtrinsicParameter timeout!\n", handle);
  395. }
  396. }
  397. void LdsLidar::SetHighSensitivityCb(livox_status status, uint8_t handle,
  398. DeviceParameterResponse *response,
  399. void *clent_data) {
  400. LdsLidar *lds_lidar = static_cast<LdsLidar *>(clent_data);
  401. if (handle >= kMaxLidarCount) {
  402. return;
  403. }
  404. LidarDevice *p_lidar = &(lds_lidar->lidars_[handle]);
  405. if (status == kStatusSuccess) {
  406. p_lidar->config.set_bits &= ~((uint32_t)(kConfigSetHighSensitivity));
  407. printf("Set high sensitivity success!\n");
  408. lock_guard<mutex> lock(lds_lidar->config_mutex_);
  409. if (!p_lidar->config.set_bits) {
  410. LidarStartSampling(handle, StartSampleCb, lds_lidar);
  411. p_lidar->connect_state = kConnectStateSampling;
  412. };
  413. } else {
  414. if (p_lidar->config.enable_high_sensitivity) {
  415. LidarEnableHighSensitivity(handle, SetHighSensitivityCb, g_lds_ldiar);
  416. } else {
  417. LidarDisableHighSensitivity(handle, SetHighSensitivityCb, g_lds_ldiar);
  418. }
  419. printf("Set high sensitivity fail, try again!\n");
  420. }
  421. }
  422. /** Callback function of starting sampling. */
  423. void LdsLidar::StartSampleCb(livox_status status, uint8_t handle,
  424. uint8_t response, void *clent_data) {
  425. LdsLidar *lds_lidar = static_cast<LdsLidar *>(clent_data);
  426. if (handle >= kMaxLidarCount) {
  427. return;
  428. }
  429. LidarDevice *p_lidar = &(lds_lidar->lidars_[handle]);
  430. if (status == kStatusSuccess) {
  431. if (response != 0) {
  432. p_lidar->connect_state = kConnectStateOn;
  433. printf("Lidar start sample fail : state[%d] handle[%d] res[%d]\n", status,
  434. handle, response);
  435. } else {
  436. printf("Lidar start sample success\n");
  437. }
  438. } else if (status == kStatusTimeout) {
  439. p_lidar->connect_state = kConnectStateOn;
  440. printf("Lidar start sample timeout : state[%d] handle[%d] res[%d]\n",
  441. status, handle, response);
  442. }
  443. }
  444. /** Callback function of stopping sampling. */
  445. void LdsLidar::StopSampleCb(livox_status status, uint8_t handle,
  446. uint8_t response, void *clent_data) {}
  447. void LdsLidar::SetRmcSyncTimeCb(livox_status status, uint8_t handle,
  448. uint8_t response, void *client_data) {
  449. if (handle >= kMaxLidarCount) {
  450. return;
  451. }
  452. printf("Set lidar[%d] sync time status[%d] response[%d]\n", handle, status,
  453. response);
  454. }
  455. void LdsLidar::ReceiveSyncTimeCallback(const char *rmc, uint32_t rmc_length,
  456. void *client_data) {
  457. LdsLidar *lds_lidar = static_cast<LdsLidar *>(client_data);
  458. // std::unique_lock<std::mutex> lock(mtx);
  459. LidarDevice *p_lidar = nullptr;
  460. for (uint8_t handle = 0; handle < kMaxLidarCount; handle++) {
  461. p_lidar = &(lds_lidar->lidars_[handle]);
  462. if (p_lidar->connect_state == kConnectStateSampling &&
  463. p_lidar->info.state == kLidarStateNormal) {
  464. livox_status status = LidarSetRmcSyncTime(handle, rmc, rmc_length,
  465. SetRmcSyncTimeCb, lds_lidar);
  466. if (status != kStatusSuccess) {
  467. printf("Set GPRMC synchronization time error code: %d.\n", status);
  468. }
  469. }
  470. }
  471. }
  472. /** Add broadcast code to whitelist */
  473. int LdsLidar::AddBroadcastCodeToWhitelist(const char *broadcast_code) {
  474. if (!broadcast_code || (strlen(broadcast_code) > kBroadcastCodeSize) ||
  475. (whitelist_count_ >= kMaxLidarCount)) {
  476. return -1;
  477. }
  478. if (LdsLidar::IsBroadcastCodeExistInWhitelist(broadcast_code)) {
  479. printf("%s is alrealy exist!\n", broadcast_code);
  480. return -1;
  481. }
  482. strcpy(broadcast_code_whitelist_[whitelist_count_], broadcast_code);
  483. ++whitelist_count_;
  484. return 0;
  485. }
  486. bool LdsLidar::IsBroadcastCodeExistInWhitelist(const char *broadcast_code) {
  487. if (!broadcast_code) {
  488. return false;
  489. }
  490. for (uint32_t i = 0; i < whitelist_count_; i++) {
  491. if (strncmp(broadcast_code, broadcast_code_whitelist_[i],
  492. kBroadcastCodeSize) == 0) {
  493. return true;
  494. }
  495. }
  496. return false;
  497. }
  498. int LdsLidar::ParseTimesyncConfig(rapidjson::Document &doc) {
  499. do {
  500. if (!doc.HasMember("timesync_config") || !doc["timesync_config"].IsObject())
  501. break;
  502. const rapidjson::Value &object = doc["timesync_config"];
  503. if (!object.IsObject()) break;
  504. if (!object.HasMember("enable_timesync") ||
  505. !object["enable_timesync"].IsBool())
  506. break;
  507. enable_timesync_ = object["enable_timesync"].GetBool();
  508. if (!object.HasMember("device_name") || !object["device_name"].IsString())
  509. break;
  510. std::string device_name = object["device_name"].GetString();
  511. std::strncpy(timesync_config_.dev_config.name, device_name.c_str(),
  512. sizeof(timesync_config_.dev_config.name));
  513. if (!object.HasMember("comm_device_type") ||
  514. !object["comm_device_type"].IsInt())
  515. break;
  516. timesync_config_.dev_config.type = object["comm_device_type"].GetInt();
  517. if (timesync_config_.dev_config.type == kCommDevUart) {
  518. if (!object.HasMember("baudrate_index") ||
  519. !object["baudrate_index"].IsInt())
  520. break;
  521. timesync_config_.dev_config.config.uart.baudrate =
  522. object["baudrate_index"].GetInt();
  523. if (!object.HasMember("parity_index") || !object["parity_index"].IsInt())
  524. break;
  525. timesync_config_.dev_config.config.uart.parity =
  526. object["parity_index"].GetInt();
  527. }
  528. if (enable_timesync_) {
  529. printf("Enable timesync : \n");
  530. if (timesync_config_.dev_config.type == kCommDevUart) {
  531. printf("Uart[%s],baudrate index[%d],parity index[%d]\n",
  532. timesync_config_.dev_config.name,
  533. timesync_config_.dev_config.config.uart.baudrate,
  534. timesync_config_.dev_config.config.uart.parity);
  535. }
  536. } else {
  537. printf("Disable timesync\n");
  538. }
  539. return 0;
  540. } while (0);
  541. return -1;
  542. }
  543. /** Config file process */
  544. int LdsLidar::ParseConfigFile(const char *pathname) {
  545. FILE *raw_file = std::fopen(pathname, "rb");
  546. if (!raw_file) {
  547. printf("Open json config file fail!\n");
  548. return -1;
  549. }
  550. char read_buffer[32768];
  551. rapidjson::FileReadStream config_file(raw_file, read_buffer,
  552. sizeof(read_buffer));
  553. rapidjson::Document doc;
  554. if (!doc.ParseStream(config_file).HasParseError()) {
  555. if (doc.HasMember("lidar_config") && doc["lidar_config"].IsArray()) {
  556. const rapidjson::Value &array = doc["lidar_config"];
  557. size_t len = array.Size();
  558. for (size_t i = 0; i < len; i++) {
  559. const rapidjson::Value &object = array[i];
  560. if (object.IsObject()) {
  561. UserRawConfig config = {0};
  562. memset(&config, 0, sizeof(config));
  563. if (object.HasMember("broadcast_code") &&
  564. object["broadcast_code"].IsString()) {
  565. std::string broadcast_code = object["broadcast_code"].GetString();
  566. std::strncpy(config.broadcast_code, broadcast_code.c_str(),
  567. sizeof(config.broadcast_code));
  568. } else {
  569. printf("User config file parse error\n");
  570. continue;
  571. }
  572. if (object.HasMember("enable_connect") &&
  573. object["enable_connect"].IsBool()) {
  574. config.enable_connect = object["enable_connect"].GetBool();
  575. }
  576. if (object.HasMember("enable_fan") && object["enable_fan"].IsBool()) {
  577. config.enable_fan = object["enable_fan"].GetBool();
  578. }
  579. if (object.HasMember("return_mode") &&
  580. object["return_mode"].IsInt()) {
  581. config.return_mode = object["return_mode"].GetInt();
  582. }
  583. if (object.HasMember("coordinate") && object["coordinate"].IsInt()) {
  584. config.coordinate = object["coordinate"].GetInt();
  585. }
  586. if (object.HasMember("imu_rate") && object["imu_rate"].IsInt()) {
  587. config.imu_rate = object["imu_rate"].GetInt();
  588. }
  589. if (object.HasMember("extrinsic_parameter_source") &&
  590. object["extrinsic_parameter_source"].IsInt()) {
  591. config.extrinsic_parameter_source =
  592. object["extrinsic_parameter_source"].GetInt();
  593. }
  594. if (object.HasMember("enable_high_sensitivity") &&
  595. object["enable_high_sensitivity"].GetBool()) {
  596. config.enable_high_sensitivity =
  597. object["enable_high_sensitivity"].GetBool();
  598. }
  599. printf("broadcast code[%s] : %d %d %d %d %d %d\n",
  600. config.broadcast_code, config.enable_connect,
  601. config.enable_fan, config.return_mode, config.coordinate,
  602. config.imu_rate, config.extrinsic_parameter_source);
  603. if (config.enable_connect) {
  604. if (!AddBroadcastCodeToWhitelist(config.broadcast_code)) {
  605. if (AddRawUserConfig(config)) {
  606. printf("Raw config is already exist : %s \n",
  607. config.broadcast_code);
  608. }
  609. }
  610. }
  611. }
  612. }
  613. }
  614. if (ParseTimesyncConfig(doc)) {
  615. printf("Parse timesync config fail\n");
  616. enable_timesync_ = false;
  617. }
  618. } else {
  619. printf("User config file parse error[%d]\n",
  620. doc.ParseStream(config_file).HasParseError());
  621. }
  622. std::fclose(raw_file);
  623. return 0;
  624. }
  625. int LdsLidar::AddRawUserConfig(UserRawConfig &config) {
  626. if (IsExistInRawConfig(config.broadcast_code)) {
  627. return -1;
  628. }
  629. raw_config_.push_back(config);
  630. printf("Add Raw user config : %s \n", config.broadcast_code);
  631. return 0;
  632. }
  633. bool LdsLidar::IsExistInRawConfig(const char *broadcast_code) {
  634. if (broadcast_code == nullptr) {
  635. return false;
  636. }
  637. for (auto ite_config : raw_config_) {
  638. if (strncmp(ite_config.broadcast_code, broadcast_code,
  639. kBroadcastCodeSize) == 0) {
  640. return true;
  641. }
  642. }
  643. return false;
  644. }
  645. int LdsLidar::GetRawConfig(const char *broadcast_code, UserRawConfig &config) {
  646. if (broadcast_code == nullptr) {
  647. return -1;
  648. }
  649. for (auto ite_config : raw_config_) {
  650. if (strncmp(ite_config.broadcast_code, broadcast_code,
  651. kBroadcastCodeSize) == 0) {
  652. config.enable_fan = ite_config.enable_fan;
  653. config.return_mode = ite_config.return_mode;
  654. config.coordinate = ite_config.coordinate;
  655. config.imu_rate = ite_config.imu_rate;
  656. config.extrinsic_parameter_source = ite_config.extrinsic_parameter_source;
  657. config.enable_high_sensitivity = ite_config.enable_high_sensitivity;
  658. return 0;
  659. }
  660. }
  661. return -1;
  662. }
  663. } // namespace livox_ros