debug_beading.py

import sys
import ast
import matplotlib.pyplot as plt
import matplotlib.patches as patches


def beadplot(total_thickness, widths, locations):
    fig, ax = plt.subplots(1, figsize=(15, 15))
    colors = ['g'] * len(widths)
    colors[0] = 'r'
    colors[-1] = 'r'
    for i in range(len(widths)):
        l = locations[i]
        w = widths[i]
        rect = patches.Rectangle(xy=(0, l - w / 2), width=total_thickness, height=w, linewidth=1, edgecolor='k',
                                 facecolor=colors[i], alpha=0.25, label=i)
        plt.axhline(y=l, color=colors[i], linestyle='-.')
        plt.text(x=total_thickness / 2 - 100, y=l + 5, s=f"idx = {i}\nwidth = {w}", horizontalalignment='center',
                 verticalalignment='center')
        ax.add_patch(rect)
    plt.axhline(y=total_thickness / 2, color='k', linestyle='-.')
    plt.grid(True)
    plt.axis('equal')
    plt.xlim([0, total_thickness])
    plt.show()


if __name__ == '__main__':
    for line in sys.stdin:
        parsed = ast.literal_eval(line)
        beadplot(**parsed)