orb.py

'''
Orbital class - each type of orbital inherits from the base orbital class. Base class is only used to represent empty orbitals

@author Silas W
@version 6-5-2024
'''
class orbital:

    # Initializes empty orbital with its energy level
    # (overridden in child classes)
    def __init__(self, energyLevel):
        self.e = 0
        self.energyLevel = energyLevel
        self.capacity = 0

    # Getters
    def isFull(self):
        return self.e == self.capacity

    def getElectrons(self):
        return self.e

    # Finds stability of orbital - Where do electrons want to be
    #ADD: Implement (see electrostablilze())
    #ADD: Make it better (I wrote this in 1 min)
    def stability(self):
        if self.e * 2 == self.capacity:
            return 2
        elif self.isFull():
            return 3
        else: 
            return 0

    # Uses energy level, type, and electrons to find its part of the orbital form of an atom
    # (Overridden in child classes)
    def __str__(self):
        return ""
    
    #Adds electrons and returns leftover
    def add(self, electrons):
        self.e += int(electrons)

        if self.e > self.capacity:
            leftover = self.e - self.capacity
            self.e = self.capacity
            return leftover

        return 0

    

    
class orbitalS(orbital):
    def __init__(self, energyLevel):
        super().__init__(energyLevel)
        self.capacity = 2
    
    def __str__(self):
        if self.e > 0:
            return str(self.energyLevel) + "S^" + str(self.e) + " "
        return ""
    

class orbitalP(orbital):
    def __init__(self, energyLevel):
        super().__init__(energyLevel)
        self.capacity = 6
    
    def __str__(self):
        if self.e > 0:
            return str(self.energyLevel) + "P^" + str(self.e) + " "
        return ""
    

class orbitalD(orbital):
    def __init__(self, energyLevel):
        super().__init__(energyLevel)
        self.capacity = 10
    
    def __str__(self):
        if self.e > 0:
            return str(self.energyLevel) + "D^" + str(self.e) + " "
        return ""
    

class orbitalF(orbital):
    def __init__(self, energyLevel):
        super().__init__(energyLevel)
        self.capacity = 12
    
    def __str__(self):
        if self.e > 0:
            return str(self.energyLevel) + "F^" + str(self.e) + " "
        return ""