from pathlib import Path

import gprMax

fn = Path(__file__)

title = gprMax.Title(name=fn.with_suffix('').name)
#title: antenna_wire_dipole_fs

domain = gprMax.Domain(p1=(0.050, 0.050, 0.200))
#domain: 0.050 0.050 0.200

dxdydz = gprMax.Discretisation(p1=(0.001, 0.001, 0.001))
#dx_dy_dz: 0.001 0.001 0.001

time_window = gprMax.TimeWindow(time=10e-9)
#time_window: 10e-9

waveform = gprMax.Waveform(wave_type='gaussian', amp=1, freq=1e9, id='mypulse')
#waveform: gaussian 1 1e9 mypulse

transmission_line = gprMax.TransmissionLine(polarisation='z',
                                            p1=(0.025, 0.025, 0.100),
                                            resistance=73,
                                            waveform_id='mypulse')
#transmission_line: z 0.025 0.025 0.100 73 mypulse

## 150mm length wire
e1 = gprMax.Edge(p1=(0.025, 0.025, 0.025),
                 p2=(0.025, 0.025, 0.175),
                 material_id='pec')
#edge: 0.025 0.025 0.025 0.025 0.025 0.175 pec

## 1mm gap at centre of dipole
e2 = gprMax.Edge(p1=(0.025, 0.025, 0.100),
                 p2=(0.025, 0.025, 0.101),
                 material_id='free_space')
#edge: 0.025 0.025 0.100 0.025 0.025 0.101 free_space

gv = gprMax.GeometryView(p1=(0.020, 0.020, 0.020),
                         p2=(0.030, 0.030, 0.180),
                         dl=(0.001, 0.001, 0.001),
                         filename=fn.with_suffix('').name,
                         output_type='n')
#geometry_view: 0.020 0.020 0.020 0.030 0.030 0.180 0.001 0.001 0.001 antenna_wire_dipole_fs f

# Create a scene
scene = gprMax.Scene()

# Add the simulation objects to the scene
scene.add(title)
scene.add(domain)
scene.add(dxdydz)
scene.add(time_window)
scene.add(waveform)
scene.add(transmission_line)
scene.add(e1)
scene.add(e2)
scene.add(gv)

# Run the simulation
gprMax.run(scenes=[scene], n=1, outputfile=fn)