forked from M-Labs/artiq-zynq
107 lines
3.1 KiB
Rust
107 lines
3.1 KiB
Rust
use libconfig::Config;
|
|
#[cfg(has_drtio_routing)]
|
|
use crate::pl::csr;
|
|
use core::fmt;
|
|
|
|
use log::{warn, info};
|
|
|
|
#[cfg(has_drtio_routing)]
|
|
pub const DEST_COUNT: usize = 256;
|
|
#[cfg(not(has_drtio_routing))]
|
|
pub const DEST_COUNT: usize = 0;
|
|
pub const MAX_HOPS: usize = 32;
|
|
pub const INVALID_HOP: u8 = 0xff;
|
|
|
|
pub struct RoutingTable(pub [[u8; MAX_HOPS]; DEST_COUNT]);
|
|
|
|
impl RoutingTable {
|
|
// default routing table is for star topology with no repeaters
|
|
pub fn default_master(default_n_links: usize) -> RoutingTable {
|
|
let mut ret = RoutingTable([[INVALID_HOP; MAX_HOPS]; DEST_COUNT]);
|
|
let n_entries = default_n_links + 1; // include local RTIO
|
|
for i in 0..n_entries {
|
|
ret.0[i][0] = i as u8;
|
|
}
|
|
for i in 1..n_entries {
|
|
ret.0[i][1] = 0x00;
|
|
}
|
|
ret
|
|
}
|
|
|
|
// use this by default on satellite, as they receive
|
|
// the routing table from the master
|
|
pub fn default_empty() -> RoutingTable {
|
|
RoutingTable([[INVALID_HOP; MAX_HOPS]; DEST_COUNT])
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for RoutingTable {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "RoutingTable {{")?;
|
|
for i in 0..DEST_COUNT {
|
|
if self.0[i][0] != INVALID_HOP {
|
|
write!(f, " {}:", i)?;
|
|
for j in 0..MAX_HOPS {
|
|
if self.0[i][j] == INVALID_HOP {
|
|
break;
|
|
}
|
|
write!(f, " {}", self.0[i][j])?;
|
|
}
|
|
write!(f, ";")?;
|
|
}
|
|
}
|
|
write!(f, " }}")?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
pub fn config_routing_table(default_n_links: usize, cfg: &Config) -> RoutingTable {
|
|
let mut ret = RoutingTable::default_master(default_n_links);
|
|
if let Ok(data) = cfg.read("routing_table") {
|
|
if data.len() == DEST_COUNT*MAX_HOPS {
|
|
for i in 0..DEST_COUNT {
|
|
for j in 0..MAX_HOPS {
|
|
ret.0[i][j] = data[i*MAX_HOPS+j];
|
|
}
|
|
}
|
|
} else {
|
|
warn!("length of the configured routing table is incorrect, using default");
|
|
}
|
|
} else {
|
|
info!("could not read routing table from configuration, using default");
|
|
}
|
|
info!("routing table: {}", ret);
|
|
ret
|
|
}
|
|
|
|
#[cfg(has_drtio_routing)]
|
|
pub fn interconnect_enable(routing_table: &RoutingTable, rank: u8, destination: u8) {
|
|
let hop = routing_table.0[destination as usize][rank as usize];
|
|
unsafe {
|
|
csr::routing_table::destination_write(destination);
|
|
csr::routing_table::hop_write(hop);
|
|
}
|
|
}
|
|
|
|
#[cfg(has_drtio_routing)]
|
|
pub fn interconnect_disable(destination: u8) {
|
|
unsafe {
|
|
csr::routing_table::destination_write(destination);
|
|
csr::routing_table::hop_write(INVALID_HOP);
|
|
}
|
|
}
|
|
|
|
#[cfg(has_drtio_routing)]
|
|
pub fn interconnect_enable_all(routing_table: &RoutingTable, rank: u8) {
|
|
for i in 0..DEST_COUNT {
|
|
interconnect_enable(routing_table, rank, i as u8);
|
|
}
|
|
}
|
|
|
|
#[cfg(has_drtio_routing)]
|
|
pub fn interconnect_disable_all() {
|
|
for i in 0..DEST_COUNT {
|
|
interconnect_disable(i as u8);
|
|
}
|
|
}
|