componentLibrary.py

"""component library defining all the usable components"""

from ursina import *
# from helperFunctions import *
from settings import *
from footprints import *
from math import asin, acos

# only needed for test in this file
def click():
    global currentEntity
    if currentEntity != {}:
        currentEntity.color = color.rgb(255, 255, 255)
    currentEntity = mouse.hovered_entity
    currentEntity.color = color.rgb(150, 255, 150)

# base class for components
class Component():
    def __init__(self, current_footprint, available_footprints, designator, clickFunction):
        self.available_footprints = available_footprints
        if len(available_footprints) > int(current_footprint):
            self.current_footprint = current_footprint
        else:
            print("selected footprint out of bounds")
            self.current_footprint = 0
        self.footprint = self.available_footprints[self.current_footprint](clickFunction, designator)
        self.designator = designator

    def getPinPos(self, pinNumber):
        return self.footprint.getPinPos(pinNumber)
    
    value = "None"


class AIRWIRE(Entity):
    def __init__(self, start, end, clickFunction, netname, startPart, endPart):
        self.net = netname
        self.startPart = startPart
        self.endPart = endPart
        self.designator = "AIRWIRE_" + str(next(counter))
        self.value = "AIRWIRE"

        midpoint = (Vec3(start) + Vec3(end)) / 2
        length = distance(start, end)
        scale = (airwire_thickness, airwire_thickness, length)

        super().__init__(model='cube', position=midpoint, scale=scale, collider='mesh', color=color.yellow, on_click=clickFunction)
        self.original_color = self.color


class WIRE(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [WIRE10X10, WIRE20X10, WIRE20X50, WIRE50X20], designator, clickFunction)
        self.value = "WIRE"

class BC847(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [TO92, SOT23_3], designator, clickFunction)
        self.value = "BC847"

class LED5MM(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [LED5MM_A, LED5MM_B], designator, clickFunction)
        self.value = "LED5MM"

class RES(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [RESTHT, RESTHT_SHORT, RES0603], designator, clickFunction)
        self.value = "RES"

class CAP(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [CAPTHT, CAP0603], designator, clickFunction)
        self.value = "CAP"

class PORT(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [PIN], designator, clickFunction)
        self.value = "PORT"

class DIODE(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [DIODETHT], designator, clickFunction)
        self.value = "DIODE"

class BC547(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [TO92, SOT23_3], designator, clickFunction)
        self.value = "BC547"

class BC557(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [TO92, SOT23_3], designator, clickFunction)
        self.value = "BC557"

class NE555(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [SOIC8, DIP8], designator, clickFunction)
        self.value = "NE555"

class TL072(Component):
    def __init__(self, clickFunction, footprint=0, designator = '?'):
        super().__init__(footprint, [SOIC8, DIP8], designator, clickFunction)
        self.value = "TL072"





if __name__ == '__main__':
    app = Ursina()
    window.borderless = False
    window.exit_button.enabled = False

    currentEntity = {}

    originArrows()

    # C = CAPTHT(click)
    L = LED5MM(click)
    # T = BC547(click)
    # D = DIP8(click)
    D = SOIC8(click)
    # R = RESTHT(click)
    # PORT(click)
    AIRWIRE(start=L.getPinPos(1), end=D.getPinPos(7))
    AIRWIRE(start=L.getPinPos(2), end=D.getPinPos(6))
    AIRWIRE(start=D.getPinPos(3), end=D.getPinPos(5))
    # AIRWIRE(start=D.getPinPos(4), end=D.getPinPos(8))
    # AIRWIRE(start=C.getPinPos(2), end=T.getPinPos(2))


    for i in range(-10, 10):
        for j in range(-10, 10):
            Entity(model="cube", position=[i*2, j*2, 0], scale=0.05)
            Entity(model="cube", position=[i*2, 0, j*2], scale=0.05)

    def input(key):
        if key == key_exit:
            app.userExit()

    EditorCamera()
    app.run()