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

#![no_std]

extern crate alloc;

mod descriptor;

pub use hotg_rune_core::{HasOutputs, Tensor};
pub use hotg_rune_proc_block_macros::ProcBlock;
pub use descriptor::*;

/// Process some data, transforming it from one form to another.
pub trait Transform<Input>: ProcBlock {
    type Output;

    fn transform(&mut self, input: Input) -> Self::Output;
}

/// The base trait that all proc blocks must implement.
///
/// This trait must be implemented using the
/// [`hotg_rune_proc_block_macros::ProcBlock`] custom derive.
pub trait ProcBlock: Default + 'static {
    /// A description of the proc block.
    const DESCRIPTOR: ProcBlockDescriptor<'static>;
}

/// An internal module used give the [`hotg_rune_proc_block_macros`] crate access to
/// all the types it will need.
#[doc(hidden)]
pub mod internal {
    pub use crate::{ProcBlock, Transform, descriptor::*};
    pub use alloc::borrow::Cow;
    pub use hotg_rune_core::{reflect::Type, Tensor};
}