from interacticle import MoleculeCollection, LennardJones, Coulomb, Simulation

import interacticle.visualizer
from interacticle.visual import WireBox, MolecularLinks

from library import Benzene, Argon

setup = MoleculeCollection()

setup.potential(LennardJones(12.011, 0.3550, 0.070))
setup.potential(LennardJones( 1.008, 0.2420, 0.030))
setup.potential(LennardJones(39.948, 0.3395, 0.971))

setup.potential(Coulomb(12.011, -0.115))
setup.potential(Coulomb( 1.008,  0.115))

for y in range(6):
    for z in range(6):
        setup.add(Benzene, (4.4,0.2+0.8*y,0.2+0.8*z))

for x in range(10):
    for y in range(12):
        for z in range(12):
            setup.add(Argon, (0.2+0.4*x,0.2+0.4*y,0.2+0.4*z))

setup.domain_size = 5
setup.tau = 0.0005
setup.cutoff = 0.3395*2.5
setup.target_temperature = 300

setup.max_coulomb = 1000
setup.max_lennard_jones = 1000

simulation = Simulation(setup, opengl = True)

interacticle.visualizer.simulate(setup, simulation, [ WireBox(setup.domain_size), MolecularLinks(simulation) ], 100)