#!/usr/bin/env python3
r"""
Simulation of active droplets in 2D
===================================

This example shows how to simulate droplets with a simple linear reaction. Note that the
simulation requires the optional `numba-scipy` package!
"""

import numba_scipy  # raises an error if the module is not available

from droplets import SphericalDroplet
from pde import ScalarField, UnitGrid

import emulsim

# set up state
grid = UnitGrid([32, 32], periodic=True)
background = emulsim.ScalarFieldElement.from_field(ScalarField(grid, 0.005))
droplet_data = [SphericalDroplet(grid.get_random_point(), 0.1) for _ in range(10)]
droplets = emulsim.SphericalDropletsElement.from_droplets(droplet_data)
state = emulsim.State({"background": background, "droplets": droplets})

# set up simulation
reaction_flux = "0.001 - 0.01 * c"
simulation = emulsim.Simulation(state)
simulation.add_actor(
    "background", emulsim.ReactionDiffusionActor({"reaction_flux": reaction_flux})
)
droplet_actor = emulsim.SphericalDropletActor(
    {"mean_reaction_inside": -0.01, "reaction_outside": reaction_flux}
)
simulation.add_actor(("droplets", "background"), droplet_actor)

# run simulation
result = simulation.run(t_range=10)

result.plot()