Update parameter service handlers to be based on latest master.
This commit is contained in:
parent
1d6cfac7f8
commit
b24af7f038
|
|
@ -1,13 +1,29 @@
|
|||
use futures::executor::LocalPool;
|
||||
use futures::task::LocalSpawnExt;
|
||||
use r2r;
|
||||
|
||||
// try to run like this
|
||||
// cargo run --example parameters -- --ros-args -p key1:=[hello,world] -p key2:=5.5 -r __ns:=/demo -r __node:=my_node
|
||||
// then run
|
||||
// ros2 param get /demo/my_node key2 # (should return 5.5)
|
||||
// ros2 param set /demo/my_node key2 false
|
||||
// ros2 param get /demo/my_node key2 # (should return false)
|
||||
|
||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
// set up executor
|
||||
let mut pool = LocalPool::new();
|
||||
let spawner = pool.spawner();
|
||||
|
||||
// set up ros node
|
||||
let ctx = r2r::Context::create()?;
|
||||
let mut node = r2r::Node::create(ctx, "to_be_replaced", "to_be_replaced")?;
|
||||
let mut timer = node.create_wall_timer(std::time::Duration::from_millis(2000))?;
|
||||
|
||||
// make a parameter handler (once per node).
|
||||
// the parameter handler is optional, only spawn one if you need it.
|
||||
let paramater_handler = node.make_parameter_handler()?;
|
||||
// run parameter handler on your executor.
|
||||
spawner.spawn_local(paramater_handler)?;
|
||||
|
||||
println!("node name: {}", node.name()?);
|
||||
println!(
|
||||
|
|
@ -16,21 +32,20 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|||
);
|
||||
println!("node namespace: {}", node.namespace()?);
|
||||
|
||||
let mut i = 0;
|
||||
// print all params every 2 seconds.
|
||||
let params = node.params.clone();
|
||||
spawner.spawn_local(async move {
|
||||
loop {
|
||||
node.spin_once(std::time::Duration::from_millis(100));
|
||||
if i % 20 == 0 {
|
||||
// every 2 seconds print all parameters
|
||||
println!("node parameters");
|
||||
node.params.lock().unwrap().iter().for_each(|(k, v)| {
|
||||
params.lock().unwrap().iter().for_each(|(k, v)| {
|
||||
println!("{} - {:?}", k, v);
|
||||
});
|
||||
let _elapsed = timer.tick().await.expect("could not tick");
|
||||
}
|
||||
i += 1;
|
||||
if i > 1000 {
|
||||
break;
|
||||
}
|
||||
}
|
||||
})?;
|
||||
|
||||
Ok(())
|
||||
loop {
|
||||
node.spin_once(std::time::Duration::from_millis(100));
|
||||
pool.run_until_stalled();
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -66,7 +66,6 @@ pub use context::Context;
|
|||
|
||||
mod parameters;
|
||||
pub use parameters::ParameterValue;
|
||||
use parameters::*;
|
||||
|
||||
mod clocks;
|
||||
pub use clocks::{Clock, ClockType};
|
||||
|
|
|
|||
67
src/nodes.rs
67
src/nodes.rs
|
|
@ -1,8 +1,9 @@
|
|||
use super::*;
|
||||
use futures::future::{self, join_all};
|
||||
|
||||
pub struct Node {
|
||||
context: Context,
|
||||
pub params: HashMap<String, ParameterValue>,
|
||||
pub params: Arc<Mutex<HashMap<String, ParameterValue>>>,
|
||||
node_handle: Box<rcl_node_t>,
|
||||
// the node owns the subscribers
|
||||
subs: Vec<Box<dyn Subscriber_>>,
|
||||
|
|
@ -104,11 +105,12 @@ impl Node {
|
|||
let param_values =
|
||||
unsafe { std::slice::from_raw_parts(np.parameter_values, np.num_params) };
|
||||
|
||||
let mut params = self.params.lock().unwrap();
|
||||
for (s, v) in param_names.iter().zip(param_values) {
|
||||
let s = unsafe { CStr::from_ptr(*s) };
|
||||
let key = s.to_str().unwrap_or("");
|
||||
let val = parameter_value_from_rcl(&*v);
|
||||
self.params.insert(key.to_owned(), val);
|
||||
let val = ParameterValue::from_rcl(&*v);
|
||||
params.insert(key.to_owned(), val);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -139,7 +141,7 @@ impl Node {
|
|||
|
||||
if res == RCL_RET_OK as i32 {
|
||||
let mut node = Node {
|
||||
params: HashMap::new(),
|
||||
params: Arc::new(Mutex::new(HashMap::new())),
|
||||
context: ctx,
|
||||
node_handle,
|
||||
subs: Vec::new(),
|
||||
|
|
@ -158,6 +160,63 @@ impl Node {
|
|||
}
|
||||
}
|
||||
|
||||
/// Returns a future which handles any parameter change requests. Spawn this onto the executor of choice.
|
||||
pub fn make_parameter_handler(&mut self) -> Result<impl Future<Output = ()>> {
|
||||
let mut handlers: Vec<std::pin::Pin<Box<dyn Future<Output = ()>>>> = Vec::new();
|
||||
|
||||
let node_name = self.name()?;
|
||||
let set_params_request_stream = self.create_service::<rcl_interfaces::srv::SetParameters::Service>(
|
||||
&format!("{}/set_parameters", node_name))?;
|
||||
|
||||
let params = self.params.clone();
|
||||
let set_params_future = set_params_request_stream
|
||||
.for_each(move |req: ServiceRequest<rcl_interfaces::srv::SetParameters::Service>| {
|
||||
let mut result = rcl_interfaces::srv::SetParameters::Response::default();
|
||||
for p in &req.message.parameters {
|
||||
let val = ParameterValue::from_parameter_value_msg(p.value.clone());
|
||||
params.lock().unwrap().insert(p.name.clone(), val);
|
||||
let r = rcl_interfaces::msg::SetParametersResult {
|
||||
successful: true,
|
||||
reason: "".into(),
|
||||
};
|
||||
result.results.push(r);
|
||||
}
|
||||
req.respond(result).expect("could not send reply to set parameter request");
|
||||
future::ready(())
|
||||
});
|
||||
handlers.push(Box::pin(set_params_future));
|
||||
|
||||
// rcl_interfaces/srv/GetParameters
|
||||
let get_params_request_stream = self.create_service::<rcl_interfaces::srv::GetParameters::Service>(
|
||||
&format!("{}/get_parameters", node_name))?;
|
||||
|
||||
let params = self.params.clone();
|
||||
let get_params_future = get_params_request_stream
|
||||
.for_each(move |req: ServiceRequest<rcl_interfaces::srv::GetParameters::Service>| {
|
||||
let params = params.lock().unwrap();
|
||||
let values = req.message.names.iter()
|
||||
.map(|n| {
|
||||
match params.get(n) {
|
||||
Some(v) => v.clone(),
|
||||
None => ParameterValue::NotSet
|
||||
}
|
||||
})
|
||||
.map(|v| v.clone().to_parameter_value_msg())
|
||||
.collect::<Vec<rcl_interfaces::msg::ParameterValue>>();
|
||||
|
||||
let result = rcl_interfaces::srv::GetParameters::Response {
|
||||
values
|
||||
};
|
||||
req.respond(result).expect("could not send reply to set parameter request");
|
||||
future::ready(())
|
||||
});
|
||||
|
||||
handlers.push(Box::pin(get_params_future));
|
||||
|
||||
// we don't care about the result, the futures will not complete anyway.
|
||||
Ok(join_all(handlers).map(|_| ()))
|
||||
}
|
||||
|
||||
pub fn subscribe<T: 'static>(&mut self, topic: &str) -> Result<impl Stream<Item = T> + Unpin>
|
||||
where
|
||||
T: WrappedTypesupport,
|
||||
|
|
|
|||
|
|
@ -14,7 +14,8 @@ pub enum ParameterValue {
|
|||
StringArray(Vec<String>),
|
||||
}
|
||||
|
||||
pub fn parameter_value_from_rcl(v: &rcl_variant_t) -> ParameterValue {
|
||||
impl ParameterValue {
|
||||
pub(crate) fn from_rcl(v: &rcl_variant_t) -> Self {
|
||||
if v.bool_value != std::ptr::null_mut() {
|
||||
ParameterValue::Bool(unsafe { *v.bool_value })
|
||||
} else if v.integer_value != std::ptr::null_mut() {
|
||||
|
|
@ -45,12 +46,18 @@ pub fn parameter_value_from_rcl(v: &rcl_variant_t) -> ParameterValue {
|
|||
ParameterValue::IntegerArray(vals.iter().cloned().collect())
|
||||
} else if v.double_array_value != std::ptr::null_mut() {
|
||||
let vals = unsafe {
|
||||
std::slice::from_raw_parts((*v.double_array_value).values, (*v.double_array_value).size)
|
||||
std::slice::from_raw_parts(
|
||||
(*v.double_array_value).values,
|
||||
(*v.double_array_value).size,
|
||||
)
|
||||
};
|
||||
ParameterValue::DoubleArray(vals.iter().cloned().collect())
|
||||
} else if v.string_array_value != std::ptr::null_mut() {
|
||||
let vals = unsafe {
|
||||
std::slice::from_raw_parts((*v.string_array_value).data, (*v.string_array_value).size)
|
||||
std::slice::from_raw_parts(
|
||||
(*v.string_array_value).data,
|
||||
(*v.string_array_value).size,
|
||||
)
|
||||
};
|
||||
let s = vals
|
||||
.iter()
|
||||
|
|
@ -63,4 +70,73 @@ pub fn parameter_value_from_rcl(v: &rcl_variant_t) -> ParameterValue {
|
|||
} else {
|
||||
ParameterValue::NotSet
|
||||
}
|
||||
}
|
||||
|
||||
pub(crate) fn from_parameter_value_msg(msg: rcl_interfaces::msg::ParameterValue) -> Self {
|
||||
// todo: use constants from ParameterType message
|
||||
match msg.type_ {
|
||||
0 => ParameterValue::NotSet,
|
||||
1 => ParameterValue::Bool(msg.bool_value),
|
||||
2 => ParameterValue::Integer(msg.integer_value),
|
||||
3 => ParameterValue::Double(msg.double_value),
|
||||
4 => ParameterValue::String(msg.string_value),
|
||||
5 => ParameterValue::ByteArray(msg.byte_array_value),
|
||||
6 => ParameterValue::BoolArray(msg.bool_array_value),
|
||||
7 => ParameterValue::IntegerArray(msg.integer_array_value),
|
||||
8 => ParameterValue::DoubleArray(msg.double_array_value),
|
||||
9 => ParameterValue::StringArray(msg.string_array_value),
|
||||
_ => {
|
||||
println!("warning: malformed parametervalue message");
|
||||
ParameterValue::NotSet
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub(crate) fn to_parameter_value_msg(self) -> rcl_interfaces::msg::ParameterValue {
|
||||
let mut ret = rcl_interfaces::msg::ParameterValue::default();
|
||||
|
||||
match self {
|
||||
ParameterValue::NotSet => {
|
||||
ret.type_ = 0; // uint8 PARAMETER_NOT_SET=0
|
||||
},
|
||||
ParameterValue::Bool(b) => {
|
||||
ret.type_ = 1; // uint8 PARAMETER_BOOL=1
|
||||
ret.bool_value = b;
|
||||
},
|
||||
ParameterValue::Integer(i) => {
|
||||
ret.type_ = 2; // uint8 PARAMETER_INTEGER=2
|
||||
ret.integer_value = i;
|
||||
},
|
||||
ParameterValue::Double(d) => {
|
||||
ret.type_ = 3; // uint8 PARAMETER_DOUBLE=3
|
||||
ret.double_value = d;
|
||||
},
|
||||
ParameterValue::String(s) => {
|
||||
ret.type_ = 4; // uint8 PARAMETER_STRING=4
|
||||
ret.string_value = s;
|
||||
},
|
||||
ParameterValue::ByteArray(ba) => {
|
||||
ret.type_ = 5; // uint8 PARAMETER_BYTE_ARRAY=5
|
||||
ret.byte_array_value = ba;
|
||||
},
|
||||
ParameterValue::BoolArray(ba) => {
|
||||
ret.type_ = 6; // uint8 PARAMETER_BOOL_ARRAY=6
|
||||
ret.bool_array_value = ba;
|
||||
},
|
||||
ParameterValue::IntegerArray(ia) => {
|
||||
ret.type_ = 7; // uint8 PARAMETER_INTEGER_ARRAY=7
|
||||
ret.integer_array_value = ia;
|
||||
},
|
||||
ParameterValue::DoubleArray(da) => {
|
||||
ret.type_ = 8; // uint8 PARAMETER_DOUBLE_ARRAY=8
|
||||
ret.double_array_value = da;
|
||||
},
|
||||
ParameterValue::StringArray(sa) => {
|
||||
ret.type_ = 9; // int PARAMETER_STRING_ARRAY=9
|
||||
ret.string_array_value = sa;
|
||||
},
|
||||
}
|
||||
ret
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue