use thiserror::Error; use crate::params; use crate::node; /// Error types for trait Network #[derive(Error, Debug)] pub enum NetworkError { #[error("Error during node insertion")] NodeInsertionError(String) } ///Network ///The Network trait define the required methods for a structure used as pgm (such as ctbn). pub trait Network { fn initialize_adj_matrix(&mut self); fn add_node(&mut self, n: node::Node) -> Result; fn add_edge(&mut self, parent: usize, child: usize); ///Get all the indices of the nodes contained inside the network fn get_node_indices(&self) -> std::ops::Range; fn get_node(&self, node_idx: usize) -> &node::Node; ///Compute the index that must be used to access the parameter of a node given a specific ///configuration of the network. Usually, the only values really used in *current_state* are ///the ones in the parent set of the *node*. fn get_param_index_network(&self, node: usize, current_state: &Vec) -> usize; fn get_parent_set(&self, node: usize) -> Vec; fn get_children_set(&self, node: usize) -> Vec; }