From e3bfcda35c755bee88cb4b1d54a0a32691d78bf7 Mon Sep 17 00:00:00 2001
From: vsitaraman <vs19g@fsu.edu>
Date: Fri, 24 Apr 2026 14:34:50 -0400
Subject: [PATCH] 	modified:   anasen_fem/garfield_sim.py changed mesh
 input for garfield

---
 anasen_fem/garfield_sim.py | 38 ++++++++++++++++++++++++++++++++------
 anasen_fem/run.py          |  3 ++-
 2 files changed, 34 insertions(+), 7 deletions(-)

diff --git a/anasen_fem/garfield_sim.py b/anasen_fem/garfield_sim.py
index 2a992d1..839f98e 100644
--- a/anasen_fem/garfield_sim.py
+++ b/anasen_fem/garfield_sim.py
@@ -4,7 +4,7 @@ import sys
 
 # 1. FIX: Manually load the Garfield library if it's not in the ROOT namespace
 # Update this path to your actual installation location
-garfield_lib_path = "/home/vsitaraman/garfieldpp/install/lib/libGarfield.so"
+garfield_lib_path = "/home/vs19g/garfieldpp/install/lib/libGarfield.so"
 
 if os.path.exists(garfield_lib_path):
     ROOT.gSystem.Load(garfield_lib_path)
@@ -44,12 +44,13 @@ else:
 # --- 3. FIELD MAP SETUP ---
 fm = ROOT.Garfield.ComponentElmer()
 
-# Update these filenames to match your Elmer SIF output exactly
-# Assuming ElmerGrid was run on 'wires2d' directory
-fm.Initialise("wires2d/mesh.nodes", 
+
+fm.Initialise("wires2d/mesh.header", 
               "wires2d/mesh.elements", 
-              "wires2d/mesh.boundary", 
-              "wires2d/elfield_anasen.result", "mm")
+              "wires2d/mesh.nodes", 
+              "wires2d/dielectrics.dat", # Dielectrics (leave as empty string)
+              "wires2d/elstatics.result", 
+              "mm")
 
 # Set the medium (Body 13 from your Gmsh script)
 fm.SetMedium(0, gas)
@@ -73,8 +74,33 @@ x0, y0, z0, t0 = 35.0, 0.0, 0.0, 0.0
 print(f"Simulating heavy ion drift from r={x0}...")
 drift.DriftIon(x0, y0, z0, t0)
 
+# Create a file to store the heavy ion track
+with open("heavy_ion_track.csv", "w") as f:
+    f.write("x,y,z,t\n")
+    
+    # After running drift.DriftIon(x0, y0, z0, t0):
+    n_points = drift.GetNumberOfDriftLinePoints()
+    for i in range(n_points):
+        xi, yi, zi, ti = ROOT.double(0), ROOT.double(0), ROOT.double(0), ROOT.double(0)
+        drift.GetDriftLinePoint(i, xi, yi, zi, ti)
+        f.write(f"{xi},{yi},{zi},{ti}\n")
+
 print(f"Simulating electron avalanche from r={x0}...")
 # AvalancheElectron(x, y, z, t, energy, dx, dy, dz)
 aval.AvalancheElectron(x0, y0, z0, t0, 0.1, 0.0, 0.0, 0.0)
 
+with open("avalanche_endpoints.csv", "w") as f:
+    f.write("x,y,z,t\n")
+    
+    # After aval.AvalancheElectron(...)
+    n_endpoints = aval.GetNumberOfEndpoints()
+    for i in range(n_endpoints):
+        # Get start and end points of each electron in the avalanche
+        x1, y1, z1, t1, e1 = ROOT.double(0), ROOT.double(0), ROOT.double(0), ROOT.double(0), ROOT.double(0)
+        x2, y2, z2, t2, e2, status = ROOT.double(0), ROOT.double(0), ROOT.double(0), ROOT.double(0), ROOT.double(0), ROOT.int(0)
+        
+        aval.GetEndpoint(i, x1, y1, z1, t1, e1, x2, y2, z2, t2, e2, status)
+        # We save the endpoint (x2, y2, z2) where the electron was collected or attached
+        f.write(f"{x2},{y2},{z2},{t2}\n")
+
 print("Simulation complete.")
\ No newline at end of file
diff --git a/anasen_fem/run.py b/anasen_fem/run.py
index 397de45..f26f85f 100755
--- a/anasen_fem/run.py
+++ b/anasen_fem/run.py
@@ -7,9 +7,10 @@ count=11
 while val<178.3+0.1:
     print(val)
     os.system("python3 wires_gmsh2d_bc.py "+str(val))
-    os.system("ElmerGrid 14 2 wires2d.msh")
+    os.system("ElmerGrid 14 2 wires2d.msh -2d")
     os.system("ElmerSolver wires2d.sif")
     os.system("./paraview_plotter.py")
+    # os.system("python3 garfield_sim.py")
     os.system("cp wires2d.msh wires2d/mesh_files/wires2d%02d_%1.4f.msh"%(count,val))
     os.system("cp wires2d.sif wires2d/sif_files/wires2d_%02d_%1.4f.sif"%(count,val))
     os.system("cp wires2d/elfield_anasen_t0001.vtu wires2d/vtu_files/elfield_anasen_%02d_%1.4f.vtu"%(count,val))