quickpcb源码

华秋PCB（https://www.hqpcb.com/zhuoluyex7?zt-499）四层板板材费降至499元/㎡，狂降16.83%；多层板全面大幅度降价，交期再提速

以下是一个简化版的 QuickPCB 电路板设计软件的 Python 实现源码。该程序使用 pygame 库实现了基本的 PCB 绘制功能，包括放置元件、绘制导线、保存/加载设计等功能：

import pygame
import pickle
import os

# 初始化pygame
pygame.init()
WIDTH, HEIGHT = 800, 600
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("QuickPCB v1.0")

# 颜色定义
BACKGROUND = (40, 44, 52)
GRID_COLOR = (60, 64, 72)
COMPONENT_COLOR = (86, 182, 194)
WIRE_COLOR = (152, 195, 121)
HIGHLIGHT = (255, 184, 108)
TEXT_COLOR = (220, 220, 220)
UI_BG = (30, 34, 42)

# PCB元件类
class Component:
    def __init__(self, x, y, type="IC"):
        self.x = x
        self.y = y
        self.type = type
        self.pins = []
        self.size = 40 if type == "IC" else 30
        self.create_pins()

    def create_pins(self):
        if self.type == "IC":
            # 创建IC的引脚 (两侧)
            for i in range(8):
                self.pins.append((self.x - 20, self.y - 15 + i * 10))
                self.pins.append((self.x + 20, self.y - 15 + i * 10))
        else:  # 电阻
            self.pins = [(self.x - 15, self.y), (self.x + 15, self.y)]

    def draw(self, surface):
        color = HIGHLIGHT if self == selected_component else COMPONENT_COLOR
        if self.type == "IC":
            pygame.draw.rect(surface, color, (self.x-20, self.y-20, 40, 40), 2)
            # 绘制引脚
            for pin in self.pins:
                pygame.draw.circle(surface, COMPONENT_COLOR, pin, 3)
        else:  # 电阻
            pygame.draw.rect(surface, color, (self.x-15, self.y-8, 30, 16), 2)
            pygame.draw.line(surface, color, (self.x-25, self.y), (self.x-15, self.y), 2)
            pygame.draw.line(surface, color, (self.x+15, self.y), (self.x+25, self.y), 2)

# PCB板类
class PCB:
    def __init__(self):
        self.components = []
        self.wires = []
        self.grid_size = 20

    def add_component(self, x, y, type):
        # 对齐到网格
        x = (x // self.grid_size) * self.grid_size
        y = (y // self.grid_size) * self.grid_size
        self.components.append(Component(x, y, type))

    def draw(self, surface):
        # 绘制网格
        for x in range(0, WIDTH, self.grid_size):
            pygame.draw.line(surface, GRID_COLOR, (x, 0), (x, HEIGHT), 1)
        for y in range(0, HEIGHT, self.grid_size):
            pygame.draw.line(surface, GRID_COLOR, (0, y), (WIDTH, y), 1)

        # 绘制导线
        for wire in self.wires:
            pygame.draw.line(surface, WIRE_COLOR, wire[0], wire[1], 3)

        # 绘制元件
        for comp in self.components:
            comp.draw(surface)

    def save_design(self, filename):
        with open(filename, 'wb') as f:
            pickle.dump((self.components, self.wires), f)

    def load_design(self, filename):
        if os.path.exists(filename):
            with open(filename, 'rb') as f:
                self.components, self.wires = pickle.load(f)

# 创建PCB实例
pcb = PCB()
selected_component = None
current_tool = "COMPONENT"  # COMPONENT, WIRE
current_component_type = "IC"
drawing_wire = False
wire_start = None

# 按钮类
class Button:
    def __init__(self, x, y, width, height, text):
        self.rect = pygame.Rect(x, y, width, height)
        self.text = text

    def draw(self, surface):
        pygame.draw.rect(surface, UI_BG, self.rect)
        pygame.draw.rect(surface, TEXT_COLOR, self.rect, 2)
        font = pygame.font.SysFont(None, 24)
        text_surf = font.render(self.text, True, TEXT_COLOR)
        surface.blit(text_surf, (self.rect.centerx - text_surf.get_width()//2, 
                               self.rect.centery - text_surf.get_height()//2))

    def is_clicked(self, pos):
        return self.rect.collidepoint(pos)

# 创建UI按钮
buttons = [
    Button(10, 10, 120, 40, "IC芯片"),
    Button(140, 10, 120, 40, "电阻"),
    Button(270, 10, 120, 40, "导线模式"),
    Button(400, 10, 120, 40, "保存"),
    Button(530, 10, 120, 40, "加载")
]

# 主循环
running = True
clock = pygame.time.Clock()

while running:
    mouse_pos = pygame.mouse.get_pos()

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

        elif event.type == pygame.MOUSEBUTTONDOWN:
            # 检查按钮点击
            for i, button in enumerate(buttons):
                if button.is_clicked(mouse_pos):
                    if i == 0:  # IC芯片
                        current_tool = "COMPONENT"
                        current_component_type = "IC"
                    elif i == 1:  # 电阻
                        current_tool = "COMPONENT"
                        current_component_type = "RESISTOR"
                    elif i == 2:  # 导线模式
                        current_tool = "WIRE"
                    elif i == 3:  # 保存
                        pcb.save_design("design.pcb")
                    elif i == 4:  # 加载
                        pcb.load_design("design.pcb")

            # 放置元件或开始画线
            if mouse_pos[1] > 60:  # 避免点击到UI区域
                if current_tool == "COMPONENT":
                    pcb.add_component(mouse_pos[0], mouse_pos[1], current_component_type)
                elif current_tool == "WIRE":
                    drawing_wire = True
                    wire_start = mouse_pos

        elif event.type == pygame.MOUSEBUTTONUP and drawing_wire:
            drawing_wire = False
            if wire_start and mouse_pos != wire_start:
                pcb.wires.append((wire_start, mouse_pos))

    # 渲染
    screen.fill(BACKGROUND)
    pcb.draw(screen)

    # 绘制按钮
    for button in buttons:
        button.draw(screen)

    # 显示当前工具
    font = pygame.font.SysFont(None, 28)
    tool_text = font.render(f"当前工具: {'元件模式' if current_tool == 'COMPONENT' else '导线模式'}", True, HIGHLIGHT)
    screen.blit(tool_text, (10, HEIGHT - 40))

    # 绘制实时导线
    if drawing_wire and wire_start:
        pygame.draw.line(screen, WIRE_COLOR, wire_start, mouse_pos, 3)

    pygame.display.flip()
    clock.tick(60)

pygame.quit()