Struct tract_pulse::internal::Registry

pub struct Registry {
    pub id: Identifier,
    pub docstrings: Option<Vec<String, Global>>,
    pub aliases: Vec<Identifier, Global>,
    pub fragments: HashMap<Identifier, FragmentDef, RandomState>,
    pub primitives: HashMap<Identifier, PrimitiveDecl, RandomState>,
    pub unit_element_wise_ops: Vec<(Identifier, Box<dyn ElementWiseMiniOp + 'static, Global>), Global>,
    pub element_wise_ops: Vec<(Identifier, TypeId, fn(_: &mut IntoAst<'_>, _: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error>, Vec<Parameter, Global>, fn(_: &mut ModelBuilder<'_>, _: &ResolvedInvocation<'_>) -> Result<Value, Error>), Global>,
    pub binary_ops: Vec<(Identifier, Box<dyn BinMiniOp + 'static, Global>), Global>,
    pub from_tract: HashMap<TypeId, fn(_: &mut IntoAst<'_>, _: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error>, RandomState>,
    pub extensions: Vec<Box<dyn Fn(&mut ModelBuilder<'_>, &[Identifier]) -> Result<ControlFlow<(), ()>, Error> + Sync + Send + 'static, Global>, Global>,
}

Fields

id: Identifier

docstrings: Option<Vec<String, Global>>

aliases: Vec<Identifier, Global>

fragments: HashMap<Identifier, FragmentDef, RandomState>

primitives: HashMap<Identifier, PrimitiveDecl, RandomState>

unit_element_wise_ops: Vec<(Identifier, Box<dyn ElementWiseMiniOp + 'static, Global>), Global>

element_wise_ops: Vec<(Identifier, TypeId, fn(_: &mut IntoAst<'_>, _: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error>, Vec<Parameter, Global>, fn(_: &mut ModelBuilder<'_>, _: &ResolvedInvocation<'_>) -> Result<Value, Error>), Global>

binary_ops: Vec<(Identifier, Box<dyn BinMiniOp + 'static, Global>), Global>

from_tract: HashMap<TypeId, fn(_: &mut IntoAst<'_>, _: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error>, RandomState>

extensions: Vec<Box<dyn Fn(&mut ModelBuilder<'_>, &[Identifier]) -> Result<ControlFlow<(), ()>, Error> + Sync + Send + 'static, Global>, Global>

Implementations

impl Registry

pub fn new(id: impl AsRef<str>) -> Registry

pub fn with_doc(self, docstring: impl Into<String>) -> Registry

pub fn register_dumper( &mut self, id: TypeId, func: fn(_: &mut IntoAst<'_>, _: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error> )

pub fn register_primitive( &mut self, id: impl AsRef<str>, params: &[Parameter], results: &[impl Into<Result_> + Clone], func: fn(_: &mut ModelBuilder<'_>, _: &ResolvedInvocation<'_>) -> Result<Value, Error> ) -> &mut PrimitiveDecl

pub fn register_fragment(&mut self, def: FragmentDef)

pub fn register_unit_element_wise( &mut self, id: impl AsRef<str>, ew: &(dyn ElementWiseMiniOp + 'static) )

pub fn register_element_wise( &mut self, id: impl AsRef<str>, type_id: TypeId, dumper: fn(_: &mut IntoAst<'_>, _: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error>, parameters: Vec<Parameter, Global>, loader: fn(_: &mut ModelBuilder<'_>, _: &ResolvedInvocation<'_>) -> Result<Value, Error> )

pub fn register_binary( &mut self, id: impl AsRef<str>, op: &(dyn BinMiniOp + 'static) )

pub fn serialize( &self, ast: &mut IntoAst<'_>, node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>> ) -> Result<Option<Arc<RValue>>, Error>

pub fn deserialize( &self, builder: &mut ModelBuilder<'_>, invocation: &Invocation, dt: &[Option<DatumType>] ) -> Result<Option<Value>, Error>