nextpnr/common/route/awooter/rust/src/lib.rs

285 lines
8.1 KiB
Rust

#![feature(c_unwind)]
use std::{ptr::NonNull, time::Instant};
use colored::Colorize;
use indicatif::MultiProgress;
use rayon::prelude::{IntoParallelRefIterator, ParallelIterator};
use crate::partition::Coord;
#[macro_use]
mod npnr;
mod partition;
mod route;
#[no_mangle]
pub extern "C-unwind" fn npnr_router_awooter(
ctx: Option<NonNull<npnr::Context>>,
pressure: f32,
history: f32,
) -> bool {
let ctx: &mut npnr::Context = unsafe { ctx.expect("non-null context").as_mut() };
route(ctx, pressure, history)
/*std::panic::catch_unwind(move || {
let ctx: &mut npnr::Context = unsafe { ctx.expect("non-null context").as_mut() };
route(ctx)
})
.unwrap_or_else(|x| {
if let Ok(x) = x.downcast::<String>() {
log_error!("caught panic: {}", x);
}
false
})*/
}
fn extract_arcs_from_nets(ctx: &npnr::Context, nets: &npnr::Nets) -> Vec<route::Arc> {
let mut arcs = vec![];
for (name, net) in nets.to_vec().iter() {
let net = unsafe { net.as_mut().unwrap() };
let str = ctx.name_of(**name).to_str().unwrap().to_string();
let verbose = false; //str == "soc0.processor.with_fpu.fpu_0.fpu_multiply_0.rin_CCU2C_S0_4$CCU2_FCI_INT";
if verbose {
dbg!(str, net.is_global());
}
if net.is_global() {
continue;
}
let port_ref = net.driver();
let port_ref = unsafe { port_ref.as_ref().unwrap() };
if let Some(cell) = port_ref.cell() {
let source = cell.location();
let source_wire = ctx.source_wire(net);
for sink_ref in nets.users_by_name(**name).unwrap().iter() {
let sink = sink_ref.cell().unwrap();
let sink = sink.location();
for sink_wire in ctx.sink_wires(net, *sink_ref) {
arcs.push(route::Arc::new(
source_wire,
source,
sink_wire,
sink,
net.index(),
));
if verbose {
let source_wire = ctx.name_of_wire(source_wire).to_str().unwrap();
let sink_wire = ctx.name_of_wire(sink_wire).to_str().unwrap();
dbg!(source_wire, sink_wire, net.index().into_inner());
}
}
}
}
}
arcs
}
fn route(ctx: &mut npnr::Context, pressure: f32, history: f32) -> bool {
log_info!(
"{}{}{}{}{}{} from Rust!\n",
"A".red(),
"w".green(),
"o".yellow(),
"o".blue(),
"o".magenta(),
"o".cyan()
);
log_info!(
"Running on a {}x{} grid\n",
ctx.grid_dim_x().to_string().bold(),
ctx.grid_dim_y().to_string().bold(),
);
let wires = ctx.wires_leaking();
log_info!("Found {} wires\n", wires.len().to_string().bold());
let pips = ctx.pips_leaking();
log_info!("Found {} pips\n", pips.len().to_string().bold());
let nets = npnr::Nets::new(ctx);
let nets_str = nets.len().to_string();
log_info!("Found {} nets\n", nets_str.bold());
let mut count = 0;
for (&name, net) in nets.to_vec().iter() {
let _src = ctx.source_wire(**net);
let net = unsafe { net.as_mut().unwrap() };
let users = nets.users_by_name(name).unwrap().iter();
for user in users {
count += ctx.sink_wires(net, *user).len();
}
}
log_info!("Found {} arcs\n", count.to_string().bold());
let binding = nets.to_vec();
let (name, net) = binding
.iter()
.max_by_key(|(name, net)| {
let net = unsafe { net.as_mut().unwrap() };
if net.is_global() {
0
} else {
nets.users_by_name(**name)
.unwrap()
.iter()
.fold(0, |acc, sink| acc + ctx.sink_wires(net, *sink).len())
}
})
.unwrap();
let net = unsafe { net.as_mut().unwrap() };
let count = nets
.users_by_name(**name)
.unwrap()
.iter()
.fold(0, |acc, sink| acc + ctx.sink_wires(net, *sink).len())
.to_string();
log_info!(
"Highest non-global fanout net is {}\n",
ctx.name_of(**name).to_str().unwrap().bold()
);
log_info!(" with {} arcs\n", count.bold());
let mut x0 = 0;
let mut y0 = 0;
let mut x1 = 0;
let mut y1 = 0;
for sink in nets.users_by_name(**name).unwrap().iter() {
let cell = sink.cell().unwrap().location();
x0 = x0.min(cell.x);
y0 = y0.min(cell.y);
x1 = x1.max(cell.x);
y1 = y1.max(cell.y);
}
let coords_min = format!("({}, {})", x0, y0);
let coords_max = format!("({}, {})", x1, y1);
log_info!(
" which spans from {} to {}\n",
coords_min.bold(),
coords_max.bold()
);
log_info!(
"rayon reports {} threads available\n",
rayon::current_num_threads().to_string().bold()
);
let start = Instant::now();
let arcs = extract_arcs_from_nets(ctx, &nets);
let mut special_arcs = vec![];
let mut partitionable_arcs = Vec::with_capacity(arcs.len());
for arc in arcs {
let src_name = ctx.name_of_wire(arc.get_source_wire()).to_str().unwrap();
let dst_name = ctx.name_of_wire(arc.get_sink_wire()).to_str().unwrap();
if src_name.contains("FCO_SLICE")
|| src_name.contains('J')
|| src_name.contains("DDR")
|| dst_name.contains("DDR")
{
special_arcs.push(arc);
} else {
partitionable_arcs.push(arc);
}
}
log_info!(
" {} arcs special-cased\n",
special_arcs.len().to_string().bold()
);
let (x_part, y_part, ne, se, sw, nw) = partition::find_partition_point_and_sanity_check(
ctx,
&nets,
&partitionable_arcs[..],
pips,
0,
ctx.grid_dim_x(),
0,
ctx.grid_dim_y(),
);
let time = format!("{:.2}", (Instant::now() - start).as_secs_f32());
log_info!("Partitioning took {}s\n", time.bold());
log_info!(
"Using pressure factor {} and history factor {}\n",
pressure,
history
);
let start = Instant::now();
log_info!("Routing partitioned arcs\n");
let progress = MultiProgress::new();
let partitions = [
(
Coord::new(0, 0),
Coord::new(x_part + 1, y_part + 1),
&ne,
"NE",
),
(
Coord::new(x_part - 1, 0),
Coord::new(ctx.grid_dim_x(), y_part + 1),
&se,
"SE",
),
(
Coord::new(x_part - 1, y_part - 1),
Coord::new(ctx.grid_dim_x(), ctx.grid_dim_y()),
&sw,
"SW",
),
(
Coord::new(0, y_part - 1),
Coord::new(x_part + 1, ctx.grid_dim_y()),
&nw,
"NW",
),
];
partitions
.par_iter()
.for_each(|(box_ne, box_sw, arcs, id)| {
let mut router = route::Router::new(*box_ne, *box_sw, pressure, history);
router.route(ctx, &nets, wires, arcs, &progress, id);
});
log_info!("Routing miscellaneous arcs\n");
let mut router = route::Router::new(
Coord::new(0, 0),
Coord::new(ctx.grid_dim_x(), ctx.grid_dim_y()),
pressure,
history,
);
router.route(ctx, &nets, wires, &special_arcs, &progress, "MISC");
let time = format!("{:.2}", (Instant::now() - start).as_secs_f32());
log_info!("Routing took {}s\n", time.bold());
//let mut router = route::Router::new(Coord::new(0, 0), Coord::new(x_part, y_part));
/*log_info!("=== level 2 NE:\n");
let _ = find_partition_point(&ne, x_start, x, y_start, y);
log_info!("=== level 2 SE:\n");
let _ = find_partition_point(&se, x, x_finish, y_start, y);
log_info!("=== level 2 SW:\n");
let _ = find_partition_point(&sw, x, x_finish, y, y_finish);
log_info!("=== level 2 NW:\n");
let _ = find_partition_point(&nw, x_start, x, y, y_finish);*/
true
}