mirror of https://github.com/E-Almqvist/hsf
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
520 B
22 lines
520 B
3 years ago
|
from phys.particle import particle
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
class quantum_system:
|
||
|
|
def __init__(self, name: str):
|
||
|
|
self.name = name
|
||
|
|
self.particles = []
|
||
|
|
|
||
|
|
def add_particle(particle: particle):
|
||
|
|
self.particles.append(particle)
|
||
|
|
print(f"Added particle {particle} to quantum system '{self.name}'")
|
||
|
|
|
||
|
|
class infbox(quantum_system):
|
||
|
|
def __init__(self, length: float):
|
||
|
|
super
|
||
|
|
self.name = "Inf Box"
|
||
|
|
self.length = length
|
||
|
|
self.norm_factor = np.sqrt(2/self.length)
|
||
|
|
|
||
|
|
|
||
|
|
|