forked from zivid/zivid-ros
-
Notifications
You must be signed in to change notification settings - Fork 1
/
sample_capture_and_save.cpp
106 lines (88 loc) · 3.33 KB
/
sample_capture_and_save.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <filesystem>
#include <rclcpp/rclcpp.hpp>
#include <stdexcept>
#include <zivid_interfaces/srv/capture_and_save.hpp>
/*
* This sample shows how to set the settings_yaml parameter of the zivid node, then invoke the
* capture_and_save service, and read the response from the service call. If successful, the
* captured frame is stored in a temporary directory.
*/
void fatal_error(const rclcpp::Logger & logger, const std::string & message)
{
RCLCPP_ERROR_STREAM(logger, message);
throw std::runtime_error(message);
}
void set_settings(const std::shared_ptr<rclcpp::Node> & node)
{
RCLCPP_INFO(node->get_logger(), "Setting parameter 'settings_yaml'");
const std::string settings_yml =
R"(
__version__:
serializer: 1
data: 22
Settings:
Acquisitions:
- Acquisition:
Aperture: 5.66
ExposureTime: 8333
Processing:
Filters:
Outlier:
Removal:
Enabled: yes
Threshold: 5
)";
auto param_client = std::make_shared<rclcpp::AsyncParametersClient>(node, "zivid_camera");
while (!param_client->wait_for_service(std::chrono::seconds(3))) {
if (!rclcpp::ok()) {
fatal_error(node->get_logger(), "Client interrupted while waiting for service to appear.");
}
RCLCPP_INFO(node->get_logger(), "Waiting for the parameters client to appear...");
}
auto result = param_client->set_parameters({rclcpp::Parameter("settings_yaml", settings_yml)});
if (
rclcpp::spin_until_future_complete(node, result, std::chrono::seconds(30)) !=
rclcpp::FutureReturnCode::SUCCESS) {
fatal_error(node->get_logger(), "Failed to set `settings_yaml` parameter");
}
}
void capture_and_save(const std::shared_ptr<rclcpp::Node> & node)
{
using zivid_interfaces::srv::CaptureAndSave;
auto client = node->create_client<CaptureAndSave>("capture_and_save");
while (!client->wait_for_service(std::chrono::seconds(3))) {
if (!rclcpp::ok()) {
fatal_error(node->get_logger(), "Client interrupted while waiting for service to appear.");
}
RCLCPP_INFO(node->get_logger(), "Waiting for the capture_and_save service to appear...");
}
const std::string filename = "zivid_sample_capture_and_save.zdf";
auto request = std::make_shared<CaptureAndSave::Request>();
request->file_path = (std::filesystem::temp_directory_path() / filename).string();
RCLCPP_INFO(
node->get_logger(), "Sending capture and save request with file path: %s",
request->file_path.c_str());
auto result = client->async_send_request(request);
if (rclcpp::spin_until_future_complete(node, result) != rclcpp::FutureReturnCode::SUCCESS) {
fatal_error(node->get_logger(), "Failed to call the capture_and_save service");
}
auto capture_response = result.get();
if (!capture_response->success) {
fatal_error(
node->get_logger(),
"Capture and save operation was unsuccessful: " + capture_response->message);
}
RCLCPP_INFO(node->get_logger(), "Capture and save operation successful");
}
int main(int argc, char * argv[])
{
rclcpp::init(argc, argv);
auto node = rclcpp::Node::make_shared("sample_capture_and_save");
RCLCPP_INFO(node->get_logger(), "Started the sample_capture_and_save node");
set_settings(node);
capture_and_save(node);
RCLCPP_INFO(node->get_logger(), "Spinning node.. Press Ctrl+C to abort.");
rclcpp::spin(node);
rclcpp::shutdown();
return EXIT_SUCCESS;
}