diff options
| -rw-r--r-- | coil_gen.py | 86 |
1 files changed, 51 insertions, 35 deletions
diff --git a/coil_gen.py b/coil_gen.py index 642a77d..78610d2 100644 --- a/coil_gen.py +++ b/coil_gen.py @@ -73,16 +73,18 @@ def generate(infile, outfile, polygon, start_angle, stop_radius, trace_width, cl segment_heights = [point_line_distance((cx, cy), (x1, y1), (x2, y2)) for (x1, y1), (x2, y2) in segments] segment_foo = list(zip(segment_heights, segments)) - closest_points = [] + midpoints = [] + for h, ((x1, y1), (x2, y2)) in segment_foo: + xb = (x1 + x2) / 2 + yb = (y1 + y2) / 2 + midpoints.append((xb, yb)) + + normals = [] for h, ((x1, y1), (x2, y2)) in segment_foo: - dc1 = dist((x1, y1), (cx, cy)) d12 = dist((x1, y1), (x2, y2)) - db = sqrt(dc1**2 - h**2) - xn = (x2 - x1) / d12 - yn = (y2 - y1) / d12 - xb = x1 + xn * db - yb = y1 + yn * db - closest_points.append((xb, yb)) + dx = x2 - x1 + dy = y2 - y1 + normals.append((-dy/d12, dx/d12)) smallest_radius = min(segment_heights) #trace_radius = smallest_radius - stop_radius @@ -90,24 +92,21 @@ def generate(infile, outfile, polygon, start_angle, stop_radius, trace_width, cl num_windings = floor((trace_radius - trace_width) / (clearance + trace_width)) print(f'Going for {num_windings} windings') - segment_foo = list(zip(segment_heights, segments, segment_angles, closest_points)) + segment_foo = list(zip(segment_heights, segments, segment_angles, midpoints, normals)) - dbg_lines = [] + dbg_lines1, dbg_lines2 = [], [] spiral_points = [] dr_tot = 0 for n in range(num_windings): - for (ha, (pa1, pa2), aa, ma), (hb, (pb1, pb2), ab, mb) in zip(segment_foo[-1:] + segment_foo[:-1], segment_foo): + for (ha, (pa1, pa2), aa, ma, na), (hb, (pb1, pb2), ab, mb, nb) in zip(segment_foo[-1:] + segment_foo[:-1], segment_foo): pitch = clearance + trace_width dr_tot_a = dr_tot dr_tot_b = dr_tot + ab/(2*pi) * pitch xma, yma = ma + xna, yna = na xmb, ymb = mb - - xra = (xma - cx) / ha - yra = (yma - cy) / ha - xrb = (xmb - cx) / hb - yrb = (ymb - cy) / hb + xnb, ynb = nb xa1, ya1 = pa1 xa2, ya2 = pa2 @@ -117,27 +116,39 @@ def generate(infile, outfile, polygon, start_angle, stop_radius, trace_width, cl dma = dist(pa2, ma) dmb = dist(pb1, mb) - qa = dr_tot_a*dma/ha - dra = hypot(dr_tot_a, qa) - xea = xa2 + (cx - xa2) / dist((cx, cy), pa2) * dra - yea = ya2 + (cy - ya2) / dist((cx, cy), pa2) * dra + x_cons_a, y_cons_a = p_cons_a = line_line_intersection((pa2, (cx, cy)), (ma, (xma-xna, yma-yna))) + d_cons_a = dist(p_cons_a, ma) + qa = dma * dr_tot_a / d_cons_a + dra = hypot(qa, dr_tot_a) + + nrax = (xa2 - cx) / dist((cx, cy), pa2) + nray = (ya2 - cy) / dist((cx, cy), pa2) + + xea = xa2 - nrax*dra + yea = ya2 - nray*dra + + x_cons_b, y_cons_b = p_cons_b = line_line_intersection((pb1, (cx, cy)), (mb, (xmb-xnb, ymb-ynb))) + d_cons_b = dist(p_cons_b, mb) + qb = dmb * dr_tot_b / d_cons_b + drb = hypot(qb, dr_tot_b) + + nrbx = (xb1 - cx) / dist((cx, cy), pb1) + nrby = (yb1 - cy) / dist((cx, cy), pb1) - qb = dr_tot_b*dmb/hb - drb = hypot(dr_tot_b, qb) - xeb = xb1 + (cx - xb1) / dist((cx, cy), pb1) * drb - yeb = yb1 + (cy - yb1) / dist((cx, cy), pb1) * drb + xeb = xb1 - nrbx*drb + yeb = yb1 - nrby*drb - xsa = xma - xra*dr_tot_a - ysa = yma - yra*dr_tot_a + xsa = xma - xna*dr_tot_a + ysa = yma - yna*dr_tot_a - xsb = xmb - xrb*dr_tot_b - ysb = ymb - yrb*dr_tot_b + xsb = xmb - xnb*dr_tot_b + ysb = ymb - ynb*dr_tot_b l1 = (xsa, ysa), (xea, yea) l2 = (xsb, ysb), (xeb, yeb) - dbg_lines.append(l1) - dbg_lines.append(l2) + dbg_lines1.append(l1) + dbg_lines2.append(l2) pic = line_line_intersection(l1, l2) spiral_points.append(pic) @@ -172,13 +183,18 @@ def generate(infile, outfile, polygon, start_angle, stop_radius, trace_width, cl f.write(f'<path fill="none" stroke="#303030" stroke-width="0.05" d="{path_d}"/>\n') f.write(f'<path fill="none" stroke="#a0a0a0" stroke-width="0.05" d="{path_d2}"/>\n') f.write(f'<path fill="none" stroke="#ff00ff" opacity="0.5" stroke-width="{trace_width}" d="{path_d3}"/>\n') - f.write(f'<circle r="0.1" fill="red" stroke="none" cx="{cx}" cy="{cy}"/>\n') - for x, y in closest_points: - f.write(f'<circle r="0.1" fill="blue" stroke="none" cx="{x}" cy="{y}"/>\n') + + for (x1, y1), (x2, y2) in dbg_lines1: + f.write(f'<path fill="none" stroke="#ff0000" opacity="0.2" stroke-width="0.05" d="M {x1} {y1} L {x2} {y2}"/>') + + for (x1, y1), (x2, y2) in dbg_lines2: + f.write(f'<path fill="none" stroke="#0000ff" opacity="0.2" stroke-width="0.05" d="M {x1} {y1} L {x2} {y2}"/>') + + for x, y in midpoints: f.write(f'<path fill="none" stroke="#a0a0ff" stroke-width="0.05" d="M {cx} {cy} L {x} {y}"/>') + f.write(f'<circle r="0.1" fill="blue" stroke="none" cx="{x}" cy="{y}"/>\n') - for (x1, y1), (x2, y2) in dbg_lines: - f.write(f'<path fill="none" stroke="#000000" opacity="0.2" stroke-width="0.05" d="M {x1} {y1} L {x2} {y2}"/>') + f.write(f'<circle r="0.1" fill="red" stroke="none" cx="{cx}" cy="{cy}"/>\n') f.write('</svg>\n') if __name__ == '__main__': |