use melodium_core::*;
use melodium_macro::{check, mel_function, mel_treatment};
#[mel_function]
pub fn equal(a: i16, b: i16) -> bool {
a == b
}
#[mel_function]
pub fn not_equal(a: i16, b: i16) -> bool {
a != b
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output result Stream<bool>
)]
pub async fn equal() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(result.send_one_bool(a == b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output result Stream<bool>
)]
pub async fn not_equal() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(result.send_one_bool(a != b).await)
}
}
#[mel_function]
pub fn pow(base: i16, exponent: i16) -> i16 {
base.pow(exponent as u32)
}
#[mel_function]
pub fn ge(a: i16, b: i16) -> bool {
a >= b
}
#[mel_function]
pub fn le(a: i16, b: i16) -> bool {
a <= b
}
#[mel_treatment(
input base Stream<i16>
input exponent Stream<i16>
output power Stream<i16>
)]
pub async fn pow() {
while let (Ok(base), Ok(exp)) = (base.recv_one_i16().await, exponent.recv_one_i16().await) {
check!(power.send_one_i16(base.pow(exp as u32)).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output is Stream<bool>
)]
pub async fn lower_equal() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(is.send_one_bool(a <= b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output is Stream<bool>
)]
pub async fn greater_equal() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(is.send_one_bool(a >= b).await)
}
}
#[mel_function]
pub fn add(a: i16, b: i16) -> i16 {
a + b
}
#[mel_function]
pub fn div(dividend: i16, divisor: i16) -> i16 {
dividend / divisor
}
#[mel_function]
pub fn mult(a: i16, b: i16) -> i16 {
a * b
}
#[mel_function]
pub fn rem(dividend: i16, divisor: i16) -> i16 {
dividend % divisor
}
#[mel_function]
pub fn sub(a: i16, b: i16) -> i16 {
a - b
}
#[mel_function]
pub fn gt(a: i16, b: i16) -> bool {
a > b
}
#[mel_function]
pub fn lt(a: i16, b: i16) -> bool {
a < b
}
#[mel_function]
pub fn min(a: i16, b: i16) -> i16 {
a.min(b)
}
#[mel_function]
pub fn max(a: i16, b: i16) -> i16 {
a.max(b)
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output sum Stream<i16>
)]
pub async fn add() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(sum.send_one_i16(a + b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output quotient Stream<i16>
)]
pub async fn div() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(quotient.send_one_i16(a / b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output product Stream<i16>
)]
pub async fn mult() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(product.send_one_i16(a * b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output remainder Stream<i16>
)]
pub async fn rem() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(remainder.send_one_i16(a % b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output diff Stream<i16>
)]
pub async fn sub() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(diff.send_one_i16(a - b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output min Stream<i16>
)]
pub async fn min() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(min.send_one_i16(a.min(b)).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output max Stream<i16>
)]
pub async fn max() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(max.send_one_i16(a.max(b)).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output is Stream<bool>
)]
pub async fn lower_than() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(is.send_one_bool(a < b).await)
}
}
#[mel_treatment(
input a Stream<i16>
input b Stream<i16>
output is Stream<bool>
)]
pub async fn greater_than() {
while let (Ok(a), Ok(b)) = (a.recv_one_i16().await, b.recv_one_i16().await) {
check!(is.send_one_bool(a > b).await)
}
}
#[mel_function]
pub fn abs(value: i16) -> i16 {
value.abs()
}
#[mel_treatment(
input value Stream<i16>
output abs Stream<i16>
)]
pub async fn abs() {
while let Ok(values) = value.recv_i16().await {
check!(
abs.send_i16(values.into_iter().map(|v| v.abs()).collect())
.await
)
}
}