def parse_net(self, args): """ Assembles a net from a list of junctions, segments, and labels. """ thisnet = Net(args) subdata = self.sub_nodes('J S A L Q B'.split()) # finish building thisnet for netpt in subdata['netpoint'][:]: # using a copy so that we can modify subdata['netpoint'] inside loop if netpt.point_id not in thisnet.points: thisnet.add_point(netpt) else: # oh yeah, a net can have a point more than once, because that # makes *great* sense. for point in netpt.connected_points: thisnet.points[netpt.point_id].add_connected_point(point) for comp in netpt.connected_components: thisnet.points[netpt.point_id].add_connected_component(comp) # update subdata['netpoint'] so that ref to netpt points to the # new combined point i = subdata['netpoint'].index(netpt) subdata['netpoint'][i] = thisnet.points[netpt.point_id] # yuck, passing in-band thisnet.ibpts = subdata['netpoint'] for pt_a, pt_b in subdata['segment']: thisnet.connect((subdata['netpoint'][pt_a - 1], subdata['netpoint'][pt_b - 1])) for annot in subdata['annot']: thisnet.add_annotation(annot) if '=' in annot.value: thisnet.add_attribute(*(annot.value.split('=', 1))) return ('net', thisnet)
class NetTests(unittest.TestCase): """ The tests of the core module net feature """ def setUp(self): """ Setup the test case. """ self.net = Net('001') def tearDown(self): """ Teardown the test case. """ pass def test_create_new_net(self): """ Test the creation of a new empty net. """ assert self.net.net_id == '001' def test_bounds_simple(self): '''Make sure bounds() uses all the included NetPoints''' for (x, y) in ((1, 3), (3, 2), (4, 3), (3, 5)): net_pt = NetPoint(str((x, y)), x, y) self.net.add_point(net_pt) # NetPoints don't actually need to be connected to affect bounds() top_left, btm_right = self.net.bounds() self.assertEqual(top_left.x, 1) self.assertEqual(top_left.y, 2) self.assertEqual(btm_right.x, 4) self.assertEqual(btm_right.y, 5)
def parse_net(self, args): """ Assembles a net from a list of junctions, segments, and labels. """ thisnet = Net(args) subdata = self.sub_nodes('J S A L Q B'.split()) # finish building thisnet for netpt in subdata['netpoint'][:]: # using a copy so that we can modify subdata['netpoint'] inside loop if netpt.point_id not in thisnet.points: thisnet.add_point(netpt) else: # oh yeah, a net can have a point more than once, because that # makes *great* sense. for point in netpt.connected_points: thisnet.points[netpt.point_id].add_connected_point(point) for comp in netpt.connected_components: thisnet.points[netpt.point_id].add_connected_component( comp) # update subdata['netpoint'] so that ref to netpt points to the # new combined point i = subdata['netpoint'].index(netpt) subdata['netpoint'][i] = thisnet.points[netpt.point_id] # yuck, passing in-band thisnet.ibpts = subdata['netpoint'] for pt_a, pt_b in subdata['segment']: thisnet.connect( (subdata['netpoint'][pt_a - 1], subdata['netpoint'][pt_b - 1])) for annot in subdata['annot']: thisnet.add_annotation(annot) if '=' in annot.value: thisnet.add_attribute(*(annot.value.split('=', 1))) return ('net', thisnet)
def test_write_net(self): """ The write_net method creates the correct kicad wires from an openjson net. """ net = Net('') p1 = NetPoint('p1', 0, 0) p2 = NetPoint('p2', 1, 0) p3 = NetPoint('p3', 0, 1) net.add_point(p1) net.add_point(p2) net.add_point(p3) net.conn_point(p1, p2) net.conn_point(p1, p3) writer = KiCAD() buf = StringIO() writer.write_net(buf, net) self.assertEqual( buf.getvalue(), 'Wire Wire Line\n\t0 0 0 ' + str(int(-1 / MULT)) + '\nWire Wire Line\n\t0 0 ' + str(int(1 / MULT)) + ' 0\n')
def calc_nets(self, design, segments): """ Return a set of Nets from segments """ coord2point = {} # (x, y) -> NetPoint def get_point(coord): """ Return a new or existing NetPoint for an (x,y) point """ coord = (int(coord[0]), int(coord[1])) if coord not in coord2point: coord2point[coord] = NetPoint('%da%d' % coord, coord[0], coord[1]) return coord2point[coord] # use this to track connected pins not yet added to a net self.make_pin_points(design, get_point) # set of points connected to pins pin_points = set(coord2point.itervalues()) # turn the (x, y) points into unique NetPoint objects segments = set((get_point(p1), get_point(p2)) for p1, p2 in segments) nets = [] # Iterate over the segments, removing segments when added to a net while segments: seg = segments.pop() # pick a point newnet = Net('') map(pin_points.discard, seg) # mark points as used newnet.connect(seg) found = True while found: found = set() for seg in segments: # iterate over segments if newnet.connected(seg): # segment touching the net map(pin_points.discard, seg) # mark points as used newnet.connect(seg) # add the segment found.add(seg) for seg in found: segments.remove(seg) nets.append(newnet) # add single-point nets for overlapping pins that are not # already in other nets for point in pin_points: if len(point.connected_components) > 1: net = Net('') net.add_point(point) nets.append(net) for net in nets: net.net_id = min(net.points) nets.sort(key=lambda net: net.net_id) return nets
def calc_nets(self, design, segments): """ Return a set of Nets from segments """ coord2point = {} # (x, y) -> NetPoint def get_point(coord): """ Return a new or existing NetPoint for an (x,y) point """ coord = (int(coord[0]), int(coord[1])) if coord not in coord2point: coord2point[coord] = NetPoint('%da%d' % coord, coord[0], coord[1]) return coord2point[coord] # use this to track connected pins not yet added to a net self.make_pin_points(design, get_point) # set of points connected to pins pin_points = set(coord2point.itervalues()) # turn the (x, y) points into unique NetPoint objects segments = set((get_point(p1), get_point(p2)) for p1, p2 in segments) nets = [] # Iterate over the segments, removing segments when added to a net while segments: seg = segments.pop() # pick a point newnet = Net('') map(pin_points.discard, seg) # mark points as used newnet.connect(seg) found = True while found: found = set() for seg in segments: # iterate over segments if newnet.connected(seg): # segment touching the net map(pin_points.discard, seg) # mark points as used newnet.connect(seg) # add the segment found.add(seg) for seg in found: segments.remove(seg) nets.append(newnet) # add single-point nets for overlapping pins that are not # already in other nets for point in pin_points: if len(point.connected_components) > 1: net = Net('') net.add_point(point) nets.append(net) for net in nets: net.net_id = min(net.points) nets.sort(key=lambda net : net.net_id) return nets
def parse_nets(self, nets): """ Extract nets. """ for net in nets: net_id = net.get('net_id') ret_net = Net(net_id) # Add Annotations for annotation in net.get('annotations'): anno = self.parse_annotation(annotation) ret_net.add_annotation(anno) # Get the Attributes for key, value in net.get('attributes').items(): ret_net.add_attribute(key, value) # Get the Points for net_point in net.get('points'): npnt = self.parse_net_point(net_point) ret_net.add_point(npnt) self.design.add_net(ret_net)
def parse_nets(self, nets): """ Extract nets. """ for net in nets: net_id = net.get('net_id') ret_net = Net(net_id) # Add Annotations for annotation in net.get('annotations'): anno = self.parse_annotation(annotation) ret_net.add_annotation(anno) # Get the Attributes for key, value in net.get('attributes').items(): ret_net.add_attribute(key, value) # Get the Points for net_point in net.get('points'): npnt = self.parse_net_point(net_point) ret_net.add_point(npnt) self.design.add_net(ret_net)
def _get_point(self, net_id, point_id, x, y): if net_id not in self.nets: n = Net(net_id) self.design.add_net(n) self.nets[n.net_id] = n else: n = self.nets[net_id] key = (x, y) if key not in self.net_points: if not point_id: point_id = str(self._id) self._id += 1 np = NetPoint(net_id + '-' + point_id, x, y) n.add_point(np) self.net_points[key] = np else: np = self.net_points[key] return np
def parse_net(self, args): """ Assembles a net from a list of junctions, segments, and labels. """ thisnet = Net(args) subdata = defaultdict(list) for phrase in self.stream: print phrase cmd, _sep, args = phrase.partition(' ') if cmd not in ('J', 'S', 'A', 'L', 'Q', 'B'): self.stream.push(phrase) break print args k, v = self.parsenode(cmd)(args) subdata[k].append(v) # finish building thisnet for netpt in subdata['netpoint'][:]: # using a copy so that we can modify subdata['netpoint'] inside loop if netpt.point_id not in thisnet.points: thisnet.add_point(netpt) else: # oh yeah, a net can have a point more than once, because that # makes *great* sense. for point in netpt.connected_points: thisnet.points[netpt.point_id].add_connected_point(point) for comp in netpt.connected_components: thisnet.points[netpt.point_id].add_connected_component(comp) # update subdata['netpoint'] so that ref to netpt points to the # new combined point i = subdata['netpoint'].index(netpt) subdata['netpoint'][i] = thisnet.points[netpt.point_id] # yuck, passing in-band thisnet.ibpts = subdata['netpoint'] for pt_a, pt_b in subdata['segment']: thisnet.connect((subdata['netpoint'][pt_a - 1], subdata['netpoint'][pt_b - 1])) for annot in subdata['annot']: thisnet.add_annotation(annot) if '=' in annot.value: thisnet.add_attribute(*(annot.value.split('=', 1))) return ('net', thisnet)
def parse_net(self, args): """ Assembles a net from a list of junctions, segments, and labels. """ thisnet = Net(args) subdata = defaultdict(list) for phrase in self.stream: cmd, _sep, args = phrase.partition(' ') if cmd not in ('J', 'S', 'A', 'L', 'Q', 'B'): self.stream.push(phrase) break k, v = self.parsenode(cmd)(args) subdata[k].append(v) # finish building thisnet for netpt in subdata['netpoint'][:]: # using a copy so that we can modify subdata['netpoint'] inside loop if netpt.point_id not in thisnet.points: thisnet.add_point(netpt) else: # oh yeah, a net can have a point more than once, because that # makes *great* sense. for point in netpt.connected_points: thisnet.points[netpt.point_id].add_connected_point(point) for comp in netpt.connected_components: thisnet.points[netpt.point_id].add_connected_component(comp) # update subdata['netpoint'] so that ref to netpt points to the # new combined point i = subdata['netpoint'].index(netpt) subdata['netpoint'][i] = thisnet.points[netpt.point_id] # yuck, passing in-band thisnet.ibpts = subdata['netpoint'] for pt_a, pt_b in subdata['segment']: thisnet.connect((subdata['netpoint'][pt_a - 1], subdata['netpoint'][pt_b - 1])) for annot in subdata['annot']: thisnet.add_annotation(annot) if '=' in annot.value: thisnet.add_attribute(*(annot.value.split('=', 1))) return ('net', thisnet)