and_gate.py

import pygame
from const import *


class AndGate:
    def __init__(self, x, y, width, height):
        self.x = x
        self.y = y
        self.width = width
        self.height = height
        self.radius = width / 2
        self.center = (x + self.radius, y + self.radius)

        self.form = '#03b'
        self.inputs = [
            {'pos': (self.x + self.width * 0.1, self.y + self.height), 'value': 0},
            {'pos': (self.x + self.width * 0.9, self.y + self.height), 'value': 0}
        ]
        self.output = {
            'pos': (self.x + self.width / 2, self.y),
            'value': 0
        }
    
    def get_value(self):
        return self.output['value']
    
    def set_value(self, index, value):
        self.inputs[index]['value'] = value
    
    def get_output_pos(self):
        return self.ouput['pos']
    
    def get_input_pos(self, index):
        return self.inputs[index]['pos']
    
    def calculate(self):
        self.output['value'] = self.inputs[0]['value'] and self.inputs[1]['value']
    
    def draw(self, screen, mouse):
        color = COLOR_ON if self.mouse_on(*mouse) else COLOR_OFF
        pygame.draw.rect(screen, color, [self.x, self.y + self.radius, self.width, self.height - self.radius])
        pygame.draw.circle(screen, color, self.center, self.radius)

        font = pygame.font.SysFont(FONT, FONT_SIZE_S, True)
        if self.mouse_on(*mouse):
            for label in self.inputs:
                value = font.render(format(label['value'], self.form), True, COLOR_RED, COLOR_WHITE)
                value_rect = value.get_rect(midtop=(label['pos'][0], label['pos'][1] + 2))
                screen.blit(value, value_rect)
            
            value = font.render(format(self.output['value'], self.form), True, COLOR_RED, COLOR_WHITE)
            value_rect = value.get_rect(midbottom=(self.output['pos'][0], self.output['pos'][1] - 2))
            screen.blit(value, value_rect)

    def mouse_on(self, x, y):
        rect = (self.x <= x <= self.x + self.width and self.y + self.radius <= y <= self.y + self.height)
        circle = (x - self.center[0])**2 + (y - self.center[1])**2 <= self.radius**2
        return rect or circle