summaryrefslogtreecommitdiff
path: root/coil_parasitics.py
diff options
context:
space:
mode:
authorjaseg <git@jaseg.de>2023-09-29 10:37:40 +0200
committerjaseg <git@jaseg.de>2023-09-29 10:37:40 +0200
commitfdaf3c7e93467634f91142668a39e1ac2a3fb725 (patch)
tree916a69e7397bf9cb0b6f0b2121dc497781163c8f /coil_parasitics.py
parent6e12adb07ef6d40303987a92a4e081619ef4b01d (diff)
downloadgerbonara-fdaf3c7e93467634f91142668a39e1ac2a3fb725.tar.gz
gerbonara-fdaf3c7e93467634f91142668a39e1ac2a3fb725.tar.bz2
gerbonara-fdaf3c7e93467634f91142668a39e1ac2a3fb725.zip
Add sim files
Diffstat (limited to 'coil_parasitics.py')
-rw-r--r--coil_parasitics.py149
1 files changed, 149 insertions, 0 deletions
diff --git a/coil_parasitics.py b/coil_parasitics.py
new file mode 100644
index 0000000..b57e1f2
--- /dev/null
+++ b/coil_parasitics.py
@@ -0,0 +1,149 @@
+#!/usr/bin/env python3
+
+from pathlib import Path
+import multiprocessing
+import re
+import tempfile
+import subprocess
+import fnmatch
+import shutil
+import numpy as np
+
+from pyelmer import elmer
+import click
+from scipy import constants
+
+def enumerate_mesh_bodies(msh_file):
+ with open(msh_file, 'r') as f:
+ for line in f:
+ if line.startswith('$PhysicalNames'):
+ break
+ else:
+ raise ValueError('No physcial bodies found in mesh file.')
+
+ _num_names = next(f)
+
+ for line in f:
+ if line.startswith('$EndPhysicalNames'):
+ break
+
+ dim, _, line = line.strip().partition(' ')
+ tag, _, name = line.partition(' ')
+ yield name.strip().strip('"'), (int(dim), int(tag))
+
+INPUT_EXT_MAP = {
+ '.grd': 1,
+ '.mesh*': 2,
+ '.ep': 3,
+ '.ansys': 4,
+ '.inp': 5,
+ '.fil': 6,
+ '.FDNEUT': 7,
+ '.unv': 8,
+ '.mphtxt': 9,
+ '.dat': 10,
+ '.node': 11,
+ '.ele': 11,
+ '.mesh': 12,
+ '.msh': 14,
+ '.ep.i': 15,
+ '.2dm': 16}
+
+OUTPUT_EXT_MAP = {
+ '.grd': 1,
+ '.mesh*': 2,
+ '.ep': 3,
+ '.msh': 4,
+ '.vtu': 5}
+
+def elmer_grid(infile, outfile=None, intype=None, outtype=None, cwd=None, **kwargs):
+ infile = Path(infile)
+ if outfile is not None:
+ outfile = Path(outfile)
+
+ if intype is None:
+ intype = str(INPUT_EXT_MAP[infile.suffix])
+
+ if outtype is None:
+ if outfile is not None and outfile.suffix:
+ outtype = str(OUTPUT_EXT_MAP[outfile.suffix])
+ else:
+ outtype = '2'
+
+ if outfile is not None:
+ kwargs['out'] = str(outfile)
+
+ args = ['ElmerGrid', intype, outtype, infile]
+ for key, value in kwargs.items():
+ args.append(f'-{key}')
+ if isinstance(value, (tuple, list)):
+ args.extend(str(v) for v in value)
+ else:
+ args.append(str(value))
+ subprocess.run(args, cwd=cwd)
+
+def elmer_solver(cwd):
+ subprocess.run(['ElmerSolver'], cwd=cwd)
+
+
+@click.command()
+@click.option('-d', '--sim-dir', type=click.Path(dir_okay=True, file_okay=False, path_type=Path))
+@click.argument('mesh_file', type=click.Path(dir_okay=False, path_type=Path))
+def run_simulation(mesh_file, sim_dir):
+ physical = dict(enumerate_mesh_bodies(mesh_file))
+ if sim_dir is not None:
+ sim_dir = Path(sim_dir)
+ sim_dir.mkdir(exist_ok=True)
+
+ sim = elmer.load_simulation('3D_steady', 'coil_parasitics_sim.yml')
+ mesh_dir = '.'
+ mesh_fn = 'mesh'
+ sim.header['Mesh DB'] = f'"{mesh_dir}" "{mesh_fn}"'
+ sim.constants.update({
+ 'Permittivity of Vacuum': str(constants.epsilon_0),
+ 'Gravity(4)': f'0 -1 0 {constants.g}',
+ 'Boltzmann Constant': str(constants.Boltzmann),
+ 'Unit Charge': str(constants.elementary_charge)})
+
+ air = elmer.load_material('air', sim, 'coil_parasitics_materials.yml')
+ ro4003c = elmer.load_material('ro4003c', sim, 'coil_parasitics_materials.yml')
+
+ solver_electrostatic = elmer.load_solver('Electrostatics_Capacitance', sim, 'coil_parasitics_solvers.yml')
+ solver_electrostatic.data['Potential Difference'] = '1.0'
+ eqn = elmer.Equation(sim, 'main', [solver_electrostatic])
+
+ bdy_sub = elmer.Body(sim, 'substrate', [physical['substrate'][1]])
+ bdy_sub.material = ro4003c
+ bdy_sub.equation = eqn
+
+ bdy_ab = elmer.Body(sim, 'airbox', [physical['airbox'][1]])
+ bdy_ab.material = air
+ bdy_ab.equation = eqn
+
+ # boundaries
+ for name, identity in physical.items():
+ if (m := re.fullmatch(r'trace([0-9]+)', name)):
+ num = int(m.group(1))
+
+ bndry_m2 = elmer.Boundary(sim, name, [identity[1]])
+ bndry_m2.data['Capacitance Body'] = str(num)
+
+ boundary_airbox = elmer.Boundary(sim, 'FarField', [physical['airbox_surface'][1]])
+ boundary_airbox.data['Electric Infinity BC'] = 'True'
+
+ with tempfile.TemporaryDirectory() as tmpdir:
+ if sim_dir:
+ tmpdir = str(sim_dir)
+
+ sim.write_startinfo(tmpdir)
+ sim.write_sif(tmpdir)
+ # Convert mesh from gmsh to elemer formats. Also scale it from 1 unit = 1 mm to 1 unit = 1 m (SI units)
+ elmer_grid(mesh_file.name, 'mesh', cwd=tmpdir, scale=[1e-3, 1e-3, 1e-3])
+ elmer_solver(tmpdir)
+
+ capacitance_matrix = np.loadtxt(tmpdir / 'capacitance.txt')
+
+
+
+if __name__ == '__main__':
+ run_simulation()