Logfile

tests/test_001_pec_cubic_cavity.py

@@ -99,7 +99,7 @@ def test_simulation(self):
         skip_cells = 12  # no. cells to skip in WP integration
         wake = WakeSolver(q=q, sigmaz=sigmaz, beta=beta, 
                     xsource=xs, ysource=ys, xtest=xt, ytest=yt,
-                    save=False, logfile=False, Ez_file='tests/001_Ez.h5',
+                    save=False, Ez_file='tests/001_Ez.h5',
                     skip_cells=skip_cells,
                     )
 

tests/test_007_mpi_lossy_cavity.py

@@ -82,6 +82,18 @@ class TestMPILossyCavity:
                     -6.04105997e+01 ,-3.06532160e+01 ,-1.17749936e+01 ,-3.12574866e+00,
                     -7.35339521e-01 ,-1.13085658e-01 , 7.18247535e-01 , 8.73829036e-02])
 
+    gridLogs = {'Nx': 60, 'Ny': 60, 'Nz': 140, 'dx': 0.008666666348775227, 'dy': 0.008666666348775227,
+                 'dz': 0.005714285799435207, 'stl_solids': ['tests/stl/007_vacuum_cavity.stl', 'tests/stl/007_lossymetal_shell.stl'],
+                 'stl_materials': ['vacuum', [30, 1.0, 30]], 'gridInitializationTime': 0}
+
+    solverLogs = {'use_gpu': False, 'use_mpi': False, 'bc_low': ['pec', 'pec', 'pec'],
+                   'bc_high': ['pec', 'pec', 'pec'], 'n_pml': 10, 'bg': 'pec',
+                   'dt': 6.970326728398968e-12, 'solverInitializationTime': 0}
+
+    wakeSolverLogs = {'ti': 2.8516132094735135e-09, 'q': 1e-09, 'sigmaz': 0.1, 'beta': 1.0,
+                       'xsource': 0.0, 'ysource': 0.0, 'xtest': 0.0, 'ytest': 0.0, 'chargedist': None,
+                       'skip_cells': 10, 'results_folder': 'tests/007_results/', 'wakelength': 10.0, 'simulationTime': 0}
+
     img_folder = 'tests/007_img/'
 
     def test_mpi_import(self):
@@ -314,4 +326,14 @@ def test_long_impedance(self):
             assert np.allclose(np.real(wake.Z)[::20], np.real(self.Z), rtol=0.1), "Real Impedance samples failed"
             assert np.allclose(np.imag(wake.Z)[::20], np.imag(self.Z), rtol=0.1), "Imag Impedance samples failed"
             assert np.cumsum(np.abs(wake.Z))[-1] == pytest.approx(250910.51090497518, 0.1), "Abs Impedance cumsum failed"
-
+
+    def test_log_file(self):
+        global solver
+        solver.logger.grid["gridInitializationTime"] = 0 #times can vary
+        solver.logger.solver["solverInitializationTime"] = 0
+        solver.logger.wakeSolver["simulationTime"] = 0
+        logfile = os.path.join(solver.logger.wakeSolver["results_folder"], "Simulation_Parameters.log")
+        assert os.path.exists(logfile), "Log file not created"
+        assert solver.logger.grid == self.gridLogs, "Grid logs do not match expected values"
+        assert solver.logger.solver == self.solverLogs, "Solver logs do not match expected values"
+        assert solver.logger.wakeSolver == self.wakeSolverLogs, "WakeSolver logs do not match expected values"

wakis/__init__.py

@@ -10,12 +10,14 @@
 from . import materials
 from . import wakeSolver
 from . import geometry
+from . import logger
 from . import field_monitors
 
 from .field_monitors import FieldMonitor
 from .field import Field
 from .gridFIT3D import GridFIT3D
 from .solverFIT3D import SolverFIT3D
 from .wakeSolver import WakeSolver
+from .logger import Logger
 
 from ._version import __version__

wakis/gridFIT3D.py

@@ -7,8 +7,10 @@
 import pyvista as pv
 from functools import partial
 from scipy.optimize import least_squares
+import time
 
 from .field import Field
+from .logger import Logger
 
 try:
     from mpi4py import MPI
@@ -63,6 +65,7 @@ def __init__(self, xmin, xmax, ymin, ymax, zmin, zmax,
                 stl_rotate=[0., 0., 0.], stl_translate=[0., 0., 0.], stl_scale=1.0,
                 stl_colors=None, verbose=1, stl_tol=1e-3):
 
+        t0 = time.time()
         if verbose: print('Generating grid...')
         self.verbose = verbose
         self.use_mpi = use_mpi
@@ -143,6 +146,24 @@ def __init__(self, xmin, xmax, ymin, ymax, zmin, zmax,
             if stl_colors is None:
                 self.assign_colors()
 
+        # Forward Parameters to logger
+        self.logger = Logger()
+        self.logger.grid["Nx"] = self.Nx
+        self.logger.grid["Ny"] = self.Ny
+        self.logger.grid["Nz"] = self.Nz
+        self.logger.grid["dx"] = self.dx
+        self.logger.grid["dy"] = self.dy
+        self.logger.grid["dz"] = self.dz
+        self.logger.grid["stl_solids"] = list(self.stl_solids.values()) if self.stl_solids is not None else []
+        self.logger.grid["stl_materials"] = list(self.stl_materials.values()) if self.stl_materials is not None else []
+        if stl_rotate != [0., 0., 0.]:
+            self.logger.grid["stl_rotate"] = self.stl_rotate
+        if stl_translate != [0., 0., 0.]:
+            self.logger.grid["stl_translate"] = self.stl_translate
+        if stl_scale != 1.0:
+            self.logger.grid["stl_scale"] = self.stl_scale
+        self.logger.grid["gridInitializationTime"] = time.time()-t0            
+
     def compute_grid(self):
         X, Y, Z = np.meshgrid(self.x, self.y, self.z, indexing='ij')
         self.grid = pv.StructuredGrid(X.transpose(), Y.transpose(), Z.transpose())
@@ -838,4 +859,4 @@ def clip(widget):
         if offscreen:
             pl.export_html('grid_inspect.html')
         else:
-            pl.show()
+            pl.show()

wakis/logger.py

@@ -0,0 +1,67 @@
+# copyright ################################# #
+# This file is part of the wakis Package.     #
+# Copyright (c) CERN, 2025.                   #
+# ########################################### #
+
+from tqdm import tqdm
+
+import numpy as np
+import time
+import h5py
+import os
+import json
+
+from scipy.constants import c as c_light, epsilon_0 as eps_0, mu_0 as mu_0
+from scipy.sparse import csc_matrix as sparse_mat
+from scipy.sparse import diags, hstack, vstack
+
+
+
+class Logger():
+
+    def __init__(self):
+        self.grid = {}
+        self.solver = {}
+        self.wakeSolver = {}
+
+    def save_logs(self):
+        """
+        Save all logs (grid, solver, wakeSolver) into log-file inside the results folder.
+        """
+        logfile = os.path.join(self.wakeSolver["results_folder"], "Simulation_Parameters.log")
+
+        # Write sections
+        if not os.path.exists(self.wakeSolver["results_folder"]): 
+            os.mkdir(self.wakeSolver["results_folder"])
+
+        with open(logfile, "w", encoding="utf-8") as fh:
+            fh.write("Simulation Parameters\n")
+            fh.write("""=====================\n\n""")
+
+            sections = [
+                ("WakeSolver Logs", self.wakeSolver),
+                ("Solver Logs", self.solver),
+                ("Grid Logs", self.grid),
+            ]
+
+            for title, data in sections:
+                fh.write(f"\n## {title} ##\n")
+                if not data:
+                    fh.write("(empty)\n")
+                    continue
+
+                # convert non-serializable values to strings recursively
+                def _convert(obj):
+                    if isinstance(obj, dict):
+                        return {k: _convert(v) for k, v in obj.items()}
+                    if isinstance(obj, (list, tuple)):
+                        return [_convert(v) for v in obj]
+                    try:
+                        json.dumps(obj)
+                        return obj
+                    except Exception:
+                        return str(obj)
+
+                clean = _convert(data)
+                fh.write(json.dumps(clean, indent=2, ensure_ascii=False))
+                fh.write("\n")

wakis/routines.py

@@ -5,6 +5,7 @@
 
 import numpy as np
 import h5py
+import time
 from tqdm import tqdm
 from scipy.constants import c as c_light
 from wakis.sources import Beam
@@ -297,6 +298,7 @@ def save_to_h5(self, hf, field, x, y, z, comp, n):
         if plot_from is None: plot_from = int(self.ti/self.dt)
 
         print('Running electromagnetic time-domain simulation...')
+        t0 = time.time()
         for n in tqdm(range(Nt)):
 
             # Initial condition
@@ -340,4 +342,9 @@ def save_to_h5(self, hf, field, x, y, z, comp, n):
 
             # Compute wakefield magnitudes is done inside WakeSolver
             self.wake.solve(compute_plane=compute_plane)
-
+
+        # Forward parameters to logger
+        self.logger.wakeSolver=self.wake.logger.wakeSolver
+        self.logger.wakeSolver["wakelength"]=wakelength            
+        self.logger.wakeSolver["simulationTime"]=time.time()-t0
+        self.logger.save_logs()

wakis/solverFIT3D.py

@@ -17,6 +17,7 @@
 from .materials import material_lib
 from .plotting import PlotMixin
 from .routines import RoutinesMixin
+from .logger import Logger
 
 try:
     from cupyx.scipy.sparse import csc_matrix as gpu_sparse_mat
@@ -93,7 +94,7 @@ def __init__(self, grid, wake=None, cfln=0.5, dt=None,
         '''
 
         self.verbose = verbose
-        if verbose:  t0 = time.time()
+        t0 = time.time()
 
         # Flags
         self.step_0 = True
@@ -113,7 +114,7 @@ def __init__(self, grid, wake=None, cfln=0.5, dt=None,
 
         # Grid 
         self.grid = grid
-
+        bgLog = bg
         self.Nx = self.grid.Nx
         self.Ny = self.grid.Ny
         self.Nz = self.grid.Nz
@@ -252,6 +253,18 @@ def __init__(self, grid, wake=None, cfln=0.5, dt=None,
 
         if verbose:  print(f'Total initialization time: {time.time() - t0} s')
 
+        # assign logs
+        self.logger = Logger()
+        self.logger.grid = self.grid.logger.grid
+        self.logger.solver["use_gpu"] = use_gpu
+        self.logger.solver["use_mpi"] = use_mpi
+        self.logger.solver["bc_low"] = bc_low
+        self.logger.solver["bc_high"] = bc_high
+        self.logger.solver["n_pml"] = n_pml
+        self.logger.solver["bg"] = bgLog
+        self.logger.solver["dt"] = self.dt
+        self.logger.solver["solverInitializationTime"] = time.time() - t0
+
     def update_tensors(self, tensor='all'):
         '''Update tensor matrices after 
         Field ieps, imu or sigma have been modified 
@@ -1107,4 +1120,4 @@ def reset_fields(self):
         for d in ['x', 'y', 'z']:
             self.E[:, :, :, d] = 0.0
             self.H[:, :, :, d] = 0.0
-            self.J[:, :, :, d] = 0.0
+            self.J[:, :, :, d] = 0.0

wakis/wakeSolver.py

@@ -12,6 +12,8 @@
 from tqdm import tqdm
 from scipy.constants import c as c_light
 
+from .logger import Logger
+
 class WakeSolver():
     ''' Class for wake potential and impedance
     calculation from 3D time domain E fields
@@ -22,7 +24,7 @@ def __init__(self, wakelength=None, q=1e-9, sigmaz=1e-3, beta=1.0,
                  chargedist=None, ti=None, 
                  compute_plane='both', skip_cells=0, add_space=None, 
                  Ez_file='Ez.h5', save=True, results_folder='results/',
-                 verbose=0, logfile=False, counter_moving=False):
+                 verbose=0, counter_moving=False):
         '''
         Parameters
         ----------
@@ -61,9 +63,6 @@ def __init__(self, wakelength=None, q=1e-9, sigmaz=1e-3, beta=1.0,
             - Charge distribution: lambda.txt, spectrum.txt
         verbose: bool, default 0
             Controls the level of verbose in the terminal output
-        logfile: bool, default False
-            Creates a `wake.log` file with the summary of the input parameters
-            and calculations performed
         counter_moving: bool, default False
             If the test charge is moving in the same or opposite direction to the source
 
@@ -161,8 +160,8 @@ def __init__(self, wakelength=None, q=1e-9, sigmaz=1e-3, beta=1.0,
 
         #user
         self.verbose = verbose
+        self.logger = Logger()
         self.save = save
-        self.logfile = logfile
         self.folder = results_folder
 
         if self.save:
@@ -173,6 +172,19 @@ def __init__(self, wakelength=None, q=1e-9, sigmaz=1e-3, beta=1.0,
         if self.log:
             self.params_to_log()
 
+        # Forward parameters to logger
+        self.logger.wakeSolver["ti"]=self.ti
+        self.logger.wakeSolver["q"]=self.q
+        self.logger.wakeSolver["sigmaz"]=self.sigmaz
+        self.logger.wakeSolver["beta"]=self.beta
+        self.logger.wakeSolver["xsource"]=self.xsource
+        self.logger.wakeSolver["ysource"]=self.ysource
+        self.logger.wakeSolver["xtest"]=self.xtest
+        self.logger.wakeSolver["ytest"]=self.ytest
+        self.logger.wakeSolver["chargedist"]=self.chargedist
+        self.logger.wakeSolver["skip_cells"]=self.skip_cells
+        self.logger.wakeSolver["results_folder"]=self.folder
+
     def solve(self, compute_plane=None, **kwargs):
         '''
         Perform the wake potential and impedance for
@@ -1114,14 +1126,6 @@ def log(self, txt):
 
         if self.verbose:
             print('\x1b[2;37m'+txt+'\x1b[0m')
-
-        if not self.logfile:
-            return
-
-        title = 'wake'
-        f = open(title + '.log', "a")
-        f.write(txt + '\r\n')
-        f.close()
 
     def params_to_log(self):
         self.log(time.asctime())