Struct tract_pulse::internal::Registry

pub struct Registry {
    pub id: String,
    pub fragments: HashMap<String, FragmentDef, RandomState>,
    pub primitives: HashMap<String, (Vec<Parameter, Global>, fn(&mut ModelBuilder<'_>, &ResolvedInvocation<'_>) -> Result<SmallVec<[OutletId; 4]>, Error>), RandomState>,
    pub unit_element_wise_ops: Vec<(String, Box<dyn ElementWiseMiniOp + 'static, Global>), Global>,
    pub element_wise_ops: Vec<(String, TypeId, fn(&mut IntoAst<'_>, &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error>, Vec<Parameter, Global>, fn(&mut ModelBuilder<'_>, &ResolvedInvocation<'_>) -> Result<SmallVec<[OutletId; 4]>, Error>), Global>,
    pub binary_ops: Vec<(String, Box<dyn BinMiniOp + 'static, Global>, Option<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>,
}

Fields

id: String

fragments: HashMap<String, FragmentDef, RandomState>

primitives: HashMap<String, (Vec<Parameter, Global>, fn(&mut ModelBuilder<'_>, &ResolvedInvocation<'_>) -> Result<SmallVec<[OutletId; 4]>, Error>), RandomState>

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

element_wise_ops: Vec<(String, TypeId, fn(&mut IntoAst<'_>, &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>) -> Result<Option<Arc<RValue>>, Error>, Vec<Parameter, Global>, fn(&mut ModelBuilder<'_>, &ResolvedInvocation<'_>) -> Result<SmallVec<[OutletId; 4]>, Error>), Global>

binary_ops: Vec<(String, Box<dyn BinMiniOp + 'static, Global>, Option<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>

Implementations

impl Registry

pub fn new(id: 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: &str,
    decl: &[Parameter],
    func: fn(&mut ModelBuilder<'_>, &ResolvedInvocation<'_>) -> Result<SmallVec<[OutletId; 4]>, Error>
)

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

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

pub fn register_element_wise(
    &mut self,
    id: impl Into<String>,
    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<SmallVec<[OutletId; 4]>, Error>
)

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

pub fn register_binary_with_flipped(
    &mut self,
    id: impl Into<String>,
    op: &(dyn BinMiniOp + 'static),
    flipped: &(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
) -> Result<Option<Value>, Error>