forked from M-Labs/humpback-dds
mqtt_mux: use nom
This commit is contained in:
parent
883e821794
commit
d47f9b4655
30
Cargo.lock
generated
30
Cargo.lock
generated
@ -244,11 +244,13 @@ dependencies = [
|
|||||||
"log",
|
"log",
|
||||||
"minimq",
|
"minimq",
|
||||||
"nb 1.0.0",
|
"nb 1.0.0",
|
||||||
|
"nom",
|
||||||
"panic-halt",
|
"panic-halt",
|
||||||
"panic-itm",
|
"panic-itm",
|
||||||
"scpi",
|
"scpi",
|
||||||
"smoltcp",
|
"smoltcp",
|
||||||
"stm32h7xx-hal",
|
"stm32h7xx-hal",
|
||||||
|
"uom 0.29.0",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -425,6 +427,12 @@ version = "0.1.8"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "7ffc5c5338469d4d3ea17d269fa8ea3512ad247247c30bd2df69e68309ed0a08"
|
checksum = "7ffc5c5338469d4d3ea17d269fa8ea3512ad247247c30bd2df69e68309ed0a08"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "memchr"
|
||||||
|
version = "2.3.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "3728d817d99e5ac407411fa471ff9800a778d88a24685968b36824eaf4bee400"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "minimq"
|
name = "minimq"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
@ -458,6 +466,16 @@ version = "0.4.0"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "2178127478ae4ee9be7180bc9c3bffb6354dd7238400db567102f98c413a9f35"
|
checksum = "2178127478ae4ee9be7180bc9c3bffb6354dd7238400db567102f98c413a9f35"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "nom"
|
||||||
|
version = "5.1.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "ffb4262d26ed83a1c0a33a38fe2bb15797329c85770da05e6b828ddb782627af"
|
||||||
|
dependencies = [
|
||||||
|
"memchr",
|
||||||
|
"version_check",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "num-integer"
|
name = "num-integer"
|
||||||
version = "0.1.43"
|
version = "0.1.43"
|
||||||
@ -601,7 +619,7 @@ dependencies = [
|
|||||||
"lexical-core",
|
"lexical-core",
|
||||||
"libm",
|
"libm",
|
||||||
"scpi_derive",
|
"scpi_derive",
|
||||||
"uom",
|
"uom 0.28.0",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -780,6 +798,16 @@ dependencies = [
|
|||||||
"typenum",
|
"typenum",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "uom"
|
||||||
|
version = "0.29.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8bb593f5252356bfb829112f8fca2d0982d48588d2d6bb5a92553b0dfc4c9aba"
|
||||||
|
dependencies = [
|
||||||
|
"num-traits",
|
||||||
|
"typenum",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "url"
|
name = "url"
|
||||||
version = "2.1.1"
|
version = "2.1.1"
|
||||||
|
16
Cargo.toml
16
Cargo.toml
@ -18,6 +18,7 @@ embedded-nal = "0.1.0"
|
|||||||
minimq = { git = "https://github.com/quartiq/minimq.git", branch = "master" }
|
minimq = { git = "https://github.com/quartiq/minimq.git", branch = "master" }
|
||||||
heapless = "0.5.5"
|
heapless = "0.5.5"
|
||||||
arrayvec = { version = "0.5.1", default-features = false, features = [ "array-sizes-33-128", "array-sizes-129-255" ] }
|
arrayvec = { version = "0.5.1", default-features = false, features = [ "array-sizes-33-128", "array-sizes-129-255" ] }
|
||||||
|
nom = { version = "5.1.2", default-features = false, features = [] }
|
||||||
|
|
||||||
# Logging and Panicking
|
# Logging and Panicking
|
||||||
panic-itm = "0.4.1"
|
panic-itm = "0.4.1"
|
||||||
@ -32,17 +33,10 @@ branch = "issue-4"
|
|||||||
default-features = false
|
default-features = false
|
||||||
features = [ "build-info", "unit-frequency", "unit-angle" ]
|
features = [ "build-info", "unit-frequency", "unit-angle" ]
|
||||||
|
|
||||||
# [dependencies.uom]
|
[dependencies.uom]
|
||||||
# version = "0.29.0"
|
version = "0.29.0"
|
||||||
# default-features = false
|
default-features = false
|
||||||
# features = [
|
features = [ "autoconvert", "f32", "f64", "si" ]
|
||||||
# "autoconvert",
|
|
||||||
# "usize", "u8", "u16", "u32", "u64",
|
|
||||||
# "isize", "i8", "i16", "i32", "i64",
|
|
||||||
# "f32", "f64",
|
|
||||||
# "si",
|
|
||||||
# "try-from"
|
|
||||||
# ]
|
|
||||||
|
|
||||||
# Use below SCPI dependency when need to modify SCPI fork offline
|
# Use below SCPI dependency when need to modify SCPI fork offline
|
||||||
# [dependencies.scpi]
|
# [dependencies.scpi]
|
||||||
|
22
src/lib.rs
22
src/lib.rs
@ -43,10 +43,11 @@ pub enum Error<E> {
|
|||||||
DDSCLKError,
|
DDSCLKError,
|
||||||
DDSRAMError,
|
DDSRAMError,
|
||||||
ParameterError,
|
ParameterError,
|
||||||
|
MqttTopicError,
|
||||||
MqttCommandError,
|
MqttCommandError,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug, Clone)]
|
||||||
pub enum ClockSource {
|
pub enum ClockSource {
|
||||||
OSC,
|
OSC,
|
||||||
SMA,
|
SMA,
|
||||||
@ -229,28 +230,47 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn set_channel_attenuation(&mut self, channel: u8, attenuation: f32) -> Result<(), Error<E>> {
|
fn set_channel_attenuation(&mut self, channel: u8, attenuation: f32) -> Result<(), Error<E>> {
|
||||||
|
if channel >= 4 || attenuation < 0.0 || attenuation > 31.5 {
|
||||||
|
return Err(Error::ParameterError);
|
||||||
|
}
|
||||||
self.attenuator.set_channel_attenuation(channel, attenuation)
|
self.attenuator.set_channel_attenuation(channel, attenuation)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn set_profile(&mut self, profile: u8) -> Result<(), Error<E>> {
|
fn set_profile(&mut self, profile: u8) -> Result<(), Error<E>> {
|
||||||
|
if profile >= 8 {
|
||||||
|
return Err(Error::ParameterError);
|
||||||
|
}
|
||||||
self.config_register.set_configurations(&mut [
|
self.config_register.set_configurations(&mut [
|
||||||
(CFGMask::PROFILE, profile.into())
|
(CFGMask::PROFILE, profile.into())
|
||||||
]).map(|_| ())
|
]).map(|_| ())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn set_channel_single_tone_profile(&mut self, channel: u8, profile: u8, frequency: f64, phase: f64, amplitude: f64) -> Result<(), Error<E>> {
|
fn set_channel_single_tone_profile(&mut self, channel: u8, profile: u8, frequency: f64, phase: f64, amplitude: f64) -> Result<(), Error<E>> {
|
||||||
|
if channel >= 4 || profile >= 8 || frequency < 0.0 || phase >= 360.0 ||
|
||||||
|
phase < 0.0 || amplitude < 0.0 || amplitude > 1.0 {
|
||||||
|
return Err(Error::ParameterError);
|
||||||
|
}
|
||||||
self.dds[usize::from(channel)].set_single_tone_profile(profile, frequency, phase, amplitude)
|
self.dds[usize::from(channel)].set_single_tone_profile(profile, frequency, phase, amplitude)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn set_channel_single_tone_profile_frequency(&mut self, channel: u8, profile: u8, frequency: f64)-> Result<(), Error<E>> {
|
fn set_channel_single_tone_profile_frequency(&mut self, channel: u8, profile: u8, frequency: f64)-> Result<(), Error<E>> {
|
||||||
|
if channel >= 4 || profile >= 8 || frequency < 0.0 {
|
||||||
|
return Err(Error::ParameterError);
|
||||||
|
}
|
||||||
self.dds[usize::from(channel)].set_single_tone_profile_frequency(profile, frequency)
|
self.dds[usize::from(channel)].set_single_tone_profile_frequency(profile, frequency)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn set_channel_single_tone_profile_phase(&mut self, channel: u8, profile: u8, phase: f64)-> Result<(), Error<E>> {
|
fn set_channel_single_tone_profile_phase(&mut self, channel: u8, profile: u8, phase: f64)-> Result<(), Error<E>> {
|
||||||
|
if channel >= 4 || profile >= 8 || phase >= 360.0 || phase < 0.0 {
|
||||||
|
return Err(Error::ParameterError);
|
||||||
|
}
|
||||||
self.dds[usize::from(channel)].set_single_tone_profile_phase(profile, phase)
|
self.dds[usize::from(channel)].set_single_tone_profile_phase(profile, phase)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn set_channel_single_tone_profile_amplitude(&mut self, channel: u8, profile: u8, amplitude: f64)-> Result<(), Error<E>> {
|
fn set_channel_single_tone_profile_amplitude(&mut self, channel: u8, profile: u8, amplitude: f64)-> Result<(), Error<E>> {
|
||||||
|
if channel >= 4 || profile >= 8 || amplitude < 0.0 || amplitude > 1.0 {
|
||||||
|
return Err(Error::ParameterError);
|
||||||
|
}
|
||||||
self.dds[usize::from(channel)].set_single_tone_profile_amplitude(profile, amplitude)
|
self.dds[usize::from(channel)].set_single_tone_profile_amplitude(profile, amplitude)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
546
src/mqtt_mux.rs
546
src/mqtt_mux.rs
@ -1,4 +1,15 @@
|
|||||||
use log::info;
|
use log::info;
|
||||||
|
use nom::IResult;
|
||||||
|
use nom::combinator::{value, map, map_res, not, opt, all_consuming};
|
||||||
|
use nom::sequence::{terminated, preceded, pair, delimited, tuple};
|
||||||
|
use nom::bytes::complete::{take, tag, tag_no_case, take_while};
|
||||||
|
use nom::character::complete::digit1;
|
||||||
|
use nom::character::is_space;
|
||||||
|
use nom::branch::alt;
|
||||||
|
use nom::number::complete::{float, double};
|
||||||
|
|
||||||
|
use uom::si::f64::Frequency;
|
||||||
|
use uom::si::frequency::{hertz, kilohertz, megahertz, gigahertz};
|
||||||
|
|
||||||
use embedded_hal::blocking::spi::Transfer;
|
use embedded_hal::blocking::spi::Transfer;
|
||||||
use core::convert::TryInto;
|
use core::convert::TryInto;
|
||||||
@ -7,27 +18,29 @@ use crate::ClockSource::*;
|
|||||||
use crate::Urukul;
|
use crate::Urukul;
|
||||||
use crate::Error;
|
use crate::Error;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug, Clone)]
|
||||||
pub enum MqttCommandType {
|
pub enum MqttTopic {
|
||||||
// Urukul/Control/Clock/Source
|
Switch(u8),
|
||||||
ClockSource(UrukulClockSource),
|
Attenuation(u8),
|
||||||
// Urukul/Control/Clock/Division
|
Singletone(u8, u8),
|
||||||
ClockDivision(u8),
|
SingletoneFrequency(u8, u8),
|
||||||
// Urukul/Control/ChannelX/Switch
|
SingletoneAmplitude(u8, u8),
|
||||||
Switch(u8, bool),
|
SingletonePhase(u8, u8),
|
||||||
// Urukul/Control/ChannelX/Attenuation
|
Profile,
|
||||||
Attenuation(u8, f32),
|
|
||||||
// Urukul/Control/ChannelX/SystemClock
|
|
||||||
SystemClock(u8, f64),
|
|
||||||
// Urukul/Control/ChannelX/ProfileY/Frequency
|
|
||||||
SingleToneFrequency(u8, u8, f64),
|
|
||||||
// Urukul/Control/ChannelX/ProfileY/Amplitude
|
|
||||||
SingleToneAmplitude(u8, u8, f64),
|
|
||||||
// Urukul/Control/ChannelX/ProfileY/Phase
|
|
||||||
SingleTonePhase(u8, u8, f64),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
use crate::mqtt_mux::MqttCommandType::*;
|
// Prossible change: Make this enum public to all comm protocol (if any)
|
||||||
|
// Such that Urukul accepts the enum directly
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub enum MqttCommand {
|
||||||
|
Switch(u8, bool),
|
||||||
|
Attenuation(u8, f32),
|
||||||
|
Singletone(u8, u8, f64, f64, f64),
|
||||||
|
SingletoneFrequency(u8, u8, f64),
|
||||||
|
SingletoneAmplitude(u8, u8, f64),
|
||||||
|
SingletonePhase(u8, u8, f64),
|
||||||
|
Profile(u8)
|
||||||
|
}
|
||||||
|
|
||||||
pub struct MqttMux<SPI> {
|
pub struct MqttMux<SPI> {
|
||||||
urukul: Urukul<SPI>
|
urukul: Urukul<SPI>
|
||||||
@ -40,158 +53,363 @@ impl<SPI, E> MqttMux<SPI> where SPI: Transfer<u8, Error = E> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn handle_command(&mut self, topic: &str, message: &[u8]) -> Result<(), Error<E>> {
|
pub fn process_mqtt(&mut self, topic: &str, message: &[u8]) -> Result<(), Error<E>> {
|
||||||
let command = self.parse(topic, message)?;
|
let (_, header) = self.parse_header(topic)
|
||||||
|
.map_err(|_| Error::MqttTopicError)?;
|
||||||
|
info!("{:?}", header);
|
||||||
|
let (_, command) = self.parse_message(header, message)
|
||||||
|
.map_err(|_| Error::MqttCommandError)?;
|
||||||
|
info!("{:?}", command);
|
||||||
self.execute(command)
|
self.execute(command)
|
||||||
}
|
}
|
||||||
|
|
||||||
// MQTT command are not case tolerant
|
fn parse_header<'a>(&mut self, topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
// If the command differs by case, space or delimiter, it is a wrong command
|
preceded(
|
||||||
// A starting forward slash ("/") is acceptable, as per MQTT standard
|
alt((
|
||||||
// Topic should contain the appropriate command header
|
tag("Urukul/Control/"),
|
||||||
// Message should provide the parameter
|
tag("/Urukul/Control/")
|
||||||
fn parse(&mut self, topic: &str, message: &[u8]) -> Result<MqttCommandType, Error<E>> {
|
)),
|
||||||
let mut assigned_channel = false;
|
alt((
|
||||||
let mut assigned_profile = false;
|
switch,
|
||||||
let mut channel :u8 = 0;
|
attenuation,
|
||||||
let mut profile :u8 = 0;
|
singletone,
|
||||||
|
singletone_frequency,
|
||||||
|
singletone_amplitude,
|
||||||
|
singletone_phase,
|
||||||
|
profile // Note: Put profile at the end
|
||||||
|
))
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
// Verify that the topic must start with Urukul/Control/ or /Urukul/Control/
|
fn parse_message<'a>(&mut self, topic: MqttTopic, message: &'a [u8]) -> IResult<&'a [u8], MqttCommand> {
|
||||||
let mut header = topic.strip_prefix("/Urukul/Control/")
|
match topic {
|
||||||
.or_else(|| topic.strip_prefix("Urukul/Control/"))
|
MqttTopic::Switch(ch) => switch_message(ch, message),
|
||||||
.ok_or(Error::MqttCommandError)?;
|
MqttTopic::Attenuation(ch) => attenuation_message(ch, message),
|
||||||
|
MqttTopic::Singletone(ch, prof) => singletone_message(ch, prof, message),
|
||||||
loop {
|
MqttTopic::SingletoneFrequency(ch, prof) => singletone_frequency_message(ch, prof, message),
|
||||||
match header {
|
MqttTopic::SingletoneAmplitude(ch, prof) => singletone_amplitude_message(ch, prof, message),
|
||||||
// The topic has a channel subtopic
|
MqttTopic::SingletonePhase(ch, prof) => singletone_phase_message(ch, prof, message),
|
||||||
_ if header.starts_with("Channel") => {
|
MqttTopic::Profile => profile_message(message),
|
||||||
// MQTT command should only mention channel once appropriately
|
|
||||||
// Channel must be referred before profile,
|
|
||||||
// as a channel is broader than a profile
|
|
||||||
if assigned_channel || assigned_profile {
|
|
||||||
return Err(Error::MqttCommandError);
|
|
||||||
}
|
|
||||||
// Remove the "Channel" part of the subtopic
|
|
||||||
header = header.strip_prefix("Channel")
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
// Remove the channel number at the end of the subtopic
|
|
||||||
// But store the channel as a char, so it can be removed easily
|
|
||||||
let numeric_char :char = header.chars()
|
|
||||||
.next()
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
// Record the channel number
|
|
||||||
channel = numeric_char.to_digit(10)
|
|
||||||
.ok_or(Error::MqttCommandError)?
|
|
||||||
.try_into()
|
|
||||||
.unwrap();
|
|
||||||
assigned_channel = true;
|
|
||||||
header = header.strip_prefix(numeric_char)
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
// Remove forward slash ("/")
|
|
||||||
header = header.strip_prefix("/")
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
},
|
|
||||||
|
|
||||||
_ if header.starts_with("Profile") => {
|
|
||||||
// MQTT command should only mention profile once appropriately
|
|
||||||
if assigned_profile {
|
|
||||||
return Err(Error::MqttCommandError);
|
|
||||||
}
|
|
||||||
// Remove the "Profile" part of the subtopic
|
|
||||||
header = header.strip_prefix("Profile")
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
// Remove the profile number at the end of the subtopic
|
|
||||||
// But store the profile as a char, so it can be removed easily
|
|
||||||
let numeric_char :char = header.chars()
|
|
||||||
.next()
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
// Record the channel number
|
|
||||||
profile = numeric_char.to_digit(10)
|
|
||||||
.ok_or(Error::MqttCommandError)?
|
|
||||||
.try_into()
|
|
||||||
.unwrap();
|
|
||||||
assigned_profile = true;
|
|
||||||
header = header.strip_prefix(numeric_char)
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
// Remove forward slash ("/")
|
|
||||||
header = header.strip_prefix("/")
|
|
||||||
.ok_or(Error::MqttCommandError)?;
|
|
||||||
},
|
|
||||||
|
|
||||||
"Clock/Source" => {
|
|
||||||
// Clock/Source refers to the Urukul clock source
|
|
||||||
// It should be common for all channels and profiles
|
|
||||||
if assigned_channel || assigned_profile {
|
|
||||||
return Err(Error::MqttCommandError);
|
|
||||||
}
|
|
||||||
let source_string = core::str::from_utf8(message).unwrap();
|
|
||||||
|
|
||||||
return match source_string {
|
|
||||||
_ if source_string.eq_ignore_ascii_case("OSC") => {
|
|
||||||
Ok(ClockSource(OSC))
|
|
||||||
},
|
|
||||||
_ if source_string.eq_ignore_ascii_case("SMA") => {
|
|
||||||
Ok(ClockSource(SMA))
|
|
||||||
},
|
|
||||||
_ if source_string.eq_ignore_ascii_case("MMCX") => {
|
|
||||||
Ok(ClockSource(MMCX))
|
|
||||||
},
|
|
||||||
_ => Err(Error::MqttCommandError),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
"Clock/Division" => {
|
|
||||||
// Clock/Division refers to the Urukul clock division
|
|
||||||
// It should be common for all channels and profiles
|
|
||||||
if assigned_channel || assigned_profile {
|
|
||||||
return Err(Error::MqttCommandError);
|
|
||||||
}
|
|
||||||
|
|
||||||
let division = u8::from_str_radix(core::str::from_utf8(message).unwrap(), 10)
|
|
||||||
.map_or_else(
|
|
||||||
|_| Err(Error::MqttCommandError),
|
|
||||||
|div| if div == 1 || div == 2 || div == 4 {
|
|
||||||
Ok(div)
|
|
||||||
} else {
|
|
||||||
Err(Error::MqttCommandError)
|
|
||||||
})?;
|
|
||||||
return Ok(ClockDivision(division));
|
|
||||||
}
|
|
||||||
|
|
||||||
"Switch" => {
|
|
||||||
// Switch is a channel specific topic
|
|
||||||
if !(assigned_channel && !assigned_profile) {
|
|
||||||
return Err(Error::MqttCommandError);
|
|
||||||
}
|
|
||||||
|
|
||||||
let switch_string = core::str::from_utf8(message).unwrap();
|
|
||||||
|
|
||||||
return match switch_string {
|
|
||||||
_ if switch_string.eq_ignore_ascii_case("on") => {
|
|
||||||
Ok(Switch(channel, true))
|
|
||||||
},
|
|
||||||
_ if switch_string.eq_ignore_ascii_case("off") => {
|
|
||||||
Ok(Switch(channel, false))
|
|
||||||
},
|
|
||||||
_ => Err(Error::MqttCommandError),
|
|
||||||
};
|
|
||||||
},
|
|
||||||
|
|
||||||
// TODO: Cover all commands
|
|
||||||
|
|
||||||
_ => return Err(Error::MqttCommandError),
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: Implement this
|
fn execute(&mut self, command: MqttCommand) -> Result<(), Error<E>> {
|
||||||
// Only need to sort the command enum
|
match command {
|
||||||
// Obviously. This is what a MUX does
|
MqttCommand::Switch(ch, state) => self.urukul.set_channel_switch(ch.into(), state),
|
||||||
fn execute(&mut self, command_type: MqttCommandType) -> Result<(), Error<E>> {
|
MqttCommand::Attenuation(ch, ampl) => self.urukul.set_channel_attenuation(ch, ampl),
|
||||||
info!("{:?}", command_type);
|
MqttCommand::Singletone(ch, prof, freq, ampl, deg) => self.urukul.set_channel_single_tone_profile(ch, prof, freq, ampl, deg),
|
||||||
match command_type {
|
MqttCommand::SingletoneFrequency(ch, prof, freq) => self.urukul.set_channel_single_tone_profile_frequency(ch, prof, freq),
|
||||||
Switch(channel, status) => self.urukul.set_channel_switch(channel as u32, status),
|
MqttCommand::SingletoneAmplitude(ch, prof, ampl) => self.urukul.set_channel_single_tone_profile_amplitude(ch, prof, ampl),
|
||||||
_ => Ok(())
|
MqttCommand::SingletonePhase(ch, prof, deg) => self.urukul.set_channel_single_tone_profile_phase(ch, prof, deg),
|
||||||
|
MqttCommand::Profile(prof) => self.urukul.set_profile(prof),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Topic separator parser
|
||||||
|
fn topic_separator<'a>(topic: &'a str) -> IResult<&'a str, ()> {
|
||||||
|
value((), tag("/"))(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Message separator parser
|
||||||
|
fn message_separator(message: &[u8]) -> IResult<&[u8], ()> {
|
||||||
|
value(
|
||||||
|
(),
|
||||||
|
preceded(
|
||||||
|
whitespace,
|
||||||
|
preceded(
|
||||||
|
tag("/"),
|
||||||
|
whitespace
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Selection parsers
|
||||||
|
fn select_channel<'a>(topic: &'a str) -> IResult<&'a str, u8> {
|
||||||
|
terminated(
|
||||||
|
map_res(
|
||||||
|
preceded(
|
||||||
|
tag("Channel"),
|
||||||
|
digit1
|
||||||
|
),
|
||||||
|
|num: &str| u8::from_str_radix(num, 10)
|
||||||
|
),
|
||||||
|
topic_separator
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn select_profile<'a>(topic: &'a str) -> IResult<&'a str, u8> {
|
||||||
|
terminated(
|
||||||
|
map_res(
|
||||||
|
preceded(
|
||||||
|
tag("Profile"),
|
||||||
|
digit1
|
||||||
|
),
|
||||||
|
|num: &str| u8::from_str_radix(num, 10)
|
||||||
|
),
|
||||||
|
topic_separator
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Selection parser for Singletone
|
||||||
|
// Note: This fucntion assumes singletone is not the most specific sub-topic
|
||||||
|
fn select_singletone<'a>(topic: &'a str) -> IResult<&'a str, ()> {
|
||||||
|
preceded(
|
||||||
|
tag("Singletone"),
|
||||||
|
topic_separator
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn check_end_slice(message: &[u8]) -> IResult<&[u8], ()> {
|
||||||
|
not(
|
||||||
|
take(1_usize)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read whitespace
|
||||||
|
fn whitespace(message: &[u8]) -> IResult<&[u8], ()> {
|
||||||
|
value((), take_while(is_space))(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reader for uom instances
|
||||||
|
fn read_frequency(message: &[u8]) -> IResult<&[u8], f64> {
|
||||||
|
map(
|
||||||
|
pair(
|
||||||
|
double,
|
||||||
|
opt(
|
||||||
|
preceded(
|
||||||
|
whitespace,
|
||||||
|
alt((
|
||||||
|
value(1.0, tag_no_case("hz")),
|
||||||
|
value(1_000.0, tag_no_case("khz")),
|
||||||
|
value(1_000_000.0, tag_no_case("mhz")),
|
||||||
|
value(1_000_000_000.0, tag_no_case("ghz"))
|
||||||
|
))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
),
|
||||||
|
|(freq, unit): (f64, Option<f64>)| {
|
||||||
|
freq * unit.map_or(1.0, |mul| mul)
|
||||||
|
}
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Switch Command Topic
|
||||||
|
fn switch<'a>(topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
terminated(
|
||||||
|
select_channel,
|
||||||
|
tag("Switch")
|
||||||
|
),
|
||||||
|
|channel: u8| MqttTopic::Switch(channel)
|
||||||
|
)
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Switch Command Message
|
||||||
|
fn switch_message(channel: u8, message: &[u8]) -> IResult<&[u8], MqttCommand> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
alt((
|
||||||
|
value(true, tag("on")),
|
||||||
|
value(false, tag("off"))
|
||||||
|
)),
|
||||||
|
|switch| MqttCommand::Switch(channel, switch)
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Attenuation Command Topic
|
||||||
|
fn attenuation<'a>(topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
terminated(
|
||||||
|
select_channel,
|
||||||
|
tag("Attenuation")
|
||||||
|
),
|
||||||
|
|channel: u8| MqttTopic::Attenuation(channel)
|
||||||
|
)
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Attenuation Command Message
|
||||||
|
fn attenuation_message(channel: u8, message: &[u8]) -> IResult<&[u8], MqttCommand> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
terminated(
|
||||||
|
float,
|
||||||
|
opt(
|
||||||
|
preceded(
|
||||||
|
whitespace,
|
||||||
|
tag_no_case("db")
|
||||||
|
)
|
||||||
|
)
|
||||||
|
),
|
||||||
|
|att: f32| MqttCommand::Attenuation(channel, att)
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Singletone frequenct Command Topic
|
||||||
|
fn singletone_frequency<'a>(topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
pair(
|
||||||
|
select_channel,
|
||||||
|
terminated(
|
||||||
|
select_profile,
|
||||||
|
preceded(
|
||||||
|
select_singletone,
|
||||||
|
tag("Frequency"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
),
|
||||||
|
|(channel, profile): (u8, u8)| MqttTopic::SingletoneFrequency(channel, profile)
|
||||||
|
)
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Singletone frequency Command Message
|
||||||
|
fn singletone_frequency_message(channel: u8, profile: u8, message: &[u8]) -> IResult<&[u8], MqttCommand> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
read_frequency,
|
||||||
|
|freq: f64| MqttCommand::SingletoneFrequency(channel, profile, freq)
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Singletone Amplitude Command Topic
|
||||||
|
fn singletone_amplitude<'a>(topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
pair(
|
||||||
|
select_channel,
|
||||||
|
terminated(
|
||||||
|
select_profile,
|
||||||
|
preceded(
|
||||||
|
select_singletone,
|
||||||
|
tag("Amplitude"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
),
|
||||||
|
|(channel, profile): (u8, u8)| MqttTopic::SingletoneAmplitude(channel, profile)
|
||||||
|
)
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Singletone AMplitude Command Message
|
||||||
|
fn singletone_amplitude_message(channel: u8, profile: u8, message: &[u8]) -> IResult<&[u8], MqttCommand> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
double,
|
||||||
|
|ampl: f64| MqttCommand::SingletoneAmplitude(channel, profile, ampl)
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Phase Command Topic
|
||||||
|
fn singletone_phase<'a>(topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
pair(
|
||||||
|
select_channel,
|
||||||
|
terminated(
|
||||||
|
select_profile,
|
||||||
|
preceded(
|
||||||
|
select_singletone,
|
||||||
|
tag("Phase"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
),
|
||||||
|
|(channel, profile): (u8, u8)| MqttTopic::SingletonePhase(channel, profile)
|
||||||
|
)
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for Phase Command Message
|
||||||
|
fn singletone_phase_message(channel: u8, profile: u8, message: &[u8]) -> IResult<&[u8], MqttCommand> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
terminated(
|
||||||
|
double,
|
||||||
|
opt(
|
||||||
|
preceded(
|
||||||
|
whitespace,
|
||||||
|
tag_no_case("deg")
|
||||||
|
)
|
||||||
|
)
|
||||||
|
),
|
||||||
|
|deg: f64| MqttCommand::SingletonePhase(channel, profile, deg)
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for one-command singletone profile Topic
|
||||||
|
fn singletone<'a>(topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
pair(
|
||||||
|
select_channel,
|
||||||
|
terminated(
|
||||||
|
select_profile,
|
||||||
|
tag("Singletone")
|
||||||
|
)
|
||||||
|
),
|
||||||
|
|(channel, profile): (u8, u8)| MqttTopic::Singletone(channel, profile)
|
||||||
|
)
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parser for one-command singletone profile Command
|
||||||
|
fn singletone_message(channel: u8, profile: u8, message: &[u8]) -> IResult<&[u8], MqttCommand> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
tuple((
|
||||||
|
read_frequency,
|
||||||
|
preceded(
|
||||||
|
message_separator,
|
||||||
|
double
|
||||||
|
),
|
||||||
|
preceded(
|
||||||
|
message_separator,
|
||||||
|
terminated(
|
||||||
|
double,
|
||||||
|
opt(
|
||||||
|
preceded(
|
||||||
|
whitespace,
|
||||||
|
tag_no_case("deg")
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
)),
|
||||||
|
|(freq, ampl, phase): (f64, f64, f64)| MqttCommand::Singletone(channel, profile, freq, ampl, phase)
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn profile<'a>(topic: &'a str) -> IResult<&'a str, MqttTopic> {
|
||||||
|
all_consuming(
|
||||||
|
value(
|
||||||
|
MqttTopic::Profile,
|
||||||
|
tag("Profile")
|
||||||
|
)
|
||||||
|
)(topic)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn profile_message(message: &[u8]) -> IResult<&[u8], MqttCommand> {
|
||||||
|
all_consuming(
|
||||||
|
map(
|
||||||
|
digit1,
|
||||||
|
|num: &[u8]| {
|
||||||
|
MqttCommand::Profile(
|
||||||
|
u8::from_str_radix(core::str::from_utf8(num).unwrap(), 10).unwrap()
|
||||||
|
)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
)(message)
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user