diff options
Diffstat (limited to 'plugin/mesh_dialog.py')
| -rw-r--r-- | plugin/mesh_dialog.py | 77 |
1 files changed, 52 insertions, 25 deletions
diff --git a/plugin/mesh_dialog.py b/plugin/mesh_dialog.py index db1c3fd..65a3a37 100644 --- a/plugin/mesh_dialog.py +++ b/plugin/mesh_dialog.py @@ -226,7 +226,7 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): exit_cell = possible_exits[0] # might overlap multiple if not orthogonal num_valid = 0 - with DebugOutput('/mnt/c/Users/jaseg/shared/test.svg') as dbg: + with DebugOutput('/mnt/c/Users/jaseg/shared/dbg_grid.svg') as dbg: dbg.add(mask, color='#00000020') for y, row in enumerate(grid): @@ -315,22 +315,33 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): not_visited = { (x, y) for x in range(grid_cols) for y in range(grid_rows) if is_valid(grid[y][x]) } num_to_visit = len(not_visited) track_count = 0 - with DebugOutput('/mnt/c/Users/jaseg/shared/test2.svg') as dbg: - dbg.add(mask, color='#00000020') + with DebugOutput('/mnt/c/Users/jaseg/shared/dbg_cells.svg') as dbg_cells,\ + DebugOutput('/mnt/c/Users/jaseg/shared/dbg_composite.svg') as dbg_composite,\ + DebugOutput('/mnt/c/Users/jaseg/shared/dbg_tiles.svg') as dbg_tiles,\ + DebugOutput('/mnt/c/Users/jaseg/shared/dbg_traces.svg') as dbg_traces: + dbg_cells.add(mask, color='#00000020') + dbg_composite.add(mask, color='#00000020') + dbg_traces.add(mask, color='#00000020') + dbg_tiles.add(mask, color='#00000020') x, y = exit_cell[1] visited = 0 key = 0 entry_dir = 0 stack = [] + depth = 0 + max_depth = 0 while not_visited or stack: for n_x, n_y, bmask in skewed_random_iter(iter_neighbors(x, y), entry_dir, settings.randomness): if (n_x, n_y) in not_visited: - dbg.add(grid[n_y][n_x], color=virihex(visited, max=num_to_visit), opacity=0.2) + dbg_composite.add(grid[n_y][n_x], color=('visit_depth', depth), opacity=1.0) + dbg_cells.add(grid[n_y][n_x], color=('visit_depth', depth), opacity=1.0) key |= bmask - stack.append((x, y, key, bmask)) + stack.append((x, y, key, bmask, depth)) not_visited.remove((n_x, n_y)) visited += 1 + depth += 1 + max_depth = max(depth, max_depth) x, y, key, entry_dir = n_x, n_y, reciprocal(bmask), bmask break else: @@ -339,32 +350,41 @@ class MeshPluginMainDialog(mesh_plugin_dialog.MainDialog): segment = affinity.translate(segment, grid_origin[0] + x*grid_cell_width, grid_origin[1] + y*grid_cell_width) segment = affinity.rotate(segment, settings.mesh_angle, origin=mask.centroid) stroke_color = { - 0b0000: '#ff00ff80', - 0b0001: '#ff000080', - 0b0010: '#ff000080', - 0b0011: '#0000ff80', - 0b0100: '#ff000080', - 0b0101: '#00ffff80', - 0b0110: '#0000ff80', - 0b0111: '#00ff0080', - 0b1000: '#ff000080', - 0b1001: '#0000ff80', - 0b1010: '#00ffff80', - 0b1011: '#00ff0080', - 0b1100: '#0000ff80', - 0b1101: '#00ff0080', - 0b1110: '#00ff0080', - 0b1111: '#ff00ff80', + 0b0000: '#ff00ffc0', + 0b0001: '#a00000c0', + 0b0010: '#a00000c0', + 0b0011: '#0000a0c0', + 0b0100: '#a00000c0', + 0b0101: '#303030c0', + 0b0110: '#0000a0c0', + 0b0111: '#00a000c0', + 0b1000: '#a00000c0', + 0b1001: '#0000a0c0', + 0b1010: '#303030c0', + 0b1011: '#00a000c0', + 0b1100: '#0000a0c0', + 0b1101: '#00a000c0', + 0b1110: '#00a000c0', + 0b1111: '#ff00ffc0', }[key] - dbg.add(segment, stroke_width=settings.trace_width, color='#ff000000', stroke_color=stroke_color) - add_track(segment, netinfos[net]) # FIXME (works, disabled for debug) + dbg_composite.add(segment, stroke_width=settings.trace_width, color='#ff000000', stroke_color='#ffffff60') + dbg_traces.add(segment, stroke_width=settings.trace_width, color='#ff000000', stroke_color='#000000ff') + dbg_tiles.add(segment, stroke_width=settings.trace_width, color='#ff000000', + stroke_color=stroke_color) + #add_track(segment, netinfos[net]) # FIXME (works, disabled for debug) track_count += 1 if not stack: break - *stack, (x, y, key, entry_dir) = stack + *stack, (x, y, key, entry_dir, depth) = stack + + dbg_cells.scale_colors('visit_depth', max_depth) + dbg_composite.scale_colors('visit_depth', max_depth) for foo in anchor_outlines: - dbg.add(foo, color='#0000ff00', stroke_width=0.05, stroke_color='#000000ff') + dbg_cells.add(foo, color='#00000080', stroke_width=0.05, stroke_color='#00000000') + dbg_traces.add(foo, color='#00000080', stroke_width=0.05, stroke_color='#00000000') + dbg_composite.add(foo, color='#00000080', stroke_width=0.05, stroke_color='#00000000') + dbg_tiles.add(foo, color='#00000080', stroke_width=0.05, stroke_color='#00000000') print(f'Added {track_count} trace segments.') @@ -476,6 +496,13 @@ class DebugOutputWrapper: self.f = f self.objs = [] + def scale_colors(self, group, max_value): + self.objs = [ + (obj, + (virihex(color[1], max=max_value) if isinstance(color, tuple) and color[0] == group else color, + *rest)) + for obj, (color, *rest) in self.objs ] + def add(self, obj, color=None, stroke_width=0, stroke_color=None, opacity=1.0): self.objs.append((obj, (color, stroke_color, stroke_width, opacity))) |