autorouter,board: use both band uids and band termsegs

This commit is contained in:
Mikolaj Wielgus 2024-07-24 01:53:06 +02:00
parent f7126dfa33
commit 3bf4d2cdb1
11 changed files with 86 additions and 62 deletions

View File

@ -56,18 +56,25 @@ impl Autoroute {
let (source, target) = autorouter.ratline_endpoints(curr_ratline); let (source, target) = autorouter.ratline_endpoints(curr_ratline);
let band = { let band_last_seg = {
let mut router = Router::new(autorouter.board.layout_mut()); let mut router = Router::new(autorouter.board.layout_mut());
let RouterStatus::Finished(band) = route.step(&mut router)? else { let RouterStatus::Finished(band_last_seg) = route.step(&mut router)? else {
return Ok(AutorouterStatus::Running); return Ok(AutorouterStatus::Running);
}; };
band band_last_seg
}; };
let band = autorouter
.board
.layout()
.drawing()
.collect()
.loose_band_uid(band_last_seg.into());
autorouter autorouter
.ratsnest .ratsnest
.assign_band_to_ratline(self.curr_ratline.unwrap(), band); .assign_band_termseg_to_ratline(self.curr_ratline.unwrap(), band_last_seg);
autorouter autorouter
.board .board

View File

@ -70,7 +70,7 @@ impl<M: AccessMesadata> Autorouter<M> {
.graph() .graph()
.edge_weight(*ratline) .edge_weight(*ratline)
.unwrap() .unwrap()
.band .band_termseg
.unwrap(); .unwrap();
self.board.layout_mut().remove_band(band); self.board.layout_mut().remove_band(band);
} }

View File

@ -12,7 +12,7 @@ use spade::{HasPosition, InsertionError, Point2};
use crate::{ use crate::{
drawing::{ drawing::{
band::BandTerminatingSegIndex, band::BandTermsegIndex,
dot::FixedDotIndex, dot::FixedDotIndex,
graph::{GetMaybeNet, MakePrimitive, PrimitiveIndex}, graph::{GetMaybeNet, MakePrimitive, PrimitiveIndex},
primitive::MakePrimitiveShape, primitive::MakePrimitiveShape,
@ -64,7 +64,7 @@ impl HasPosition for RatvertexWeight {
#[derive(Debug, Default, Clone, Copy)] #[derive(Debug, Default, Clone, Copy)]
pub struct RatlineWeight { pub struct RatlineWeight {
pub band: Option<BandTerminatingSegIndex>, pub band_termseg: Option<BandTermsegIndex>,
} }
pub struct Ratsnest { pub struct Ratsnest {
@ -165,12 +165,12 @@ impl Ratsnest {
Ok(this) Ok(this)
} }
pub fn assign_band_to_ratline( pub fn assign_band_termseg_to_ratline(
&mut self, &mut self,
ratline: EdgeIndex<usize>, ratline: EdgeIndex<usize>,
band: BandTerminatingSegIndex, termseg: BandTermsegIndex,
) { ) {
self.graph.edge_weight_mut(ratline).unwrap().band = Some(band); self.graph.edge_weight_mut(ratline).unwrap().band_termseg = Some(termseg);
} }
pub fn graph(&self) -> &UnGraph<RatvertexWeight, RatlineWeight, usize> { pub fn graph(&self) -> &UnGraph<RatvertexWeight, RatlineWeight, usize> {

View File

@ -1,9 +1,11 @@
use std::collections::HashMap; use std::{cmp::Ordering, collections::HashMap};
use bimap::BiHashMap;
use crate::{ use crate::{
board::mesadata::AccessMesadata, board::mesadata::AccessMesadata,
drawing::{ drawing::{
band::BandTerminatingSegIndex, band::{BandTermsegIndex, BandUid},
dot::{FixedDotIndex, FixedDotWeight}, dot::{FixedDotIndex, FixedDotWeight},
graph::{GetLayer, GetMaybeNet}, graph::{GetLayer, GetMaybeNet},
seg::{FixedSegIndex, FixedSegWeight}, seg::{FixedSegIndex, FixedSegWeight},
@ -17,11 +19,24 @@ use crate::{
math::Circle, math::Circle,
}; };
#[derive(Debug, Hash, Clone, PartialEq, Eq)]
pub struct BandName(String, String);
impl BandName {
pub fn new(pinname1: String, pinname2: String) -> Self {
if pinname1.cmp(&pinname2) == Ordering::Greater {
BandName(pinname2, pinname1)
} else {
BandName(pinname1, pinname2)
}
}
}
#[derive(Debug)] #[derive(Debug)]
pub struct Board<M: AccessMesadata> { pub struct Board<M: AccessMesadata> {
layout: Layout<M>, layout: Layout<M>,
node_to_pinname: HashMap<NodeIndex, String>, node_to_pinname: HashMap<NodeIndex, String>,
pinname_pair_to_band: HashMap<(String, String), BandTerminatingSegIndex>, band_bandname: BiHashMap<BandUid, BandName>,
} }
impl<M: AccessMesadata> Board<M> { impl<M: AccessMesadata> Board<M> {
@ -29,7 +44,7 @@ impl<M: AccessMesadata> Board<M> {
Self { Self {
layout, layout,
node_to_pinname: HashMap::new(), node_to_pinname: HashMap::new(),
pinname_pair_to_band: HashMap::new(), band_bandname: BiHashMap::new(),
} }
} }
@ -140,7 +155,7 @@ impl<M: AccessMesadata> Board<M> {
&mut self, &mut self,
source: FixedDotIndex, source: FixedDotIndex,
target: FixedDotIndex, target: FixedDotIndex,
band: BandTerminatingSegIndex, band: BandUid,
) { ) {
let source_pinname = self let source_pinname = self
.node_pinname(GenericNode::Primitive(source.into())) .node_pinname(GenericNode::Primitive(source.into()))
@ -150,23 +165,19 @@ impl<M: AccessMesadata> Board<M> {
.node_pinname(GenericNode::Primitive(target.into())) .node_pinname(GenericNode::Primitive(target.into()))
.unwrap() .unwrap()
.to_string(); .to_string();
self.pinname_pair_to_band self.band_bandname
.insert((source_pinname, target_pinname), band); .insert(band, BandName::new(source_pinname, target_pinname));
} }
pub fn band_between_pins( pub fn band_between_pins(&self, pinname1: &str, pinname2: &str) -> Option<BandUid> {
&self,
pinname1: &str,
pinname2: &str,
) -> Option<BandTerminatingSegIndex> {
if let Some(band) = self if let Some(band) = self
.pinname_pair_to_band .band_bandname
.get(&(pinname1.to_string(), pinname2.to_string())) .get_by_right(&BandName::new(pinname1.to_string(), pinname2.to_string()))
{ {
Some(*band) Some(*band)
} else if let Some(band) = self } else if let Some(band) = self
.pinname_pair_to_band .band_bandname
.get(&(pinname2.to_string(), pinname1.to_string())) .get_by_right(&BandName::new(pinname2.to_string(), pinname1.to_string()))
{ {
Some(*band) Some(*band)
} else { } else {

View File

@ -16,10 +16,10 @@ use super::{
}; };
#[derive(Debug, Hash, Clone, Copy)] #[derive(Debug, Hash, Clone, Copy)]
pub struct BandUid(BandTerminatingSegIndex, BandTerminatingSegIndex); pub struct BandUid(BandTermsegIndex, BandTermsegIndex);
impl BandUid { impl BandUid {
pub fn new(first_seg1: BandTerminatingSegIndex, first_seg2: BandTerminatingSegIndex) -> Self { pub fn new(first_seg1: BandTermsegIndex, first_seg2: BandTermsegIndex) -> Self {
if first_seg1.petgraph_index() <= first_seg2.petgraph_index() { if first_seg1.petgraph_index() <= first_seg2.petgraph_index() {
BandUid(first_seg1, first_seg2) BandUid(first_seg1, first_seg2)
} else { } else {
@ -39,13 +39,22 @@ impl Eq for BandUid {}
#[enum_dispatch(GetPetgraphIndex)] #[enum_dispatch(GetPetgraphIndex)]
#[derive(Debug, Hash, Clone, Copy)] #[derive(Debug, Hash, Clone, Copy)]
pub enum BandTerminatingSegIndex { pub enum BandTermsegIndex {
Straight(LoneLooseSegIndex), Straight(LoneLooseSegIndex),
Bended(SeqLooseSegIndex), Bended(SeqLooseSegIndex),
} }
impl From<BandTermsegIndex> for LooseIndex {
fn from(terminating_seg: BandTermsegIndex) -> Self {
match terminating_seg {
BandTermsegIndex::Straight(seg) => LooseIndex::LoneSeg(seg),
BandTermsegIndex::Bended(seg) => LooseIndex::SeqSeg(seg),
}
}
}
impl<'a, CW: Copy, R: AccessRules> MakeRef<'a, BandRef<'a, CW, R>, Drawing<CW, R>> impl<'a, CW: Copy, R: AccessRules> MakeRef<'a, BandRef<'a, CW, R>, Drawing<CW, R>>
for BandTerminatingSegIndex for BandTermsegIndex
{ {
fn ref_(&self, drawing: &'a Drawing<CW, R>) -> BandRef<'a, CW, R> { fn ref_(&self, drawing: &'a Drawing<CW, R>) -> BandRef<'a, CW, R> {
BandRef::new(*self, drawing) BandRef::new(*self, drawing)
@ -53,15 +62,12 @@ impl<'a, CW: Copy, R: AccessRules> MakeRef<'a, BandRef<'a, CW, R>, Drawing<CW, R
} }
pub struct BandRef<'a, CW: Copy, R: AccessRules> { pub struct BandRef<'a, CW: Copy, R: AccessRules> {
first_seg: BandTerminatingSegIndex, first_seg: BandTermsegIndex,
drawing: &'a Drawing<CW, R>, drawing: &'a Drawing<CW, R>,
} }
impl<'a, CW: Copy, R: AccessRules> BandRef<'a, CW, R> { impl<'a, CW: Copy, R: AccessRules> BandRef<'a, CW, R> {
pub fn new( pub fn new(first_seg: BandTermsegIndex, drawing: &'a Drawing<CW, R>) -> BandRef<'a, CW, R> {
first_seg: BandTerminatingSegIndex,
drawing: &'a Drawing<CW, R>,
) -> BandRef<'a, CW, R> {
Self { first_seg, drawing } Self { first_seg, drawing }
} }
} }
@ -69,10 +75,10 @@ impl<'a, CW: Copy, R: AccessRules> BandRef<'a, CW, R> {
impl<'a, CW: Copy, R: AccessRules> MeasureLength for BandRef<'a, CW, R> { impl<'a, CW: Copy, R: AccessRules> MeasureLength for BandRef<'a, CW, R> {
fn length(&self) -> f64 { fn length(&self) -> f64 {
match self.first_seg { match self.first_seg {
BandTerminatingSegIndex::Straight(seg) => { BandTermsegIndex::Straight(seg) => {
self.drawing.geometry().seg_shape(seg.into()).length() self.drawing.geometry().seg_shape(seg.into()).length()
} }
BandTerminatingSegIndex::Bended(first_loose_seg) => { BandTermsegIndex::Bended(first_loose_seg) => {
let mut maybe_loose: Option<LooseIndex> = Some(first_loose_seg.into()); let mut maybe_loose: Option<LooseIndex> = Some(first_loose_seg.into());
let mut prev = None; let mut prev = None;
let mut length = 0.0; let mut length = 0.0;

View File

@ -1,7 +1,7 @@
use crate::graph::{GenericIndex, GetPetgraphIndex}; use crate::graph::{GenericIndex, GetPetgraphIndex};
use super::{ use super::{
band::{BandTerminatingSegIndex, BandUid}, band::{BandTermsegIndex, BandUid},
bend::LooseBendIndex, bend::LooseBendIndex,
graph::PrimitiveIndex, graph::PrimitiveIndex,
loose::{GetPrevNextLoose, LooseIndex}, loose::{GetPrevNextLoose, LooseIndex},
@ -28,9 +28,9 @@ impl<'a, CW: Copy, R: AccessRules> Collect<'a, CW, R> {
) )
} }
fn loose_band_first_seg(&self, start_loose: LooseIndex) -> BandTerminatingSegIndex { fn loose_band_first_seg(&self, start_loose: LooseIndex) -> BandTermsegIndex {
if let LooseIndex::LoneSeg(seg) = start_loose { if let LooseIndex::LoneSeg(seg) = start_loose {
return BandTerminatingSegIndex::Straight(seg); return BandTermsegIndex::Straight(seg);
} }
let mut loose = start_loose; let mut loose = start_loose;
@ -41,14 +41,14 @@ impl<'a, CW: Copy, R: AccessRules> Collect<'a, CW, R> {
prev = Some(loose); prev = Some(loose);
loose = next_loose; loose = next_loose;
} else { } else {
return BandTerminatingSegIndex::Bended(GenericIndex::new(loose.petgraph_index())); return BandTermsegIndex::Bended(GenericIndex::new(loose.petgraph_index()));
} }
} }
} }
fn loose_band_last_seg(&self, start_loose: LooseIndex) -> BandTerminatingSegIndex { fn loose_band_last_seg(&self, start_loose: LooseIndex) -> BandTermsegIndex {
if let LooseIndex::LoneSeg(seg) = start_loose { if let LooseIndex::LoneSeg(seg) = start_loose {
return BandTerminatingSegIndex::Straight(seg); return BandTermsegIndex::Straight(seg);
} }
let mut loose = start_loose; let mut loose = start_loose;
@ -59,7 +59,7 @@ impl<'a, CW: Copy, R: AccessRules> Collect<'a, CW, R> {
next = Some(loose); next = Some(loose);
loose = prev_loose; loose = prev_loose;
} else { } else {
return BandTerminatingSegIndex::Bended(GenericIndex::new(loose.petgraph_index())); return BandTermsegIndex::Bended(GenericIndex::new(loose.petgraph_index()));
} }
} }
} }

View File

@ -6,7 +6,7 @@ use rstar::{RTree, AABB};
use thiserror::Error; use thiserror::Error;
use crate::drawing::{ use crate::drawing::{
band::BandTerminatingSegIndex, band::BandTermsegIndex,
bend::{BendIndex, BendWeight, FixedBendIndex, LooseBendIndex, LooseBendWeight}, bend::{BendIndex, BendWeight, FixedBendIndex, LooseBendIndex, LooseBendWeight},
cane::Cane, cane::Cane,
collect::Collect, collect::Collect,
@ -88,12 +88,12 @@ impl<CW: Copy, R: AccessRules> Drawing<CW, R> {
} }
} }
pub fn remove_band(&mut self, band: BandTerminatingSegIndex) { pub fn remove_band(&mut self, band: BandTermsegIndex) {
match band { match band {
BandTerminatingSegIndex::Straight(seg) => { BandTermsegIndex::Straight(seg) => {
self.geometry_with_rtree.remove_seg(seg.into()); self.geometry_with_rtree.remove_seg(seg.into());
} }
BandTerminatingSegIndex::Bended(first_loose_seg) => { BandTermsegIndex::Bended(first_loose_seg) => {
let mut dots = vec![]; let mut dots = vec![];
let mut segs = vec![]; let mut segs = vec![];
let mut bends = vec![]; let mut bends = vec![];

View File

@ -5,7 +5,7 @@ use rstar::AABB;
use crate::{ use crate::{
drawing::{ drawing::{
band::BandTerminatingSegIndex, band::BandTermsegIndex,
bend::LooseBendWeight, bend::LooseBendWeight,
cane::Cane, cane::Cane,
dot::{DotIndex, FixedDotIndex, FixedDotWeight, LooseDotIndex, LooseDotWeight}, dot::{DotIndex, FixedDotIndex, FixedDotWeight, LooseDotIndex, LooseDotWeight},
@ -204,7 +204,7 @@ impl<R: AccessRules> Layout<R> {
) )
} }
pub fn remove_band(&mut self, band: BandTerminatingSegIndex) { pub fn remove_band(&mut self, band: BandTermsegIndex) {
self.drawing.remove_band(band); self.drawing.remove_band(band);
} }

View File

@ -4,7 +4,7 @@ use thiserror::Error;
use crate::{ use crate::{
drawing::{ drawing::{
band::BandTerminatingSegIndex, band::BandTermsegIndex,
bend::{BendIndex, LooseBendWeight}, bend::{BendIndex, LooseBendWeight},
dot::{DotIndex, FixedDotIndex, LooseDotIndex, LooseDotWeight}, dot::{DotIndex, FixedDotIndex, LooseDotIndex, LooseDotWeight},
graph::{GetLayer, GetMaybeNet, MakePrimitive}, graph::{GetLayer, GetMaybeNet, MakePrimitive},
@ -51,7 +51,7 @@ impl<'a, R: AccessRules> Draw<'a, R> {
head: Head, head: Head,
into: FixedDotIndex, into: FixedDotIndex,
width: f64, width: f64,
) -> Result<BandTerminatingSegIndex, DrawException> { ) -> Result<BandTermsegIndex, DrawException> {
let tangent = self let tangent = self
.guide() .guide()
.head_into_dot_segment(&head, into, width) .head_into_dot_segment(&head, into, width)
@ -62,8 +62,8 @@ impl<'a, R: AccessRules> Draw<'a, R> {
let layer = head.face().primitive(self.layout.drawing()).layer(); let layer = head.face().primitive(self.layout.drawing()).layer();
let maybe_net = head.face().primitive(self.layout.drawing()).maybe_net(); let maybe_net = head.face().primitive(self.layout.drawing()).maybe_net();
Ok::<BandTerminatingSegIndex, DrawException>(match head.face() { Ok::<BandTermsegIndex, DrawException>(match head.face() {
DotIndex::Fixed(dot) => BandTerminatingSegIndex::Straight( DotIndex::Fixed(dot) => BandTermsegIndex::Straight(
self.layout self.layout
.add_lone_loose_seg( .add_lone_loose_seg(
dot, dot,
@ -76,7 +76,7 @@ impl<'a, R: AccessRules> Draw<'a, R> {
) )
.map_err(|err| DrawException::CannotFinishIn(into, err.into()))?, .map_err(|err| DrawException::CannotFinishIn(into, err.into()))?,
), ),
DotIndex::Loose(dot) => BandTerminatingSegIndex::Bended( DotIndex::Loose(dot) => BandTermsegIndex::Bended(
self.layout self.layout
.add_seq_loose_seg( .add_seq_loose_seg(
into.into(), into.into(),

View File

@ -4,7 +4,7 @@ use thiserror::Error;
use crate::{ use crate::{
drawing::{ drawing::{
band::BandTerminatingSegIndex, band::BandTermsegIndex,
dot::{DotIndex, FixedDotIndex}, dot::{DotIndex, FixedDotIndex},
graph::{MakePrimitive, PrimitiveIndex}, graph::{MakePrimitive, PrimitiveIndex},
head::GetFace, head::GetFace,
@ -39,7 +39,7 @@ pub enum RouterError {
#[derive(Debug)] #[derive(Debug)]
pub enum RouterStatus { pub enum RouterStatus {
Running, Running,
Finished(BandTerminatingSegIndex), Finished(BandTermsegIndex),
} }
#[derive(Debug)] #[derive(Debug)]
@ -78,7 +78,7 @@ impl<'a, R: AccessRules> RouterAstarStrategy<'a, R> {
} }
} }
impl<'a, R: AccessRules> AstarStrategy<Navmesh, f64, BandTerminatingSegIndex> impl<'a, R: AccessRules> AstarStrategy<Navmesh, f64, BandTermsegIndex>
for RouterAstarStrategy<'a, R> for RouterAstarStrategy<'a, R>
{ {
fn is_goal( fn is_goal(
@ -86,7 +86,7 @@ impl<'a, R: AccessRules> AstarStrategy<Navmesh, f64, BandTerminatingSegIndex>
navmesh: &Navmesh, navmesh: &Navmesh,
vertex: NavvertexIndex, vertex: NavvertexIndex,
tracker: &PathTracker<Navmesh>, tracker: &PathTracker<Navmesh>,
) -> Option<BandTerminatingSegIndex> { ) -> Option<BandTermsegIndex> {
let new_path = tracker.reconstruct_path_to(vertex); let new_path = tracker.reconstruct_path_to(vertex);
let width = self.trace.width; let width = self.trace.width;
@ -176,7 +176,7 @@ impl<'a, R: AccessRules> Router<'a, R> {
from: FixedDotIndex, from: FixedDotIndex,
to: FixedDotIndex, to: FixedDotIndex,
width: f64, width: f64,
) -> Result<BandTerminatingSegIndex, RouterError> { ) -> Result<BandTermsegIndex, RouterError> {
let mut route = self.route_walk(from, to, width)?; let mut route = self.route_walk(from, to, width)?;
loop { loop {

View File

@ -2,7 +2,7 @@ use contracts::{debug_ensures, debug_requires};
use thiserror::Error; use thiserror::Error;
use crate::{ use crate::{
drawing::{band::BandTerminatingSegIndex, dot::FixedDotIndex, rules::AccessRules}, drawing::{band::BandTermsegIndex, dot::FixedDotIndex, rules::AccessRules},
layout::Layout, layout::Layout,
router::{ router::{
draw::{Draw, DrawException}, draw::{Draw, DrawException},
@ -44,7 +44,7 @@ impl<'a, R: AccessRules> Tracer<'a, R> {
trace: &mut Trace, trace: &mut Trace,
target: FixedDotIndex, target: FixedDotIndex,
width: f64, width: f64,
) -> Result<BandTerminatingSegIndex, TracerException> { ) -> Result<BandTermsegIndex, TracerException> {
Ok(Draw::new(self.layout).finish_in_dot(trace.head, target, width)?) Ok(Draw::new(self.layout).finish_in_dot(trace.head, target, width)?)
} }