你已经派生过 gprMax
镜像自地址
https://gitee.com/sunhf/gprMax.git
已同步 2025-08-08 07:24:19 +08:00
123 行
3.7 KiB
Python
123 行
3.7 KiB
Python
# Copyright (C) 2015-2019: The University of Edinburgh
|
|
# Authors: Craig Warren and Antonis Giannopoulos
|
|
#
|
|
# This file is part of gprMax.
|
|
#
|
|
# gprMax is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# gprMax is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with gprMax. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import gprMax.config as config
|
|
from .grid import FDTDGrid
|
|
from .grid import CUDAGrid
|
|
from .subgrids.updates import create_updates as create_subgrid_updates
|
|
from .updates import CPUUpdates
|
|
from .updates import CUDAUpdates
|
|
|
|
|
|
def create_G(sim_config):
|
|
"""Create grid object according to solver.
|
|
|
|
Args:
|
|
sim_config (SimConfig): Simulation level configuration object.
|
|
|
|
Returns:
|
|
G (FDTDGrid): Holds essential parameters describing the model.
|
|
"""
|
|
|
|
if sim_config.general['cuda']:
|
|
G = CUDAGrid()
|
|
elif sim_config.subgrid:
|
|
G = FDTDGrid()
|
|
else:
|
|
G = FDTDGrid()
|
|
|
|
return G
|
|
|
|
|
|
def create_solver(G, sim_config):
|
|
"""Create configured solver object.
|
|
|
|
Args:
|
|
G (FDTDGrid): Holds essential parameters describing the model.
|
|
sim_config (SimConfig): simulation level configuration object.
|
|
|
|
Returns:
|
|
solver (Solver): solver object.
|
|
"""
|
|
|
|
if sim_config.gpu:
|
|
updates = CUDAUpdates(G)
|
|
solver = Solver(updates)
|
|
elif sim_config.subgrid:
|
|
updates = create_subgrid_updates(G)
|
|
solver = Solver(updates, hsg=True)
|
|
# A large range of different functions exist to advance the time step for
|
|
# dispersive materials. The correct function is set here based on the
|
|
# the required numerical precision and dispersive material type.
|
|
props = updates.adapt_dispersive_config(config)
|
|
updates.set_dispersive_updates(props)
|
|
else:
|
|
updates = CPUUpdates(G)
|
|
solver = Solver(updates)
|
|
props = updates.adapt_dispersive_config(config)
|
|
updates.set_dispersive_updates(props)
|
|
|
|
return solver
|
|
|
|
|
|
class Solver:
|
|
"""Generic solver for Update objects"""
|
|
|
|
def __init__(self, updates, hsg=False):
|
|
"""
|
|
Args:
|
|
updates (Updates): Updates contains methods to run FDTD algorithm.
|
|
hsg (bool): Use sub-gridding.
|
|
"""
|
|
|
|
self.updates = updates
|
|
self.hsg = hsg
|
|
|
|
def get_G(self):
|
|
return self.updates.G
|
|
|
|
def solve(self, iterator):
|
|
"""Time step the FDTD model.
|
|
|
|
Args:
|
|
iterator (iterator): can be range() or tqdm()
|
|
"""
|
|
|
|
self.updates.time_start()
|
|
|
|
for iteration in iterator:
|
|
self.updates.store_outputs()
|
|
self.updates.store_snapshots(iteration)
|
|
self.updates.update_magnetic()
|
|
self.updates.update_magnetic_pml()
|
|
self.updates.update_magnetic_sources()
|
|
if self.hsg:
|
|
self.updates.hsg_2()
|
|
self.updates.update_electric_a()
|
|
self.updates.update_electric_pml()
|
|
self.updates.update_electric_sources()
|
|
if self.hsg:
|
|
self.updates.hsg_1()
|
|
self.updates.update_electric_b()
|
|
|
|
self.updates.finalise()
|
|
tsolve = self.updates.calculate_tsolve()
|
|
self.updates.cleanup()
|
|
|
|
return tsolve
|