RB_Layout/visualize.py at master · bu-gsLab/RB_Layout · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import pygame
import numpy as np

W, H = 900, 900
RB_COL = (180, 210, 240)
PB_COL = (210, 240, 185)
BG = (255, 255, 255)
GRAY = (50, 50, 50)

def show(rows, row_index):
    pygame.init()
    screen = pygame.display.set_mode((W, H))
    font = pygame.font.SysFont("monospace", 12)
    clock = pygame.time.Clock()
    idx = 0

    while True:
        for ev in pygame.event.get():
            if ev.type == pygame.QUIT:
                pygame.quit()
                return
            if ev.type == pygame.KEYDOWN:
                if ev.key == pygame.K_RIGHT:
                    idx = (idx + 1) % len(rows)
                if ev.key == pygame.K_LEFT:
                    idx = (idx - 1) % len(rows)

        screen.fill(BG)
        row = rows[idx]
        mods = row.modules
        x_off = W / 12
        DEE_R = 1185
        inner_DEE_R = 302
        scale = H / (DEE_R * 2.2)
        DEE_R_scaled = DEE_R * scale
        inner_DEE_R_scaled = inner_DEE_R * scale

        total_rb_width = sum(m.RB.total_long for m in mods) * scale
        x = W - x_off - total_rb_width
        y = H / 2 + 100

        R = row.R * scale
        rb_keepout_x = x - (row.rb_distance * scale)
        pb_keepout_x = x + (mods[::-1][0].RB_long - mods[::-1][0].PB_long) * scale - (row.pb_distance * scale)

        if row_index <= 12:
            rb_keepout_y = y
            pb_keepout_y = y - (mods[::-1][0].RB_PB_short - mods[::-1][0].RB_short) * scale
        else:
            rb_keepout_y = y + mods[::-1][0].RB_short * scale
            pb_keepout_y = y


        p1 = np.array([rb_keepout_x, rb_keepout_y])
        p2 = np.array([pb_keepout_x, pb_keepout_y])
        d = np.linalg.norm(p2 - p1)

        mid = (p1 + p2) / 2
        h = np.sqrt(R**2 - (d/2)**2)

        perp = np.array([-(p2 - p1)[1], (p2 - p1)[0]]) / d
        center1 = mid + h * perp
        center2 = mid - h * perp
        if row_index <= 12:
            center = center1 if center1[1] > center2[1] else center2
        else:
            center = center1 if center1[1] < center2[1] else center2
        cx, cy = center
        shift_x = W / 2 - (cx - 4*R / (3*np.pi))
        shift_y = H / 2 - cy
        cx += shift_x
        cy += shift_y
        x  += shift_x
        y  += shift_y


        for m in mods[::-1]:
            rw = m.RB.total_long * scale
            rh = m.RB.total_short * scale
            pw = m.PB.total_long * scale
            ph = m.PB.total_short * scale
            pygame.draw.rect(screen, RB_COL, (x, y, rw, rh))
            pygame.draw.rect(screen, PB_COL, (x + (rw - pw), y - ph, pw, ph))
            pygame.draw.rect(screen, (0,0,0), (x, y, rw, rh), 1)
            pygame.draw.rect(screen, (0,0,0), (x + (rw - pw), y - ph, pw, ph), 1)
            lbl = font.render(m.label, True, (30, 30, 30))
            screen.blit(lbl, lbl.get_rect(center=(x + rw/2, y + rh/2)))
            x += rw

        info = f"{idx+1}/{len(rows)}  {row}"
        screen.blit(font.render(info, True, (0, 0, 0)), (20, 10))
        pygame.draw.circle(screen, (255, 0, 0), (cx, cy), R, 2)
        pygame.draw.circle(screen, GRAY, (int(cx), int(cy)), int(inner_DEE_R_scaled), 1)
        pygame.draw.circle(screen, GRAY, (int(cx), int(cy)), int(DEE_R_scaled), 1)
        pygame.draw.rect(screen, BG, (int(cx), int(cy - DEE_R_scaled), W - int(cx), 3*DEE_R_scaled))
        pygame.draw.line(screen, GRAY, (int(cx), int(cy - inner_DEE_R_scaled)), (int(cx), int(cy - DEE_R_scaled)), 1)
        pygame.draw.line(screen, GRAY, (int(cx), int(cy + inner_DEE_R_scaled)), (int(cx), int(cy + DEE_R_scaled)), 1)
        pygame.display.flip()
        clock.tick(60)