From 80316be63b67d27453a54c030019d84c8e253073 Mon Sep 17 00:00:00 2001 From: John Hartley Date: Thu, 1 Aug 2019 09:03:35 +0100 Subject: [PATCH] rename --- gprMax/{solver.py => solvers.py} | 20 ++++++++------------ 1 file changed, 8 insertions(+), 12 deletions(-) rename gprMax/{solver.py => solvers.py} (86%) diff --git a/gprMax/solver.py b/gprMax/solvers.py similarity index 86% rename from gprMax/solver.py rename to gprMax/solvers.py index 6be7f1d0..0bc7afa5 100644 --- a/gprMax/solver.py +++ b/gprMax/solvers.py @@ -16,26 +16,22 @@ # You should have received a copy of the GNU General Public License # along with gprMax. If not, see . from gprMax.updates import CPUUpdates -from gprMax.updates import SubGridsUpdates from gprMax.updates import GPUUpdates from gprMax.utilities import timer def create_solver(sim_config): """Returns the configured solver.""" - if sim_config.cpu: - from gprMax.Grid import FDTDGrid - G = FDTDGrid() - updates = CPUUpdates(G) - - elif sim_config.gpu: - from gprMax.Grid import GPUGrid + if sim_config.gpu: + from .grid import GPUGrid G = GPUGrid() updates = GPUUpdates(G) else: - raise NotImplementedError + from .grid import FDTDGrid + G = FDTDGrid() + updates = CPUUpdates(G) - solver = Solver(updates, iterator) + solver = Solver(updates) return solver @@ -44,7 +40,7 @@ class Solver: """Generic solver for Update objects""" - def __init__(self, updates, iterator): + def __init__(self, updates): """Context for the model to run in. Sub-class this with contexts i.e. an MPI context. @@ -53,7 +49,7 @@ class Solver: iterator (iterator): can be range() or tqdm() """ self.updates = updates - self.iterator = iterator + #self.iterator = iterator def solve(self): """Time step the FDTD model."""