mirror of https://codeberg.org/topola/topola.git
747 lines
24 KiB
Rust
747 lines
24 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use geo::{point, Point, Rotate};
|
|
use thiserror::Error;
|
|
|
|
use crate::{
|
|
board::{mesadata::MesadataTrait, Board},
|
|
drawing::{dot::FixedDotWeight, seg::FixedSegWeight, Drawing},
|
|
layout::{zone::SolidZoneWeight, Layout},
|
|
math::Circle,
|
|
specctra::{
|
|
read,
|
|
mesadata::SpecctraMesadata,
|
|
structure::{self, Layer, Pcb, Shape, SpecctraFile},
|
|
},
|
|
};
|
|
|
|
#[derive(Error, Debug)]
|
|
pub enum LoadingError {
|
|
#[error(transparent)]
|
|
Io(#[from] std::io::Error),
|
|
#[error(transparent)]
|
|
Parse(#[from] read::ParseError),
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct SpecctraDesign {
|
|
pcb: Pcb,
|
|
}
|
|
|
|
impl SpecctraDesign {
|
|
pub fn load_from_file(filename: &str) -> Result<Self, LoadingError> {
|
|
let file = std::fs::File::open(filename)?;
|
|
let reader = std::io::BufReader::new(file);
|
|
let mut list_reader = read::ListTokenizer::new(reader);
|
|
|
|
if let Ok(file) = list_reader.read_value::<structure::SpecctraFile>() {
|
|
//use super::structure::*;
|
|
|
|
// (this entire if let block does not belong here)
|
|
|
|
/*let ses_name = filename.replace(".dsn", ".ses");
|
|
let file2 = std::fs::File::create(ses_name).unwrap();
|
|
let writer = std::io::BufWriter::new(file2);
|
|
let mut list_writer = super::write::ListWriter::new(writer);
|
|
|
|
let mut net_outs = HashMap::<String, NetOut>::new();
|
|
for mut wire in dsn.pcb.wiring.wires {
|
|
// move wires to double check that importing the resulting file into KiCad does something
|
|
for point in &mut wire.path.coords {
|
|
point.x += 1000.0;
|
|
}
|
|
|
|
if let Some(net) = net_outs.get_mut(&wire.net) {
|
|
net.wire.push(wire);
|
|
} else {
|
|
net_outs.insert(
|
|
wire.net.clone(),
|
|
NetOut {
|
|
name: wire.net.clone(),
|
|
wire: vec![wire],
|
|
via: Vec::new(),
|
|
},
|
|
);
|
|
}
|
|
}
|
|
for via in dsn.pcb.wiring.vias {
|
|
if let Some(net) = net_outs.get_mut(&via.net) {
|
|
net.via.push(via);
|
|
} else {
|
|
net_outs.insert(
|
|
via.net.clone(),
|
|
NetOut {
|
|
name: via.net.clone(),
|
|
wire: Vec::new(),
|
|
via: vec![via],
|
|
},
|
|
);
|
|
}
|
|
}
|
|
|
|
// build a basic .ses file from what was loaded
|
|
let ses = SesFile {
|
|
session: Session {
|
|
id: "ID".to_string(),
|
|
routes: Routes {
|
|
resolution: Resolution {
|
|
unit: "um".into(),
|
|
// TODO: why does resolution need to be adjusted from what was imported?
|
|
value: 1.0,
|
|
},
|
|
library_out: Library {
|
|
images: Vec::new(),
|
|
padstacks: dsn.pcb.library.padstacks,
|
|
},
|
|
network_out: NetworkOut {
|
|
net: net_outs.into_values().collect(),
|
|
},
|
|
},
|
|
},
|
|
};*/
|
|
|
|
//println!("{:?}", list_writer.write_value(&ses));
|
|
|
|
Ok(Self { pcb: file.pcb })
|
|
} else {
|
|
todo!();
|
|
}
|
|
}
|
|
|
|
pub fn load_from_string(contents: String) -> Result<Self, LoadingError> {
|
|
/*let dsn = de::from_str::<DsnFile>(&contents)
|
|
.map_err(|err| LoadingError::Syntax(err))?
|
|
.pcb;
|
|
|
|
Ok(Self { pcb })*/
|
|
let mut list_reader = read::ListTokenizer::new(contents.as_bytes());
|
|
let dsn = list_reader.read_value::<structure::SpecctraFile>();
|
|
|
|
Ok(Self {
|
|
pcb: dsn.unwrap().pcb,
|
|
})
|
|
}
|
|
|
|
pub fn make_board(&self) -> Board<SpecctraMesadata> {
|
|
let mesadata = SpecctraMesadata::from_pcb(&self.pcb);
|
|
let mut board = Board::new(Layout::new(Drawing::new(
|
|
mesadata,
|
|
self.pcb.structure.layers.len(),
|
|
)));
|
|
|
|
// mapping of pin -> net prepared for adding pins
|
|
let pin_nets = HashMap::<String, usize>::from_iter(
|
|
self.pcb
|
|
.network
|
|
.nets
|
|
.iter()
|
|
.map(|net_pin_assignments| {
|
|
// resolve the id so we don't work with strings
|
|
let net = board
|
|
.layout()
|
|
.drawing()
|
|
.rules()
|
|
.netname_net(&net_pin_assignments.name)
|
|
.unwrap();
|
|
|
|
// take the list of pins
|
|
// and for each pin id output (pin id, net id)
|
|
net_pin_assignments
|
|
.pins
|
|
.names
|
|
.iter()
|
|
.map(move |pinname| (pinname.clone(), net))
|
|
})
|
|
// flatten the nested iters into a single stream of tuples
|
|
.flatten(),
|
|
);
|
|
|
|
// add pins from components
|
|
for component in &self.pcb.placement.components {
|
|
for place in &component.places {
|
|
let image = self
|
|
.pcb
|
|
.library
|
|
.images
|
|
.iter()
|
|
.find(|image| image.name == component.name)
|
|
.unwrap();
|
|
|
|
for pin in &image.pins {
|
|
let pinname = format!("{}-{}", place.name, pin.id);
|
|
let net = pin_nets.get(&pinname).unwrap();
|
|
|
|
let padstack = &self
|
|
.pcb
|
|
.library
|
|
.padstacks
|
|
.iter()
|
|
.find(|padstack| padstack.name == pin.name)
|
|
.unwrap();
|
|
|
|
for shape in padstack.shapes.iter() {
|
|
match shape {
|
|
Shape::Circle(circle) => {
|
|
let layer = Self::layer(
|
|
&mut board,
|
|
&self.pcb.structure.layers,
|
|
&circle.layer,
|
|
place.side == "front",
|
|
);
|
|
Self::add_circle(
|
|
&mut board,
|
|
(place.x as f64, place.y as f64).into(),
|
|
place.rotation as f64,
|
|
(pin.x as f64, pin.y as f64).into(),
|
|
pin.rotate.unwrap_or(0.0) as f64,
|
|
circle.diameter as f64 / 2.0,
|
|
layer as usize,
|
|
*net,
|
|
Some(pinname.clone()),
|
|
)
|
|
}
|
|
Shape::Rect(rect) => {
|
|
let layer = Self::layer(
|
|
&mut board,
|
|
&self.pcb.structure.layers,
|
|
&rect.layer,
|
|
place.side == "front",
|
|
);
|
|
Self::add_rect(
|
|
&mut board,
|
|
(place.x as f64, place.y as f64).into(),
|
|
place.rotation as f64,
|
|
(pin.x as f64, pin.y as f64).into(),
|
|
pin.rotate.unwrap_or(0.0) as f64,
|
|
rect.x1 as f64,
|
|
rect.y1 as f64,
|
|
rect.x2 as f64,
|
|
rect.y2 as f64,
|
|
layer as usize,
|
|
*net,
|
|
Some(pinname.clone()),
|
|
)
|
|
}
|
|
Shape::Path(path) => {
|
|
let layer = Self::layer(
|
|
&mut board,
|
|
&self.pcb.structure.layers,
|
|
&path.layer,
|
|
place.side == "front",
|
|
);
|
|
Self::add_path(
|
|
&mut board,
|
|
(place.x as f64, place.y as f64).into(),
|
|
place.rotation as f64,
|
|
(pin.x as f64, pin.y as f64).into(),
|
|
pin.rotate.unwrap_or(0.0) as f64,
|
|
&path.coords,
|
|
path.width as f64,
|
|
layer as usize,
|
|
*net,
|
|
Some(pinname.clone()),
|
|
)
|
|
}
|
|
Shape::Polygon(polygon) => {
|
|
let layer = Self::layer(
|
|
&mut board,
|
|
&self.pcb.structure.layers,
|
|
&polygon.layer,
|
|
place.side == "front",
|
|
);
|
|
Self::add_polygon(
|
|
&mut board,
|
|
(place.x as f64, place.y as f64).into(),
|
|
place.rotation as f64,
|
|
(pin.x as f64, pin.y as f64).into(),
|
|
pin.rotate.unwrap_or(0.0) as f64,
|
|
&polygon.coords,
|
|
polygon.width as f64,
|
|
layer as usize,
|
|
*net,
|
|
Some(pinname.clone()),
|
|
)
|
|
}
|
|
};
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
for via in &self.pcb.wiring.vias {
|
|
let net = board
|
|
.layout()
|
|
.drawing()
|
|
.rules()
|
|
.netname_net(&via.net)
|
|
.unwrap();
|
|
|
|
// find the padstack referenced by this via placement
|
|
let padstack = &self
|
|
.pcb
|
|
.library
|
|
.padstacks
|
|
.iter()
|
|
.find(|padstack| padstack.name == via.name)
|
|
.unwrap();
|
|
|
|
for shape in &padstack.shapes {
|
|
match shape {
|
|
Shape::Circle(circle) => {
|
|
let layer = Self::layer(
|
|
&mut board,
|
|
&self.pcb.structure.layers,
|
|
&circle.layer,
|
|
true,
|
|
);
|
|
Self::add_circle(
|
|
&mut board,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
circle.diameter as f64 / 2.0,
|
|
layer as usize,
|
|
net,
|
|
None,
|
|
)
|
|
}
|
|
Shape::Rect(rect) => {
|
|
let layer =
|
|
Self::layer(&mut board, &self.pcb.structure.layers, &rect.layer, true);
|
|
Self::add_rect(
|
|
&mut board,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
rect.x1 as f64,
|
|
rect.y1 as f64,
|
|
rect.x2 as f64,
|
|
rect.y2 as f64,
|
|
layer as usize,
|
|
net,
|
|
None,
|
|
)
|
|
}
|
|
Shape::Path(path) => {
|
|
let layer =
|
|
Self::layer(&mut board, &self.pcb.structure.layers, &path.layer, true);
|
|
Self::add_path(
|
|
&mut board,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
&path.coords,
|
|
path.width as f64,
|
|
layer as usize,
|
|
net,
|
|
None,
|
|
)
|
|
}
|
|
Shape::Polygon(polygon) => {
|
|
let layer = Self::layer(
|
|
&mut board,
|
|
&self.pcb.structure.layers,
|
|
&polygon.layer,
|
|
true,
|
|
);
|
|
Self::add_polygon(
|
|
&mut board,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
&polygon.coords,
|
|
polygon.width as f64,
|
|
layer as usize,
|
|
net,
|
|
None,
|
|
)
|
|
}
|
|
};
|
|
}
|
|
}
|
|
|
|
for wire in self.pcb.wiring.wires.iter() {
|
|
let layer = board
|
|
.layout()
|
|
.drawing()
|
|
.rules()
|
|
.layername_layer(&wire.path.layer)
|
|
.unwrap();
|
|
let net = board
|
|
.layout()
|
|
.drawing()
|
|
.rules()
|
|
.netname_net(&wire.net)
|
|
.unwrap();
|
|
|
|
Self::add_path(
|
|
&mut board,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
(0.0, 0.0).into(),
|
|
0.0,
|
|
&wire.path.coords,
|
|
wire.path.width as f64,
|
|
layer,
|
|
net,
|
|
None,
|
|
);
|
|
}
|
|
|
|
// The clones here are bad, we'll have something better later on.
|
|
|
|
let layername_to_layer = &board.layout().drawing().rules().layer_layername.clone();
|
|
|
|
for (layer, layername) in layername_to_layer.iter() {
|
|
board
|
|
.layout_mut()
|
|
.rules_mut()
|
|
.bename_layer(*layer, layername.to_string());
|
|
}
|
|
|
|
let netname_to_net = &board.layout().drawing().rules().net_netname.clone();
|
|
|
|
for (net, netname) in netname_to_net.iter() {
|
|
board
|
|
.layout_mut()
|
|
.rules_mut()
|
|
.bename_net(*net, netname.to_string());
|
|
}
|
|
|
|
board
|
|
}
|
|
|
|
fn layer(
|
|
board: &Board<SpecctraMesadata>,
|
|
layers: &Vec<Layer>,
|
|
layername: &str,
|
|
front: bool,
|
|
) -> usize {
|
|
let image_layer = board
|
|
.layout()
|
|
.drawing()
|
|
.rules()
|
|
.layername_layer(layername)
|
|
.unwrap();
|
|
|
|
if front {
|
|
image_layer as usize
|
|
} else {
|
|
layers.len() - image_layer as usize - 1
|
|
}
|
|
}
|
|
|
|
fn add_circle(
|
|
board: &mut Board<SpecctraMesadata>,
|
|
place_pos: Point,
|
|
place_rot: f64,
|
|
pin_pos: Point,
|
|
pin_rot: f64,
|
|
r: f64,
|
|
layer: usize,
|
|
net: usize,
|
|
maybe_pin: Option<String>,
|
|
) {
|
|
let circle = Circle {
|
|
pos: Self::pos(place_pos, place_rot, pin_pos, pin_rot, 0.0, 0.0),
|
|
r,
|
|
};
|
|
|
|
board.add_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle,
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
maybe_pin.clone(),
|
|
);
|
|
}
|
|
|
|
fn add_rect(
|
|
board: &mut Board<SpecctraMesadata>,
|
|
place_pos: Point,
|
|
place_rot: f64,
|
|
pin_pos: Point,
|
|
pin_rot: f64,
|
|
x1: f64,
|
|
y1: f64,
|
|
x2: f64,
|
|
y2: f64,
|
|
layer: usize,
|
|
net: usize,
|
|
maybe_pin: Option<String>,
|
|
) {
|
|
let zone = board.add_zone(
|
|
SolidZoneWeight {
|
|
layer,
|
|
maybe_net: Some(net),
|
|
}
|
|
.into(),
|
|
maybe_pin.clone(),
|
|
);
|
|
|
|
// Corners.
|
|
let dot_1_1 = board.add_zone_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos: Self::pos(place_pos, place_rot, pin_pos, pin_rot, x1, y1),
|
|
r: 0.5,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
let dot_2_1 = board.add_zone_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos: Self::pos(place_pos, place_rot, pin_pos, pin_rot, x2, y1),
|
|
r: 0.5,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
let dot_2_2 = board.add_zone_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos: Self::pos(place_pos, place_rot, pin_pos, pin_rot, x2, y2),
|
|
r: 0.5,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
let dot_1_2 = board.add_zone_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos: Self::pos(place_pos, place_rot, pin_pos, pin_rot, x1, y2),
|
|
r: 0.5,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
// Sides.
|
|
board.add_zone_fixed_seg_infringably(
|
|
dot_1_1,
|
|
dot_2_1,
|
|
FixedSegWeight {
|
|
width: 1.0,
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
board.add_zone_fixed_seg_infringably(
|
|
dot_2_1,
|
|
dot_2_2,
|
|
FixedSegWeight {
|
|
width: 1.0,
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
board.add_zone_fixed_seg_infringably(
|
|
dot_2_2,
|
|
dot_1_2,
|
|
FixedSegWeight {
|
|
width: 1.0,
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
board.add_zone_fixed_seg_infringably(
|
|
dot_1_2,
|
|
dot_1_1,
|
|
FixedSegWeight {
|
|
width: 1.0,
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
zone,
|
|
);
|
|
}
|
|
|
|
fn add_path(
|
|
board: &mut Board<SpecctraMesadata>,
|
|
place_pos: Point,
|
|
place_rot: f64,
|
|
pin_pos: Point,
|
|
pin_rot: f64,
|
|
coords: &Vec<structure::Point>,
|
|
width: f64,
|
|
layer: usize,
|
|
net: usize,
|
|
maybe_pin: Option<String>,
|
|
) {
|
|
// add the first coordinate in the wire path as a dot and save its index
|
|
let mut prev_pos = Self::pos(
|
|
place_pos,
|
|
place_rot,
|
|
pin_pos,
|
|
pin_rot,
|
|
coords[0].x as f64,
|
|
coords[0].y as f64,
|
|
);
|
|
let mut prev_index = board.add_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos: prev_pos,
|
|
r: width / 2.0,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
maybe_pin.clone(),
|
|
);
|
|
|
|
// iterate through path coords starting from the second
|
|
for coord in coords.iter().skip(1) {
|
|
let pos = Self::pos(
|
|
place_pos,
|
|
place_rot,
|
|
pin_pos,
|
|
pin_rot,
|
|
coord.x as f64,
|
|
coord.y as f64,
|
|
);
|
|
|
|
if pos == prev_pos {
|
|
continue;
|
|
}
|
|
|
|
let index = board.add_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos,
|
|
r: width / 2.0,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
maybe_pin.clone(),
|
|
);
|
|
|
|
// add a seg between the current and previous coords
|
|
let _ = board.add_fixed_seg_infringably(
|
|
prev_index,
|
|
index,
|
|
FixedSegWeight {
|
|
width,
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
maybe_pin.clone(),
|
|
);
|
|
|
|
prev_index = index;
|
|
prev_pos = pos;
|
|
}
|
|
}
|
|
|
|
fn add_polygon(
|
|
board: &mut Board<SpecctraMesadata>,
|
|
place_pos: Point,
|
|
place_rot: f64,
|
|
pin_pos: Point,
|
|
pin_rot: f64,
|
|
coords: &Vec<structure::Point>,
|
|
width: f64,
|
|
layer: usize,
|
|
net: usize,
|
|
maybe_pin: Option<String>,
|
|
) {
|
|
let zone = board.add_zone(
|
|
SolidZoneWeight {
|
|
layer,
|
|
maybe_net: Some(net),
|
|
}
|
|
.into(),
|
|
maybe_pin.clone(),
|
|
);
|
|
|
|
// add the first coordinate in the wire path as a dot and save its index
|
|
let mut prev_index = board.add_zone_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos: Self::pos(
|
|
place_pos,
|
|
place_rot,
|
|
pin_pos,
|
|
pin_rot,
|
|
coords[0].x as f64,
|
|
coords[0].y as f64,
|
|
),
|
|
r: width / 2.0,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
// TODO: This manual retagging shouldn't be necessary, `.into()` should suffice.
|
|
//GenericIndex::new(zone.node_index()).into(),
|
|
zone,
|
|
);
|
|
|
|
// iterate through path coords starting from the second
|
|
for coord in coords.iter().skip(1) {
|
|
let index = board.add_zone_fixed_dot_infringably(
|
|
FixedDotWeight {
|
|
circle: Circle {
|
|
pos: Self::pos(
|
|
place_pos,
|
|
place_rot,
|
|
pin_pos,
|
|
pin_rot,
|
|
coord.x as f64,
|
|
coord.y as f64,
|
|
)
|
|
.into(),
|
|
r: width / 2.0,
|
|
},
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
// TODO: This manual retagging shouldn't be necessary, `.into()` should suffice.
|
|
zone,
|
|
);
|
|
|
|
// add a seg between the current and previous coords
|
|
let _ = board.add_zone_fixed_seg_infringably(
|
|
prev_index,
|
|
index,
|
|
FixedSegWeight {
|
|
width,
|
|
layer,
|
|
maybe_net: Some(net),
|
|
},
|
|
// TODO: This manual retagging shouldn't be necessary, `.into()` should suffice.
|
|
zone,
|
|
);
|
|
|
|
prev_index = index;
|
|
}
|
|
}
|
|
|
|
fn pos(
|
|
place_pos: Point,
|
|
place_rot: f64,
|
|
pin_pos: Point,
|
|
pin_rot: f64,
|
|
x: f64,
|
|
y: f64,
|
|
) -> Point {
|
|
let pos = (point! {x: x, y: y} + pin_pos).rotate_around_point(pin_rot, pin_pos);
|
|
(pos + place_pos).rotate_around_point(place_rot, place_pos)
|
|
}
|
|
}
|