from pathlib import Path

import gprMax
from toolboxes.GPRAntennaModels.MALA import antenna_like_MALA_1200

# File path for output
fn = Path(__file__)

# Discretisation
dl = 0.001

# Domain
x = 0.264
y = 0.189
z = 0.220

scene = gprMax.Scene()

title = gprMax.Title(name=fn.with_suffix('').name)
domain = gprMax.Domain(p1=(x, y, z))
dxdydz = gprMax.Discretisation(p1=(dl, dl, dl))
time_window = gprMax.TimeWindow(time=6e-9)

scene.add(title)
scene.add(domain)
scene.add(dxdydz)
scene.add(time_window)

# Import antenna model and add to model
ant_pos = (0.132, 0.095, 0.100)
mala_objects = antenna_like_MALA_1200(ant_pos[0], ant_pos[1], ant_pos[2], 
                                      resolution=dl)
for obj in mala_objects:
    scene.add(obj)

# Run model
gprMax.run(scenes=[scene], geometry_only=False, outputfile=fn, gpu=None)